Fix/no response (#463)

* fix: response error

* fix: response error

* fix: response error

---------

Co-authored-by: harvey_xiang <[email protected]>

Author: CarltonXiang

src/memos/api/middleware/request_context.py

@@ -72,10 +72,18 @@ async def dispatch(self, request: Request, call_next: Callable) -> Response:
             f"headers: {request.headers}"
         )
 
+        response = await call_next(request)
+        end_time = time.time()
+
         # Process the request
         try:
-            response = await call_next(request)
-            end_time = time.time()
+            if not response:
+                logger.error(
+                    f"Request Failed No Response, path: {request.url.path}, status: {response.status_code}, cost: {(end_time - start_time) * 1000:.2f}ms"
+                )
+
+                return response
+
             if response.status_code == 200:
                 logger.info(
                     f"Request completed: source: {self.source}, path: {request.url.path}, status: {response.status_code}, cost: {(end_time - start_time) * 1000:.2f}ms"
@@ -89,6 +97,5 @@ async def dispatch(self, request: Request, call_next: Callable) -> Response:
             logger.error(
                 f"Request Exception Error: source: {self.source}, path: {request.url.path}, error: {e}, cost: {(end_time - start_time) * 1000:.2f}ms"
             )
-            raise e
 
         return response

src/memos/api/routers/server_router.py

@@ -349,7 +349,7 @@ def search_memories(search_req: APISearchRequest):
         mem_cube_id=search_req.mem_cube_id,
         session_id=search_req.session_id or "default_session",
     )
-    logger.info(f"Search user_id is: {user_context.mem_cube_id}")
+    logger.info(f"Search Req is: {search_req}")
     memories_result: MOSSearchResult = {
         "text_mem": [],
         "act_mem": [],
@@ -502,6 +502,9 @@ def add_memories(add_req: APIADDRequest):
         mem_cube_id=add_req.mem_cube_id,
         session_id=add_req.session_id or "default_session",
     )
+
+    logger.info(f"Add Req is: {add_req}")
+
     target_session_id = add_req.session_id
     if not target_session_id:
         target_session_id = "default_session"