# Python 3. Set XAI_API_KEY, XAI_MODEL and QANUN_ACCESS_TOKEN privately.
import json, os, urllib.request
payload = {
    "model": os.environ["XAI_MODEL"],
    "input": "Find Health Care Commission legislation. Preserve source links and coverage limitations.",
    "tools": [{"type": "mcp", "server_label": "qanun_maps",
               "server_url": "https://www.jusnrem.legal/mcp",
               "headers": {"Authorization": "Bearer " + os.environ["QANUN_ACCESS_TOKEN"]},
               "allowed_tools": ["find_legislation"]}]
}
request = urllib.request.Request("https://api.x.ai/v1/responses",
    data=json.dumps(payload).encode(), headers={"Content-Type": "application/json",
    "Authorization": "Bearer " + os.environ["XAI_API_KEY"]})
with urllib.request.urlopen(request, timeout=90) as response:
    print(response.read().decode())

