複製代码

1. # 初始化 OpenAI 客户端
2. from openai import OpenAI
4. text = input("请输入对话:\n")
5. print(" 正在AI对话... 请稍等.....")
7. # 请替换为你的 API 密钥
8. client = OpenAI(api_key="sk-xx", base_url="https://api.deepseek.com")
10. messages = [{"role": "user", "content": text}]
12. # 使用 stream=True 来逐行获取响应
13. response = client.chat.completions.create(
14.     model="deepseek-chat",  # 请确保模型名称正确
15.     messages=messages,
16.     stream=True  # 启用流式响应
17. )
19. # 逐行显示响应内容
20. print("AI回复:")
21. for chunk in response:
22.     if chunk.choices[0].delta.content:  # 检查是否有内容
23.         print(chunk.choices[0].delta.content, end="", flush=True)  # 逐行显示
24. print()  # 换行
26. while True:
27.     text = input("追加对话:\n")
28.     messages.append({"role": "user", "content": text})
29.     response = client.chat.completions.create(
30.         model="deepseek-chat",
31.         messages=messages,
32.         stream=True
33.     )
35.     # 逐行显示响应内容
36.     print("AI回复:")
37.     for chunk in response:
38.         if chunk.choices[0].delta.content:  # 检查是否有内容
39.             print(chunk.choices[0].delta.content, end="", flush=True)  # 逐行显示
40.     print()  # 换行

複製代码