mirror of
https://github.com/openclaw/openclaw.git
synced 2026-05-08 03:01:25 +00:00
122 lines
4.2 KiB
TypeScript
122 lines
4.2 KiB
TypeScript
import fs from "node:fs/promises";
|
|
import path from "node:path";
|
|
import { describe, expect, it } from "vitest";
|
|
import { resolveOpenClawAgentDir } from "./agent-paths.js";
|
|
import {
|
|
CUSTOM_PROXY_MODELS_CONFIG,
|
|
installModelsConfigTestHooks,
|
|
MODELS_CONFIG_IMPLICIT_ENV_VARS,
|
|
unsetEnv,
|
|
withTempEnv,
|
|
withModelsTempHome as withTempHome,
|
|
} from "./models-config.e2e-harness.js";
|
|
import { ensureOpenClawModelsJson } from "./models-config.js";
|
|
|
|
installModelsConfigTestHooks();
|
|
|
|
describe("models-config", () => {
|
|
it("skips writing models.json when no env token or profile exists", async () => {
|
|
await withTempHome(async (home) => {
|
|
await withTempEnv([...MODELS_CONFIG_IMPLICIT_ENV_VARS, "KIMI_API_KEY"], async () => {
|
|
unsetEnv([...MODELS_CONFIG_IMPLICIT_ENV_VARS, "KIMI_API_KEY"]);
|
|
|
|
const agentDir = path.join(home, "agent-empty");
|
|
// ensureAuthProfileStore merges the main auth store into non-main dirs; point main at our temp dir.
|
|
process.env.OPENCLAW_AGENT_DIR = agentDir;
|
|
process.env.PI_CODING_AGENT_DIR = agentDir;
|
|
|
|
const result = await ensureOpenClawModelsJson(
|
|
{
|
|
models: { providers: {} },
|
|
},
|
|
agentDir,
|
|
);
|
|
|
|
await expect(fs.stat(path.join(agentDir, "models.json"))).rejects.toThrow();
|
|
expect(result.wrote).toBe(false);
|
|
});
|
|
});
|
|
});
|
|
|
|
it("writes models.json for configured providers", async () => {
|
|
await withTempHome(async () => {
|
|
await ensureOpenClawModelsJson(CUSTOM_PROXY_MODELS_CONFIG);
|
|
|
|
const modelPath = path.join(resolveOpenClawAgentDir(), "models.json");
|
|
const raw = await fs.readFile(modelPath, "utf8");
|
|
const parsed = JSON.parse(raw) as {
|
|
providers: Record<string, { baseUrl?: string }>;
|
|
};
|
|
|
|
expect(parsed.providers["custom-proxy"]?.baseUrl).toBe("http://localhost:4000/v1");
|
|
});
|
|
});
|
|
|
|
it("adds minimax provider when MINIMAX_API_KEY is set", async () => {
|
|
await withTempHome(async () => {
|
|
const prevKey = process.env.MINIMAX_API_KEY;
|
|
process.env.MINIMAX_API_KEY = "sk-minimax-test";
|
|
try {
|
|
await ensureOpenClawModelsJson({});
|
|
|
|
const modelPath = path.join(resolveOpenClawAgentDir(), "models.json");
|
|
const raw = await fs.readFile(modelPath, "utf8");
|
|
const parsed = JSON.parse(raw) as {
|
|
providers: Record<
|
|
string,
|
|
{
|
|
baseUrl?: string;
|
|
apiKey?: string;
|
|
models?: Array<{ id: string }>;
|
|
}
|
|
>;
|
|
};
|
|
expect(parsed.providers.minimax?.baseUrl).toBe("https://api.minimax.io/anthropic");
|
|
expect(parsed.providers.minimax?.apiKey).toBe("MINIMAX_API_KEY");
|
|
const ids = parsed.providers.minimax?.models?.map((model) => model.id);
|
|
expect(ids).toContain("MiniMax-M2.1");
|
|
expect(ids).toContain("MiniMax-VL-01");
|
|
} finally {
|
|
if (prevKey === undefined) {
|
|
delete process.env.MINIMAX_API_KEY;
|
|
} else {
|
|
process.env.MINIMAX_API_KEY = prevKey;
|
|
}
|
|
}
|
|
});
|
|
});
|
|
|
|
it("adds synthetic provider when SYNTHETIC_API_KEY is set", async () => {
|
|
await withTempHome(async () => {
|
|
const prevKey = process.env.SYNTHETIC_API_KEY;
|
|
process.env.SYNTHETIC_API_KEY = "sk-synthetic-test";
|
|
try {
|
|
await ensureOpenClawModelsJson({});
|
|
|
|
const modelPath = path.join(resolveOpenClawAgentDir(), "models.json");
|
|
const raw = await fs.readFile(modelPath, "utf8");
|
|
const parsed = JSON.parse(raw) as {
|
|
providers: Record<
|
|
string,
|
|
{
|
|
baseUrl?: string;
|
|
apiKey?: string;
|
|
models?: Array<{ id: string }>;
|
|
}
|
|
>;
|
|
};
|
|
expect(parsed.providers.synthetic?.baseUrl).toBe("https://api.synthetic.new/anthropic");
|
|
expect(parsed.providers.synthetic?.apiKey).toBe("SYNTHETIC_API_KEY");
|
|
const ids = parsed.providers.synthetic?.models?.map((model) => model.id);
|
|
expect(ids).toContain("hf:MiniMaxAI/MiniMax-M2.1");
|
|
} finally {
|
|
if (prevKey === undefined) {
|
|
delete process.env.SYNTHETIC_API_KEY;
|
|
} else {
|
|
process.env.SYNTHETIC_API_KEY = prevKey;
|
|
}
|
|
}
|
|
});
|
|
});
|
|
});
|