@mastra/client-js 0.0.0-default-storage-virtual-file-20250410035748 → 0.0.0-experimental-agent-builder-20250815195917

package/src/resources/a2a.ts

@@ -0,0 +1,98 @@
+import type {
+  AgentCard,
+  GetTaskResponse,
+  MessageSendParams,
+  SendMessageResponse,
+  SendStreamingMessageResponse,
+  Task,
+  TaskQueryParams,
+} from '@mastra/core/a2a';
+import type { ClientOptions } from '../types';
+import { BaseResource } from './base';
+
+/**
+ * Class for interacting with an agent via the A2A protocol
+ */
+export class A2A extends BaseResource {
+  constructor(
+    options: ClientOptions,
+    private agentId: string,
+  ) {
+    super(options);
+  }
+
+  /**
+   * Get the agent card with metadata about the agent
+   * @returns Promise containing the agent card information
+   */
+  async getCard(): Promise<AgentCard> {
+    return this.request(`/.well-known/${this.agentId}/agent-card.json`);
+  }
+
+  /**
+   * Send a message to the agent and gets a message or task response
+   * @param params - Parameters for the task
+   * @returns Promise containing the response
+   */
+  async sendMessage(params: MessageSendParams): Promise<SendMessageResponse> {
+    const response = await this.request<SendMessageResponse>(`/a2a/${this.agentId}`, {
+      method: 'POST',
+      body: {
+        method: 'message/send',
+        params,
+      },
+    });
+
+    return response;
+  }
+
+  /**
+   * Sends a message to an agent to initiate/continue a task and subscribes
+   * the client to real-time updates for that task via Server-Sent Events (SSE).
+   * @param params - Parameters for the task
+   * @returns A stream of Server-Sent Events. Each SSE `data` field contains a `SendStreamingMessageResponse`
+   */
+  async sendStreamingMessage(params: MessageSendParams): Promise<AsyncIterable<SendStreamingMessageResponse>> {
+    const response = await this.request<AsyncIterable<SendStreamingMessageResponse>>(`/a2a/${this.agentId}`, {
+      method: 'POST',
+      body: {
+        method: 'message/stream',
+        params,
+      },
+    });
+
+    return response;
+  }
+
+  /**
+   * Get the status and result of a task
+   * @param params - Parameters for querying the task
+   * @returns Promise containing the task response
+   */
+  async getTask(params: TaskQueryParams): Promise<GetTaskResponse> {
+    const response = await this.request<GetTaskResponse>(`/a2a/${this.agentId}`, {
+      method: 'POST',
+      body: {
+        method: 'tasks/get',
+        params,
+      },
+    });
+
+    return response;
+  }
+
+  /**
+   * Cancel a running task
+   * @param params - Parameters identifying the task to cancel
+   * @returns Promise containing the task response
+   */
+  async cancelTask(params: TaskQueryParams): Promise<Task> {
+    return this.request(`/a2a/${this.agentId}`, {
+      method: 'POST',
+      body: {
+        method: 'tasks/cancel',
+        params,
+      },
+    });
+  }
+}