Update app/agent/planner.py

app/agent/planner.py

@@ -5,22 +5,26 @@ from config import Config
 
 def create_plan(goal: str):
     """
-    Robust goal planner with full safety checks
+    Robust AI planner:
+    - safe LLM call
+    - strict JSON parsing
+    - fallback protection
     """
 
     prompt = f"""
 You are a planning agent.
 
-Break the goal into small executable steps.
+Break the goal into executable steps.
 
 GOAL:
 {goal}
 
 RULES:
-- Output MUST be valid JSON only
-- No explanation
-- No markdown
-- No extra text
+- Output ONLY valid JSON
+- NO explanation
+- NO markdown
+- NO extra text
+- steps must be simple actionable instructions
 
 FORMAT:
 {{
@@ -34,7 +38,7 @@ FORMAT:
 
     try:
         # =========================
-        # CALL LLM SAFELY
+        # CALL LLM
         # =========================
         res = requests.post(
             Config.LLM_API_URL,
@@ -59,54 +63,51 @@ FORMAT:
             return {"steps": [goal]}
 
         # =========================
-        # SAFE JSON PARSE
+        # JSON PARSE SAFE
         # =========================
         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:
+        # =========================
+        # EXTRACT CONTENT
+        # =========================
+        choices = data.get("choices")
+        if not choices or not isinstance(choices, list):
             return {"steps": [goal]}
 
-        content = (
-            choices[0]
-            .get("message", {})
-            .get("content", "")
-        )
+        message = choices[0].get("message", {})
+        content = message.get("content", "")
 
         if not content:
             return {"steps": [goal]}
 
         # =========================
-        # PARSE JSON CONTENT
+        # PARSE PLAN JSON
         # =========================
         try:
             parsed = json.loads(content)
 
-            if isinstance(parsed, dict) and "steps" in parsed:
-                if isinstance(parsed["steps"], list):
-                    return parsed
+            if (
+                isinstance(parsed, dict)
+                and "steps" in parsed
+                and isinstance(parsed["steps"], list)
+                and len(parsed["steps"]) > 0
+            ):
+                return parsed
 
         except Exception:
             pass
 
         # =========================
-        # FALLBACK STEP SPLIT
+        # FALLBACK (SAFE)
         # =========================
-        return {
-            "steps": [goal]
-        }
+        return {"steps": [goal]}
 
     except Exception:
-        # HARD FALLBACK
-        return {
-            "steps": [goal]
-        }
+        # HARD FAIL SAFE
+        return {"steps": [goal]}