Update agentos_core.py

agentos_core.py

@@ -13,35 +13,29 @@ class AgentCore:
         self.context = ContextGraph()
 
         self.telemetry.log("init", "success", {"agent_id": self.agent_id})
-        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:
-            recall_hits = self.context.query_context(self.agent_id, prompt)
-            response = None
-
-            if recall_hits and recall_hits != ["No matching context found."]:
-                for hit in recall_hits:
-                    if " is " in hit:
-                        parts = hit.split(" is ", 1)
-                        subject = parts[0].replace("My ", "Your ").strip()
-                        fact = parts[1].strip().rstrip(".")
-                        if subject.lower() in prompt.lower():
-                            response = f"{subject} is {fact}."
-                            break
-
-            if not response:
-                response = f"Agent {self.agent_id} processed: {prompt}"
-
-            self.memory.save({"prompt": prompt, "response": response})
-            self.context.link_context(self.agent_id, prompt, response)
-
-            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)})
-            print(f"[ERROR] {e}")
-            return f"Error: {e}"
+        print(f"[INIT] Agent {self.agent_id} initialized with 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