From 8143f86be6e95731b319e3bd97b5c97e58e137bd Mon Sep 17 00:00:00 2001 From: Dev Khant Date: Fri, 28 Feb 2025 15:05:26 +0530 Subject: [PATCH] Fix proxy pytests (#2272) --- tests/test_proxy.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/tests/test_proxy.py b/tests/test_proxy.py index a41be267..e3207a38 100644 --- a/tests/test_proxy.py +++ b/tests/test_proxy.py @@ -9,7 +9,9 @@ from mem0.proxy.main import Chat, Completions, Mem0 @pytest.fixture def mock_memory_client(): - return Mock(spec=MemoryClient) + mock_client = Mock(spec=MemoryClient) + mock_client.user_email = None + return mock_client @pytest.fixture @@ -65,6 +67,7 @@ def test_completions_create(mock_memory_client, mock_litellm): messages = [{"role": "user", "content": "Hello, how are you?"}] mock_memory_client.search.return_value = [{"memory": "Some relevant memory"}] mock_litellm.completion.return_value = {"choices": [{"message": {"content": "I'm doing well, thank you!"}}]} + mock_litellm.supports_function_calling.return_value = True response = completions.create(model="gpt-4o-mini", messages=messages, user_id="test_user", temperature=0.7) @@ -89,6 +92,7 @@ def test_completions_create_with_system_message(mock_memory_client, mock_litellm ] mock_memory_client.search.return_value = [{"memory": "Some relevant memory"}] mock_litellm.completion.return_value = {"choices": [{"message": {"content": "I'm doing well, thank you!"}}]} + mock_litellm.supports_function_calling.return_value = True completions.create(model="gpt-4o-mini", messages=messages, user_id="test_user")