Bigger LLM and new search tool.

app.py

@@ -10,7 +10,8 @@ from llama_index.core.llms import ChatMessage, TextBlock, ImageBlock, AudioBlock
 from llama_index.llms.openai import OpenAI
 from llama_index.core.agent.workflow import ReActAgent, AgentOutput
 from llama_index.core.tools import FunctionTool
-from llama_index.tools.duckduckgo import DuckDuckGoSearchToolSpec
+#from llama_index.tools.duckduckgo import DuckDuckGoSearchToolSpec
+from llama_index.tools.tavily_research.base import TavilyToolSpec
 from llama_index.tools.wikipedia import WikipediaToolSpec
 from dotenv import load_dotenv
 from pydantic import ValidationError
@@ -33,13 +34,13 @@ SYSTEM_PROMPT = (Path(__file__).parent / 'system_prompt.txt').read_text()
 # --- Basic Agent Definition ---
 class BasicAgent:
     def __init__(self, max_calls_per_minute=15):
-        search_tool = FunctionTool.from_defaults(DuckDuckGoSearchToolSpec().duckduckgo_full_search)
-        wikipedia_load_tool = FunctionTool.from_defaults(WikipediaToolSpec().load_data)
-        wikipedia_search_tool = FunctionTool.from_defaults(WikipediaToolSpec().search_data)
-        self._tools = [search_tool, wikipedia_load_tool, wikipedia_search_tool]
+        self._tools = TavilyToolSpec(api_key=os.environ["TAVILY_API_KEY"]).to_tool_list()
+        #FunctionTool.from_defaults(TavilyToolSpec().duckduckgo_full_search)
+        self._tools.append(FunctionTool.from_defaults(WikipediaToolSpec().load_data))
+        self._tools.append(FunctionTool.from_defaults(WikipediaToolSpec().search_data))
 
         #self._llm = GoogleGenAI(api_key=GOOGLE_API_KEY, model="gemini-2.0-flash", max_tokens=1600)
-        self._llm = OpenAI(model="gpt-4o-mini")
+        self._llm = OpenAI(model="gpt-o3", temperature=0.0)
         self._agent = ReActAgent(tools=self._tools, llm=self._llm)
         # Modify the react prompt.
         self._agent.update_prompts({"react_header": SYSTEM_PROMPT})

requirements.txt

@@ -4,5 +4,5 @@ requests
 pandas
 llama-index
 llama-index-llms-openai
-llama-index-tools-duckduckgo
+lama-index-tools-tavily-research
 llama-index-tools-wikipedia