Update hedis_engine.py

hedis_engine.py

@@ -31,14 +31,16 @@ class InteractiveHedisComplianceEngine:
             role="HEDIS Measure Expert",
             goal=(
                 f"Provide accurate, official details for HEDIS measure {measure_code} "
-                "including required tests, codes, timeframes, and exclusions."
+                "including required tests, codes, timeframes, and exclusions. "
+                "ALWAYS respond with valid JSON format only."
             ),
             verbose=True,
             memory=False,
             tools=[],  # design choice: no web/search tools here; rely on model knowledge
             backstory=(
                 "Healthcare quality expert with deep knowledge of NCQA HEDIS specifications. "
-                "Must compute concrete absolute date ranges given the measurement year."
+                "Must compute concrete absolute date ranges given the measurement year. "
+                "CRITICAL: Always respond with properly formatted JSON without any markdown or extra text."
             ),
             llm=self.model,
         )
@@ -88,8 +90,13 @@ class InteractiveHedisComplianceEngine:
             hedis_task_description += f"\n\nIncorporate these suggestions: {suggestions}\n"
         
         hedis_task_description += (
-            "IMPORTANT: Return ONLY valid JSON without any markdown formatting, code blocks, or extra text. "
-            "The response must start with { and end with } and be valid JSON that can be parsed directly."
+            "\n\nCRITICAL FORMATTING REQUIREMENTS:\n"
+            "- Return ONLY valid JSON without any markdown formatting, code blocks, or extra text\n"
+            "- Do not wrap the JSON in ```json``` blocks\n"
+            "- Do not add any explanatory text before or after the JSON\n"
+            "- The response must start with { and end with }\n"
+            "- Ensure all JSON keys and string values are properly quoted\n"
+            "- Example format: {\"measure\":\"...\",\"required_tests\":[...],\"codes\":{...}}"
         )
 
         hedis_task = Task(