@@ -8,7 +8,7 @@ import { describe, expect, it } from "vitest";
88import plugin from "./index.js";
991010type NvidiaManifest = {
11-providerAuthChoices?: Array<{ choiceId?: string; method?: string; provider?: string }>;
11+providerAuthChoices?: Array<Record<string, unknown>>;
1212};
1313type RegisteredModelCatalogProvider = Parameters<
1414ReturnType<typeof createTestPluginApi>["registerModelCatalogProvider"]
@@ -45,15 +45,21 @@ describe("nvidia provider hooks", () => {
4545});
4646expect(choice?.provider.id).toBe("nvidia");
4747expect(choice?.method.id).toBe("api-key");
48-expect(readManifest().providerAuthChoices).toEqual(
49-expect.arrayContaining([
50-expect.objectContaining({
51-provider: "nvidia",
52-method: "api-key",
53-choiceId: "nvidia-api-key",
54-}),
55-]),
56-);
48+expect(readManifest().providerAuthChoices).toStrictEqual([
49+{
50+provider: "nvidia",
51+method: "api-key",
52+choiceId: "nvidia-api-key",
53+choiceLabel: "NVIDIA API key",
54+groupId: "nvidia",
55+groupLabel: "NVIDIA",
56+groupHint: "Direct API key",
57+optionKey: "nvidiaApiKey",
58+cliFlag: "--nvidia-api-key",
59+cliOption: "--nvidia-api-key <key>",
60+cliDescription: "NVIDIA API key",
61+},
62+]);
5763});
58645965it("keeps nvidia auth setup metadata aligned", async () => {
@@ -85,20 +91,24 @@ describe("nvidia provider hooks", () => {
8591it("keeps nvidia wizard setup metadata aligned", async () => {
8692const provider = await registerNvidiaProvider();
879388-expect(provider.wizard?.setup).toMatchObject({
94+expect(provider.wizard?.setup).toStrictEqual({
8995choiceId: "nvidia-api-key",
9096choiceLabel: "NVIDIA API key",
9197groupId: "nvidia",
9298groupLabel: "NVIDIA",
9399groupHint: "Direct API key",
94100methodId: "api-key",
101+modelSelection: {
102+promptWhenAuthChoiceProvided: true,
103+allowKeepCurrent: false,
104+},
95105});
96106});
9710798108it("keeps nvidia model picker metadata aligned", async () => {
99109const provider = await registerNvidiaProvider();
100110101-expect(provider.wizard?.modelPicker).toMatchObject({
111+expect(provider.wizard?.modelPicker).toStrictEqual({
102112label: "NVIDIA (custom)",
103113hint: "Use NVIDIA-hosted open models",
104114methodId: "api-key",
@@ -162,9 +172,9 @@ describe("nvidia provider hooks", () => {
162172}),
163173);
164174165-expect(registeredProviders).toContain("nvidia");
166-expect(registeredModelCatalogProviders.map((provider) => provider.provider)).toContain(
175+expect(registeredProviders).toStrictEqual(["nvidia"]);
176+expect(registeredModelCatalogProviders.map((provider) => provider.provider)).toStrictEqual([
167177"nvidia",
168-);
178+]);
169179});
170180});
此内容由惯性聚合(RSS阅读器)自动聚合整理，仅供阅读参考。 原文来自 — 版权归原作者所有。