koishi-plugin-openai-compatible 1.0.5 → 1.0.6

package/dist/index.js

@@ -1,97 +0,0 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.Config = exports.name = void 0;
-exports.apply = apply;
-const koishi_1 = require("koishi");
-exports.name = 'openai-compatible';
-exports.Config = koishi_1.Schema.object({
-    apiEndpoint: koishi_1.Schema.string().required().description('API endpoint URL, e.g., "https://api.openai.com/v1"'),
-    apiKey: koishi_1.Schema.string().required().description('API key'),
-    model: koishi_1.Schema.string().required().description('Model name, e.g., "gpt-3.5-turbo"'),
-    temperature: koishi_1.Schema.number().min(0).max(2).default(1).description('Controls randomness of responses (0-2)'),
-    maxTokens: koishi_1.Schema.number().min(1).default(2048).description('Maximum number of tokens to generate'),
-    systemPrompt: koishi_1.Schema.string().description('System prompt (optional)'),
-    commandName: koishi_1.Schema.string().description('Custom command name (optional)'),
-    triggerPrefix: koishi_1.Schema.string().description('Trigger prefix (optional, used when no commandName is set)'),
-    userPrompts: koishi_1.Schema.dict(koishi_1.Schema.string()).description('User-specific prompts (optional)'),
-    thinkingPrompt: koishi_1.Schema.string().description('Thinking prompt (optional)'),
-    blacklist: koishi_1.Schema.array(koishi_1.Schema.string()).description('List of blacklisted user IDs (optional)'),
-    blacklistResponse: koishi_1.Schema.string().description('Response for blacklisted users (text or image URL)'),
-});
-function apply(ctx, config) {
-    const { apiEndpoint, apiKey, model, temperature, maxTokens, systemPrompt, commandName, triggerPrefix, userPrompts, thinkingPrompt, blacklist, blacklistResponse } = config;
-    // Register command if commandName is provided
-    if (commandName) {
-        ctx.command(commandName, 'Chat with OpenAI-compatible API')
-            .action(async ({ session }, ...args) => {
-            if (!session || !session.userId || blacklist?.includes(session.userId)) {
-                return blacklistResponse || 'You are not allowed to use this command.';
-            }
-            const userPrompt = !session || !session.userId ? '' : userPrompts?.[session.userId] || '';
-            const prompt = args.join(' ');
-            const fullPrompt = systemPrompt ? `${systemPrompt}\n${userPrompt}\n${prompt}` : `${userPrompt}\n${prompt}`;
-            if (thinkingPrompt) {
-                if (session)
-                    session.send(thinkingPrompt);
-            }
-            try {
-                const response = await ctx.http.post(`${apiEndpoint}/chat/completions`, {
-                    model,
-                    messages: [
-                        { role: 'user', content: fullPrompt },
-                    ],
-                    temperature,
-                    max_tokens: maxTokens,
-                }, {
-                    headers: {
-                        'Authorization': `Bearer ${apiKey}`,
-                        'Content-Type': 'application/json',
-                    },
-                });
-                return response.choices[0].message.content;
-            }
-            catch (error) {
-                ctx.logger.error('OpenAI API error:', error);
-                return 'Failed to get response from the API.';
-            }
-        });
-    }
-    else {
-        // Handle all messages if no commandName is provided
-        ctx.middleware(async (session, next) => {
-            if (!session || !session.userId || blacklist?.includes(session.userId)) {
-                return blacklistResponse || 'You are not allowed to use this command.';
-            }
-            const content = session.content;
-            if (!content || (triggerPrefix && !content.startsWith(triggerPrefix)))
-                return next();
-            const userPrompt = !session || !session.userId ? '' : userPrompts?.[session.userId] || '';
-            const prompt = !content ? '' : triggerPrefix ? content.slice(triggerPrefix.length) : content;
-            const fullPrompt = systemPrompt ? `${systemPrompt}\n${userPrompt}\n${prompt}` : `${userPrompt}\n${prompt}`;
-            if (thinkingPrompt) {
-                if (session)
-                    session.send(thinkingPrompt);
-            }
-            try {
-                const response = await ctx.http.post(`${apiEndpoint}/chat/completions`, {
-                    model,
-                    messages: [
-                        { role: 'user', content: fullPrompt },
-                    ],
-                    temperature,
-                    max_tokens: maxTokens,
-                }, {
-                    headers: {
-                        'Authorization': `Bearer ${apiKey}`,
-                        'Content-Type': 'application/json',
-                    },
-                });
-                return response.choices[0].message.content;
-            }
-            catch (error) {
-                ctx.logger.error('OpenAI API error:', error);
-                return 'Failed to get response from the API.';
-            }
-        });
-    }
-}