Skip to content

Commit 733dce2

Browse files
committed
fix: update mcp.ts and protocol.ts after rebase - use getMethodLiteral, add createToolError
1 parent 2fce53c commit 733dce2

File tree

4 files changed

+174
-85
lines changed

4 files changed

+174
-85
lines changed

package-lock.json

Lines changed: 9 additions & 15 deletions
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.

src/server/mcp.ts

Lines changed: 27 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -45,7 +45,7 @@ import {
4545
} from '../types.js';
4646

4747
import { UriTemplate, Variables } from '../shared/uriTemplate.js';
48-
import { RequestHandlerExtra } from '../shared/protocol.js';
48+
import { RequestHandlerExtra, getMethodLiteral } from '../shared/protocol.js';
4949
import { Transport } from '../shared/transport.js';
5050

5151
import { toJsonSchemaCompat } from './zod-json-schema-compat.js';
@@ -96,8 +96,8 @@ export class McpServer {
9696
return;
9797
}
9898

99-
this.server.assertCanSetRequestHandler(ListToolsRequestSchema.shape.method.value);
100-
this.server.assertCanSetRequestHandler(CallToolRequestSchema.shape.method.value);
99+
this.server.assertCanSetRequestHandler(getMethodLiteral(ListToolsRequestSchema));
100+
this.server.assertCanSetRequestHandler(getMethodLiteral(CallToolRequestSchema));
101101

102102
this.server.registerCapabilities({
103103
tools: {
@@ -207,14 +207,32 @@ export class McpServer {
207207
this._toolHandlersInitialized = true;
208208
}
209209

210+
/**
211+
* Creates a tool error result.
212+
*
213+
* @param errorMessage - The error message.
214+
* @returns The tool error result.
215+
*/
216+
private createToolError(errorMessage: string): CallToolResult {
217+
return {
218+
content: [
219+
{
220+
type: 'text',
221+
text: errorMessage
222+
}
223+
],
224+
isError: true
225+
};
226+
}
227+
210228
private _completionHandlerInitialized = false;
211229

212230
private setCompletionRequestHandler() {
213231
if (this._completionHandlerInitialized) {
214232
return;
215233
}
216234

217-
this.server.assertCanSetRequestHandler(CompleteRequestSchema.shape.method.value);
235+
this.server.assertCanSetRequestHandler(getMethodLiteral(CompleteRequestSchema));
218236

219237
this.server.registerCapabilities({
220238
completions: {}
@@ -301,9 +319,9 @@ export class McpServer {
301319
return;
302320
}
303321

304-
this.server.assertCanSetRequestHandler(ListResourcesRequestSchema.shape.method.value);
305-
this.server.assertCanSetRequestHandler(ListResourceTemplatesRequestSchema.shape.method.value);
306-
this.server.assertCanSetRequestHandler(ReadResourceRequestSchema.shape.method.value);
322+
this.server.assertCanSetRequestHandler(getMethodLiteral(ListResourcesRequestSchema));
323+
this.server.assertCanSetRequestHandler(getMethodLiteral(ListResourceTemplatesRequestSchema));
324+
this.server.assertCanSetRequestHandler(getMethodLiteral(ReadResourceRequestSchema));
307325

308326
this.server.registerCapabilities({
309327
resources: {
@@ -384,8 +402,8 @@ export class McpServer {
384402
return;
385403
}
386404

387-
this.server.assertCanSetRequestHandler(ListPromptsRequestSchema.shape.method.value);
388-
this.server.assertCanSetRequestHandler(GetPromptRequestSchema.shape.method.value);
405+
this.server.assertCanSetRequestHandler(getMethodLiteral(ListPromptsRequestSchema));
406+
this.server.assertCanSetRequestHandler(getMethodLiteral(GetPromptRequestSchema));
389407

390408
this.server.registerCapabilities({
391409
prompts: {

0 commit comments

Comments
 (0)