Update hedis_engine.py

hedis_engine.py

@@ -1,5 +1,4 @@
 # hedis_engine.py
-# interactive_hedis_engine.py
 import os
 import json
 from crewai import Agent, Task, Crew, Process
@@ -89,13 +88,13 @@ class InteractiveHedisComplianceEngine:
             hedis_task_description += f"\n\nIncorporate these suggestions: {suggestions}\n"
         
         hedis_task_description += (
-            "Return JSON with fields: 'measure','required_tests','required_medications',"
-            "'codes','timeframes','inclusions','exclusions'."
+            "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."
         )
 
         hedis_task = Task(
             description=hedis_task_description,
-            expected_output="Structured JSON with absolute date ranges.",
+            expected_output="Valid JSON object (no markdown, no code blocks) with fields: 'measure','required_tests','required_medications','codes','timeframes','inclusions','exclusions'.",
             agent=self.hedis_expert,
         )
 
@@ -116,10 +115,27 @@ class InteractiveHedisComplianceEngine:
         result = criteria_crew.kickoff(inputs=inputs)
         
         try:
+            # Clean the result string - remove markdown code blocks and extra whitespace
+            result_str = str(result).strip()
+            
+            # Remove markdown code blocks if present
+            if result_str.startswith("```json"):
+                result_str = result_str[7:]  # Remove ```json
+            elif result_str.startswith("```"):
+                result_str = result_str[3:]   # Remove ```
+            
+            if result_str.endswith("```"):
+                result_str = result_str[:-3]  # Remove closing ```
+            
+            # Remove any leading/trailing whitespace and newlines
+            result_str = result_str.strip()
+            
             # Try to parse as JSON
-            criteria_json = json.loads(str(result))
+            criteria_json = json.loads(result_str)
             return criteria_json
-        except json.JSONDecodeError:
+        except json.JSONDecodeError as e:
+            print(f"JSON parsing error: {e}")
+            print(f"Raw result: {repr(str(result))}")
             # If not valid JSON, return as string
             return {"raw_output": str(result)}