88 lines
2.7 KiB
TypeScript
88 lines
2.7 KiB
TypeScript
import { describe, expect, it, vi } from "vitest";
|
|
import { createSessionsSpawnTool } from "./tools/sessions-spawn-tool.js";
|
|
|
|
vi.mock("../config/config.js", async () => {
|
|
const actual = await vi.importActual("../config/config.js");
|
|
return {
|
|
...actual,
|
|
loadConfig: () => ({
|
|
agents: {
|
|
defaults: {
|
|
subagents: {
|
|
thinking: "high",
|
|
},
|
|
},
|
|
},
|
|
routing: {
|
|
sessions: {
|
|
mainKey: "agent:test:main",
|
|
},
|
|
},
|
|
}),
|
|
};
|
|
});
|
|
|
|
vi.mock("../gateway/call.js", () => {
|
|
return {
|
|
callGateway: vi.fn(async ({ method }: { method: string }) => {
|
|
if (method === "agent") {
|
|
return { runId: "run-123" };
|
|
}
|
|
return {};
|
|
}),
|
|
};
|
|
});
|
|
|
|
type GatewayCall = { method: string; params?: Record<string, unknown> };
|
|
|
|
async function getGatewayCalls(): Promise<GatewayCall[]> {
|
|
const { callGateway } = await import("../gateway/call.js");
|
|
return (callGateway as unknown as ReturnType<typeof vi.fn>).mock.calls.map(
|
|
(call) => call[0] as GatewayCall,
|
|
);
|
|
}
|
|
|
|
function findLastCall(calls: GatewayCall[], predicate: (call: GatewayCall) => boolean) {
|
|
for (let i = calls.length - 1; i >= 0; i -= 1) {
|
|
const call = calls[i];
|
|
if (call && predicate(call)) {
|
|
return call;
|
|
}
|
|
}
|
|
return undefined;
|
|
}
|
|
|
|
describe("sessions_spawn thinking defaults", () => {
|
|
it("applies agents.defaults.subagents.thinking when thinking is omitted", async () => {
|
|
const tool = createSessionsSpawnTool({ agentSessionKey: "agent:test:main" });
|
|
const result = await tool.execute("call-1", { task: "hello" });
|
|
expect(result.details).toMatchObject({ status: "accepted" });
|
|
|
|
const calls = await getGatewayCalls();
|
|
const agentCall = findLastCall(calls, (call) => call.method === "agent");
|
|
const thinkingPatch = findLastCall(
|
|
calls,
|
|
(call) => call.method === "sessions.patch" && call.params?.thinkingLevel !== undefined,
|
|
);
|
|
|
|
expect(agentCall?.params?.thinking).toBe("high");
|
|
expect(thinkingPatch?.params?.thinkingLevel).toBe("high");
|
|
});
|
|
|
|
it("prefers explicit sessions_spawn.thinking over config default", async () => {
|
|
const tool = createSessionsSpawnTool({ agentSessionKey: "agent:test:main" });
|
|
const result = await tool.execute("call-2", { task: "hello", thinking: "low" });
|
|
expect(result.details).toMatchObject({ status: "accepted" });
|
|
|
|
const calls = await getGatewayCalls();
|
|
const agentCall = findLastCall(calls, (call) => call.method === "agent");
|
|
const thinkingPatch = findLastCall(
|
|
calls,
|
|
(call) => call.method === "sessions.patch" && call.params?.thinkingLevel !== undefined,
|
|
);
|
|
|
|
expect(agentCall?.params?.thinking).toBe("low");
|
|
expect(thinkingPatch?.params?.thinkingLevel).toBe("low");
|
|
});
|
|
});
|