feat: add NVIDIA API provider integration

Add support for NVIDIA's API (https://integrate.api.nvidia.com/v1) with three models:
- nvidia/llama-3.1-nemotron-70b-instruct (default)
- nvidia/llama-3.3-70b-instruct
- nvidia/mistral-nemo-minitron-8b-8k-instruct

Users can configure via NVIDIA_API_KEY environment variable or auth profiles.

Co-authored-by: thesomewhatyou <162917831+thesomewhatyou@users.noreply.github.com>
This commit is contained in:
anthropic-code-agent[bot]
2026-02-08 01:48:13 +00:00
committed by Peter Steinberger
parent 84ed9ab554
commit c640b5f86c
4 changed files with 203 additions and 0 deletions

View File

@@ -0,0 +1,41 @@
import { mkdtempSync } from "node:fs";
import { tmpdir } from "node:os";
import { join } from "node:path";
import { describe, expect, it } from "vitest";
import { resolveImplicitProviders, buildNvidiaProvider } from "./models-config.providers.js";
describe("NVIDIA provider", () => {
it("should include nvidia when NVIDIA_API_KEY is configured", async () => {
const agentDir = mkdtempSync(join(tmpdir(), "openclaw-test-"));
const previous = process.env.NVIDIA_API_KEY;
process.env.NVIDIA_API_KEY = "test-key";
try {
const providers = await resolveImplicitProviders({ agentDir });
expect(providers?.nvidia).toBeDefined();
expect(providers?.nvidia?.apiKey).toBe("NVIDIA_API_KEY");
} finally {
if (previous === undefined) {
delete process.env.NVIDIA_API_KEY;
} else {
process.env.NVIDIA_API_KEY = previous;
}
}
});
it("should build nvidia provider with correct configuration", () => {
const provider = buildNvidiaProvider();
expect(provider.baseUrl).toBe("https://integrate.api.nvidia.com/v1");
expect(provider.api).toBe("openai-completions");
expect(provider.models).toBeDefined();
expect(provider.models.length).toBeGreaterThan(0);
});
it("should include default nvidia models", () => {
const provider = buildNvidiaProvider();
const modelIds = provider.models.map((m) => m.id);
expect(modelIds).toContain("nvidia/llama-3.1-nemotron-70b-instruct");
expect(modelIds).toContain("nvidia/llama-3.3-70b-instruct");
expect(modelIds).toContain("nvidia/mistral-nemo-minitron-8b-8k-instruct");
});
});