diff --git a/src/commands/models/list.probe.test.ts b/src/commands/models/list.probe.test.ts index a57df70c4ad..932908572dc 100644 --- a/src/commands/models/list.probe.test.ts +++ b/src/commands/models/list.probe.test.ts @@ -27,12 +27,13 @@ describe("mapFailoverReasonToProbeStatus", () => { expect(mapFailoverReasonToProbeStatus("overloaded")).toBe("rate_limit"); expect(mapFailoverReasonToProbeStatus("billing")).toBe("billing"); expect(mapFailoverReasonToProbeStatus("timeout")).toBe("timeout"); + expect(mapFailoverReasonToProbeStatus("model_not_found")).toBe("format"); expect(mapFailoverReasonToProbeStatus("format")).toBe("format"); }); it("falls back to unknown for unrecognized values", () => { expect(mapFailoverReasonToProbeStatus(undefined)).toBe("unknown"); expect(mapFailoverReasonToProbeStatus(null)).toBe("unknown"); - expect(mapFailoverReasonToProbeStatus("model_not_found")).toBe("unknown"); + expect(mapFailoverReasonToProbeStatus("something_else")).toBe("unknown"); }); }); diff --git a/src/commands/models/list.probe.ts b/src/commands/models/list.probe.ts index 18df0a0a673..9020e0e4556 100644 --- a/src/commands/models/list.probe.ts +++ b/src/commands/models/list.probe.ts @@ -121,6 +121,9 @@ export function mapFailoverReasonToProbeStatus(reason?: string | null): AuthProb if (reason === "timeout") { return "timeout"; } + if (reason === "model_not_found") { + return "format"; + } if (reason === "format") { return "format"; }