@adcp/sdk 13.0.0-rc.7 → 13.0.0-rc.8

package/dist/lib/core/SingleAgentClient.js

@@ -45,6 +45,7 @@ var import_idempotency = require('../utils/idempotency.js');
 var import_client = require("@a2a-js/sdk/client");
 var import_TaskExecutor = require('./TaskExecutor.js');
 var import_match = require('./match.js');
+var import_task_deadline = require('./task-deadline.js');
 var import_auth = require('../auth/index.js');
 var import_abort = require('../protocols/abort.js');
 var import_errors = require('../errors/index.js');
@@ -890,7 +891,8 @@ class SingleAgentClient {
       await ensureClientCredentialsTokens(this.normalizedAgent, {
         storage: getAgentStorage(this.normalizedAgent),
         allowPrivateIp: (0, import_net.isLikelyPrivateUrl)(this.normalizedAgent.agent_uri),
-        fetch: transport?.fetchFn
+        fetch: transport?.fetchFn,
+        signal: options?.signal
       });
     }
     const discoveredEndpoint = await this.discoverMCPEndpoint(this.normalizedAgent.agent_uri, options);
@@ -939,7 +941,8 @@ class SingleAgentClient {
       await ensureClientCredentialsTokens(this.normalizedAgent, {
         storage: getAgentStorage(this.normalizedAgent),
         allowPrivateIp: (0, import_net.isLikelyPrivateUrl)(this.normalizedAgent.agent_uri),
-        fetch: transport?.fetchFn
+        fetch: transport?.fetchFn,
+        signal: options?.signal
       });
     }
     const canonicalUrl = await this.fetchA2ACanonicalUrl(this.normalizedAgent.agent_uri, options);
@@ -1013,16 +1016,21 @@ class SingleAgentClient {
       if ((0, import_errors.is401Error)(error, got401)) {
         const requirements = await (0, import_authorization_required.discoverAuthorizationRequirements)(agentUri, {
           allowPrivateIp: (0, import_net.isLikelyPrivateUrl)(agentUri),
-          fetchFn: transport?.fetchFn
+          fetchFn: transport?.fetchFn,
+          signal: readOptions?.signal
         });
         if (requirements) {
           throw new import_authorization_required.NeedsAuthorizationError(requirements);
         }
         const challenge = await (0, import_authorization_required.probeAuthChallenge)(agentUri, {
           allowPrivateIp: (0, import_net.isLikelyPrivateUrl)(agentUri),
-          fetchFn: transport?.fetchFn
+          fetchFn: transport?.fetchFn,
+          signal: readOptions?.signal
+        });
+        const oauthMetadata = await (0, import_discovery.discoverOAuthMetadata)(agentUri, {
+          fetch: transport?.fetchFn,
+          signal: readOptions?.signal
         });
-        const oauthMetadata = await (0, import_discovery.discoverOAuthMetadata)(agentUri, { fetch: transport?.fetchFn });
         throw new import_errors.AuthenticationRequiredError(agentUri, oauthMetadata || void 0, void 0, challenge ?? void 0);
       }
       throw error;
@@ -1146,17 +1154,20 @@ class SingleAgentClient {
     if (got401) {
       const requirements = await (0, import_authorization_required.discoverAuthorizationRequirements)(providedUri, {
         allowPrivateIp: (0, import_net.isLikelyPrivateUrl)(providedUri),
-        fetchFn: options?.transport?.fetchFn ?? this.config.transport?.fetchFn
+        fetchFn: options?.transport?.fetchFn ?? this.config.transport?.fetchFn,
+        signal: options?.signal
       });
       if (requirements) {
         throw new import_authorization_required.NeedsAuthorizationError(requirements);
       }
       const challenge = await (0, import_authorization_required.probeAuthChallenge)(providedUri, {
         allowPrivateIp: (0, import_net.isLikelyPrivateUrl)(providedUri),
-        fetchFn: options?.transport?.fetchFn ?? this.config.transport?.fetchFn
+        fetchFn: options?.transport?.fetchFn ?? this.config.transport?.fetchFn,
+        signal: options?.signal
       });
       const oauthMetadata = await (0, import_discovery.discoverOAuthMetadata)(providedUri, {
-        fetch: options?.transport?.fetchFn ?? this.config.transport?.fetchFn
+        fetch: options?.transport?.fetchFn ?? this.config.transport?.fetchFn,
+        signal: options?.signal
       });
       throw new import_errors.AuthenticationRequiredError(providedUri, oauthMetadata || void 0, void 0, challenge ?? void 0);
     }
@@ -1760,6 +1771,21 @@ Hint: this usually means agent_uri does not include the MCP endpoint path. The S
    * Automatically adapts requests for v2 servers and normalizes responses.
    */
   async executeAndHandle(taskType, handlerName, params, inputHandler, options, transformCompletedResponse, legacyFormatConverter, canonicalRequest) {
+    return (0, import_task_deadline.withTaskDeadline)(
+      options,
+      (effectiveOptions) => this.executeAndHandleWithinDeadline(
+        taskType,
+        handlerName,
+        params,
+        inputHandler,
+        effectiveOptions,
+        transformCompletedResponse,
+        legacyFormatConverter,
+        canonicalRequest
+      )
+    );
+  }
+  async executeAndHandleWithinDeadline(taskType, handlerName, params, inputHandler, options, transformCompletedResponse, legacyFormatConverter, canonicalRequest) {
     (0, import_abort2.throwIfAborted)(options?.signal);
     let normalizedParams = (0, import_request_normalizer.normalizeRequestParams)(taskType, params, {
       skipIdempotencyAutoInject: options?.skipIdempotencyAutoInject,
@@ -1777,14 +1803,18 @@ Hint: this usually means agent_uri does not include the MCP endpoint path. The S
       this.validateRequest(taskType, normalizedParams);
     }
     await this.validateTaskFeatures(taskType, options);
+    (0, import_abort2.throwIfAborted)(options?.signal);
     if (this.config.requireV3ForMutations && (0, import_idempotency.isMutatingTask)(taskType)) {
       await this.requireSupportedMajor(taskType, options);
+      (0, import_abort2.throwIfAborted)(options?.signal);
     }
     const earlyResult = await this.getEarlyResultForUnsupportedFeatures(taskType, normalizedParams, options);
+    (0, import_abort2.throwIfAborted)(options?.signal);
     if (earlyResult) {
       return (0, import_match.attachMatch)(earlyResult);
     }
     const agent = await this.ensureEndpointDiscovered(options);
+    (0, import_abort2.throwIfAborted)(options?.signal);
     if (!options?.skipIdempotencyAutoInject && !options?.skipAccountValidation) {
       this.executor.validateRequest(taskType, normalizedParams);
     }
@@ -1794,6 +1824,7 @@ Hint: this usually means agent_uri does not include the MCP endpoint path. The S
       [CAPABILITY_DISCOVERY_CONTEXT]: capabilityDiscoveryContext
     };
     const serverVersion = await this.detectServerVersion(detectionOptions);
+    (0, import_abort2.throwIfAborted)(options?.signal);
     const inputSchemaStripLogs = [];
     const { params: adaptedParams, driftLogs: adaptDriftLogs } = this.adaptRequest(
       taskType,
@@ -1819,6 +1850,7 @@ Hint: this usually means agent_uri does not include the MCP endpoint path. The S
         serverVersion
       )
     );
+    (0, import_abort2.throwIfAborted)(effectiveOptions?.signal);
     const postAdapterLogs = [...inputSchemaStripLogs, ...v25DriftLogs];
     if (postAdapterLogs.length > 0) {
       result.debug_logs = [...result.debug_logs ?? [], ...postAdapterLogs];
@@ -1832,6 +1864,7 @@ Hint: this usually means agent_uri does not include the MCP endpoint path. The S
     }
     this.rememberLegacyFormatConverter(taskType, legacyFormatConverter, result, options);
     result = await this.applyProductPropertyPolicy(result, taskType, normalizedParams);
+    (0, import_abort2.throwIfAborted)(effectiveOptions?.signal);
     if (CANONICAL_CREATIVE_ACTIVITY_TASKS.has(taskType)) {
       const routingSnapshot = canonicalCreativeRoutingSnapshot(taskType, canonicalRequest ?? normalizedParams);
       result = this.canonicalizeCreativeTaskResult(result, taskType, transformCompletedResponse, legacyFormatConverter);
@@ -1849,6 +1882,7 @@ Hint: this usually means agent_uri does not include the MCP endpoint path. The S
     if (result.status === "completed" && result.success && this.asyncHandler) {
       const handler = this.config.handlers?.[handlerName];
       if (handler) {
+        (0, import_abort2.throwIfAborted)(effectiveOptions?.signal);
         const metadata = {
           operation_id: options?.contextId || "sync",
           context_id: options?.contextId,
@@ -1859,6 +1893,7 @@ Hint: this usually means agent_uri does not include the MCP endpoint path. The S
           timestamp: (/* @__PURE__ */ new Date()).toISOString()
         };
         await handler(result.data, metadata);
+        (0, import_abort2.throwIfAborted)(effectiveOptions?.signal);
       }
     }
     return result;
@@ -2739,6 +2774,12 @@ Hint: this usually means agent_uri does not include the MCP endpoint path. The S
    * @param options - Task execution options
    */
   async createMediaBuy(params, inputHandler, options) {
+    return (0, import_task_deadline.withTaskDeadline)(
+      options,
+      (effectiveOptions) => this.createMediaBuyWithinDeadline(params, inputHandler, effectiveOptions)
+    );
+  }
+  async createMediaBuyWithinDeadline(params, inputHandler, options) {
     const { legacyFormatConverter, projectionCatalogs, canonicalFormatLegacyResolver, ...taskOptions } = options ?? {};
     const effectiveLegacyFormatConverter = this.resolveLegacyFormatConverter(
       legacyFormatConverter,
@@ -2829,6 +2870,12 @@ Hint: this usually means agent_uri does not include the MCP endpoint path. The S
    * @param options - Task execution options
    */
   async updateMediaBuy(params, inputHandler, options) {
+    return (0, import_task_deadline.withTaskDeadline)(
+      options,
+      (effectiveOptions) => this.updateMediaBuyWithinDeadline(params, inputHandler, effectiveOptions)
+    );
+  }
+  async updateMediaBuyWithinDeadline(params, inputHandler, options) {
     const { legacyFormatConverter, projectionCatalogs, canonicalFormatLegacyResolver, ...taskOptions } = options ?? {};
     const effectiveLegacyFormatConverter = this.resolveLegacyFormatConverter(
       legacyFormatConverter,
@@ -2881,6 +2928,12 @@ Hint: this usually means agent_uri does not include the MCP endpoint path. The S
    * @param options - Task execution options
    */
   async syncCreatives(params, inputHandler, options) {
+    return (0, import_task_deadline.withTaskDeadline)(
+      options,
+      (effectiveOptions) => this.syncCreativesWithinDeadline(params, inputHandler, effectiveOptions)
+    );
+  }
+  async syncCreativesWithinDeadline(params, inputHandler, options) {
     const {
       creativeFormatProjection,
       legacyFormatConverter,
@@ -3167,14 +3220,25 @@ Hint: this usually means agent_uri does not include the MCP endpoint path. The S
    *
    * Use this when a task returned status 'submitted' or 'working' and
    * later resolves via polling or webhooks. The checkId is available
-   * on the original TaskResult at result.governance.checkId.
+   * on the original TaskResult at result.governance.checkId. After a
+   * TaskTimeoutError during governance postflight, pass the error's
+   * governanceRecovery.outcomeIdempotencyKey in `options` to safely retry.
    */
-  async reportGovernanceOutcome(checkId, outcome, governanceContext, sellerResponse, error) {
+  async reportGovernanceOutcome(checkId, outcome, governanceContext, sellerResponse, error, options) {
     const middleware = this.executor.getGovernanceMiddleware();
     if (!middleware) {
       throw new Error("No governance middleware configured. Set config.governance.campaign to enable governance.");
     }
-    return middleware.reportOutcome(checkId, outcome, sellerResponse, error, [], governanceContext);
+    return middleware.reportOutcome(
+      checkId,
+      outcome,
+      sellerResponse,
+      error,
+      [],
+      governanceContext,
+      options?.signal,
+      options?.outcomeIdempotencyKey
+    );
   }
   getGovernanceAgent() {
     const agent = this.config.governance?.campaign?.agent;
@@ -3194,6 +3258,12 @@ Hint: this usually means agent_uri does not include the MCP endpoint path. The S
    * @param options - Task execution options
    */
   async getAdcpCapabilities(params, inputHandler, options) {
+    return (0, import_task_deadline.withTaskDeadline)(
+      options,
+      (effectiveOptions) => this.getAdcpCapabilitiesWithinDeadline(params, inputHandler, effectiveOptions)
+    );
+  }
+  async getAdcpCapabilitiesWithinDeadline(params, inputHandler, options) {
     const agent = await this.ensureEndpointDiscovered(options);
     this.executor.validateRequest("get_adcp_capabilities", params);
     return this.executor.executeTask(
@@ -3469,6 +3539,12 @@ Hint: this usually means agent_uri does not include the MCP endpoint path. The S
     return this.executeTaskUnprojected(taskName, params, inputHandler, options);
   }
   async executeTaskUnprojected(taskName, params, inputHandler, options) {
+    return (0, import_task_deadline.withTaskDeadline)(
+      options,
+      (effectiveOptions) => this.executeTaskUnprojectedWithinDeadline(taskName, params, inputHandler, effectiveOptions)
+    );
+  }
+  async executeTaskUnprojectedWithinDeadline(taskName, params, inputHandler, options) {
     (0, import_abort2.throwIfAborted)(options?.signal);
     const startTime = Date.now();
     try {
@@ -3981,7 +4057,8 @@ Hint: this usually means agent_uri does not include the MCP endpoint path. The S
       await ensureClientCredentialsTokens(this.normalizedAgent, {
         storage: getAgentStorage(this.normalizedAgent),
         allowPrivateIp: (0, import_net.isLikelyPrivateUrl)(this.normalizedAgent.agent_uri),
-        fetch: transport?.fetchFn
+        fetch: transport?.fetchFn,
+        signal: options?.signal
       });
       return this.normalizedAgent.oauth_tokens?.access_token;
     };