mirror of
https://github.com/diegosouzapw/OmniRoute.git
synced 2026-07-26 09:52:11 +03:00
4ae488b25b
Introduce a runtime settings layer that hydrates persisted config at startup and reapplies aliases, payload rules, cache behavior, CLI compatibility, usage tuning, and related switches when settings change or SQLite updates. Replace the legacy prompt injection middleware path with a guardrail registry that supports prompt injection detection, PII masking, disabled guardrail overrides, and post-call response handling across the chat pipeline. Add a metadata registry for model catalog and alias resolution so catalog endpoints return enriched capabilities plus diagnostic headers and typed alias errors instead of ad hoc responses. Convert unsupported built-in web_search tools into an OmniRoute fallback tool, execute them through builtin skills, and preserve Responses API function call output with sanitized usage fields. Centralize provider header fingerprints for GitHub, Cursor, Qwen, Qoder, Kiro, and Antigravity, and migrate management passwords from env or plaintext storage into persisted bcrypt hashes during startup and login.
90 lines
3.1 KiB
TypeScript
90 lines
3.1 KiB
TypeScript
import test from "node:test";
|
|
import assert from "node:assert/strict";
|
|
import fs from "node:fs";
|
|
import os from "node:os";
|
|
import path from "node:path";
|
|
|
|
const TEST_DATA_DIR = fs.mkdtempSync(path.join(os.tmpdir(), "omniroute-model-metadata-"));
|
|
process.env.DATA_DIR = TEST_DATA_DIR;
|
|
|
|
const core = await import("../../src/lib/db/core.ts");
|
|
const modelsDb = await import("../../src/lib/db/models.ts");
|
|
const modelsDevSync = await import("../../src/lib/modelsDevSync.ts");
|
|
const registry = await import("../../src/lib/modelMetadataRegistry.ts");
|
|
|
|
async function resetStorage() {
|
|
core.resetDbInstance();
|
|
fs.rmSync(TEST_DATA_DIR, { recursive: true, force: true });
|
|
fs.mkdirSync(TEST_DATA_DIR, { recursive: true });
|
|
}
|
|
|
|
test.beforeEach(async () => {
|
|
await resetStorage();
|
|
});
|
|
|
|
test.after(async () => {
|
|
await resetStorage();
|
|
fs.rmSync(TEST_DATA_DIR, { recursive: true, force: true });
|
|
});
|
|
|
|
test("canonical model metadata merges static and synced capabilities into one record", async () => {
|
|
modelsDevSync.saveModelsDevCapabilities({
|
|
openai: {
|
|
"gpt-4o": {
|
|
tool_call: true,
|
|
reasoning: true,
|
|
attachment: true,
|
|
structured_output: true,
|
|
temperature: true,
|
|
modalities_input: JSON.stringify(["text", "image"]),
|
|
modalities_output: JSON.stringify(["text"]),
|
|
knowledge_cutoff: "2025-03",
|
|
release_date: "2025-01-01",
|
|
last_updated: "2025-04-01",
|
|
status: "stable",
|
|
family: "gpt-4",
|
|
open_weights: false,
|
|
limit_context: 256000,
|
|
limit_input: 256000,
|
|
limit_output: 16384,
|
|
interleaved_field: null,
|
|
},
|
|
},
|
|
});
|
|
|
|
const metadata = registry.getCanonicalModelMetadata({ provider: "openai", model: "gpt-4o" });
|
|
|
|
assert.ok(metadata);
|
|
assert.equal(metadata.provider, "openai");
|
|
assert.equal(metadata.providerAlias, "openai");
|
|
assert.equal(metadata.capabilities.toolCalling, true);
|
|
assert.equal(metadata.capabilities.vision, true);
|
|
assert.equal(metadata.limits.contextWindow, 256000);
|
|
assert.equal(metadata.limits.maxOutputTokens, 16384);
|
|
assert.deepEqual(metadata.modalities.input, ["text", "image"]);
|
|
assert.equal(metadata.metadata.family, "gpt-4");
|
|
assert.equal(metadata.metadata.source.syncedCapability, true);
|
|
});
|
|
|
|
test("resolveModelAliasLookup returns stored alias resolution with metadata", async () => {
|
|
await modelsDb.setModelAlias("fast-default", "openai/gpt-4o");
|
|
|
|
const resolved = await registry.resolveModelAliasLookup("fast-default");
|
|
|
|
assert.equal(resolved.ok, true);
|
|
if (!resolved.ok) return;
|
|
assert.equal(resolved.value.source, "stored_alias");
|
|
assert.equal(resolved.value.resolvedAlias, "openai/gpt-4o");
|
|
assert.equal(resolved.value.metadata.model, "gpt-4o");
|
|
});
|
|
|
|
test("resolveModelAliasLookup flags ambiguous raw model IDs instead of guessing", async () => {
|
|
const resolved = await registry.resolveModelAliasLookup("claude-sonnet-4-6");
|
|
|
|
assert.equal(resolved.ok, false);
|
|
if (resolved.ok) return;
|
|
assert.equal(resolved.error.code, registry.MODEL_ALIAS_AMBIGUOUS);
|
|
assert.equal(resolved.error.status, 409);
|
|
assert.ok((resolved.error.candidates || []).some((candidate) => candidate.startsWith("cc/")));
|
|
});
|