tools.py

import json
from urllib.parse import quote_plus
from urllib.request import Request, urlopen

from crewai.tools import tool


def _fetch_json(url: str) -> dict:
	request = Request(url, headers={"User-Agent": "Mozilla/5.0"})
	with urlopen(request, timeout=12) as response:
		return json.loads(response.read().decode("utf-8"))


@tool("wikipedia_summary")
def wikipedia_summary(topic: str) -> str:
	"""Return a short Wikipedia summary for a topic."""
	if not topic.strip():
		return "No topic provided."

	url = (
		"https://en.wikipedia.org/api/rest_v1/page/summary/"
		f"{quote_plus(topic.strip())}"
	)

	try:
		data = _fetch_json(url)
		summary = data.get("extract") or "No summary found on Wikipedia."
		title = data.get("title", topic)
		return f"{title}: {summary}"
	except Exception as exc:
		return f"Wikipedia lookup failed: {exc}"


@tool("duckduckgo_instant_answer")
def duckduckgo_instant_answer(query: str) -> str:
	"""Return quick facts/snippets from DuckDuckGo instant answer API."""
	if not query.strip():
		return "No query provided."

	url = (
		"https://api.duckduckgo.com/?format=json&no_html=1&skip_disambig=1&q="
		f"{quote_plus(query.strip())}"
	)

	try:
		data = _fetch_json(url)
		abstract = data.get("AbstractText", "").strip()
		heading = data.get("Heading", "").strip()

		if abstract:
			if heading:
				return f"{heading}: {abstract}"
			return abstract

		related = data.get("RelatedTopics", [])
		snippets = []
		for item in related[:5]:
			text = item.get("Text") if isinstance(item, dict) else None
			if text:
				snippets.append(f"- {text}")

		if snippets:
			return "No direct abstract. Related snippets:\n" + "\n".join(snippets)
		return "No instant answer found."
	except Exception as exc:
		return f"DuckDuckGo lookup failed: {exc}"