mirror of
https://github.com/diegosouzapw/OmniRoute.git
synced 2026-08-03 22:02:08 +03:00
d78a836d3c
Squash merge from local merge-train (Hard Rule owner-approved). Tip 029cdf4215cf465f0e1716ac9f84a84692b1e881 validated on 192.168.0.113: 26631/26653 pass.
110 lines
4.8 KiB
TypeScript
110 lines
4.8 KiB
TypeScript
import { test, after, beforeEach } from "node:test";
|
|
import assert from "node:assert/strict";
|
|
import {
|
|
getLearnedThinkingCap,
|
|
recordLearnedThinkingCap,
|
|
parseThinkingBudgetMax,
|
|
__test_resetLearnedThinkingCaps,
|
|
} from "../../open-sse/services/learnedThinkingCaps.ts";
|
|
|
|
beforeEach(() => {
|
|
__test_resetLearnedThinkingCaps();
|
|
});
|
|
|
|
after(() => {
|
|
__test_resetLearnedThinkingCaps();
|
|
});
|
|
|
|
// ── parseThinkingBudgetMax ─────────────────────────────────────────────────
|
|
|
|
test("parseThinkingBudgetMax extracts 65535 from the real Gemini error body", () => {
|
|
const err =
|
|
"* GenerateContentRequest.generation_config.thinking_config.thinking_budget: " +
|
|
"thinking_budget must be in the range [-1, 65535]";
|
|
assert.equal(parseThinkingBudgetMax(err), 65535);
|
|
});
|
|
|
|
test("parseThinkingBudgetMax handles camelCase thinkingBudget variant", () => {
|
|
const err = "generationConfig.thinkingConfig.thinkingBudget must be in the range [-1, 24576]";
|
|
assert.equal(parseThinkingBudgetMax(err), 24576);
|
|
});
|
|
|
|
test("parseThinkingBudgetMax returns null for unrelated 400 text", () => {
|
|
assert.equal(parseThinkingBudgetMax("some random upstream error"), null);
|
|
assert.equal(parseThinkingBudgetMax(""), null);
|
|
assert.equal(parseThinkingBudgetMax(null), null);
|
|
assert.equal(parseThinkingBudgetMax(undefined), null);
|
|
});
|
|
|
|
test("parseThinkingBudgetMax returns null when thinking_budget mentioned but no range", () => {
|
|
assert.equal(parseThinkingBudgetMax("thinking_budget is deprecated"), null);
|
|
});
|
|
|
|
// ── recordLearnedThinkingCap step ladder ───────────────────────────────────
|
|
|
|
test("recordLearnedThinkingCap steps 131072 → 32768 on first failure", () => {
|
|
const next = recordLearnedThinkingCap("gemini", "gemini-2.5-pro", 131072);
|
|
assert.equal(next, 32768);
|
|
assert.equal(getLearnedThinkingCap("gemini", "gemini-2.5-pro"), 32768);
|
|
});
|
|
|
|
test("recordLearnedThinkingCap steps 32768 → 24576 on second failure", () => {
|
|
recordLearnedThinkingCap("gemini", "gemini-2.5-pro", 131072);
|
|
const next = recordLearnedThinkingCap("gemini", "gemini-2.5-pro", 32768);
|
|
assert.equal(next, 24576);
|
|
assert.equal(getLearnedThinkingCap("gemini", "gemini-2.5-pro"), 24576);
|
|
});
|
|
|
|
test("recordLearnedThinkingCap steps 24576 → 8192 on third failure", () => {
|
|
recordLearnedThinkingCap("gemini", "gemini-2.5-pro", 131072);
|
|
recordLearnedThinkingCap("gemini", "gemini-2.5-pro", 32768);
|
|
const next = recordLearnedThinkingCap("gemini", "gemini-2.5-pro", 24576);
|
|
assert.equal(next, 8192);
|
|
assert.equal(getLearnedThinkingCap("gemini", "gemini-2.5-pro"), 8192);
|
|
});
|
|
|
|
test("recordLearnedThinkingCap returns null when steps exhausted (8192 failed)", () => {
|
|
recordLearnedThinkingCap("gemini", "gemini-2.5-pro", 131072);
|
|
recordLearnedThinkingCap("gemini", "gemini-2.5-pro", 32768);
|
|
recordLearnedThinkingCap("gemini", "gemini-2.5-pro", 24576);
|
|
const next = recordLearnedThinkingCap("gemini", "gemini-2.5-pro", 8192);
|
|
assert.equal(next, null);
|
|
});
|
|
|
|
test("recordLearnedThinkingCap picks the first step strictly below failedBudget", () => {
|
|
// A budget of 30000 failing should skip 32768 (not below) and land on 24576.
|
|
const next = recordLearnedThinkingCap("gemini", "gemini-3.1-pro", 30000);
|
|
assert.equal(next, 24576);
|
|
});
|
|
|
|
// ── getLearnedThinkingCap key handling ─────────────────────────────────────
|
|
|
|
test("getLearnedThinkingCap returns null for unknown provider+model", () => {
|
|
assert.equal(getLearnedThinkingCap("gemini", "unknown-model"), null);
|
|
});
|
|
|
|
test("getLearnedThinkingCap is keyed case-insensitively on provider+model", () => {
|
|
recordLearnedThinkingCap("Gemini", "Gemini-2.5-Pro", 131072);
|
|
assert.equal(getLearnedThinkingCap("gemini", "gemini-2.5-pro"), 32768);
|
|
assert.equal(getLearnedThinkingCap("GEMINI", "GEMINI-2.5-PRO"), 32768);
|
|
});
|
|
|
|
test("different providers for the same model id have independent caps", () => {
|
|
recordLearnedThinkingCap("gemini", "gemini-2.5-pro", 131072);
|
|
assert.equal(getLearnedThinkingCap("openrouter", "gemini-2.5-pro"), null);
|
|
});
|
|
|
|
test("different models for the same provider have independent caps", () => {
|
|
recordLearnedThinkingCap("gemini", "gemini-2.5-pro", 131072);
|
|
assert.equal(getLearnedThinkingCap("gemini", "gemini-2.5-flash"), null);
|
|
});
|
|
|
|
test("handles empty/null provider or model gracefully", () => {
|
|
assert.equal(getLearnedThinkingCap("", "gemini-2.5-pro"), null);
|
|
assert.equal(getLearnedThinkingCap("gemini", ""), null);
|
|
assert.equal(getLearnedThinkingCap(null, "gemini-2.5-pro"), null);
|
|
assert.equal(getLearnedThinkingCap("gemini", null), null);
|
|
assert.equal(recordLearnedThinkingCap("", "gemini-2.5-pro", 131072), null);
|
|
assert.equal(recordLearnedThinkingCap("gemini", "", 131072), null);
|
|
});
|