mirror of
https://github.com/diegosouzapw/OmniRoute.git
synced 2026-08-04 22:32:12 +03:00
ca03d619a4
* feat(mitm): add Antigravity reasoning-effort overrides
The Antigravity MITM alias mapping only ever swapped the destination model;
there was no way to override the reasoning effort Antigravity's own
thinkingConfig requested. Alias entries are now `{ model?, reasoningEffort? }`
(a legacy plain-string mapping still normalizes to `{ model }`, so no DB
migration is required). The standalone proxy (server.cjs) forwards the chosen
tier as a top-level `reasoningEffortOverride` on the intercepted request; the
antigravity->openai translator honors it ahead of its thinkingConfig-derived
guess, and an explicit "none" suppresses reasoning_effort entirely even when
Antigravity's own request asked for thinking. Reuses the existing canonical
5-tier reasoning vocabulary (`@/shared/reasoning/effortStandardization.ts`,
with max/extra aliasing to xhigh) instead of introducing a new one. The API
route validates the reasoning-effort value at the boundary and the Antigravity
tool card UI now exposes a per-model reasoning-effort selector alongside the
existing model-mapping input.
Co-authored-by: Truong Fiu <gnourtf@gmail.com>
Inspired-by: https://github.com/decolua/9router/pull/2584
* chore(changelog): fragment for #7228
---------
Co-authored-by: Truong Fiu <gnourtf@gmail.com>
67 lines
3.1 KiB
TypeScript
67 lines
3.1 KiB
TypeScript
import test from "node:test";
|
|
import assert from "node:assert/strict";
|
|
import { createRequire } from "node:module";
|
|
|
|
// CJS mirror consumed by the standalone proxy process (server.cjs — cannot import the
|
|
// ESM/TS `src/mitm/aliasConfig.ts`). Ported from upstream decolua/9router#2584 ("add
|
|
// Antigravity reasoning effort overrides"). Keep assertions aligned with
|
|
// `tests/unit/mitm-antigravity-reasoning-effort-override.test.ts` (the TS counterpart).
|
|
const require = createRequire(import.meta.url);
|
|
const {
|
|
normalizeReasoningEffort,
|
|
normalizeAliasEntry,
|
|
normalizeAliasMappings,
|
|
applyAntigravityOverride,
|
|
} = require("../../src/mitm/_internal/aliasConfig.cjs");
|
|
|
|
test("normalizeReasoningEffort canonicalizes case and the max/extra UI synonyms", () => {
|
|
assert.equal(normalizeReasoningEffort(" HIGH "), "high");
|
|
assert.equal(normalizeReasoningEffort("max"), "xhigh");
|
|
assert.equal(normalizeReasoningEffort("extra"), "xhigh");
|
|
assert.equal(normalizeReasoningEffort("extreme"), undefined);
|
|
assert.equal(normalizeReasoningEffort(42), undefined);
|
|
});
|
|
|
|
test("normalizeAliasEntry upgrades a legacy string mapping", () => {
|
|
assert.deepEqual(normalizeAliasEntry("provider/model-id"), { model: "provider/model-id" });
|
|
assert.equal(normalizeAliasEntry(""), null);
|
|
});
|
|
|
|
test("normalizeAliasMappings resolves the stored SQLite row shape used by getMappedOverride", () => {
|
|
const stored = {
|
|
"gemini-3-flash-agent": "cx/gpt-5.6-sol",
|
|
"gemini-3-pro-agent": { model: "cx/gpt-5.6-sol", reasoningEffort: "high" },
|
|
"gemini-3-flash-thinking-agent": { reasoningEffort: "none" },
|
|
};
|
|
const normalized = normalizeAliasMappings(stored);
|
|
assert.deepEqual(normalized["gemini-3-flash-agent"], { model: "cx/gpt-5.6-sol" });
|
|
assert.deepEqual(normalized["gemini-3-pro-agent"], {
|
|
model: "cx/gpt-5.6-sol",
|
|
reasoningEffort: "high",
|
|
});
|
|
assert.deepEqual(normalized["gemini-3-flash-thinking-agent"], { reasoningEffort: "none" });
|
|
});
|
|
|
|
test("applyAntigravityOverride swaps model and sets the top-level reasoningEffortOverride", () => {
|
|
const body = { model: "gemini-3-flash-agent", request: { contents: [] } };
|
|
const result = applyAntigravityOverride(body, { model: "cx/gpt-5.6-sol", reasoningEffort: "high" });
|
|
assert.equal(result.model, "cx/gpt-5.6-sol");
|
|
assert.equal(result.reasoningEffortOverride, "high");
|
|
// Original body is untouched (server.cjs still needs it for logging/capture).
|
|
assert.equal(body.model, "gemini-3-flash-agent");
|
|
assert.equal("reasoningEffortOverride" in body, false);
|
|
});
|
|
|
|
test("applyAntigravityOverride applies a reasoning-only override without touching model", () => {
|
|
const body = { model: "gemini-3-flash-agent", request: { contents: [] } };
|
|
const result = applyAntigravityOverride(body, { reasoningEffort: "none" });
|
|
assert.equal(result.model, "gemini-3-flash-agent");
|
|
assert.equal(result.reasoningEffortOverride, "none");
|
|
});
|
|
|
|
test("applyAntigravityOverride is a no-op pass-through when override carries neither field", () => {
|
|
const body = { model: "gemini-3-flash-agent" };
|
|
const result = applyAntigravityOverride(body, {});
|
|
assert.deepEqual(result, body);
|
|
});
|