patched tavily to latest

agent/tools.py

@@ -3,12 +3,13 @@ from typing import List
 from langchain_core.tools import tool
 from langchain_community.embeddings import HuggingFaceEmbeddings
 from langchain_chroma import Chroma
-from langchain_community.tools.tavily_search import TavilySearchResults
+from langchain_tavily import TavilySearch
 from langchain_community.document_loaders import WikipediaLoader, ArxivLoader 
-
+from dotenv import load_dotenv
 
 from config import configs
 
+load_dotenv()
 
 def intialize_chroma_vectorstore():
     """Initialize and return the Chroma vector store."""
@@ -33,7 +34,10 @@ def web_search_tavily(query: str) -> dict:
         'source', 'page', and 'content'.
     """
     try:
-        search_docs = TavilySearchResults(max_results=3).invoke(input=query)
+        search_docs = TavilySearch(
+            max_results=3,
+            topic="general",
+            ).invoke({"query": query})["results"]
         results = [
             {
                 "title": doc.get("title", ""),

requirements.txt

@@ -14,4 +14,7 @@ langchain_text_splitters
 pymupdf-layout
 sentence_transformers
 gradio
-python-dotenv
+python-dotenv
+langchain-tavily
+arxiv
+wikipedia