Update agentos_core.py

agentos_core.py

@@ -1,42 +1,44 @@
-from langchain.memory import ConversationBufferMemory
-from langchain_community.chat_models import ChatOpenAI
-from langchain.chains import ConversationChain
-from memory import MemoryManager, ContextGraph
-class AgentCore:
+import uuid
+import json
+import os
+
+class MemoryManager:
     def __init__(self, agent_id):
-        self.memory = MemoryManager(agent_id)
-    def __init__(self, agent_id):  # Add this function
-        self.memory = MemoryManager(agent_id)  # Move it here
-class AgentCore:
-    def __init__(self, agent_id):  # Add this if it’s not there
-        self.memory = MemoryManager(self.agent_id)  # Move it here
-
-  File "/app/app.py", line 2, in <module>
-    from agentos_core import AgentCore
-  File "/app/agentos_core.py", line 13, in <module>
-    self.context_graph = ContextGraph()
-NameError: name 'self' is not defined
-self.memory.save({
-    "event": "init",
-    "details": f"Agent {self.agent_id} initialized."
-})
-self.context_graph = ContextGraph()
-self.memory.save({
-    "event": "init",
-    "details": f"Agent {self.agent_id} initialized."
-})
+        self.agent_id = agent_id
+        self.file_path = f"memory_{agent_id}.json"
+        self._init_memory()
+
+    def _init_memory(self):
+        if not os.path.exists(self.file_path):
+            with open(self.file_path, "w") as f:
+                json.dump({"history": []}, f)
+
+    def save(self, data):
+        with open(self.file_path, "r") as f:
+            memory = json.load(f)
+        memory["history"].append(data)
+        with open(self.file_path, "w") as f:
+            json.dump(memory, f, indent=2)
+
+
+class ContextGraph:
+    def __init__(self):
+        self.context = {}
+
+    def update(self, key, value):
+        self.context[key] = value
+
+
+def create_agent_identity():
+    return str(uuid.uuid4())
+
+
 class AgentCore:
-    def __init__(self, model="gpt-4-turbo"):
-        # Rest of the code here
-        self.memory = ConversationBufferMemory()
-        self.agent = ConversationChain(
-            llm=ChatOpenAI(model_name=model),
-            memory=self.memory
-        )
-
-    def run(self, prompt):
-        response = self.agent.run(prompt)
-        return {
-            "response": response,
-            "memory_summary": self.memory.buffer
-        }
+    def __init__(self, model="gpt-4o-mini"):
+        self.agent_id = create_agent_identity()
+        self.memory = MemoryManager(self.agent_id)
+        self.context_graph = ContextGraph()
+        self.memory.save({
+            "event": "init",
+            "details": f"Agent {self.agent_id} initialized."
+        })