from openai import OpenAI
# 初始化 阶跃星辰 Client
STEPFUN_KEY = ""
client = OpenAI(base_url="https://api.stepfun.com/v1", api_key=STEPFUN_KEY)
# 调用补全接口进行补全
stream = client.chat.completions.create(
model="step-1-8k",
messages=[{"role": "user", "content": "全季酒店怎么样?"}],
stream=True,
)
# 对流式返回的内容进行打印 / 渲染输出
for chunk in stream:
if chunk.choices[0].delta.content is not None:
print(chunk.choices[0].delta.content, end="")