@a2a-js/sdk 0.3.2 → 0.3.4

package/dist/server/index.d.ts

@@ -1,5 +1,5 @@
 import { EventEmitter } from 'events';
-import { B as Message, aw as Task, aO as TaskStatusUpdateEvent, aQ as TaskArtifactUpdateEvent, ac as AgentCard, w as MessageSendParams, V as TaskQueryParams, X as TaskIdParams, Z as TaskPushNotificationConfig, a1 as GetTaskPushNotificationConfigParams, a5 as ListTaskPushNotificationConfigParams, a7 as DeleteTaskPushNotificationConfigParams, i as JSONRPCResponse, au as JSONRPCError } from '../types-DNKcmF0f.js';
+import { B as Message, aw as Task, aO as TaskStatusUpdateEvent, aQ as TaskArtifactUpdateEvent, y as PushNotificationConfig, ac as AgentCard, w as MessageSendParams, V as TaskQueryParams, X as TaskIdParams, Z as TaskPushNotificationConfig, a1 as GetTaskPushNotificationConfigParams, a5 as ListTaskPushNotificationConfigParams, a7 as DeleteTaskPushNotificationConfigParams, i as JSONRPCResponse, au as JSONRPCError } from '../types-DNKcmF0f.js';
 import { A as A2ARequestHandler } from '../a2a_request_handler-B5t-IxgA.js';
 
 type AgentExecutionEvent = Message | Task | TaskStatusUpdateEvent | TaskArtifactUpdateEvent;
@@ -119,14 +119,31 @@ declare class InMemoryTaskStore implements TaskStore {
     save(task: Task): Promise<void>;
 }
 
+interface PushNotificationStore {
+    save(taskId: string, pushNotificationConfig: PushNotificationConfig): Promise<void>;
+    load(taskId: string): Promise<PushNotificationConfig[]>;
+    delete(taskId: string, configId?: string): Promise<void>;
+}
+declare class InMemoryPushNotificationStore implements PushNotificationStore {
+    private store;
+    save(taskId: string, pushNotificationConfig: PushNotificationConfig): Promise<void>;
+    load(taskId: string): Promise<PushNotificationConfig[]>;
+    delete(taskId: string, configId?: string): Promise<void>;
+}
+
+interface PushNotificationSender {
+    send(task: Task): Promise<void>;
+}
+
 declare class DefaultRequestHandler implements A2ARequestHandler {
     private readonly agentCard;
     private readonly extendedAgentCard?;
     private readonly taskStore;
     private readonly agentExecutor;
     private readonly eventBusManager;
-    private readonly pushNotificationConfigs;
-    constructor(agentCard: AgentCard, taskStore: TaskStore, agentExecutor: AgentExecutor, eventBusManager?: ExecutionEventBusManager, extendedAgentCard?: AgentCard);
+    private readonly pushNotificationStore?;
+    private readonly pushNotificationSender?;
+    constructor(agentCard: AgentCard, taskStore: TaskStore, agentExecutor: AgentExecutor, eventBusManager?: ExecutionEventBusManager, pushNotificationStore?: PushNotificationStore, pushNotificationSender?: PushNotificationSender, extendedAgentCard?: AgentCard);
     getAgentCard(): Promise<AgentCard>;
     getAuthenticatedExtendedAgentCard(): Promise<AgentCard>;
     private _createRequestContext;
@@ -140,6 +157,7 @@ declare class DefaultRequestHandler implements A2ARequestHandler {
     listTaskPushNotificationConfigs(params: ListTaskPushNotificationConfigParams): Promise<TaskPushNotificationConfig[]>;
     deleteTaskPushNotificationConfig(params: DeleteTaskPushNotificationConfigParams): Promise<void>;
     resubscribe(params: TaskIdParams): AsyncGenerator<Task | TaskStatusUpdateEvent | TaskArtifactUpdateEvent, void, undefined>;
+    private _sendPushNotificationIfNeeded;
 }
 
 declare class ResultManager {
@@ -207,4 +225,22 @@ declare class A2AError extends Error {
     static authenticatedExtendedCardNotConfigured(): A2AError;
 }
 
-export { A2AError, A2ARequestHandler, type AgentExecutionEvent, type AgentExecutor, DefaultExecutionEventBus, DefaultExecutionEventBusManager, DefaultRequestHandler, type ExecutionEventBus, type ExecutionEventBusManager, ExecutionEventQueue, InMemoryTaskStore, JsonRpcTransportHandler, RequestContext, ResultManager, type TaskStore };
+interface DefaultPushNotificationSenderOptions {
+    /**
+     * Timeout in milliseconds for the abort controller. Defaults to 5000ms.
+     */
+    timeout?: number;
+    /**
+     * Custom header name for the token. Defaults to 'X-A2A-Notification-Token'.
+     */
+    tokenHeaderName?: string;
+}
+declare class DefaultPushNotificationSender implements PushNotificationSender {
+    private readonly pushNotificationStore;
+    private readonly options;
+    constructor(pushNotificationStore: PushNotificationStore, options?: DefaultPushNotificationSenderOptions);
+    send(task: Task): Promise<void>;
+    private _dispatchNotification;
+}
+
+export { A2AError, A2ARequestHandler, type AgentExecutionEvent, type AgentExecutor, DefaultExecutionEventBus, DefaultExecutionEventBusManager, DefaultPushNotificationSender, type DefaultPushNotificationSenderOptions, DefaultRequestHandler, type ExecutionEventBus, type ExecutionEventBusManager, ExecutionEventQueue, InMemoryPushNotificationStore, InMemoryTaskStore, JsonRpcTransportHandler, type PushNotificationSender, type PushNotificationStore, RequestContext, ResultManager, type TaskStore };

package/dist/server/index.js

@@ -257,6 +257,97 @@ var ResultManager = class {
   }
 };
 
+// src/server/push_notification/push_notification_store.ts
+var InMemoryPushNotificationStore = class {
+  store = /* @__PURE__ */ new Map();
+  async save(taskId, pushNotificationConfig) {
+    const configs = this.store.get(taskId) || [];
+    if (!pushNotificationConfig.id) {
+      pushNotificationConfig.id = taskId;
+    }
+    const existingIndex = configs.findIndex((config) => config.id === pushNotificationConfig.id);
+    if (existingIndex !== -1) {
+      configs.splice(existingIndex, 1);
+    }
+    configs.push(pushNotificationConfig);
+    this.store.set(taskId, configs);
+  }
+  async load(taskId) {
+    const configs = this.store.get(taskId);
+    return configs || [];
+  }
+  async delete(taskId, configId) {
+    if (configId === void 0) {
+      configId = taskId;
+    }
+    const configs = this.store.get(taskId);
+    if (!configs) {
+      return;
+    }
+    const configIndex = configs.findIndex((config) => config.id === configId);
+    if (configIndex !== -1) {
+      configs.splice(configIndex, 1);
+    }
+    if (configs.length === 0) {
+      this.store.delete(taskId);
+    } else {
+      this.store.set(taskId, configs);
+    }
+  }
+};
+
+// src/server/push_notification/default_push_notification_sender.ts
+var DefaultPushNotificationSender = class {
+  pushNotificationStore;
+  options;
+  constructor(pushNotificationStore, options = {}) {
+    this.pushNotificationStore = pushNotificationStore;
+    this.options = {
+      timeout: 5e3,
+      tokenHeaderName: "X-A2A-Notification-Token",
+      ...options
+    };
+  }
+  async send(task) {
+    const pushConfigs = await this.pushNotificationStore.load(task.id);
+    if (!pushConfigs || pushConfigs.length === 0) {
+      return;
+    }
+    pushConfigs.forEach((pushConfig) => {
+      this._dispatchNotification(task, pushConfig).catch((error) => {
+        console.error(`Error sending push notification for task_id=${task.id} to URL: ${pushConfig.url}. Error:`, error);
+      });
+    });
+  }
+  async _dispatchNotification(task, pushConfig) {
+    const url = pushConfig.url;
+    const controller = new AbortController();
+    const timeoutId = setTimeout(() => controller.abort(), this.options.timeout);
+    try {
+      const headers = {
+        "Content-Type": "application/json"
+      };
+      if (pushConfig.token) {
+        headers[this.options.tokenHeaderName] = pushConfig.token;
+      }
+      const response = await fetch(url, {
+        method: "POST",
+        headers,
+        body: JSON.stringify(task),
+        signal: controller.signal
+      });
+      if (!response.ok) {
+        throw new Error(`HTTP ${response.status}: ${response.statusText}`);
+      }
+      console.info(`Push notification sent for task_id=${task.id} to URL: ${url}`);
+    } catch (error) {
+      console.error(`Error sending push notification for task_id=${task.id} to URL: ${url}. Error:`, error);
+    } finally {
+      clearTimeout(timeoutId);
+    }
+  }
+};
+
 // src/server/request_handler/default_request_handler.ts
 var terminalStates = ["completed", "failed", "canceled", "rejected"];
 var DefaultRequestHandler = class {
@@ -265,14 +356,18 @@ var DefaultRequestHandler = class {
   taskStore;
   agentExecutor;
   eventBusManager;
-  // Store for push notification configurations (could be part of TaskStore or separate)
-  pushNotificationConfigs = /* @__PURE__ */ new Map();
-  constructor(agentCard, taskStore, agentExecutor, eventBusManager = new DefaultExecutionEventBusManager(), extendedAgentCard) {
+  pushNotificationStore;
+  pushNotificationSender;
+  constructor(agentCard, taskStore, agentExecutor, eventBusManager = new DefaultExecutionEventBusManager(), pushNotificationStore, pushNotificationSender, extendedAgentCard) {
     this.agentCard = agentCard;
     this.taskStore = taskStore;
     this.agentExecutor = agentExecutor;
     this.eventBusManager = eventBusManager;
     this.extendedAgentCard = extendedAgentCard;
+    if (agentCard.capabilities.pushNotifications) {
+      this.pushNotificationStore = pushNotificationStore || new InMemoryPushNotificationStore();
+      this.pushNotificationSender = pushNotificationSender || new DefaultPushNotificationSender(this.pushNotificationStore);
+    }
   }
   async getAgentCard() {
     return this.agentCard;
@@ -324,6 +419,7 @@ var DefaultRequestHandler = class {
     try {
       for await (const event of eventQueue.events()) {
         await resultManager.processEvent(event);
+        await this._sendPushNotificationIfNeeded(event);
         if (options?.firstResultResolver && !firstResultSent) {
           if (event.kind === "message" || event.kind === "task") {
             options.firstResultResolver(event);
@@ -355,6 +451,9 @@ var DefaultRequestHandler = class {
     resultManager.setContext(incomingMessage);
     const requestContext = await this._createRequestContext(incomingMessage, taskId, false);
     const finalMessageForAgent = requestContext.userMessage;
+    if (params.configuration?.pushNotificationConfig && this.agentCard.capabilities.pushNotifications) {
+      await this.pushNotificationStore?.save(taskId, params.configuration.pushNotificationConfig);
+    }
     const eventBus = this.eventBusManager.createOrGetByTaskId(taskId);
     const eventQueue = new ExecutionEventQueue(eventBus);
     this.agentExecutor.execute(requestContext, eventBus).catch((err) => {
@@ -422,6 +521,9 @@ var DefaultRequestHandler = class {
     const finalMessageForAgent = requestContext.userMessage;
     const eventBus = this.eventBusManager.createOrGetByTaskId(taskId);
     const eventQueue = new ExecutionEventQueue(eventBus);
+    if (params.configuration?.pushNotificationConfig && this.agentCard.capabilities.pushNotifications) {
+      await this.pushNotificationStore?.save(taskId, params.configuration.pushNotificationConfig);
+    }
     this.agentExecutor.execute(requestContext, eventBus).catch((err) => {
       console.error(`Agent execution failed for stream message ${finalMessageForAgent.messageId}:`, err);
       const errorTaskStatus = {
@@ -449,6 +551,7 @@ var DefaultRequestHandler = class {
     try {
       for await (const event of eventQueue.events()) {
         await resultManager.processEvent(event);
+        await this._sendPushNotificationIfNeeded(event);
         yield event;
       }
     } finally {
@@ -513,10 +616,7 @@ var DefaultRequestHandler = class {
     if (!pushNotificationConfig.id) {
       pushNotificationConfig.id = taskId;
     }
-    const configs = this.pushNotificationConfigs.get(taskId) || [];
-    const updatedConfigs = configs.filter((c) => c.id !== pushNotificationConfig.id);
-    updatedConfigs.push(pushNotificationConfig);
-    this.pushNotificationConfigs.set(taskId, updatedConfigs);
+    await this.pushNotificationStore?.save(taskId, pushNotificationConfig);
     return params;
   }
   async getTaskPushNotificationConfig(params) {
@@ -527,7 +627,7 @@ var DefaultRequestHandler = class {
     if (!task) {
       throw A2AError.taskNotFound(params.id);
     }
-    const configs = this.pushNotificationConfigs.get(params.id) || [];
+    const configs = await this.pushNotificationStore?.load(params.id) || [];
     if (configs.length === 0) {
       throw A2AError.internalError(`Push notification config not found for task ${params.id}.`);
     }
@@ -551,7 +651,7 @@ var DefaultRequestHandler = class {
     if (!task) {
       throw A2AError.taskNotFound(params.id);
     }
-    const configs = this.pushNotificationConfigs.get(params.id) || [];
+    const configs = await this.pushNotificationStore?.load(params.id) || [];
     return configs.map((config) => ({
       taskId: params.id,
       pushNotificationConfig: config
@@ -566,16 +666,7 @@ var DefaultRequestHandler = class {
       throw A2AError.taskNotFound(params.id);
     }
     const { id: taskId, pushNotificationConfigId } = params;
-    const configs = this.pushNotificationConfigs.get(taskId);
-    if (!configs) {
-      return;
-    }
-    const updatedConfigs = configs.filter((c) => c.id !== pushNotificationConfigId);
-    if (updatedConfigs.length === 0) {
-      this.pushNotificationConfigs.delete(taskId);
-    } else if (updatedConfigs.length < configs.length) {
-      this.pushNotificationConfigs.set(taskId, updatedConfigs);
-    }
+    await this.pushNotificationStore?.delete(taskId, pushNotificationConfigId);
   }
   async *resubscribe(params) {
     if (!this.agentCard.capabilities.streaming) {
@@ -610,6 +701,28 @@ var DefaultRequestHandler = class {
       eventQueue.stop();
     }
   }
+  async _sendPushNotificationIfNeeded(event) {
+    if (!this.agentCard.capabilities.pushNotifications) {
+      return;
+    }
+    let taskId = "";
+    if (event.kind == "task") {
+      const task2 = event;
+      taskId = task2.id;
+    } else {
+      taskId = event.taskId;
+    }
+    if (!taskId) {
+      console.error(`Task ID not found for event ${event.kind}.`);
+      return;
+    }
+    const task = await this.taskStore.load(taskId);
+    if (!task) {
+      console.error(`Task ${taskId} not found.`);
+      return;
+    }
+    this.pushNotificationSender?.send(task);
+  }
 };
 
 // src/server/store.ts
@@ -627,8 +740,10 @@ export {
   A2AError,
   DefaultExecutionEventBus,
   DefaultExecutionEventBusManager,
+  DefaultPushNotificationSender,
   DefaultRequestHandler,
   ExecutionEventQueue,
+  InMemoryPushNotificationStore,
   InMemoryTaskStore,
   JsonRpcTransportHandler,
   RequestContext,

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@a2a-js/sdk",
-  "version": "0.3.2",
+  "version": "0.3.4",
   "description": "Server & Client SDK for Agent2Agent protocol",
   "repository": {
     "type": "git",
@@ -43,13 +43,13 @@
     "@genkit-ai/googleai": "^1.8.0",
     "@genkit-ai/vertexai": "^1.8.0",
     "@types/chai": "^5.2.2",
-    "@types/express": "^4.17.23",
+    "@types/express": "^5.0.3",
     "@types/mocha": "^10.0.10",
     "@types/node": "^22.13.14",
     "@types/sinon": "^17.0.4",
     "c8": "^10.1.3",
     "chai": "^5.2.0",
-    "express": "^4.21.2",
+    "express": "^5.1.0",
     "genkit": "^1.8.0",
     "gts": "^6.0.2",
     "json-schema-to-typescript": "^15.0.4",
@@ -73,7 +73,7 @@
     "uuid": "^11.1.0"
   },
   "peerDependencies": {
-    "express": "^4.21.2"
+    "express": "^4.21.2 || ^5.1.0"
   },
   "peerDependenciesMeta": {
     "express": {