update tools.py

tools.py

@@ -1,21 +1,22 @@
 
 
-from langchain_community.tools import WikipediaQueryRun, ArxivQueryRun
-from langchain_community.utilities import WikipediaAPIWrapper,ArxivAPIWrapper
-from langchain_community.tools.tavily_search import TavilySearchResults
+from langchain_community.tools import WikipediaQueryRun, ArxivQueryRun, DuckDuckGoSearchRun
+from langchain_community.utilities import WikipediaAPIWrapper, ArxivAPIWrapper, DuckDuckGoSearchAPIWrapper
+
 from langchain_core.tools import tool
 from datetime import datetime
-
+@tool
 def wikipedia_tool() -> WikipediaQueryRun:
     """best search for biography, facts and history"""
     return WikipediaQueryRun(
-        api_wrapper=WikipediaAPIWrapper(doc_content_chars_max= 4000)
+        api_wrapper=WikipediaAPIWrapper(top_k_results=2, doc_content_chars_max= 4000)
     )
-
-def tavily_tool() -> TavilySearchResults:
-    """web search for recents events and facts"""
-    return TavilySearchResults(max_results=5)
-
+@tool
+def ddg_search_tool() -> DuckDuckGoSearchRun:
+    """DuckDuckGo search"""
+    wrapper = DuckDuckGoSearchAPIWrapper(max_results=5)
+    return DuckDuckGoSearchRun(api_wrapper=wrapper)
+@tool
 def arxiv_tool() -> ArxivQueryRun:
     """returns scientific and research papers"""
     return ArxivQueryRun(
@@ -26,3 +27,5 @@ def arxiv_tool() -> ArxivQueryRun:
 def get_current_year() -> str:
     """returns the current year"""
     return str(datetime.now().year)
+
+