54 lines
2.1 KiB
TypeScript
54 lines
2.1 KiB
TypeScript
import { describe, it, expect, beforeEach, afterEach } from "vitest";
|
|
import { listK8sModels } from "./models.js";
|
|
|
|
describe("listK8sModels", () => {
|
|
const savedEnv: Record<string, string | undefined> = {};
|
|
|
|
beforeEach(() => {
|
|
savedEnv.CLAUDE_CODE_USE_BEDROCK = process.env.CLAUDE_CODE_USE_BEDROCK;
|
|
savedEnv.ANTHROPIC_BEDROCK_BASE_URL = process.env.ANTHROPIC_BEDROCK_BASE_URL;
|
|
delete process.env.CLAUDE_CODE_USE_BEDROCK;
|
|
delete process.env.ANTHROPIC_BEDROCK_BASE_URL;
|
|
});
|
|
|
|
afterEach(() => {
|
|
for (const [key, value] of Object.entries(savedEnv)) {
|
|
if (value === undefined) delete process.env[key];
|
|
else process.env[key] = value;
|
|
}
|
|
});
|
|
|
|
it("returns direct API models by default", async () => {
|
|
const models = await listK8sModels();
|
|
expect(models.some((m) => m.id === "claude-opus-4-7")).toBe(true);
|
|
expect(models.some((m) => m.id === "claude-opus-4-6")).toBe(true);
|
|
expect(models.every((m) => !m.id.includes("anthropic."))).toBe(true);
|
|
});
|
|
|
|
it("returns Bedrock models when CLAUDE_CODE_USE_BEDROCK=1", async () => {
|
|
process.env.CLAUDE_CODE_USE_BEDROCK = "1";
|
|
const models = await listK8sModels();
|
|
expect(models.some((m) => m.id.includes("anthropic."))).toBe(true);
|
|
expect(models.every((m) => !m.id.startsWith("claude-"))).toBe(true);
|
|
});
|
|
|
|
it("returns Bedrock models when CLAUDE_CODE_USE_BEDROCK=true", async () => {
|
|
process.env.CLAUDE_CODE_USE_BEDROCK = "true";
|
|
const models = await listK8sModels();
|
|
expect(models.some((m) => m.id.includes("anthropic."))).toBe(true);
|
|
});
|
|
|
|
it("returns Bedrock models when ANTHROPIC_BEDROCK_BASE_URL is set", async () => {
|
|
process.env.ANTHROPIC_BEDROCK_BASE_URL = "https://bedrock.us-east-1.amazonaws.com";
|
|
const models = await listK8sModels();
|
|
expect(models.some((m) => m.id.includes("anthropic."))).toBe(true);
|
|
});
|
|
|
|
it("ignores blank ANTHROPIC_BEDROCK_BASE_URL", async () => {
|
|
process.env.ANTHROPIC_BEDROCK_BASE_URL = " ";
|
|
const models = await listK8sModels();
|
|
expect(models.some((m) => m.id === "claude-opus-4-7")).toBe(true);
|
|
});
|
|
});
|
|
|