From 6aadd88ee00b8bce896a5e59ded2272a4c3d395e Mon Sep 17 00:00:00 2001
From: "glin1993@outlook.com" <>
Date: Wed, 3 Dec 2025 22:26:33 +0800
Subject: [PATCH] chore(feedback): propagate source_doc_id in KB logs if
 available

---
 src/memos/mem_scheduler/general_scheduler.py | 24 ++++++++++++++++----
 1 file changed, 19 insertions(+), 5 deletions(-)

diff --git a/src/memos/mem_scheduler/general_scheduler.py b/src/memos/mem_scheduler/general_scheduler.py
index a2e4f5d4e..ad34530bc 100644
--- a/src/memos/mem_scheduler/general_scheduler.py
+++ b/src/memos/mem_scheduler/general_scheduler.py
@@ -625,12 +625,26 @@ def _extract_fields(mem_item):
                         or mem_item.get("old_memory")
                         or mem_item.get("original_content")
                     )
-                    return mem_id, mem_memory, original_content
+                    source_doc_id = None
+                    if isinstance(mem_item, dict):
+                        source_doc_id = (
+                            mem_item.get("source_doc_id")
+                            or mem_item.get("doc_id")
+                            or (mem_item.get("metadata") or {}).get("source_doc_id")
+                        )
+                    else:
+                        metadata = getattr(mem_item, "metadata", None)
+                        if metadata:
+                            source_doc_id = getattr(metadata, "source_doc_id", None) or getattr(
+                                metadata, "doc_id", None
+                            )
+
+                    return mem_id, mem_memory, original_content, source_doc_id
 
                 kb_log_content: list[dict] = []
 
                 for mem_item in add_records or []:
-                    mem_id, mem_memory, _ = _extract_fields(mem_item)
+                    mem_id, mem_memory, _, source_doc_id = _extract_fields(mem_item)
                     if mem_id and mem_memory:
                         kb_log_content.append(
                             {
@@ -640,7 +654,7 @@ def _extract_fields(mem_item):
                                 "memory_id": mem_id,
                                 "content": mem_memory,
                                 "original_content": None,
-                                "source_doc_id": None,
+                                "source_doc_id": source_doc_id,
                             }
                         )
                     else:
@@ -654,7 +668,7 @@ def _extract_fields(mem_item):
                         )
 
                 for mem_item in update_records or []:
-                    mem_id, mem_memory, original_content = _extract_fields(mem_item)
+                    mem_id, mem_memory, original_content, source_doc_id = _extract_fields(mem_item)
                     if mem_id and mem_memory:
                         kb_log_content.append(
                             {
@@ -664,7 +678,7 @@ def _extract_fields(mem_item):
                                 "memory_id": mem_id,
                                 "content": mem_memory,
                                 "original_content": original_content,
-                                "source_doc_id": None,
+                                "source_doc_id": source_doc_id,
                             }
                         )
                     else: