Update agent.py

agent.py

@@ -4,6 +4,7 @@ import json
 import requests
 import pandas as pd
 from huggingface_hub import InferenceClient
+from openai import OpenAI
 
 class GetFileTool(Tool):
     name = "get_file_tool"
@@ -126,6 +127,38 @@ 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": "The question to search the web for"}
+    }
+    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(
@@ -187,8 +220,8 @@ manager_agent = CodeAgent(
         model_id="gemini/gemini-2.5-flash-preview-04-17",
         api_key=os.getenv("GOOGLE_API_KEY"),
     ),
-    tools=[GetFileTool(), LoadXlsxFileTool(), LoadTextFileTool(), ImageAnalysisTool()],
-    managed_agents=[web_agent, audio_agent],
+    tools=[GetFileTool(), LoadXlsxFileTool(), LoadTextFileTool(), ImageAnalysisTool(), WebSearchPerplexityTool()],
+    managed_agents=[audio_agent],
     additional_authorized_imports=[
         "pandas"
     ],