chore: import upstream snapshot with attribution
This commit is contained in:
@@ -0,0 +1,65 @@
|
||||
import test from "node:test";
|
||||
import assert from "node:assert/strict";
|
||||
|
||||
import {
|
||||
buildAliasEnvVar,
|
||||
buildCustomCliEnvScript,
|
||||
buildCustomCliJsonConfig,
|
||||
normalizeOpenAiBaseUrl,
|
||||
} from "../../src/app/(dashboard)/dashboard/cli-code/components/customCliConfig.ts";
|
||||
|
||||
test("normalizeOpenAiBaseUrl appends /v1 only when needed", () => {
|
||||
assert.equal(normalizeOpenAiBaseUrl("http://localhost:20128"), "http://localhost:20128/v1");
|
||||
assert.equal(
|
||||
normalizeOpenAiBaseUrl("https://example.com/proxy/v1"),
|
||||
"https://example.com/proxy/v1"
|
||||
);
|
||||
});
|
||||
|
||||
test("buildAliasEnvVar sanitizes aliases for env variable export", () => {
|
||||
assert.equal(buildAliasEnvVar("review"), "OMNIROUTE_MODEL_REVIEW");
|
||||
assert.equal(buildAliasEnvVar("plan mode"), "OMNIROUTE_MODEL_PLAN_MODE");
|
||||
assert.equal(buildAliasEnvVar(""), null);
|
||||
});
|
||||
|
||||
test("custom CLI generators include default model and alias mappings", () => {
|
||||
const envScript = buildCustomCliEnvScript({
|
||||
cliName: "My Team CLI",
|
||||
baseUrl: "http://localhost:20128",
|
||||
apiKey: "sk_test_123",
|
||||
defaultModel: "omniroute/fast",
|
||||
aliasMappings: [
|
||||
{ alias: "review", model: "cc/claude-sonnet-4-5-20250929" },
|
||||
{ alias: "vision", model: "gemini/gemini-3-flash" },
|
||||
],
|
||||
});
|
||||
|
||||
assert.match(envScript, /export OPENAI_BASE_URL="http:\/\/localhost:20128\/v1"/);
|
||||
assert.match(envScript, /export OPENAI_MODEL="omniroute\/fast"/);
|
||||
assert.match(envScript, /export OMNIROUTE_MODEL_REVIEW="cc\/claude-sonnet-4-5-20250929"/);
|
||||
assert.match(envScript, /# http:\/\/localhost:20128\/v1\/chat\/completions/);
|
||||
assert.match(envScript, /my-team-cli --base-url "\$OPENAI_BASE_URL"/);
|
||||
|
||||
const jsonConfig = JSON.parse(
|
||||
buildCustomCliJsonConfig({
|
||||
cliName: "My Team CLI",
|
||||
baseUrl: "http://localhost:20128",
|
||||
apiKey: "sk_test_123",
|
||||
defaultModel: "omniroute/fast",
|
||||
aliasMappings: [{ alias: "review", model: "cc/claude-sonnet-4-5-20250929" }],
|
||||
})
|
||||
);
|
||||
|
||||
assert.deepEqual(jsonConfig, {
|
||||
name: "My Team CLI",
|
||||
provider: {
|
||||
type: "openai",
|
||||
baseURL: "http://localhost:20128/v1",
|
||||
apiKey: "sk_test_123",
|
||||
model: "omniroute/fast",
|
||||
},
|
||||
modelAliases: {
|
||||
review: "cc/claude-sonnet-4-5-20250929",
|
||||
},
|
||||
});
|
||||
});
|
||||
Reference in New Issue
Block a user