tools/WikipediaTool.py

from smolagents import Tool
import wikipedia
class LocalWikipediaTool(Tool):
    name = "wikipedia_search"
    description = "Search Wikipedia and return a summary."
    inputs = {
        "query": {"type": "string", "description": "Search term"}
    }
    output_type = "string"

    def forward(self, query: str) -> str:
        try:
            page = wikipedia.page(query)
            return page.content  # full article text
        except Exception:
            return ""