Python 调用

使用 OpenAI 官方 SDK（推荐）

pip install openai

from openai import OpenAI

client = OpenAI(
    api_key="sk-heima-xxxxxxxx",
    base_url="https://api.token.heimatianqi.com/v1",
)

resp = client.chat.completions.create(
    model="deepseek-flash",
    messages=[{"role": "user", "content": "你好"}],
)
print(resp.choices[0].message.content)
print(resp.usage)

流式输出

stream = client.chat.completions.create(
    model="deepseek-flash",
    messages=[{"role": "user", "content": "用三句话介绍你自己"}],
    stream=True,
    stream_options={"include_usage": True},
)
for chunk in stream:
    if chunk.choices and chunk.choices[0].delta.content:
        print(chunk.choices[0].delta.content, end="", flush=True)

不依赖 SDK（标准库）

import json, os, urllib.request

BASE = os.environ.get("HEIMA_BASE_URL", "https://api.token.heimatianqi.com/v1")
KEY = os.environ["HEIMA_API_KEY"]

req = urllib.request.Request(
    BASE + "/chat/completions",
    data=json.dumps({
        "model": "mock-chat",
        "messages": [{"role": "user", "content": "你好"}],
    }).encode(),
    headers={"Authorization": f"Bearer {KEY}", "Content-Type": "application/json"},
    method="POST",
)
with urllib.request.urlopen(req, timeout=60) as resp:
    data = json.loads(resp.read())
print(data["choices"][0]["message"]["content"][:50])
print("usage:", data["usage"])