Files
diegosouzapw--omniroute/tests/unit/nvidia-validation-model-3116.test.ts
2026-07-13 13:39:12 +08:00

30 lines
1.3 KiB
TypeScript

/**
* #3116 — NVIDIA key validation probed the first catalog model (`z-ai/glm-5.1`), which
* requires the "Public API Endpoints" account permission and can hang/be DEGRADED,
* making a *valid* key fail with a misleading "Upstream Error". The probe now defaults to
* the universally-available `meta/llama-3.1-8b-instruct`, with a per-connection override.
*/
import test from "node:test";
import assert from "node:assert/strict";
import {
NVIDIA_DEFAULT_VALIDATION_MODEL,
resolveNvidiaValidationModel,
} from "../../src/lib/providers/nvidiaValidationModel.ts";
test("defaults to a stable, permission-free NVIDIA model (not the gated glm-5.1)", () => {
assert.equal(NVIDIA_DEFAULT_VALIDATION_MODEL, "meta/llama-3.1-8b-instruct");
assert.equal(resolveNvidiaValidationModel(), "meta/llama-3.1-8b-instruct");
assert.equal(resolveNvidiaValidationModel({}), "meta/llama-3.1-8b-instruct");
assert.notEqual(resolveNvidiaValidationModel(undefined), "z-ai/glm-5.1");
});
test("honors a per-connection validationModelId override", () => {
assert.equal(
resolveNvidiaValidationModel({ validationModelId: "nvidia/llama-3.3-nemotron-super-49b" }),
"nvidia/llama-3.3-nemotron-super-49b"
);
// blank/whitespace override falls back to the default
assert.equal(resolveNvidiaValidationModel({ validationModelId: " " }), NVIDIA_DEFAULT_VALIDATION_MODEL);
});