diff --git a/src/lib/validation.test.ts b/src/lib/validation.test.ts index 0006aa60a08..d9b766b7692 100644 --- a/src/lib/validation.test.ts +++ b/src/lib/validation.test.ts @@ -280,6 +280,10 @@ describe("shouldSkipResponsesProbe", () => { expect(shouldSkipResponsesProbe("nvidia-prod")).toBe(true); }); + it("skips the Responses probe for nvidia-nim (same endpoint as nvidia-prod, no /v1/responses)", () => { + expect(shouldSkipResponsesProbe("nvidia-nim")).toBe(true); + }); + it("skips the Responses probe for gemini-api (Gemini does not support /v1/responses)", () => { expect(shouldSkipResponsesProbe("gemini-api")).toBe(true); }); diff --git a/src/lib/validation.ts b/src/lib/validation.ts index 9b142c9ea66..7125abc77cf 100644 --- a/src/lib/validation.ts +++ b/src/lib/validation.ts @@ -143,7 +143,7 @@ export function nvcfFunctionNotFoundMessage(model: string): string { * See issue #1601 (Bug 1) and issue #1960. */ export function shouldSkipResponsesProbe(provider: string): boolean { - return provider === "nvidia-prod" || provider === "gemini-api"; + return provider === "nvidia-prod" || provider === "nvidia-nim" || provider === "gemini-api"; } /**