Files
openclaw/src/agents/models-config.providers.normalize-keys.test.ts
Peter Steinberger 6bea38b21f fix(models): land #31202 normalize custom provider keys (@stakeswky)
Landed from contributor PR #31202 by @stakeswky.

Co-authored-by: stakeswky <stakeswky@users.noreply.github.com>
2026-03-02 03:11:55 +00:00

77 lines
2.8 KiB
TypeScript

import fs from "node:fs/promises";
import os from "node:os";
import path from "node:path";
import { describe, expect, it } from "vitest";
import type { OpenClawConfig } from "../config/config.js";
import { normalizeProviders } from "./models-config.providers.js";
describe("normalizeProviders", () => {
it("trims provider keys so image models remain discoverable for custom providers", async () => {
const agentDir = await fs.mkdtemp(path.join(os.tmpdir(), "openclaw-agent-"));
try {
const providers: NonNullable<NonNullable<OpenClawConfig["models"]>["providers"]> = {
" dashscope-vision ": {
baseUrl: "https://dashscope.aliyuncs.com/compatible-mode/v1",
api: "openai-completions",
apiKey: "DASHSCOPE_API_KEY",
models: [
{
id: "qwen-vl-max",
name: "Qwen VL Max",
input: ["text", "image"],
reasoning: false,
cost: { input: 0, output: 0, cacheRead: 0, cacheWrite: 0 },
contextWindow: 32000,
maxTokens: 4096,
},
],
},
};
const normalized = normalizeProviders({ providers, agentDir });
expect(Object.keys(normalized ?? {})).toEqual(["dashscope-vision"]);
expect(normalized?.["dashscope-vision"]?.models?.[0]?.id).toBe("qwen-vl-max");
} finally {
await fs.rm(agentDir, { recursive: true, force: true });
}
});
it("keeps the latest provider config when duplicate keys only differ by whitespace", async () => {
const agentDir = await fs.mkdtemp(path.join(os.tmpdir(), "openclaw-agent-"));
try {
const providers: NonNullable<NonNullable<OpenClawConfig["models"]>["providers"]> = {
openai: {
baseUrl: "https://api.openai.com/v1",
api: "openai-completions",
apiKey: "OPENAI_API_KEY",
models: [],
},
" openai ": {
baseUrl: "https://example.com/v1",
api: "openai-completions",
apiKey: "CUSTOM_OPENAI_API_KEY",
models: [
{
id: "gpt-4.1-mini",
name: "GPT-4.1 mini",
input: ["text"],
reasoning: false,
cost: { input: 0, output: 0, cacheRead: 0, cacheWrite: 0 },
contextWindow: 128000,
maxTokens: 16384,
},
],
},
};
const normalized = normalizeProviders({ providers, agentDir });
expect(Object.keys(normalized ?? {})).toEqual(["openai"]);
expect(normalized?.openai?.baseUrl).toBe("https://example.com/v1");
expect(normalized?.openai?.apiKey).toBe("CUSTOM_OPENAI_API_KEY");
expect(normalized?.openai?.models?.[0]?.id).toBe("gpt-4.1-mini");
} finally {
await fs.rm(agentDir, { recursive: true, force: true });
}
});
});