koishi-plugin-aka-ai-generator 0.1.9 → 0.2.1

package/lib/index.d.ts

@@ -1,6 +1,13 @@
 import { Context, Schema } from 'koishi';
 export declare const name = "aka-ai-generator";
 export type ImageProvider = 'yunwu' | 'gptgod';
+export interface ModelMappingConfig {
+    suffix: string;
+    modelId: string;
+    provider?: ImageProvider;
+    promptSuffix?: string;
+    description?: string;
+}
 export interface StyleConfig {
     commandName: string;
     prompt: string;
@@ -27,6 +34,7 @@ export interface Config {
     yunwuModelId: string;
     gptgodApiKey: string;
     gptgodModelId: string;
+    modelMappings?: ModelMappingConfig[];
     apiTimeout: number;
     commandTimeout: number;
     defaultNumImages: number;

package/lib/index.js

@@ -571,6 +571,16 @@ var Config = import_koishi.Schema.intersect([
     yunwuModelId: import_koishi.Schema.string().default("gemini-2.5-flash-image").description("云雾图像生成模型ID"),
     gptgodApiKey: import_koishi.Schema.string().description("GPTGod API 密钥").role("secret").default(""),
     gptgodModelId: import_koishi.Schema.string().default("nano-banana").description("GPTGod 模型ID"),
+    modelMappings: import_koishi.Schema.array(import_koishi.Schema.object({
+      suffix: import_koishi.Schema.string().required().description("指令后缀（例如 4K，对应输入 -4K）"),
+      provider: import_koishi.Schema.union([
+        import_koishi.Schema.const("yunwu").description("云雾 Gemini 服务"),
+        import_koishi.Schema.const("gptgod").description("GPTGod 服务")
+      ]).description("可选：覆盖供应商"),
+      modelId: import_koishi.Schema.string().required().description("触发该后缀时使用的模型 ID"),
+      promptSuffix: import_koishi.Schema.string().role("textarea", { rows: 2 }).description("附加在风格 prompt 末尾的提示语"),
+      description: import_koishi.Schema.string().description("该后缀的说明，方便记忆")
+    })).role("table").default([]).description("根据 -后缀切换模型/供应商与附加 prompt"),
     apiTimeout: import_koishi.Schema.number().default(120).description("API请求超时时间（秒）"),
     commandTimeout: import_koishi.Schema.number().default(180).description("命令执行总超时时间（秒）"),
     // 默认设置
@@ -625,17 +635,89 @@ function apply(ctx, config) {
   const logger = ctx.logger("aka-ai-generator");
   const activeTasks = /* @__PURE__ */ new Map();
   const rateLimitMap = /* @__PURE__ */ new Map();
-  const imageProvider = createImageProvider({
-    provider: config.provider,
-    yunwuApiKey: config.yunwuApiKey,
-    yunwuModelId: config.yunwuModelId,
-    gptgodApiKey: config.gptgodApiKey,
-    gptgodModelId: config.gptgodModelId,
-    apiTimeout: config.apiTimeout,
-    logLevel: config.logLevel,
-    logger,
-    ctx
-  });
+  const providerCache = /* @__PURE__ */ new Map();
+  function getProviderInstance(providerType, modelId) {
+    const cacheKey = `${providerType}:${modelId || "default"}`;
+    if (!providerCache.has(cacheKey)) {
+      providerCache.set(cacheKey, createImageProvider({
+        provider: providerType,
+        yunwuApiKey: config.yunwuApiKey,
+        yunwuModelId: providerType === "yunwu" ? modelId || config.yunwuModelId : config.yunwuModelId,
+        gptgodApiKey: config.gptgodApiKey,
+        gptgodModelId: providerType === "gptgod" ? modelId || config.gptgodModelId : config.gptgodModelId,
+        apiTimeout: config.apiTimeout,
+        logLevel: config.logLevel,
+        logger,
+        ctx
+      }));
+    }
+    return providerCache.get(cacheKey);
+  }
+  __name(getProviderInstance, "getProviderInstance");
+  getProviderInstance(config.provider);
+  const modelMappingIndex = buildModelMappingIndex(config.modelMappings);
+  function normalizeSuffix(value) {
+    return value?.replace(/^\-+/, "").trim().toLowerCase();
+  }
+  __name(normalizeSuffix, "normalizeSuffix");
+  function buildModelMappingIndex(mappings) {
+    const map = /* @__PURE__ */ new Map();
+    if (!Array.isArray(mappings)) return map;
+    for (const mapping of mappings) {
+      const key = normalizeSuffix(mapping?.suffix);
+      if (!key || !mapping?.modelId) continue;
+      map.set(key, mapping);
+    }
+    return map;
+  }
+  __name(buildModelMappingIndex, "buildModelMappingIndex");
+  function parseStyleCommandModifiers(argv) {
+    const rawArgs = [...argv.args || []].map((arg) => typeof arg === "string" ? arg.trim() : "").filter(Boolean);
+    if (!rawArgs.length && !argv.rest) return {};
+    const modifiers = { customAdditions: [] };
+    const flagCandidates = [];
+    let index = 0;
+    while (index < rawArgs.length) {
+      const token = rawArgs[index];
+      if (!token) {
+        index++;
+        continue;
+      }
+      const lower = token.toLowerCase();
+      if (lower.startsWith("-prompt:")) {
+        const promptHead = token.substring(token.indexOf(":") + 1);
+        const restTokens = rawArgs.slice(index + 1);
+        modifiers.customPromptSuffix = [promptHead, ...restTokens].join(" ").trim();
+        break;
+      }
+      if (lower === "-add") {
+        index++;
+        const additionTokens = [];
+        while (index < rawArgs.length && !rawArgs[index].startsWith("-")) {
+          additionTokens.push(rawArgs[index]);
+          index++;
+        }
+        if (additionTokens.length) {
+          modifiers.customAdditions.push(additionTokens.join(" "));
+        }
+        continue;
+      }
+      flagCandidates.push(token);
+      index++;
+    }
+    for (const arg of flagCandidates) {
+      if (!arg.startsWith("-")) continue;
+      const key = normalizeSuffix(arg);
+      if (!key) continue;
+      const mapping = modelMappingIndex.get(key);
+      if (mapping) {
+        modifiers.modelMapping = mapping;
+        break;
+      }
+    }
+    return modifiers;
+  }
+  __name(parseStyleCommandModifiers, "parseStyleCommandModifiers");
   function getStyleCommands() {
     if (!config.styles || !Array.isArray(config.styles)) return [];
     return config.styles.filter((style) => style.commandName && style.prompt).map((style) => ({
@@ -933,13 +1015,23 @@ function apply(ctx, config) {
     return url;
   }
   __name(getImageUrl, "getImageUrl");
-  async function requestProviderImages(prompt, imageUrls, numImages) {
-    return await imageProvider.generateImages(prompt, imageUrls, numImages);
+  async function requestProviderImages(prompt, imageUrls, numImages, requestContext) {
+    const providerType = requestContext?.provider || config.provider;
+    const targetModelId = requestContext?.modelId;
+    const providerInstance = getProviderInstance(providerType, targetModelId);
+    if (config.logLevel === "debug") {
+      logger.debug("准备调用图像供应商", {
+        providerType,
+        modelId: targetModelId || "default",
+        numImages
+      });
+    }
+    return await providerInstance.generateImages(prompt, imageUrls, numImages);
   }
   __name(requestProviderImages, "requestProviderImages");
-  async function processImageWithTimeout(session, img, prompt, styleName, numImages) {
+  async function processImageWithTimeout(session, img, prompt, styleName, requestContext) {
     return Promise.race([
-      processImage(session, img, prompt, styleName, numImages),
+      processImage(session, img, prompt, styleName, requestContext),
       new Promise(
         (_, reject) => setTimeout(() => reject(new Error("命令执行超时")), config.commandTimeout * 1e3)
       )
@@ -951,12 +1043,12 @@ function apply(ctx, config) {
     });
   }
   __name(processImageWithTimeout, "processImageWithTimeout");
-  async function processImage(session, img, prompt, styleName, numImages) {
+  async function processImage(session, img, prompt, styleName, requestContext) {
     const userId = session.userId;
     if (activeTasks.has(userId)) {
       return "您有一个图像处理任务正在进行中，请等待完成";
     }
-    const imageCount = numImages || config.defaultNumImages;
+    const imageCount = requestContext?.numImages || config.defaultNumImages;
     if (imageCount < 1 || imageCount > 4) {
       return "生成数量必须在 1-4 之间";
     }
@@ -964,17 +1056,21 @@ function apply(ctx, config) {
     if (!imageUrl) {
       return;
     }
+    const providerType = requestContext?.provider || config.provider;
+    const providerModelId = requestContext?.modelId || (providerType === "yunwu" ? config.yunwuModelId : config.gptgodModelId);
     logger.info("开始图像处理", {
       userId,
       imageUrl,
       styleName,
       prompt,
-      numImages: imageCount
+      numImages: imageCount,
+      provider: providerType,
+      modelId: providerModelId
     });
     await session.send(`开始处理图片（${styleName}）...`);
     try {
       activeTasks.set(userId, "processing");
-      const images = await requestProviderImages(prompt, imageUrl, imageCount);
+      const images = await requestProviderImages(prompt, imageUrl, imageCount, requestContext);
       if (images.length === 0) {
         activeTasks.delete(userId);
         return "图像处理失败：未能生成图片";
@@ -1001,13 +1097,40 @@ function apply(ctx, config) {
   if (config.styles && Array.isArray(config.styles)) {
     for (const style of config.styles) {
       if (style.commandName && style.prompt) {
-        ctx.command(`${style.commandName} [img:text]`, "图像风格转换").option("num", "-n <num:number> 生成图片数量 (1-4)").action(async ({ session, options }, img) => {
+        ctx.command(`${style.commandName} [img:text]`, "图像风格转换").option("num", "-n <num:number> 生成图片数量 (1-4)").action(async (argv, img) => {
+          const { session, options } = argv;
           if (!session?.userId) return "会话无效";
           const limitCheck = await checkDailyLimit(session.userId);
           if (!limitCheck.allowed) {
             return limitCheck.message;
           }
-          return processImageWithTimeout(session, img, style.prompt, style.commandName, options?.num);
+          const modifiers = parseStyleCommandModifiers(argv);
+          const promptSegments = [style.prompt];
+          if (modifiers.modelMapping?.promptSuffix) {
+            promptSegments.push(modifiers.modelMapping.promptSuffix);
+          }
+          if (modifiers.customAdditions?.length) {
+            promptSegments.push(...modifiers.customAdditions);
+          }
+          if (modifiers.customPromptSuffix) {
+            promptSegments.push(modifiers.customPromptSuffix);
+          }
+          const mergedPrompt = promptSegments.filter(Boolean).join("\n\n");
+          const requestContext = {
+            numImages: options?.num
+          };
+          if (modifiers.modelMapping?.provider) {
+            requestContext.provider = modifiers.modelMapping.provider;
+          }
+          if (modifiers.modelMapping?.modelId) {
+            requestContext.modelId = modifiers.modelMapping.modelId;
+          }
+          if (modifiers.modelMapping?.description) {
+            await session.send(`已启用 ${modifiers.modelMapping.description} 配置`);
+          } else if (modifiers.modelMapping?.suffix) {
+            await session.send(`已启用后缀「-${modifiers.modelMapping.suffix}」配置`);
+          }
+          return processImageWithTimeout(session, img, mergedPrompt, style.commandName, requestContext);
         });
         logger.info(`已注册命令: ${style.commandName}`);
       }

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "koishi-plugin-aka-ai-generator",
   "description": "自用AI生成插件(GPTGod & Yunwu)",
-  "version": "0.1.9",
+  "version": "0.2.1",
   "main": "lib/index.js",
   "typings": "lib/index.d.ts",
   "files": [