Update agentos_core.py

agentos_core.py

@@ -1,28 +1,43 @@
-def run(self, prompt):
-    # Log start
-    self.telemetry.log("run_start", "in_progress", {"prompt": prompt})
+import json, os, time
+from memory_core import MemoryManager
+from context_graph import ContextGraph
+from telemetry import Telemetry
+from identity_core import create_agent_identity
 
-    try:
-        # Create a dummy response for now
-        response = f"Agent {self.agent_id} processed: {prompt}"
 
-        # Save to memory
-        self.memory.save({"prompt": prompt, "response": response})
+class AgentCore:
+    def __init__(self, model="gpt-4o-mini"):
+        self.agent_id = create_agent_identity()
+        self.memory = MemoryManager(self.agent_id)
+        self.context = ContextGraph()
+        self.telemetry = Telemetry(self.agent_id)
+        self.model = model
 
-        # Retrieve past context for this agent
-        previous_context = self.context.get_context(self.agent_id)
+    def run(self, prompt):
+        # Log start
+        self.telemetry.log("run_start", "in_progress", {"prompt": prompt})
 
-        # Check for matches in past context
-        for key, val in previous_context.items():
-            if key.lower() in prompt.lower():
-                response = f"You told me earlier: {val}"
-                break
+        try:
+            # Create a dummy response for now
+            response = f"Agent {self.agent_id} processed: {prompt}"
 
-        # Log completion
-        self.telemetry.log("run_complete", "success", {"response": response})
+            # Save to memory
+            self.memory.save({"prompt": prompt, "response": response})
 
-        return response
+            # Retrieve past context for this agent
+            previous_context = self.context.get_context(self.agent_id)
 
-    except Exception as e:
-        self.telemetry.log("run_failed", "error", {"error": str(e)})
-        raise e
+            # Check for matches in past context
+            for key, val in previous_context.items():
+                if key.lower() in prompt.lower():
+                    response = f"You told me earlier: {val}"
+                    break
+
+            # Log completion
+            self.telemetry.log("run_complete", "success", {"response": response})
+
+            return response
+
+        except Exception as e:
+            self.telemetry.log("run_failed", "error", {"error": str(e)})
+            raise e