Update hedis_engine.py

hedis_engine.py

@@ -20,7 +20,7 @@ class HedisComplianceEngine:
         self.vectordb = vectordb
         self.measure_code = measure_code
         self.measurement_year = measurement_year
-        self.model = ChatOpenAI(model="gpt-4o-mini", temperature=0)
+        self.model = ChatOpenAI(model="gpt-4o", temperature=0, max_tokens=5096)
 
         @tool("patient_chart_search")
         def patient_chart_search(query: str) -> str:
@@ -48,7 +48,7 @@ class HedisComplianceEngine:
         # 2) Patient chart analyzer (RAG over embeddings)
         self.patient_analyzer = Agent(
             role="Patient Chart Analyzer",
-            goal=f"Extract tests, diagnoses, procedures, meds & dates relevant to HEDIS {measure_code}.",
+            goal=f"Extract tests, diagnoses, procedures, meds & dates relevant to HEDIS {measure_code} only. Don't include anything which is not related to this measure.",
             verbose=True,
             memory=False,
             tools=[patient_chart_search],