koishi-plugin-deepseekapi-chat 1.0.0

package/package.json

@@ -0,0 +1,41 @@
+{
+  "name": "koishi-plugin-deepseekapi-chat",
+  "version": "1.0.0",
+  "description": "高性能 DeepSeek 大模型接入插件，支持自定义人设、多轮对话记忆及详细参数调节",
+  "main": "./src/index.js",
+  "scripts": {
+    "test": "echo \"Error: no test specified\" && exit 1"
+  },
+  "keywords": [
+    "koishi",
+    "plugin",
+    "deepseek",
+    "ai",
+    "chat",
+    "llm",
+    "context",
+    "rpg"
+  ],
+  "author": "YourName",
+  "license": "MIT",
+  "repository": {
+    "type": "git",
+    "url": ""
+  },
+  "koishi": {
+    "description": {
+      "en": "High-performance DeepSeek AI integration with customizable persona, context memory, and advanced parameter tuning.",
+      "zh": "高性能 DeepSeek 大模型接入插件。支持自定义系统人设、多轮对话记忆（可配置轮数）、温度/惩罚系数调节，提供 .ds/.ai 指令，适合群聊智能助手。"
+    },
+    "service": {
+      "required": [],
+      "optional": []
+    }
+  },
+  "peerDependencies": {
+    "koishi": "^4.16.0"
+  },
+  "engines": {
+    "node": ">=18.0.0"
+  }
+}

package/src/index.js

@@ -0,0 +1,183 @@
+const { Context } = require('koishi');
+
+exports.name = 'deepseekapi-chat';
+exports.usage = '使用 .ds <问题> 向 DeepSeek 提问，支持通过配置文件自定义人设和参数';
+
+// 定义插件的配置架构
+// 这里的 schema 定义了网页后台的配置表单结构，同时也作为默认值的来源
+exports.Config = (ctx) => ({
+  // --- 核心认证 ---
+  apiKey: '', 
+  endpoint: 'https://api.deepseek.com',
+  
+  // --- 模型设置 ---
+  model: 'deepseek-chat',
+  
+  // --- 人设与提示词 (核心修改点) ---
+  // 默认系统提示词，可以在这里写很长的 Prompt
+  systemPrompt: '你是一个 helpful 的 AI 助手。回答请简洁、幽默，适合 QQ 群聊环境。如果用户问的是代码问题，请直接给出代码片段并简要解释。',
+  
+  // --- 对话上下文管理 ---
+  // 最大保留的对话轮数 (1轮 = 1问1答)。设置为 0 则不保留历史，每次都是新对话。
+  maxContextRounds: 10, 
+  // 是否在每次请求时强制重新注入 systemPrompt (建议开启，防止被历史记录冲淡)
+  injectSystemPromptEveryTime: true,
+
+  // --- 生成参数 (LLM Hyperparameters) ---
+  // 创造性 (0-2)。越高越随机，越低越严谨。
+  temperature: 0.7,
+  // 核采样概率 (0-1)。与 temperature 配合使用。
+  top_p: 0.9,
+  // 最大生成令牌数
+  max_tokens: 1024,
+  // 频率惩罚 (-2 到 2)。降低重复度。
+  frequency_penalty: 0,
+  // 存在惩罚 (-2 到 2)。鼓励谈论新话题。
+  presence_penalty: 0,
+
+  // --- 行为控制 ---
+  // 是否启用流式输出 (目前命令模式暂不支持流式回传，但保留配置供未来扩展或日志记录)
+  stream: false,
+  // 出错时是否显示详细调试信息 (生产环境建议关闭)
+  verboseError: false
+});
+
+exports.apply = (ctx, config) => {
+  // 存储每个用户的上下文
+  // 结构：{ 'session_id': [{role, content}, ...] }
+  const sessions = new Map();
+
+  ctx.command('ds <text:text>', '向 DeepSeek 提问')
+    .alias('ai', 'ask', 'deepseek')
+    .option('reset', '-r, --reset', { desc: '清空当前会话历史' })
+    .action(async ({ session, options }, text) => {
+      
+      // 1. 基础检查
+      if (!text && !options.reset) {
+        return '请提供问题，例如：.ds 今天天气怎么样？\n使用 .ds -r 可清空记忆。';
+      }
+
+      if (!config.apiKey) {
+        return '❌ 插件未配置 API Key，请在配置文件或网页后台填入 deepseek-chat.apiKey。';
+      }
+
+      // 2. 生成唯一会话 ID
+      // 策略：群聊中区分用户 (group:gid:user:uid)，私聊直接用 uid
+      const sessionId = session.groupId 
+        ? `group:${session.groupId}:user:${session.userId}` 
+        : `user:${session.userId}`;
+
+      // 3. 处理重置选项
+      if (options.reset) {
+        sessions.delete(sessionId);
+        return '✅ 已清空当前的对话记忆，我们可以重新开始聊了！';
+      }
+
+      // 初始化或获取历史
+      if (!sessions.has(sessionId)) {
+        sessions.set(sessionId, []);
+      }
+      const history = sessions.get(sessionId);
+
+      // 4. 更新历史记录
+      // 只有当用户真的发了文字时才加入历史 (reset 选项不加入)
+      if (text) {
+        history.push({ role: 'user', content: text });
+        
+        // 限制历史记录长度
+        // 注意：maxContextRounds 是“轮数”，每轮包含 1 个 user 和 1 个 assistant
+        const maxMessages = config.maxContextRounds * 2;
+        
+        if (config.maxContextRounds > 0 && history.length > maxMessages) {
+          // 移除最旧的记录，直到符合长度
+          history.splice(0, history.length - maxMessages);
+        }
+      }
+
+      // 5. 构造发送给 API 的消息体
+      const messages = [];
+
+      // 添加系统提示词
+      // 策略：如果配置了 injectSystemPromptEveryTime，每次都加在最前面
+      // 否则，只在历史记录为空时加一次 (这种策略较少用，因为历史记录可能被截断导致丢失人设)
+      if (config.systemPrompt) {
+        messages.push({ role: 'system', content: config.systemPrompt });
+      }
+
+      // 添加历史对话
+      messages.push(...history);
+
+      // 6. 调用 DeepSeek API
+      try {
+        const response = await fetch(`${config.endpoint}/v1/chat/completions`, {
+          method: 'POST',
+          headers: {
+            'Content-Type': 'application/json',
+            'Authorization': `Bearer ${config.apiKey}`
+          },
+          body: JSON.stringify({
+            model: config.model,
+            messages: messages,
+            stream: config.stream,
+            max_tokens: config.max_tokens,
+            temperature: config.temperature,
+            top_p: config.top_p,
+            frequency_penalty: config.frequency_penalty,
+            presence_penalty: config.presence_penalty,
+            // DeepSeek 特有或通用参数可根据需要添加
+          })
+        });
+
+        if (!response.ok) {
+          let errDetail = '';
+          try {
+            const errData = await response.json();
+            errDetail = errData.error?.message || JSON.stringify(errData);
+          } catch (e) {
+            errDetail = await response.text();
+          }
+          
+          const errorMsg = config.verboseError 
+            ? `API 错误 (${response.status}): ${errDetail}` 
+            : `请求失败 (状态码 ${response.status})`;
+            
+          throw new Error(errorMsg);
+        }
+
+        const data = await response.json();
+        
+        if (!data.choices || data.choices.length === 0) {
+          throw new Error('API 返回了空响应');
+        }
+
+        const aiContent = data.choices[0].message.content;
+
+        // 7. 将 AI 回复加入历史 (仅当成功时)
+        if (text) { // 确保是因为有用户输入才记录的
+           history.push({ role: 'assistant', content: aiContent });
+        }
+
+        return aiContent;
+
+      } catch (error) {
+        console.error('[DeepSeek Chat] Error:', error.message);
+        // 如果是网络错误或 API 密钥错误，可以特殊处理
+        if (error.message.includes('401')) {
+          return '❌ API Key 无效或已过期，请检查配置。';
+        }
+        if (error.message.includes('429')) {
+          return '⏳ 请求过于频繁，请稍后再试。';
+        }
+        return `❌ 发生错误：${error.message}`;
+      }
+    });
+    
+  // 可选：添加一个管理员命令来热重载配置或查看状态
+  if (ctx.config.admin) {
+     ctx.command('deepseek-status', '查看 DeepSeek 插件状态')
+       .action(() => {
+         const count = sessions.size;
+         return `🤖 DeepSeek 插件运行中\n当前活跃会话数: ${count}\n模型: ${config.model}\n上下文轮数限制: ${config.maxContextRounds}`;
+       });
+  }
+};