wikipedia_searcher.py

from smolagents import Tool
from langchain_community.tools import WikipediaQueryRun 
from langchain_community.utilities import WikipediaAPIWrapper

class WikipediaSearcher(Tool):
    name = "wikipedia_search"
    description = "Search Wikipedia for factual answers to questions."

    def run(self, input_data: dict) -> str:
        query = input_data.get("query", "")
        if not query:
            return "No query provided for Wikipedia search."
        try:
            wikipedia = WikipediaQueryRun(api_wrapper=WikipediaAPIWrapper())
            return wikipedia.run(query)
        except Exception as e:
            return f"Error retrieving Wikipedia data: {e}"