diff --git a/src/memos/memories/textual/tree_text_memory/organize/manager.py b/src/memos/memories/textual/tree_text_memory/organize/manager.py
index 59675bdc2..5e9c74f61 100644
--- a/src/memos/memories/textual/tree_text_memory/organize/manager.py
+++ b/src/memos/memories/textual/tree_text_memory/organize/manager.py
@@ -183,7 +183,9 @@ def _add_memories_batch(
                 "ToolTrajectoryMemory",
                 "SkillMemory",
             ):
-                graph_node_id = str(uuid.uuid4())
+                if not memory.id:
+                    logger.error("Memory ID is not set, generating a new one")
+                graph_node_id = memory.id or str(uuid.uuid4())
                 metadata_dict = memory.metadata.model_dump(exclude_none=True)
                 metadata_dict["updated_at"] = datetime.now().isoformat()
 
@@ -384,7 +386,9 @@ def _add_to_graph_memory(
         """
         Generalized method to add memory to a graph-based memory type (e.g., LongTermMemory, UserMemory).
         """
-        node_id = str(uuid.uuid4())
+        if not memory.id:
+            logger.error("Memory ID is not set, generating a new one")
+        node_id = memory.id or str(uuid.uuid4())
         # Step 2: Add new node to graph
         metadata_dict = memory.metadata.model_dump(exclude_none=True)
         tags = metadata_dict.get("tags") or []