Update agentos_core.py

agentos_core.py

@@ -2,13 +2,19 @@ import uuid
 import json
 import os
 import time
-from identity_core import IdentityCore
 from context_graph import ContextGraph
+from identity_core import IdentityCore
+
+
+# ---------- TELEMETRY ----------
 class Telemetry:
     @staticmethod
     def log(event, status, data=None):
+        """Simple logger for system events."""
         print(f"[Telemetry] {event} - {status} | {data or {}}")
 
+
+# ---------- MEMORY ----------
 class MemoryManager:
     def __init__(self, agent_id):
         self.agent_id = agent_id
@@ -18,6 +24,7 @@ class MemoryManager:
                 json.dump({"history": []}, f)
 
     def save(self, data):
+        """Save each interaction into memory."""
         with open(self.file_path, "r") as f:
             memory = json.load(f)
         memory["history"].append(data)
@@ -25,27 +32,49 @@ class MemoryManager:
             json.dump(memory, f, indent=2)
         print(f"[Memory] {self.agent_id} learned {data}")
 
+    def recall(self):
+        """Return the memory history."""
+        with open(self.file_path, "r") as f:
+            memory = json.load(f)
+        return memory["history"]
+
+
+# ---------- IDENTITY ----------
 def create_agent_identity():
+    """Generate a unique ID for every agent instance."""
     return str(uuid.uuid4())
 
+
+# ---------- AGENT CORE ----------
 class AgentCore:
     def __init__(self, model="gpt-4o-mini"):
-        """Initialize agent with memory + telemetry."""
-        self.identity = IdentityCore()
-agent_data = self.identity.create_identity()
-self.agent_signature = agent_data["signature"]
+        """Initialize agent with memory, telemetry, context, and identity."""
         self.agent_id = create_agent_identity()
         self.model = model
         self.memory = MemoryManager(self.agent_id)
         self.context = ContextGraph()
-        Telemetry.log("init", "success", {"agent_id": self.agent_id})
+        self.identity = IdentityCore()
+        agent_data = self.identity.create_identity()
+        self.agent_signature = agent_data["signature"]
+
+        Telemetry.log("init", "success", {
+            "agent_id": self.agent_id,
+            "signature": self.agent_signature
+        })
         print(f"[INIT] Agent {self.agent_id} initialized with model {model}")
+
     def run(self, prompt):
+        """Main logic for handling a user prompt."""
         Telemetry.log("run_start", "in_progress", {"prompt": prompt})
         time.sleep(0.5)
+
+        # Simulated processing (replace with model later)
         response = f"Agent {self.agent_id} processed: {prompt}"
+
+        # Save to memory + link to context graph
         self.memory.save({"prompt": prompt, "response": response})
         self.context.link_context(self.agent_id, prompt, response)
+
         Telemetry.log("run_complete", "success", {"response": response})
         print(f"[RUN] {response}")
         return response