koishi-plugin-aka-ai-generator 0.3.1 → 0.3.3

package/lib/index.d.ts

@@ -9,7 +9,6 @@ export interface ModelMappingConfig {
 export interface StyleConfig {
     commandName: string;
     prompt: string;
-    mode?: 'single' | 'multiple';
 }
 export interface StyleGroupConfig {
     prompts: StyleConfig[];

package/lib/index.js

@@ -753,12 +753,8 @@ var COMMANDS = {
   IMAGE_COMMANDS: "图像指令"
 };
 var StyleItemSchema = import_koishi.Schema.object({
-  commandName: import_koishi.Schema.string().required().description("命令名称（不含前缀斜杠）"),
-  prompt: import_koishi.Schema.string().role("textarea", { rows: 4 }).required().description("生成 prompt"),
-  mode: import_koishi.Schema.union([
-    import_koishi.Schema.const("single").description("单图模式"),
-    import_koishi.Schema.const("multiple").description("多图模式")
-  ]).default("single").description("图片输入模式")
+  commandName: import_koishi.Schema.string().required().description("命令名称").role("table-cell", { width: 100 }),
+  prompt: import_koishi.Schema.string().role("textarea", { rows: 4 }).required().description("生成 prompt")
 });
 var Config = import_koishi.Schema.intersect([
   import_koishi.Schema.object({
@@ -805,13 +801,11 @@ var Config = import_koishi.Schema.intersect([
     styles: import_koishi.Schema.array(StyleItemSchema).role("table").default([
       {
         commandName: "变手办",
-        prompt: "将这张照片变成手办模型。在它后面放置一个印有图像主体的盒子，桌子上有一台电脑显示Blender建模过程。在盒子前面添加一个圆形塑料底座，角色手办站在上面。如果可能的话，将场景设置在室内",
-        mode: "single"
+        prompt: "将这张照片变成手办模型。在它后面放置一个印有图像主体的盒子，桌子上有一台电脑显示Blender建模过程。在盒子前面添加一个圆形塑料底座，角色手办站在上面。如果可能的话，将场景设置在室内"
       },
       {
         commandName: "变写实",
-        prompt: "请根据用户提供的图片，在严格保持主体身份、外观特征与姿态不变的前提下，生成一张照片级真实感的超写实摄影作品。要求：1. 采用专业相机拍摄（如佳能EOS R5），使用85mm f/1.4人像镜头，呈现柯达Portra 400胶片质感，8K超高清画质，HDR高动态范围，电影级打光效果；2. 画面应具有照片级真实感、超现实主义风格和高细节表现，确保光影、皮肤质感、服饰纹理与背景环境都贴近真实世界；3. 使用自然光影营造真实氛围，呈现raw and natural的原始自然感，具有authentic film snapshot的真实胶片质感；4. 整体需具备tactile feel触感质感和simulated texture模拟纹理细节，可以适度优化噪点与瑕疵，但不要改变主体特征或添加额外元素；5. 整体效果需像专业摄影棚拍摄的真实照片，具有电影级画质；6. 如果主体是人物脸部，脸部生成效果应参考欧美混血白人精致美丽帅气英俊的外观特征进行生成，保持精致立体的五官轮廓、健康光泽的肌肤质感、优雅的气质和自然的表情，确保面部特征协调美观。",
-        mode: "single"
+        prompt: "请根据用户提供的图片，在严格保持主体身份、外观特征与姿态不变的前提下，生成一张照片级真实感的超写实摄影作品。要求：1. 采用专业相机拍摄（如佳能EOS R5），使用85mm f/1.4人像镜头，呈现柯达Portra 400胶片质感，8K超高清画质，HDR高动态范围，电影级打光效果；2. 画面应具有照片级真实感、超现实主义风格和高细节表现，确保光影、皮肤质感、服饰纹理与背景环境都贴近真实世界；3. 使用自然光影营造真实氛围，呈现raw and natural的原始自然感，具有authentic film snapshot的真实胶片质感；4. 整体需具备tactile feel触感质感和simulated texture模拟纹理细节，可以适度优化噪点与瑕疵，但不要改变主体特征或添加额外元素；5. 整体效果需像专业摄影棚拍摄的真实照片，具有电影级画质；6. 如果主体是人物脸部，脸部生成效果应参考欧美混血白人精致美丽帅气英俊的外观特征进行生成，保持精致立体的五官轮廓、健康光泽的肌肤质感、优雅的气质和自然的表情，确保面部特征协调美观。"
       }
     ]).description("自定义风格命令配置")
   }),
@@ -1426,7 +1420,7 @@ ${infoParts.join("\n")}`;
   if (styleDefinitions.length > 0) {
     for (const style of styleDefinitions) {
       if (style.commandName && style.prompt) {
-        ctx.command(`${style.commandName} [img:text]`, "图像风格转换").option("num", "-n <num:number> 生成图片数量 (1-4)").action(async (argv, img) => {
+        ctx.command(`${style.commandName} [img:text]`, "图像风格转换").option("num", "-n <num:number> 生成图片数量 (1-4)").option("multiple", "-m 允许多图输入").action(async (argv, img) => {
           const { session, options } = argv;
           if (!session?.userId) return "会话无效";
           const modifiers = parseStyleCommandModifiers(argv, img);
@@ -1465,7 +1459,7 @@ ${infoParts.join("\n")}`;
             displayInfo.modelId = modifiers.modelMapping.modelId;
             displayInfo.modelDescription = modifiers.modelMapping.suffix || modifiers.modelMapping.modelId;
           }
-          const mode = style.mode || "single";
+          const mode = options?.multiple ? "multiple" : "single";
           return processImageWithTimeout(session, img, mergedPrompt, style.commandName, requestContext, displayInfo, mode);
         });
         logger.info(`已注册命令: ${style.commandName}`);
@@ -1484,113 +1478,18 @@ ${infoParts.join("\n")}`;
     };
     return processImageWithTimeout(session, prompt, "", COMMANDS.TXT_TO_IMG, requestContext, {}, "text");
   });
-  ctx.command(COMMANDS.IMG_TO_IMG, "使用自定义prompt进行图像处理").option("num", "-n <num:number> 生成图片数量 (1-4)").action(async ({ session, options }) => {
+  ctx.command(`${COMMANDS.IMG_TO_IMG} [img:text]`, "使用自定义prompt进行图像处理").option("num", "-n <num:number> 生成图片数量 (1-4)").option("multiple", "-m 允许多图输入").action(async ({ session, options }, img) => {
     if (!session?.userId) return "会话无效";
-    return Promise.race([
-      (async () => {
-        const userId = session.userId;
-        if (!userId) return "会话无效";
-        if (activeTasks.has(userId)) {
-          return "您有一个图像处理任务正在进行中，请等待完成";
-        }
-        await session.send("图片+描述");
-        const collectedImages = [];
-        let prompt = "";
-        while (true) {
-          const msg = await session.prompt(6e4);
-          if (!msg) {
-            return "等待超时，请重试";
-          }
-          const elements = import_koishi.h.parse(msg);
-          const images = import_koishi.h.select(elements, "img");
-          const textElements = import_koishi.h.select(elements, "text");
-          const text = textElements.map((el) => el.attrs.content).join(" ").trim();
-          if (images.length > 0) {
-            if (collectedImages.length > 0) {
-              return '本功能仅支持处理一张图片，如需合成多张图片请使用"合成图"命令';
-            }
-            if (images.length > 1) {
-              return '本功能仅支持处理一张图片，检测到多张图片。如需合成多张图片请使用"合成图"命令';
-            }
-            for (const img of images) {
-              collectedImages.push(img.attrs.src);
-            }
-            if (text) {
-              prompt = text;
-              break;
-            }
-            await session.send("请发送描述");
-            continue;
-          }
-          if (text) {
-            if (collectedImages.length === 0) {
-              return "未检测到图片，请先发送图片";
-            }
-            prompt = text;
-            break;
-          }
-          return "未检测到有效内容，请重新发送";
-        }
-        if (collectedImages.length === 0) {
-          return "未检测到图片，请重新发送";
-        }
-        if (collectedImages.length > 1) {
-          return '本功能仅支持处理一张图片，检测到多张图片。如需合成多张图片请使用"合成图"命令';
-        }
-        if (!prompt) {
-          return "未检测到prompt描述，请重新发送";
-        }
-        const imageUrl = collectedImages[0];
-        const imageCount = options?.num || config.defaultNumImages;
-        if (imageCount < 1 || imageCount > 4) {
-          return "生成数量必须在 1-4 之间";
-        }
-        const limitCheck = await checkDailyLimit(userId, imageCount);
-        if (!limitCheck.allowed) {
-          return limitCheck.message;
-        }
-        logger.info("开始自定义图像处理", {
-          userId,
-          imageUrl,
-          prompt,
-          numImages: imageCount
-        });
-        await session.send(`开始处理图片（自定义prompt）...
-Prompt: ${prompt}`);
-        try {
-          activeTasks.set(userId, "processing");
-          const resultImages = await requestProviderImages(prompt, imageUrl, imageCount);
-          if (resultImages.length === 0) {
-            activeTasks.delete(userId);
-            return "图像处理失败：未能生成图片";
-          }
-          await session.send("图像处理完成！");
-          for (let i = 0; i < resultImages.length; i++) {
-            await session.send(import_koishi.h.image(resultImages[i]));
-            if (resultImages.length > 1 && i < resultImages.length - 1) {
-              await new Promise((resolve) => setTimeout(resolve, 1e3));
-            }
-          }
-          await recordUserUsage(session, COMMANDS.IMG_TO_IMG, resultImages.length);
-          activeTasks.delete(userId);
-        } catch (error) {
-          activeTasks.delete(userId);
-          logger.error("自定义图像处理失败", { userId, error });
-          if (error?.message) {
-            return `图像处理失败：${error.message}`;
-          }
-          return "图像处理失败，请稍后重试";
-        }
-      })(),
-      new Promise(
-        (_, reject) => setTimeout(() => reject(new Error("命令执行超时")), config.commandTimeout * 1e3)
-      )
-    ]).catch((error) => {
-      const userId = session.userId;
-      if (userId) activeTasks.delete(userId);
-      logger.error("自定义图像处理超时或失败", { userId, error });
-      return error.message === "命令执行超时" ? "图像处理超时，请重试" : `图像处理失败：${error.message}`;
-    });
+    const numImages = options?.num || config.defaultNumImages;
+    const mode = options?.multiple ? "multiple" : "single";
+    const limitCheck = await checkDailyLimit(session.userId, numImages);
+    if (!limitCheck.allowed) {
+      return limitCheck.message;
+    }
+    const requestContext = {
+      numImages
+    };
+    return processImageWithTimeout(session, img, "", COMMANDS.IMG_TO_IMG, requestContext, {}, mode);
   });
   ctx.command(COMMANDS.COMPOSE_IMAGE, "合成多张图片，使用自定义prompt控制合成效果").option("num", "-n <num:number> 生成图片数量 (1-4)").action(async ({ session, options }) => {
     if (!session?.userId) return "会话无效";

package/package.json

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