Rename tools.py to wikipedia_searcher.py

tools.py

@@ -1,61 +0,0 @@
-from smolagents import Tool
-import requests
-
-def fetch_wikipedia_article(query: str) -> str:
-    url = "https://en.wikipedia.org/w/api.php"
-    params = {
-        "action": "query",
-        "prop": "extracts",
-        "explaintext": True,
-        "format": "json",
-        "titles": query,
-    }
-    response = requests.get(url, params=params)
-    data = response.json()
-    pages = data.get("query", {}).get("pages", {})
-    page = next(iter(pages.values()))
-    return page.get("extract", "No content found for this query.")
-
-def search_wikipedia(query: str) -> str:
-    url = "https://en.wikipedia.org/w/api.php"
-    params = {
-        "action": "query",
-        "list": "search",
-        "srsearch": query,
-        "format": "json",
-    }
-    response = requests.get(url, params=params)
-    data = response.json()
-    results = data.get("query", {}).get("search", [])
-    if results:
-        return "\n".join([r['title'] for r in results])
-    return "No search results found."
-
-class WikipediaTool(Tool):
-    name = "WikipediaTool"
-    description = "Fetches plain text from a Wikipedia article. Input should be the article title as a string."
-    inputs = {"query": "string"}
-    output_type = "string"
-
-    def __call__(self, inputs) -> str:
-        if isinstance(inputs, str):
-            inputs = {"query": inputs}
-        elif isinstance(inputs, dict) and "query" not in inputs:
-            inputs = {"query": next(iter(inputs.values()))}
-        return fetch_wikipedia_article(inputs["query"])
-
-class WikipediaSearchTool(Tool):
-    name = "WikipediaSearchTool"
-    description = "Searches Wikipedia for article titles. Input should be a question or topic as a string."
-    inputs = {"query": "string"}
-    output_type = "string"
-
-    def __call__(self, inputs) -> str:
-        if isinstance(inputs, str):
-            inputs = {"query": inputs}
-        elif isinstance(inputs, dict) and "query" not in inputs:
-            inputs = {"query": next(iter(inputs.values()))}
-        return search_wikipedia(inputs["query"])
-
-
-

wikipedia_searcher.py

@@ -0,0 +1,12 @@
+from langchain_community.tools import WikipediaQueryRun
+from langchain_community.utilities import WikipediaAPIWrapper
+
+class WikipediaSearcher:
+    def __init__(self):
+        self.wikipedia = WikipediaQueryRun(api_wrapper=WikipediaAPIWrapper())
+
+    def search(self, query: str) -> str:
+        return self.wikipedia.run(query)
+
+
+