Add Tavily search tool

app.py

@@ -6,6 +6,8 @@ import pandas as pd
 import spaces
 from langchain_openai import ChatOpenAI
 from langchain.agents import create_agent
+from langchain_tavily import TavilySearch
+
 # (Keep Constants as is)
 # --- Constants ---
 DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
@@ -23,7 +25,12 @@ def zerogpu_function():
 #         fixed_answer = "This is a default answer."
 #         print(f"Agent returning fixed answer: {fixed_answer}")
 #         return fixed_answer
+search_tool = TavilySearch(
+    max_results=5,
+    topic="general",
+    search_depth="advanced",
 
+)
 model = ChatOpenAI(
     model="qwen-plus",
     api_key=os.environ["DASHSCOPE_API_KEY"],
@@ -39,16 +46,26 @@ class BasicAgent:
 
         self.agent = create_agent(
             model=agent_model,
-            tools=[],
+            tools=[search_tool],
             system_prompt="""
 You are solving GAIA benchmark questions.
 
-You have NO external tools.
-You cannot browse the web.
-You cannot access files or attachments.
-You cannot use a calculator or Python.
+You have access to a web search tool.
+
+Use web search whenever:
+- The question asks about specific facts that may not be in your knowledge.
+- The question refers to Wikipedia, websites, historical records,
+  specific people, articles, dates, publications, or other information
+  that should be verified.
+- You are uncertain about a factual answer.
+
+You still cannot access files or attachments.
+You do not have Python or a calculator.
 
-Solve each question using only your internal knowledge and reasoning.
+When searching:
+- Search multiple times if necessary.
+- Use different search queries when the first search is insufficient.
+- Carefully reason over the search results before answering.
 
 IMPORTANT:
 - Return only the final answer.