gemini

agent.py

@@ -4,7 +4,6 @@ import json
 import requests
 import pandas as pd
 from huggingface_hub import InferenceClient
-from openai import OpenAI
 
 class GetFileTool(Tool):
     name = "get_file_tool"
@@ -127,38 +126,6 @@ class ImageAnalysisTool(Tool):
 
         return completion.choices[0].message.content
 
-class WebSearchPerplexityTool(Tool):
-    name = "web_search_perplexity_tool"
-    description = """This tool searches the web for information using Perplexity, try first to forward the entire question"""
-    inputs = {
-        "question": {"type": "string", "description": "Use the entire question at first and then iterate with follow-up questions"}
-    }
-    output_type = "string"
-
-    def forward(self, question: str) -> str:
-        messages = [
-            {
-                "role": "system",
-                "content": (
-                    "Answer the question based on the information provided and following the instructions"
-                ),
-            },
-            {   
-                "role": "user",
-                "content": question,
-            },
-        ]
-
-        client = OpenAI(api_key=os.getenv("PERPLEXITY_API_KEY"), base_url="https://api.perplexity.ai")
-
-        # chat completion without streaming
-        response = client.chat.completions.create(
-            model="sonar-pro",
-            messages=messages,
-        )
-        
-        return response.choices[0].message.content
-
 
 def final_answer_formatting(answer, question):
     model = LiteLLMModel(
@@ -167,14 +134,7 @@ def final_answer_formatting(answer, question):
     )
 
     prompt = f"""
-    You are an AI assistant specialized in the GAIA benchmark. For the question provided, generate the answer in the exact format requested by the question. Do not include any other text, formatting or creative additions.
-
-    For example:
-    Question: "what is the capital of France ?"
-    Answer: "The capital of France is Paris. It is the largest city in France and serves as the country's political, cultural, and economic center. Paris has been the capital since its liberation in 1944, although it has held this status intermittently throughout French history"
-    Return: "Paris"
-
-    Your turn:
+    You are an AI assistant specialized in the GAIA benchmark. For the question provided, generate the answer in the exact format requested by the question. Do not include any other text or creative additions.
     Question: {question}
     Answer: {answer}
     """
@@ -223,15 +183,15 @@ audio_agent = CodeAgent(
 manager_agent = CodeAgent(
     name="manager_agent",
     model=LiteLLMModel(
-        model_id="gemini/gemini-2.0-flash",
+        model_id="gemini/gemini-2.5-flash-preview-04-17",
         api_key=os.getenv("GOOGLE_API_KEY"),
     ),
-    tools=[GetFileTool(), LoadXlsxFileTool(), LoadTextFileTool(), ImageAnalysisTool(), WebSearchPerplexityTool()],
-    managed_agents=[audio_agent],
+    tools=[GetFileTool(), LoadXlsxFileTool(), LoadTextFileTool(), ImageAnalysisTool()],
+    managed_agents=[web_agent, audio_agent],
     additional_authorized_imports=[
         "pandas"
     ],
     planning_interval=5,
     verbosity_level=1,
     max_steps=10,
-)
+)