Abstraction for Project in MemoryClient (#3067)

2025-07-08 11:33:20 +05:30
parent aae5989e78
commit 70d6f9231b
11 changed files with 1071 additions and 39 deletions
--- a/tests/llms/test_gemini.py
+++ b/tests/llms/test_gemini.py
@@ -72,6 +72,9 @@ def test_generate_response_with_tools(mock_gemini_client: Mock):
        }
    ]

+    # Create a proper mock for the function call arguments
+    mock_args = {"data": "Today is a sunny day."}
+    
    mock_tool_call = Mock()
    mock_tool_call.name = "add_memory"
    mock_tool_call.args = {"data": "Today is a sunny day."}