From 065d5578c9b19869b789e6917f91cadbafd6c3dc Mon Sep 17 00:00:00 2001
From: "glin1993@outlook.com" <>
Date: Sat, 6 Dec 2025 19:38:35 +0800
Subject: [PATCH] Fix: Populate source_doc_id in memory metadata for scheduler
 logging

---
 src/memos/multi_mem_cube/single_cube.py | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/src/memos/multi_mem_cube/single_cube.py b/src/memos/multi_mem_cube/single_cube.py
index 081056473..b51429376 100644
--- a/src/memos/multi_mem_cube/single_cube.py
+++ b/src/memos/multi_mem_cube/single_cube.py
@@ -662,6 +662,13 @@ def _process_text_mem(
             mode=extract_mode,
         )
         flattened_local = [mm for m in memories_local for mm in m]
+
+        # Explicitly set source_doc_id to metadata if present in info
+        source_doc_id = (add_req.info or {}).get("source_doc_id")
+        if source_doc_id:
+            for memory in flattened_local:
+                memory.metadata.source_doc_id = source_doc_id
+
         self.logger.info(f"Memory extraction completed for user {add_req.user_id}")
 
         # Add memories to text_mem