步骤一:安装必要的库,如openai;
步骤二:deepseek平台申请api,并充值(可先充10元),费用大概一个查询2分钱的样子;
步骤三:设置环境变量:DEEPSEEK_API_KEY,安全起见
步骤三:代码如下:
import os
api_key = os.getenv('DEEPSEEK_API_KEY')
from openai import OpenAI
client = OpenAI(api_key=api_key, base_url="https://api.deepseek.com")
# 初始化对话历史(包含系统消息)
messages = [{"role": "system", "content": "You are a helpful assistant"}]
while True:
# 获取用户输入
user_input = input("\nYou: ")
# 退出条件
if user_input.lower() in ["exit", "quit"]:
print("对话结束")
break
# 添加用户消息到历史
messages.append({"role": "user", "content": user_input})
try:
# 发起流式请求
response = client.chat.completions.create(
model="deepseek-reasoner", # 可替换为deepseek-chat,v3模型,当前r1
messages=messages,
stream=True # 启用流式输出
)
# 初始化回复收集
full_response = []
print("\nAssistant: ", end="", flush=True) # 先打印前缀
# 实时处理流式响应
for chunk in response:
content = chunk.choices[0].delta.content
if content: # 过滤空内容
print(content, end="", flush=True) # 逐字打印
full_response.append(content)
# 添加完整回复到历史
ai_response = "".join(full_response)
messages.append({"role": "assistant", "content": ai_response})
print() # 换行
except Exception as e:
print(f"\n发生错误: {str(e)}")
break
即可实现简单问答。
费用可在deepseek开放平台查询: