Update hedis_engine.py

hedis_engine.py

@@ -4,6 +4,12 @@ from crewai import Agent, Task, Crew, Process
 from crewai_tools import tool
 from langchain_community.vectorstores import Chroma
 
+from langchain_openai import ChatOpenAI
+
+
+
+
+
 class HedisComplianceEngine:
     """
     Crew that: (1) states official HEDIS measure criteria,
@@ -14,6 +20,7 @@ class HedisComplianceEngine:
         self.vectordb = vectordb
         self.measure_code = measure_code
         self.measurement_year = measurement_year
+        self.model = ChatOpenAI(model="gpt-4o", temperature=0, max_new_tokens=10000)
 
         @tool("patient_chart_search")
         def patient_chart_search(query: str) -> str:
@@ -35,6 +42,7 @@ class HedisComplianceEngine:
                 "Healthcare quality expert with deep knowledge of NCQA HEDIS specifications. "
                 "Must compute concrete absolute date ranges given the measurement year."
             ),
+            llm=self.model,
         )
 
         # 2) Patient chart analyzer (RAG over embeddings)
@@ -45,6 +53,7 @@ class HedisComplianceEngine:
             memory=False,
             tools=[patient_chart_search],
             backstory="Specialist in retrieving structured clinical data from patient chart embeddings.",
+            llm=self.model,
         )
 
         # 3) Reporter who composes a clean markdown report
@@ -58,6 +67,7 @@ class HedisComplianceEngine:
                 "Clinical documentation specialist who organizes evidence and highlights "
                 "relevant data for quality measure analysis."
             ),
+            llm=self.model,
         )
 
         # Tasks