Update app/agent/agent.py

app/agent/agent.py

@@ -13,10 +13,11 @@ class Agent:
         self.history = []
         self.executor = Executor(llm=self.call_llm)
 
-    # ========================
-    # LLM CALL LAYER (SAFE)
-    # ========================
+    # =====================================================
+    # LLM CALL LAYER (ROBUST)
+    # =====================================================
     def call_llm(self, messages):
+
         try:
             res = requests.post(
                 Config.LLM_API_URL,
@@ -32,27 +33,51 @@ class Agent:
                 timeout=60
             )
 
+            # HTTP check
+            res.raise_for_status()
+
             data = res.json()
 
-            # safety check
+            # structure validation
+            if not isinstance(data, dict):
+                return self._llm_fallback("invalid_response")
+
+            if "choices" not in data:
+                return self._llm_fallback("missing_choices")
+
             return data
 
         except Exception as e:
-            return {
-                "error": str(e),
-                "choices": [{
+            return self._llm_fallback(str(e))
+
+    # =====================================================
+    # SAFE FALLBACK RESPONSE
+    # =====================================================
+    def _llm_fallback(self, error_msg):
+
+        return {
+            "error": error_msg,
+            "choices": [
+                {
                     "message": {
                         "content": "LLM_ERROR"
                     }
-                }]
-            }
+                }
+            ]
+        }
 
-    # ========================
+    # =====================================================
     # GOAL MODE (AUTONOMOUS)
-    # ========================
+    # =====================================================
     def run_goal(self, goal: str):
 
-        plan = create_plan(goal)
+        try:
+            plan = create_plan(goal)
+        except Exception as e:
+            return {
+                "mode": "error",
+                "error": f"plan_failed: {str(e)}"
+            }
 
         results = self.executor.run_plan(plan)
 
@@ -63,9 +88,9 @@ class Agent:
             "results": results
         }
 
-    # ========================
-    # TOOL MODE (DIRECT)
-    # ========================
+    # =====================================================
+    # TOOL / CHAT MODE
+    # =====================================================
     def run(self, user_input):
 
         messages = [
@@ -76,8 +101,8 @@ class Agent:
 You are an autonomous coding + browser + planning agent.
 
 MODES:
-1) TOOL MODE → STRICT JSON ONLY
-2) NORMAL MODE → plain text
+- TOOL MODE → STRICT JSON ONLY
+- NORMAL MODE → plain text
 
 TOOLS:
 - run_shell(cmd)
@@ -89,6 +114,10 @@ TOOLS:
 - browser_click(selector)
 - browser_type(selector, text)
 - browser_screenshot(path)
+
+RULE:
+If action → JSON ONLY
+If explanation → text only
 """
             },
             {
@@ -99,20 +128,27 @@ TOOLS:
 
         response = self.call_llm(messages)
 
-        # ========================
-        # SAFE PARSING
-        # ========================
+        # =================================================
+        # SAFE CONTENT EXTRACTION
+        # =================================================
+        content = None
+
         try:
-            content = response["choices"][0]["message"]["content"]
+            content = response.get("choices", [{}])[0] \
+                             .get("message", {}) \
+                             .get("content", None)
         except Exception:
+            content = None
+
+        if not content:
             return {
                 "mode": "error",
-                "response": "Invalid LLM response structure"
+                "response": "Empty LLM response"
             }
 
-        # ========================
+        # =================================================
         # TOOL EXECUTION PATH
-        # ========================
+        # =================================================
         try:
             tool_call = json.loads(content)
 
@@ -120,11 +156,13 @@ TOOLS:
 
                 result = execute_tool_call(tool_call)
 
-                # NO CRASH FIX (healer optional safe guard)
+                # SAFE HEALER CALL (NO CRASH)
                 try:
                     fix = self.executor.healer.heal(user_input, result)
-                    if fix:
+
+                    if isinstance(fix, dict) and "tool" in fix:
                         result = execute_tool_call(fix)
+
                 except Exception:
                     pass
 
@@ -142,18 +180,18 @@ TOOLS:
         except json.JSONDecodeError:
             pass
 
-        # ========================
-        # NORMAL RESPONSE PATH
-        # ========================
+        # =================================================
+        # NORMAL RESPONSE
+        # =================================================
         return {
             "mode": "normal_response",
             "response": content
         }
 
 
-# ========================
+# =====================================================
 # DEBUG
-# ========================
+# =====================================================
 if __name__ == "__main__":
     agent = Agent()