Update tools.py

tools.py

@@ -1,8 +1,21 @@
+import getpass
+import os
 from langchain.tools import Tool
-from langchain_community.tools import DuckDuckGoSearchRun
+# from langchain_community.tools import DuckDuckGoSearchRun
+from langchain_tavily import TavilySearch
 from langchain_experimental.utilities import PythonREPL
 
-web_search_tool = DuckDuckGoSearchRun()
+
+if not os.environ.get("TAVILY_API_KEY"):
+    os.environ["TAVILY_API_KEY"] = getpass.getpass("Tavily API key:\n")
+
+# web_search_tool = DuckDuckGoSearchRun()
+
+# Initialize Tavily Search Tool
+tavily_search_tool = TavilySearch(
+    max_results=5,
+    topic="general",
+)
 
 python_repl = PythonREPL()
 repl_tool = Tool(