Update agentos_core.py

agentos_core.py

@@ -1,43 +1,42 @@
-from telemetry import Telemetry
-from memory_core import MemoryManager
+import os
+from memory import MemoryManager
 from context_graph import ContextGraph
+from telemetry import Telemetry
 from identity_core import create_agent_identity
 
 
 class AgentCore:
     def __init__(self, model="gpt-4o-mini"):
         self.agent_id = create_agent_identity()
-        self.telemetry = Telemetry(agent_id=self.agent_id)
-        self.model = model
+        self.telemetry = Telemetry(self.agent_id)
         self.memory = MemoryManager(self.agent_id)
         self.context = ContextGraph()
-
-        self.telemetry.log("init", "success", {"agent_id": self.agent_id})
+        self.model = model
         print(f"[INIT] Agent {self.agent_id} initialized with model {self.model}")
 
     def run(self, prompt):
-    self.telemetry.log("run_start", "in_progress", {"prompt": prompt})
-
-    try:
-        # 1. Recall related context
-        previous_context = self.context.query_context(self.agent_id, "favorite")
-
-        # 2. Build a contextual response
-        if previous_context and not prompt.lower().startswith("my"):
-            recall_text = " ".join(previous_context)
-            response = f"From memory: {recall_text}"
-        else:
-            response = f"Agent {self.agent_id} processed: {prompt}"
-
-        # 3. Save memory + context
-        self.memory.save({"prompt": prompt, "response": response})
-        self.context.link_context(self.agent_id, prompt, response)
-
-        # 4. Telemetry log success
-        self.telemetry.log("run_complete", "success", {"response": response})
-        print(f"[RUN] {response}")
-        return response
-
-    except Exception as e:
-        self.telemetry.log("run_failed", "error", {"error": str(e)})
-        raise e
+        self.telemetry.log("run_start", "in_progress", {"prompt": prompt})
+
+        try:
+            # 1️⃣ Recall related context from past messages containing “favorite”
+            previous_context = self.context.query_context(self.agent_id, "favorite")
+
+            # 2️⃣ Build contextual recall logic
+            if previous_context and not prompt.lower().startswith("my"):
+                recall_text = " ".join(previous_context)
+                response = f"From memory: {recall_text}"
+            else:
+                response = f"Agent {self.agent_id} processed: {prompt}"
+
+            # 3️⃣ Save new info to memory + link context
+            self.memory.save({"prompt": prompt, "response": response})
+            self.context.link_context(self.agent_id, prompt, response)
+
+            # 4️⃣ Log success
+            self.telemetry.log("run_complete", "success", {"response": response})
+            print(f"[RUN] {response}")
+            return response
+
+        except Exception as e:
+            self.telemetry.log("run_failed", "error", {"error": str(e)})
+            raise e