Return id when creating

elizaOS · Jul 27, 2023 · 1838402 · 1838402
1 parent 7cf8a28
commit 1838402
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/agentmemory/main.py b/agentmemory/main.py
@@ -59,6 +59,7 @@ def create_memory(category, text, metadata={}, embedding=None, id=None):
     )
 
     debug_log(f"Created memory {id}: {text}", metadata)
+    return id
 
 
 def create_unique_memory(category, content, metadata={}, similarity=0.95):
@@ -95,6 +96,7 @@ def create_unique_memory(category, content, metadata={}, similarity=0.95):
     metadata["related_to"] = memories[0]["id"]
     metadata["related_document"] = memories[0]["document"]
     create_memory(category, content, metadata=metadata)
+    return id
 
 
 def search_memory(