tmcp 1.7.0 → 1.7.1

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "tmcp",
-  "version": "1.7.0",
+  "version": "1.7.1",
   "description": "The main tmcp library",
   "main": "src/index.js",
   "type": "module",
@@ -36,10 +36,11 @@
     "@standard-schema/spec": "^1.0.0",
     "@types/json-schema": "^7.0.15",
     "@types/node": "^24.0.13",
-    "publint": "^0.3.12"
+    "publint": "^0.3.12",
+    "vitest": "^3.2.4"
   },
   "scripts": {
-    "test": "echo \"Error: no test specified\" && exit 1",
+    "test": "vitest",
     "typecheck": "tsc --noEmit",
     "generate:types": "find ./src -name \"*.d.ts\" -not -path \"./src/internal/*\" -delete && tsc ./src/*.js --declaration --emitDeclarationOnly --target esnext --allowJs --moduleresolution nodenext --module nodenext && publint",
     "prepublish": "pnpm generate:types"

package/src/index.d.ts

@@ -41,7 +41,7 @@ export class McpServer<StandardSchema extends StandardSchemaV1> {
      * @param {McpEvents[TEvent]} callback
      * @param {AddEventListenerOptions} [options]
      */
-    on<TEvent extends keyof McpEvents>(event: TEvent, callback: McpEvents[TEvent], options?: AddEventListenerOptions): void;
+    on<TEvent extends keyof McpEvents>(event: TEvent, callback: McpEvents[TEvent], options?: AddEventListenerOptions): () => void;
     /**
      * @template {StandardSchema | undefined} [TSchema=undefined]
      * @param {{ name: string; description: string; title?: string; schema?: StandardSchemaV1.InferInput<TSchema extends undefined ? never : TSchema> extends Record<string, unknown> ? TSchema : never; annotations?: ToolAnnotations }} options

package/src/index.js

@@ -270,13 +270,18 @@ export class McpServer {
 			this.#lazyily_create_client();
 		}
 
-		this.#event_target.addEventListener(
-			event,
-			(e) => {
-				callback(/** @type {CustomEvent} */ (e).detail);
-			},
-			options,
-		);
+		/**
+		 * @param {Event} e
+		 */
+		const listener = (e) => {
+			callback(/** @type {CustomEvent} */ (e).detail);
+		};
+
+		this.#event_target.addEventListener(event, listener, options);
+
+		return () => {
+			this.#event_target.removeEventListener(event, listener, options);
+		};
 	}
 
 	/**