This commit is contained in:
@@ -11,7 +11,9 @@ import { Server } from "@modelcontextprotocol/sdk/server/index.js";
|
||||
import { StdioServerTransport } from "@modelcontextprotocol/sdk/server/stdio.js";
|
||||
import {
|
||||
CallToolRequestSchema,
|
||||
ElicitResultSchema,
|
||||
ListToolsRequestSchema,
|
||||
ResultSchema,
|
||||
} from "@modelcontextprotocol/sdk/types.js";
|
||||
|
||||
import { loadConfig, type EnabledAgent } from "./config.js";
|
||||
@@ -136,6 +138,18 @@ export async function buildGateway(opts: GatewayOptions = {}): Promise<GatewayHa
|
||||
// best effort
|
||||
}
|
||||
},
|
||||
onRequest: async (upstreamReq) => {
|
||||
const schema = upstreamReq.method === "elicitation/create"
|
||||
? ElicitResultSchema
|
||||
: ResultSchema;
|
||||
return extra.sendRequest(
|
||||
{
|
||||
method: upstreamReq.method,
|
||||
params: upstreamReq.params,
|
||||
} as any,
|
||||
schema as any,
|
||||
);
|
||||
},
|
||||
},
|
||||
);
|
||||
// Pass through the upstream's CallToolResult as-is. Upstreams already
|
||||
|
||||
Reference in New Issue
Block a user