Skip to content

Commit 1de25cf

Browse files
committed
fix: wrong format
1 parent 5744985 commit 1de25cf

File tree

1 file changed

+5
-5
lines changed

1 file changed

+5
-5
lines changed

src/memos/mem_reader/simple_struct.py

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -224,6 +224,11 @@ 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+
for idx, item in enumerate(scene_data_info):
228+
role = item.get("role", "")
229+
content = item.get("content", "")
230+
chat_time = item.get("chat_time", None)
231+
parts = []
227232
if role and str(role).lower() != "mix":
228233
parts.append(f"{role}: ")
229234
if chat_time:
@@ -243,11 +248,6 @@ def _iter_chat_windows(self, scene_data_info, max_tokens=None, overlap=200):
243248
buf.append(line)
244249
sources.append({"type": "chat", "index": idx, "role": role, "chat_time": chat_time, "content": content})
245250
cur_text = "".join(buf)
246-
for idx, item in enumerate(scene_data_info):
247-
role = item.get("role", "")
248-
content = item.get("content", "")
249-
chat_time = item.get("chat_time", None)
250-
parts = []
251251

252252

253253
if buf:

0 commit comments

Comments
 (0)