Update app/agent/planner.py

app/agent/planner.py

@@ -5,18 +5,24 @@ from config import Config
 
 def create_plan(goal: str):
     """
-    Break goal into steps
+    Robust goal planner with full safety checks
     """
 
     prompt = f"""
 You are a planning agent.
 
-Break this goal into steps:
+Break the goal into small executable steps.
 
 GOAL:
 {goal}
 
-Return ONLY JSON like:
+RULES:
+- Output MUST be valid JSON only
+- No explanation
+- No markdown
+- No extra text
+
+FORMAT:
 {{
   "steps": [
     "step 1",
@@ -26,24 +32,81 @@ Return ONLY JSON like:
 }}
 """
 
-    res = requests.post(
-        Config.LLM_API_URL,
-        headers={
-            "Authorization": f"Bearer {Config.LLM_API_KEY}",
-            "Content-Type": "application/json"
-        },
-        json={
-            "model": Config.MODEL,
-            "messages": [
-                {"role": "user", "content": prompt}
-            ],
-            "temperature": 0.2
-        }
-    )
+    try:
+        # =========================
+        # CALL LLM SAFELY
+        # =========================
+        res = requests.post(
+            Config.LLM_API_URL,
+            headers={
+                "Authorization": f"Bearer {Config.LLM_API_KEY}",
+                "Content-Type": "application/json"
+            },
+            json={
+                "model": Config.MODEL,
+                "messages": [
+                    {"role": "user", "content": prompt}
+                ],
+                "temperature": 0.2
+            },
+            timeout=30
+        )
 
-    content = res.json()["choices"][0]["message"]["content"]
+        # =========================
+        # HTTP CHECK
+        # =========================
+        if res.status_code != 200:
+            return {"steps": [goal]}
 
-    try:
-        return json.loads(content)
-    except:
-        return {"steps": [goal]}
+        # =========================
+        # SAFE JSON PARSE
+        # =========================
+        try:
+            data = res.json()
+        except Exception:
+            return {"steps": [goal]}
+
+        # =========================
+        # SAFE STRUCTURE CHECK
+        # =========================
+        if not isinstance(data, dict):
+            return {"steps": [goal]}
+
+        choices = data.get("choices", [])
+        if not choices:
+            return {"steps": [goal]}
+
+        content = (
+            choices[0]
+            .get("message", {})
+            .get("content", "")
+        )
+
+        if not content:
+            return {"steps": [goal]}
+
+        # =========================
+        # PARSE JSON CONTENT
+        # =========================
+        try:
+            parsed = json.loads(content)
+
+            if isinstance(parsed, dict) and "steps" in parsed:
+                if isinstance(parsed["steps"], list):
+                    return parsed
+
+        except Exception:
+            pass
+
+        # =========================
+        # FALLBACK STEP SPLIT
+        # =========================
+        return {
+            "steps": [goal]
+        }
+
+    except Exception:
+        # HARD FALLBACK
+        return {
+            "steps": [goal]
+        }