Update app/agent/agent.py

app/agent/agent.py

@@ -11,7 +11,7 @@ class Agent:
 
     def __init__(self):
         self.history = []
-        self.executor = Executor()
+        self.executor = Executor(llm=self.call_llm)
 
     # ========================
     # LLM CALL LAYER
@@ -33,14 +33,12 @@ class Agent:
         return res.json()
 
     # ========================
-    # STEP MODE (AUTONOMOUS GOAL EXECUTION)
+    # GOAL MODE (AUTONOMOUS)
     # ========================
     def run_goal(self, goal: str):
 
-        # STEP 1: PLAN
         plan = create_plan(goal)
 
-        # STEP 2: EXECUTE PLAN
         results = self.executor.run_plan(plan)
 
         return {
@@ -51,7 +49,7 @@ class Agent:
         }
 
     # ========================
-    # TOOL MODE (DIRECT AGENT CALL)
+    # TOOL MODE (DIRECT)
     # ========================
     def run(self, user_input):
 
@@ -119,6 +117,12 @@ RULE:
 
                 result = execute_tool_call(tool_call)
 
+                # self-healing (direct mode)
+                fix = self.executor.healer.heal(user_input, result)
+
+                if fix:
+                    result = execute_tool_call(fix)
+
                 self.history.append({
                     "input": user_input,
                     "tool_call": tool_call,
@@ -143,13 +147,10 @@ RULE:
 
 
 # ========================
-# DEBUG RUN
+# DEBUG
 # ========================
 if __name__ == "__main__":
     agent = Agent()
 
-    # Example 1: normal tool call
     print(agent.run("List files in current directory"))
-
-    # Example 2: autonomous goal mode
     print(agent.run_goal("Create a simple python web server"))