koishi-plugin-aka-ai-generator 0.0.4 → 0.0.6

package/lib/index.js

@@ -415,6 +415,34 @@ var GptGodProvider = class {
         }
       );
       logger.success("GPTGod 图像编辑 API 调用成功");
+      if (response?.choices?.length > 0) {
+        const firstChoice = response.choices[0];
+        const messageContent = firstChoice.message?.content;
+        let errorMessage = "";
+        if (typeof messageContent === "string") {
+          errorMessage = messageContent;
+        } else if (Array.isArray(messageContent)) {
+          const textParts = messageContent.filter((part) => part?.type === "text" && part?.text).map((part) => part.text).join(" ");
+          errorMessage = textParts;
+        } else if (messageContent?.text) {
+          errorMessage = messageContent.text;
+        }
+        if (errorMessage && (errorMessage.includes("PROHIBITED_CONTENT") || errorMessage.includes("blocked by Google Gemini") || errorMessage.includes("prohibited under official usage policies") || errorMessage.toLowerCase().includes("content is prohibited"))) {
+          logger.error("内容被 Google Gemini 政策拦截", {
+            errorMessage: errorMessage.substring(0, 200),
+            finishReason: firstChoice.finish_reason
+          });
+          throw new Error("内容被安全策略拦截");
+        }
+        if (errorMessage && (errorMessage.toLowerCase().includes("error") || errorMessage.toLowerCase().includes("failed") || errorMessage.toLowerCase().includes("blocked")) && !errorMessage.match(/https?:\/\//)) {
+          logger.error("API 返回错误消息", {
+            errorMessage: errorMessage.substring(0, 200),
+            finishReason: firstChoice.finish_reason
+          });
+          const shortError = errorMessage.length > 50 ? errorMessage.substring(0, 50) + "..." : errorMessage;
+          throw new Error(`处理失败：${shortError}`);
+        }
+      }
       if (this.config.logLevel === "debug") {
         logger.debug("GPTGod API 响应结构", {
           hasChoices: !!response?.choices,
@@ -435,9 +463,20 @@ var GptGodProvider = class {
           warnData.responsePreview = JSON.stringify(response).substring(0, 500);
         }
         logger.warn("生成的图片数量不足", warnData);
+        if (images.length === 0 && response?.choices?.[0]?.message?.content) {
+          const content = response.choices[0].message.content;
+          const contentText = typeof content === "string" ? content : Array.isArray(content) ? content.map((p) => p?.text || "").join(" ") : "";
+          if (contentText && !contentText.match(/https?:\/\//)) {
+            const shortError = contentText.length > 50 ? contentText.substring(0, 50) + "..." : contentText;
+            throw new Error(`生成失败：${shortError}`);
+          }
+        }
       }
       return images;
     } catch (error) {
+      if (error?.message && (error.message.includes("内容被安全策略拦截") || error.message.includes("生成失败") || error.message.includes("处理失败"))) {
+        throw error;
+      }
       logger.error("GPTGod 图像编辑 API 调用失败", {
         message: error?.message || "未知错误",
         code: error?.code,
@@ -530,7 +569,7 @@ var Config = import_koishi.Schema.intersect([
       {
         commandName: "变写实",
         commandDescription: "以真实摄影风格重建主体",
-        prompt: "请根据用户提供的图片，在保持主体身份、外观与姿态的前提下生成一张超写实的摄影作品。确保光影、皮肤质感、服饰纹理与背景环境都贴近真实世界，可以适度优化噪点与瑕疵，但不要改变主体特征或添加额外元素，整体效果需像专业摄影棚拍摄。",
+        prompt: "请根据用户提供的图片，在保持主体身份、外观与姿态的前提下生成一张超写实的摄影作品。采用Ultra-realistic 3D rendered风格，确保光影、皮肤质感、服饰纹理与背景环境都贴近真实世界。画面应呈现raw and natural的原始自然感，具有authentic film snapshot的真实胶片质感。使用strong contrast between light and dark营造强烈明暗对比，产生deep shadows深阴影效果。整体需具备tactile feel触感质感和simulated texture模拟纹理细节，可以适度优化噪点与瑕疵，但不要改变主体特征或添加额外元素，整体效果需像专业摄影棚拍摄的真实照片。",
         enabled: true
       },
       {
@@ -918,6 +957,9 @@ function apply(ctx, config) {
     } catch (error) {
       activeTasks.delete(userId);
       logger.error("图像处理失败", { userId, error });
+      if (error?.message && (error.message.includes("内容被安全策略拦截") || error.message.includes("生成失败") || error.message.includes("处理失败"))) {
+        return error.message;
+      }
       return "图像处理失败，请稍后重试";
     }
   }
@@ -950,7 +992,7 @@ function apply(ctx, config) {
         if (activeTasks.has(userId)) {
           return "您有一个图像处理任务正在进行中，请等待完成";
         }
-        await session.send('请发送一张图片和prompt，支持两种方式：\n1. 同时发送：[图片] + prompt描述\n2. 分步发送：先发送一张图片，再发送prompt文字\n\n例如：[图片] 让这张图片变成油画风格\n\n注意：本功能仅支持处理一张图片，多张图片请使用"合成图像"命令');
+        await session.send('图片+描述\n\n多张图片使用"合成图像"指令');
         const collectedImages = [];
         let prompt = "";
         while (true) {
@@ -1029,6 +1071,9 @@ 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;
+          }
           return "图像处理失败，请稍后重试";
         }
       })(),
@@ -1055,7 +1100,7 @@ Prompt: ${prompt}`);
         if (activeTasks.has(userId)) {
           return "您有一个图像处理任务正在进行中，请等待完成";
         }
-        await session.send("请发送多张图片和prompt，支持两种方式：\n1. 同时发送：[图片1] [图片2]... + prompt描述\n2. 分步发送：先发送多张图片，再发送prompt文字\n\n例如：[图片1] [图片2] 将这两张图片合成一张");
+        await session.send("多张图片+描述");
         const collectedImages = [];
         let prompt = "";
         while (true) {
@@ -1125,6 +1170,9 @@ 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;
+          }
           return "图片合成失败，请稍后重试";
         }
       })(),

package/package.json

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