jiang
6a49d2209e
Move project and host queries into native MCP with host-owned OAuth/context, while retaining memory, skills, actions and publication plugins. Add nearby-model queries and explicit chat/task/learning model defaults. Adopt Responses HTTP/SSE and native steer/queue inbox delivery, preserving scoped session ownership, pending input edits and investigation communication. Disable local-provider discovery: Server /health allowed native vLLM to probe /v1/models on port 8000. A native before/after regression prevents recurrence. Validation: 170 focused Bun tests; service/plugin typechecks; native discovery, provider-discovery and input-inbox smoke checks; no paid model inference.
27 lines
1.6 KiB
TypeScript
27 lines
1.6 KiB
TypeScript
import { expect, it } from "bun:test";
|
|
import type { Tool } from "@modelcontextprotocol/client";
|
|
import { nativeQueryTimeout, queryTimeouts } from "../../src/mcp/query-timeouts.js";
|
|
|
|
const simulation: Tool = { name: "run_simulation", inputSchema: { type: "object", properties: {
|
|
timeout_seconds: { type: "integer", minimum: 1, maximum: 3600, default: 120 },
|
|
} } };
|
|
|
|
it("budgets calculations independently from their simulated duration and from ordinary reads", () => {
|
|
expect(queryTimeouts(simulation, { duration: 86400, timeout_seconds: 900 })).toEqual({ executionMs: 900_000, callMs: 905_000, requestMs: 960_000 });
|
|
expect(queryTimeouts(simulation, {})).toEqual({ executionMs: 120_000, callMs: 125_000, requestMs: 180_000 });
|
|
expect(queryTimeouts({ name: "list_elements", inputSchema: { type: "object" } }, {})).toEqual({ executionMs: 30_000, callMs: 30_000, requestMs: 35_000 });
|
|
});
|
|
|
|
it("honors the effective advertised maximum and rejects invalid per-call budgets", () => {
|
|
for (const timeout_seconds of [0, -1, 1.5, 3601, "900", null, true, NaN, Infinity]) {
|
|
expect(() => queryTimeouts(simulation, { timeout_seconds })).toThrow();
|
|
}
|
|
const limited = { ...simulation, inputSchema: { type: "object" as const, properties: {
|
|
timeout_seconds: { type: "integer", maximum: 120, default: 120 },
|
|
} } };
|
|
expect(queryTimeouts(limited, { timeout_seconds: 120 }).executionMs).toBe(120_000);
|
|
expect(() => queryTimeouts(limited, { timeout_seconds: 121 })).toThrow();
|
|
expect(nativeQueryTimeout([simulation, limited])).toBe(3_665_000);
|
|
expect(nativeQueryTimeout([limited])).toBe(185_000);
|
|
});
|