@hiper2d/ai-agents 0.5.1 → 0.5.3

package/dist/index.mjs

@@ -714,9 +714,11 @@ var ModelInvalidResponseError = class extends ModelError {
   }
 };
 var ModelRefusalError = class extends ModelError {
-  constructor(modelType, message = `${modelType} refused to answer (stop_reason: refusal)`) {
+  reason;
+  constructor(modelType, message = `${modelType} refused to answer (stop_reason: refusal)`, reason) {
     super(message, modelType);
     this.name = "ModelRefusalError";
+    this.reason = reason;
   }
 };
 
@@ -2654,7 +2656,7 @@ ${schemaDescription}`;
         throw new Error(this.errorMessages.apiError(apiError));
       }
       if (response.stop_reason === "refusal") {
-        throw new ModelRefusalError(this.model);
+        throw new ModelRefusalError(this.model, void 0, "refusal");
       }
       if (!("content" in response) || !Array.isArray(response.content) || response.content.length === 0) {
         throw new Error(this.errorMessages.emptyResponse);
@@ -2748,7 +2750,7 @@ ${schemaDescription}`;
         throw new Error(this.errorMessages.apiError(apiError));
       }
       if (response.stop_reason === "refusal") {
-        throw new ModelRefusalError(this.model);
+        throw new ModelRefusalError(this.model, void 0, "refusal");
       }
       if (!("content" in response) || !Array.isArray(response.content) || response.content.length === 0) {
         throw new Error(this.errorMessages.emptyResponse);
@@ -2804,6 +2806,45 @@ ${schemaDescription}`;
 
 // src/agents/google-agent.ts
 import { GoogleGenAI as GoogleGenAI3 } from "@google/genai";
+var GEMINI_REFUSAL_FINISH_REASONS = /* @__PURE__ */ new Set([
+  "SAFETY",
+  "RECITATION",
+  "BLOCKLIST",
+  "PROHIBITED_CONTENT",
+  "SPII",
+  "IMAGE_SAFETY",
+  "IMAGE_PROHIBITED_CONTENT",
+  "IMAGE_RECITATION"
+]);
+function throwForEmptyGeminiResponse(model, response, fallback) {
+  const blockReason = response?.promptFeedback?.blockReason;
+  const candidate = response?.candidates?.[0];
+  const finishReason = candidate?.finishReason;
+  const finishMessage = candidate?.finishMessage;
+  const ratings = (candidate?.safetyRatings ?? response?.promptFeedback?.safetyRatings ?? []).filter((r) => r?.blocked || r?.probability && r.probability !== "NEGLIGIBLE");
+  const ratingText = ratings.length ? `; safetyRatings: ${ratings.map((r) => `${r.category}=${r.probability ?? (r.blocked ? "blocked" : "?")}`).join(", ")}` : "";
+  if (blockReason) {
+    throw new ModelRefusalError(
+      model,
+      `${model} refused the prompt (blockReason: ${blockReason}${finishMessage ? `, ${finishMessage}` : ""}${ratingText})`,
+      blockReason
+    );
+  }
+  if (finishReason && GEMINI_REFUSAL_FINISH_REASONS.has(finishReason)) {
+    throw new ModelRefusalError(
+      model,
+      `${model} refused to answer (finishReason: ${finishReason}${finishMessage ? `, ${finishMessage}` : ""}${ratingText})`,
+      finishReason
+    );
+  }
+  if (finishReason === "MAX_TOKENS") {
+    throw new ModelInvalidResponseError(model, "empty response, finishReason MAX_TOKENS (thinking consumed the whole output budget)", true);
+  }
+  const bits = [`finishReason=${finishReason ?? "none"}`];
+  if (!response?.candidates?.length) bits.push("no candidates");
+  if (finishMessage) bits.push(`finishMessage="${finishMessage}"`);
+  throw new Error(`${fallback} (${bits.join(", ")})`);
+}
 var GoogleAgent = class extends AbstractAgent {
   client;
   defaultConfig = {
@@ -2815,7 +2856,7 @@ var GoogleAgent = class extends AbstractAgent {
   };
   // Error message templates
   errorMessages = {
-    emptyResponse: "Empty response from Google API - check logs for detailed response info",
+    emptyResponse: "Empty response from Google API",
     invalidFormat: "Invalid response format from Google API",
     apiError: (error) => `Failed to get response from Google API: ${error instanceof Error ? error.message : String(error)}`,
     unsupportedRole: (role) => `Unsupported role type: ${role}`
@@ -2987,7 +3028,7 @@ var GoogleAgent = class extends AbstractAgent {
       }
       this.logger(`Zod schema response received - hasText: ${!!response.text}, textLength: ${response.text ? response.text.length : 0}`);
       if (!response.text) {
-        throw new Error(this.errorMessages.emptyResponse);
+        throwForEmptyGeminiResponse(this.model, response, this.errorMessages.emptyResponse);
       }
       const parsedData = parseAndValidateLlmJson(response.text, zodSchema, (m) => this.logger(m));
       if (parsedData) {
@@ -3075,7 +3116,7 @@ var GoogleAgent = class extends AbstractAgent {
       }
       this.logger(`Plain text response received - hasText: ${!!response.text}, textLength: ${response.text ? response.text.length : 0}`);
       if (!response.text) {
-        throw new Error(this.errorMessages.emptyResponse);
+        throwForEmptyGeminiResponse(this.model, response, this.errorMessages.emptyResponse);
       }
       this.logReply(response.text, thinkingContent || void 0, tokenUsage);
       return [response.text, thinkingContent, tokenUsage, googleThoughtSignature || void 0];
@@ -4371,6 +4412,18 @@ ${schemaDescription}`;
 
 // src/agents/qwen-agent.ts
 import { OpenAI as OpenAI9 } from "openai";
+function qwenRefusalFrom(model, apiError) {
+  const message = apiError instanceof Error ? apiError.message : String(apiError);
+  if (!/DataInspectionFailed|inappropriate content/i.test(message)) {
+    return void 0;
+  }
+  const detail = message.replace(/^[\s\S]*DataInspectionFailed:\s*/, "").trim();
+  return new ModelRefusalError(
+    model,
+    `${model} refused the prompt (refusalReason: DataInspectionFailed${detail ? `; ${detail}` : ""})`,
+    "DataInspectionFailed"
+  );
+}
 var QwenAgent = class extends AbstractAgent {
   client;
   // A getter, not a field: `maxOutputTokens` can be raised after construction, and a field
@@ -4495,7 +4548,7 @@ ${schemaDescription}`;
         completion = await this.client.chat.completions.create(params);
       } catch (apiError) {
         this.logger(this.logTemplates.error(this.name, apiError));
-        throw new Error(this.errorMessages.apiError(apiError));
+        throw qwenRefusalFrom(this.model, apiError) ?? new Error(this.errorMessages.apiError(apiError));
       }
       const rawReply = completion.choices[0]?.message?.content;
       if (!rawReply) {
@@ -4515,6 +4568,9 @@ ${schemaDescription}`;
       return [validated, thinkingContent, tokenUsage];
     } catch (error) {
       this.logger(this.logTemplates.error(this.name, error));
+      if (error instanceof ModelError) {
+        throw error;
+      }
       throw new Error(this.errorMessages.apiError(error));
     }
   }
@@ -4548,7 +4604,7 @@ ${openAIMessages[0].content}`;
         completion = await this.client.chat.completions.create(params);
       } catch (apiError) {
         this.logger(this.logTemplates.error(this.name, apiError));
-        throw new Error(this.errorMessages.apiError(apiError));
+        throw qwenRefusalFrom(this.model, apiError) ?? new Error(this.errorMessages.apiError(apiError));
       }
       const rawReply = completion.choices[0]?.message?.content;
       if (!rawReply) {
@@ -4564,6 +4620,9 @@ ${openAIMessages[0].content}`;
       return [reply, thinkingContent, tokenUsage];
     } catch (error) {
       this.logger(this.logTemplates.error(this.name, error));
+      if (error instanceof ModelError) {
+        throw error;
+      }
       throw new Error(this.errorMessages.apiError(error));
     }
   }