koishi-plugin-aka-ai-generator 0.3.9 → 0.4.0

package/lib/index.js

@@ -587,11 +587,31 @@ async function downloadImageAsBase643(ctx, url, timeout, logger) {
   }
 }
 __name(downloadImageAsBase643, "downloadImageAsBase64");
-function parseGeminiResponse(response) {
+function parseGeminiResponse(response, logger) {
   try {
     const images = [];
+    if (!response) {
+      logger?.error("Gemini API 响应为空");
+      return [];
+    }
+    if (response.error) {
+      logger?.error("Gemini API 返回错误", { error: response.error });
+      throw new Error(`Gemini API 错误: ${response.error.message || JSON.stringify(response.error)}`);
+    }
     if (response.candidates && response.candidates.length > 0) {
       for (const candidate of response.candidates) {
+        if (candidate.finishReason && candidate.finishReason !== "STOP") {
+          logger?.warn("Gemini 响应 finishReason 异常", {
+            finishReason: candidate.finishReason,
+            safetyRatings: candidate.safetyRatings
+          });
+          if (candidate.finishReason === "SAFETY" || candidate.finishReason === "RECITATION") {
+            throw new Error(`内容被阻止: ${candidate.finishReason}，可能包含不安全的内容`);
+          }
+          if (candidate.finishReason !== "MAX_TOKENS") {
+            logger?.warn("Gemini 响应可能不完整", { finishReason: candidate.finishReason });
+          }
+        }
         if (candidate.content && candidate.content.parts) {
           for (const part of candidate.content.parts) {
             if (part.inlineData && part.inlineData.data) {
@@ -599,21 +619,39 @@ function parseGeminiResponse(response) {
               const mimeType = part.inlineData.mimeType || "image/jpeg";
               const dataUrl = `data:${mimeType};base64,${base64Data}`;
               images.push(dataUrl);
+              logger?.debug("从响应中提取到图片 (inlineData)", { mimeType, dataLength: base64Data.length });
             } else if (part.inline_data && part.inline_data.data) {
               const base64Data = part.inline_data.data;
               const mimeType = part.inline_data.mime_type || "image/jpeg";
               const dataUrl = `data:${mimeType};base64,${base64Data}`;
               images.push(dataUrl);
+              logger?.debug("从响应中提取到图片 (inline_data)", { mimeType, dataLength: base64Data.length });
             } else if (part.fileData && part.fileData.fileUri) {
               images.push(part.fileData.fileUri);
+              logger?.debug("从响应中提取到图片 (fileData)", { fileUri: part.fileData.fileUri });
+            } else if (part.text) {
+              logger?.warn("响应中包含文本而非图片", { text: part.text.substring(0, 100) });
             }
           }
+        } else {
+          logger?.warn("候选响应中没有 content.parts", { candidate: JSON.stringify(candidate).substring(0, 200) });
         }
       }
+    } else {
+      logger?.warn("Gemini API 响应中没有 candidates", { response: JSON.stringify(response).substring(0, 500) });
+    }
+    if (images.length === 0) {
+      logger?.error("未能从 Gemini API 响应中提取到任何图片", {
+        hasCandidates: !!response.candidates,
+        candidatesCount: response.candidates?.length || 0,
+        responseKeys: Object.keys(response),
+        firstCandidate: response.candidates?.[0] ? JSON.stringify(response.candidates[0]).substring(0, 300) : null
+      });
     }
     return images;
   } catch (error) {
-    return [];
+    logger?.error("解析 Gemini 响应时出错", { error: error.message, stack: error.stack });
+    throw error;
   }
 }
 __name(parseGeminiResponse, "parseGeminiResponse");
@@ -626,12 +664,18 @@ var GeminiProvider = class {
     this.config = config;
   }
   async generateImages(prompt, imageUrls, numImages) {
-    const urls = Array.isArray(imageUrls) ? imageUrls : [imageUrls];
+    let urls = [];
+    if (Array.isArray(imageUrls)) {
+      urls = imageUrls.filter((url) => url && typeof url === "string" && url.trim());
+    } else if (imageUrls && typeof imageUrls === "string" && imageUrls.trim()) {
+      urls = [imageUrls];
+    }
     const logger = this.config.logger;
     const ctx = this.config.ctx;
-    logger.debug("开始下载图片并转换为Base64", { urls });
+    logger.debug("开始处理图片输入", { urls, promptLength: prompt.length, isTextToImage: urls.length === 0 });
     const imageParts = [];
     for (const url of urls) {
+      if (!url || !url.trim()) continue;
       const { data, mimeType } = await downloadImageAsBase643(
         ctx,
         url,
@@ -678,14 +722,24 @@ var GeminiProvider = class {
             timeout: this.config.apiTimeout * 1e3
           }
         );
-        const images = parseGeminiResponse(response);
+        const images = parseGeminiResponse(response, logger);
+        if (images.length === 0) {
+          logger.warn("Gemini API 调用成功但未解析到图片", {
+            current: i + 1,
+            total: numImages,
+            responseHasCandidates: !!response.candidates,
+            responseKeys: Object.keys(response)
+          });
+        } else {
+          logger.success("Gemini API 调用成功", { current: i + 1, total: numImages, imagesCount: images.length });
+        }
         allImages.push(...images);
-        logger.success("Gemini API 调用成功", { current: i + 1, total: numImages });
       } catch (error) {
         logger.error("Gemini API 调用失败", {
           message: error?.message || "未知错误",
           code: error?.code,
           status: error?.response?.status,
+          responseData: error?.response?.data ? JSON.stringify(error.response.data).substring(0, 500) : void 0,
           current: i + 1,
           total: numImages
         });
@@ -696,6 +750,10 @@ var GeminiProvider = class {
         throw new Error(`图像处理API调用失败: ${error?.message || "未知错误"}`);
       }
     }
+    if (allImages.length === 0) {
+      logger.error("所有 Gemini API 调用都未生成图片", { numImages });
+      throw new Error("未能从 Gemini API 生成图片，请检查 prompt 和模型配置");
+    }
     return allImages;
   }
 };

package/package.json

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