Updated memory related tasks

src/manager/manager.py

@@ -158,6 +158,8 @@ class GeminiManager:
 
     def get_k_memories(self, query, k=5, threshold=0.0):
         memories = MemoryManager().get_memories()
+        for i in range(len(memories)):
+            memories[i] = memories[i]['memory']
         if len(memories) == 0:
             return []
         top_k = min(k, len(memories))

src/tools/default_tools/memory_manager.py

@@ -9,22 +9,22 @@ class MemoryManager():
 
     inputSchema = {
         "name": "MemoryManager",
-        "description": "Updates, retrieves, or deletes the memory of an AI agent. The memory is stored in a JSON file.",
+        "description": "Updates, retrieves, or deletes the memory of an AI agent.",
         "parameters": {
             "type": "object",
             "properties":{
                 "action": {
                     "type": "string",
-                    "enum": ["add_memory", "get_memory", "delete_memory"],
-                    "description": "The action to perform: add_memory, get_memory, or delete_memory.",
+                    "enum": ["add_memory", "get_all_memories", "delete_memory"],
+                    "description": "The action to perform: add_memory, get_all_memories, or delete_memory.",
                 },
                 "memory": {
                     "type": "string",
                     "description": "The memory to add. Required for 'add_memory' action.",
                 },
-                "index": {
-                    "type": "integer",
-                    "description": "The index of the memory to delete. Required for 'delete_memory' action.",
+                "key": {
+                    "type": "string",
+                    "description": "The key to delete or add memory."
                 },
             },
             "required": ["action"],
@@ -45,29 +45,40 @@ class MemoryManager():
     def update_memories(self, memories):
         # save the memory to src/data/memory.json
         with open("src/data/memory.json", "w") as f:
-            json.dump(memories, f)
+            json.dump(memories, f, indent=4)
 
 
     def run(self, **kwargs):
         # save it to src/data/memory.json
         action = kwargs.get("action")
         memory = kwargs.get("memory")
-        index = kwargs.get("index")
+        key = kwargs.get("key")
         memories = self.get_memories()
-        if action == "get_memory":
+        if action == "get_all_memories":
             return {
                 "status": "success",
                 "message": "Memory retrieved successfully",
                 "output": memories
             }
         if action == "add_memory":
-            if memory is None:
+            if memory is None or key is None:
                 return {
                     "status": "error",
-                    "message": "Memory is required for add_memory action",
+                    "message": "Memory and key are required for add_memory action",
                     "output": None
                 }
-            memories.append(memory)
+            # check if the key already exists
+            for mem in memories:
+                if mem["key"] == key:
+                    return {
+                        "status": "error",
+                        "message": f"Memory with key {key} already exists",
+                        "output": None
+                    }
+            memories.append({
+                "key": key,
+                "memory": memory
+            })
             self.update_memories(memories)
             return {
                 "status": "success",
@@ -75,22 +86,24 @@ class MemoryManager():
                 "output": None
             }
         if action == "delete_memory":
-            if index is None:
+            if key is None:
                 return {
                     "status": "error",
-                    "message": "Index is required for delete_memory action",
+                    "message": "Key is required for delete_memory action",
                     "output": None
                 }
-            if index < 0 or index >= len(memories):
-                return {
-                    "status": "error",
-                    "message": "Index out of range",
-                    "output": None
-                }
-            memories.pop(index)
-            self.update_memories(memories)
+            # check if the key exists
+            for mem in memories:
+                if mem["key"] == key:
+                    memories.remove(mem)
+                    self.update_memories(memories)
+                    return {
+                        "status": "success",
+                        "message": "Memory deleted successfully",
+                        "output": None
+                    }
             return {
-                "status": "success",
-                "message": "Memory deleted successfully",
+                "status": "error",
+                "message": f"Memory with key {key} not found",
                 "output": None
             }