koishi-plugin-openai-compatible 1.0.7 → 1.0.9

package/src/types.ts

@@ -1,34 +1,222 @@
-export interface Message {
-  role: 'system' | 'user' | 'assistant';
-  content: string;
-}
-
-export interface ChatCompletionRequest {
-  model: string;
-  messages: Message[];
-  temperature?: number;
-  max_tokens?: number;
-  top_p?: number;
-  frequency_penalty?: number;
-  presence_penalty?: number;
-}
-
-export interface ChatCompletionResponse {
-  id: string;
-  object: string;
-  created: number;
-  model: string;
-  choices: Array<{
-    index: number;
-    message: {
-      role: string;
-      content: string;
-    };
-    finish_reason: string;
-  }>;
-  usage?: {
-    prompt_tokens: number;
-    completion_tokens: number;
-    total_tokens: number;
-  };
-}
+import { Schema } from 'koishi'
+
+// 情绪类型定义
+export type EmotionType = 
+  | 'happy'      // 开心
+  | 'sad'        // 悲伤
+  | 'angry'      // 愤怒
+  | 'neutral'    // 中性
+  | 'surprised'  // 惊讶
+  | 'fearful'    // 害怕
+  | 'disgusted'  // 厌恶
+  | 'excited'    // 兴奋
+  | 'calm'       // 平静
+  | 'confused'   // 困惑
+  | string       // 自定义情绪类型
+
+// 表情包配置接口
+export interface EmojiConfig {
+  text: string   // 文本表情
+  image?: string // 图片链接（可选）
+}
+
+export interface Config {
+  // 主模型配置
+  endpoint: string
+  apiKey: string
+  model: string
+  prompt: string
+  
+  // 情绪分析模型配置
+  emotionAnalysis: {
+    enabled: boolean
+    endpoint: string
+    apiKey: string
+    model: string
+    prompt: string
+  }
+  
+  // 情绪表情配置
+  emotionEmojis: Record<string, EmojiConfig>
+  
+  // 系统提示
+  systemPrompt: string
+  
+  // 交互控制
+  blacklist: string[]
+  cooldown: number
+  showError: boolean
+  
+  // 模型参数
+  maxTokens: number
+  temperature: number
+  topP: number
+  presencePenalty: number
+  frequencyPenalty: number
+  
+  // 情绪分析参数
+  emotionMaxTokens: number
+  emotionTemperature: number
+  
+  // 网络设置
+  timeout: number
+  maxRetries: number
+  emotionTimeout: number
+  
+  // 图片显示选项
+  showEmotionImage: boolean
+  imageAsMarkdown: boolean
+}
+
+export const Config: Schema<Config> = Schema.intersect([
+  Schema.object({
+    endpoint: Schema.string()
+      .description('OpenAI兼容API端点')
+      .default('https://api.openai.com/v1'),
+    apiKey: Schema.string()
+      .description('API密钥')
+      .role('secret'),
+    model: Schema.string()
+      .description('使用的模型')
+      .default('gpt-3.5-turbo'),
+    prompt: Schema.string()
+      .description('默认提示词（将被附加到用户消息前）')
+      .default(''),
+  }).description('基础配置'),
+  
+  Schema.object({
+    emotionAnalysis: Schema.object({
+      enabled: Schema.boolean()
+        .description('是否启用情绪分析')
+        .default(true),
+      endpoint: Schema.string()
+        .description('情绪分析API端点（留空则使用主端点）')
+        .default(''),
+      apiKey: Schema.string()
+        .description('情绪分析API密钥（留空则使用主密钥）')
+        .role('secret')
+        .default(''),
+      model: Schema.string()
+        .description('情绪分析模型（建议使用较小较快的模型）')
+        .default('gpt-3.5-turbo'),
+      prompt: Schema.string()
+        .description('情绪分析提示词')
+        .default('请分析以下文本的情绪类型，只能返回以下情绪类型之一：happy（开心）、sad（悲伤）、angry（愤怒）、neutral（中性）、surprised（惊讶）、fearful（害怕）、disgusted（厌恶）、excited（兴奋）、calm（平静）、confused（困惑）。不要返回其他内容，只返回情绪类型单词。'),
+    }).description('情绪分析配置'),
+  }).description('情绪分析模块'),
+  
+  Schema.object({
+    emotionEmojis: Schema.dict(
+      Schema.object({
+        text: Schema.string()
+          .description('文本表情符号')
+          .required(),
+        image: Schema.string()
+          .description('图片链接（可选，支持jpg/png/gif）')
+          .default(''),
+      })
+    )
+    .description('情绪类型对应的表情包配置')
+    .default({
+      'happy': { text: '😊', image: '' },
+      'sad': { text: '😢', image: '' },
+      'angry': { text: '😠', image: '' },
+      'neutral': { text: '😐', image: '' },
+      'surprised': { text: '😲', image: '' },
+      'fearful': { text: '😨', image: '' },
+      'disgusted': { text: '🤢', image: '' },
+      'excited': { text: '🤩', image: '' },
+      'calm': { text: '😌', image: '' },
+      'confused': { text: '😕', image: '' },
+    }),
+    showEmotionImage: Schema.boolean()
+      .description('是否显示情绪图片（如果配置了图片链接）')
+      .default(true),
+    imageAsMarkdown: Schema.boolean()
+      .description('是否使用Markdown格式显示图片（否则使用图片CQ码）')
+      .default(false),
+  }).description('情绪表情配置'),
+  
+  Schema.object({
+    systemPrompt: Schema.string()
+      .description('系统提示词（用于system role）')
+      .default('你是一个有帮助的AI助手。'),
+  }).description('系统提示'),
+  
+  Schema.object({
+    blacklist: Schema.array(String)
+      .description('黑名单用户ID')
+      .default([]),
+    cooldown: Schema.number()
+      .description('冷却时间（秒）')
+      .min(0)
+      .default(0),
+    showError: Schema.boolean()
+      .description('是否显示错误提示')
+      .default(true),
+  }).description('交互控制'),
+  
+  Schema.object({
+    maxTokens: Schema.number()
+      .description('最大输出token数')
+      .min(1)
+      .max(4096)
+      .default(2048),
+    temperature: Schema.number()
+      .description('温度参数（0-2）')
+      .min(0)
+      .max(2)
+      .step(0.1)
+      .default(0.7),
+    topP: Schema.number()
+      .description('Top-p采样参数')
+      .min(0)
+      .max(1)
+      .step(0.1)
+      .default(1),
+    presencePenalty: Schema.number()
+      .description('存在惩罚（-2到2）')
+      .min(-2)
+      .max(2)
+      .step(0.1)
+      .default(0),
+    frequencyPenalty: Schema.number()
+      .description('频率惩罚（-2到2）')
+      .min(-2)
+      .max(2)
+      .step(0.1)
+      .default(0),
+  }).description('模型参数'),
+  
+  Schema.object({
+    emotionMaxTokens: Schema.number()
+      .description('情绪分析最大token数')
+      .min(1)
+      .max(100)
+      .default(10),
+    emotionTemperature: Schema.number()
+      .description('情绪分析温度参数')
+      .min(0)
+      .max(1)
+      .step(0.1)
+      .default(0.1),
+  }).description('情绪分析参数'),
+  
+  Schema.object({
+    timeout: Schema.number()
+      .description('主请求超时时间（毫秒）')
+      .min(1000)
+      .max(60000)
+      .default(30000),
+    emotionTimeout: Schema.number()
+      .description('情绪分析请求超时时间（毫秒）')
+      .min(1000)
+      .max(30000)
+      .default(5000),
+    maxRetries: Schema.number()
+      .description('最大重试次数')
+      .min(0)
+      .max(5)
+      .default(3),
+  }).description('网络设置'),
+])

package/lib/api.d.ts

@@ -1,7 +0,0 @@
-import type { ChatCompletionRequest } from './types';
-export declare class OpenAICompatibleAPI {
-    private endpoint;
-    private apiKey;
-    constructor(endpoint: string, apiKey: string);
-    chatCompletion(request: ChatCompletionRequest): Promise<string>;
-}

package/lib/api.js

@@ -1,32 +0,0 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.OpenAICompatibleAPI = void 0;
-class OpenAICompatibleAPI {
-    constructor(endpoint, apiKey) {
-        this.endpoint = endpoint;
-        this.apiKey = apiKey;
-    }
-    async chatCompletion(request) {
-        const headers = {
-            'Content-Type': 'application/json',
-        };
-        if (this.apiKey) {
-            headers['Authorization'] = `Bearer ${this.apiKey}`;
-        }
-        const response = await fetch(this.endpoint, {
-            method: 'POST',
-            headers,
-            body: JSON.stringify(request),
-        });
-        if (!response.ok) {
-            const errorText = await response.text();
-            throw new Error(`API 请求失败: ${response.status} ${response.statusText}\n${errorText}`);
-        }
-        const data = await response.json();
-        if (!data.choices || data.choices.length === 0) {
-            throw new Error('API 返回了空的选择列表');
-        }
-        return data.choices[0].message.content;
-    }
-}
-exports.OpenAICompatibleAPI = OpenAICompatibleAPI;

package/lib/config.d.ts

@@ -1,23 +0,0 @@
-import { Schema } from 'koishi';
-export interface Config {
-    endpoint: string;
-    apiKey: string;
-    model: string;
-    enableEmotion: boolean;
-    emotionEndpoint: string;
-    emotionApiKey: string;
-    emotionModel: string;
-    emotionImages: Record<string, string>;
-    systemPrompt: string;
-    emotionSystemPrompt: string;
-    temperature: number;
-    maxTokens: number;
-    topP: number;
-    frequencyPenalty: number;
-    presencePenalty: number;
-    blacklist: string[];
-    cooldown: number;
-    commandName: string;
-    enableMessage: boolean;
-}
-export declare const Config: Schema<Config>;

package/lib/config.js

@@ -1,54 +0,0 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.Config = void 0;
-const koishi_1 = require("koishi");
-exports.Config = koishi_1.Schema.intersect([
-    koishi_1.Schema.object({
-        endpoint: koishi_1.Schema.string().default('https://api.openai.com/v1/chat/completions'),
-        apiKey: koishi_1.Schema.string().default('').description('OpenAI API Key'),
-        model: koishi_1.Schema.string().default('gpt-3.5-turbo'),
-    }).description('API 配置'),
-    koishi_1.Schema.object({
-        enableEmotion: koishi_1.Schema.boolean().default(false).description('是否启用情绪分析'),
-        emotionEndpoint: koishi_1.Schema.string().default('https://api.openai.com/v1/chat/completions'),
-        emotionApiKey: koishi_1.Schema.string().default('').description('情绪分析 API Key'),
-        emotionModel: koishi_1.Schema.string().default('gpt-3.5-turbo').description('推荐使用较快的模型'),
-        emotionImages: koishi_1.Schema.dict(koishi_1.Schema.string().description('情绪表情包图片链接')).default({
-            '快乐': 'https://example.com/happy.gif',
-            '开心': 'https://example.com/happy.gif',
-            '高兴': 'https://example.com/happy.gif',
-            '伤心': 'https://example.com/sad.gif',
-            '难过': 'https://example.com/sad.gif',
-            '悲伤': 'https://example.com/sad.gif',
-            '生气': 'https://example.com/angry.gif',
-            '愤怒': 'https://example.com/angry.gif',
-            '惊讶': 'https://example.com/surprised.gif',
-            '疑惑': 'https://example.com/confused.gif',
-            '害怕': 'https://example.com/afraid.gif',
-            '恐惧': 'https://example.com/afraid.gif',
-            '平静': 'https://example.com/calm.gif',
-            '中立': 'https://example.com/neutral.gif',
-        }).description('情绪名称对应的表情包图片链接'),
-    }).description('情绪分析配置'),
-    koishi_1.Schema.object({
-        systemPrompt: koishi_1.Schema.string()
-            .default('你是一个友好的 AI 助手，请用简洁、准确的语言回答用户的问题。')
-            .description('系统提示词'),
-        emotionSystemPrompt: koishi_1.Schema.string()
-            .default('你是一个情绪分析助手。分析给定的文本内容，并判断其中表达的情绪。\n\n情绪类型包括：快乐、开心、高兴、伤心、难过、悲伤、生气、愤怒、惊讶、疑惑、害怕、恐惧、平静、中立。\n\n请直接输出一个情绪名称，不要输出任何其他内容，不要解释，不要添加标点符号。\n\n例如：\n- "今天天气真好，我很开心！" -> 快乐\n- "这次考试失败了，我好难过。" -> 伤心\n- "什么？这太不可思议了！" -> 惊讶\n- "我不知道该怎么办。" -> 疑惑\n- "这真是太可怕了！" -> 害怕\n- "一切都很平静。" -> 平静')
-            .description('情绪分析系统提示词'),
-    }).description('提示词配置'),
-    koishi_1.Schema.object({
-        temperature: koishi_1.Schema.number().min(0).max(2).default(0.7).description('温度参数'),
-        maxTokens: koishi_1.Schema.number().min(1).default(2000).description('最大令牌数'),
-        topP: koishi_1.Schema.number().min(0).max(1).default(1).description('Top P 参数'),
-        frequencyPenalty: koishi_1.Schema.number().min(-2).max(2).default(0).description('频率惩罚'),
-        presencePenalty: koishi_1.Schema.number().min(-2).max(2).default(0).description('存在惩罚'),
-    }).description('参数配置'),
-    koishi_1.Schema.object({
-        blacklist: koishi_1.Schema.array(koishi_1.Schema.string()).default([]).description('黑名单用户 ID'),
-        cooldown: koishi_1.Schema.number().min(0).default(10000).description('请求冷却时间（毫秒）'),
-        commandName: koishi_1.Schema.string().default('ai').description('自定义指令名称'),
-        enableMessage: koishi_1.Schema.boolean().default(true).description('是否开启提示（关闭后所有错误提示都不会输出）'),
-    }).description('其他配置'),
-]);

package/lib/emotion.d.ts

@@ -1,9 +0,0 @@
-import type { OpenAICompatibleAPI } from './api';
-import type { Config } from './config';
-export declare class EmotionAnalyzer {
-    private api;
-    private config;
-    constructor(api: OpenAICompatibleAPI, config: Config);
-    analyzeEmotion(text: string): Promise<string>;
-    getEmotionImage(emotion: string): string | null;
-}

package/lib/emotion.js

@@ -1,32 +0,0 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.EmotionAnalyzer = void 0;
-class EmotionAnalyzer {
-    constructor(api, config) {
-        this.api = api;
-        this.config = config;
-    }
-    async analyzeEmotion(text) {
-        const request = {
-            model: this.config.emotionModel,
-            messages: [
-                { role: 'system', content: this.config.emotionSystemPrompt },
-                { role: 'user', content: text },
-            ],
-            temperature: 0.3,
-            max_tokens: 50,
-        };
-        try {
-            const emotion = await this.api.chatCompletion(request);
-            return emotion.trim();
-        }
-        catch (error) {
-            console.error('情绪分析失败:', error);
-            return '';
-        }
-    }
-    getEmotionImage(emotion) {
-        return this.config.emotionImages[emotion] || null;
-    }
-}
-exports.EmotionAnalyzer = EmotionAnalyzer;

package/lib/locales/zh.d.ts

@@ -1,44 +0,0 @@
-declare const _default: {
-    commands: {
-        ai: {
-            name: string;
-            description: string;
-            input: string;
-            cooldown: string;
-            blacklist: string;
-            error: string;
-        };
-        admin: {
-            add: {
-                description: string;
-                success: string;
-            };
-            remove: {
-                description: string;
-                success: string;
-                notFound: string;
-            };
-            list: {
-                description: string;
-                empty: string;
-                title: string;
-            };
-            clearcooldown: {
-                description: string;
-                success: string;
-            };
-            status: {
-                description: string;
-                title: string;
-                model: string;
-                endpoint: string;
-                emotion: string;
-                enabled: string;
-                disabled: string;
-                cooldown: string;
-                blacklistCount: string;
-            };
-        };
-    };
-};
-export default _default;

package/lib/locales/zh.js

@@ -1,45 +0,0 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.default = {
-    commands: {
-        ai: {
-            name: 'AI 对话',
-            description: '与 AI 进行对话',
-            input: '请输入要聊天的内容。',
-            cooldown: '请求过于频繁，请等待 {0} 秒后再试。',
-            blacklist: '您在黑名单中，无法使用此功能。',
-            error: '请求失败: {0}',
-        },
-        admin: {
-            add: {
-                description: '添加黑名单',
-                success: '已将用户 {0} 添加到黑名单。',
-            },
-            remove: {
-                description: '移除黑名单',
-                success: '已将用户 {0} 从黑名单中移除。',
-                notFound: '用户 {0} 不在黑名单中。',
-            },
-            list: {
-                description: '查看黑名单',
-                empty: '黑名单为空。',
-                title: '黑名单列表:',
-            },
-            clearcooldown: {
-                description: '清除用户冷却',
-                success: '已清除用户 {0} 的冷却时间。',
-            },
-            status: {
-                description: '查看插件状态',
-                title: '插件状态:',
-                model: '模型: {0}',
-                endpoint: '端点: {0}',
-                emotion: '情绪分析: {0}',
-                enabled: '已启用',
-                disabled: '未启用',
-                cooldown: '冷却时间: {0} 秒',
-                blacklistCount: '黑名单数量: {0}',
-            },
-        },
-    },
-};

package/lib/storage.d.ts

@@ -1,10 +0,0 @@
-export interface CooldownData {
-    lastRequestTime: number;
-}
-export declare class Storage {
-    private cooldowns;
-    getCooldown(userId: string): CooldownData | undefined;
-    setCooldown(userId: string, data: CooldownData): void;
-    deleteCooldown(userId: string): void;
-    isOnCooldown(userId: string, cooldownTime: number): boolean;
-}

package/lib/storage.js

@@ -1,25 +0,0 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.Storage = void 0;
-class Storage {
-    constructor() {
-        this.cooldowns = new Map();
-    }
-    getCooldown(userId) {
-        return this.cooldowns.get(userId);
-    }
-    setCooldown(userId, data) {
-        this.cooldowns.set(userId, data);
-    }
-    deleteCooldown(userId) {
-        this.cooldowns.delete(userId);
-    }
-    isOnCooldown(userId, cooldownTime) {
-        const data = this.getCooldown(userId);
-        if (!data)
-            return false;
-        const elapsed = Date.now() - data.lastRequestTime;
-        return elapsed < cooldownTime;
-    }
-}
-exports.Storage = Storage;

package/src/api.ts

@@ -1,40 +0,0 @@
-import type { ChatCompletionRequest, ChatCompletionResponse } from './types';
-
-export class OpenAICompatibleAPI {
-  private endpoint: string;
-  private apiKey: string;
-
-  constructor(endpoint: string, apiKey: string) {
-    this.endpoint = endpoint;
-    this.apiKey = apiKey;
-  }
-
-  async chatCompletion(request: ChatCompletionRequest): Promise<string> {
-    const headers: Record<string, string> = {
-      'Content-Type': 'application/json',
-    };
-
-    if (this.apiKey) {
-      headers['Authorization'] = `Bearer ${this.apiKey}`;
-    }
-
-    const response = await fetch(this.endpoint, {
-      method: 'POST',
-      headers,
-      body: JSON.stringify(request),
-    });
-
-    if (!response.ok) {
-      const errorText = await response.text();
-      throw new Error(`API 请求失败: ${response.status} ${response.statusText}\n${errorText}`);
-    }
-
-    const data: ChatCompletionResponse = await response.json();
-
-    if (!data.choices || data.choices.length === 0) {
-      throw new Error('API 返回了空的选择列表');
-    }
-
-    return data.choices[0].message.content;
-  }
-}