Update app/agent/agent.py

app/agent/agent.py

@@ -1,6 +1,9 @@
 import requests
 import json
+
 from config import Config
+from planner import create_plan
+from executor import Executor
 from tools_runtime import execute_tool_call
 
 
@@ -8,6 +11,7 @@ class Agent:
 
     def __init__(self):
         self.history = []
+        self.executor = Executor()
 
     # ========================
     # LLM CALL LAYER
@@ -29,7 +33,25 @@ class Agent:
         return res.json()
 
     # ========================
-    # MAIN EXECUTION LOOP
+    # STEP MODE (AUTONOMOUS GOAL EXECUTION)
+    # ========================
+    def run_goal(self, goal: str):
+
+        # STEP 1: PLAN
+        plan = create_plan(goal)
+
+        # STEP 2: EXECUTE PLAN
+        results = self.executor.run_plan(plan)
+
+        return {
+            "mode": "goal_execution",
+            "goal": goal,
+            "plan": plan,
+            "results": results
+        }
+
+    # ========================
+    # TOOL MODE (DIRECT AGENT CALL)
     # ========================
     def run(self, user_input):
 
@@ -38,18 +60,13 @@ class Agent:
                 "role": "system",
                 "content": Config.SYSTEM_PROMPT + """
 
-You are an autonomous coding + browser agent.
+You are an autonomous coding + browser + planning agent.
 
-You can respond in TWO ways:
+You can operate in TWO modes:
 
-========================
-1) NORMAL RESPONSE
-========================
-Explain or answer normally.
-
-========================
-2) TOOL CALL (STRICT JSON ONLY)
-========================
+================================================
+1) TOOL MODE (STRICT JSON ONLY)
+================================================
 {
   "tool": "write_file",
   "args": {
@@ -58,28 +75,28 @@ Explain or answer normally.
   }
 }
 
-========================
-AVAILABLE TOOLS
-========================
-
-SYSTEM TOOLS:
+Available tools:
+SYSTEM:
 - run_shell(cmd)
 - write_file(path, content)
 - read_file(path)
 - list_files(path)
 - delete_file(path)
 
-BROWSER TOOLS:
+BROWSER:
 - browser_open(url)
 - browser_click(selector)
 - browser_type(selector, text)
 - browser_screenshot(path)
 
-========================
-RULE
-========================
-If action required → return ONLY JSON
-If explanation → normal text only
+================================================
+2) NORMAL MODE
+================================================
+Explain or answer normally.
+
+RULE:
+- If action needed → JSON ONLY
+- If explanation → text only
 """
             },
             {
@@ -90,13 +107,10 @@ If explanation → normal text only
 
         response = self.call_llm(messages)
 
-        # ========================
-        # Extract LLM output
-        # ========================
         content = response["choices"][0]["message"]["content"]
 
         # ========================
-        # TOOL PARSING LAYER
+        # TOOL EXECUTION PATH
         # ========================
         try:
             tool_call = json.loads(content)
@@ -105,22 +119,25 @@ If explanation → normal text only
 
                 result = execute_tool_call(tool_call)
 
-                # save memory
                 self.history.append({
                     "input": user_input,
                     "tool_call": tool_call,
                     "result": result
                 })
 
-                return result
+                return {
+                    "mode": "tool_execution",
+                    "result": result
+                }
 
         except json.JSONDecodeError:
             pass
 
         # ========================
-        # NORMAL RESPONSE
+        # NORMAL RESPONSE PATH
         # ========================
         return {
+            "mode": "normal_response",
             "response": content
         }
 
@@ -130,4 +147,9 @@ If explanation → normal text only
 # ========================
 if __name__ == "__main__":
     agent = Agent()
-    print(agent.run("Open google.com and explain what you see"))
+
+    # 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"))