@@ -68,6 +68,7 @@ def mock_init_server():
6868 "pref_retriever" : None ,
6969 "pref_mem" : None ,
7070 "online_bot" : None ,
71+ "chat_llms" : Mock (),
7172 }
7273
7374 with patch ("memos.api.handlers.init_server" , return_value = mock_components ):
@@ -184,6 +185,7 @@ def test_search_response_format(self, mock_handlers, client):
184185
185186 request_data = {
186187 "query" : "test query" ,
188+ "user_id" : "test_user_id" ,
187189 "mem_cube_id" : "test_cube" ,
188190 }
189191
@@ -226,18 +228,6 @@ def test_add_valid_input_output(self, mock_handlers, client):
226228 assert call_args .mem_cube_id == "test_cube"
227229 assert call_args .user_id == "test_user"
228230
229- def test_add_invalid_input_missing_cube_id (self , mock_handlers , client ):
230- """Test add endpoint with missing required field."""
231- request_data = {
232- "user_id" : "test_user" ,
233- "memory_content" : "test memory content" ,
234- }
235-
236- response = client .post ("/product/add" , json = request_data )
237-
238- # Should return validation error
239- assert response .status_code == 422
240-
241231 def test_add_response_format (self , mock_handlers , client ):
242232 """Test add endpoint returns MemoryResponse format."""
243233 mock_handlers ["add" ].handle_add_memories .return_value = MemoryResponse (
0 commit comments