@hiper2d/ai-agents 0.5.2 → 0.5.3

package/dist/index.js

@@ -4581,6 +4581,18 @@ ${schemaDescription}`;
 
 // src/agents/qwen-agent.ts
 var import_openai9 = require("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
@@ -4705,7 +4717,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) {
@@ -4725,6 +4737,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));
     }
   }
@@ -4758,7 +4773,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) {
@@ -4774,6 +4789,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));
     }
   }