diff --git a/src/memos/api/handlers/chat_handler.py b/src/memos/api/handlers/chat_handler.py index 2ab60d5de..c42157245 100644 --- a/src/memos/api/handlers/chat_handler.py +++ b/src/memos/api/handlers/chat_handler.py @@ -395,16 +395,6 @@ def generate_chat_response() -> Generator[str, None, None]: [chat_req.mem_cube_id] if chat_req.mem_cube_id else [chat_req.user_id] ) - # for playground, add the query to memory without response - self._start_add_to_memory( - user_id=chat_req.user_id, - writable_cube_ids=writable_cube_ids, - session_id=chat_req.session_id or "default_session", - query=chat_req.query, - full_response=None, - async_mode="sync", - ) - # ====== first search text mem with parse goal ====== search_req = APISearchRequest( query=chat_req.query, @@ -506,6 +496,16 @@ def generate_chat_response() -> Generator[str, None, None]: end_time = time.time() self.logger.info(f"second search time: {end_time - start_time}") + # for playground, add the query to memory without response + self._start_add_to_memory( + user_id=chat_req.user_id, + writable_cube_ids=writable_cube_ids, + session_id=chat_req.session_id or "default_session", + query=chat_req.query, + full_response=None, + async_mode="sync", + ) + # Extract memories from search results (second search) memories_list = [] if search_response.data and search_response.data.get("text_mem"):