Skip to content

Commit d181339

Browse files
author
yuan.wang
committed
Merge branch 'dev' into feat/fix_palyground_bug
2 parents 70373f9 + 53aa48c commit d181339

File tree

1 file changed

+2
-12
lines changed

1 file changed

+2
-12
lines changed

src/memos/mem_scheduler/general_scheduler.py

Lines changed: 2 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -626,18 +626,8 @@ def _extract_fields(mem_item):
626626
or mem_item.get("original_content")
627627
)
628628
source_doc_id = None
629-
if isinstance(mem_item, dict):
630-
source_doc_id = (
631-
mem_item.get("source_doc_id")
632-
or mem_item.get("doc_id")
633-
or (mem_item.get("metadata") or {}).get("source_doc_id")
634-
)
635-
else:
636-
metadata = getattr(mem_item, "metadata", None)
637-
if metadata:
638-
source_doc_id = getattr(metadata, "source_doc_id", None) or getattr(
639-
metadata, "doc_id", None
640-
)
629+
if "archived_id" in mem_item:
630+
source_doc_id = mem_item.get("archived_id")
641631

642632
return mem_id, mem_memory, original_content, source_doc_id
643633

0 commit comments

Comments
 (0)