Update app/core/agents.py

app/core/agents.py

@@ -1122,7 +1122,21 @@ def llm_judge(original_payload: Dict[str, Any], generated_plan: Dict[str, Any])
     }}
     Do NOT include any extra text.
     """
-        
+
+
+        judge_output_schema = {
+          "approved": "boolean",
+          "score": "float",
+          "feedback": "string",
+          "validation": {
+            "feature_issues": [],
+            "execution_issues": [],
+            "ml_best_practice_violations": [],
+            "notes": []
+          }
+        }
+
+       
         ml_judge_prompt = f"""
 You are the **RiverGen ML Quality Assurance Judge**.
 
@@ -1200,17 +1214,7 @@ SCORING GUIDELINES
 OUTPUT FORMAT (JSON ONLY)
 ────────────────────────────────────────
 Return ONLY:
-{
-  "approved": boolean,
-  "score": float,
-  "feedback": "string",
-  "validation": {
-    "feature_issues": [],
-    "execution_issues": [],
-    "ml_best_practice_violations": [],
-    "notes": []
-  }
-}
+{json.dumps(judge_output_schema, indent=2)}
 
 NO extra text.
 """