add tavily search

agent.py

@@ -31,11 +31,16 @@ import pytesseract
 import fitz  # PyMuPDF
 from dotenv import load_dotenv
 from contextlib import redirect_stdout
+from langchain_community.tools import TavilySearchResults
+from tavily import TavilyClient
 
 # Load environment variables from .env file
 # in HF Spaces, the .env file is saved in Variables and secrets in settings
 load_dotenv()
 
+# Initialize Tavily client (after loading environment variables)
+tavily_client = TavilyClient(api_key=os.environ.get("TAVILY_API_KEY"))
+
 # === System Prompt ===
 SYSTEM_PROMPT = """
 You are a general AI assistant. I will ask you a question. Report your thoughts, and finish your answer with the following template:
@@ -102,6 +107,21 @@ def wikipedia_search(query: str) -> str:
         return WikipediaAPIWrapper().run(query)
     except Exception as e:
         return f"Error searching Wikipedia: {e}"
+    
+    
+@tool
+def tavily_search(query: str) -> str:
+    """Search the web using Tavily for comprehensive results."""
+    try:
+        results = tavily_client.search(query)
+        # You can format the results as needed; here we just return the summary or first result
+        if isinstance(results, dict) and "results" in results:
+            return results["results"][0].get("content", "No content found.")
+        elif isinstance(results, list) and results:
+            return results[0].get("content", "No content found.")
+        return str(results)
+    except Exception as e:
+        return f"Error performing Tavily search: {e}"
 
 @tool
 def image_recognition(image_path: str) -> str:
@@ -253,12 +273,13 @@ def download_file(url_or_path: str, save_dir: str = "./downloads") -> str:
     except Exception as e:
         return f"Error downloading/copying file: {e}"
 
+
 # Update tools list
 tools: List[StructuredTool] = [
     calculate, web_search, wikipedia_search, image_recognition,
     read_pdf, read_csv, read_spreadsheet, transcribe_audio,
     youtube_transcript_tool, youtube_transcript_api, read_jsonl,
-    python_interpreter, download_file  # Add download_file
+    python_interpreter, download_file, tavily_search  # Add tavily_search here
 ]
 
 class AgentState(TypedDict):
@@ -462,3 +483,4 @@ class MyAgent:
 
 
 
+