Update agentos_core.py

agentos_core.py

@@ -1,74 +1,40 @@
-import uuid
-import json
 import os
 import time
-from identity_core import IdentityManager
-from context_graph import ContextGraph
 from telemetry import log_event
 from memory_core import update_memory, load_memory
+from identity_core import IdentityManager
+from context_graph import ContextGraph
 
 
-# --- Memory Manager ---
-class MemoryManager:
-    def __init__(self, agent_id):
-        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)
-
-
-# --- Context Graph ---
-class ContextGraph:
-    def __init__(self):
-        self.context = {}
-
-    def update(self, key, value):
-        self.context[key] = value
-
-
-# --- Agent Identity ---
-def create_agent_identity():
-    return str(uuid.uuid4())
-
-
-# --- Agent Core ---
 class AgentCore:
     def __init__(self, model="gpt-4o-mini"):
         self.identity = IdentityManager()
-self.context = ContextGraph(self.identity.agent_id)
-        self.agent_id = create_agent_identity()
-        self.memory = MemoryManager(self.agent_id)
-        self.context_graph = ContextGraph()
+        self.agent_id = self.identity.agent_id
+        self.context = ContextGraph(self.agent_id)
 
-        self.memory.save({
-            "event": "init",
-            "details": f"Agent {self.agent_id} initialized."
-        })
         log_event(self.agent_id, "init", "success", {"model": model})
+        print(f"[INIT] Agent {self.agent_id} initialized with model {model}")
 
     def run(self, prompt):
         log_event(self.agent_id, "run_start", "in_progress", {"prompt": prompt})
 
         try:
-            # simulate model generation
+            # Simulate model output
             response = f"Agent {self.agent_id} processed: {prompt}"
+
+            # Save memory + context
             update_memory(self.agent_id, "last_prompt", prompt)
             update_memory(self.agent_id, "last_response", response)
+            prompt_node = self.context.add_node("prompt", prompt)
+            response_node = self.context.add_node("response", response)
+            self.context.connect(prompt_node, response_node)
 
+            # Telemetry success
             log_event(self.agent_id, "run_complete", "success", {"response": response})
+            print(f"[RUN] {response}")
             return response
 
         except Exception as e:
             log_event(self.agent_id, "run_failed", "error", {"error": str(e)})
+            print(f"[ERROR] {e}")
             raise e