Update crew.py

crew.py

@@ -13,7 +13,8 @@ from openinference.instrumentation.crewai import CrewAIInstrumentor
 from phoenix.otel import register
 from util import get_final_answer
 
-PLANNING_MODEL = "gpt-4.5-preview"
+CREW_MODEL = "gpt-4.5-preview"
+AGENT_MODEL = "gpt-4.5-preview"
 
 PHOENIX_API_KEY = os.environ["PHOENIX_API_KEY"]
 
@@ -43,9 +44,10 @@ def run_crew(question, file_name):
         goal="Search an image for question \"{topic}\". ",
         backstory="As an expert image search assistant, you search an image for the question. ",
         allow_delegation=False,
-        max_iter=5,
+        llm=AGENT_MODEL,
+        max_iter=1,
         tools=[image_search_tool],
-        verbose=True
+        verbose=False
     )
     
     python_coding_agent = Agent(
@@ -53,6 +55,7 @@ def run_crew(question, file_name):
         goal="Write and execute Python code to solve question \"{topic}\". ",
         backstory="As an expert Python coding assistant, you write and execute Python for the question. ",
         allow_delegation=False,
+        llm=AGENT_MODEL,
         max_iter=1,
         tools=[python_coding_tool],
         verbose=False,
@@ -63,6 +66,7 @@ def run_crew(question, file_name):
         goal="Search the web for question \"{topic}\" and scrape the most relevant web page. ",
         backstory="As an expert web search assistant, you search the web for the question and scrape the most relevant web page. ",
         allow_delegation=False,
+        llm=AGENT_MODEL,
         max_iter=1,
         tools=[web_search_tool, web_rag_tool],
         verbose=False
@@ -73,6 +77,7 @@ def run_crew(question, file_name):
         goal="Search a YouTube video for question \"{topic}\". ",
         backstory="As an expert YouTube search assistant, you search a YouTube video for the question. ",
         allow_delegation=False,
+        llm=AGENT_MODEL,
         max_iter=1,
         tools=[youtube_search_tool],
         verbose=False
@@ -109,7 +114,7 @@ def run_crew(question, file_name):
     crew = Crew(
         agents=[image_search_agent, python_coding_agent, web_search_agent, youtube_search_agent],
         planning=True,
-        planning_llm=PLANNING_MODEL,
+        planning_llm=CREW_MODEL,
         process=Process.sequential,
         tasks=[image_search_task, python_coding_task, web_search_task, youtube_search_task],
         verbose=False