Added some better logic

business_continuity.py

@@ -114,6 +114,8 @@ Your output MUST be in this EXACT format with NO DEVIATIONS:
   "third_party_vendors_unavailability_strategy": "Strategy for supplier disruptions (2-3 sentences).",
   "vendor_reasoning": "Why this vendor strategy works (1 sentence)."
 }
+
+CRITICAL: Your output MUST be a valid JSON object with the above structure. If you cannot generate this, return an error message in the same JSON format with a "message" field indicating the error.
 GIVE ONLY this JSON output. DO NOT include any other text, explanations, or formatting.
 THE ENTIRE CODE DEPENDS ON THIS JSON STRUCTURE BEING CORRECT. PLEASE ensure the JSON is valid and parsable.
 DO NOT add markdown code blocks, explanations, or any text before or after the JSON. ONLY output the raw JSON object."""
@@ -131,7 +133,9 @@ Description: {request.business_process.process_description}"""
         # Parse JSON response directly
         try:
             # Simple direct parsing - trust the LLM to format correctly
-            strategies_data = json.loads(api_response+'}')
+            if not api_response.strip().endswith('}'):
+                api_response += '}'
+            strategies_data = json.loads(api_response)
             logger.info(f"Request {request_id}: Successfully parsed JSON")
         except json.JSONDecodeError as e:
             # If parsing fails, just return the error