Update crew.py

crew.py

@@ -1,10 +1,10 @@
 import os
 from crewai import Agent, Crew, Process, Task
 from crewai.tools import tool
-from crewai_tools import (
-    SerperDevTool,
-    WebsiteSearchTool
-)
+#from crewai_tools import (
+#    SerperDevTool,
+#    WebsiteSearchTool
+#)
 from google import genai
 from google.genai import types
 from openinference.instrumentation.crewai import CrewAIInstrumentor
@@ -13,17 +13,19 @@ from util import get_final_answer
 
 ## LLMs
 
-MANAGER_MODEL      = "gpt-4.1-mini"
-AGENT_MODEL        = "gpt-4.1-mini"
+MANAGER_MODEL           = "gpt-4.1-mini"
+AGENT_MODEL             = "gpt-4.1-mini"
 
-FINAL_ANSWER_MODEL = "gemini-2.5-flash-preview-04-17"
+FINAL_ANSWER_MODEL      = "gemini-2.5-flash-preview-04-17"
 
-IMAGE_MODEL        = "gemini-2.5-flash-preview-04-17"
-AUDIO_MODEL        = "gemini-2.5-flash-preview-04-17"
-VIDEO_MODEL        = "gemini-2.5-flash-preview-04-17"
-YOUTUBE_MODEL      = "gemini-2.5-flash-preview-04-17"
-DOCUMENT_MODEL     = "gemini-2.5-flash-preview-04-17"
-CODE_MODEL         = "gemini-2.5-flash-preview-04-17"
+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"
+CODE_GENERATION_MODEL   = "gemini-2.5-flash-preview-04-17"
+CODE_EXECUTION_MODEL    = "gemini-2.5-flash-preview-04-17"
 
 # LLM evaluation
 
@@ -42,8 +44,45 @@ CrewAIInstrumentor().instrument(tracer_provider=tracer_provider)
 def run_crew(question, file_path):
     # Tools
 
-    web_search_tool = SerperDevTool()
-    web_rag_tool = WebsiteSearchTool()
+    #web_search_tool = SerperDevTool()
+    #web_rag_tool = WebsiteSearchTool()
+
+    @tool("Web Search Tool")
+    def web_search_tool(question: str) -> str:
+        """Search the web to answer a question.
+    
+           Args:
+               question (str): Question to answer
+                
+           Returns:
+               str: Answer to the question
+                
+           Raises:
+               RuntimeError: If processing fails"""
+        try:
+            #client = genai.Client(api_key=os.environ["GEMINI_API_KEY"])
+            
+            #response = client.models.generate_content(
+            #    model=IMAGE_MODEL, 
+            #    contents=[file, question]
+            #)
+          
+            #return response.text
+            ###
+            client = genai.Client(api_key=os.environ["GEMINI_API_KEY"])
+            
+            response = client.models.generate_content(
+                model=WEB_SEARCH_MODEL,
+                contents=question,
+                config=types.GenerateContentConfig(
+                    tools=[types.Tool(google_search_retrieval=types.GoogleSearchRetrieval())]
+                )
+            )
+
+            return response.text
+            ###
+        except Exception as e:
+            raise RuntimeError(f"Processing failed: {str(e)}")
     
     @tool("Image Analysis Tool")
     def image_analysis_tool(question: str, file_path: str) -> str:
@@ -64,7 +103,7 @@ def run_crew(question, file_path):
             file = client.files.upload(file=file_path)
 
             response = client.models.generate_content(
-                model=IMAGE_MODEL, 
+                model=IMAGE_ANALYSIS_MODEL, 
                 contents=[file, question]
             )
           
@@ -91,7 +130,7 @@ def run_crew(question, file_path):
             file = client.files.upload(file=file_path)
             
             response = client.models.generate_content(
-                model=AUDIO_MODEL, 
+                model=AUDIO_ANALYSIS_MODEL, 
                 contents=[file, question]
             )
           
@@ -118,7 +157,7 @@ def run_crew(question, file_path):
             file = client.files.upload(file=file_path)
             
             response = client.models.generate_content(
-                model=VIDEO_MODEL, 
+                model=VIDEO_ANALYSIS_MODEL, 
                 contents=[file, question]
             )
 
@@ -143,7 +182,7 @@ def run_crew(question, file_path):
             client = genai.Client(api_key=os.environ["GEMINI_API_KEY"])
 
             return client.models.generate_content(
-                model=YOUTUBE_MODEL,
+                model=YOUTUBE_ANALYSIS_MODEL,
                 contents=types.Content(
                     parts=[types.Part(file_data=types.FileData(file_uri=url)),
                            types.Part(text=question)]
@@ -172,7 +211,7 @@ def run_crew(question, file_path):
             file = client.files.upload(file=file_path)
             
             response = client.models.generate_content(
-                model=DOCUMENT_MODEL, 
+                model=DOCUMENT_ANALYSIS_MODEL, 
                 contents=[file, question]
             )
           
@@ -198,7 +237,7 @@ def run_crew(question, file_path):
             file = client.files.upload(file=file_path)
 
             response = client.models.generate_content(
-                model=CODE_MODEL,
+                model=CODE_GENERATION_MODEL,
                 contents=[question],
                 config=types.GenerateContentConfig(
                     tools=[types.Tool(code_execution=types.ToolCodeExecution)]
@@ -230,7 +269,7 @@ def run_crew(question, file_path):
             file = client.files.upload(file=file_path)
 
             response = client.models.generate_content(
-                model=CODE_MODEL,
+                model=CODE_EXECUTION_MODEL,
                 contents=[file, question],
                 config=types.GenerateContentConfig(
                     tools=[types.Tool(code_execution=types.ToolCodeExecution)]
@@ -247,12 +286,12 @@ def run_crew(question, file_path):
 
     web_search_agent = Agent(
         role="Web Search Agent",
-        goal="Search the web to help answer question \"{question}\", then scrape the most relevant web page.",
+        goal="Search the web to help answer question \"{question}\"",
         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,
-        tools=[web_search_tool, web_rag_tool],
+        tools=[web_search_tool],
         verbose=False
     )