diff --git a/src/memos/mem_reader/simple_struct.py b/src/memos/mem_reader/simple_struct.py index 9f5eb9832..0f74adead 100644 --- a/src/memos/mem_reader/simple_struct.py +++ b/src/memos/mem_reader/simple_struct.py @@ -224,7 +224,6 @@ def _iter_chat_windows(self, scene_data_info, max_tokens=None, overlap=200): max_tokens = max_tokens or self.chat_window_max_tokens buf, sources, start_idx = [], [], 0 cur_text = "" - for idx, item in enumerate(scene_data_info): role = item.get("role", "") content = item.get("content", "") @@ -247,7 +246,15 @@ def _iter_chat_windows(self, scene_data_info, max_tokens=None, overlap=200): cur_text = "".join(buf) buf.append(line) - sources.append({"type": "chat", "index": idx, "role": role, "chat_time": chat_time}) + sources.append( + { + "type": "chat", + "index": idx, + "role": role, + "chat_time": chat_time, + "content": content, + } + ) cur_text = "".join(buf) if buf: diff --git a/src/memos/memories/textual/simple_tree.py b/src/memos/memories/textual/simple_tree.py index 9c67db288..52bf62c6d 100644 --- a/src/memos/memories/textual/simple_tree.py +++ b/src/memos/memories/textual/simple_tree.py @@ -151,9 +151,6 @@ def search( list[TextualMemoryItem]: List of matching memories. """ if (self.internet_retriever is not None) and manual_close_internet: - logger.warning( - "Internet retriever is init by config , but this search set manual_close_internet is True and will close it" - ) searcher = Searcher( self.dispatcher_llm, self.graph_store,