koishi-plugin-aka-ai-generator 0.2.14 → 0.2.16

package/lib/index.d.ts

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

package/lib/index.js

@@ -604,7 +604,11 @@ var 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")
+  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("图片输入模式")
 });
 var Config = import_koishi.Schema.intersect([
   import_koishi.Schema.object({
@@ -646,11 +650,13 @@ var Config = import_koishi.Schema.intersect([
     styles: import_koishi.Schema.array(StyleItemSchema).role("table").default([
       {
         commandName: "变手办",
-        prompt: "将这张照片变成手办模型。在它后面放置一个印有图像主体的盒子，桌子上有一台电脑显示Blender建模过程。在盒子前面添加一个圆形塑料底座，角色手办站在上面。如果可能的话，将场景设置在室内"
+        prompt: "将这张照片变成手办模型。在它后面放置一个印有图像主体的盒子，桌子上有一台电脑显示Blender建模过程。在盒子前面添加一个圆形塑料底座，角色手办站在上面。如果可能的话，将场景设置在室内",
+        mode: "single"
       },
       {
         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. 如果主体是人物脸部，脸部生成效果应参考欧美混血白人精致美丽帅气英俊的外观特征进行生成，保持精致立体的五官轮廓、健康光泽的肌肤质感、优雅的气质和自然的表情，确保面部特征协调美观。"
+        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"
       }
     ]).description("自定义风格命令配置")
   }),
@@ -1111,55 +1117,72 @@ function apply(ctx, config) {
     });
   }
   __name(recordUserUsage, "recordUserUsage");
-  async function getImageUrl(img, session) {
-    let url = null;
-    if (img) {
-      url = img.attrs?.src || null;
-      if (url) {
-        if (config.logLevel === "debug") {
-          logger.debug("从命令参数获取图片", { url });
+  async function getInputData(session, imgParam, mode) {
+    const collectedImages = [];
+    let collectedText = "";
+    if (imgParam) {
+      if (typeof imgParam === "object" && imgParam.attrs?.src) {
+        collectedImages.push(imgParam.attrs.src);
+      } else if (typeof imgParam === "string") {
+        if (imgParam.startsWith("http") || imgParam.startsWith("data:")) {
+          collectedImages.push(imgParam);
         }
-        return url;
       }
     }
-    let elements = session.quote?.elements;
-    if (elements) {
-      const images2 = import_koishi.h.select(elements, "img");
-      if (images2.length > 0) {
-        if (images2.length > 1) {
-          await session.send('本功能仅支持处理一张图片，检测到多张图片。如需合成多张图片请使用"合成图像"命令');
-          return null;
+    if (session.quote?.elements) {
+      const quoteImages = import_koishi.h.select(session.quote.elements, "img");
+      for (const img of quoteImages) {
+        if (img.attrs.src) collectedImages.push(img.attrs.src);
+      }
+    }
+    if (collectedImages.length > 0) {
+      if (mode === "single") {
+        if (collectedImages.length > 1) {
+          return { error: '本功能仅支持处理一张图片，检测到多张图片。如需合成多张图片请使用"合成图像"命令' };
+        }
+        return { images: collectedImages };
+      }
+      return { images: collectedImages };
+    }
+    const promptMsg = mode === "single" ? "请在30秒内发送一张图片" : "请发送图片（发送纯文字结束，至少需要2张）";
+    await session.send(promptMsg);
+    while (true) {
+      const msg = await session.prompt(mode === "multiple" ? 6e4 : 3e4);
+      if (!msg) return { error: "等待超时" };
+      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) {
+        for (const img of images) {
+          collectedImages.push(img.attrs.src);
         }
-        url = images2[0].attrs.src;
-        if (config.logLevel === "debug") {
-          logger.debug("从引用消息获取图片", { url });
+        if (mode === "single") {
+          if (collectedImages.length > 1) {
+            return { error: "本功能仅支持处理一张图片，检测到多张图片" };
+          }
+          if (text) collectedText = text;
+          break;
+        }
+        if (text) {
+          collectedText = text;
+          break;
         }
-        return url;
+        await session.send(`已收到 ${collectedImages.length} 张图片，继续发送或输入文字结束`);
+        continue;
+      }
+      if (text) {
+        if (collectedImages.length === 0) {
+          await session.send("未检测到图片，请先发送图片");
+          continue;
+        }
+        collectedText = text;
+        break;
       }
     }
-    await session.send("请在30秒内发送一张图片");
-    const msg = await session.prompt(3e4);
-    if (!msg) {
-      await session.send("等待超时");
-      return null;
-    }
-    elements = import_koishi.h.parse(msg);
-    const images = import_koishi.h.select(elements, "img");
-    if (images.length === 0) {
-      await session.send("未检测到图片，请重试");
-      return null;
-    }
-    if (images.length > 1) {
-      await session.send('本功能仅支持处理一张图片，检测到多张图片。如需合成多张图片请使用"合成图像"命令');
-      return null;
-    }
-    url = images[0].attrs.src;
-    if (config.logLevel === "debug") {
-      logger.debug("从用户输入获取图片", { url });
-    }
-    return url;
+    return { images: collectedImages, text: collectedText };
   }
-  __name(getImageUrl, "getImageUrl");
+  __name(getInputData, "getInputData");
   async function requestProviderImages(prompt, imageUrls, numImages, requestContext) {
     const providerType = requestContext?.provider || config.provider;
     const targetModelId = requestContext?.modelId;
@@ -1174,9 +1197,9 @@ function apply(ctx, config) {
     return await providerInstance.generateImages(prompt, imageUrls, numImages);
   }
   __name(requestProviderImages, "requestProviderImages");
-  async function processImageWithTimeout(session, img, prompt, styleName, requestContext, displayInfo) {
+  async function processImageWithTimeout(session, img, prompt, styleName, requestContext, displayInfo, mode = "single") {
     return Promise.race([
-      processImage(session, img, prompt, styleName, requestContext, displayInfo),
+      processImage(session, img, prompt, styleName, requestContext, displayInfo, mode),
       new Promise(
         (_, reject) => setTimeout(() => reject(new Error("命令执行超时")), config.commandTimeout * 1e3)
       )
@@ -1184,11 +1207,11 @@ function apply(ctx, config) {
       const userId = session.userId;
       if (userId) activeTasks.delete(userId);
       logger.error("图像处理超时或失败", { userId, error });
-      return error.message === "命令执行超时" ? "图像处理超时，请重试" : "图像处理失败，请稍后重试";
+      return error.message === "命令执行超时" ? "图像处理超时，请重试" : `图像处理失败：${error.message}`;
     });
   }
   __name(processImageWithTimeout, "processImageWithTimeout");
-  async function processImage(session, img, prompt, styleName, requestContext, displayInfo) {
+  async function processImage(session, img, prompt, styleName, requestContext, displayInfo, mode = "single") {
     const userId = session.userId;
     if (activeTasks.has(userId)) {
       return "您有一个图像处理任务正在进行中，请等待完成";
@@ -1197,17 +1220,22 @@ function apply(ctx, config) {
     if (imageCount < 1 || imageCount > 4) {
       return "生成数量必须在 1-4 之间";
     }
-    const imageUrl = await getImageUrl(img, session);
-    if (!imageUrl) {
-      return;
+    const inputResult = await getInputData(session, img, mode);
+    if ("error" in inputResult) {
+      return inputResult.error;
+    }
+    const { images: imageUrls, text: extraText } = inputResult;
+    let finalPrompt = prompt;
+    if (extraText) {
+      finalPrompt += " " + extraText;
     }
     const providerType = requestContext?.provider || config.provider;
     const providerModelId = requestContext?.modelId || (providerType === "yunwu" ? config.yunwuModelId : config.gptgodModelId);
     logger.info("开始图像处理", {
       userId,
-      imageUrl,
+      imageUrls,
       styleName,
-      prompt,
+      prompt: finalPrompt,
       numImages: imageCount,
       provider: providerType,
       modelId: providerModelId
@@ -1229,7 +1257,7 @@ ${infoParts.join("\n")}`;
     await session.send(statusMessage);
     try {
       activeTasks.set(userId, "processing");
-      const images = await requestProviderImages(prompt, imageUrl, imageCount, requestContext);
+      const images = await requestProviderImages(finalPrompt, imageUrls, imageCount, requestContext);
       if (images.length === 0) {
         activeTasks.delete(userId);
         return "图像处理失败：未能生成图片";
@@ -1246,8 +1274,8 @@ ${infoParts.join("\n")}`;
     } catch (error) {
       activeTasks.delete(userId);
       logger.error("图像处理失败", { userId, error });
-      if (error?.message && (error.message.includes("内容被安全策略拦截") || error.message.includes("生成失败") || error.message.includes("处理失败"))) {
-        return error.message;
+      if (error?.message) {
+        return `图像处理失败：${error.message}`;
       }
       return "图像处理失败，请稍后重试";
     }
@@ -1307,7 +1335,8 @@ ${infoParts.join("\n")}`;
             displayInfo.modelId = modifiers.modelMapping.modelId;
             displayInfo.modelDescription = modifiers.modelMapping.suffix || modifiers.modelMapping.modelId;
           }
-          return processImageWithTimeout(session, img, mergedPrompt, style.commandName, requestContext, displayInfo);
+          const mode = style.mode || "single";
+          return processImageWithTimeout(session, img, mergedPrompt, style.commandName, requestContext, displayInfo, mode);
         });
         logger.info(`已注册命令: ${style.commandName}`);
       }
@@ -1412,8 +1441,8 @@ Prompt: ${prompt}`);
         } catch (error) {
           activeTasks.delete(userId);
           logger.error("自定义图像处理失败", { userId, error });
-          if (error?.message && (error.message.includes("内容被安全策略拦截") || error.message.includes("生成失败") || error.message.includes("处理失败"))) {
-            return error.message;
+          if (error?.message) {
+            return `图像处理失败：${error.message}`;
           }
           return "图像处理失败，请稍后重试";
         }
@@ -1425,7 +1454,7 @@ Prompt: ${prompt}`);
       const userId = session.userId;
       if (userId) activeTasks.delete(userId);
       logger.error("自定义图像处理超时或失败", { userId, error });
-      return error.message === "命令执行超时" ? "图像处理超时，请重试" : "图像处理失败，请稍后重试";
+      return error.message === "命令执行超时" ? "图像处理超时，请重试" : `图像处理失败：${error.message}`;
     });
   });
   ctx.command(COMMANDS.COMPOSE_IMAGE, "合成多张图片，使用自定义prompt控制合成效果").option("num", "-n <num:number> 生成图片数量 (1-4)").action(async ({ session, options }) => {
@@ -1518,8 +1547,8 @@ Prompt: ${prompt}`);
         } catch (error) {
           activeTasks.delete(userId);
           logger.error("图片合成失败", { userId, error });
-          if (error?.message && (error.message.includes("内容被安全策略拦截") || error.message.includes("生成失败") || error.message.includes("处理失败"))) {
-            return error.message;
+          if (error?.message) {
+            return `图片合成失败：${error.message}`;
           }
           return "图片合成失败，请稍后重试";
         }
@@ -1531,7 +1560,7 @@ Prompt: ${prompt}`);
       const userId = session.userId;
       if (userId) activeTasks.delete(userId);
       logger.error("图片合成超时或失败", { userId, error });
-      return error.message === "命令执行超时" ? "图片合成超时，请重试" : "图片合成失败，请稍后重试";
+      return error.message === "命令执行超时" ? "图片合成超时，请重试" : `图片合成失败：${error.message}`;
     });
   });
   ctx.command(`${COMMANDS.RECHARGE} [content:text]`, "为用户充值次数（仅管理员）").action(async ({ session }, content) => {

package/package.json

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