Skip to content

Commit e4eb9db

Browse files
author
yuan.wang
committed
fix bug
1 parent 4964d2b commit e4eb9db

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

src/memos/mem_reader/simple_struct.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -295,6 +295,9 @@ def _iter_chat_windows(self, scene_data_info, max_tokens=None, overlap=200):
295295
def _process_chat_data(self, scene_data_info, info, **kwargs):
296296
mode = kwargs.get("mode", "fine")
297297
windows = list(self._iter_chat_windows(scene_data_info))
298+
custom_tags = info.pop(
299+
"custom_tags", None
300+
) # msut pop here, avoid add to info, only used in sync fine mode
298301

299302
if mode == "fast":
300303
logger.debug("Using unified Fast Mode")
@@ -324,7 +327,6 @@ def _build_fast_node(w):
324327
else:
325328
logger.debug("Using unified Fine Mode")
326329
chat_read_nodes = []
327-
custom_tags = info.pop("custom_tags", None)
328330
for w in windows:
329331
resp = self._get_llm_response(w["text"], custom_tags)
330332
for m in resp.get("memory list", []):
@@ -353,6 +355,7 @@ def _process_transfer_chat_data(
353355
):
354356
raw_memory = raw_node.memory
355357
response_json = self._get_llm_response(raw_memory, custom_tags)
358+
356359
chat_read_nodes = []
357360
for memory_i_raw in response_json.get("memory list", []):
358361
try:

0 commit comments

Comments
 (0)