@a2a-js/sdk 0.3.6 → 0.3.7

package/dist/server/index.cjs

@@ -54,16 +54,194 @@ var RequestContext = class {
 };
 
 // src/server/events/execution_event_bus.ts
-var import_events = require("events");
-var DefaultExecutionEventBus = class extends import_events.EventEmitter {
-  constructor() {
-    super();
+var CustomEventImpl = typeof CustomEvent !== "undefined" ? CustomEvent : class CustomEventPolyfill extends Event {
+  detail;
+  constructor(type, eventInitDict) {
+    super(type, eventInitDict);
+    this.detail = eventInitDict?.detail ?? null;
   }
+};
+function isAgentExecutionCustomEvent(e) {
+  return e instanceof CustomEventImpl;
+}
+var DefaultExecutionEventBus = class extends EventTarget {
+  // Separate storage for each event type - both use the interface's Listener type
+  // but are invoked differently (with event payload vs. no arguments)
+  eventListeners = /* @__PURE__ */ new Map();
+  finishedListeners = /* @__PURE__ */ new Map();
   publish(event) {
-    this.emit("event", event);
+    this.dispatchEvent(new CustomEventImpl("event", { detail: event }));
   }
   finished() {
-    this.emit("finished");
+    this.dispatchEvent(new Event("finished"));
+  }
+  /**
+   * EventEmitter-compatible 'on' method.
+   * Wraps the listener to extract event detail from CustomEvent.
+   * Supports multiple registrations of the same listener (like EventEmitter).
+   * @param eventName The event name to listen for.
+   * @param listener The callback function to invoke when the event is emitted.
+   * @returns This instance for method chaining.
+   */
+  on(eventName, listener) {
+    if (eventName === "event") {
+      this.addEventListenerInternal(listener);
+    } else {
+      this.addFinishedListenerInternal(listener);
+    }
+    return this;
+  }
+  /**
+   * EventEmitter-compatible 'off' method.
+   * Uses the stored wrapped listener for proper removal.
+   * Removes at most one instance of a listener per call (like EventEmitter).
+   * @param eventName The event name to stop listening for.
+   * @param listener The callback function to remove.
+   * @returns This instance for method chaining.
+   */
+  off(eventName, listener) {
+    if (eventName === "event") {
+      this.removeEventListenerInternal(listener);
+    } else {
+      this.removeFinishedListenerInternal(listener);
+    }
+    return this;
+  }
+  /**
+   * EventEmitter-compatible 'once' method.
+   * Listener is automatically removed after first invocation.
+   * Supports multiple registrations of the same listener (like EventEmitter).
+   * @param eventName The event name to listen for once.
+   * @param listener The callback function to invoke when the event is emitted.
+   * @returns This instance for method chaining.
+   */
+  once(eventName, listener) {
+    if (eventName === "event") {
+      this.addEventListenerOnceInternal(listener);
+    } else {
+      this.addFinishedListenerOnceInternal(listener);
+    }
+    return this;
+  }
+  /**
+   * EventEmitter-compatible 'removeAllListeners' method.
+   * Removes all listeners for a specific event or all events.
+   * @param eventName Optional event name to remove listeners for. If omitted, removes all.
+   * @returns This instance for method chaining.
+   */
+  removeAllListeners(eventName) {
+    if (eventName === void 0 || eventName === "event") {
+      for (const wrappedListeners of this.eventListeners.values()) {
+        for (const wrapped of wrappedListeners) {
+          this.removeEventListener("event", wrapped);
+        }
+      }
+      this.eventListeners.clear();
+    }
+    if (eventName === void 0 || eventName === "finished") {
+      for (const wrappedListeners of this.finishedListeners.values()) {
+        for (const wrapped of wrappedListeners) {
+          this.removeEventListener("finished", wrapped);
+        }
+      }
+      this.finishedListeners.clear();
+    }
+    return this;
+  }
+  // ========================
+  // Helper methods for listener tracking
+  // ========================
+  /**
+   * Adds a wrapped listener to the tracking map.
+   */
+  trackListener(listenerMap, listener, wrapped) {
+    const existing = listenerMap.get(listener);
+    if (existing) {
+      existing.push(wrapped);
+    } else {
+      listenerMap.set(listener, [wrapped]);
+    }
+  }
+  /**
+   * Removes a wrapped listener from the tracking map (for once cleanup).
+   */
+  untrackWrappedListener(listenerMap, listener, wrapped) {
+    const wrappedList = listenerMap.get(listener);
+    if (wrappedList && wrappedList.length > 0) {
+      const index = wrappedList.indexOf(wrapped);
+      if (index !== -1) {
+        wrappedList.splice(index, 1);
+        if (wrappedList.length === 0) {
+          listenerMap.delete(listener);
+        }
+      }
+    }
+  }
+  // ========================
+  // Internal methods for 'event' listeners
+  // ========================
+  addEventListenerInternal(listener) {
+    const wrapped = (e) => {
+      if (!isAgentExecutionCustomEvent(e)) {
+        throw new Error('Internal error: expected CustomEvent for "event" type');
+      }
+      listener.call(this, e.detail);
+    };
+    this.trackListener(this.eventListeners, listener, wrapped);
+    this.addEventListener("event", wrapped);
+  }
+  removeEventListenerInternal(listener) {
+    const wrappedList = this.eventListeners.get(listener);
+    if (wrappedList && wrappedList.length > 0) {
+      const wrapped = wrappedList.pop();
+      if (wrappedList.length === 0) {
+        this.eventListeners.delete(listener);
+      }
+      this.removeEventListener("event", wrapped);
+    }
+  }
+  addEventListenerOnceInternal(listener) {
+    const wrapped = (e) => {
+      if (!isAgentExecutionCustomEvent(e)) {
+        throw new Error('Internal error: expected CustomEvent for "event" type');
+      }
+      this.untrackWrappedListener(this.eventListeners, listener, wrapped);
+      listener.call(this, e.detail);
+    };
+    this.trackListener(this.eventListeners, listener, wrapped);
+    this.addEventListener("event", wrapped, { once: true });
+  }
+  // ========================
+  // Internal methods for 'finished' listeners
+  // ========================
+  // The interface declares listeners as (event: AgentExecutionEvent) => void,
+  // but for 'finished' events they are invoked with no arguments (EventEmitter behavior).
+  // We use Function.prototype.call to invoke with `this` as the event bus (matching
+  // EventEmitter semantics) and no arguments, which is type-safe.
+  addFinishedListenerInternal(listener) {
+    const wrapped = () => {
+      listener.call(this);
+    };
+    this.trackListener(this.finishedListeners, listener, wrapped);
+    this.addEventListener("finished", wrapped);
+  }
+  removeFinishedListenerInternal(listener) {
+    const wrappedList = this.finishedListeners.get(listener);
+    if (wrappedList && wrappedList.length > 0) {
+      const wrapped = wrappedList.pop();
+      if (wrappedList.length === 0) {
+        this.finishedListeners.delete(listener);
+      }
+      this.removeEventListener("finished", wrapped);
+    }
+  }
+  addFinishedListenerOnceInternal(listener) {
+    const wrapped = () => {
+      this.untrackWrappedListener(this.finishedListeners, listener, wrapped);
+      listener.call(this);
+    };
+    this.trackListener(this.finishedListeners, listener, wrapped);
+    this.addEventListener("finished", wrapped, { once: true });
   }
 };
 

package/dist/server/index.d.cts

@@ -1,21 +1,81 @@
-import { EventEmitter } from 'events';
 import { F as Message, ay as Task, aQ as TaskStatusUpdateEvent, aS as TaskArtifactUpdateEvent, z as PushNotificationConfig, ae as AgentCard, x as MessageSendParams, X as TaskQueryParams, Z as TaskIdParams, $ as TaskPushNotificationConfig, a3 as GetTaskPushNotificationConfigParams, a7 as ListTaskPushNotificationConfigParams, a9 as DeleteTaskPushNotificationConfigParams, j as JSONRPCResponse, aw as JSONRPCError } from '../extensions-DvruCIzw.cjs';
 import { S as ServerCallContext, A as A2ARequestHandler } from '../a2a_request_handler-B3LxMq3P.cjs';
 export { a as UnauthenticatedUser, U as User } from '../a2a_request_handler-B3LxMq3P.cjs';
 
 type AgentExecutionEvent = Message | Task | TaskStatusUpdateEvent | TaskArtifactUpdateEvent;
+/**
+ * Event names supported by ExecutionEventBus.
+ */
+type ExecutionEventName = 'event' | 'finished';
 interface ExecutionEventBus {
     publish(event: AgentExecutionEvent): void;
-    on(eventName: 'event' | 'finished', listener: (event: AgentExecutionEvent) => void): this;
-    off(eventName: 'event' | 'finished', listener: (event: AgentExecutionEvent) => void): this;
-    once(eventName: 'event' | 'finished', listener: (event: AgentExecutionEvent) => void): this;
-    removeAllListeners(eventName?: 'event' | 'finished'): this;
+    on(eventName: ExecutionEventName, listener: (event: AgentExecutionEvent) => void): this;
+    off(eventName: ExecutionEventName, listener: (event: AgentExecutionEvent) => void): this;
+    once(eventName: ExecutionEventName, listener: (event: AgentExecutionEvent) => void): this;
+    removeAllListeners(eventName?: ExecutionEventName): this;
     finished(): void;
 }
-declare class DefaultExecutionEventBus extends EventEmitter implements ExecutionEventBus {
-    constructor();
+/**
+ * Web-compatible ExecutionEventBus using EventTarget.
+ * Works across all modern runtimes: Node.js 15+, browsers, Cloudflare Workers, Deno, Bun.
+ *
+ * This implementation provides the subset of EventEmitter methods defined in the
+ * ExecutionEventBus interface. Users extending DefaultExecutionEventBus should note
+ * that other EventEmitter methods (e.g., listenerCount, rawListeners) are not available.
+ */
+declare class DefaultExecutionEventBus extends EventTarget implements ExecutionEventBus {
+    private readonly eventListeners;
+    private readonly finishedListeners;
     publish(event: AgentExecutionEvent): void;
     finished(): void;
+    /**
+     * EventEmitter-compatible 'on' method.
+     * Wraps the listener to extract event detail from CustomEvent.
+     * Supports multiple registrations of the same listener (like EventEmitter).
+     * @param eventName The event name to listen for.
+     * @param listener The callback function to invoke when the event is emitted.
+     * @returns This instance for method chaining.
+     */
+    on(eventName: ExecutionEventName, listener: (event: AgentExecutionEvent) => void): this;
+    /**
+     * EventEmitter-compatible 'off' method.
+     * Uses the stored wrapped listener for proper removal.
+     * Removes at most one instance of a listener per call (like EventEmitter).
+     * @param eventName The event name to stop listening for.
+     * @param listener The callback function to remove.
+     * @returns This instance for method chaining.
+     */
+    off(eventName: ExecutionEventName, listener: (event: AgentExecutionEvent) => void): this;
+    /**
+     * EventEmitter-compatible 'once' method.
+     * Listener is automatically removed after first invocation.
+     * Supports multiple registrations of the same listener (like EventEmitter).
+     * @param eventName The event name to listen for once.
+     * @param listener The callback function to invoke when the event is emitted.
+     * @returns This instance for method chaining.
+     */
+    once(eventName: ExecutionEventName, listener: (event: AgentExecutionEvent) => void): this;
+    /**
+     * EventEmitter-compatible 'removeAllListeners' method.
+     * Removes all listeners for a specific event or all events.
+     * @param eventName Optional event name to remove listeners for. If omitted, removes all.
+     * @returns This instance for method chaining.
+     */
+    removeAllListeners(eventName?: ExecutionEventName): this;
+    /**
+     * Adds a wrapped listener to the tracking map.
+     */
+    private trackListener;
+    /**
+     * Removes a wrapped listener from the tracking map (for once cleanup).
+     */
+    private untrackWrappedListener;
+    private addEventListenerInternal;
+    private removeEventListenerInternal;
+    private addEventListenerOnceInternal;
+    private addFinishedListenerInternal;
+    private removeFinishedListenerInternal;
+    private addFinishedListenerOnceInternal;
 }
 
 declare class RequestContext {
@@ -253,4 +313,4 @@ declare class DefaultPushNotificationSender implements PushNotificationSender {
     private _dispatchNotification;
 }
 
-export { A2AError, A2ARequestHandler, type AgentExecutionEvent, type AgentExecutor, DefaultExecutionEventBus, DefaultExecutionEventBusManager, DefaultPushNotificationSender, type DefaultPushNotificationSenderOptions, DefaultRequestHandler, type ExecutionEventBus, type ExecutionEventBusManager, ExecutionEventQueue, type ExtendedAgentCardProvider, InMemoryPushNotificationStore, InMemoryTaskStore, JsonRpcTransportHandler, type PushNotificationSender, type PushNotificationStore, RequestContext, ResultManager, ServerCallContext, type TaskStore };
+export { A2AError, A2ARequestHandler, type AgentExecutionEvent, type AgentExecutor, DefaultExecutionEventBus, DefaultExecutionEventBusManager, DefaultPushNotificationSender, type DefaultPushNotificationSenderOptions, DefaultRequestHandler, type ExecutionEventBus, type ExecutionEventBusManager, type ExecutionEventName, ExecutionEventQueue, type ExtendedAgentCardProvider, InMemoryPushNotificationStore, InMemoryTaskStore, JsonRpcTransportHandler, type PushNotificationSender, type PushNotificationStore, RequestContext, ResultManager, ServerCallContext, type TaskStore };

package/dist/server/index.d.ts

@@ -1,21 +1,81 @@
-import { EventEmitter } from 'events';
 import { F as Message, ay as Task, aQ as TaskStatusUpdateEvent, aS as TaskArtifactUpdateEvent, z as PushNotificationConfig, ae as AgentCard, x as MessageSendParams, X as TaskQueryParams, Z as TaskIdParams, $ as TaskPushNotificationConfig, a3 as GetTaskPushNotificationConfigParams, a7 as ListTaskPushNotificationConfigParams, a9 as DeleteTaskPushNotificationConfigParams, j as JSONRPCResponse, aw as JSONRPCError } from '../extensions-DvruCIzw.js';
 import { S as ServerCallContext, A as A2ARequestHandler } from '../a2a_request_handler-BuP9LgXH.js';
 export { a as UnauthenticatedUser, U as User } from '../a2a_request_handler-BuP9LgXH.js';
 
 type AgentExecutionEvent = Message | Task | TaskStatusUpdateEvent | TaskArtifactUpdateEvent;
+/**
+ * Event names supported by ExecutionEventBus.
+ */
+type ExecutionEventName = 'event' | 'finished';
 interface ExecutionEventBus {
     publish(event: AgentExecutionEvent): void;
-    on(eventName: 'event' | 'finished', listener: (event: AgentExecutionEvent) => void): this;
-    off(eventName: 'event' | 'finished', listener: (event: AgentExecutionEvent) => void): this;
-    once(eventName: 'event' | 'finished', listener: (event: AgentExecutionEvent) => void): this;
-    removeAllListeners(eventName?: 'event' | 'finished'): this;
+    on(eventName: ExecutionEventName, listener: (event: AgentExecutionEvent) => void): this;
+    off(eventName: ExecutionEventName, listener: (event: AgentExecutionEvent) => void): this;
+    once(eventName: ExecutionEventName, listener: (event: AgentExecutionEvent) => void): this;
+    removeAllListeners(eventName?: ExecutionEventName): this;
     finished(): void;
 }
-declare class DefaultExecutionEventBus extends EventEmitter implements ExecutionEventBus {
-    constructor();
+/**
+ * Web-compatible ExecutionEventBus using EventTarget.
+ * Works across all modern runtimes: Node.js 15+, browsers, Cloudflare Workers, Deno, Bun.
+ *
+ * This implementation provides the subset of EventEmitter methods defined in the
+ * ExecutionEventBus interface. Users extending DefaultExecutionEventBus should note
+ * that other EventEmitter methods (e.g., listenerCount, rawListeners) are not available.
+ */
+declare class DefaultExecutionEventBus extends EventTarget implements ExecutionEventBus {
+    private readonly eventListeners;
+    private readonly finishedListeners;
     publish(event: AgentExecutionEvent): void;
     finished(): void;
+    /**
+     * EventEmitter-compatible 'on' method.
+     * Wraps the listener to extract event detail from CustomEvent.
+     * Supports multiple registrations of the same listener (like EventEmitter).
+     * @param eventName The event name to listen for.
+     * @param listener The callback function to invoke when the event is emitted.
+     * @returns This instance for method chaining.
+     */
+    on(eventName: ExecutionEventName, listener: (event: AgentExecutionEvent) => void): this;
+    /**
+     * EventEmitter-compatible 'off' method.
+     * Uses the stored wrapped listener for proper removal.
+     * Removes at most one instance of a listener per call (like EventEmitter).
+     * @param eventName The event name to stop listening for.
+     * @param listener The callback function to remove.
+     * @returns This instance for method chaining.
+     */
+    off(eventName: ExecutionEventName, listener: (event: AgentExecutionEvent) => void): this;
+    /**
+     * EventEmitter-compatible 'once' method.
+     * Listener is automatically removed after first invocation.
+     * Supports multiple registrations of the same listener (like EventEmitter).
+     * @param eventName The event name to listen for once.
+     * @param listener The callback function to invoke when the event is emitted.
+     * @returns This instance for method chaining.
+     */
+    once(eventName: ExecutionEventName, listener: (event: AgentExecutionEvent) => void): this;
+    /**
+     * EventEmitter-compatible 'removeAllListeners' method.
+     * Removes all listeners for a specific event or all events.
+     * @param eventName Optional event name to remove listeners for. If omitted, removes all.
+     * @returns This instance for method chaining.
+     */
+    removeAllListeners(eventName?: ExecutionEventName): this;
+    /**
+     * Adds a wrapped listener to the tracking map.
+     */
+    private trackListener;
+    /**
+     * Removes a wrapped listener from the tracking map (for once cleanup).
+     */
+    private untrackWrappedListener;
+    private addEventListenerInternal;
+    private removeEventListenerInternal;
+    private addEventListenerOnceInternal;
+    private addFinishedListenerInternal;
+    private removeFinishedListenerInternal;
+    private addFinishedListenerOnceInternal;
 }
 
 declare class RequestContext {
@@ -253,4 +313,4 @@ declare class DefaultPushNotificationSender implements PushNotificationSender {
     private _dispatchNotification;
 }
 
-export { A2AError, A2ARequestHandler, type AgentExecutionEvent, type AgentExecutor, DefaultExecutionEventBus, DefaultExecutionEventBusManager, DefaultPushNotificationSender, type DefaultPushNotificationSenderOptions, DefaultRequestHandler, type ExecutionEventBus, type ExecutionEventBusManager, ExecutionEventQueue, type ExtendedAgentCardProvider, InMemoryPushNotificationStore, InMemoryTaskStore, JsonRpcTransportHandler, type PushNotificationSender, type PushNotificationStore, RequestContext, ResultManager, ServerCallContext, type TaskStore };
+export { A2AError, A2ARequestHandler, type AgentExecutionEvent, type AgentExecutor, DefaultExecutionEventBus, DefaultExecutionEventBusManager, DefaultPushNotificationSender, type DefaultPushNotificationSenderOptions, DefaultRequestHandler, type ExecutionEventBus, type ExecutionEventBusManager, type ExecutionEventName, ExecutionEventQueue, type ExtendedAgentCardProvider, InMemoryPushNotificationStore, InMemoryTaskStore, JsonRpcTransportHandler, type PushNotificationSender, type PushNotificationStore, RequestContext, ResultManager, ServerCallContext, type TaskStore };

package/dist/server/index.js

@@ -25,16 +25,194 @@ var RequestContext = class {
 };
 
 // src/server/events/execution_event_bus.ts
-import { EventEmitter } from "events";
-var DefaultExecutionEventBus = class extends EventEmitter {
-  constructor() {
-    super();
+var CustomEventImpl = typeof CustomEvent !== "undefined" ? CustomEvent : class CustomEventPolyfill extends Event {
+  detail;
+  constructor(type, eventInitDict) {
+    super(type, eventInitDict);
+    this.detail = eventInitDict?.detail ?? null;
   }
+};
+function isAgentExecutionCustomEvent(e) {
+  return e instanceof CustomEventImpl;
+}
+var DefaultExecutionEventBus = class extends EventTarget {
+  // Separate storage for each event type - both use the interface's Listener type
+  // but are invoked differently (with event payload vs. no arguments)
+  eventListeners = /* @__PURE__ */ new Map();
+  finishedListeners = /* @__PURE__ */ new Map();
   publish(event) {
-    this.emit("event", event);
+    this.dispatchEvent(new CustomEventImpl("event", { detail: event }));
   }
   finished() {
-    this.emit("finished");
+    this.dispatchEvent(new Event("finished"));
+  }
+  /**
+   * EventEmitter-compatible 'on' method.
+   * Wraps the listener to extract event detail from CustomEvent.
+   * Supports multiple registrations of the same listener (like EventEmitter).
+   * @param eventName The event name to listen for.
+   * @param listener The callback function to invoke when the event is emitted.
+   * @returns This instance for method chaining.
+   */
+  on(eventName, listener) {
+    if (eventName === "event") {
+      this.addEventListenerInternal(listener);
+    } else {
+      this.addFinishedListenerInternal(listener);
+    }
+    return this;
+  }
+  /**
+   * EventEmitter-compatible 'off' method.
+   * Uses the stored wrapped listener for proper removal.
+   * Removes at most one instance of a listener per call (like EventEmitter).
+   * @param eventName The event name to stop listening for.
+   * @param listener The callback function to remove.
+   * @returns This instance for method chaining.
+   */
+  off(eventName, listener) {
+    if (eventName === "event") {
+      this.removeEventListenerInternal(listener);
+    } else {
+      this.removeFinishedListenerInternal(listener);
+    }
+    return this;
+  }
+  /**
+   * EventEmitter-compatible 'once' method.
+   * Listener is automatically removed after first invocation.
+   * Supports multiple registrations of the same listener (like EventEmitter).
+   * @param eventName The event name to listen for once.
+   * @param listener The callback function to invoke when the event is emitted.
+   * @returns This instance for method chaining.
+   */
+  once(eventName, listener) {
+    if (eventName === "event") {
+      this.addEventListenerOnceInternal(listener);
+    } else {
+      this.addFinishedListenerOnceInternal(listener);
+    }
+    return this;
+  }
+  /**
+   * EventEmitter-compatible 'removeAllListeners' method.
+   * Removes all listeners for a specific event or all events.
+   * @param eventName Optional event name to remove listeners for. If omitted, removes all.
+   * @returns This instance for method chaining.
+   */
+  removeAllListeners(eventName) {
+    if (eventName === void 0 || eventName === "event") {
+      for (const wrappedListeners of this.eventListeners.values()) {
+        for (const wrapped of wrappedListeners) {
+          this.removeEventListener("event", wrapped);
+        }
+      }
+      this.eventListeners.clear();
+    }
+    if (eventName === void 0 || eventName === "finished") {
+      for (const wrappedListeners of this.finishedListeners.values()) {
+        for (const wrapped of wrappedListeners) {
+          this.removeEventListener("finished", wrapped);
+        }
+      }
+      this.finishedListeners.clear();
+    }
+    return this;
+  }
+  // ========================
+  // Helper methods for listener tracking
+  // ========================
+  /**
+   * Adds a wrapped listener to the tracking map.
+   */
+  trackListener(listenerMap, listener, wrapped) {
+    const existing = listenerMap.get(listener);
+    if (existing) {
+      existing.push(wrapped);
+    } else {
+      listenerMap.set(listener, [wrapped]);
+    }
+  }
+  /**
+   * Removes a wrapped listener from the tracking map (for once cleanup).
+   */
+  untrackWrappedListener(listenerMap, listener, wrapped) {
+    const wrappedList = listenerMap.get(listener);
+    if (wrappedList && wrappedList.length > 0) {
+      const index = wrappedList.indexOf(wrapped);
+      if (index !== -1) {
+        wrappedList.splice(index, 1);
+        if (wrappedList.length === 0) {
+          listenerMap.delete(listener);
+        }
+      }
+    }
+  }
+  // ========================
+  // Internal methods for 'event' listeners
+  // ========================
+  addEventListenerInternal(listener) {
+    const wrapped = (e) => {
+      if (!isAgentExecutionCustomEvent(e)) {
+        throw new Error('Internal error: expected CustomEvent for "event" type');
+      }
+      listener.call(this, e.detail);
+    };
+    this.trackListener(this.eventListeners, listener, wrapped);
+    this.addEventListener("event", wrapped);
+  }
+  removeEventListenerInternal(listener) {
+    const wrappedList = this.eventListeners.get(listener);
+    if (wrappedList && wrappedList.length > 0) {
+      const wrapped = wrappedList.pop();
+      if (wrappedList.length === 0) {
+        this.eventListeners.delete(listener);
+      }
+      this.removeEventListener("event", wrapped);
+    }
+  }
+  addEventListenerOnceInternal(listener) {
+    const wrapped = (e) => {
+      if (!isAgentExecutionCustomEvent(e)) {
+        throw new Error('Internal error: expected CustomEvent for "event" type');
+      }
+      this.untrackWrappedListener(this.eventListeners, listener, wrapped);
+      listener.call(this, e.detail);
+    };
+    this.trackListener(this.eventListeners, listener, wrapped);
+    this.addEventListener("event", wrapped, { once: true });
+  }
+  // ========================
+  // Internal methods for 'finished' listeners
+  // ========================
+  // The interface declares listeners as (event: AgentExecutionEvent) => void,
+  // but for 'finished' events they are invoked with no arguments (EventEmitter behavior).
+  // We use Function.prototype.call to invoke with `this` as the event bus (matching
+  // EventEmitter semantics) and no arguments, which is type-safe.
+  addFinishedListenerInternal(listener) {
+    const wrapped = () => {
+      listener.call(this);
+    };
+    this.trackListener(this.finishedListeners, listener, wrapped);
+    this.addEventListener("finished", wrapped);
+  }
+  removeFinishedListenerInternal(listener) {
+    const wrappedList = this.finishedListeners.get(listener);
+    if (wrappedList && wrappedList.length > 0) {
+      const wrapped = wrappedList.pop();
+      if (wrappedList.length === 0) {
+        this.finishedListeners.delete(listener);
+      }
+      this.removeEventListener("finished", wrapped);
+    }
+  }
+  addFinishedListenerOnceInternal(listener) {
+    const wrapped = () => {
+      this.untrackWrappedListener(this.finishedListeners, listener, wrapped);
+      listener.call(this);
+    };
+    this.trackListener(this.finishedListeners, listener, wrapped);
+    this.addEventListener("finished", wrapped, { once: true });
   }
 };
 

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@a2a-js/sdk",
-  "version": "0.3.6",
+  "version": "0.3.7",
   "description": "Server & Client SDK for Agent2Agent protocol",
   "license": "Apache-2.0",
   "repository": {
@@ -41,15 +41,12 @@
     "README.md"
   ],
   "devDependencies": {
+    "@cloudflare/vitest-pool-workers": "^0.10.15",
     "@eslint/js": "^9.39.1",
-    "@types/chai": "^5.2.2",
     "@types/express": "^5.0.3",
-    "@types/mocha": "^10.0.10",
     "@types/node": "^22.13.14",
-    "@types/sinon": "^17.0.4",
     "@types/supertest": "^6.0.3",
-    "c8": "^10.1.3",
-    "chai": "^5.2.0",
+    "@vitest/coverage-v8": "^3.2.4",
     "esbuild": "^0.27.0",
     "eslint": "^9.39.1",
     "eslint-config-prettier": "^10.1.8",
@@ -57,25 +54,25 @@
     "express": "^5.1.0",
     "gts": "^6.0.2",
     "json-schema-to-typescript": "^15.0.4",
-    "mocha": "^11.6.0",
     "prettier": "^3.6.2",
-    "sinon": "^20.0.0",
     "supertest": "^7.1.4",
     "tsup": "^8.5.0",
     "tsx": "^4.19.3",
     "typescript": "^5.9.3",
-    "typescript-eslint": "^8.46.4"
+    "typescript-eslint": "^8.46.4",
+    "vitest": "^3.2.4"
   },
   "scripts": {
     "clean": "gts clean",
     "build": "tsup",
-    "test": "mocha \"test/**/*.spec.ts\"",
+    "test": "vitest run",
+    "test:edge": "vitest run --config vitest.edge.config.ts",
     "lint": "tsc --noEmit && npx eslint .",
     "format:readme": "prettier --write ./README.md",
     "lint:fix": "npx eslint . --fix",
-    "coverage": "c8 npm run test",
+    "coverage": "vitest run --coverage",
     "generate": "curl https://raw.githubusercontent.com/google-a2a/A2A/refs/heads/main/specification/json/a2a.json > spec.json && node scripts/generateTypes.js && rm spec.json",
-    "test-build:browser": "esbuild ./dist/client/index.js ./dist/index.js --bundle --platform=browser --outdir=dist/tmp-checks --outbase=./dist"
+    "test-build": "esbuild ./dist/client/index.js ./dist/server/index.js ./dist/index.js --bundle --platform=neutral --outdir=dist/tmp-checks --outbase=./dist"
   },
   "dependencies": {
     "uuid": "^11.1.0"
@@ -87,8 +84,5 @@
     "express": {
       "optional": true
     }
-  },
-  "mocha": {
-    "require": "tsx"
   }
 }