From 0de0bf845b4afe9fca4ed0e4bcd42e3152e09b63 Mon Sep 17 00:00:00 2001 From: fridayL Date: Tue, 28 Oct 2025 11:19:14 +0800 Subject: [PATCH 1/3] feat: add content and remove log waring --- src/memos/mem_reader/simple_struct.py | 14 +++++++------- src/memos/memories/textual/simple_tree.py | 3 --- 2 files changed, 7 insertions(+), 10 deletions(-) diff --git a/src/memos/mem_reader/simple_struct.py b/src/memos/mem_reader/simple_struct.py index 9f5eb9832..8f503df98 100644 --- a/src/memos/mem_reader/simple_struct.py +++ b/src/memos/mem_reader/simple_struct.py @@ -224,12 +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", "") - chat_time = item.get("chat_time", None) - parts = [] if role and str(role).lower() != "mix": parts.append(f"{role}: ") if chat_time: @@ -247,8 +241,14 @@ 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) + for idx, item in enumerate(scene_data_info): + role = item.get("role", "") + content = item.get("content", "") + chat_time = item.get("chat_time", None) + parts = [] + if buf: yield {"text": "".join(buf), "sources": sources.copy(), "start_idx": start_idx} 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, From 1de25cf5122c5236cdc6446a39426fb20f88917b Mon Sep 17 00:00:00 2001 From: fridayL Date: Tue, 28 Oct 2025 11:27:48 +0800 Subject: [PATCH 2/3] fix: wrong format --- src/memos/mem_reader/simple_struct.py | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/src/memos/mem_reader/simple_struct.py b/src/memos/mem_reader/simple_struct.py index 8f503df98..c61a8185e 100644 --- a/src/memos/mem_reader/simple_struct.py +++ b/src/memos/mem_reader/simple_struct.py @@ -224,6 +224,11 @@ 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", "") + chat_time = item.get("chat_time", None) + parts = [] if role and str(role).lower() != "mix": parts.append(f"{role}: ") if chat_time: @@ -243,11 +248,6 @@ def _iter_chat_windows(self, scene_data_info, max_tokens=None, overlap=200): buf.append(line) sources.append({"type": "chat", "index": idx, "role": role, "chat_time": chat_time, "content": content}) cur_text = "".join(buf) - for idx, item in enumerate(scene_data_info): - role = item.get("role", "") - content = item.get("content", "") - chat_time = item.get("chat_time", None) - parts = [] if buf: From eeaf82a9b829f40e47ded447f6261cfa290ee00a Mon Sep 17 00:00:00 2001 From: fridayL Date: Tue, 28 Oct 2025 11:34:13 +0800 Subject: [PATCH 3/3] fix: fix code format --- src/memos/mem_reader/simple_struct.py | 11 +++++++++-- 1 file changed, 9 insertions(+), 2 deletions(-) diff --git a/src/memos/mem_reader/simple_struct.py b/src/memos/mem_reader/simple_struct.py index c61a8185e..0f74adead 100644 --- a/src/memos/mem_reader/simple_struct.py +++ b/src/memos/mem_reader/simple_struct.py @@ -246,10 +246,17 @@ 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, "content": content}) + sources.append( + { + "type": "chat", + "index": idx, + "role": role, + "chat_time": chat_time, + "content": content, + } + ) cur_text = "".join(buf) - if buf: yield {"text": "".join(buf), "sources": sources.copy(), "start_idx": start_idx}