Spaces:
Running on Zero
Running on Zero
| from urllib.parse import quote | |
| import requests | |
| from smolagents import tool | |
| def search_wikipedia(query: str) -> str: | |
| """Fetches an encyclopedic summary of a topic from Wikipedia. | |
| Args: | |
| query: The subject or topic name to search on Wikipedia (e.g., 'Alan Turing', 'Artificial intelligence'). | |
| """ | |
| try: | |
| encoded_query = quote(query) | |
| url = f"https://en.wikipedia.org/api/rest_v1/page/summary/{encoded_query}" | |
| headers = {"User-Agent": "HF-Agents-Course-ArkAssistant/1.0"} | |
| resp = requests.get(url, headers=headers, timeout=10) | |
| if resp.status_code == 200: | |
| data = resp.json() | |
| extract = data.get("extract", "") | |
| title = data.get("title", query) | |
| if extract: | |
| return f"**{title}**: {extract}" | |
| return f"No extract available for '{query}'." | |
| if resp.status_code == 404: | |
| return f"No Wikipedia article found for '{query}'." | |
| return f"Wikipedia query returned status code {resp.status_code}." | |
| except Exception as e: | |
| return f"Error querying Wikipedia for '{query}': {str(e)}" | |