Update agent.py

agent.py

@@ -12,6 +12,7 @@ from langchain_core.tools import tool
 from langgraph.graph import START, StateGraph, MessagesState
 from langgraph.prebuilt import tools_condition
 from langgraph.prebuilt import ToolNode
+from langchain_tavily import TavilySearch
 
 import requests
 
@@ -55,33 +56,13 @@ def search_wiki(query: str, max_results: int = 3) -> str:
     return json.dumps(response, indent=2)
 
 
-@tool
-def search_web(query: str, max_results: int = 3) -> str:
-    """
-    Searches the web for the given query and returns a maximum of 'max_results'
-    relevant hits.
-
-    Args:
-        query (str): The search query for the web search.
-        max_results (int): The maximum number of search results to retrieve (default is 3).
-
-    Returns:
-        str: A JSON string containing a list, where each entry
-             represents a search result article with its snippet, title, link and other metadata.
-             Returns an empty list if no results are found or an error occurs.
-    """
-    try:
-        wrapper = DuckDuckGoSearchAPIWrapper(max_results=max_results)
-        search = DuckDuckGoSearchResults(api_wrapper=wrapper)
-        #search = DuckDuckGoSearchResults()
-        results = search.invoke(query)
-        return results
-    except Exception as e:
-        print(f"An error occurred during web search: {e}")
-        return json.dumps([]) # Return an empty JSON list on error
+tavily_search_tool = TavilySearch(
+    max_results=5,
+    topic="general",
+)
 
 tools = [
-    search_web,
+    tavily_search_tool,
     search_wiki
 ]