Update crew.py

crew.py

@@ -13,8 +13,8 @@ from openinference.instrumentation.crewai import CrewAIInstrumentor
 from phoenix.otel import register
 from util import get_final_answer
 
-MANAGER_MODEL = "gpt-4.5-preview"
-AGENT_MODEL = "gpt-4.5-preview"
+MANAGER_MODEL = "gpt-4o"
+AGENT_MODEL = "gpt-4o"
 
 PHOENIX_API_KEY = os.environ["PHOENIX_API_KEY"]
 
@@ -40,90 +40,64 @@ def run_crew(question, file_name):
     # Agents
 
     image_analysis_agent = Agent(
-        role="image_analysis_agent",
+        role="Image Analysis Agent",
         goal="Analyze image to help answer question \"{topic}\". ",
         backstory="As an expert image analysis assistant, you analyze the image to help answer the question. ",
         allow_delegation=False,
         llm=AGENT_MODEL,
-        max_iter=2,
+        max_iter=3,
         tools=[image_analysis_tool],
         verbose=False
     )
     
     python_coding_agent = Agent(
-        role="python_coding_agent",
+        role="Python Coding Agent",
         goal="Write and/or execute Python code to help answer question \"{topic}\". ",
         backstory="As an expert Python coding assistant, you write and/or execute Python code to help answer the question. ",
         allow_delegation=False,
         llm=AGENT_MODEL,
-        max_iter=2,
+        max_iter=3,
         tools=[python_coding_tool],
         verbose=False,
     )
    
     video_analysis_agent = Agent(
-        role="video_analysis_agent",
+        role="Video Analysis Agent",
         goal="Analyze video to help answer question \"{topic}\". ",
         backstory="As an expert video analysis assistant, you analyze the video to help answer the question. ",
         allow_delegation=False,
         llm=AGENT_MODEL,
-        max_iter=2,
+        max_iter=3,
         tools=[video_analysis_tool],
         verbose=False
     )
 
     web_search_agent = Agent(
-        role="web_search_agent",
+        role="Web Search Agent",
         goal="Search the web to help answer question \"{topic}\", then scrape the most relevant web page. ",
         backstory="As an expert web search assistant, you search the web to help answer the question. ",
         allow_delegation=False,
         llm=AGENT_MODEL,
-        max_iter=2,
+        max_iter=3,
         tools=[web_search_tool, web_rag_tool],
         verbose=False
     )
 
     manager_agent = Agent(
-        role="manager_agent",
+        role="Manager Agent",
         goal="Try to answer the following question. If needed, delegate to **one** of your coworkers, image_analysis_agent, python_coding_agent, video_analysis_agent, or web_search_agent for help. Question: \"{topic}\" ",
         backstory="As an expert manager assistant, you answer the question. ",
         allow_delegation=True,
         llm=MANAGER_MODEL,
         max_iter=5,
-        verbose=True
+        verbose=False
     )
 
     # Tasks
-
-    """
-    image_analysis_task = Task(
-        agent=image_analysis_agent,
-        description="Analyze image to help answer question \"{topic}\". ",
-        expected_output="Content to help answer the question. "
-    )
-    
-    python_coding_task = Task(
-        agent=python_coding_agent,
-        description="Write and/or execute Python code to help answer question \"{topic}\". ",
-        expected_output="Content to help answer the question. "
-    )
-
-    video_analysis_task = Task(
-        agent=video_analysis_agent,
-        description="Analyze video to help answer question \"{topic}\". ",
-        expected_output="Content to help answer the question. "
-    )
-    
-    web_search_task = Task(
-        agent=web_search_agent,
-        description="Search the web to help answer question \"{topic}\", then scrape the most relevant web page. ",
-        expected_output="Content to help answer the question. "
-    )
-    """
-    
+  
     manager_task = Task(
         agent=manager_agent,
-        description="Try to answer the following question. If needed, delegate to **one** of your coworkers, image_analysis_agent, python_coding_agent, video_analysis_agent, or web_search_agent for help. Question: \"{topic}\" ",
+        description="Try to answer the following question. If needed, delegate to **one** of your coworkers, Image Analysis Agent, Python Coding Agent, Video Analysis Agent, or Web Search Agent for help. Question: \"{topic}\" ",
         expected_output="The answer to the question. "
     )
     
@@ -133,7 +107,7 @@ def run_crew(question, file_name):
         agents=[image_analysis_agent, python_coding_agent, video_analysis_agent, web_search_agent],
         manager_agent=manager_agent,
         tasks=[manager_task],
-        verbose=True
+        verbose=False
     )
 
     print(crew.agents)