Added user_id while updating memory (#1613)

mem0ai · Aug 2, 2024 · 784b607 · 784b607
1 parent 44aa16a
commit 784b607
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/mem0/memory/main.py b/mem0/memory/main.py
@@ -377,6 +377,14 @@ def _update_memory_tool(self, memory_id, data, metadata=None):
         new_metadata["data"] = data
         new_metadata["created_at"] = existing_memory.payload.get("created_at")
         new_metadata["updated_at"] = datetime.now(pytz.timezone('US/Pacific')).isoformat()
+
+        if "user_id" in existing_memory.payload: 
+            new_metadata["user_id"] = existing_memory.payload["user_id"]
+        if "agent_id" in existing_memory.payload:
+            new_metadata["agent_id"] = existing_memory.payload["agent_id"]
+        if "run_id" in existing_memory.payload:
+            new_metadata["run_id"] = existing_memory.payload["run_id"]
+
         embeddings = self.embedding_model.embed(data)
         self.vector_store.update(
             name=self.collection_name,