expect(opts.req.method).toBe("sessions.get");

expect(opts.req.params).toEqual({ key: "s-legacy" });

opts.respond(true, { messages: [{ id: "m-2" }] });

})

.mockImplementationOnce(async (opts: HandleGatewayRequestOptions) => {

expect(opts.req.method).toBe("sessions.get");

expect(opts.req.params).toEqual({ key: "s-limited", limit: 1_000 });

opts.respond(true, { messages: [{ id: "m-3" }] });

});

await expect(runtime.getSessionMessages({ sessionKey: "s-read" })).resolves.toEqual({

await expect(runtime.getSession({ sessionKey: "s-legacy" })).resolves.toEqual({

messages: [{ id: "m-2" }],

});

await expect(

runtime.getSessionMessages({

sessionKey: "s-limited",

limit: 9e15,

}),

).resolves.toEqual({

messages: [{ id: "m-3" }],

});

});

test("times out while waiting for the first in-process gateway response", async () => {

return await dispatchGatewayMethod<T>(method, params, options);

}

const PLUGIN_SUBAGENT_SESSION_MESSAGES_MAX_LIMIT = 1_000;

export function createGatewaySubagentRuntime(): PluginRuntime["subagent"] {

const getSessionMessages: PluginRuntime["subagent"]["getSessionMessages"] = async (params) => {

const limit =

params.limit == null || !Number.isFinite(params.limit)

? undefined

: Math.min(

PLUGIN_SUBAGENT_SESSION_MESSAGES_MAX_LIMIT,

Math.max(1, Math.floor(params.limit)),

);

const payload = await dispatchGatewayMethod<{ messages?: unknown[] }>("sessions.get", {

key: params.sessionKey,

...(params.limit != null && { limit: params.limit }),

...(limit != null && { limit }),

});

return { messages: Array.isArray(payload?.messages) ? payload.messages : [] };

};