import json, os, urllib.request

base = os.environ.get("MISTRAL_API_BASE", "https://api.mistral.ai/v1")
payload = {
    "model": "mistral-large-4",
    "messages": [{"role": "user", "content": "Explain mixture-of-experts in two sentences."}],
    "max_tokens": 128,
}
request = urllib.request.Request(
    base.rstrip("/") + "/chat/completions",
    data=json.dumps(payload).encode(),
    headers={"Authorization": "Bearer " + os.environ["MISTRAL_API_KEY"],
             "Content-Type": "application/json"},
)
with urllib.request.urlopen(request, timeout=60) as response:
    result = json.load(response)
print(result["choices"][0]["message"]["content"])
print(json.dumps(result.get("usage", {})))
