可执行任务接口自动化接入指南工业级高并发推理接口 · 极简接入技术手册

from openai import OpenAI

# 初始化 DeepSeek 客户端，无缝复用官方生态
client = OpenAI(
    api_key="sk-your-deepseek-api-key",
    base_url="https://api.deepseek.com"
)

response = client.chat.completions.create(
    model="deepseek-reasoner", # 对应 R1 显式思维链推理模型
    messages=[
        {"role": "user", "content": "写一个符合 MLA 潜在注意力机制计算流程的 PyTorch 前向推演片段"}
    ],
    stream=True
)

for chunk in response:
    if chunk.choices[0].delta.reasoning_content:
        print(chunk.choices[0].delta.reasoning_content, end="") # 流式输出思维链
    elif chunk.choices[0].delta.content:
        print(chunk.choices[0].delta.content, end="")