chore: import upstream snapshot with attribution
This commit is contained in:
@@ -0,0 +1,55 @@
|
||||
import { test } from "node:test";
|
||||
import assert from "node:assert/strict";
|
||||
import {
|
||||
buildContinueModels,
|
||||
mergeContinueConfig,
|
||||
resolveContinueTarget,
|
||||
} from "../../../bin/cli/commands/setup-continue.mjs";
|
||||
|
||||
test("buildContinueModels emits provider:openai + apiBase + secret ref + roles", () => {
|
||||
const models = buildContinueModels(["glm/glm-5.2"], "http://vps:20128/v1");
|
||||
assert.equal(models.length, 1);
|
||||
const m = models[0];
|
||||
assert.equal(m.provider, "openai");
|
||||
assert.equal(m.model, "glm/glm-5.2");
|
||||
assert.equal(m.apiBase, "http://vps:20128/v1");
|
||||
assert.equal(m.apiKey, "${{ secrets.OMNIROUTE_API_KEY }}");
|
||||
assert.ok(m.roles.includes("chat") && m.roles.includes("edit") && m.roles.includes("apply"));
|
||||
});
|
||||
|
||||
test("buildContinueModels gives the fast tier an autocomplete role", () => {
|
||||
const fast = buildContinueModels(["glm/glm-5-turbo"], "http://x/v1")[0]; // fast → effort low
|
||||
assert.ok(fast.roles.includes("autocomplete"));
|
||||
});
|
||||
|
||||
test("buildContinueModels skips uncategorised models", () => {
|
||||
assert.equal(buildContinueModels(["some/unknown-model"], "http://x/v1").length, 0);
|
||||
});
|
||||
|
||||
test("mergeContinueConfig replaces prior OmniRoute models, keeps others", () => {
|
||||
const existing = {
|
||||
name: "My Config",
|
||||
models: [
|
||||
{ name: "Local Ollama", provider: "ollama", model: "llama3", apiBase: "http://localhost:11434" },
|
||||
{ name: "old omni", provider: "openai", model: "x", apiBase: "http://vps:20128/v1" },
|
||||
],
|
||||
};
|
||||
const fresh = buildContinueModels(["glm/glm-5.2"], "http://vps:20128/v1");
|
||||
const merged = mergeContinueConfig(existing, fresh, "http://vps:20128/v1");
|
||||
// kept the ollama model; dropped the old omni one (same apiBase); added the new
|
||||
const apiBases = merged.models.map((m) => m.apiBase);
|
||||
assert.ok(merged.models.some((m) => m.provider === "ollama"));
|
||||
assert.equal(merged.models.filter((m) => m.apiBase === "http://vps:20128/v1").length, 1);
|
||||
assert.equal(merged.name, "My Config", "preserves existing top-level keys");
|
||||
});
|
||||
|
||||
test("mergeContinueConfig sets defaults on an empty config", () => {
|
||||
const merged = mergeContinueConfig({}, buildContinueModels(["glm/glm-5.2"], "http://x/v1"), "http://x/v1");
|
||||
assert.equal(merged.schema, "v1");
|
||||
assert.ok(merged.name && merged.version);
|
||||
});
|
||||
|
||||
test("resolveContinueTarget ensures /v1 on apiBase", () => {
|
||||
assert.equal(resolveContinueTarget({ remote: "http://vps:20128" }).apiBase, "http://vps:20128/v1");
|
||||
assert.equal(resolveContinueTarget({ remote: "http://vps:20128/v1/" }).apiBase, "http://vps:20128/v1");
|
||||
});
|
||||
Reference in New Issue
Block a user