import os, requests
HEADERS = {"Authorization": f"Bearer {os.environ['NINJACHAT_API_KEY']}"}
def search(query, group="web", depth="basic"):
r = requests.post("https://ninjachat.ai/api/v1/search",
headers=HEADERS,
json={
"query": query,
"group": group,
"search_depth": depth,
"include_answer": True,
}
)
r.raise_for_status()
return r.json()
# Web search
result = search("best Python web frameworks 2026")
print(result["answer"])
for src in result["sources"][:3]:
print(f" - {src['title']}: {src['url']}")
# Academic search
result = search("transformer architecture improvements", group="academic")
print(result["answer"])
# News search
result = search("AI regulation", group="news")
print(result["answer"])