第三章：概念升级与 Chatbot 实战¶

# chatbot.py —— 带记忆的终端 Chatbot

import os
from openai import OpenAI
from dotenv import load_dotenv

load_dotenv()

client = OpenAI(
    api_key=os.getenv("DEEPSEEK_API_KEY"),
    base_url=os.getenv("DEEPSEEK_BASE_URL"),
)

# 初始化消息历史，system 消息设定 AI 的角色
messages = [
    {
        "role": "system",
        "content": "你是一个友好、耐心的 AI 助手。请用简洁清晰的语言回答问题。"
    }
]

print("=" * 50)
print("🤖 Chatbot 已启动！输入 quit 或 exit 退出。")
print("=" * 50)

while True:
    # 获取用户输入
    user_input = input("\n你：").strip()

    # 退出条件
    if user_input.lower() in ("quit", "exit", "退出", "q"):
        print("\n再见！下次见～ 👋")
        break

    # 跳过空输入
    if not user_input:
        continue

    # 将用户消息追加到历史
    messages.append({"role": "user", "content": user_input})

    try:
        # 发送完整历史，获取回复
        response = client.chat.completions.create(
            model="deepseek-chat",
            messages=messages,
            temperature=0.7,
            max_tokens=1000,
        )

        # 提取回复
        assistant_reply = response.choices[0].message.content

        # 将 AI 回复也追加到历史（下次调用时一并发送）
        messages.append({"role": "assistant", "content": assistant_reply})

        print(f"\nAI：{assistant_reply}")

    except Exception as e:
        print(f"\n⚠️  请求失败：{e}")
        # 请求失败时移除刚才添加的 user 消息，避免历史污染
        messages.pop()