Update hedis_engine.py

hedis_engine.py

@@ -20,12 +20,12 @@ class HedisComplianceEngine:
         self.vectordb = vectordb
         self.measure_code = measure_code
         self.measurement_year = measurement_year
-        self.model = ChatOpenAI(model="gpt-4o", temperature=0, max_tokens=5096)
+        self.model = ChatOpenAI(model="gpt-4o-mini", temperature=0)
 
         @tool("patient_chart_search")
         def patient_chart_search(query: str) -> str:
             """Return top patient chart passages relevant to the query."""
-            results = self.vectordb.similarity_search(query, k=10)
+            results = self.vectordb.similarity_search(query, k=15)
             return "\n".join([res.page_content for res in results])
 
         # 1) HEDIS measure expert (external knowledge via model)
@@ -113,7 +113,7 @@ class HedisComplianceEngine:
                 "### Summary\n"
                 "Use patient_chart_search multiple times for broad coverage."
             ),
-            expected_output="Comprehensive HEDIS in well-formatted markdown.",
+            expected_output="A thorough markdown report of HEDIS findings.",
             agent=self.hedis_chart_reporter,
             context=[self.hedis_task, self.patient_task],
         )