Fix: Populate source_doc_id in memory metadata for scheduler logging (#625)

Co-authored-by: [email protected] <>

Author: glin93

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