Update agent.py

agent.py

@@ -11,38 +11,17 @@ from langchain_huggingface import HuggingFaceEndpoint, ChatHuggingFace
 from langchain_core.tools import tool
 from langchain_community.document_loaders import WikipediaLoader
 from langchain_google_genai import ChatGoogleGenerativeAI
-from langchain_community.tools import DuckDuckGoSearchRun
+from langchain_tavily import TavilySearch
 
 load_dotenv()
 
 # ReAct System Prompt
-REACT_SYSTEM_PROMPT = """You are a helpful assistant tasked with answering questions using a set of tools.
-
-You use the ReAct (Reasoning and Acting) methodology to solve problems.
-
-For each user query, you should:
-1. **Thought**: Think step by step about what you need to do to answer the question
-2. **Action**: Use the appropriate tool(s) to gather information or perform calculations
-3. **Observation**: Analyze the results from your actions
-4. Repeat the Thought-Action-Observation cycle as needed until you have enough information to provide a complete answer
-
-When you need to use tools:
-- Think about which tool is most appropriate for the task
-- Use tools to gather information or perform calculations
-- Analyze the results and determine if you need additional information
-- Continue until you can provide a comprehensive answer
-
-Available tools:
-- multiply, add, subtract, divide: For mathematical calculations
-- wikidata_search: Search Wikipedia for factual information
-- duckduckgo_search: Search the web for current information
-
-Now, I will ask you a question. Report your thoughts, and finish your answer with the following template:
-FINAL ANSWER: [YOUR FINAL ANSWER].
-YOUR FINAL ANSWER should be a number OR as few words as possible OR a comma separated list of numbers and/or strings. If you are asked for a number, don't use comma to write your number neither use units such as $ or percent sign unless specified otherwise. If you are asked for a string, don't use articles, neither abbreviations (e.g. for cities), and write the digits in plain text unless specified otherwise. If you are asked for a comma separated list, apply the above rules depending of whether the element to be put in the list is a number or a string.
-Your answer should only start with "FINAL ANSWER: ", then follows with the answer.
-
-Always explain your reasoning process and show your work step by step."""
+REACT_SYSTEM_PROMPT = """You are a helpful assistant tasked with answering questions using a set of tools. 
+Now, I will ask you a question. Report your thoughts, and finish your answer with the following template: 
+FINAL ANSWER: [YOUR FINAL ANSWER]. 
+YOUR FINAL ANSWER should be a number OR as few words as possible OR a comma separated list of numbers and/or strings. If you are asked for a number, don't use comma to write your number neither use units such as $ or percent sign unless specified otherwise. If you are asked for a string, don't use articles, neither abbreviations (e.g. for cities), and write the digits in plain text unless specified otherwise. If you are asked for a comma separated list, Apply the rules above for each element (number or string), ensure there is exactly one space after each comma.
+Your answer should only start with "FINAL ANSWER: ", then follows with the answer. 
+"""
 
 @tool
 def multiply(a:int, b:int) -> int:
@@ -89,21 +68,13 @@ def wikidata_search(query: str) -> str:
         ])
     return {"wiki_results": formatted_search_docs}
 
-@tool
-def duckduckgo_search(query: str) -> str:
-    """
-    Search for information on DuckDuckGo and return maximum 3 results.
-    
-    Args:
-        query: The search query.
-    """
-    search = DuckDuckGoSearchRun()
-    # DuckDuckGo returns a string, so we need to format it differently
-    results = search.run(query)
-    formatted_search_docs = f'<Document source="DuckDuckGo Search" page=""/>\n{results}\n</Document>'
-    return {"web_results": formatted_search_docs}
+# Initialize Tavily Search Tool
+tavily_search_tool = TavilySearch(
+    max_results=3,
+    topic="general",
+)
 
-tools = [multiply, add, subtract, divide, wikidata_search, duckduckgo_search]
+tools = [multiply, add, subtract, divide, wikidata_search, tavily_search_tool]
 
 def build_graph():
     llm = ChatGoogleGenerativeAI(model="gemini-2.0-flash", api_key=os.getenv("GOOGLE_API_KEY"))