koishi-plugin-openai-compatible 1.0.2 → 1.0.4

package/lib/database.d.ts

@@ -1,29 +0,0 @@
-export interface UserPrompt {
-    userId: string;
-    prompt: string;
-    createdAt: Date;
-    updatedAt: Date;
-}
-export interface ChatHistory {
-    userId: string;
-    messages: {
-        role: 'system' | 'user' | 'assistant';
-        content: string;
-        timestamp: Date;
-    }[];
-    lastUpdated: Date;
-}
-declare const tables: {
-    'openai.userPrompts': {
-        userId: string;
-        prompt: string;
-        createdAt: string;
-        updatedAt: string;
-    };
-    'openai.chatHistories': {
-        userId: string;
-        messages: string;
-        lastUpdated: string;
-    };
-};
-export default tables;

package/lib/database.js

@@ -1,16 +0,0 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-const tables = {
-    'openai.userPrompts': {
-        userId: 'string',
-        prompt: 'text',
-        createdAt: 'timestamp',
-        updatedAt: 'timestamp'
-    },
-    'openai.chatHistories': {
-        userId: 'string',
-        messages: 'json',
-        lastUpdated: 'timestamp'
-    }
-};
-exports.default = tables;

package/lib/index.d.ts

@@ -1,6 +0,0 @@
-import { Context, Schema } from 'koishi';
-import { Config as ConfigInterface } from './config';
-export declare const name = "openai-compatible";
-export declare const usage = "## \uD83E\uDD16 OpenAI\u517C\u5BB9\u804A\u5929\u63D2\u4EF6\n\n\u8BE5\u63D2\u4EF6\u652F\u6301\u4E0E\u6240\u6709\u517C\u5BB9OpenAI API\u7684\u5927\u6A21\u578B\u8FDB\u884C\u804A\u5929\u4EA4\u4E92\u3002\n\n### \u914D\u7F6E\u8BF4\u660E\n\n- **apiEndpoint**: API\u7AEF\u70B9URL\uFF0C\u4F8B\u5982 \"https://api.openai.com/v1\"\n- **apiKey**: API\u5BC6\u94A5\n- **model**: \u4F7F\u7528\u7684\u6A21\u578B\u540D\u79F0\uFF0C\u4F8B\u5982 \"gpt-3.5-turbo\"\n- **temperature**: \u6E29\u5EA6\u53C2\u6570\uFF0C\u63A7\u5236\u56DE\u590D\u7684\u968F\u673A\u6027 (0-2)\n- **maxTokens**: \u6700\u5927\u751F\u6210\u7684token\u6570\u91CF\n- **systemPrompt**: \u7CFB\u7EDF\u9884\u8BBE\u63D0\u793A\u8BCD\uFF08\u53EF\u9009\uFF09\n- **commandName**: \u81EA\u5B9A\u4E49\u6307\u4EE4\u540D\uFF08\u53EF\u9009\uFF0C\u5982\u4E0D\u8BBE\u7F6E\u5219\u9ED8\u8BA4\u5904\u7406\u6240\u6709\u6D88\u606F\uFF09\n- **triggerPrefix**: \u89E6\u53D1\u524D\u7F00\uFF08\u53EF\u9009\uFF0C\u4EC5\u5728\u672A\u8BBE\u7F6EcommandName\u65F6\u6709\u6548\uFF09\n- **userPrompts**: \u7279\u5B9A\u7528\u6237\u7684\u9884\u8BBE\u63D0\u793A\u8BCD\uFF08\u53EF\u9009\uFF0C\u683C\u5F0F\u4E3A {\"\u7528\u6237ID\": \"\u9884\u8BBE\u63D0\u793A\u8BCD\"}\uFF09\n- **thinkingPrompt**: \u601D\u8003\u63D0\u793A\u8BCD\uFF08\u53EF\u9009\uFF0C\u7559\u7A7A\u5219\u4E0D\u663E\u793A\u601D\u8003\u63D0\u793A\uFF09\n\n### \u4F7F\u7528\u65B9\u6CD5\n\n\u5982\u679C\u8BBE\u7F6E\u4E86\u81EA\u5B9A\u4E49\u6307\u4EE4\u540D\uFF0C\u53EF\u4EE5\u901A\u8FC7 `/\u6307\u4EE4\u540D \u4F60\u7684\u95EE\u9898` \u6765\u4F7F\u7528\u3002\n\u5982\u679C\u672A\u8BBE\u7F6E\u6307\u4EE4\u540D\uFF0C\u5219\u6240\u6709\u6D88\u606F\u90FD\u4F1A\u89E6\u53D1\u63D2\u4EF6\uFF08\u53EF\u901A\u8FC7\u524D\u7F00\u8FC7\u6EE4\uFF09\u3002\n\n### \u793A\u4F8B\n\n```\n/chat \u4ECA\u5929\u5929\u6C14\u600E\u4E48\u6837\uFF1F\n```\n\n\u6216\u8005\u5728\u672A\u8BBE\u7F6E\u6307\u4EE4\u540D\u7684\u60C5\u51B5\u4E0B\u76F4\u63A5\u53D1\u9001\u6D88\u606F\uFF1A\n\n```\n\u4ECA\u5929\u5929\u6C14\u600E\u4E48\u6837\uFF1F\n```\n";
-export declare const Config: Schema<ConfigInterface>;
-export declare function apply(ctx: Context, config: ConfigInterface): void;

package/lib/index.js

@@ -1,173 +0,0 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.Config = exports.usage = exports.name = void 0;
-exports.apply = apply;
-const koishi_1 = require("koishi");
-const service_1 = require("./service");
-exports.name = 'openai-compatible';
-exports.usage = `## 🤖 OpenAI兼容聊天插件
-
-该插件支持与所有兼容OpenAI API的大模型进行聊天交互。
-
-### 配置说明
-
-- **apiEndpoint**: API端点URL，例如 "https://api.openai.com/v1"
-- **apiKey**: API密钥
-- **model**: 使用的模型名称，例如 "gpt-3.5-turbo"
-- **temperature**: 温度参数，控制回复的随机性 (0-2)
-- **maxTokens**: 最大生成的token数量
-- **systemPrompt**: 系统预设提示词（可选）
-- **commandName**: 自定义指令名（可选，如不设置则默认处理所有消息）
-- **triggerPrefix**: 触发前缀（可选，仅在未设置commandName时有效）
-- **userPrompts**: 特定用户的预设提示词（可选，格式为 {"用户ID": "预设提示词"}）
-- **thinkingPrompt**: 思考提示词（可选，留空则不显示思考提示）
-
-### 使用方法
-
-如果设置了自定义指令名，可以通过 \`/指令名 你的问题\` 来使用。
-如果未设置指令名，则所有消息都会触发插件（可通过前缀过滤）。
-
-### 示例
-
-\`\`\`
-/chat 今天天气怎么样？
-\`\`\`
-
-或者在未设置指令名的情况下直接发送消息：
-
-\`\`\`
-今天天气怎么样？
-\`\`\`
-`;
-exports.Config = koishi_1.Schema.object({
-    apiEndpoint: koishi_1.Schema.string()
-        .description('API端点URL')
-        .default('https://api.openai.com/v1')
-        .required(),
-    apiKey: koishi_1.Schema.string()
-        .description('API密钥')
-        .role('secret')
-        .required(),
-    model: koishi_1.Schema.string()
-        .description('使用的模型名称')
-        .default('gpt-3.5-turbo')
-        .required(),
-    temperature: koishi_1.Schema.number()
-        .description('温度参数，控制回复的随机性 (0-2)')
-        .default(0.7)
-        .min(0)
-        .max(2),
-    maxTokens: koishi_1.Schema.number()
-        .description('最大生成的token数量')
-        .default(2000)
-        .min(1),
-    systemPrompt: koishi_1.Schema.string()
-        .description('系统预设提示词（可选）')
-        .default('你是一个有用的AI助手。'),
-    commandName: koishi_1.Schema.string()
-        .description('自定义指令名（可选，如不设置则默认处理所有消息）'),
-    triggerPrefix: koishi_1.Schema.string()
-        .description('触发前缀（可选，仅在未设置commandName时有效）')
-        .default(''),
-    userPrompts: koishi_1.Schema.dict(koishi_1.Schema.string())
-        .description('特定用户的预设提示词（可选，格式为 {"用户ID": "预设提示词"}）')
-        .default({}),
-    thinkingPrompt: koishi_1.Schema.string()
-        .description('思考提示词（可选，留空则不显示思考提示）')
-        .default(''),
-});
-function apply(ctx, config) {
-    // 创建服务实例
-    const service = new service_1.OpenAICompatibleService(ctx, config);
-    // 注册管理命令
-    ctx.command('ai管理', '管理AI插件')
-        .usage('管理OpenAI兼容插件的设置和用户预设')
-        .userFields(['authority'])
-        .action(({ session }) => {
-        if (session.user.authority < 3)
-            return '权限不足';
-        return '管理功能已启用';
-    });
-    // 设置用户预设命令
-    ctx.command('ai预设 <userId:string> <prompt:text>', '设置用户专属AI行为')
-        .usage('设置用户专属AI行为\n示例：ai预设 123456 你是一个专业翻译')
-        .example('ai预设 123456 你是一个专业翻译')
-        .action(async ({ session }, userId, prompt) => {
-        if (!userId || !prompt)
-            return '请提供用户ID和预设提示词';
-        service.setUserPrompt(userId, prompt);
-        return `已为用户 ${userId} 设置预设提示词`;
-    });
-    // 移除用户预设命令
-    ctx.command('ai清除 <userId:string>', '删除用户专属设置')
-        .usage('移除特定用户的预设提示词')
-        .example('ai清除 123456')
-        .action(async ({ session }, userId) => {
-        if (!userId)
-            return '请提供用户ID';
-        service.removeUserPrompt(userId);
-        return `已移除用户 ${userId} 的预设提示词`;
-    });
-    // 清除用户历史命令
-    ctx.command('ai清除记录 <userId:string>', '清除用户聊天历史')
-        .usage('清除特定用户的聊天历史')
-        .example('ai清除记录 123456')
-        .action(async ({ session }, userId) => {
-        if (!userId)
-            return '请提供用户ID';
-        service.clearHistory(userId);
-        return `已清除用户 ${userId} 的聊天历史`;
-    });
-    // 如果设置了指令名，则注册指令
-    if (config.commandName) {
-        ctx.command(config.commandName, '与AI助手对话')
-            .action(async ({ session }, text) => {
-            if (!text)
-                return '请输入您想问的内容。';
-            try {
-                // 仅当配置了思考提示时才发送
-                if (config.thinkingPrompt) {
-                    session.send(koishi_1.h.text(config.thinkingPrompt));
-                }
-                const response = await service.chat(session.userId, text);
-                return response;
-            }
-            catch (error) {
-                ctx.logger('openai-compatible').error(error);
-                return `发生错误: ${error.message || '未知错误'}`;
-            }
-        });
-    }
-    else {
-        // 如果没有设置指令名，则处理所有消息
-        ctx.middleware(async (session, next) => {
-            // 如果是自己发送的消息，则跳过
-            if (session.userId === session.bot.selfId)
-                return next();
-            // 检查前缀
-            const content = session.content;
-            if (config.triggerPrefix && !content.startsWith(config.triggerPrefix)) {
-                return next();
-            }
-            // 去除前缀
-            const text = config.triggerPrefix
-                ? content.slice(config.triggerPrefix.length).trim()
-                : content.trim();
-            if (!text)
-                return next();
-            try {
-                // 仅当配置了思考提示时才发送
-                if (config.thinkingPrompt) {
-                    await session.send(koishi_1.h.text(config.thinkingPrompt));
-                }
-                // 获取回复
-                const response = await service.chat(session.userId, text);
-                return response;
-            }
-            catch (error) {
-                ctx.logger('openai-compatible').error(error);
-                return `发生错误: ${error.message || '未知错误'}`;
-            }
-        });
-    }
-}

package/lib/service.d.ts

@@ -1,37 +0,0 @@
-import { Context, Service } from 'koishi';
-import { Config } from './config';
-export declare class OpenAICompatibleService extends Service {
-    config: Config;
-    private readonly historyMaxLength;
-    private readonly historyExpireTime;
-    private histories;
-    constructor(ctx: Context, config: Config);
-    /**
-     * 清理过期的历史记录
-     */
-    private cleanExpiredHistories;
-    /**
-     * 获取用户的聊天历史
-     */
-    private getUserHistory;
-    /**
-     * 添加消息到历史记录
-     */
-    private addMessageToHistory;
-    /**
-     * 与AI进行聊天
-     */
-    chat(userId: string, content: string): Promise<string>;
-    /**
-     * 清除用户的聊天历史
-     */
-    clearHistory(userId: string): void;
-    /**
-     * 设置特定用户的预设提示词
-     */
-    setUserPrompt(userId: string, prompt: string): Promise<void>;
-    /**
-     * 移除特定用户的预设提示词
-     */
-    removeUserPrompt(userId: string): void;
-}

package/lib/service.js

@@ -1,190 +0,0 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.OpenAICompatibleService = void 0;
-const koishi_1 = require("koishi");
-class OpenAICompatibleService extends koishi_1.Service {
-    constructor(ctx, config) {
-        super(ctx, 'openai-compatible');
-        this.historyMaxLength = 10; // 历史记录最大长度
-        this.historyExpireTime = 30 * 60 * 1000; // 历史记录过期时间（30分钟）
-        this.histories = new Map();
-        this.config = config;
-        this.ctx = ctx;
-        // 初始化数据库
-        ctx.model.extend('openai.userPrompts', {
-            userId: 'string',
-            prompt: 'text',
-            createdAt: 'timestamp',
-            updatedAt: 'timestamp',
-        }, {
-            primary: 'userId'
-        });
-        ctx.model.extend('openai.chatHistories', {
-            userId: 'string',
-            messages: 'json',
-            lastUpdated: 'timestamp',
-        }, {
-            primary: 'userId'
-        });
-        // 定期清理过期的历史记录
-        ctx.setInterval(() => {
-            this.cleanExpiredHistories();
-        }, 5 * 60 * 1000); // 每5分钟清理一次
-    }
-    /**
-     * 清理过期的历史记录
-     */
-    async cleanExpiredHistories() {
-        const expireTime = new Date(Date.now() - this.historyExpireTime);
-        await this.ctx.database.remove('openai.chatHistories', {
-            lastUpdated: { $lt: expireTime }
-        });
-    }
-    /**
-     * 获取用户的聊天历史
-     */
-    getUserHistory(userId) {
-        if (!this.histories.has(userId)) {
-            this.histories.set(userId, {
-                userId,
-                messages: [],
-                lastUpdated: Date.now()
-            });
-        }
-        return this.histories.get(userId);
-    }
-    /**
-     * 添加消息到历史记录
-     */
-    addMessageToHistory(userId, message) {
-        const history = this.getUserHistory(userId);
-        // 如果是系统消息，检查是否已存在系统消息，如果存在则替换
-        if (message.role === 'system') {
-            const systemIndex = history.messages.findIndex(m => m.role === 'system');
-            if (systemIndex >= 0) {
-                history.messages[systemIndex] = message;
-                history.lastUpdated = Date.now();
-                return;
-            }
-        }
-        // 添加新消息
-        history.messages.push(message);
-        // 如果超过最大长度，移除最早的非系统消息
-        if (history.messages.length > this.historyMaxLength) {
-            const systemMessage = history.messages.find(m => m.role === 'system');
-            history.messages = systemMessage
-                ? [systemMessage, ...history.messages.slice(-this.historyMaxLength + 1)]
-                : history.messages.slice(-this.historyMaxLength);
-        }
-        history.lastUpdated = Date.now();
-    }
-    /**
-     * 与AI进行聊天
-     */
-    async chat(userId, content) {
-        // 获取用户历史记录
-        const history = this.getUserHistory(userId);
-        // 如果是首次对话且有系统提示，添加系统提示
-        if (history.messages.length === 0) {
-            // 检查是否有特定用户的预设提示词
-            const userPrompt = this.config.userPrompts?.[userId];
-            const systemPrompt = userPrompt || this.config.systemPrompt;
-            if (systemPrompt) {
-                this.addMessageToHistory(userId, {
-                    role: 'system',
-                    content: systemPrompt
-                });
-            }
-        }
-        // 添加用户消息
-        this.addMessageToHistory(userId, {
-            role: 'user',
-            content
-        });
-        try {
-            // 调用API
-            const response = await this.ctx.http.post(`${this.config.apiEndpoint}/chat/completions`, {
-                model: this.config.model,
-                messages: history.messages,
-                temperature: this.config.temperature,
-                max_tokens: this.config.maxTokens
-            }, {
-                headers: {
-                    'Content-Type': 'application/json',
-                    'Authorization': `Bearer ${this.config.apiKey}`
-                }
-            });
-            // 处理响应
-            const assistantMessage = response.choices[0]?.message;
-            if (!assistantMessage || !assistantMessage.content) {
-                throw new Error('API返回的响应格式不正确');
-            }
-            // 添加助手回复到历史记录
-            this.addMessageToHistory(userId, {
-                role: 'assistant',
-                content: assistantMessage.content
-            });
-            return assistantMessage.content;
-        }
-        catch (error) {
-            this.ctx.logger('openai-compatible').error('API调用失败:', error);
-            // 从历史记录中移除最后一条用户消息（因为没有得到回复）
-            history.messages.pop();
-            // 处理 ctx.http 的错误
-            if (error.status) {
-                throw new Error(`API调用失败: ${error.status} ${error.statusText || 'Bad Request'}`);
-            }
-            else if (error.code === 'ETIMEDOUT' || error.code === 'ECONNABORTED') {
-                throw new Error('API请求超时或无响应');
-            }
-            else {
-                throw new Error(`请求错误: ${error.message || '未知错误'}`);
-            }
-        }
-    }
-    /**
-     * 清除用户的聊天历史
-     */
-    clearHistory(userId) {
-        this.histories.delete(userId);
-    }
-    /**
-     * 设置特定用户的预设提示词
-     */
-    async setUserPrompt(userId, prompt) {
-        await this.ctx.database.upsert('openai.userPrompts', [{
-                userId,
-                prompt,
-                updatedAt: new Date()
-            }]);
-        // 如果用户已有历史记录，更新系统提示
-        if (this.histories.has(userId)) {
-            const history = this.getUserHistory(userId);
-            const systemIndex = history.messages.findIndex(m => m.role === 'system');
-            if (systemIndex >= 0) {
-                // 替换现有的系统提示
-                history.messages[systemIndex] = {
-                    role: 'system',
-                    content: prompt
-                };
-            }
-            else if (history.messages.length > 0) {
-                // 在开头插入系统提示
-                history.messages.unshift({
-                    role: 'system',
-                    content: prompt
-                });
-            }
-            history.lastUpdated = Date.now();
-        }
-    }
-    /**
-     * 移除特定用户的预设提示词
-     */
-    removeUserPrompt(userId) {
-        if (this.config.userPrompts && this.config.userPrompts[userId]) {
-            delete this.config.userPrompts[userId];
-        }
-    }
-}
-exports.OpenAICompatibleService = OpenAICompatibleService;

package/src/config.ts

@@ -1,23 +0,0 @@
-export interface Config {
-    apiEndpoint: string
-    apiKey: string
-    model: string
-    temperature: number
-    maxTokens: number
-    systemPrompt: string
-    commandName?: string
-    triggerPrefix?: string
-    userPrompts?: Record<string, string>
-    thinkingPrompt?: string
-  }
-  
-  export interface Message {
-    role: 'system' | 'user' | 'assistant'
-    content: string
-  }
-  
-  export interface ChatHistory {
-    userId: string
-    messages: Message[]
-    lastUpdated: number
-  }

package/src/database.ts

@@ -1,34 +0,0 @@
-import { Tables } from 'koishi'
-
-export interface UserPrompt {
-  userId: string
-  prompt: string
-  createdAt: Date
-  updatedAt: Date
-}
-
-export interface ChatHistory {
-  userId: string
-  messages: {
-    role: 'system' | 'user' | 'assistant'
-    content: string
-    timestamp: Date
-  }[]
-  lastUpdated: Date
-}
-
-const tables = {
-  'openai.userPrompts': {
-    userId: 'string',
-    prompt: 'text',
-    createdAt: 'timestamp',
-    updatedAt: 'timestamp'
-  },
-  'openai.chatHistories': {
-    userId: 'string',
-    messages: 'json',
-    lastUpdated: 'timestamp'
-  }
-};
-
-export default tables;