Update tools.py

tools.py

@@ -1,5 +1,6 @@
 from langchain.tools import tool
 from langchain_community.tools.ddg_search import DuckDuckGoSearchRun
+import wikipedia
 
 search = DuckDuckGoSearchRun()
 
@@ -8,4 +9,17 @@ def web_search(query: str) -> str:
     """Perform a web search using DuckDuckGo."""
     return search.run(query)
 
-tools = [web_search]
+@tool
+def wiki_search(query: str) -> str:
+    """Search Wikipedia and return a summary of the most relevant result."""
+    try:
+        page = wikipedia.page(query)
+        return page.summary
+    except wikipedia.DisambiguationError as e:
+        return f"Disambiguation: {', '.join(e.options[:5])}"
+    except wikipedia.PageError:
+        return "Page not found."
+    except Exception as e:
+        return f"Error: {e}"
+
+tools = [web_search, wiki_search]