fix: playground chat bug (MemTensor#975)

CarltonXiang · web-flow · commit f83c5d52f7e3 · 2026-01-28T20:06:47.000+08:00
## Description  Summary: (summary) Fix: #(issue) Docs Issue/PR: (docs-issue-or-pr-link) Reviewer: @(reviewer) ## Checklist: - [ ] I have performed a self-review of my own code | 我已自行检查了自己的代码 - [ ] I have commented my code in hard-to-understand areas | 我已在难以理解的地方对代码进行了注释 - [ ] I have added tests that prove my fix is effective or that my feature works | 我已添加测试以证明我的修复有效或功能正常 - [ ] I have created related documentation issue/PR in [MemOS-Docs](https://github.com/MemTensor/MemOS-Docs) (if applicable) | 我已在 [MemOS-Docs](https://github.com/MemTensor/MemOS-Docs) 中创建了相关的文档 issue/PR（如果适用） - [ ] I have linked the issue to this PR (if applicable) | 我已将 issue 链接到此 PR（如果适用） - [ ] I have mentioned the person who will review this PR | 我已提及将审查此 PR 的人
diff --git a/src/memos/api/handlers/chat_handler.py b/src/memos/api/handlers/chat_handler.py
@@ -110,6 +110,7 @@ def handle_chat_complete(self, chat_req: APIChatCompleteRequest) -> dict[str, An
         Raises:
             HTTPException: If chat fails
         """
+        self.logger.info(f"[ChatHandler] Chat Req is: {chat_req}")
         try:
             # Resolve readable cube IDs (for search)
             readable_cube_ids = chat_req.readable_cube_ids or [chat_req.user_id]
@@ -241,6 +242,7 @@ def handle_chat_stream(self, chat_req: ChatRequest) -> StreamingResponse:
         Raises:
             HTTPException: If stream initialization fails
         """
+        self.logger.info(f"[ChatHandler] Chat Req is: {chat_req}")
         try:
 
             def generate_chat_response() -> Generator[str, None, None]:
@@ -422,6 +424,7 @@ def handle_chat_stream_playground(self, chat_req: ChatPlaygroundRequest) -> Stre
         Raises:
             HTTPException: If stream initialization fails
         """
+        self.logger.info(f"[ChatHandler] Chat Req is: {chat_req}")
         try:
 
             def generate_chat_response() -> Generator[str, None, None]:
@@ -585,6 +588,8 @@ def generate_chat_response() -> Generator[str, None, None]:
                     # get internet reference
                     internet_reference = self._get_internet_reference(
                         search_response.data.get("text_mem")[0]["memories"]
+                        if search_response.data.get("text_mem")
+                        else []
                     )
                     yield f"data: {json.dumps({'type': 'reference', 'data': reference}, ensure_ascii=False)}\n\n"
 
diff --git a/src/memos/mem_reader/multi_modal_struct.py b/src/memos/mem_reader/multi_modal_struct.py
@@ -464,13 +464,6 @@ def _get_maybe_merged_memory(
                 status="activated",
                 threshold=merge_threshold,
                 user_name=user_name,
-                filter={
-                    "or": [
-                        {"memory_type": "LongTermMemory"},
-                        {"memory_type": "UserMemory"},
-                        {"memory_type": "WorkingMemory"},
-                    ]
-                },
             )
 
             if not search_results:
diff --git a/src/memos/mem_reader/read_multi_modal/base.py b/src/memos/mem_reader/read_multi_modal/base.py
@@ -15,6 +15,7 @@
     TextualMemoryItem,
     TreeNodeTextualMemoryMetadata,
 )
+from memos.utils import timed
 
 from .utils import detect_lang, get_text_splitter
 
@@ -245,6 +246,7 @@ def parse(
         else:
             raise ValueError(f"Unknown mode: {mode}. Must be 'fast' or 'fine'")
 
+    @timed
     def _split_text(self, text: str, is_markdown: bool = False) -> list[str]:
         """
         Split text into chunks using text splitter from utils.
