From a355e485156c793d97fb39add64ed31ffa18af47 Mon Sep 17 00:00:00 2001 From: xiwuqi Date: Wed, 18 Mar 2026 16:34:13 -0500 Subject: [PATCH] fix: surface invalid-model probe failures --- src/commands/models/list.probe.test.ts | 3 ++- src/commands/models/list.probe.ts | 3 +++ 2 files changed, 5 insertions(+), 1 deletion(-) 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"; }