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

package/lib/index.js

@@ -598,6 +598,39 @@ function parseGeminiResponse(response, logger) {
       logger?.error("Gemini API 返回错误", { error: response.error });
       throw new Error(`Gemini API 错误: ${response.error.message || JSON.stringify(response.error)}`);
     }
+    if (response.promptFeedback) {
+      const blockReason = response.promptFeedback.blockReason;
+      const safetyRatings = response.promptFeedback.safetyRatings;
+      if (blockReason) {
+        logger?.error("Gemini API 请求被阻止", {
+          blockReason,
+          safetyRatings,
+          blockReasonMessage: response.promptFeedback.blockReasonMessage
+        });
+        let errorMessage = "请求被 Gemini API 阻止";
+        switch (blockReason) {
+          case "SAFETY":
+            errorMessage = "内容被安全策略阻止，可能包含不安全的内容";
+            break;
+          case "OTHER":
+            errorMessage = "请求被阻止（原因：OTHER），可能是内容不符合使用政策或模型无法处理";
+            if (response.promptFeedback.blockReasonMessage) {
+              errorMessage += `：${response.promptFeedback.blockReasonMessage}`;
+            }
+            break;
+          case "RECITATION":
+            errorMessage = "内容包含受版权保护的内容";
+            break;
+          default:
+            errorMessage = `请求被阻止（原因：${blockReason}）`;
+        }
+        if (safetyRatings && Array.isArray(safetyRatings) && safetyRatings.length > 0) {
+          const ratings = safetyRatings.map((r) => `${r.category}:${r.probability}`).join(", ");
+          errorMessage += ` [安全评分: ${ratings}]`;
+        }
+        throw new Error(errorMessage);
+      }
+    }
     if (response.candidates && response.candidates.length > 0) {
       for (const candidate of response.candidates) {
         if (candidate.finishReason && candidate.finishReason !== "STOP") {
@@ -638,14 +671,29 @@ function parseGeminiResponse(response, logger) {
         }
       }
     } else {
-      logger?.warn("Gemini API 响应中没有 candidates", { response: JSON.stringify(response).substring(0, 500) });
+      const hasPromptFeedback = !!response.promptFeedback;
+      const responseKeys = Object.keys(response);
+      logger?.error("Gemini API 响应中没有 candidates", {
+        response: JSON.stringify(response).substring(0, 500),
+        hasPromptFeedback,
+        responseKeys
+      });
+      if (!hasPromptFeedback) {
+        throw new Error("Gemini API 响应格式异常：既没有生成内容也没有反馈信息");
+      }
+      if (hasPromptFeedback && !response.promptFeedback.blockReason) {
+        logger?.warn("有 promptFeedback 但没有 blockReason，也没有 candidates", {
+          promptFeedback: response.promptFeedback
+        });
+      }
     }
     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
+        firstCandidate: response.candidates?.[0] ? JSON.stringify(response.candidates[0]).substring(0, 300) : null,
+        promptFeedback: response.promptFeedback ? JSON.stringify(response.promptFeedback) : null
       });
     }
     return images;

package/package.json

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