mirror of
https://github.com/diegosouzapw/OmniRoute.git
synced 2026-07-31 04:12:10 +03:00
a95da4a902
Phases 3-4 of #3384 (Phases 1-2 shipped DB schema + resolveInterceptSearch in release/v3.8.47). Adds resolveInterceptFetch(provider, model) as a structural twin of resolveInterceptSearch, and open-sse/services/webFetchInterception.ts (mirroring webSearchFallback.ts) to rewrite a provider-native web_fetch tool declaration into a synthetic omniroute_web_fetch function tool. The synthetic tool call is dispatched through the existing handleToolCallExecution path (same as omniroute_web_search) to a new web_fetch builtin skill handler that resolves credentials and calls handleWebFetch() against /v1/web/fetch. Strictly opt-in: with no interceptFetch DB row configured (the default), the outgoing request body is byte-identical to pre-change behavior — no heuristic default bypass like the interceptSearch sibling, to guarantee zero overhead when disabled (Hard Rule #20). Also ships the dashboard toggle (owner decision, overriding the plan's backend-only recommendation): ProviderInterceptionSection.tsx on the provider detail page, backed by GET/PUT/DELETE /api/providers/[id]/interception-rules, covering both interceptSearch and interceptFetch from one control. chatCore.ts touch is minimal (frozen file): one resolver call + one prepareWebFetchFallbackBody call mirroring the existing interceptSearch block, plus threading provider/model into the existing handleToolCallExecution call.
64 lines
2.5 KiB
TypeScript
64 lines
2.5 KiB
TypeScript
import { describe, it, beforeEach, after } from "node:test";
|
|
import assert from "node:assert/strict";
|
|
import os from "node:os";
|
|
import path from "node:path";
|
|
import fs from "node:fs";
|
|
|
|
// Set DATA_DIR to a temp dir before any imports that touch the DB.
|
|
const tmpDir = fs.mkdtempSync(path.join(os.tmpdir(), "omniroute-test-intercept-fetch-resolver-"));
|
|
process.env.DATA_DIR = tmpDir;
|
|
|
|
const core = await import("../../src/lib/db/core.ts");
|
|
const { setInterceptionRules, resolveInterceptFetch } = await import(
|
|
"../../src/lib/db/interceptionRules.ts"
|
|
);
|
|
|
|
// #7339 — resolveInterceptFetch, a structural twin of resolveInterceptSearch
|
|
// (tests/unit/interception-rules.test.ts), covering Phase 3 of #3384.
|
|
describe("db/interceptionRules — resolveInterceptFetch precedence (#7339)", () => {
|
|
function resetDb() {
|
|
core.resetDbInstance();
|
|
fs.rmSync(tmpDir, { recursive: true, force: true });
|
|
fs.mkdirSync(tmpDir, { recursive: true });
|
|
}
|
|
|
|
beforeEach(() => {
|
|
resetDb();
|
|
});
|
|
|
|
after(() => {
|
|
core.resetDbInstance();
|
|
fs.rmSync(tmpDir, { recursive: true, force: true });
|
|
});
|
|
|
|
it("returns undefined when no provider/model rule exists", () => {
|
|
assert.equal(resolveInterceptFetch("anthropic", "claude-opus-4"), undefined);
|
|
});
|
|
|
|
it("returns the provider-level interceptFetch value when only a provider rule is set", () => {
|
|
setInterceptionRules("anthropic", { interceptFetch: true });
|
|
assert.equal(resolveInterceptFetch("anthropic", "claude-opus-4"), true);
|
|
assert.equal(resolveInterceptFetch("anthropic", "claude-haiku-4"), true);
|
|
});
|
|
|
|
it("model-level interceptFetch wins over the provider-level rule when both are set", () => {
|
|
setInterceptionRules("anthropic", {
|
|
interceptFetch: false,
|
|
models: { "claude-opus-4": { interceptFetch: true } },
|
|
});
|
|
assert.equal(resolveInterceptFetch("anthropic", "claude-opus-4"), true);
|
|
assert.equal(resolveInterceptFetch("anthropic", "claude-haiku-4"), false);
|
|
});
|
|
|
|
it("does not read interceptSearch when resolving interceptFetch (fields stay independent)", () => {
|
|
setInterceptionRules("anthropic", { interceptSearch: true, interceptFetch: false });
|
|
assert.equal(resolveInterceptFetch("anthropic", "claude-opus-4"), false);
|
|
});
|
|
|
|
it("returns undefined for an empty/missing provider", () => {
|
|
assert.equal(resolveInterceptFetch("", "claude-opus-4"), undefined);
|
|
assert.equal(resolveInterceptFetch(null, "claude-opus-4"), undefined);
|
|
assert.equal(resolveInterceptFetch(undefined, "claude-opus-4"), undefined);
|
|
});
|
|
});
|