使用 DeepSeek API 开发简易对话机器人

# -*- coding: utf-8 -*-
# @Time    : 2025/3/7 20:38
# @Author  : AI悦创
# @FileName: deepseekapi.py
# @Software: PyCharm
# @Blog    ：https://bornforthis.cn/
# code is far away from bugs with the god animal protecting
#    I love animals. They taste delicious.
# Please install OpenAI SDK first: `pip3 install openai`

from openai import OpenAI

api_key = "sk-b5b6d742d6ed4daea4a75ec6fe0b483b"  # <DeepSeek API Key>
client = OpenAI(api_key=api_key, base_url="https://api.deepseek.com")

response = client.chat.completions.create(
    model="deepseek-chat",
    messages=[
        {"role": "system", "content": "您是个乐于助人的助手"},
        {"role": "user", "content": "你好，我是悦创。"},
    ],
    stream=False
)

print(response.choices[0].message.content)

# main.py

from openai import OpenAI
from config import API_KEY, BASE_URL, MODEL

# 初始化客户端
client = OpenAI(api_key=API_KEY, base_url=BASE_URL)

# 初始上下文
messages = [
    {"role": "system", "content": "你是一个乐于助人的中文 AI 助手。"}
]

def chat():
    print("🤖 DeepSeek 聊天助手已启动（输入 exit 退出）")
    while True:
        user_input = input("🧑 你：")
        if user_input.lower() in ["exit", "退出", "bye"]:
            print("👋 再见！")
            break

        # 添加用户输入
        messages.append({"role": "user", "content": user_input})

        try:
            # 生成回复
            response = client.chat.completions.create(
                model=MODEL,
                messages=messages,
                stream=False
            )
            reply = response.choices[0].message.content.strip()

            # 打印助手回复
            print("🤖 助手：" + reply)

            # 添加助手回复
            messages.append({"role": "assistant", "content": reply})
        except Exception as e:
            print(f"❌ 出错了: {e}")
            break

if __name__ == "__main__":
    chat()

🤖 DeepSeek 聊天助手已启动（输入 exit 退出）
🧑 你：我是AI悦创，你呢？
🤖 助手：你好，AI悦创！😊 我是 **DeepSeek Chat**，由深度求索公司开发的智能助手。我的目标是为你提供各种有用的信息、解答问题，或者陪你聊天！无论是学习、工作，还是日常生活中的疑问，都可以来找我哦～  

很高兴认识你！有什么我可以帮你的吗？✨
🧑 你：给我一个你的API代码
🤖 助手：你可以通过 **DeepSeek AI 官方 API** 来调用我的服务。以下是一个简单的 Python 示例代码，展示如何使用 API 与我进行交互：

### **Python 示例代码（使用 requests 库）**
```python
import requests

# 替换为你的 API Key（如果有的话，目前 DeepSeek Chat 可能仍处于免费阶段）
API_KEY = "your_api_key_here"  # 如果需要的话
API_URL = "https://api.deepseek.com/v1/chat/completions"  # 示例地址，请以官方文档为准

headers = {
    "Authorization": f"Bearer {API_KEY}",
    "Content-Type": "application/json"
}

data = {
    "model": "deepseek-chat",  # 或其他可用模型
    "messages": [
        {"role": "user", "content": "你好！请介绍一下你自己。"}
    ],
    "temperature": 0.7,  # 控制回答的随机性（0-1）
    "max_tokens": 1024   # 限制返回的最大 token 数
}

response = requests.post(API_URL, headers=headers, json=data)

if response.status_code == 200:
    print(response.json()["choices"][0]["message"]["content"])
else:
    print("请求失败:", response.text)
```

### **注意事项**
1. **API 地址和认证方式** 可能会变化，请查阅 [DeepSeek 官方 API 文档](https://platform.deepseek.com/docs) 获取最新信息。
2. **API Key**：如果 DeepSeek 提供 API 访问，你可能需要申请 API Key。
3. **可用模型**：目前可能支持 `deepseek-chat` 或其他版本，请根据官方文档调整。

### **Node.js 示例（如果需要）**
如果你想要 Node.js 版本的代码，可以这样写：
```javascript
const axios = require('axios');

const API_KEY = "your_api_key_here";
const API_URL = "https://api.deepseek.com/v1/chat/completions";

const data = {
    model: "deepseek-chat",
    messages: [{ role: "user", content: "你好！你是谁？" }],
    temperature: 0.7,
    max_tokens: 1024,
};

axios.post(API_URL, data, {
    headers: {
        "Authorization": `Bearer ${API_KEY}`,
        "Content-Type": "application/json",
    },
})
.then(response => console.log(response.data.choices[0].message.content))
.catch(error => console.error("请求失败:", error.response?.data || error.message));
```

如果你需要更详细的 API 参数或遇到问题，可以告诉我，我会尽力帮你优化代码！🚀
🧑 你：你还记得，我是谁吗？
🤖 助手：当然记得！你是 **AI悦创**，我们刚刚才互相介绍过呢～ 😊  

不过作为AI，虽然我可以根据当前对话上下文记住你的身份，但每次新的对话开始时，我的记忆会重置（不保留之前的聊天记录）。所以如果你稍后重新打开聊天，可能需要再次提醒我哦！  

有什么需要帮忙的吗？我会尽力协助你！✨
🧑 你：exit
👋 再见！

# writer.py

from openai import OpenAI
from config import API_KEY, BASE_URL, MODEL
import datetime

client = OpenAI(api_key=API_KEY, base_url=BASE_URL)

styles = {
    "1": "日常风格",
    "2": "科幻风格",
    "3": "技术说明风格"
}

def get_style_prompt(style_choice):
    if style_choice == "1":
        return "请用日常生活化的语气续写下文："
    elif style_choice == "2":
        return "请用科幻小说风格续写下文："
    elif style_choice == "3":
        return "请用专业技术文档风格续写下文："
    else:
        return "请续写下文："

def generate_writing(style_prompt, user_input):
    messages = [
        {"role": "system", "content": "你是一个专业写作助手，擅长多种文风写作。"},
        {"role": "user", "content": f"{style_prompt}\n\n{user_input}"}
    ]

    response = client.chat.completions.create(
        model=MODEL,
        messages=messages,
        stream=False
    )
    return response.choices[0].message.content.strip()

def save_output(content):
    timestamp = datetime.datetime.now().strftime("%Y%m%d_%H%M%S")
    filename = f"output_{timestamp}.md"
    with open(filename, "w", encoding="utf-8") as f:
        f.write(content)
    print(f"✅ 写作内容已保存至：{filename}")

def main():
    print("✍️ AI 写作助手启动")
    print("请选择写作风格：")
    for k, v in styles.items():
        print(f"{k}. {v}")

    style_choice = input("请输入数字选择风格：").strip()
    user_input = input("请输入你想写的主题或段落开头：\n").strip()

    style_prompt = get_style_prompt(style_choice)
    result = generate_writing(style_prompt, user_input)

    print("\n📝 生成内容：\n")
    print(result)

    save_output(result)

if __name__ == "__main__":
    main()

import threading
import time
import sys

# 动画线程函数
def show_loading(stop_event):
    i = 0
    loading_chars = ['.', '..', '...', '....']
    while not stop_event.is_set():
        sys.stdout.write('\r🤖 助手正在思考中' + loading_chars[i % len(loading_chars)])
        sys.stdout.flush()
        i += 1
        time.sleep(0.5)
    sys.stdout.write('\r' + ' ' * 30 + '\r')  # 清空行

# 主聊天循环中修改部分
        try:
            # 启动动画线程
            stop_event = threading.Event()
            loading_thread = threading.Thread(target=show_loading, args=(stop_event,))
            loading_thread.start()

            # 生成回复
            response = client.chat.completions.create(
                model=MODEL,
                messages=messages,
                stream=False
            )

            # 停止动画
            stop_event.set()
            loading_thread.join()

            reply = response.choices[0].message.content.strip()
            print("🤖 助手：" + reply)

def generate_writing(style_prompt, user_input):
    import threading, sys, time

    def show_loading(stop_event):
        i = 0
        dots = ['.', '..', '...', '....']
        while not stop_event.is_set():
            sys.stdout.write('\r⏳ AI 正在写作中' + dots[i % len(dots)])
            sys.stdout.flush()
            i += 1
            time.sleep(0.5)
        sys.stdout.write('\r' + ' ' * 30 + '\r')  # 清空行

    messages = [
        {"role": "system", "content": "你是一个专业写作助手，擅长多种文风写作。"},
        {"role": "user", "content": f"{style_prompt}\n\n{user_input}"}
    ]

    # 启动 loading 动画
    stop_event = threading.Event()
    t = threading.Thread(target=show_loading, args=(stop_event,))
    t.start()

    # 发起请求
    response = client.chat.completions.create(
        model=MODEL,
        messages=messages,
        stream=False
    )

    # 停止 loading 动画
    stop_event.set()
    t.join()

    return response.choices[0].message.content.strip()

世界名称：格雷姆·齿轮纪元

世界背景：
在一个由浮空岛屿组成的世界中，古老的蒸汽科技驱动着一切。天空被云层永远遮蔽，阳光只能穿透机械之神允许的缝隙。

科技系统：
- 动力来源：永动蒸汽核心
- 通信方式：飞鸽机械鸟
- 核心交通工具：巨型空中飞船“齿轮鲸”

种族设定：
- 机灵族：半机械半有机体，体内由齿轮与神经协作控制。
- 烬人族：曾是人类，如今依靠蒸汽心脏维持生命。

历史大事件：
- 齿轮圣战（146年）：机灵族与人类围绕“神圣主引擎”爆发的战争。

from openai import OpenAI
from config import API_KEY, BASE_URL, MODEL
import datetime

client = OpenAI(api_key=API_KEY, base_url=BASE_URL)

def build_world(keywords):
    messages = [
        {"role": "system", "content": "你是一个世界观设计师，擅长创造详细的虚拟世界背景、种族、科技和历史。"},
        {"role": "user", "content": f"请根据以下关键词构建一个原创幻想世界：{keywords}。\n请按以下格式输出：\n1. 世界名称\n2. 世界背景\n3. 种族设定\n4. 科技/魔法系统\n5. 地理区域\n6. 历史事件"}
    ]
    response = client.chat.completions.create(
        model=MODEL,
        messages=messages,
        stream=False
    )
    return response.choices[0].message.content.strip()

def save_world(content):
    filename = f"world_{datetime.datetime.now().strftime('%Y%m%d_%H%M%S')}.md"
    with open(filename, "w", encoding="utf-8") as f:
        f.write(content)
    print(f"✅ 世界观文档已保存为：{filename}")

def main():
    print("🌍 AI 世界观生成器")
    keywords = input("请输入关键词（多个用逗号分隔）：")
    world = build_world(keywords)
    print("\n🧾 生成的世界设定：\n")
    print(world)
    save_world(world)

if __name__ == "__main__":
    main()