Update tools.py

tools.py

@@ -1,14 +1,16 @@
-
-from langchain.tools import Tool
-from langchain_community.tools import DuckDuckGoSearchRun
-from langchain.tools import tool
+from langchain.tools import Tool, tool
+from langchain_community.tools.tavily_search import TavilySearchResults
 import math
+import os
 
-# Tool 1: Web Search
-search = DuckDuckGoSearchRun()
+# Tool 1: Web Search (Tavily)
+search = TavilySearchResults(
+    tavily_api_key=os.getenv("TAVILY_API_KEY"),
+    max_results=3
+)
 search_tool = Tool(
     name="web_search",
-    func=search.run,
+    func=lambda q: str(search.invoke(q)),
     description="Useful for searching the internet for current events, facts, or any real-time information. Input should be a search query."
 )