Update tool.py

tool.py

@@ -1,10 +1,10 @@
 from langchain_core.tools import tool
 from langchain_community.utilities.duckduckgo_search import DuckDuckGoSearchAPIWrapper
+from langchain_community.tools.tavily_search import TavilySearchResults
 from langchain_community.document_loaders.wikipedia import WikipediaLoader
 from langchain_community.document_loaders.arxiv import ArxivLoader
 from langchain_community.document_loaders.pubmed import PubMedLoader
 
-
 ## Simple algebra tools
 @tool
 def add(a: float, b: float) -> float: 
@@ -52,7 +52,7 @@ def divide(a: float, b: float) -> float:
 
 ## Search Tools
 def DuckDuckGoSearchTool(query: str) -> str:
-    """Search DuckDuckGo for a query and return maximum 3 results.
+    """Search DuckDuckGo for a query and return maximum 5 results.
     
     Args:
         query: The search query.
@@ -65,6 +65,20 @@ def DuckDuckGoSearchTool(query: str) -> str:
         ])
     return {"web_results": formatted_search_docs}
 
+@tool
+def TavilySearchTool(query: str) -> str:
+    """Search Tavily for a query and return maximum 3 results.
+    
+    Args:
+        query: The search query."""
+    search_docs = TavilySearchResults(max_results=5).invoke(query=query)
+    formatted_search_docs = "\n\n---\n\n".join(
+        [
+            f'<Document source="{doc.metadata["source"]}" page="{doc.metadata.get("page", "")}"/>\n{doc.page_content}\n</Document>'
+            for doc in search_docs
+        ])
+    return {"web_results": formatted_search_docs}
+
 @tool
 def WikipediaSearchTool(query: str) -> str:
     """Search Wikipedia for a query and return maximum 2 results.