@mastra/server 1.29.0-alpha.5 → 1.29.0-alpha.6

package/dist/server/a2a/store.cjs

@@ -1,22 +1,101 @@
 'use strict';
 
 // src/server/a2a/store.ts
+function createAbortError() {
+  return new DOMException("The operation was aborted.", "AbortError");
+}
 var InMemoryTaskStore = class {
   store = /* @__PURE__ */ new Map();
+  versions = /* @__PURE__ */ new Map();
+  listeners = /* @__PURE__ */ new Map();
   activeCancellations = /* @__PURE__ */ new Set();
+  getKey(agentId, taskId) {
+    return `${agentId}-${taskId}`;
+  }
   async load({ agentId, taskId }) {
-    const entry = this.store.get(`${agentId}-${taskId}`);
+    const snapshot = this.loadWithVersion({ agentId, taskId });
+    if (!snapshot) {
+      return null;
+    }
+    return snapshot.task;
+  }
+  loadWithVersion({ agentId, taskId }) {
+    const key = this.getKey(agentId, taskId);
+    const entry = this.store.get(key);
     if (!entry) {
       return null;
     }
-    return { ...entry };
+    return {
+      task: { ...entry },
+      version: this.versions.get(key) ?? 0
+    };
   }
   async save({ agentId, data }) {
-    const key = `${agentId}-${data.id}`;
+    const key = this.getKey(agentId, data.id);
     if (!data.id) {
       throw new Error("Task ID is required");
     }
-    this.store.set(key, { ...data });
+    const storedTask = { ...data };
+    const nextVersion = (this.versions.get(key) ?? 0) + 1;
+    this.store.set(key, storedTask);
+    this.versions.set(key, nextVersion);
+    const listeners = this.listeners.get(key);
+    if (listeners) {
+      for (const listener of listeners) {
+        listener({ task: { ...storedTask }, version: nextVersion });
+      }
+    }
+  }
+  getVersion({ agentId, taskId }) {
+    return this.versions.get(this.getKey(agentId, taskId)) ?? 0;
+  }
+  async waitForNextUpdate({
+    agentId,
+    taskId,
+    afterVersion,
+    signal
+  }) {
+    const key = this.getKey(agentId, taskId);
+    const snapshot = this.loadWithVersion({ agentId, taskId });
+    if (snapshot && snapshot.version > afterVersion) {
+      return snapshot;
+    }
+    if (signal?.aborted) {
+      return Promise.reject(createAbortError());
+    }
+    return new Promise((resolve, reject) => {
+      const listeners = this.listeners.get(key) ?? /* @__PURE__ */ new Set();
+      let settled = false;
+      const cleanup = () => {
+        listeners.delete(listener);
+        if (listeners.size === 0) {
+          this.listeners.delete(key);
+        }
+        signal?.removeEventListener("abort", onAbort);
+      };
+      const onAbort = () => {
+        if (settled) {
+          return;
+        }
+        settled = true;
+        cleanup();
+        reject(createAbortError());
+      };
+      const listener = (update) => {
+        if (settled || signal?.aborted) {
+          return;
+        }
+        if (update.version <= afterVersion) {
+          return;
+        }
+        settled = true;
+        cleanup();
+        resolve({ task: { ...update.task }, version: update.version });
+      };
+      listeners.add(listener);
+      this.listeners.set(key, listeners);
+      signal?.addEventListener("abort", onAbort, { once: true });
+    });
   }
 };
 

package/dist/server/a2a/store.cjs.map

@@ -1 +1 @@
-{"version":3,"sources":["../../../src/server/a2a/store.ts"],"names":[],"mappings":";;;AAEO,IAAM,oBAAN,MAAwB;AAAA,EACrB,KAAA,uBAA+B,GAAA,EAAI;AAAA,EACpC,mBAAA,uBAA0B,GAAA,EAAY;AAAA,EAE7C,MAAM,IAAA,CAAK,EAAE,OAAA,EAAS,QAAO,EAA8D;AACzF,IAAA,MAAM,KAAA,GAAQ,KAAK,KAAA,CAAM,GAAA,CAAI,GAAG,OAAO,CAAA,CAAA,EAAI,MAAM,CAAA,CAAE,CAAA;AAEnD,IAAA,IAAI,CAAC,KAAA,EAAO;AACV,MAAA,OAAO,IAAA;AAAA,IACT;AAGA,IAAA,OAAO,EAAE,GAAG,KAAA,EAAM;AAAA,EACpB;AAAA,EAEA,MAAM,IAAA,CAAK,EAAE,OAAA,EAAS,MAAK,EAAmD;AAE5E,IAAA,MAAM,GAAA,GAAM,CAAA,EAAG,OAAO,CAAA,CAAA,EAAI,KAAK,EAAE,CAAA,CAAA;AACjC,IAAA,IAAI,CAAC,KAAK,EAAA,EAAI;AACZ,MAAA,MAAM,IAAI,MAAM,qBAAqB,CAAA;AAAA,IACvC;AACA,IAAA,IAAA,CAAK,MAAM,GAAA,CAAI,GAAA,EAAK,EAAE,GAAG,MAAM,CAAA;AAAA,EACjC;AACF","file":"store.cjs","sourcesContent":["import type { Task } from '@mastra/core/a2a';\n\nexport class InMemoryTaskStore {\n  private store: Map<string, Task> = new Map();\n  public activeCancellations = new Set<string>();\n\n  async load({ agentId, taskId }: { agentId: string; taskId: string }): Promise<Task | null> {\n    const entry = this.store.get(`${agentId}-${taskId}`);\n\n    if (!entry) {\n      return null;\n    }\n\n    // Return copies to prevent external mutation\n    return { ...entry };\n  }\n\n  async save({ agentId, data }: { agentId: string; data: Task }): Promise<void> {\n    // Store copies to prevent internal mutation if caller reuses objects\n    const key = `${agentId}-${data.id}`;\n    if (!data.id) {\n      throw new Error('Task ID is required');\n    }\n    this.store.set(key, { ...data });\n  }\n}\n"]}
+{"version":3,"sources":["../../../src/server/a2a/store.ts"],"names":[],"mappings":";;;AAEA,SAAS,gBAAA,GAAmB;AAC1B,EAAA,OAAO,IAAI,YAAA,CAAa,4BAAA,EAA8B,YAAY,CAAA;AACpE;AAEO,IAAM,oBAAN,MAAwB;AAAA,EACrB,KAAA,uBAA+B,GAAA,EAAI;AAAA,EACnC,QAAA,uBAAoC,GAAA,EAAI;AAAA,EACxC,SAAA,uBAAqF,GAAA,EAAI;AAAA,EAC1F,mBAAA,uBAA0B,GAAA,EAAY;AAAA,EAErC,MAAA,CAAO,SAAiB,MAAA,EAAgB;AAC9C,IAAA,OAAO,CAAA,EAAG,OAAO,CAAA,CAAA,EAAI,MAAM,CAAA,CAAA;AAAA,EAC7B;AAAA,EAEA,MAAM,IAAA,CAAK,EAAE,OAAA,EAAS,QAAO,EAA8D;AACzF,IAAA,MAAM,WAAW,IAAA,CAAK,eAAA,CAAgB,EAAE,OAAA,EAAS,QAAQ,CAAA;AAEzD,IAAA,IAAI,CAAC,QAAA,EAAU;AACb,MAAA,OAAO,IAAA;AAAA,IACT;AAEA,IAAA,OAAO,QAAA,CAAS,IAAA;AAAA,EAClB;AAAA,EAEA,eAAA,CAAgB,EAAE,OAAA,EAAS,MAAA,EAAO,EAAgF;AAChH,IAAA,MAAM,GAAA,GAAM,IAAA,CAAK,MAAA,CAAO,OAAA,EAAS,MAAM,CAAA;AACvC,IAAA,MAAM,KAAA,GAAQ,IAAA,CAAK,KAAA,CAAM,GAAA,CAAI,GAAG,CAAA;AAEhC,IAAA,IAAI,CAAC,KAAA,EAAO;AACV,MAAA,OAAO,IAAA;AAAA,IACT;AAEA,IAAA,OAAO;AAAA,MACL,IAAA,EAAM,EAAE,GAAG,KAAA,EAAM;AAAA,MACjB,OAAA,EAAS,IAAA,CAAK,QAAA,CAAS,GAAA,CAAI,GAAG,CAAA,IAAK;AAAA,KACrC;AAAA,EACF;AAAA,EAEA,MAAM,IAAA,CAAK,EAAE,OAAA,EAAS,MAAK,EAAmD;AAE5E,IAAA,MAAM,GAAA,GAAM,IAAA,CAAK,MAAA,CAAO,OAAA,EAAS,KAAK,EAAE,CAAA;AACxC,IAAA,IAAI,CAAC,KAAK,EAAA,EAAI;AACZ,MAAA,MAAM,IAAI,MAAM,qBAAqB,CAAA;AAAA,IACvC;AAEA,IAAA,MAAM,UAAA,GAAa,EAAE,GAAG,IAAA,EAAK;AAC7B,IAAA,MAAM,eAAe,IAAA,CAAK,QAAA,CAAS,GAAA,CAAI,GAAG,KAAK,CAAA,IAAK,CAAA;AAEpD,IAAA,IAAA,CAAK,KAAA,CAAM,GAAA,CAAI,GAAA,EAAK,UAAU,CAAA;AAC9B,IAAA,IAAA,CAAK,QAAA,CAAS,GAAA,CAAI,GAAA,EAAK,WAAW,CAAA;AAElC,IAAA,MAAM,SAAA,GAAY,IAAA,CAAK,SAAA,CAAU,GAAA,CAAI,GAAG,CAAA;AACxC,IAAA,IAAI,SAAA,EAAW;AACb,MAAA,KAAA,MAAW,YAAY,SAAA,EAAW;AAChC,QAAA,QAAA,CAAS,EAAE,MAAM,EAAE,GAAG,YAAW,EAAG,OAAA,EAAS,aAAa,CAAA;AAAA,MAC5D;AAAA,IACF;AAAA,EACF;AAAA,EAEA,UAAA,CAAW,EAAE,OAAA,EAAS,MAAA,EAAO,EAAgD;AAC3E,IAAA,OAAO,IAAA,CAAK,SAAS,GAAA,CAAI,IAAA,CAAK,OAAO,OAAA,EAAS,MAAM,CAAC,CAAA,IAAK,CAAA;AAAA,EAC5D;AAAA,EAEA,MAAM,iBAAA,CAAkB;AAAA,IACtB,OAAA;AAAA,IACA,MAAA;AAAA,IACA,YAAA;AAAA,IACA;AAAA,GACF,EAK6C;AAC3C,IAAA,MAAM,GAAA,GAAM,IAAA,CAAK,MAAA,CAAO,OAAA,EAAS,MAAM,CAAA;AACvC,IAAA,MAAM,WAAW,IAAA,CAAK,eAAA,CAAgB,EAAE,OAAA,EAAS,QAAQ,CAAA;AAEzD,IAAA,IAAI,QAAA,IAAY,QAAA,CAAS,OAAA,GAAU,YAAA,EAAc;AAC/C,MAAA,OAAO,QAAA;AAAA,IACT;AAEA,IAAA,IAAI,QAAQ,OAAA,EAAS;AACnB,MAAA,OAAO,OAAA,CAAQ,MAAA,CAAO,gBAAA,EAAkB,CAAA;AAAA,IAC1C;AAEA,IAAA,OAAO,IAAI,OAAA,CAAQ,CAAC,OAAA,EAAS,MAAA,KAAW;AACtC,MAAA,MAAM,YAAY,IAAA,CAAK,SAAA,CAAU,IAAI,GAAG,CAAA,wBAAS,GAAA,EAAuD;AACxG,MAAA,IAAI,OAAA,GAAU,KAAA;AAEd,MAAA,MAAM,UAAU,MAAM;AACpB,QAAA,SAAA,CAAU,OAAO,QAAQ,CAAA;AACzB,QAAA,IAAI,SAAA,CAAU,SAAS,CAAA,EAAG;AACxB,UAAA,IAAA,CAAK,SAAA,CAAU,OAAO,GAAG,CAAA;AAAA,QAC3B;AACA,QAAA,MAAA,EAAQ,mBAAA,CAAoB,SAAS,OAAO,CAAA;AAAA,MAC9C,CAAA;AAEA,MAAA,MAAM,UAAU,MAAM;AACpB,QAAA,IAAI,OAAA,EAAS;AACX,UAAA;AAAA,QACF;AAEA,QAAA,OAAA,GAAU,IAAA;AACV,QAAA,OAAA,EAAQ;AACR,QAAA,MAAA,CAAO,kBAAkB,CAAA;AAAA,MAC3B,CAAA;AAEA,MAAA,MAAM,QAAA,GAAW,CAAC,MAAA,KAA4C;AAC5D,QAAA,IAAI,OAAA,IAAW,QAAQ,OAAA,EAAS;AAC9B,UAAA;AAAA,QACF;AAEA,QAAA,IAAI,MAAA,CAAO,WAAW,YAAA,EAAc;AAClC,UAAA;AAAA,QACF;AAEA,QAAA,OAAA,GAAU,IAAA;AACV,QAAA,OAAA,EAAQ;AAER,QAAA,OAAA,CAAQ,EAAE,IAAA,EAAM,EAAE,GAAG,MAAA,CAAO,MAAK,EAAG,OAAA,EAAS,MAAA,CAAO,OAAA,EAAS,CAAA;AAAA,MAC/D,CAAA;AAEA,MAAA,SAAA,CAAU,IAAI,QAAQ,CAAA;AACtB,MAAA,IAAA,CAAK,SAAA,CAAU,GAAA,CAAI,GAAA,EAAK,SAAS,CAAA;AACjC,MAAA,MAAA,EAAQ,iBAAiB,OAAA,EAAS,OAAA,EAAS,EAAE,IAAA,EAAM,MAAM,CAAA;AAAA,IAC3D,CAAC,CAAA;AAAA,EACH;AACF","file":"store.cjs","sourcesContent":["import type { Task } from '@mastra/core/a2a';\n\nfunction createAbortError() {\n  return new DOMException('The operation was aborted.', 'AbortError');\n}\n\nexport class InMemoryTaskStore {\n  private store: Map<string, Task> = new Map();\n  private versions: Map<string, number> = new Map();\n  private listeners: Map<string, Set<(update: { task: Task; version: number }) => void>> = new Map();\n  public activeCancellations = new Set<string>();\n\n  private getKey(agentId: string, taskId: string) {\n    return `${agentId}-${taskId}`;\n  }\n\n  async load({ agentId, taskId }: { agentId: string; taskId: string }): Promise<Task | null> {\n    const snapshot = this.loadWithVersion({ agentId, taskId });\n\n    if (!snapshot) {\n      return null;\n    }\n\n    return snapshot.task;\n  }\n\n  loadWithVersion({ agentId, taskId }: { agentId: string; taskId: string }): { task: Task; version: number } | null {\n    const key = this.getKey(agentId, taskId);\n    const entry = this.store.get(key);\n\n    if (!entry) {\n      return null;\n    }\n\n    return {\n      task: { ...entry },\n      version: this.versions.get(key) ?? 0,\n    };\n  }\n\n  async save({ agentId, data }: { agentId: string; data: Task }): Promise<void> {\n    // Store copies to prevent internal mutation if caller reuses objects\n    const key = this.getKey(agentId, data.id);\n    if (!data.id) {\n      throw new Error('Task ID is required');\n    }\n\n    const storedTask = { ...data };\n    const nextVersion = (this.versions.get(key) ?? 0) + 1;\n\n    this.store.set(key, storedTask);\n    this.versions.set(key, nextVersion);\n\n    const listeners = this.listeners.get(key);\n    if (listeners) {\n      for (const listener of listeners) {\n        listener({ task: { ...storedTask }, version: nextVersion });\n      }\n    }\n  }\n\n  getVersion({ agentId, taskId }: { agentId: string; taskId: string }): number {\n    return this.versions.get(this.getKey(agentId, taskId)) ?? 0;\n  }\n\n  async waitForNextUpdate({\n    agentId,\n    taskId,\n    afterVersion,\n    signal,\n  }: {\n    agentId: string;\n    taskId: string;\n    afterVersion: number;\n    signal?: AbortSignal;\n  }): Promise<{ task: Task; version: number }> {\n    const key = this.getKey(agentId, taskId);\n    const snapshot = this.loadWithVersion({ agentId, taskId });\n\n    if (snapshot && snapshot.version > afterVersion) {\n      return snapshot;\n    }\n\n    if (signal?.aborted) {\n      return Promise.reject(createAbortError());\n    }\n\n    return new Promise((resolve, reject) => {\n      const listeners = this.listeners.get(key) ?? new Set<(update: { task: Task; version: number }) => void>();\n      let settled = false;\n\n      const cleanup = () => {\n        listeners.delete(listener);\n        if (listeners.size === 0) {\n          this.listeners.delete(key);\n        }\n        signal?.removeEventListener('abort', onAbort);\n      };\n\n      const onAbort = () => {\n        if (settled) {\n          return;\n        }\n\n        settled = true;\n        cleanup();\n        reject(createAbortError());\n      };\n\n      const listener = (update: { task: Task; version: number }) => {\n        if (settled || signal?.aborted) {\n          return;\n        }\n\n        if (update.version <= afterVersion) {\n          return;\n        }\n\n        settled = true;\n        cleanup();\n\n        resolve({ task: { ...update.task }, version: update.version });\n      };\n\n      listeners.add(listener);\n      this.listeners.set(key, listeners);\n      signal?.addEventListener('abort', onAbort, { once: true });\n    });\n  }\n}\n"]}

package/dist/server/a2a/store.d.ts

@@ -1,14 +1,37 @@
 import type { Task } from '@mastra/core/a2a';
 export declare class InMemoryTaskStore {
     private store;
+    private versions;
+    private listeners;
     activeCancellations: Set<string>;
+    private getKey;
     load({ agentId, taskId }: {
         agentId: string;
         taskId: string;
     }): Promise<Task | null>;
+    loadWithVersion({ agentId, taskId }: {
+        agentId: string;
+        taskId: string;
+    }): {
+        task: Task;
+        version: number;
+    } | null;
     save({ agentId, data }: {
         agentId: string;
         data: Task;
     }): Promise<void>;
+    getVersion({ agentId, taskId }: {
+        agentId: string;
+        taskId: string;
+    }): number;
+    waitForNextUpdate({ agentId, taskId, afterVersion, signal, }: {
+        agentId: string;
+        taskId: string;
+        afterVersion: number;
+        signal?: AbortSignal;
+    }): Promise<{
+        task: Task;
+        version: number;
+    }>;
 }
 //# sourceMappingURL=store.d.ts.map

package/dist/server/a2a/store.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"store.d.ts","sourceRoot":"","sources":["../../../src/server/a2a/store.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,IAAI,EAAE,MAAM,kBAAkB,CAAC;AAE7C,qBAAa,iBAAiB;IAC5B,OAAO,CAAC,KAAK,CAAgC;IACtC,mBAAmB,cAAqB;IAEzC,IAAI,CAAC,EAAE,OAAO,EAAE,MAAM,EAAE,EAAE;QAAE,OAAO,EAAE,MAAM,CAAC;QAAC,MAAM,EAAE,MAAM,CAAA;KAAE,GAAG,OAAO,CAAC,IAAI,GAAG,IAAI,CAAC;IAWpF,IAAI,CAAC,EAAE,OAAO,EAAE,IAAI,EAAE,EAAE;QAAE,OAAO,EAAE,MAAM,CAAC;QAAC,IAAI,EAAE,IAAI,CAAA;KAAE,GAAG,OAAO,CAAC,IAAI,CAAC;CAQ9E"}
+{"version":3,"file":"store.d.ts","sourceRoot":"","sources":["../../../src/server/a2a/store.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,IAAI,EAAE,MAAM,kBAAkB,CAAC;AAM7C,qBAAa,iBAAiB;IAC5B,OAAO,CAAC,KAAK,CAAgC;IAC7C,OAAO,CAAC,QAAQ,CAAkC;IAClD,OAAO,CAAC,SAAS,CAAkF;IAC5F,mBAAmB,cAAqB;IAE/C,OAAO,CAAC,MAAM;IAIR,IAAI,CAAC,EAAE,OAAO,EAAE,MAAM,EAAE,EAAE;QAAE,OAAO,EAAE,MAAM,CAAC;QAAC,MAAM,EAAE,MAAM,CAAA;KAAE,GAAG,OAAO,CAAC,IAAI,GAAG,IAAI,CAAC;IAU1F,eAAe,CAAC,EAAE,OAAO,EAAE,MAAM,EAAE,EAAE;QAAE,OAAO,EAAE,MAAM,CAAC;QAAC,MAAM,EAAE,MAAM,CAAA;KAAE,GAAG;QAAE,IAAI,EAAE,IAAI,CAAC;QAAC,OAAO,EAAE,MAAM,CAAA;KAAE,GAAG,IAAI;IAc3G,IAAI,CAAC,EAAE,OAAO,EAAE,IAAI,EAAE,EAAE;QAAE,OAAO,EAAE,MAAM,CAAC;QAAC,IAAI,EAAE,IAAI,CAAA;KAAE,GAAG,OAAO,CAAC,IAAI,CAAC;IAqB7E,UAAU,CAAC,EAAE,OAAO,EAAE,MAAM,EAAE,EAAE;QAAE,OAAO,EAAE,MAAM,CAAC;QAAC,MAAM,EAAE,MAAM,CAAA;KAAE,GAAG,MAAM;IAItE,iBAAiB,CAAC,EACtB,OAAO,EACP,MAAM,EACN,YAAY,EACZ,MAAM,GACP,EAAE;QACD,OAAO,EAAE,MAAM,CAAC;QAChB,MAAM,EAAE,MAAM,CAAC;QACf,YAAY,EAAE,MAAM,CAAC;QACrB,MAAM,CAAC,EAAE,WAAW,CAAC;KACtB,GAAG,OAAO,CAAC;QAAE,IAAI,EAAE,IAAI,CAAC;QAAC,OAAO,EAAE,MAAM,CAAA;KAAE,CAAC;CAsD7C"}

package/dist/server/a2a/store.js

@@ -1,20 +1,99 @@
 // src/server/a2a/store.ts
+function createAbortError() {
+  return new DOMException("The operation was aborted.", "AbortError");
+}
 var InMemoryTaskStore = class {
   store = /* @__PURE__ */ new Map();
+  versions = /* @__PURE__ */ new Map();
+  listeners = /* @__PURE__ */ new Map();
   activeCancellations = /* @__PURE__ */ new Set();
+  getKey(agentId, taskId) {
+    return `${agentId}-${taskId}`;
+  }
   async load({ agentId, taskId }) {
-    const entry = this.store.get(`${agentId}-${taskId}`);
+    const snapshot = this.loadWithVersion({ agentId, taskId });
+    if (!snapshot) {
+      return null;
+    }
+    return snapshot.task;
+  }
+  loadWithVersion({ agentId, taskId }) {
+    const key = this.getKey(agentId, taskId);
+    const entry = this.store.get(key);
     if (!entry) {
       return null;
     }
-    return { ...entry };
+    return {
+      task: { ...entry },
+      version: this.versions.get(key) ?? 0
+    };
   }
   async save({ agentId, data }) {
-    const key = `${agentId}-${data.id}`;
+    const key = this.getKey(agentId, data.id);
     if (!data.id) {
       throw new Error("Task ID is required");
     }
-    this.store.set(key, { ...data });
+    const storedTask = { ...data };
+    const nextVersion = (this.versions.get(key) ?? 0) + 1;
+    this.store.set(key, storedTask);
+    this.versions.set(key, nextVersion);
+    const listeners = this.listeners.get(key);
+    if (listeners) {
+      for (const listener of listeners) {
+        listener({ task: { ...storedTask }, version: nextVersion });
+      }
+    }
+  }
+  getVersion({ agentId, taskId }) {
+    return this.versions.get(this.getKey(agentId, taskId)) ?? 0;
+  }
+  async waitForNextUpdate({
+    agentId,
+    taskId,
+    afterVersion,
+    signal
+  }) {
+    const key = this.getKey(agentId, taskId);
+    const snapshot = this.loadWithVersion({ agentId, taskId });
+    if (snapshot && snapshot.version > afterVersion) {
+      return snapshot;
+    }
+    if (signal?.aborted) {
+      return Promise.reject(createAbortError());
+    }
+    return new Promise((resolve, reject) => {
+      const listeners = this.listeners.get(key) ?? /* @__PURE__ */ new Set();
+      let settled = false;
+      const cleanup = () => {
+        listeners.delete(listener);
+        if (listeners.size === 0) {
+          this.listeners.delete(key);
+        }
+        signal?.removeEventListener("abort", onAbort);
+      };
+      const onAbort = () => {
+        if (settled) {
+          return;
+        }
+        settled = true;
+        cleanup();
+        reject(createAbortError());
+      };
+      const listener = (update) => {
+        if (settled || signal?.aborted) {
+          return;
+        }
+        if (update.version <= afterVersion) {
+          return;
+        }
+        settled = true;
+        cleanup();
+        resolve({ task: { ...update.task }, version: update.version });
+      };
+      listeners.add(listener);
+      this.listeners.set(key, listeners);
+      signal?.addEventListener("abort", onAbort, { once: true });
+    });
   }
 };
 

package/dist/server/a2a/store.js.map

@@ -1 +1 @@
-{"version":3,"sources":["../../../src/server/a2a/store.ts"],"names":[],"mappings":";AAEO,IAAM,oBAAN,MAAwB;AAAA,EACrB,KAAA,uBAA+B,GAAA,EAAI;AAAA,EACpC,mBAAA,uBAA0B,GAAA,EAAY;AAAA,EAE7C,MAAM,IAAA,CAAK,EAAE,OAAA,EAAS,QAAO,EAA8D;AACzF,IAAA,MAAM,KAAA,GAAQ,KAAK,KAAA,CAAM,GAAA,CAAI,GAAG,OAAO,CAAA,CAAA,EAAI,MAAM,CAAA,CAAE,CAAA;AAEnD,IAAA,IAAI,CAAC,KAAA,EAAO;AACV,MAAA,OAAO,IAAA;AAAA,IACT;AAGA,IAAA,OAAO,EAAE,GAAG,KAAA,EAAM;AAAA,EACpB;AAAA,EAEA,MAAM,IAAA,CAAK,EAAE,OAAA,EAAS,MAAK,EAAmD;AAE5E,IAAA,MAAM,GAAA,GAAM,CAAA,EAAG,OAAO,CAAA,CAAA,EAAI,KAAK,EAAE,CAAA,CAAA;AACjC,IAAA,IAAI,CAAC,KAAK,EAAA,EAAI;AACZ,MAAA,MAAM,IAAI,MAAM,qBAAqB,CAAA;AAAA,IACvC;AACA,IAAA,IAAA,CAAK,MAAM,GAAA,CAAI,GAAA,EAAK,EAAE,GAAG,MAAM,CAAA;AAAA,EACjC;AACF","file":"store.js","sourcesContent":["import type { Task } from '@mastra/core/a2a';\n\nexport class InMemoryTaskStore {\n  private store: Map<string, Task> = new Map();\n  public activeCancellations = new Set<string>();\n\n  async load({ agentId, taskId }: { agentId: string; taskId: string }): Promise<Task | null> {\n    const entry = this.store.get(`${agentId}-${taskId}`);\n\n    if (!entry) {\n      return null;\n    }\n\n    // Return copies to prevent external mutation\n    return { ...entry };\n  }\n\n  async save({ agentId, data }: { agentId: string; data: Task }): Promise<void> {\n    // Store copies to prevent internal mutation if caller reuses objects\n    const key = `${agentId}-${data.id}`;\n    if (!data.id) {\n      throw new Error('Task ID is required');\n    }\n    this.store.set(key, { ...data });\n  }\n}\n"]}
+{"version":3,"sources":["../../../src/server/a2a/store.ts"],"names":[],"mappings":";AAEA,SAAS,gBAAA,GAAmB;AAC1B,EAAA,OAAO,IAAI,YAAA,CAAa,4BAAA,EAA8B,YAAY,CAAA;AACpE;AAEO,IAAM,oBAAN,MAAwB;AAAA,EACrB,KAAA,uBAA+B,GAAA,EAAI;AAAA,EACnC,QAAA,uBAAoC,GAAA,EAAI;AAAA,EACxC,SAAA,uBAAqF,GAAA,EAAI;AAAA,EAC1F,mBAAA,uBAA0B,GAAA,EAAY;AAAA,EAErC,MAAA,CAAO,SAAiB,MAAA,EAAgB;AAC9C,IAAA,OAAO,CAAA,EAAG,OAAO,CAAA,CAAA,EAAI,MAAM,CAAA,CAAA;AAAA,EAC7B;AAAA,EAEA,MAAM,IAAA,CAAK,EAAE,OAAA,EAAS,QAAO,EAA8D;AACzF,IAAA,MAAM,WAAW,IAAA,CAAK,eAAA,CAAgB,EAAE,OAAA,EAAS,QAAQ,CAAA;AAEzD,IAAA,IAAI,CAAC,QAAA,EAAU;AACb,MAAA,OAAO,IAAA;AAAA,IACT;AAEA,IAAA,OAAO,QAAA,CAAS,IAAA;AAAA,EAClB;AAAA,EAEA,eAAA,CAAgB,EAAE,OAAA,EAAS,MAAA,EAAO,EAAgF;AAChH,IAAA,MAAM,GAAA,GAAM,IAAA,CAAK,MAAA,CAAO,OAAA,EAAS,MAAM,CAAA;AACvC,IAAA,MAAM,KAAA,GAAQ,IAAA,CAAK,KAAA,CAAM,GAAA,CAAI,GAAG,CAAA;AAEhC,IAAA,IAAI,CAAC,KAAA,EAAO;AACV,MAAA,OAAO,IAAA;AAAA,IACT;AAEA,IAAA,OAAO;AAAA,MACL,IAAA,EAAM,EAAE,GAAG,KAAA,EAAM;AAAA,MACjB,OAAA,EAAS,IAAA,CAAK,QAAA,CAAS,GAAA,CAAI,GAAG,CAAA,IAAK;AAAA,KACrC;AAAA,EACF;AAAA,EAEA,MAAM,IAAA,CAAK,EAAE,OAAA,EAAS,MAAK,EAAmD;AAE5E,IAAA,MAAM,GAAA,GAAM,IAAA,CAAK,MAAA,CAAO,OAAA,EAAS,KAAK,EAAE,CAAA;AACxC,IAAA,IAAI,CAAC,KAAK,EAAA,EAAI;AACZ,MAAA,MAAM,IAAI,MAAM,qBAAqB,CAAA;AAAA,IACvC;AAEA,IAAA,MAAM,UAAA,GAAa,EAAE,GAAG,IAAA,EAAK;AAC7B,IAAA,MAAM,eAAe,IAAA,CAAK,QAAA,CAAS,GAAA,CAAI,GAAG,KAAK,CAAA,IAAK,CAAA;AAEpD,IAAA,IAAA,CAAK,KAAA,CAAM,GAAA,CAAI,GAAA,EAAK,UAAU,CAAA;AAC9B,IAAA,IAAA,CAAK,QAAA,CAAS,GAAA,CAAI,GAAA,EAAK,WAAW,CAAA;AAElC,IAAA,MAAM,SAAA,GAAY,IAAA,CAAK,SAAA,CAAU,GAAA,CAAI,GAAG,CAAA;AACxC,IAAA,IAAI,SAAA,EAAW;AACb,MAAA,KAAA,MAAW,YAAY,SAAA,EAAW;AAChC,QAAA,QAAA,CAAS,EAAE,MAAM,EAAE,GAAG,YAAW,EAAG,OAAA,EAAS,aAAa,CAAA;AAAA,MAC5D;AAAA,IACF;AAAA,EACF;AAAA,EAEA,UAAA,CAAW,EAAE,OAAA,EAAS,MAAA,EAAO,EAAgD;AAC3E,IAAA,OAAO,IAAA,CAAK,SAAS,GAAA,CAAI,IAAA,CAAK,OAAO,OAAA,EAAS,MAAM,CAAC,CAAA,IAAK,CAAA;AAAA,EAC5D;AAAA,EAEA,MAAM,iBAAA,CAAkB;AAAA,IACtB,OAAA;AAAA,IACA,MAAA;AAAA,IACA,YAAA;AAAA,IACA;AAAA,GACF,EAK6C;AAC3C,IAAA,MAAM,GAAA,GAAM,IAAA,CAAK,MAAA,CAAO,OAAA,EAAS,MAAM,CAAA;AACvC,IAAA,MAAM,WAAW,IAAA,CAAK,eAAA,CAAgB,EAAE,OAAA,EAAS,QAAQ,CAAA;AAEzD,IAAA,IAAI,QAAA,IAAY,QAAA,CAAS,OAAA,GAAU,YAAA,EAAc;AAC/C,MAAA,OAAO,QAAA;AAAA,IACT;AAEA,IAAA,IAAI,QAAQ,OAAA,EAAS;AACnB,MAAA,OAAO,OAAA,CAAQ,MAAA,CAAO,gBAAA,EAAkB,CAAA;AAAA,IAC1C;AAEA,IAAA,OAAO,IAAI,OAAA,CAAQ,CAAC,OAAA,EAAS,MAAA,KAAW;AACtC,MAAA,MAAM,YAAY,IAAA,CAAK,SAAA,CAAU,IAAI,GAAG,CAAA,wBAAS,GAAA,EAAuD;AACxG,MAAA,IAAI,OAAA,GAAU,KAAA;AAEd,MAAA,MAAM,UAAU,MAAM;AACpB,QAAA,SAAA,CAAU,OAAO,QAAQ,CAAA;AACzB,QAAA,IAAI,SAAA,CAAU,SAAS,CAAA,EAAG;AACxB,UAAA,IAAA,CAAK,SAAA,CAAU,OAAO,GAAG,CAAA;AAAA,QAC3B;AACA,QAAA,MAAA,EAAQ,mBAAA,CAAoB,SAAS,OAAO,CAAA;AAAA,MAC9C,CAAA;AAEA,MAAA,MAAM,UAAU,MAAM;AACpB,QAAA,IAAI,OAAA,EAAS;AACX,UAAA;AAAA,QACF;AAEA,QAAA,OAAA,GAAU,IAAA;AACV,QAAA,OAAA,EAAQ;AACR,QAAA,MAAA,CAAO,kBAAkB,CAAA;AAAA,MAC3B,CAAA;AAEA,MAAA,MAAM,QAAA,GAAW,CAAC,MAAA,KAA4C;AAC5D,QAAA,IAAI,OAAA,IAAW,QAAQ,OAAA,EAAS;AAC9B,UAAA;AAAA,QACF;AAEA,QAAA,IAAI,MAAA,CAAO,WAAW,YAAA,EAAc;AAClC,UAAA;AAAA,QACF;AAEA,QAAA,OAAA,GAAU,IAAA;AACV,QAAA,OAAA,EAAQ;AAER,QAAA,OAAA,CAAQ,EAAE,IAAA,EAAM,EAAE,GAAG,MAAA,CAAO,MAAK,EAAG,OAAA,EAAS,MAAA,CAAO,OAAA,EAAS,CAAA;AAAA,MAC/D,CAAA;AAEA,MAAA,SAAA,CAAU,IAAI,QAAQ,CAAA;AACtB,MAAA,IAAA,CAAK,SAAA,CAAU,GAAA,CAAI,GAAA,EAAK,SAAS,CAAA;AACjC,MAAA,MAAA,EAAQ,iBAAiB,OAAA,EAAS,OAAA,EAAS,EAAE,IAAA,EAAM,MAAM,CAAA;AAAA,IAC3D,CAAC,CAAA;AAAA,EACH;AACF","file":"store.js","sourcesContent":["import type { Task } from '@mastra/core/a2a';\n\nfunction createAbortError() {\n  return new DOMException('The operation was aborted.', 'AbortError');\n}\n\nexport class InMemoryTaskStore {\n  private store: Map<string, Task> = new Map();\n  private versions: Map<string, number> = new Map();\n  private listeners: Map<string, Set<(update: { task: Task; version: number }) => void>> = new Map();\n  public activeCancellations = new Set<string>();\n\n  private getKey(agentId: string, taskId: string) {\n    return `${agentId}-${taskId}`;\n  }\n\n  async load({ agentId, taskId }: { agentId: string; taskId: string }): Promise<Task | null> {\n    const snapshot = this.loadWithVersion({ agentId, taskId });\n\n    if (!snapshot) {\n      return null;\n    }\n\n    return snapshot.task;\n  }\n\n  loadWithVersion({ agentId, taskId }: { agentId: string; taskId: string }): { task: Task; version: number } | null {\n    const key = this.getKey(agentId, taskId);\n    const entry = this.store.get(key);\n\n    if (!entry) {\n      return null;\n    }\n\n    return {\n      task: { ...entry },\n      version: this.versions.get(key) ?? 0,\n    };\n  }\n\n  async save({ agentId, data }: { agentId: string; data: Task }): Promise<void> {\n    // Store copies to prevent internal mutation if caller reuses objects\n    const key = this.getKey(agentId, data.id);\n    if (!data.id) {\n      throw new Error('Task ID is required');\n    }\n\n    const storedTask = { ...data };\n    const nextVersion = (this.versions.get(key) ?? 0) + 1;\n\n    this.store.set(key, storedTask);\n    this.versions.set(key, nextVersion);\n\n    const listeners = this.listeners.get(key);\n    if (listeners) {\n      for (const listener of listeners) {\n        listener({ task: { ...storedTask }, version: nextVersion });\n      }\n    }\n  }\n\n  getVersion({ agentId, taskId }: { agentId: string; taskId: string }): number {\n    return this.versions.get(this.getKey(agentId, taskId)) ?? 0;\n  }\n\n  async waitForNextUpdate({\n    agentId,\n    taskId,\n    afterVersion,\n    signal,\n  }: {\n    agentId: string;\n    taskId: string;\n    afterVersion: number;\n    signal?: AbortSignal;\n  }): Promise<{ task: Task; version: number }> {\n    const key = this.getKey(agentId, taskId);\n    const snapshot = this.loadWithVersion({ agentId, taskId });\n\n    if (snapshot && snapshot.version > afterVersion) {\n      return snapshot;\n    }\n\n    if (signal?.aborted) {\n      return Promise.reject(createAbortError());\n    }\n\n    return new Promise((resolve, reject) => {\n      const listeners = this.listeners.get(key) ?? new Set<(update: { task: Task; version: number }) => void>();\n      let settled = false;\n\n      const cleanup = () => {\n        listeners.delete(listener);\n        if (listeners.size === 0) {\n          this.listeners.delete(key);\n        }\n        signal?.removeEventListener('abort', onAbort);\n      };\n\n      const onAbort = () => {\n        if (settled) {\n          return;\n        }\n\n        settled = true;\n        cleanup();\n        reject(createAbortError());\n      };\n\n      const listener = (update: { task: Task; version: number }) => {\n        if (settled || signal?.aborted) {\n          return;\n        }\n\n        if (update.version <= afterVersion) {\n          return;\n        }\n\n        settled = true;\n        cleanup();\n\n        resolve({ task: { ...update.task }, version: update.version });\n      };\n\n      listeners.add(listener);\n      this.listeners.set(key, listeners);\n      signal?.addEventListener('abort', onAbort, { once: true });\n    });\n  }\n}\n"]}

package/dist/server/handlers/a2a.cjs

@@ -1,40 +1,44 @@
 'use strict';
 
-var chunk6VT7WJZZ_cjs = require('../../chunk-6VT7WJZZ.cjs');
+var chunkSJGVGNXJ_cjs = require('../../chunk-SJGVGNXJ.cjs');
 
 
 
 Object.defineProperty(exports, "AGENT_EXECUTION_ROUTE", {
   enumerable: true,
-  get: function () { return chunk6VT7WJZZ_cjs.AGENT_EXECUTION_ROUTE; }
+  get: function () { return chunkSJGVGNXJ_cjs.AGENT_EXECUTION_ROUTE; }
 });
 Object.defineProperty(exports, "GET_AGENT_CARD_ROUTE", {
   enumerable: true,
-  get: function () { return chunk6VT7WJZZ_cjs.GET_AGENT_CARD_ROUTE; }
+  get: function () { return chunkSJGVGNXJ_cjs.GET_AGENT_CARD_ROUTE; }
 });
 Object.defineProperty(exports, "getAgentCardByIdHandler", {
   enumerable: true,
-  get: function () { return chunk6VT7WJZZ_cjs.getAgentCardByIdHandler; }
+  get: function () { return chunkSJGVGNXJ_cjs.getAgentCardByIdHandler; }
 });
 Object.defineProperty(exports, "getAgentExecutionHandler", {
   enumerable: true,
-  get: function () { return chunk6VT7WJZZ_cjs.getAgentExecutionHandler; }
+  get: function () { return chunkSJGVGNXJ_cjs.getAgentExecutionHandler; }
 });
 Object.defineProperty(exports, "handleMessageSend", {
   enumerable: true,
-  get: function () { return chunk6VT7WJZZ_cjs.handleMessageSend; }
+  get: function () { return chunkSJGVGNXJ_cjs.handleMessageSend; }
 });
 Object.defineProperty(exports, "handleMessageStream", {
   enumerable: true,
-  get: function () { return chunk6VT7WJZZ_cjs.handleMessageStream; }
+  get: function () { return chunkSJGVGNXJ_cjs.handleMessageStream; }
 });
 Object.defineProperty(exports, "handleTaskCancel", {
   enumerable: true,
-  get: function () { return chunk6VT7WJZZ_cjs.handleTaskCancel; }
+  get: function () { return chunkSJGVGNXJ_cjs.handleTaskCancel; }
 });
 Object.defineProperty(exports, "handleTaskGet", {
   enumerable: true,
-  get: function () { return chunk6VT7WJZZ_cjs.handleTaskGet; }
+  get: function () { return chunkSJGVGNXJ_cjs.handleTaskGet; }
+});
+Object.defineProperty(exports, "handleTaskResubscribe", {
+  enumerable: true,
+  get: function () { return chunkSJGVGNXJ_cjs.handleTaskResubscribe; }
 });
 //# sourceMappingURL=a2a.cjs.map
 //# sourceMappingURL=a2a.cjs.map