mirror of https://github.com/openclaw/openclaw.git
86 lines
3.5 KiB
TypeScript
86 lines
3.5 KiB
TypeScript
import { mkdtempSync } from "node:fs";
|
|
import { tmpdir } from "node:os";
|
|
import { join } from "node:path";
|
|
import { describe, expect, it } from "vitest";
|
|
import { withEnvAsync } from "../test-utils/env.js";
|
|
import { resolveApiKeyForProvider } from "./model-auth.js";
|
|
import { buildNvidiaProvider, resolveImplicitProviders } 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-"));
|
|
await withEnvAsync({ NVIDIA_API_KEY: "test-key" }, async () => {
|
|
const providers = await resolveImplicitProviders({ agentDir });
|
|
expect(providers?.nvidia).toBeDefined();
|
|
expect(providers?.nvidia?.models?.length).toBeGreaterThan(0);
|
|
});
|
|
});
|
|
|
|
it("resolves the nvidia api key value from env", async () => {
|
|
const agentDir = mkdtempSync(join(tmpdir(), "openclaw-test-"));
|
|
await withEnvAsync({ NVIDIA_API_KEY: "nvidia-test-api-key" }, async () => {
|
|
const auth = await resolveApiKeyForProvider({
|
|
provider: "nvidia",
|
|
agentDir,
|
|
});
|
|
|
|
expect(auth.apiKey).toBe("nvidia-test-api-key");
|
|
expect(auth.mode).toBe("api-key");
|
|
expect(auth.source).toContain("NVIDIA_API_KEY");
|
|
});
|
|
});
|
|
|
|
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("meta/llama-3.3-70b-instruct");
|
|
expect(modelIds).toContain("nvidia/mistral-nemo-minitron-8b-8k-instruct");
|
|
});
|
|
});
|
|
|
|
describe("MiniMax implicit provider (#15275)", () => {
|
|
it("should use anthropic-messages API for API-key provider", async () => {
|
|
const agentDir = mkdtempSync(join(tmpdir(), "openclaw-test-"));
|
|
await withEnvAsync({ MINIMAX_API_KEY: "test-key" }, async () => {
|
|
const providers = await resolveImplicitProviders({ agentDir });
|
|
expect(providers?.minimax).toBeDefined();
|
|
expect(providers?.minimax?.api).toBe("anthropic-messages");
|
|
expect(providers?.minimax?.baseUrl).toBe("https://api.minimax.io/anthropic");
|
|
});
|
|
});
|
|
});
|
|
|
|
describe("vLLM provider", () => {
|
|
it("should not include vllm when no API key is configured", async () => {
|
|
const agentDir = mkdtempSync(join(tmpdir(), "openclaw-test-"));
|
|
await withEnvAsync({ VLLM_API_KEY: undefined }, async () => {
|
|
const providers = await resolveImplicitProviders({ agentDir });
|
|
expect(providers?.vllm).toBeUndefined();
|
|
});
|
|
});
|
|
|
|
it("should include vllm when VLLM_API_KEY is set", async () => {
|
|
const agentDir = mkdtempSync(join(tmpdir(), "openclaw-test-"));
|
|
await withEnvAsync({ VLLM_API_KEY: "test-key" }, async () => {
|
|
const providers = await resolveImplicitProviders({ agentDir });
|
|
|
|
expect(providers?.vllm).toBeDefined();
|
|
expect(providers?.vllm?.apiKey).toBe("VLLM_API_KEY");
|
|
expect(providers?.vllm?.baseUrl).toBe("http://127.0.0.1:8000/v1");
|
|
expect(providers?.vllm?.api).toBe("openai-completions");
|
|
|
|
// Note: discovery is disabled in test environments (VITEST check)
|
|
expect(providers?.vllm?.models).toEqual([]);
|
|
});
|
|
});
|
|
});
|