Adding web tools to main agent

agents/llama_index_agent.py

@@ -5,6 +5,10 @@ from typing import Optional, List, Any
 from llama_index.llms.openai import OpenAI
 from llama_index.llms.anthropic import Anthropic
 
+from tools.web_tools import (
+    tavily_tool,
+    wikipedia_tool
+)
 class GaiaAgent(ReActAgent):
     """
     A flexible ReActAgent for GAIA benchmark tasks that supports multiple LLM providers.
@@ -47,7 +51,11 @@ class GaiaAgent(ReActAgent):
         
         # Use default tools if not provided
         if tools is None:
-            tools = [reverse_text_tool]
+            tools = [
+                reverse_text_tool,
+                wikipedia_tool,
+                tavily_tool
+                ]
             
         # Use default system prompt if not provided
         if system_prompt is None:
@@ -72,19 +80,7 @@ class GaiaAgent(ReActAgent):
             
         elif model_provider == "anthropic":
             return Anthropic(model=model_name, api_key=api_key or os.getenv("ANTHROPIC_API_KEY"))
-            
-        elif model_provider == "cohere":
-            from llama_index.llms.cohere import Cohere
-            return Cohere(model=model_name, api_key=api_key or os.getenv("COHERE_API_KEY"))
-            
-        elif model_provider == "huggingface":
-            from llama_index.llms.huggingface import HuggingFaceLLM
-            return HuggingFaceLLM(model_name=model_name, tokenizer_name=model_name)
-            
-        elif model_provider == "llama":
-            from llama_index.llms.llama_cpp import LlamaCPP
-            return LlamaCPP(model_path=model_name)
-            
+                        
         else:
             raise ValueError(f"Unsupported model provider: {model_provider}. "
                             f"Supported providers are: openai, anthropic, cohere, huggingface, llama")

tools/web_tools.py

@@ -0,0 +1,8 @@
+from llama_index.tools.wikipedia import WikipediaToolSpec
+from llama_index.tools.tavily_research import TavilyToolSpec
+import os
+
+wikipedia_tool = WikipediaToolSpec()
+tavily_tool = TavilyToolSpec(
+    api_key=os.getenv("TAVILY_API_KEY")
+)