31 lines
851 B
JavaScript
31 lines
851 B
JavaScript
import { Client } from "@modelcontextprotocol/client";
|
|
import { StdioClientTransport } from "@modelcontextprotocol/client/stdio";
|
|
|
|
/** Own one SDK client with an explicit receive budget for a large complete result. */
|
|
export const withLargeResultMcp = async (
|
|
{ entrypoint, environment, maxBufferSize },
|
|
useClient,
|
|
) => {
|
|
const client = new Client({ name: "large-result-verifier", version: "1" });
|
|
const transport = new StdioClientTransport({
|
|
command: process.execPath,
|
|
args: [entrypoint, "mcp"],
|
|
env: {
|
|
...environment,
|
|
REA_MCP_MAX_RESPONSE_BYTES: String(maxBufferSize),
|
|
},
|
|
stderr: "inherit",
|
|
maxBufferSize,
|
|
});
|
|
try {
|
|
await client.connect(transport);
|
|
return await useClient(client);
|
|
} finally {
|
|
try {
|
|
await client.close();
|
|
} finally {
|
|
await transport.close();
|
|
}
|
|
}
|
|
};
|