Skip to content

Commit 018d759

Browse files
authored
Feat: remove long waring for internet and add content for memreader (#401)
* feat: add content and remove log waring * fix: wrong format * fix: fix code format
1 parent 83a7c34 commit 018d759

File tree

2 files changed

+9
-5
lines changed

2 files changed

+9
-5
lines changed

src/memos/mem_reader/simple_struct.py

Lines changed: 9 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -224,7 +224,6 @@ def _iter_chat_windows(self, scene_data_info, max_tokens=None, overlap=200):
224224
max_tokens = max_tokens or self.chat_window_max_tokens
225225
buf, sources, start_idx = [], [], 0
226226
cur_text = ""
227-
228227
for idx, item in enumerate(scene_data_info):
229228
role = item.get("role", "")
230229
content = item.get("content", "")
@@ -247,7 +246,15 @@ def _iter_chat_windows(self, scene_data_info, max_tokens=None, overlap=200):
247246
cur_text = "".join(buf)
248247

249248
buf.append(line)
250-
sources.append({"type": "chat", "index": idx, "role": role, "chat_time": chat_time})
249+
sources.append(
250+
{
251+
"type": "chat",
252+
"index": idx,
253+
"role": role,
254+
"chat_time": chat_time,
255+
"content": content,
256+
}
257+
)
251258
cur_text = "".join(buf)
252259

253260
if buf:

src/memos/memories/textual/simple_tree.py

Lines changed: 0 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -151,9 +151,6 @@ def search(
151151
list[TextualMemoryItem]: List of matching memories.
152152
"""
153153
if (self.internet_retriever is not None) and manual_close_internet:
154-
logger.warning(
155-
"Internet retriever is init by config , but this search set manual_close_internet is True and will close it"
156-
)
157154
searcher = Searcher(
158155
self.dispatcher_llm,
159156
self.graph_store,

0 commit comments

Comments
 (0)