@a2a-js/sdk 0.3.7 → 0.3.8

package/dist/client/index.js

@@ -1,18 +1,28 @@
 import {
   A2A_ERROR_CODE,
+  AgentCard,
   AuthenticatedExtendedCardNotConfiguredError,
   ContentTypeNotSupportedError,
+  FromProto,
   InvalidAgentResponseError,
+  ListTaskPushNotificationConfigResponse,
   PushNotificationNotSupportedError,
+  SendMessageRequest,
+  SendMessageResponse,
+  StreamResponse,
+  Task,
   TaskNotCancelableError,
   TaskNotFoundError,
+  TaskPushNotificationConfig,
+  ToProto,
   UnsupportedOperationError,
   parseSseStream
-} from "../chunk-SJNAG4AL.js";
+} from "../chunk-DHC2REQH.js";
 import {
   AGENT_CARD_PATH,
   HTTP_EXTENSION_HEADER
 } from "../chunk-3QDLXHKS.js";
+import "../chunk-UHZEIZLS.js";
 import {
   Extensions
 } from "../chunk-ZX6KNMCP.js";
@@ -1003,23 +1013,44 @@ var RestTransport = class _RestTransport {
     this.customFetchImpl = options.fetchImpl;
   }
   async getExtendedAgentCard(options) {
-    return this._sendRequest("GET", "/v1/card", void 0, options);
+    const response = await this._sendRequest(
+      "GET",
+      "/v1/card",
+      void 0,
+      options,
+      void 0,
+      AgentCard
+    );
+    return FromProto.agentCard(response);
   }
   async sendMessage(params, options) {
-    return this._sendRequest("POST", "/v1/message:send", params, options);
+    const requestBody = ToProto.messageSendParams(params);
+    const response = await this._sendRequest(
+      "POST",
+      "/v1/message:send",
+      requestBody,
+      options,
+      SendMessageRequest,
+      SendMessageResponse
+    );
+    return FromProto.sendMessageResult(response);
   }
   async *sendMessageStream(params, options) {
-    yield* this._sendStreamingRequest("/v1/message:stream", params, options);
+    const protoParams = ToProto.messageSendParams(params);
+    const requestBody = SendMessageRequest.toJSON(protoParams);
+    yield* this._sendStreamingRequest("/v1/message:stream", requestBody, options);
   }
   async setTaskPushNotificationConfig(params, options) {
-    return this._sendRequest(
+    const requestBody = ToProto.taskPushNotificationConfig(params);
+    const response = await this._sendRequest(
       "POST",
       `/v1/tasks/${encodeURIComponent(params.taskId)}/pushNotificationConfigs`,
-      {
-        pushNotificationConfig: params.pushNotificationConfig
-      },
-      options
+      requestBody,
+      options,
+      TaskPushNotificationConfig,
+      TaskPushNotificationConfig
     );
+    return FromProto.taskPushNotificationConfig(response);
   }
   async getTaskPushNotificationConfig(params, options) {
     const { pushNotificationConfigId } = params;
@@ -1028,27 +1059,35 @@ var RestTransport = class _RestTransport {
         "pushNotificationConfigId is required for getTaskPushNotificationConfig with REST transport."
       );
     }
-    return this._sendRequest(
+    const response = await this._sendRequest(
       "GET",
       `/v1/tasks/${encodeURIComponent(params.id)}/pushNotificationConfigs/${encodeURIComponent(pushNotificationConfigId)}`,
       void 0,
-      options
+      options,
+      void 0,
+      TaskPushNotificationConfig
     );
+    return FromProto.taskPushNotificationConfig(response);
   }
   async listTaskPushNotificationConfig(params, options) {
-    return this._sendRequest(
+    const response = await this._sendRequest(
       "GET",
       `/v1/tasks/${encodeURIComponent(params.id)}/pushNotificationConfigs`,
       void 0,
-      options
+      options,
+      void 0,
+      ListTaskPushNotificationConfigResponse
     );
+    return FromProto.listTaskPushNotificationConfig(response);
   }
   async deleteTaskPushNotificationConfig(params, options) {
     await this._sendRequest(
       "DELETE",
       `/v1/tasks/${encodeURIComponent(params.id)}/pushNotificationConfigs/${encodeURIComponent(params.pushNotificationConfigId)}`,
       void 0,
-      options
+      options,
+      void 0,
+      void 0
     );
   }
   async getTask(params, options) {
@@ -1058,15 +1097,26 @@ var RestTransport = class _RestTransport {
     }
     const queryString = queryParams.toString();
     const path = `/v1/tasks/${encodeURIComponent(params.id)}${queryString ? `?${queryString}` : ""}`;
-    return this._sendRequest("GET", path, void 0, options);
+    const response = await this._sendRequest(
+      "GET",
+      path,
+      void 0,
+      options,
+      void 0,
+      Task
+    );
+    return FromProto.task(response);
   }
   async cancelTask(params, options) {
-    return this._sendRequest(
+    const response = await this._sendRequest(
       "POST",
       `/v1/tasks/${encodeURIComponent(params.id)}:cancel`,
       void 0,
-      options
+      options,
+      void 0,
+      Task
     );
+    return FromProto.task(response);
   }
   async *resubscribeTask(params, options) {
     yield* this._sendStreamingRequest(
@@ -1093,7 +1143,7 @@ var RestTransport = class _RestTransport {
       Accept: acceptHeader
     };
   }
-  async _sendRequest(method, path, body, options) {
+  async _sendRequest(method, path, body, options, requestType, responseType) {
     const url = `${this.endpoint}${path}`;
     const requestInit = {
       method,
@@ -1101,17 +1151,22 @@ var RestTransport = class _RestTransport {
       signal: options?.signal
     };
     if (body !== void 0 && method !== "GET") {
-      requestInit.body = JSON.stringify(body);
+      if (!requestType) {
+        throw new Error(
+          `Bug: Request body provided for ${method} ${path} but no toJson serializer provided.`
+        );
+      }
+      requestInit.body = JSON.stringify(requestType.toJSON(body));
     }
     const response = await this._fetch(url, requestInit);
     if (!response.ok) {
       await this._handleErrorResponse(response, path);
     }
-    if (response.status === 204) {
+    if (response.status === 204 || !responseType) {
       return void 0;
     }
     const result = await response.json();
-    return result;
+    return responseType.fromJSON(result);
   }
   async _handleErrorResponse(response, path) {
     let errorBodyText = "(empty or non-JSON response)";
@@ -1167,8 +1222,9 @@ var RestTransport = class _RestTransport {
       throw new Error("Attempted to process empty SSE event data.");
     }
     try {
-      const data = JSON.parse(jsonData);
-      return data;
+      const response = JSON.parse(jsonData);
+      const protoResponse = StreamResponse.fromJSON(response);
+      return FromProto.messageStreamResult(protoResponse);
     } catch (e) {
       console.error("Failed to parse SSE event data:", jsonData, e);
       throw new Error(
@@ -1262,8 +1318,7 @@ var ClientFactory = class {
     }
     this.transportsByName = transportsByName(options.transports);
     for (const transport of options.preferredTransports ?? []) {
-      const factory = this.options.transports.find((t) => t.protocolName === transport);
-      if (!factory) {
+      if (!this.transportsByName.has(transport)) {
         throw new Error(
           `Unknown preferred transport: ${transport}, available transports: ${[...this.transportsByName.keys()].join()}`
         );
@@ -1279,7 +1334,7 @@ var ClientFactory = class {
   async createFromAgentCard(agentCard) {
     const agentCardPreferred = agentCard.preferredTransport ?? JsonRpcTransportFactory.name;
     const additionalInterfaces = agentCard.additionalInterfaces ?? [];
-    const urlsPerAgentTransports = new Map([
+    const urlsPerAgentTransports = new CaseInsensitiveMap([
       [agentCardPreferred, agentCard.url],
       ...additionalInterfaces.map((i) => [i.transport, i.url])
     ]);
@@ -1335,7 +1390,7 @@ function mergeTransports(original, overrides) {
   return Array.from(result.values());
 }
 function transportsByName(transports) {
-  const result = /* @__PURE__ */ new Map();
+  const result = new CaseInsensitiveMap();
   if (!transports) {
     return result;
   }
@@ -1353,6 +1408,23 @@ function mergeArrays(a1, a2) {
   }
   return [...a1 ?? [], ...a2 ?? []];
 }
+var CaseInsensitiveMap = class extends Map {
+  normalizeKey(key) {
+    return key.toUpperCase();
+  }
+  set(key, value) {
+    return super.set(this.normalizeKey(key), value);
+  }
+  get(key) {
+    return super.get(this.normalizeKey(key));
+  }
+  has(key) {
+    return super.has(this.normalizeKey(key));
+  }
+  delete(key) {
+    return super.delete(this.normalizeKey(key));
+  }
+};
 
 // src/client/service-parameters.ts
 var ServiceParameters = {