move add position

Author: yuan.wang

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"):