@a2a-js/sdk 0.3.4 → 0.3.6

package/dist/server/index.d.cts

@@ -1,6 +1,7 @@
 import { EventEmitter } from 'events';
-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.cjs';
-import { A as A2ARequestHandler } from '../a2a_request_handler-DUvKWfix.cjs';
+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;
 interface ExecutionEventBus {
@@ -19,11 +20,12 @@ declare class DefaultExecutionEventBus extends EventEmitter implements Execution
 
 declare class RequestContext {
     readonly userMessage: Message;
-    readonly task?: Task;
-    readonly referenceTasks?: Task[];
     readonly taskId: string;
     readonly contextId: string;
-    constructor(userMessage: Message, taskId: string, contextId: string, task?: Task, referenceTasks?: Task[]);
+    readonly task?: Task;
+    readonly referenceTasks?: Task[];
+    readonly context?: ServerCallContext;
+    constructor(userMessage: Message, taskId: string, contextId: string, task?: Task, referenceTasks?: Task[], context?: ServerCallContext);
 }
 
 interface AgentExecutor {
@@ -53,13 +55,13 @@ declare class DefaultExecutionEventBusManager implements ExecutionEventBusManage
     /**
      * Creates or retrieves an existing ExecutionEventBus based on the taskId.
      * @param taskId The ID of the task.
-     * @returns An instance of IExecutionEventBus.
+     * @returns An instance of ExecutionEventBus.
      */
     createOrGetByTaskId(taskId: string): ExecutionEventBus;
     /**
      * Retrieves an existing ExecutionEventBus based on the taskId.
      * @param taskId The ID of the task.
-     * @returns An instance of IExecutionEventBus or undefined if not found.
+     * @returns An instance of ExecutionEventBus or undefined if not found.
      */
     getByTaskId(taskId: string): ExecutionEventBus | undefined;
     /**
@@ -102,16 +104,18 @@ interface TaskStore {
     /**
      * Saves a task.
      * Overwrites existing data if the task ID exists.
-     * @param data An object containing the task.
+     * @param task The task to save.
+     * @param context The context of the current call.
      * @returns A promise resolving when the save operation is complete.
      */
-    save(task: Task): Promise<void>;
+    save(task: Task, context?: ServerCallContext): Promise<void>;
     /**
      * Loads a task by task ID.
      * @param taskId The ID of the task to load.
+     * @param context The context of the current call.
      * @returns A promise resolving to an object containing the Task, or undefined if not found.
      */
-    load(taskId: string): Promise<Task | undefined>;
+    load(taskId: string, context?: ServerCallContext): Promise<Task | undefined>;
 }
 declare class InMemoryTaskStore implements TaskStore {
     private store;
@@ -137,35 +141,38 @@ interface PushNotificationSender {
 
 declare class DefaultRequestHandler implements A2ARequestHandler {
     private readonly agentCard;
-    private readonly extendedAgentCard?;
     private readonly taskStore;
     private readonly agentExecutor;
     private readonly eventBusManager;
     private readonly pushNotificationStore?;
     private readonly pushNotificationSender?;
-    constructor(agentCard: AgentCard, taskStore: TaskStore, agentExecutor: AgentExecutor, eventBusManager?: ExecutionEventBusManager, pushNotificationStore?: PushNotificationStore, pushNotificationSender?: PushNotificationSender, extendedAgentCard?: AgentCard);
+    private readonly extendedAgentCardProvider?;
+    constructor(agentCard: AgentCard, taskStore: TaskStore, agentExecutor: AgentExecutor, eventBusManager?: ExecutionEventBusManager, pushNotificationStore?: PushNotificationStore, pushNotificationSender?: PushNotificationSender, extendedAgentCardProvider?: AgentCard | ExtendedAgentCardProvider);
     getAgentCard(): Promise<AgentCard>;
-    getAuthenticatedExtendedAgentCard(): Promise<AgentCard>;
+    getAuthenticatedExtendedAgentCard(context?: ServerCallContext): Promise<AgentCard>;
     private _createRequestContext;
     private _processEvents;
-    sendMessage(params: MessageSendParams): Promise<Message | Task>;
-    sendMessageStream(params: MessageSendParams): AsyncGenerator<Message | Task | TaskStatusUpdateEvent | TaskArtifactUpdateEvent, void, undefined>;
-    getTask(params: TaskQueryParams): Promise<Task>;
-    cancelTask(params: TaskIdParams): Promise<Task>;
-    setTaskPushNotificationConfig(params: TaskPushNotificationConfig): Promise<TaskPushNotificationConfig>;
-    getTaskPushNotificationConfig(params: TaskIdParams | GetTaskPushNotificationConfigParams): Promise<TaskPushNotificationConfig>;
-    listTaskPushNotificationConfigs(params: ListTaskPushNotificationConfigParams): Promise<TaskPushNotificationConfig[]>;
-    deleteTaskPushNotificationConfig(params: DeleteTaskPushNotificationConfigParams): Promise<void>;
-    resubscribe(params: TaskIdParams): AsyncGenerator<Task | TaskStatusUpdateEvent | TaskArtifactUpdateEvent, void, undefined>;
+    sendMessage(params: MessageSendParams, context?: ServerCallContext): Promise<Message | Task>;
+    sendMessageStream(params: MessageSendParams, context?: ServerCallContext): AsyncGenerator<Message | Task | TaskStatusUpdateEvent | TaskArtifactUpdateEvent, void, undefined>;
+    getTask(params: TaskQueryParams, context?: ServerCallContext): Promise<Task>;
+    cancelTask(params: TaskIdParams, context?: ServerCallContext): Promise<Task>;
+    setTaskPushNotificationConfig(params: TaskPushNotificationConfig, context?: ServerCallContext): Promise<TaskPushNotificationConfig>;
+    getTaskPushNotificationConfig(params: TaskIdParams | GetTaskPushNotificationConfigParams, context?: ServerCallContext): Promise<TaskPushNotificationConfig>;
+    listTaskPushNotificationConfigs(params: ListTaskPushNotificationConfigParams, context?: ServerCallContext): Promise<TaskPushNotificationConfig[]>;
+    deleteTaskPushNotificationConfig(params: DeleteTaskPushNotificationConfigParams, context?: ServerCallContext): Promise<void>;
+    resubscribe(params: TaskIdParams, context?: ServerCallContext): AsyncGenerator<Task | TaskStatusUpdateEvent | TaskArtifactUpdateEvent, void, undefined>;
     private _sendPushNotificationIfNeeded;
+    private _handleProcessingError;
 }
+type ExtendedAgentCardProvider = (context?: ServerCallContext) => Promise<AgentCard>;
 
 declare class ResultManager {
-    private taskStore;
+    private readonly taskStore;
+    private readonly serverCallContext?;
     private currentTask?;
     private latestUserMessage?;
     private finalMessageResult?;
-    constructor(taskStore: TaskStore);
+    constructor(taskStore: TaskStore, serverCallContext?: ServerCallContext);
     setContext(latestUserMessage: Message): void;
     /**
      * Processes an agent execution event and updates the task store.
@@ -198,7 +205,9 @@ declare class JsonRpcTransportHandler {
      * For streaming methods, it returns an AsyncGenerator of JSONRPCResult.
      * For non-streaming methods, it returns a Promise of a single JSONRPCMessage (Result or ErrorResponse).
      */
-    handle(requestBody: any): Promise<JSONRPCResponse | AsyncGenerator<JSONRPCResponse, void, undefined>>;
+    handle(requestBody: any, context?: ServerCallContext): Promise<JSONRPCResponse | AsyncGenerator<JSONRPCResponse, void, undefined>>;
+    private isRequestValid;
+    private paramsAreValid;
 }
 
 /**
@@ -237,10 +246,11 @@ interface DefaultPushNotificationSenderOptions {
 }
 declare class DefaultPushNotificationSender implements PushNotificationSender {
     private readonly pushNotificationStore;
+    private notificationChain;
     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 };
+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 };

package/dist/server/index.d.ts

@@ -1,6 +1,7 @@
 import { EventEmitter } from 'events';
-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';
+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;
 interface ExecutionEventBus {
@@ -19,11 +20,12 @@ declare class DefaultExecutionEventBus extends EventEmitter implements Execution
 
 declare class RequestContext {
     readonly userMessage: Message;
-    readonly task?: Task;
-    readonly referenceTasks?: Task[];
     readonly taskId: string;
     readonly contextId: string;
-    constructor(userMessage: Message, taskId: string, contextId: string, task?: Task, referenceTasks?: Task[]);
+    readonly task?: Task;
+    readonly referenceTasks?: Task[];
+    readonly context?: ServerCallContext;
+    constructor(userMessage: Message, taskId: string, contextId: string, task?: Task, referenceTasks?: Task[], context?: ServerCallContext);
 }
 
 interface AgentExecutor {
@@ -53,13 +55,13 @@ declare class DefaultExecutionEventBusManager implements ExecutionEventBusManage
     /**
      * Creates or retrieves an existing ExecutionEventBus based on the taskId.
      * @param taskId The ID of the task.
-     * @returns An instance of IExecutionEventBus.
+     * @returns An instance of ExecutionEventBus.
      */
     createOrGetByTaskId(taskId: string): ExecutionEventBus;
     /**
      * Retrieves an existing ExecutionEventBus based on the taskId.
      * @param taskId The ID of the task.
-     * @returns An instance of IExecutionEventBus or undefined if not found.
+     * @returns An instance of ExecutionEventBus or undefined if not found.
      */
     getByTaskId(taskId: string): ExecutionEventBus | undefined;
     /**
@@ -102,16 +104,18 @@ interface TaskStore {
     /**
      * Saves a task.
      * Overwrites existing data if the task ID exists.
-     * @param data An object containing the task.
+     * @param task The task to save.
+     * @param context The context of the current call.
      * @returns A promise resolving when the save operation is complete.
      */
-    save(task: Task): Promise<void>;
+    save(task: Task, context?: ServerCallContext): Promise<void>;
     /**
      * Loads a task by task ID.
      * @param taskId The ID of the task to load.
+     * @param context The context of the current call.
      * @returns A promise resolving to an object containing the Task, or undefined if not found.
      */
-    load(taskId: string): Promise<Task | undefined>;
+    load(taskId: string, context?: ServerCallContext): Promise<Task | undefined>;
 }
 declare class InMemoryTaskStore implements TaskStore {
     private store;
@@ -137,35 +141,38 @@ interface PushNotificationSender {
 
 declare class DefaultRequestHandler implements A2ARequestHandler {
     private readonly agentCard;
-    private readonly extendedAgentCard?;
     private readonly taskStore;
     private readonly agentExecutor;
     private readonly eventBusManager;
     private readonly pushNotificationStore?;
     private readonly pushNotificationSender?;
-    constructor(agentCard: AgentCard, taskStore: TaskStore, agentExecutor: AgentExecutor, eventBusManager?: ExecutionEventBusManager, pushNotificationStore?: PushNotificationStore, pushNotificationSender?: PushNotificationSender, extendedAgentCard?: AgentCard);
+    private readonly extendedAgentCardProvider?;
+    constructor(agentCard: AgentCard, taskStore: TaskStore, agentExecutor: AgentExecutor, eventBusManager?: ExecutionEventBusManager, pushNotificationStore?: PushNotificationStore, pushNotificationSender?: PushNotificationSender, extendedAgentCardProvider?: AgentCard | ExtendedAgentCardProvider);
     getAgentCard(): Promise<AgentCard>;
-    getAuthenticatedExtendedAgentCard(): Promise<AgentCard>;
+    getAuthenticatedExtendedAgentCard(context?: ServerCallContext): Promise<AgentCard>;
     private _createRequestContext;
     private _processEvents;
-    sendMessage(params: MessageSendParams): Promise<Message | Task>;
-    sendMessageStream(params: MessageSendParams): AsyncGenerator<Message | Task | TaskStatusUpdateEvent | TaskArtifactUpdateEvent, void, undefined>;
-    getTask(params: TaskQueryParams): Promise<Task>;
-    cancelTask(params: TaskIdParams): Promise<Task>;
-    setTaskPushNotificationConfig(params: TaskPushNotificationConfig): Promise<TaskPushNotificationConfig>;
-    getTaskPushNotificationConfig(params: TaskIdParams | GetTaskPushNotificationConfigParams): Promise<TaskPushNotificationConfig>;
-    listTaskPushNotificationConfigs(params: ListTaskPushNotificationConfigParams): Promise<TaskPushNotificationConfig[]>;
-    deleteTaskPushNotificationConfig(params: DeleteTaskPushNotificationConfigParams): Promise<void>;
-    resubscribe(params: TaskIdParams): AsyncGenerator<Task | TaskStatusUpdateEvent | TaskArtifactUpdateEvent, void, undefined>;
+    sendMessage(params: MessageSendParams, context?: ServerCallContext): Promise<Message | Task>;
+    sendMessageStream(params: MessageSendParams, context?: ServerCallContext): AsyncGenerator<Message | Task | TaskStatusUpdateEvent | TaskArtifactUpdateEvent, void, undefined>;
+    getTask(params: TaskQueryParams, context?: ServerCallContext): Promise<Task>;
+    cancelTask(params: TaskIdParams, context?: ServerCallContext): Promise<Task>;
+    setTaskPushNotificationConfig(params: TaskPushNotificationConfig, context?: ServerCallContext): Promise<TaskPushNotificationConfig>;
+    getTaskPushNotificationConfig(params: TaskIdParams | GetTaskPushNotificationConfigParams, context?: ServerCallContext): Promise<TaskPushNotificationConfig>;
+    listTaskPushNotificationConfigs(params: ListTaskPushNotificationConfigParams, context?: ServerCallContext): Promise<TaskPushNotificationConfig[]>;
+    deleteTaskPushNotificationConfig(params: DeleteTaskPushNotificationConfigParams, context?: ServerCallContext): Promise<void>;
+    resubscribe(params: TaskIdParams, context?: ServerCallContext): AsyncGenerator<Task | TaskStatusUpdateEvent | TaskArtifactUpdateEvent, void, undefined>;
     private _sendPushNotificationIfNeeded;
+    private _handleProcessingError;
 }
+type ExtendedAgentCardProvider = (context?: ServerCallContext) => Promise<AgentCard>;
 
 declare class ResultManager {
-    private taskStore;
+    private readonly taskStore;
+    private readonly serverCallContext?;
     private currentTask?;
     private latestUserMessage?;
     private finalMessageResult?;
-    constructor(taskStore: TaskStore);
+    constructor(taskStore: TaskStore, serverCallContext?: ServerCallContext);
     setContext(latestUserMessage: Message): void;
     /**
      * Processes an agent execution event and updates the task store.
@@ -198,7 +205,9 @@ declare class JsonRpcTransportHandler {
      * For streaming methods, it returns an AsyncGenerator of JSONRPCResult.
      * For non-streaming methods, it returns a Promise of a single JSONRPCMessage (Result or ErrorResponse).
      */
-    handle(requestBody: any): Promise<JSONRPCResponse | AsyncGenerator<JSONRPCResponse, void, undefined>>;
+    handle(requestBody: any, context?: ServerCallContext): Promise<JSONRPCResponse | AsyncGenerator<JSONRPCResponse, void, undefined>>;
+    private isRequestValid;
+    private paramsAreValid;
 }
 
 /**
@@ -237,10 +246,11 @@ interface DefaultPushNotificationSenderOptions {
 }
 declare class DefaultPushNotificationSender implements PushNotificationSender {
     private readonly pushNotificationStore;
+    private notificationChain;
     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 };
+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 };