lunanting/ai-chatbot
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix(agent): handle legacy image service failure messages

Browse Source
This commit is contained in:
134 2026-10-21 10:26:00 +08:00
parent f022fda584
commit f27d8dc5d2
2 changed files with 5 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
lib/ai/runtime/agent-image-failover.ts
View File

@ -8,12 +8,14 @@ export function agentImageFailurePolicy(error: unknown): { retry: boolean; pause
const unknown = value?.providerOutcomeUnknown || /request_outcome_unknown|may still be running|outcome.unknown|timed out|timeout|超时|fetch failed|econnreset|socket hang up/.test(message);
if (unknown && !/task failed|task.*(?:failed|cancelled|canceled)|任务.*(?:失败|取消)/.test(message)) return { retry: false, pause: null };
if (/content.?policy|moderation|safety|违规|敏感|积分不足|本月.*配额|user cancel|user abort/.test(message)) return { retry: false, pause: null };
const status = Number(value?.statusCode || value?.status) || Number(message.match(/(?:http[\s:：]*|status(?:code)?[\s:=]*)([45]\d{2})/i)?.[1]) || null;
const status = Number(value?.statusCode || value?.status) || Number(message.match(/(?:http[\s:：]*|status(?:code)?[\s:=]*|api request failed[\s:：]*)([45]\d{2})/i)?.[1]) || null;
if (/unauthorized|invalid[ _]?(?:api[ _]?)?key|authentication/.test(message)) return { retry: true, pause: { reason: "authentication", status } };
if (/forbidden|access denied/.test(message)) return { retry: true, pause: { reason: "access_denied", status } };
if (/quota|insufficient.*(?:balance|credit)|credits? insufficient|余额|额度|配额|rate.?limit|too many requests/.test(message))
return { retry: true, pause: { reason: "rate_or_quota", status } };
const fault = agentModelFault(status ? { message, status } : error);
if (fault) return { retry: true, pause: fault };
if (/task failed|image generation failed|生成失败|任务.*失败|未返回.*图片|no images? (?:in|url)|no valid data|response body is null|empty.result|empty.response|provider.error|上游.*错误|模型.*维护/.test(message))
if (/task failed|image generation failed|生成失败|任务.*失败|未返回.*图片|no images? (?:in|url)|no valid data|response body is null|empty.result|empty.response|provider.error|上游.*错误|模型.*维护|internal.server.error|service.unavailable|bad.gateway|server.busy|系统繁忙/.test(message))
return { retry: true, pause: { reason: "image_generation_failed", status: Number(value?.statusCode || value?.status) || null } };
return { retry: false, pause: null };
}

2
tests/routes/canvas-agent-runtime-image-failover.test.ts
View File

@ -71,7 +71,7 @@ test("the existing non-Agent auto alias continues to select its configured defau
test("provider adapters that include HTTP status or quota only in their message still recover", () => {
for (const message of ["API request failed: HTTP 503", "server busy（HTTP 429）", "insufficient balance", "模型配额不足", "No images in response", "BananaImage 生成失败"]) {
for (const message of ["API request failed: 503", "Internal server error", "Service Unavailable", "Unauthorized", "API request failed: HTTP 503", "server busy（HTTP 429）", "insufficient balance", "模型配额不足", "No images in response", "BananaImage 生成失败"]) {
expect(agentImageFailurePolicy(new Error(message)).retry).toBe(true);
}
expect(agentImageFailurePolicy(new Error("invalid input (HTTP 400)")).retry).toBe(false);