1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
|
import os
from openai import OpenAI
from dotenv import load_dotenv
class DeepSeekChat:
def __init__(self):
load_dotenv()
self.client = OpenAI(
api_key=os.getenv("DEEPSEEK_API_KEY"),
base_url="https://api.deepseek.com"
)
self.history = []
def chat_stream(self, user_input):
"""流式对话"""
self.history.append({"role": "user", "content": user_input})
messages = self.history.copy()
try:
response = self.client.chat.completions.create(
model="deepseek-chat",
messages=messages,
stream=True,
temperature=0.7,
max_tokens=2000
)
full_response = ""
print("\n🤖 DeepSeek: ", end="", flush=True)
for chunk in response:
if chunk.choices[0].delta.content:
content = chunk.choices[0].delta.content
print(content, end="", flush=True)
full_response += content
print("\n")
self.history.append({"role": "assistant", "content": full_response})
except Exception as e:
print(f"\n❌ 错误：{e}")
def chat_once(self, user_input):
"""一次性对话"""
self.history.append({"role": "user", "content": user_input})
try:
response = self.client.chat.completions.create(
model="deepseek-chat",
messages=self.history,
stream=False,
temperature=0.7,
max_tokens=2000
)
reply = response.choices[0].message.content
print(f"\n🤖 DeepSeek: {reply}\n")
self.history.append({"role": "assistant", "content": reply})
except Exception as e:
print(f"\n❌ 错误：{e}")
def clear_history(self):
"""清空对话历史"""
self.history = []
print("✅ 对话历史已清空")
def show_history(self):
"""显示对话历史"""
print("\n" + "="*50)
print("📜 对话历史:")
print("="*50)
for i, msg in enumerate(self.history):
role = "👤 用户" if msg["role"] == "user" else "🤖 AI"
print(f"{i+1}. {role}: {msg['content'][:100]}...")
print("="*50)
def main():
chat = DeepSeekChat()
print("🚀 DeepSeek 交互式对话开始")
print("输入 '退出' 或 'exit' 结束对话")
print("输入 '清空' 或 'clear' 清空历史")
print("输入 '历史' 或 'history' 查看历史")
print("输入 '流式' 或 'stream' 切换流式输出")
print("="*50)
use_stream = False
while True:
user_input = input("\n👤 你：").strip()
if user_input.lower() in ['退出', 'exit', 'quit']:
print("👋 再见！")
break
elif user_input.lower() in ['清空', 'clear']:
chat.clear_history()
continue
elif user_input.lower() in ['历史', 'history']:
chat.show_history()
continue
elif user_input.lower() in ['流式', 'stream']:
use_stream = not use_stream
mode = "流式" if use_stream else "非流式"
print(f"✅ 已切换到{mode}模式")
continue
elif not user_input:
continue
if use_stream:
chat.chat_stream(user_input)
else:
chat.chat_once(user_input)
if __name__ == "__main__":
main()
|
