Update summarizer.py

summarizer.py

@@ -3,6 +3,8 @@ from typing import Callable
 from crewai import Agent, Task, Crew, Process
 from crewai_tools import tool
 from langchain_community.vectorstores import Chroma
+from langchain_openai import ChatOpenAI
+
 
 class PatientChartSummarizer:
     """
@@ -11,6 +13,7 @@ class PatientChartSummarizer:
     """
     def __init__(self, vectordb: Chroma):
         self.vectordb = vectordb
+        self.model = ChatOpenAI(model="gpt-4o", temperature=0, max_new_tokens=10000)
 
         @tool("patient_chart_search")
         def patient_chart_search(query: str) -> str:
@@ -28,6 +31,7 @@ class PatientChartSummarizer:
                 "You coordinate information extraction for diagnoses, procedures, labs, vitals, and medications, "
                 "then synthesize a clear, clinically relevant chart summary."
             ),
+            llm=self.model
         )
 
         self.summary_task = Task(