mirror of
https://github.com/diegosouzapw/OmniRoute.git
synced 2026-09-11 09:22:48 +03:00
Compare commits
1 Commits
fix/12172-
...
fix/12783-
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
a37c39ef73 |
@@ -35,16 +35,24 @@ export function resolveOpencodeTarget(opts = {}) {
|
||||
baseUrl = `http://localhost:${Number(opts.port ?? process.env.PORT ?? 20128) || 20128}`;
|
||||
}
|
||||
|
||||
// Precedence: explicit --api-key flag > OMNIROUTE_API_KEY env var > active
|
||||
// context's management token. A context's accessToken/apiKey is a CLI
|
||||
// management credential (oma_live_...) with no /v1/* inference scope — it
|
||||
// must never silently outrank a real inference key the caller supplied
|
||||
// either as a flag or via the ambient env var (mirrors the explicit >
|
||||
// ambient-env > context precedence documented in bin/cli/api.mjs's
|
||||
// buildHeaders()). Only fall back to the context token when neither an
|
||||
// explicit flag nor the env var is set.
|
||||
let apiKey = opts.apiKey ?? opts["api-key"];
|
||||
if (!apiKey) apiKey = process.env.OMNIROUTE_API_KEY || "";
|
||||
if (!apiKey) {
|
||||
try {
|
||||
const c = resolveActiveContext(opts.context ?? process.env.OMNIROUTE_CONTEXT);
|
||||
apiKey = c?.accessToken || c?.apiKey;
|
||||
apiKey = c?.accessToken || c?.apiKey || "";
|
||||
} catch {
|
||||
/* no context auth */
|
||||
}
|
||||
}
|
||||
if (!apiKey) apiKey = process.env.OMNIROUTE_API_KEY || "";
|
||||
return { baseUrl: baseUrl.replace(/\/+$/, ""), apiKey };
|
||||
}
|
||||
|
||||
@@ -177,8 +185,17 @@ export function registerSetupOpencode(program) {
|
||||
"--allow-container-write",
|
||||
"Write even when the target is inside a container and not mounted from the host"
|
||||
)
|
||||
.action(async (opts) => {
|
||||
const code = await runSetupOpencodeCommand(opts);
|
||||
.action(async (opts, cmd) => {
|
||||
// Commander parses the ancestor program's own global --api-key option
|
||||
// (bin/cli/program.mjs, bound to .env("OMNIROUTE_API_KEY")) against any
|
||||
// occurrence of the flag in argv, so it wins the value even when the
|
||||
// user typed --api-key AFTER `setup-opencode` — this local option's own
|
||||
// `opts.apiKey` never sees it. cmd.optsWithGlobals() resolves to the
|
||||
// correct value either way ("globals overwrite locals" is exactly the
|
||||
// outcome we want here, since the global option is where the value
|
||||
// always actually lands).
|
||||
const resolvedOpts = { ...opts, apiKey: cmd.optsWithGlobals().apiKey ?? opts.apiKey };
|
||||
const code = await runSetupOpencodeCommand(resolvedOpts);
|
||||
if (code !== 0) process.exit(code);
|
||||
});
|
||||
}
|
||||
|
||||
@@ -1 +0,0 @@
|
||||
- fix(db): scope model visibility overrides by modality so hiding a Chat model no longer hides an identically-ID'd Image/Embeddings/etc. model (#12172)
|
||||
@@ -0,0 +1 @@
|
||||
- fix(cli): setup-opencode no longer sends an active context's management token to `/v1/models` when `--api-key`/`OMNIROUTE_API_KEY` is supplied — an explicit flag or the env var now always outranks the context's token, and the flag itself is no longer swallowed by the parent program's global `--api-key` option (#12783)
|
||||
@@ -212,10 +212,7 @@ export function useModelVisibilityHandlers({
|
||||
{
|
||||
method: "PATCH",
|
||||
headers: { "Content-Type": "application/json" },
|
||||
// #12172: this page manages only Chat models — scope the hide/unhide to
|
||||
// "chat" so it never suppresses an identically-ID'd model registered
|
||||
// under a different modality's registry (e.g. Image).
|
||||
body: JSON.stringify({ isHidden: hidden, modality: "chat" }),
|
||||
body: JSON.stringify({ isHidden: hidden }),
|
||||
}
|
||||
);
|
||||
if (!res.ok) {
|
||||
@@ -242,8 +239,7 @@ export function useModelVisibilityHandlers({
|
||||
const res = await fetch(`/api/provider-models?provider=${encodeURIComponent(providerKey)}`, {
|
||||
method: "PATCH",
|
||||
headers: { "Content-Type": "application/json" },
|
||||
// #12172: same "chat"-only scoping as the single-model toggle above.
|
||||
body: JSON.stringify({ isHidden: hidden, modelIds, modality: "chat" }),
|
||||
body: JSON.stringify({ isHidden: hidden, modelIds }),
|
||||
});
|
||||
if (!res.ok) {
|
||||
const detail = await res.text().catch(() => "");
|
||||
|
||||
@@ -412,17 +412,6 @@ export async function PATCH(request) {
|
||||
);
|
||||
}
|
||||
|
||||
// #12172: optional modality scope (e.g. "chat", "images") so hiding a model on one
|
||||
// registry surface does not also hide an identically-ID'd model on another one.
|
||||
// Omitted = legacy "hide everywhere" behavior, unchanged for existing callers.
|
||||
if (typeof body.modality !== "undefined" && typeof body.modality !== "string") {
|
||||
return Response.json(
|
||||
{ error: { message: "modality must be a string when provided", type: "validation_error" } },
|
||||
{ status: 400 }
|
||||
);
|
||||
}
|
||||
const modality = typeof body.modality === "string" && body.modality ? body.modality : undefined;
|
||||
|
||||
const modelIds = normalizeRequestedModelIds(searchParams, body);
|
||||
if (modelIds.length === 0) {
|
||||
return Response.json(
|
||||
@@ -439,7 +428,7 @@ export async function PATCH(request) {
|
||||
for (const modelId of modelIds) {
|
||||
const updatedModel = await updateCustomModel(provider, modelId, { isHidden: body.isHidden });
|
||||
if (!updatedModel) {
|
||||
mergeModelCompatOverride(provider, modelId, { isHidden: body.isHidden, modality });
|
||||
mergeModelCompatOverride(provider, modelId, { isHidden: body.isHidden });
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
@@ -290,21 +290,17 @@ async function buildUnifiedModelsResponseCore(
|
||||
}
|
||||
};
|
||||
try {
|
||||
// #9147/#12172: bulk-load the hidden-model map once PER MODALITY (memoized below,
|
||||
// one SQLite query per modality actually used) instead of `getModelIsHidden()`'s
|
||||
// per-call read — per-modality because chat/images/etc. registries can share a
|
||||
// literal model id and must be hideable independently (#12172). Deliberately kept
|
||||
// INSIDE this try block: the builder's catch below sanitizes a build-time failure
|
||||
// into a 500 instead of a rejected promise.
|
||||
const hiddenModelsByModality = new Map<string, Map<string, Set<string>>>();
|
||||
const getHiddenModelsForModality = (modality: string): Map<string, Set<string>> => {
|
||||
let m = hiddenModelsByModality.get(modality);
|
||||
if (!m) {
|
||||
m = getHiddenModelsByProvider(modality);
|
||||
hiddenModelsByModality.set(modality, m);
|
||||
}
|
||||
return m;
|
||||
};
|
||||
// #9147: `getModelIsHidden()` is a SQLite read per call (custom row + compat list)
|
||||
// and the build consults it ~16× per entry. Bulk-load the hidden-model map once
|
||||
// (one query — `getHiddenModelsByProvider`) and resolve from memory for the whole
|
||||
// build. A provider absent from the map has no hidden models at all — `false`,
|
||||
// no on-demand fallback (that would reintroduce the per-call SQLite reads).
|
||||
// Deliberately kept INSIDE this try block (not hoisted above it): the builder's
|
||||
// own catch below is what converts a build-time failure into a sanitized 500
|
||||
// Response instead of a rejected promise — hoisting this bulk read above the
|
||||
// try would let a crash here propagate as an unhandled rejection instead
|
||||
// (catalogCache.ts's in-flight coalescing does not fully consume rejections).
|
||||
const hiddenModelsByProvider = getHiddenModelsByProvider();
|
||||
let settings: Record<string, any> = {};
|
||||
try {
|
||||
settings = await getSettings();
|
||||
@@ -432,8 +428,7 @@ async function buildUnifiedModelsResponseCore(
|
||||
const isModelHiddenBulk = (
|
||||
providerKey: string | null | undefined,
|
||||
modelId: string,
|
||||
canonicalProviderId?: string | null,
|
||||
modality: string = "chat"
|
||||
canonicalProviderId?: string | null
|
||||
): boolean => {
|
||||
if (!providerKey || !modelId) return false;
|
||||
const canonical = canonicalProviderId || resolveCanonicalProviderId(providerKey);
|
||||
@@ -442,9 +437,8 @@ async function buildUnifiedModelsResponseCore(
|
||||
const keysToCheck = [providerKey, canonical, alias, nodePrefix].filter((k): k is string =>
|
||||
Boolean(k)
|
||||
);
|
||||
const hiddenModelsForModality = getHiddenModelsForModality(modality);
|
||||
for (const key of keysToCheck) {
|
||||
const hiddenSet = hiddenModelsForModality.get(key);
|
||||
const hiddenSet = hiddenModelsByProvider.get(key);
|
||||
if (hiddenSet?.has(modelId)) return true;
|
||||
}
|
||||
return false;
|
||||
@@ -1473,7 +1467,7 @@ async function buildUnifiedModelsResponseCore(
|
||||
if (!isProviderActive(embModel.provider)) continue;
|
||||
const rawModelId = getSpecialtyModelRelativeId(embModel.id, embModel.provider);
|
||||
if (!providerSupportsModel(embModel.provider, rawModelId)) continue;
|
||||
if (isModelHiddenBulk(embModel.provider, rawModelId, null, "embeddings")) continue;
|
||||
if (isModelHiddenBulk(embModel.provider, rawModelId)) continue;
|
||||
const existingEmbedding = findEquivalentSpecialtyModel(
|
||||
embModel.provider,
|
||||
rawModelId,
|
||||
@@ -1516,7 +1510,7 @@ async function buildUnifiedModelsResponseCore(
|
||||
if (!isProviderActive(imgModel.provider)) continue;
|
||||
const rawModelId = getSpecialtyModelRelativeId(imgModel.id, imgModel.provider);
|
||||
if (!providerSupportsModel(imgModel.provider, rawModelId)) continue;
|
||||
if (isModelHiddenBulk(imgModel.provider, rawModelId, null, "images")) continue;
|
||||
if (isModelHiddenBulk(imgModel.provider, rawModelId)) continue;
|
||||
models.push({
|
||||
id: imgModel.id,
|
||||
object: "model",
|
||||
@@ -1536,7 +1530,7 @@ async function buildUnifiedModelsResponseCore(
|
||||
if (!isProviderActive(rerankModel.provider)) continue;
|
||||
const rawModelId = getSpecialtyModelRelativeId(rerankModel.id, rerankModel.provider);
|
||||
if (!providerSupportsModel(rerankModel.provider, rawModelId)) continue;
|
||||
if (isModelHiddenBulk(rerankModel.provider, rawModelId, null, "rerank")) continue;
|
||||
if (isModelHiddenBulk(rerankModel.provider, rawModelId)) continue;
|
||||
if (hasEquivalentSpecialtyModel(rerankModel.provider, rawModelId, "rerank", rerankModel.id)) {
|
||||
continue;
|
||||
}
|
||||
@@ -1555,7 +1549,7 @@ async function buildUnifiedModelsResponseCore(
|
||||
if (!isProviderActive(audioModel.provider)) continue;
|
||||
const rawModelId = getSpecialtyModelRelativeId(audioModel.id, audioModel.provider);
|
||||
if (!providerSupportsModel(audioModel.provider, rawModelId)) continue;
|
||||
if (isModelHiddenBulk(audioModel.provider, rawModelId, null, "audio")) continue;
|
||||
if (isModelHiddenBulk(audioModel.provider, rawModelId)) continue;
|
||||
models.push({
|
||||
id: audioModel.id,
|
||||
object: "model",
|
||||
@@ -1571,7 +1565,7 @@ async function buildUnifiedModelsResponseCore(
|
||||
if (!isProviderActive(modModel.provider)) continue;
|
||||
const rawModelId = getSpecialtyModelRelativeId(modModel.id, modModel.provider);
|
||||
if (!providerSupportsModel(modModel.provider, rawModelId)) continue;
|
||||
if (isModelHiddenBulk(modModel.provider, rawModelId, null, "moderation")) continue;
|
||||
if (isModelHiddenBulk(modModel.provider, rawModelId)) continue;
|
||||
models.push({
|
||||
id: modModel.id,
|
||||
object: "model",
|
||||
@@ -1586,7 +1580,7 @@ async function buildUnifiedModelsResponseCore(
|
||||
if (!isProviderActive(videoModel.provider)) continue;
|
||||
const rawModelId = getSpecialtyModelRelativeId(videoModel.id, videoModel.provider);
|
||||
if (!providerSupportsModel(videoModel.provider, rawModelId)) continue;
|
||||
if (isModelHiddenBulk(videoModel.provider, rawModelId, null, "videos")) continue;
|
||||
if (isModelHiddenBulk(videoModel.provider, rawModelId)) continue;
|
||||
models.push({
|
||||
id: videoModel.id,
|
||||
object: "model",
|
||||
@@ -1607,7 +1601,7 @@ async function buildUnifiedModelsResponseCore(
|
||||
if (!isProviderActive(musicModel.provider)) continue;
|
||||
const rawModelId = getSpecialtyModelRelativeId(musicModel.id, musicModel.provider);
|
||||
if (!providerSupportsModel(musicModel.provider, rawModelId)) continue;
|
||||
if (isModelHiddenBulk(musicModel.provider, rawModelId, null, "music")) continue;
|
||||
if (isModelHiddenBulk(musicModel.provider, rawModelId)) continue;
|
||||
models.push({
|
||||
id: musicModel.id,
|
||||
object: "model",
|
||||
|
||||
@@ -28,8 +28,6 @@ import {
|
||||
isCompatProtocolKey,
|
||||
sanitizeUpstreamHeadersMap,
|
||||
removeModelCompatOverride,
|
||||
mergeModelCompatOverride,
|
||||
isOverrideHiddenForModality,
|
||||
type CompatByProtocolMap,
|
||||
type ModelCompatProtocolKey,
|
||||
type ModelCompatOverride,
|
||||
@@ -977,30 +975,22 @@ export function getModelPreserveOpenAIDeveloperRole(
|
||||
|
||||
/**
|
||||
* Check if the model is flagged as hidden from the public catalog.
|
||||
* `modality` (default "chat") scopes the check to one endpoint/registry — see
|
||||
* {@link isOverrideHiddenForModality} — so an identically-ID'd model in a different
|
||||
* modality's registry (e.g. Chat vs Image, #12172) is not silently suppressed too.
|
||||
*/
|
||||
export function getModelIsHidden(
|
||||
providerId: string,
|
||||
modelId: string,
|
||||
modality: string = "chat"
|
||||
): boolean {
|
||||
export function getModelIsHidden(providerId: string, modelId: string): boolean {
|
||||
const m = getCustomModelRow(providerId, modelId);
|
||||
if (m && Object.prototype.hasOwnProperty.call(m, "isHidden")) {
|
||||
return Boolean(m.isHidden);
|
||||
}
|
||||
const co = readCompatList(providerId).find((e) => e.id === modelId);
|
||||
return isOverrideHiddenForModality(co, modality);
|
||||
return Boolean(co?.isHidden);
|
||||
}
|
||||
|
||||
/**
|
||||
* Get a map of provider ID → set of hidden model IDs from all modelCompatOverrides
|
||||
* and customModels, scoped to one `modality` (default "chat", matching every
|
||||
* pre-#12172 caller's original chat-only intent). Used by auto-combo candidate
|
||||
* building to skip user-hidden models. Single bulk DB query — not N+1 per model.
|
||||
* and customModels. Used by auto-combo candidate building to skip user-hidden models.
|
||||
* Single bulk DB query — not N+1 per model.
|
||||
*/
|
||||
export function getHiddenModelsByProvider(modality: string = "chat"): Map<string, Set<string>> {
|
||||
export function getHiddenModelsByProvider(): Map<string, Set<string>> {
|
||||
const db = getDbInstance();
|
||||
const visibilityByProvider = new Map<string, Map<string, boolean>>();
|
||||
const rows = db
|
||||
@@ -1019,33 +1009,13 @@ export function getHiddenModelsByProvider(modality: string = "chat"): Map<string
|
||||
if (!entry || typeof entry !== "object") continue;
|
||||
const modelId = (entry as { id?: unknown }).id;
|
||||
if (typeof modelId !== "string" || modelId.length === 0) continue;
|
||||
const record = entry as { isHidden?: unknown; hiddenModalities?: unknown };
|
||||
const hasHiddenInfo =
|
||||
Object.prototype.hasOwnProperty.call(record, "isHidden") ||
|
||||
(namespace === "modelCompatOverrides" &&
|
||||
record.hiddenModalities &&
|
||||
typeof record.hiddenModalities === "object");
|
||||
if (!hasHiddenInfo) continue;
|
||||
// #12172: customModels rows have no modality scope (single user-managed
|
||||
// entry) — legacy global isHidden applies to every modality unchanged.
|
||||
const isHidden =
|
||||
namespace === "modelCompatOverrides"
|
||||
? isOverrideHiddenForModality(
|
||||
{
|
||||
isHidden: Boolean(record.isHidden),
|
||||
hiddenModalities: record.hiddenModalities as
|
||||
| Record<string, boolean>
|
||||
| undefined,
|
||||
},
|
||||
modality
|
||||
)
|
||||
: Boolean(record.isHidden);
|
||||
if (!Object.prototype.hasOwnProperty.call(entry, "isHidden")) continue;
|
||||
let visibility = visibilityByProvider.get(row.key);
|
||||
if (!visibility) {
|
||||
visibility = new Map<string, boolean>();
|
||||
visibilityByProvider.set(row.key, visibility);
|
||||
}
|
||||
visibility.set(modelId, isHidden);
|
||||
visibility.set(modelId, Boolean((entry as { isHidden?: unknown }).isHidden));
|
||||
}
|
||||
} catch {
|
||||
// Skip malformed entries
|
||||
@@ -1068,12 +1038,7 @@ export function getHiddenModelsByProvider(modality: string = "chat"): Map<string
|
||||
* row when one exists, otherwise on the compat-override list. Setting
|
||||
* `hidden = false` is a no-op when the model is already visible.
|
||||
*/
|
||||
export function setModelIsHidden(
|
||||
providerId: string,
|
||||
modelId: string,
|
||||
hidden: boolean,
|
||||
modality?: string
|
||||
): void {
|
||||
export function setModelIsHidden(providerId: string, modelId: string, hidden: boolean): void {
|
||||
const customRow = getCustomModelRow(providerId, modelId);
|
||||
if (customRow) {
|
||||
if (hidden) {
|
||||
@@ -1084,14 +1049,6 @@ export function setModelIsHidden(
|
||||
return;
|
||||
}
|
||||
|
||||
// #12172: a modality-scoped write never touches the legacy all-modalities
|
||||
// `isHidden` flag — it only sets/clears that one modality's override, so an
|
||||
// identically-ID'd model in a different modality's registry is unaffected.
|
||||
if (modality) {
|
||||
mergeModelCompatOverride(providerId, modelId, { isHidden: hidden, modality });
|
||||
return;
|
||||
}
|
||||
|
||||
const list = readCompatList(providerId);
|
||||
const idx = list.findIndex((e) => e.id === modelId);
|
||||
if (hidden) {
|
||||
|
||||
@@ -114,36 +114,11 @@ export type ModelCompatOverride = {
|
||||
compatByProtocol?: CompatByProtocolMap;
|
||||
upstreamHeaders?: Record<string, string>;
|
||||
isHidden?: boolean;
|
||||
/**
|
||||
* #12172: per-modality visibility override, keyed by endpoint/modality id
|
||||
* (e.g. "chat", "images", "embeddings", ...). A key present here always wins
|
||||
* over the legacy top-level `isHidden` for that specific modality — this is
|
||||
* what lets an operator hide a model from Chat without also suppressing an
|
||||
* identically-ID'd model in the Image (or any other) registry. A modality
|
||||
* with no entry here falls back to `isHidden` (the pre-#12172 "hide
|
||||
* everywhere" behavior), so existing rows keep working unchanged.
|
||||
*/
|
||||
hiddenModalities?: Record<string, boolean>;
|
||||
apiFormat?: string;
|
||||
targetFormat?: string;
|
||||
supportsVision?: boolean;
|
||||
};
|
||||
|
||||
/**
|
||||
* Resolve whether an override hides its model for a given modality.
|
||||
* Precedence: an explicit `hiddenModalities[modality]` entry always wins;
|
||||
* otherwise fall back to the legacy all-modalities `isHidden` flag.
|
||||
*/
|
||||
export function isOverrideHiddenForModality(
|
||||
override: Pick<ModelCompatOverride, "isHidden" | "hiddenModalities"> | null | undefined,
|
||||
modality: string
|
||||
): boolean {
|
||||
if (!override) return false;
|
||||
const scoped = override.hiddenModalities?.[modality];
|
||||
if (scoped !== undefined) return Boolean(scoped);
|
||||
return Boolean(override.isHidden);
|
||||
}
|
||||
|
||||
export function readCompatList(providerId: string): ModelCompatOverride[] {
|
||||
const db = getDbInstance();
|
||||
const row = db
|
||||
@@ -196,13 +171,6 @@ export type ModelCompatPatch = {
|
||||
/** Replace top-level extra headers for override-only rows; omit to leave unchanged. */
|
||||
upstreamHeaders?: Record<string, string> | null;
|
||||
isHidden?: boolean | null;
|
||||
/**
|
||||
* #12172: when set alongside `isHidden`, scopes the write to that one
|
||||
* modality (see {@link ModelCompatOverride.hiddenModalities}) instead of
|
||||
* the legacy all-modalities flag. `isHidden: null` with a `modality` clears
|
||||
* just that modality's override (reverting it to inherit the legacy flag).
|
||||
*/
|
||||
modality?: string | null;
|
||||
apiFormat?: string | null;
|
||||
targetFormat?: string | null;
|
||||
supportsVision?: boolean | null;
|
||||
@@ -262,17 +230,7 @@ export function mergeModelCompatOverride(
|
||||
const hasVideoUrlFlag = Object.prototype.hasOwnProperty.call(next, "preserveVideoUrl");
|
||||
const hasTopUpstream = next.upstreamHeaders && Object.keys(next.upstreamHeaders).length > 0;
|
||||
if ("isHidden" in patch) {
|
||||
const modality = typeof patch.modality === "string" && patch.modality ? patch.modality : null;
|
||||
if (modality) {
|
||||
const hiddenModalities = { ...(next.hiddenModalities || {}) };
|
||||
if (patch.isHidden === null) {
|
||||
delete hiddenModalities[modality];
|
||||
} else {
|
||||
hiddenModalities[modality] = Boolean(patch.isHidden);
|
||||
}
|
||||
if (Object.keys(hiddenModalities).length > 0) next.hiddenModalities = hiddenModalities;
|
||||
else delete next.hiddenModalities;
|
||||
} else if (patch.isHidden === null) {
|
||||
if (patch.isHidden === null) {
|
||||
delete next.isHidden;
|
||||
} else {
|
||||
next.isHidden = Boolean(patch.isHidden);
|
||||
@@ -299,9 +257,7 @@ export function mergeModelCompatOverride(
|
||||
next.supportsVision = Boolean(patch.supportsVision);
|
||||
}
|
||||
}
|
||||
const hasHiddenFlag =
|
||||
Object.prototype.hasOwnProperty.call(next, "isHidden") ||
|
||||
(!!next.hiddenModalities && Object.keys(next.hiddenModalities).length > 0);
|
||||
const hasHiddenFlag = Object.prototype.hasOwnProperty.call(next, "isHidden");
|
||||
const hasApiFormat = Object.prototype.hasOwnProperty.call(next, "apiFormat");
|
||||
const hasTargetFormat = Object.prototype.hasOwnProperty.call(next, "targetFormat");
|
||||
const hasVisionFlag = Object.prototype.hasOwnProperty.call(next, "supportsVision");
|
||||
|
||||
@@ -1,82 +0,0 @@
|
||||
/**
|
||||
* Regression test for #12172 — a model ID collision between the Chat registry and a
|
||||
* specialty modality registry (Image) prevented independent visibility toggling,
|
||||
* because `modelCompatOverrides` was keyed only by (providerId, modelId) with no
|
||||
* modality/endpoint field: hiding the chat model also hid the identically-ID'd
|
||||
* image model, and vice versa.
|
||||
*/
|
||||
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-12172-modality-"));
|
||||
process.env.DATA_DIR = TEST_DATA_DIR;
|
||||
process.env.API_KEY_SECRET = process.env.API_KEY_SECRET || "modality-collision-test-secret";
|
||||
|
||||
const { setModelIsHidden, getModelIsHidden, getHiddenModelsByProvider } = await import(
|
||||
"../../src/lib/db/models.ts"
|
||||
);
|
||||
const { resetDbInstance } = await import("../../src/lib/db/core.ts");
|
||||
const { codexProvider } = await import(
|
||||
"../../open-sse/config/providers/registry/codex/index.ts"
|
||||
);
|
||||
const { IMAGE_PROVIDERS } = await import("../../open-sse/config/imageRegistry.ts");
|
||||
|
||||
test.after(() => {
|
||||
resetDbInstance();
|
||||
fs.rmSync(TEST_DATA_DIR, { recursive: true, force: true });
|
||||
});
|
||||
|
||||
test("#12172: codex chat and image registries collide on the same model id (premise)", () => {
|
||||
const chatModel = codexProvider.models.find((m) => m.id === "gpt-5.6-sol");
|
||||
const imageModel = IMAGE_PROVIDERS.codex.models.find((m) => m.id === "gpt-5.6-sol");
|
||||
assert.ok(chatModel, "codex chat registry must define gpt-5.6-sol (premise check)");
|
||||
assert.ok(imageModel, "codex image registry must define gpt-5.6-sol (premise check)");
|
||||
});
|
||||
|
||||
test("#12172: hiding the codex CHAT model gpt-5.6-sol must not suppress the codex IMAGE model", () => {
|
||||
setModelIsHidden("codex", "gpt-5.6-sol", true, "chat");
|
||||
|
||||
const chatHidden = getHiddenModelsByProvider("chat").get("codex");
|
||||
const imageHidden = getHiddenModelsByProvider("images").get("codex");
|
||||
|
||||
assert.equal(chatHidden?.has("gpt-5.6-sol"), true, "chat model must be hidden after the toggle");
|
||||
assert.equal(
|
||||
imageHidden?.has("gpt-5.6-sol") ?? false,
|
||||
false,
|
||||
"BUG #12172: hiding the chat model must not also hide the identically-ID'd image model"
|
||||
);
|
||||
|
||||
assert.equal(getModelIsHidden("codex", "gpt-5.6-sol", "chat"), true);
|
||||
assert.equal(getModelIsHidden("codex", "gpt-5.6-sol", "images"), false);
|
||||
});
|
||||
|
||||
test("#12172: unhiding one modality does not affect the other", () => {
|
||||
setModelIsHidden("codex", "gpt-5.6-terra", true, "chat");
|
||||
setModelIsHidden("codex", "gpt-5.6-terra", true, "images");
|
||||
assert.equal(getModelIsHidden("codex", "gpt-5.6-terra", "chat"), true);
|
||||
assert.equal(getModelIsHidden("codex", "gpt-5.6-terra", "images"), true);
|
||||
|
||||
setModelIsHidden("codex", "gpt-5.6-terra", false, "chat");
|
||||
assert.equal(
|
||||
getModelIsHidden("codex", "gpt-5.6-terra", "chat"),
|
||||
false,
|
||||
"chat toggle must not be affected by the images toggle"
|
||||
);
|
||||
assert.equal(
|
||||
getModelIsHidden("codex", "gpt-5.6-terra", "images"),
|
||||
true,
|
||||
"images toggle must remain hidden after unhiding chat only"
|
||||
);
|
||||
});
|
||||
|
||||
test("#12172: a legacy (no-modality) hide keeps suppressing every modality — backward compatible", () => {
|
||||
setModelIsHidden("codex", "gpt-5.6-luna", true);
|
||||
|
||||
assert.equal(getModelIsHidden("codex", "gpt-5.6-luna", "chat"), true);
|
||||
assert.equal(getModelIsHidden("codex", "gpt-5.6-luna", "images"), true);
|
||||
assert.equal(getHiddenModelsByProvider("chat").get("codex")?.has("gpt-5.6-luna"), true);
|
||||
assert.equal(getHiddenModelsByProvider("images").get("codex")?.has("gpt-5.6-luna"), true);
|
||||
});
|
||||
121
tests/unit/repro-12783-setup-opencode-apikey.test.ts
Normal file
121
tests/unit/repro-12783-setup-opencode-apikey.test.ts
Normal file
@@ -0,0 +1,121 @@
|
||||
import assert from "node:assert/strict";
|
||||
import { mkdtempSync, rmSync, writeFileSync } from "node:fs";
|
||||
import { tmpdir } from "node:os";
|
||||
import { join } from "node:path";
|
||||
import { test } from "node:test";
|
||||
|
||||
import { resolveOpencodeTarget } from "../../bin/cli/commands/setup-opencode.mjs";
|
||||
|
||||
/** Point OMNIROUTE_CONTEXT config resolution at an isolated, throwaway DATA_DIR. */
|
||||
function withIsolatedContext(contextConfig, fn) {
|
||||
const dir = mkdtempSync(join(tmpdir(), "omniroute-setup-opencode-test-"));
|
||||
const originalDataDir = process.env.DATA_DIR;
|
||||
process.env.DATA_DIR = dir;
|
||||
writeFileSync(
|
||||
join(dir, "config.json"),
|
||||
JSON.stringify({
|
||||
version: 1,
|
||||
currentContext: "remote",
|
||||
contexts: { remote: contextConfig },
|
||||
})
|
||||
);
|
||||
try {
|
||||
return fn();
|
||||
} finally {
|
||||
if (originalDataDir === undefined) delete process.env.DATA_DIR;
|
||||
else process.env.DATA_DIR = originalDataDir;
|
||||
rmSync(dir, { recursive: true, force: true });
|
||||
}
|
||||
}
|
||||
|
||||
function withEnvApiKey(value, fn) {
|
||||
const original = process.env.OMNIROUTE_API_KEY;
|
||||
if (value === undefined) delete process.env.OMNIROUTE_API_KEY;
|
||||
else process.env.OMNIROUTE_API_KEY = value;
|
||||
try {
|
||||
return fn();
|
||||
} finally {
|
||||
if (original === undefined) delete process.env.OMNIROUTE_API_KEY;
|
||||
else process.env.OMNIROUTE_API_KEY = original;
|
||||
}
|
||||
}
|
||||
|
||||
test("setup-opencode: --api-key typed AFTER the subcommand name is not stolen by the parent program's global option", async () => {
|
||||
const { createProgram } = await import("../../bin/cli/program.mjs");
|
||||
const program = createProgram();
|
||||
const setupOpencode = program.commands.find((c) => c.name() === "setup-opencode");
|
||||
assert.ok(setupOpencode, "setup-opencode subcommand must be registered");
|
||||
|
||||
let capturedApiKey;
|
||||
setupOpencode._actionHandler = null; // avoid the real network-calling action
|
||||
setupOpencode.action((opts, cmd) => {
|
||||
capturedApiKey = cmd.optsWithGlobals().apiKey ?? opts.apiKey;
|
||||
});
|
||||
|
||||
await program.parseAsync(
|
||||
[
|
||||
"node",
|
||||
"omniroute",
|
||||
"setup-opencode",
|
||||
"--remote",
|
||||
"http://100.64.0.1:20128",
|
||||
"--api-key",
|
||||
"sk-TESTKEY123",
|
||||
],
|
||||
{ from: "node" }
|
||||
);
|
||||
|
||||
assert.equal(
|
||||
capturedApiKey,
|
||||
"sk-TESTKEY123",
|
||||
"the CLI-supplied --api-key value must reach the setup-opencode action handler"
|
||||
);
|
||||
});
|
||||
|
||||
test("resolveOpencodeTarget: (a) explicit --api-key flag wins over an active context's management token", () => {
|
||||
withEnvApiKey(undefined, () => {
|
||||
withIsolatedContext(
|
||||
{ baseUrl: "http://100.64.0.1:20128", accessToken: "oma_live_CONTEXT_TOKEN" },
|
||||
() => {
|
||||
const { apiKey } = resolveOpencodeTarget({ apiKey: "sk-FLAG", context: "remote" });
|
||||
assert.equal(apiKey, "sk-FLAG");
|
||||
}
|
||||
);
|
||||
});
|
||||
});
|
||||
|
||||
test("resolveOpencodeTarget: (b) OMNIROUTE_API_KEY env wins over an active context's management token when no flag is passed", () => {
|
||||
withEnvApiKey("sk-ENVKEY", () => {
|
||||
withIsolatedContext(
|
||||
{ baseUrl: "http://100.64.0.1:20128", accessToken: "oma_live_CONTEXT_TOKEN" },
|
||||
() => {
|
||||
const { apiKey } = resolveOpencodeTarget({ context: "remote" });
|
||||
assert.equal(apiKey, "sk-ENVKEY");
|
||||
}
|
||||
);
|
||||
});
|
||||
});
|
||||
|
||||
test("resolveOpencodeTarget: (c) the context's token is used only when neither a flag nor the env var is set", () => {
|
||||
withEnvApiKey(undefined, () => {
|
||||
withIsolatedContext(
|
||||
{ baseUrl: "http://100.64.0.1:20128", accessToken: "oma_live_CONTEXT_TOKEN" },
|
||||
() => {
|
||||
const { apiKey } = resolveOpencodeTarget({ context: "remote" });
|
||||
assert.equal(apiKey, "oma_live_CONTEXT_TOKEN");
|
||||
}
|
||||
);
|
||||
});
|
||||
});
|
||||
|
||||
test("resolveOpencodeTarget: falls back to '' when neither a flag, env var, nor a resolvable context is present", () => {
|
||||
withEnvApiKey(undefined, () => {
|
||||
withIsolatedContext({ baseUrl: "http://100.64.0.1:20128" }, () => {
|
||||
const { apiKey } = resolveOpencodeTarget({
|
||||
remote: "http://100.64.0.1:20128",
|
||||
context: "__no-such-context__",
|
||||
});
|
||||
assert.equal(apiKey, "");
|
||||
});
|
||||
});
|
||||
});
|
||||
Reference in New Issue
Block a user