tools updated

tools.py

@@ -1,9 +1,11 @@
-from typing import Optional
 from langchain_core.tools import tool
 import math
 import pandas as pd
 import requests
 import io
+from langchain_tavily import TavilySearch
+import os
+from dotenv import load_dotenv
 
 
 @tool
@@ -44,35 +46,20 @@ def describe_table_from_url(file_url: str) -> str:
         return f"Error loading table: {e}"
 
 
-@tool
-def search_web(query: str) -> str:
-    """
-    Perform a simple web search using DuckDuckGo Instant Answer API.
-    Returns a short summary or answer.
-    """
-    try:
-        response = requests.get("https://api.duckduckgo.com/", params={
-            "q": query,
-            "format": "json",
-            "no_redirect": 1,
-            "no_html": 1
-        }, timeout=10)
-        data = response.json()
-        if data.get("AbstractText"):
-            return data["AbstractText"]
-        elif data.get("Answer"):
-            return data["Answer"]
-        elif data.get("RelatedTopics"):
-            return data["RelatedTopics"][0].get("Text", "No summary found.")
-        else:
-            return "No results found."
-    except Exception as e:
-        return f"Web search error: {e}"
-
+load_dotenv()  # Load variables from .env into environment
+api_key = os.getenv("TAVILY_API_KEY")
+# Initialize Tavily Search Tool
+tavily_search_tool = TavilySearch(
+    max_results=5,
+    topic="general",
+    tavily_api_key=api_key
+)
 
 # Export the tools in a list
 TOOLS = [
     compute_expression,
     describe_table_from_url,
-    search_web,
+    tavily_search_tool
 ]
+
+