Spaces:
Sleeping
Sleeping
| from smolagents.tools import Tool | |
| import wikipedia | |
| #@tool | |
| class WikipediaLookupTool(Tool): | |
| name = "wikipedia_lookup" | |
| description = "Look up content from the English Wikipedia based on a query string." | |
| def __call__(self, query: str) -> str: | |
| try: | |
| page = wikipedia.page(query) | |
| return page.content # full text | |
| except wikipedia.DisambiguationError as e: | |
| return f"Disambiguation error. Options: {e.options[:5]}" | |
| except wikipedia.PageError: | |
| return f"Page not found for: {query}" | |
| except Exception as e: | |
| return f"Unexpected error: {e}" |