Update crew.py

crew.py

@@ -10,6 +10,7 @@ from crewai.agents.agent_builder.base_agent import BaseAgent
 from crewai.project import CrewBase, agent, crew, task
 from crewai.tools import tool
 from google import genai
+from google.genai import types
 from openinference.instrumentation.crewai import CrewAIInstrumentor
 from phoenix.otel import register
 from tools import web_search_tool, image_analysis_tool, audio_analysis_tool, video_analysis_tool, youtube_analysis_tool, document_analysis_tool, arithmetic_tool, code_execution_tool, code_generation_tool, add, subtract, multiply, divide, modulus
@@ -23,6 +24,16 @@ AGENT_MODEL             = "gpt-4.1-mini"
 
 FINAL_ANSWER_MODEL      = "gemini-2.5-pro-preview-03-25"
 
+WEB_SEARCH_MODEL        = "gemini-2.5-flash-preview-04-17"
+IMAGE_ANALYSIS_MODEL    = "gemini-2.5-flash-preview-04-17"
+AUDIO_ANALYSIS_MODEL    = "gemini-2.5-flash-preview-04-17"
+VIDEO_ANALYSIS_MODEL    = "gemini-2.5-flash-preview-04-17"
+YOUTUBE_ANALYSIS_MODEL  = "gemini-2.5-flash-preview-04-17"
+DOCUMENT_ANALYSIS_MODEL = "gemini-2.5-flash-preview-04-17"
+ARITHMETIC_MODEL        = "gemini-2.5-flash-preview-04-17"
+CODE_GENERATION_MODEL   = "gemini-2.5-flash-preview-04-17"
+CODE_EXECUTION_MODEL    = "gemini-2.5-flash-preview-04-17"
+
 # LLM evaluation
 
 PHOENIX_API_KEY = os.environ["PHOENIX_API_KEY"]