Update crew.py

crew.py

@@ -11,6 +11,8 @@ from openinference.instrumentation.crewai import CrewAIInstrumentor
 from phoenix.otel import register
 from util import get_final_answer
 
+PLANNING_MODEL = "o3"
+
 PHOENIX_API_KEY = os.environ["PHOENIX_API_KEY"]
 
 os.environ["PHOENIX_CLIENT_HEADERS"] = f"api_key={PHOENIX_API_KEY}"
@@ -33,6 +35,7 @@ def run_crew(question):
         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,
+        max_iter=1,
         tools=[web_search_tool, web_rag_tool],
         verbose=True
     )
@@ -42,6 +45,7 @@ def run_crew(question):
         goal="Search a YouTube video for question \"{topic}\".",
         backstory="As an expert YouTube search assistant, you search a YouTube video.",
         allow_delegation=False,
+        max_iter=1,
         tools=[web_search_tool, web_rag_tool],
         verbose=True
     )
@@ -60,7 +64,8 @@ def run_crew(question):
     
     crew = Crew(
         agents=[web_search_agent],
-        #planning=True,
+        planning=True,
+        planning_llm=PLANNING_MODEL
         process=Process.sequential,
         tasks=[web_search],
         verbose=True