WIP shttp changes
This commit is contained in:
parent
8579d446aa
commit
c2119ed9f9
1 changed files with 88 additions and 58 deletions
|
|
@ -15,13 +15,16 @@ import {
|
|||
SetLevelRequestSchema,
|
||||
SubscribeRequestSchema,
|
||||
Tool,
|
||||
ToolSchema,
|
||||
UnsubscribeRequestSchema,
|
||||
} from "@modelcontextprotocol/sdk/types.js";
|
||||
import { z } from "zod";
|
||||
import { zodToJsonSchema } from "zod-to-json-schema";
|
||||
|
||||
type ToolInput = z.infer<typeof ToolSchema.shape.inputSchema>;
|
||||
type ToolInput = {
|
||||
type: "object";
|
||||
properties?: Record<string, unknown>;
|
||||
required?: string[];
|
||||
};
|
||||
|
||||
/* Input schemas for tools implemented in this server */
|
||||
const EchoSchema = z.object({
|
||||
|
|
@ -109,18 +112,6 @@ export const createMcpServer = () => {
|
|||
}
|
||||
);
|
||||
|
||||
// Add logging to track all incoming messages
|
||||
const originalSetRequestHandler = server.setRequestHandler.bind(server);
|
||||
server.setRequestHandler = function(schema: any, handler: any) {
|
||||
const wrappedHandler = async (request: any) => {
|
||||
console.log(`[MCP Server] Handling request: ${request.method}`);
|
||||
const result = await handler(request);
|
||||
console.log(`[MCP Server] Request ${request.method} completed`);
|
||||
return result;
|
||||
};
|
||||
return originalSetRequestHandler(schema, wrappedHandler);
|
||||
};
|
||||
|
||||
const subscriptions: Set<string> = new Set();
|
||||
|
||||
// Set up update interval for subscribed resources
|
||||
|
|
|
|||
Loading…
Reference in a new issue