|
|
import importlib |
|
|
|
|
|
__all__ = ['WikipediaTool'] |
|
|
|
|
|
|
|
|
class WikipediaTool(): |
|
|
dependencies = ["requests==2.32.3", "beautifulsoup4==4.13.3"] |
|
|
|
|
|
inputSchema = { |
|
|
"name": "WikipediaTool", |
|
|
"description": "Searches Wikipedia for a given question and returns a short summary.", |
|
|
"parameters": { |
|
|
"type": "object", |
|
|
"properties": { |
|
|
"question": { |
|
|
"type": "string", |
|
|
"description": "The topic or question to search on Wikipedia.", |
|
|
} |
|
|
}, |
|
|
"required": ["question"], |
|
|
} |
|
|
} |
|
|
|
|
|
def run(self, **kwargs): |
|
|
question = kwargs.get("question") |
|
|
if not question: |
|
|
return { |
|
|
"status": "error", |
|
|
"message": "Missing required parameter: 'question'", |
|
|
"output": None |
|
|
} |
|
|
|
|
|
print(f"Searching Wikipedia for: {question}") |
|
|
|
|
|
requests = importlib.import_module("requests") |
|
|
bs4 = importlib.import_module("bs4") |
|
|
BeautifulSoup = bs4.BeautifulSoup |
|
|
|
|
|
search_url = "https://en.wikipedia.org/w/api.php" |
|
|
search_params = { |
|
|
"action": "query", |
|
|
"format": "json", |
|
|
"list": "search", |
|
|
"srsearch": question, |
|
|
"srlimit": 1, |
|
|
} |
|
|
|
|
|
try: |
|
|
response = requests.get(search_url, params=search_params) |
|
|
if response.status_code != 200: |
|
|
return { |
|
|
"status": "error", |
|
|
"message": "Wikipedia API request failed.", |
|
|
"output": None, |
|
|
} |
|
|
|
|
|
data = response.json() |
|
|
search_results = data.get("query", {}).get("search", []) |
|
|
|
|
|
if not search_results: |
|
|
return { |
|
|
"status": "error", |
|
|
"message": "No results found on Wikipedia.", |
|
|
"output": None, |
|
|
} |
|
|
|
|
|
top_result = search_results[0]["title"] |
|
|
page_url = f"https://en.wikipedia.org/wiki/{top_result.replace(' ', '_')}" |
|
|
print(f"Fetching full content from: {page_url}") |
|
|
|
|
|
html_url = f"https://en.wikipedia.org/api/rest_v1/page/html/{top_result.replace(' ', '_')}" |
|
|
html_response = requests.get(html_url) |
|
|
|
|
|
if html_response.status_code != 200: |
|
|
return { |
|
|
"status": "error", |
|
|
"message": "Failed to fetch article content.", |
|
|
"output": None, |
|
|
} |
|
|
|
|
|
soup = BeautifulSoup(html_response.text, "html.parser") |
|
|
paragraphs = [p.get_text() for p in soup.find_all("p") if p.get_text()] |
|
|
full_text = " ".join(paragraphs) |
|
|
summary = " ".join(full_text.split(". ")[:5]) |
|
|
|
|
|
output_text = f"**{top_result}**\n{summary}...\n[Read more]({page_url})" |
|
|
|
|
|
return { |
|
|
"status": "success", |
|
|
"message": "Wikipedia article summary retrieved successfully.", |
|
|
"output": output_text, |
|
|
} |
|
|
|
|
|
except Exception as e: |
|
|
return { |
|
|
"status": "error", |
|
|
"message": f"Exception occurred: {str(e)}", |
|
|
"output": None, |
|
|
} |
|
|
|
