import { agentModelFault, type AgentModelFault } from "./agent-model-failover";

export function agentImageFailurePolicy(error: unknown): { retry: boolean; pause: AgentModelFault | null } {
  const value = error as { message?: string; providerOutcomeUnknown?: boolean; status?: number; statusCode?: number } | null;
  const message = String(value?.message || "").toLowerCase();
  // A submitted image job may still succeed. Retain its task and reconcile it;
  // never submit another paid job just because a poll/connection timed out.
  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 fault = agentModelFault(error);
  if (fault) return { retry: true, pause: fault };
  if (/task failed|image generation failed|图片生成失败|未返回.*图片|empty.result|empty.response|provider.error|上游.*错误|模型.*维护/.test(message))
    return { retry: true, pause: { reason: "image_generation_failed", status: Number(value?.statusCode || value?.status) || null } };
  return { retry: false, pause: null };
}