Merge pull request #58 from modelcontextprotocol/justin/improve-abort

Add message to `abortAfterTimeout`, add missing request `options`
This commit is contained in:
Justin Spahr-Summers 2024-11-16 16:03:59 +00:00 • committed by GitHub
commit c32e057c72
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
2 changed files with 15 additions and 6 deletions

View file

@ -278,8 +278,8 @@ export class Client<
}
}
async ping() {
return this.request({ method: "ping" }, EmptyResultSchema);
async ping(options?: RequestOptions) {
return this.request({ method: "ping" }, EmptyResultSchema, options);
}
async complete(params: CompleteRequest["params"], options?: RequestOptions) {
@ -290,10 +290,11 @@ export class Client<
);
}
async setLoggingLevel(level: LoggingLevel) {
async setLoggingLevel(level: LoggingLevel, options?: RequestOptions) {
return this.request(
{ method: "logging/setLevel", params: { level } },
EmptyResultSchema,
options,
);
}
@ -352,17 +353,25 @@ export class Client<
);
}
async subscribeResource(params: SubscribeRequest["params"]) {
async subscribeResource(
params: SubscribeRequest["params"],
options?: RequestOptions,
) {
return this.request(
{ method: "resources/subscribe", params },
EmptyResultSchema,
options,
);
}
async unsubscribeResource(params: UnsubscribeRequest["params"]) {
async unsubscribeResource(
params: UnsubscribeRequest["params"],
options?: RequestOptions,
) {
return this.request(
{ method: "resources/unsubscribe", params },
EmptyResultSchema,
options,
);
}

View file

@ -4,7 +4,7 @@
export function abortAfterTimeout(timeoutMs: number): AbortSignal {
const controller = new AbortController();
setTimeout(() => {
controller.abort();
controller.abort("Timeout exceeded");
}, timeoutMs);
return controller.signal;