deepseek-ai/DeepSeek-R1 在线推理调用 API 代码
API 文档cURL
添加令牌为：
import os
import requests
import json
API_URL = "https://api-ai.gitcode.com/v1/chat/completions"
headers = {
"Authorization": f"Bearer {os.environ['ACCESS_TOKEN']}",
}
def query(payload):
response = requests.post(API_URL, headers=headers, json=payload, stream=True)
for line in response.iter_lines():
if not line.startswith(b"data:"):
continue
if line.strip() == b"data:[DONE]" or line.strip() == b"data: [DONE]":
return
yield json.loads(line.decode("utf-8").lstrip("data:").rstrip("/n"))
chunks = query({
"model": "deepseek-ai/DeepSeek-R1",
"messages": [
{
"role": "user",
"content": "告诉我一个有关宇宙的有趣事实？"
}
],
"stream": True
})
for chunk in chunks:
print(chunk["choices"])