@astrasyncai/verification-gateway 5.15.0 → 5.17.0

package/dist/agent/index.mjs

@@ -1,5 +1,5 @@
 // src/version.ts
-var SDK_VERSION = "5.15.0";
+var SDK_VERSION = "5.17.0";
 
 // src/http.ts
 var SDK_USER_AGENT = `astrasync-sdk/${SDK_VERSION}`;
@@ -185,6 +185,7 @@ var OwnershipMismatchError = class extends AstraSyncSdkError {
 // src/agent/client.ts
 var AgentClient = class _AgentClient {
   constructor(config) {
+    this.challengeHandler = config.challengeHandler;
     this.credentials = {
       agentId: config.agentId,
       verifyUrl: config.verifyUrl ?? "https://astrasync.ai/api/agents/verify-access",
@@ -239,7 +240,7 @@ var AgentClient = class _AgentClient {
    * Make an HTTP request with AstraSync headers automatically injected.
    */
   async fetch(url, options) {
-    const { purpose, action, ...fetchOptions } = options ?? {};
+    const { purpose, action, counterpartyId, ...fetchOptions } = options ?? {};
     const creds = { ...this.credentials };
     if (purpose) {
       creds.pdlss = {
@@ -262,10 +263,16 @@ var AgentClient = class _AgentClient {
       }
     }
     const enrichedHeaders = setHttpHeaders(existingHeaders, creds);
-    return sdkFetch(url, {
-      ...fetchOptions,
-      headers: enrichedHeaders
-    });
+    const pending = this.challengeHandler ? [counterpartyId, new URL(url).origin].filter((v) => Boolean(v)) : [];
+    pending.forEach((p) => this.challengeHandler.registerPending(p));
+    try {
+      return await sdkFetch(url, {
+        ...fetchOptions,
+        headers: enrichedHeaders
+      });
+    } finally {
+      pending.forEach((p) => this.challengeHandler.removePending(p));
+    }
   }
   /**
    * Prepare A2A task metadata with AstraSync credentials.
@@ -303,26 +310,32 @@ var AgentClient = class _AgentClient {
 // src/agent/challenge-handler.ts
 var ChallengeHandler = class {
   constructor(config) {
-    this.pendingCounterparties = /* @__PURE__ */ new Set();
+    /** Counterparty → in-flight interactions (two concurrent calls to one counterparty both count). */
+    this.pendingCounterparties = /* @__PURE__ */ new Map();
     this.agentId = config.agentId;
   }
   /**
    * Register a counterparty as pending (before initiating contact).
    */
   registerPending(counterpartyId) {
-    this.pendingCounterparties.add(counterpartyId);
+    this.pendingCounterparties.set(
+      counterpartyId,
+      (this.pendingCounterparties.get(counterpartyId) ?? 0) + 1
+    );
   }
   /**
    * Remove a counterparty from pending list (after interaction complete).
    */
   removePending(counterpartyId) {
-    this.pendingCounterparties.delete(counterpartyId);
+    const n = (this.pendingCounterparties.get(counterpartyId) ?? 0) - 1;
+    if (n > 0) this.pendingCounterparties.set(counterpartyId, n);
+    else this.pendingCounterparties.delete(counterpartyId);
   }
   /**
    * Get current pending counterparties list.
    */
   getPendingList() {
-    return [...this.pendingCounterparties];
+    return [...this.pendingCounterparties.keys()];
   }
   /**
    * Express middleware for the challenge endpoint.
@@ -378,10 +391,7 @@ var ChallengeHandler = class {
         }
       };
     }
-    const explicitList = this.getPendingList();
-    const pendingCounterparties = explicitList.length > 0 ? explicitList : [payload.counterpartyId, payload.counterpartyUrl].filter(
-      (v) => typeof v === "string" && v.length > 0
-    );
+    const pendingCounterparties = this.getPendingList();
     return {
       status: 200,
       body: {