@mastra/mongodb 1.23.0 → 1.23.1-alpha.1

package/dist/storage/connectors/MongoDBConnector.d.ts

@@ -20,9 +20,16 @@ export declare class MongoDBConnector {
     getCollection(collectionName: string): Promise<import("mongodb").Collection<import("mongodb").Document>>;
     collectionExists(collectionName: string): Promise<boolean>;
     /**
-     * Returns true when the deployment supports multi-document transactions
-     * (replica set or sharded cluster). Standalone servers and custom connector
-     * handlers return false. Probed once and cached.
+     * Whether the deployment supports multi-document transactions (replica set
+     * or sharded cluster). Standalone servers and custom connector handlers
+     * don't. The answer is cached once a probe completes; a probe that fails
+     * rejects and is retried on the next call, so the answer never changes.
+     */
+    probeTransactions(): Promise<boolean>;
+    /**
+     * Like {@link probeTransactions}, but a failed probe reads as no
+     * transaction support (and is retried on the next call), so a momentary
+     * outage degrades writes to non-atomic instead of failing them.
      */
     supportsTransactions(): Promise<boolean>;
     /**

package/dist/storage/connectors/MongoDBConnector.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"MongoDBConnector.d.ts","sourceRoot":"","sources":["../../../src/storage/connectors/MongoDBConnector.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,WAAW,EAAE,MAAM,SAAS,CAAC;AACtC,OAAO,KAAK,EAAE,aAAa,EAAM,MAAM,SAAS,CAAC;AAEjD,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,UAAU,CAAC;AAC/C,OAAO,KAAK,EAAE,gBAAgB,EAAE,MAAM,QAAQ,CAAC;AAE/C,KAAK,uBAAuB,GACxB;IACE,MAAM,EAAE,WAAW,CAAC;IACpB,MAAM,EAAE,MAAM,CAAC;IACf,OAAO,EAAE,SAAS,CAAC;CACpB,GACD;IACE,MAAM,EAAE,SAAS,CAAC;IAClB,MAAM,EAAE,SAAS,CAAC;IAClB,OAAO,EAAE,gBAAgB,CAAC;CAC3B,CAAC;AAEN,qBAAa,gBAAgB;;IAQ3B,YAAY,OAAO,EAAE,uBAAuB,EAK3C;IAED,MAAM,CAAC,kBAAkB,CAAC,MAAM,EAAE,cAAc,GAAG,gBAAgB,CAyBlE;IAED,MAAM,CAAC,qBAAqB,CAAC,OAAO,EAAE,gBAAgB,GAAG,gBAAgB,CAMxE;YAEa,aAAa;IAcrB,aAAa,CAAC,cAAc,EAAE,MAAM,qEAMzC;IAEK,gBAAgB,CAAC,cAAc,EAAE,MAAM,GAAG,OAAO,CAAC,OAAO,CAAC,CAM/D;IAED;;;;OAIG;IACG,oBAAoB,IAAI,OAAO,CAAC,OAAO,CAAC,CAmB7C;IAED;;;;;OAKG;IACG,eAAe,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,OAAO,CAAC,EAAE,aAAa,KAAK,OAAO,CAAC,CAAC,CAAC,GAAG,OAAO,CAAC,CAAC,CAAC,CAehF;IAEK,KAAK,kBAUV;CACF"}
+{"version":3,"file":"MongoDBConnector.d.ts","sourceRoot":"","sources":["../../../src/storage/connectors/MongoDBConnector.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,WAAW,EAAE,MAAM,SAAS,CAAC;AACtC,OAAO,KAAK,EAAE,aAAa,EAAM,MAAM,SAAS,CAAC;AAEjD,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,UAAU,CAAC;AAC/C,OAAO,KAAK,EAAE,gBAAgB,EAAE,MAAM,QAAQ,CAAC;AAE/C,KAAK,uBAAuB,GACxB;IACE,MAAM,EAAE,WAAW,CAAC;IACpB,MAAM,EAAE,MAAM,CAAC;IACf,OAAO,EAAE,SAAS,CAAC;CACpB,GACD;IACE,MAAM,EAAE,SAAS,CAAC;IAClB,MAAM,EAAE,SAAS,CAAC;IAClB,OAAO,EAAE,gBAAgB,CAAC;CAC3B,CAAC;AAEN,qBAAa,gBAAgB;;IAQ3B,YAAY,OAAO,EAAE,uBAAuB,EAK3C;IAED,MAAM,CAAC,kBAAkB,CAAC,MAAM,EAAE,cAAc,GAAG,gBAAgB,CAyBlE;IAED,MAAM,CAAC,qBAAqB,CAAC,OAAO,EAAE,gBAAgB,GAAG,gBAAgB,CAMxE;YAEa,aAAa;IAcrB,aAAa,CAAC,cAAc,EAAE,MAAM,qEAMzC;IAEK,gBAAgB,CAAC,cAAc,EAAE,MAAM,GAAG,OAAO,CAAC,OAAO,CAAC,CAM/D;IAED;;;;;OAKG;IACG,iBAAiB,IAAI,OAAO,CAAC,OAAO,CAAC,CAa1C;IAED;;;;OAIG;IACG,oBAAoB,IAAI,OAAO,CAAC,OAAO,CAAC,CAM7C;IAED;;;;;OAKG;IACG,eAAe,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,OAAO,CAAC,EAAE,aAAa,KAAK,OAAO,CAAC,CAAC,CAAC,GAAG,OAAO,CAAC,CAAC,CAAC,CAehF;IAEK,KAAK,kBAUV;CACF"}

package/dist/storage/domains/memory/index.d.ts

@@ -1,7 +1,8 @@
+import { type UpdateActiveObservationsResult, type UpdateBufferedObservationsResult } from '../../../_types/@internal_core/dist/memory/index.js';
 import type { MastraMessageContentV2 } from '@mastra/core/agent';
 import type { MastraDBMessage, StorageThreadType } from '@mastra/core/memory';
 import { MemoryStorage } from '@mastra/core/storage';
-import type { PruneOptions, PruneResult, RetentionTablesDescriptor, TableRetentionPolicy, StorageResourceType, StorageListMessagesInput, StorageListMessagesByResourceIdInput, StorageListMessagesOutput, StorageListThreadsInput, StorageListThreadsOutput, StorageCloneThreadInput, StorageCloneThreadOutput, ObservationalMemoryRecord, ObservationalMemoryHistoryOptions, CreateObservationalMemoryInput, UpdateActiveObservationsInput, UpdateBufferedObservationsInput, SwapBufferedToActiveInput, SwapBufferedToActiveResult, UpdateBufferedReflectionInput, SwapBufferedReflectionToActiveInput, CreateReflectionGenerationInput, UpdateObservationalMemoryConfigInput } from '@mastra/core/storage';
+import type { PruneOptions, PruneResult, RetentionTablesDescriptor, TableRetentionPolicy, StorageResourceType, StorageListMessagesInput, StorageListMessagesByResourceIdInput, StorageListMessagesOutput, StorageListThreadsInput, StorageListThreadsOutput, StorageCloneThreadInput, StorageCloneThreadOutput, ObservationalMemoryRecord, ObservationalMemoryHistoryOptions, CreateObservationalMemoryInput, UpdateActiveObservationsInput, UpdateBufferedObservationsInput, SwapBufferedToActiveInput, SwapBufferedToActiveResult, UpdateBufferedReflectionInput, SwapBufferedReflectionToActiveInput, CreateReflectionGenerationInput, UpdateObservationalMemoryConfigInput, RunFence } from '@mastra/core/storage';
 import type { MongoDBDomainConfig, MongoDBIndexConfig } from '../../types.js';
 export declare class MemoryStorageMongoDB extends MemoryStorage {
     #private;
@@ -9,21 +10,31 @@ export declare class MemoryStorageMongoDB extends MemoryStorage {
     readonly supportsObservationalMemory = true;
     readonly supportsObservationalMemoryHistorySearch = true;
     /** Collections managed by this domain */
-    static readonly MANAGED_COLLECTIONS: readonly ["mastra_threads", "mastra_messages", "mastra_resources", "mastra_observational_memory"];
+    static readonly MANAGED_COLLECTIONS: readonly ["mastra_threads", "mastra_messages", "mastra_resources", "mastra_observational_memory", "mastra_memory_run_fences"];
     /**
      * Retention-eligible collections. The observational-memory collection is
      * excluded: it has no timestamp anchor to age on. All anchors are BSON dates.
+     * A run fence's `retiredAt` stays null until the execution holding it
+     * settles, and null is never pruned.
      */
     static readonly retentionTables: RetentionTablesDescriptor;
     constructor(config: MongoDBDomainConfig);
     private getCollection;
     init(): Promise<void>;
+    /**
+     * Fenced writes need multi-document transactions, so only replica sets and
+     * sharded clusters fence. Rejects while the deployment can't be probed.
+     */
+    supportsRunFencing(): Promise<boolean>;
+    raiseRunFence(fence: RunFence): Promise<boolean>;
+    retireRunFence(fence: RunFence): Promise<boolean>;
     /**
      * Delete memory rows older than each table's `maxAge`, batched. Order is
      * messages → resources → threads so child rows never outlive the delete of
-     * their thread. Like `deleteThread()`, this does not sweep vector-store
-     * embeddings — semantic-recall vectors live in a separate vector store the
-     * memory domain cannot reach; cleaning those up is the operator's concern.
+     * their thread; run fences are independent and go last. Like
+     * `deleteThread()`, this does not sweep vector-store embeddings —
+     * semantic-recall vectors live in a separate vector store the memory domain
+     * cannot reach; cleaning those up is the operator's concern.
      */
     prune(policies: Record<string, TableRetentionPolicy>, options?: PruneOptions): Promise<PruneResult[]>;
     /**
@@ -56,12 +67,13 @@ export declare class MemoryStorageMongoDB extends MemoryStorage {
     }>;
     listMessages(args: StorageListMessagesInput): Promise<StorageListMessagesOutput>;
     listMessagesByResourceId(args: StorageListMessagesByResourceIdInput): Promise<StorageListMessagesOutput>;
-    saveMessages({ messages }: {
+    saveMessages({ messages, fence, }: {
         messages: MastraDBMessage[];
+        fence?: RunFence;
     }): Promise<{
         messages: MastraDBMessage[];
     }>;
-    updateMessages({ messages, }: {
+    updateMessages({ messages, fence, }: {
         messages: (Partial<Omit<MastraDBMessage, 'createdAt'>> & {
             id: string;
             content?: {
@@ -69,6 +81,7 @@ export declare class MemoryStorageMongoDB extends MemoryStorage {
                 content?: MastraMessageContentV2['content'];
             };
         })[];
+        fence?: RunFence;
     }): Promise<MastraDBMessage[]>;
     getResourceById({ resourceId }: {
         resourceId: string;
@@ -76,28 +89,33 @@ export declare class MemoryStorageMongoDB extends MemoryStorage {
     saveResource({ resource }: {
         resource: StorageResourceType;
     }): Promise<StorageResourceType>;
-    updateResource({ resourceId, workingMemory, metadata, }: {
+    updateResource({ resourceId, workingMemory, metadata, fence, }: {
         resourceId: string;
         workingMemory?: string;
         metadata?: Record<string, unknown>;
+        fence?: RunFence;
     }): Promise<StorageResourceType>;
     getThreadById({ threadId, resourceId, }: {
         threadId: string;
         resourceId?: string;
     }): Promise<StorageThreadType | null>;
     listThreads(args: StorageListThreadsInput): Promise<StorageListThreadsOutput>;
-    saveThread({ thread }: {
+    saveThread({ thread, fence }: {
         thread: StorageThreadType;
+        fence?: RunFence;
     }): Promise<StorageThreadType>;
-    updateThread({ id, title, metadata, }: {
+    updateThread({ id, title, metadata, fence, }: {
         id: string;
         title?: string;
         metadata?: Record<string, unknown>;
+        fence?: RunFence;
     }): Promise<StorageThreadType>;
     deleteThread({ threadId }: {
         threadId: string;
     }): Promise<void>;
-    deleteMessages(messageIds: string[]): Promise<void>;
+    deleteMessages(messageIds: string[], options?: {
+        fence?: RunFence;
+    }): Promise<void>;
     cloneThread(args: StorageCloneThreadInput): Promise<StorageCloneThreadOutput>;
     private getOMKey;
     private parseOMDocument;
@@ -106,6 +124,7 @@ export declare class MemoryStorageMongoDB extends MemoryStorage {
     initializeObservationalMemory(input: CreateObservationalMemoryInput): Promise<ObservationalMemoryRecord>;
     insertObservationalMemoryRecord(record: ObservationalMemoryRecord): Promise<void>;
     updateActiveObservations(input: UpdateActiveObservationsInput): Promise<void>;
+    commitActiveObservations(input: UpdateActiveObservationsInput): Promise<UpdateActiveObservationsResult>;
     createReflectionGeneration(input: CreateReflectionGenerationInput): Promise<ObservationalMemoryRecord>;
     setReflectingFlag(id: string, isReflecting: boolean): Promise<void>;
     setObservingFlag(id: string, isObserving: boolean): Promise<void>;
@@ -115,6 +134,7 @@ export declare class MemoryStorageMongoDB extends MemoryStorage {
     setPendingMessageTokens(id: string, tokenCount: number): Promise<void>;
     updateObservationalMemoryConfig(input: UpdateObservationalMemoryConfigInput): Promise<void>;
     updateBufferedObservations(input: UpdateBufferedObservationsInput): Promise<void>;
+    appendBufferedObservations(input: UpdateBufferedObservationsInput): Promise<UpdateBufferedObservationsResult>;
     swapBufferedToActive(input: SwapBufferedToActiveInput): Promise<SwapBufferedToActiveResult>;
     updateBufferedReflection(input: UpdateBufferedReflectionInput): Promise<void>;
     swapBufferedReflectionToActive(input: SwapBufferedReflectionToActiveInput): Promise<ObservationalMemoryRecord>;

package/dist/storage/domains/memory/index.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../../src/storage/domains/memory/index.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,sBAAsB,EAAE,MAAM,oBAAoB,CAAC;AAEjE,OAAO,KAAK,EAAmB,eAAe,EAAE,iBAAiB,EAAE,MAAM,qBAAqB,CAAC;AAC/F,OAAO,EAEL,aAAa,EASd,MAAM,sBAAsB,CAAC;AAQ9B,OAAO,KAAK,EACV,YAAY,EACZ,WAAW,EACX,yBAAyB,EACzB,oBAAoB,EACpB,mBAAmB,EACnB,wBAAwB,EACxB,oCAAoC,EACpC,yBAAyB,EACzB,uBAAuB,EACvB,wBAAwB,EACxB,uBAAuB,EACvB,wBAAwB,EAExB,yBAAyB,EACzB,iCAAiC,EAEjC,8BAA8B,EAC9B,6BAA6B,EAC7B,+BAA+B,EAC/B,yBAAyB,EACzB,0BAA0B,EAC1B,6BAA6B,EAC7B,mCAAmC,EACnC,+BAA+B,EAC/B,oCAAoC,EACrC,MAAM,sBAAsB,CAAC;AAI9B,OAAO,KAAK,EAAE,mBAAmB,EAAE,kBAAkB,EAAE,MAAM,aAAa,CAAC;AAG3E,qBAAa,oBAAqB,SAAQ,aAAa;;IACrD,SAAkB,2BAA2B,QAAQ;IACrD,QAAQ,CAAC,2BAA2B,QAAQ;IAC5C,QAAQ,CAAC,wCAAwC,QAAQ;IAMzD,yCAAyC;IACzC,MAAM,CAAC,QAAQ,CAAC,mBAAmB,oGAAuE;IAE1G;;;OAGG;IACH,gBAAyB,eAAe,EAAE,yBAAyB,CAIjE;IAEF,YAAY,MAAM,EAAE,mBAAmB,EAQtC;YAEa,aAAa;IAIrB,IAAI,IAAI,OAAO,CAAC,IAAI,CAAC,CAG1B;IAED;;;;;;OAMG;IACG,KAAK,CAAC,QAAQ,EAAE,MAAM,CAAC,MAAM,EAAE,oBAAoB,CAAC,EAAE,OAAO,CAAC,EAAE,YAAY,GAAG,OAAO,CAAC,WAAW,EAAE,CAAC,CAO1G;IAED;;OAEG;IACH,0BAA0B,IAAI,kBAAkB,EAAE,CAsBjD;IAED;;OAEG;IACG,oBAAoB,IAAI,OAAO,CAAC,IAAI,CAAC,CAsC1C;IAED;;OAEG;IACG,mBAAmB,IAAI,OAAO,CAAC,IAAI,CAAC,CAczC;IAEK,mBAAmB,IAAI,OAAO,CAAC,IAAI,CAAC,CAczC;IAED,OAAO,CAAC,QAAQ;IAuBhB,OAAO,CAAC,aAAa;IAerB;;;;;;OAMG;YACW,oBAAoB;IA6ErB,gBAAgB,CAAC,EAAE,UAAU,EAAE,EAAE;QAAE,UAAU,EAAE,MAAM,EAAE,CAAA;KAAE,GAAG,OAAO,CAAC;QAAE,QAAQ,EAAE,eAAe,EAAE,CAAA;KAAE,CAAC,CAyBhH;IAEY,YAAY,CAAC,IAAI,EAAE,wBAAwB,GAAG,OAAO,CAAC,yBAAyB,CAAC,CAgL5F;IAEY,wBAAwB,CACnC,IAAI,EAAE,oCAAoC,GACzC,OAAO,CAAC,yBAAyB,CAAC,CAsKpC;IAEK,YAAY,CAAC,EAAE,QAAQ,EAAE,EAAE;QAAE,QAAQ,EAAE,eAAe,EAAE,CAAA;KAAE,GAAG,OAAO,CAAC;QAAE,QAAQ,EAAE,eAAe,EAAE,CAAA;KAAE,CAAC,CA0E1G;IAEK,cAAc,CAAC,EACnB,QAAQ,GACT,EAAE;QACD,QAAQ,EAAE,CAAC,OAAO,CAAC,IAAI,CAAC,eAAe,EAAE,WAAW,CAAC,CAAC,GAAG;YACvD,EAAE,EAAE,MAAM,CAAC;YACX,OAAO,CAAC,EAAE;gBAAE,QAAQ,CAAC,EAAE,sBAAsB,CAAC,UAAU,CAAC,CAAC;gBAAC,OAAO,CAAC,EAAE,sBAAsB,CAAC,SAAS,CAAC,CAAA;aAAE,CAAC;SAC1G,CAAC,EAAE,CAAC;KACN,GAAG,OAAO,CAAC,eAAe,EAAE,CAAC,CA6F7B;IAEK,eAAe,CAAC,EAAE,UAAU,EAAE,EAAE;QAAE,UAAU,EAAE,MAAM,CAAA;KAAE,GAAG,OAAO,CAAC,mBAAmB,GAAG,IAAI,CAAC,CA2BjG;IAEK,YAAY,CAAC,EAAE,QAAQ,EAAE,EAAE;QAAE,QAAQ,EAAE,mBAAmB,CAAA;KAAE,GAAG,OAAO,CAAC,mBAAmB,CAAC,CAuBhG;IAEK,cAAc,CAAC,EACnB,UAAU,EACV,aAAa,EACb,QAAQ,GACT,EAAE;QACD,UAAU,EAAE,MAAM,CAAC;QACnB,aAAa,CAAC,EAAE,MAAM,CAAC;QACvB,QAAQ,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;KACpC,GAAG,OAAO,CAAC,mBAAmB,CAAC,CAgD/B;IAEK,aAAa,CAAC,EAClB,QAAQ,EACR,UAAU,GACX,EAAE;QACD,QAAQ,EAAE,MAAM,CAAC;QACjB,UAAU,CAAC,EAAE,MAAM,CAAC;KACrB,GAAG,OAAO,CAAC,iBAAiB,GAAG,IAAI,CAAC,CAuBpC;IAEY,WAAW,CAAC,IAAI,EAAE,uBAAuB,GAAG,OAAO,CAAC,wBAAwB,CAAC,CAoHzF;IAEK,UAAU,CAAC,EAAE,MAAM,EAAE,EAAE;QAAE,MAAM,EAAE,iBAAiB,CAAA;KAAE,GAAG,OAAO,CAAC,iBAAiB,CAAC,CAyBtF;IAEK,YAAY,CAAC,EACjB,EAAE,EACF,KAAK,EACL,QAAQ,GACT,EAAE;QACD,EAAE,EAAE,MAAM,CAAC;QACX,KAAK,CAAC,EAAE,MAAM,CAAC;QACf,QAAQ,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;KACpC,GAAG,OAAO,CAAC,iBAAiB,CAAC,CAgD7B;IAEK,YAAY,CAAC,EAAE,QAAQ,EAAE,EAAE;QAAE,QAAQ,EAAE,MAAM,CAAA;KAAE,GAAG,OAAO,CAAC,IAAI,CAAC,CA0BpE;IAEK,cAAc,CAAC,UAAU,EAAE,MAAM,EAAE,GAAG,OAAO,CAAC,IAAI,CAAC,CA6BxD;IAEK,WAAW,CAAC,IAAI,EAAE,uBAAuB,GAAG,OAAO,CAAC,wBAAwB,CAAC,CA2KlF;IAMD,OAAO,CAAC,QAAQ;IAIhB,OAAO,CAAC,eAAe;IAmDjB,sBAAsB,CAAC,QAAQ,EAAE,MAAM,GAAG,IAAI,EAAE,UAAU,EAAE,MAAM,GAAG,OAAO,CAAC,yBAAyB,GAAG,IAAI,CAAC,CAkBnH;IAEK,6BAA6B,CACjC,QAAQ,EAAE,MAAM,GAAG,IAAI,EACvB,UAAU,EAAE,MAAM,EAClB,KAAK,GAAE,MAAW,EAClB,OAAO,CAAC,EAAE,iCAAiC,GAC1C,OAAO,CAAC,yBAAyB,EAAE,CAAC,CA4DtC;IAEK,6BAA6B,CAAC,KAAK,EAAE,8BAA8B,GAAG,OAAO,CAAC,yBAAyB,CAAC,CAsE7G;IAEK,+BAA+B,CAAC,MAAM,EAAE,yBAAyB,GAAG,OAAO,CAAC,IAAI,CAAC,CAkDtF;IAEK,wBAAwB,CAAC,KAAK,EAAE,6BAA6B,GAAG,OAAO,CAAC,IAAI,CAAC,CA8ClF;IAEK,0BAA0B,CAAC,KAAK,EAAE,+BAA+B,GAAG,OAAO,CAAC,yBAAyB,CAAC,CAwE3G;IAEK,iBAAiB,CAAC,EAAE,EAAE,MAAM,EAAE,YAAY,EAAE,OAAO,GAAG,OAAO,CAAC,IAAI,CAAC,CA4BxE;IAEK,gBAAgB,CAAC,EAAE,EAAE,MAAM,EAAE,WAAW,EAAE,OAAO,GAAG,OAAO,CAAC,IAAI,CAAC,CA4BtE;IAEK,2BAA2B,CAAC,EAAE,EAAE,MAAM,EAAE,WAAW,EAAE,OAAO,EAAE,oBAAoB,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC,CAqChH;IAEK,0BAA0B,CAAC,EAAE,EAAE,MAAM,EAAE,WAAW,EAAE,OAAO,GAAG,OAAO,CAAC,IAAI,CAAC,CA+BhF;IAEK,wBAAwB,CAAC,QAAQ,EAAE,MAAM,GAAG,IAAI,EAAE,UAAU,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC,CAgBzF;IAEK,uBAAuB,CAAC,EAAE,EAAE,MAAM,EAAE,UAAU,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC,CA4C3E;IAEK,+BAA+B,CAAC,KAAK,EAAE,oCAAoC,GAAG,OAAO,CAAC,IAAI,CAAC,CAmChG;IAMK,0BAA0B,CAAC,KAAK,EAAE,+BAA+B,GAAG,OAAO,CAAC,IAAI,CAAC,CA0DtF;IAEK,oBAAoB,CAAC,KAAK,EAAE,yBAAyB,GAAG,OAAO,CAAC,0BAA0B,CAAC,CAkMhG;IAEK,wBAAwB,CAAC,KAAK,EAAE,6BAA6B,GAAG,OAAO,CAAC,IAAI,CAAC,CA6DlF;IAEK,8BAA8B,CAAC,KAAK,EAAE,mCAAmC,GAAG,OAAO,CAAC,yBAAyB,CAAC,CA+EnH;CACF"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../../src/storage/domains/memory/index.ts"],"names":[],"mappings":"AACA,OAAO,EAML,KAAK,8BAA8B,EACnC,KAAK,gCAAgC,EACtC,MAAM,uBAAuB,CAAC;AAG/B,OAAO,KAAK,EAAE,sBAAsB,EAAE,MAAM,oBAAoB,CAAC;AAEjE,OAAO,KAAK,EAAmB,eAAe,EAAE,iBAAiB,EAAE,MAAM,qBAAqB,CAAC;AAC/F,OAAO,EAGL,aAAa,EAWd,MAAM,sBAAsB,CAAC;AAQ9B,OAAO,KAAK,EACV,YAAY,EACZ,WAAW,EACX,yBAAyB,EACzB,oBAAoB,EACpB,mBAAmB,EACnB,wBAAwB,EACxB,oCAAoC,EACpC,yBAAyB,EACzB,uBAAuB,EACvB,wBAAwB,EACxB,uBAAuB,EACvB,wBAAwB,EAExB,yBAAyB,EACzB,iCAAiC,EAEjC,8BAA8B,EAC9B,6BAA6B,EAC7B,+BAA+B,EAC/B,yBAAyB,EACzB,0BAA0B,EAC1B,6BAA6B,EAC7B,mCAAmC,EACnC,+BAA+B,EAC/B,oCAAoC,EACpC,QAAQ,EACT,MAAM,sBAAsB,CAAC;AAK9B,OAAO,KAAK,EAAE,mBAAmB,EAAE,kBAAkB,EAAE,MAAM,aAAa,CAAC;AAwH3E,qBAAa,oBAAqB,SAAQ,aAAa;;IACrD,SAAkB,2BAA2B,QAAQ;IACrD,QAAQ,CAAC,2BAA2B,QAAQ;IAC5C,QAAQ,CAAC,wCAAwC,QAAQ;IAMzD,yCAAyC;IACzC,MAAM,CAAC,QAAQ,CAAC,mBAAmB,gIAMxB;IAEX;;;;;OAKG;IACH,gBAAyB,eAAe,EAAE,yBAAyB,CAKjE;IAEF,YAAY,MAAM,EAAE,mBAAmB,EAQtC;YAEa,aAAa;IAIrB,IAAI,IAAI,OAAO,CAAC,IAAI,CAAC,CAK1B;IAED;;;OAGG;IACM,kBAAkB,IAAI,OAAO,CAAC,OAAO,CAAC,CAE9C;IAEc,aAAa,CAAC,KAAK,EAAE,QAAQ,GAAG,OAAO,CAAC,OAAO,CAAC,CAwC9D;IAEc,cAAc,CAAC,KAAK,EAAE,QAAQ,GAAG,OAAO,CAAC,OAAO,CAAC,CAmB/D;IAQD;;;;;;;OAOG;IACG,KAAK,CAAC,QAAQ,EAAE,MAAM,CAAC,MAAM,EAAE,oBAAoB,CAAC,EAAE,OAAO,CAAC,EAAE,YAAY,GAAG,OAAO,CAAC,WAAW,EAAE,CAAC,CAO1G;IAED;;OAEG;IACH,0BAA0B,IAAI,kBAAkB,EAAE,CAsBjD;IAED;;OAEG;IACG,oBAAoB,IAAI,OAAO,CAAC,IAAI,CAAC,CAsC1C;IAED;;OAEG;IACG,mBAAmB,IAAI,OAAO,CAAC,IAAI,CAAC,CAczC;IAEK,mBAAmB,IAAI,OAAO,CAAC,IAAI,CAAC,CAiBzC;IAED,OAAO,CAAC,QAAQ;IAuBhB,OAAO,CAAC,aAAa;IAerB;;;;;;OAMG;YACW,oBAAoB;IA6ErB,gBAAgB,CAAC,EAAE,UAAU,EAAE,EAAE;QAAE,UAAU,EAAE,MAAM,EAAE,CAAA;KAAE,GAAG,OAAO,CAAC;QAAE,QAAQ,EAAE,eAAe,EAAE,CAAA;KAAE,CAAC,CAyBhH;IAEY,YAAY,CAAC,IAAI,EAAE,wBAAwB,GAAG,OAAO,CAAC,yBAAyB,CAAC,CAgL5F;IAEY,wBAAwB,CACnC,IAAI,EAAE,oCAAoC,GACzC,OAAO,CAAC,yBAAyB,CAAC,CAsKpC;IAEK,YAAY,CAAC,EACjB,QAAQ,EACR,KAAK,GACN,EAAE;QACD,QAAQ,EAAE,eAAe,EAAE,CAAC;QAC5B,KAAK,CAAC,EAAE,QAAQ,CAAC;KAClB,GAAG,OAAO,CAAC;QAAE,QAAQ,EAAE,eAAe,EAAE,CAAA;KAAE,CAAC,CA8E3C;IAEK,cAAc,CAAC,EACnB,QAAQ,EACR,KAAK,GACN,EAAE;QACD,QAAQ,EAAE,CAAC,OAAO,CAAC,IAAI,CAAC,eAAe,EAAE,WAAW,CAAC,CAAC,GAAG;YACvD,EAAE,EAAE,MAAM,CAAC;YACX,OAAO,CAAC,EAAE;gBAAE,QAAQ,CAAC,EAAE,sBAAsB,CAAC,UAAU,CAAC,CAAC;gBAAC,OAAO,CAAC,EAAE,sBAAsB,CAAC,SAAS,CAAC,CAAA;aAAE,CAAC;SAC1G,CAAC,EAAE,CAAC;QACL,KAAK,CAAC,EAAE,QAAQ,CAAC;KAClB,GAAG,OAAO,CAAC,eAAe,EAAE,CAAC,CAiG7B;IAEK,eAAe,CAAC,EAAE,UAAU,EAAE,EAAE;QAAE,UAAU,EAAE,MAAM,CAAA;KAAE,GAAG,OAAO,CAAC,mBAAmB,GAAG,IAAI,CAAC,CA2BjG;IAEK,YAAY,CAAC,EAAE,QAAQ,EAAE,EAAE;QAAE,QAAQ,EAAE,mBAAmB,CAAA;KAAE,GAAG,OAAO,CAAC,mBAAmB,CAAC,CAuBhG;IAEK,cAAc,CAAC,EACnB,UAAU,EACV,aAAa,EACb,QAAQ,EACR,KAAK,GACN,EAAE;QACD,UAAU,EAAE,MAAM,CAAC;QACnB,aAAa,CAAC,EAAE,MAAM,CAAC;QACvB,QAAQ,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;QACnC,KAAK,CAAC,EAAE,QAAQ,CAAC;KAClB,GAAG,OAAO,CAAC,mBAAmB,CAAC,CAuD/B;IAEK,aAAa,CAAC,EAClB,QAAQ,EACR,UAAU,GACX,EAAE;QACD,QAAQ,EAAE,MAAM,CAAC;QACjB,UAAU,CAAC,EAAE,MAAM,CAAC;KACrB,GAAG,OAAO,CAAC,iBAAiB,GAAG,IAAI,CAAC,CAuBpC;IAEY,WAAW,CAAC,IAAI,EAAE,uBAAuB,GAAG,OAAO,CAAC,wBAAwB,CAAC,CAoHzF;IAEK,UAAU,CAAC,EAAE,MAAM,EAAE,KAAK,EAAE,EAAE;QAAE,MAAM,EAAE,iBAAiB,CAAC;QAAC,KAAK,CAAC,EAAE,QAAQ,CAAA;KAAE,GAAG,OAAO,CAAC,iBAAiB,CAAC,CA4B/G;IAEK,YAAY,CAAC,EACjB,EAAE,EACF,KAAK,EACL,QAAQ,EACR,KAAK,GACN,EAAE;QACD,EAAE,EAAE,MAAM,CAAC;QACX,KAAK,CAAC,EAAE,MAAM,CAAC;QACf,QAAQ,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;QACnC,KAAK,CAAC,EAAE,QAAQ,CAAC;KAClB,GAAG,OAAO,CAAC,iBAAiB,CAAC,CAqD7B;IAEK,YAAY,CAAC,EAAE,QAAQ,EAAE,EAAE;QAAE,QAAQ,EAAE,MAAM,CAAA;KAAE,GAAG,OAAO,CAAC,IAAI,CAAC,CA0BpE;IAEK,cAAc,CAAC,UAAU,EAAE,MAAM,EAAE,EAAE,OAAO,CAAC,EAAE;QAAE,KAAK,CAAC,EAAE,QAAQ,CAAA;KAAE,GAAG,OAAO,CAAC,IAAI,CAAC,CAoCxF;IAEK,WAAW,CAAC,IAAI,EAAE,uBAAuB,GAAG,OAAO,CAAC,wBAAwB,CAAC,CA2KlF;IAMD,OAAO,CAAC,QAAQ;IAIhB,OAAO,CAAC,eAAe;IAoDjB,sBAAsB,CAAC,QAAQ,EAAE,MAAM,GAAG,IAAI,EAAE,UAAU,EAAE,MAAM,GAAG,OAAO,CAAC,yBAAyB,GAAG,IAAI,CAAC,CAgBnH;IAEK,6BAA6B,CACjC,QAAQ,EAAE,MAAM,GAAG,IAAI,EACvB,UAAU,EAAE,MAAM,EAClB,KAAK,GAAE,MAAW,EAClB,OAAO,CAAC,EAAE,iCAAiC,GAC1C,OAAO,CAAC,yBAAyB,EAAE,CAAC,CA4DtC;IAEK,6BAA6B,CAAC,KAAK,EAAE,8BAA8B,GAAG,OAAO,CAAC,yBAAyB,CAAC,CAuE7G;IAEK,+BAA+B,CAAC,MAAM,EAAE,yBAAyB,GAAG,OAAO,CAAC,IAAI,CAAC,CAmDtF;IAEK,wBAAwB,CAAC,KAAK,EAAE,6BAA6B,GAAG,OAAO,CAAC,IAAI,CAAC,CAElF;IAEK,wBAAwB,CAAC,KAAK,EAAE,6BAA6B,GAAG,OAAO,CAAC,8BAA8B,CAAC,CAuD5G;IAEK,0BAA0B,CAAC,KAAK,EAAE,+BAA+B,GAAG,OAAO,CAAC,yBAAyB,CAAC,CAwC3G;IAEK,iBAAiB,CAAC,EAAE,EAAE,MAAM,EAAE,YAAY,EAAE,OAAO,GAAG,OAAO,CAAC,IAAI,CAAC,CA4BxE;IAEK,gBAAgB,CAAC,EAAE,EAAE,MAAM,EAAE,WAAW,EAAE,OAAO,GAAG,OAAO,CAAC,IAAI,CAAC,CA4BtE;IAEK,2BAA2B,CAAC,EAAE,EAAE,MAAM,EAAE,WAAW,EAAE,OAAO,EAAE,oBAAoB,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC,CA0BhH;IAEK,0BAA0B,CAAC,EAAE,EAAE,MAAM,EAAE,WAAW,EAAE,OAAO,GAAG,OAAO,CAAC,IAAI,CAAC,CA+BhF;IAEK,wBAAwB,CAAC,QAAQ,EAAE,MAAM,GAAG,IAAI,EAAE,UAAU,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC,CAgBzF;IAEK,uBAAuB,CAAC,EAAE,EAAE,MAAM,EAAE,UAAU,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC,CA8B3E;IAEK,+BAA+B,CAAC,KAAK,EAAE,oCAAoC,GAAG,OAAO,CAAC,IAAI,CAAC,CAmChG;IAMK,0BAA0B,CAAC,KAAK,EAAE,+BAA+B,GAAG,OAAO,CAAC,IAAI,CAAC,CAEtF;IAEK,0BAA0B,CAAC,KAAK,EAAE,+BAA+B,GAAG,OAAO,CAAC,gCAAgC,CAAC,CAgIlH;IAEK,oBAAoB,CAAC,KAAK,EAAE,yBAAyB,GAAG,OAAO,CAAC,0BAA0B,CAAC,CA2MhG;IAEK,wBAAwB,CAAC,KAAK,EAAE,6BAA6B,GAAG,OAAO,CAAC,IAAI,CAAC,CAmElF;IAEK,8BAA8B,CAAC,KAAK,EAAE,mCAAmC,GAAG,OAAO,CAAC,yBAAyB,CAAC,CA2DnH;CA6LF"}

package/dist/storage/domains/run-fencing.d.ts

@@ -0,0 +1,39 @@
+import type { RunFence } from '@mastra/core/storage';
+import type { ClientSession, Collection } from 'mongodb';
+import type { MongoDBConnector } from '../connectors/MongoDBConnector.js';
+/** A write's fence, and the collection holding the run's current claim to check it against. */
+export interface RunFenceCheck {
+    claims: string;
+    fence: RunFence;
+    operation: string;
+}
+/**
+ * A run's claim: its ownership document (workflows) or raised fence (memory),
+ * keyed by runId.
+ */
+export interface RunClaimDocument {
+    _id: string;
+    generation: number;
+    ownerId: string;
+    leaseExpiresAt?: Date | null;
+    updatedAt?: Date;
+    retiredAt?: Date | null;
+    fencedWrites?: number;
+}
+export declare function getRunClaims(connector: MongoDBConnector, name: string): Promise<Collection<RunClaimDocument>>;
+/** Duplicate-key error: an upsert lost the race to insert the document. */
+export declare function isDuplicateKeyError(error: unknown): boolean;
+/**
+ * Runs a write, fenced when `check` is set. A fenced write runs in a
+ * transaction that first updates the run's claim document, matching it on
+ * the fence. A fence that is no longer current matches nothing, and the write
+ * throws `RunFenceConflictError` before anything is written. A claim that
+ * changes while the transaction runs makes the update conflict, so the
+ * transaction retries and sees the new claim; a claim attempted after the
+ * update waits until the transaction commits. Updating the claim document,
+ * rather than reading it, is what makes concurrent claims conflict.
+ *
+ * Fencing is declared only on deployments that support transactions.
+ */
+export declare function withRunFence<T>(connector: MongoDBConnector, check: RunFenceCheck | undefined, write: (session?: ClientSession) => Promise<T>): Promise<T>;
+//# sourceMappingURL=run-fencing.d.ts.map

package/dist/storage/domains/run-fencing.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"run-fencing.d.ts","sourceRoot":"","sources":["../../../src/storage/domains/run-fencing.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,QAAQ,EAAE,MAAM,sBAAsB,CAAC;AAErD,OAAO,KAAK,EAAE,aAAa,EAAE,UAAU,EAAE,MAAM,SAAS,CAAC;AACzD,OAAO,KAAK,EAAE,gBAAgB,EAAE,MAAM,gCAAgC,CAAC;AAEvE,+FAA+F;AAC/F,MAAM,WAAW,aAAa;IAC5B,MAAM,EAAE,MAAM,CAAC;IACf,KAAK,EAAE,QAAQ,CAAC;IAChB,SAAS,EAAE,MAAM,CAAC;CACnB;AAED;;;GAGG;AACH,MAAM,WAAW,gBAAgB;IAC/B,GAAG,EAAE,MAAM,CAAC;IACZ,UAAU,EAAE,MAAM,CAAC;IACnB,OAAO,EAAE,MAAM,CAAC;IAChB,cAAc,CAAC,EAAE,IAAI,GAAG,IAAI,CAAC;IAC7B,SAAS,CAAC,EAAE,IAAI,CAAC;IACjB,SAAS,CAAC,EAAE,IAAI,GAAG,IAAI,CAAC;IACxB,YAAY,CAAC,EAAE,MAAM,CAAC;CACvB;AAED,wBAAsB,YAAY,CAAC,SAAS,EAAE,gBAAgB,EAAE,IAAI,EAAE,MAAM,GAAG,OAAO,CAAC,UAAU,CAAC,gBAAgB,CAAC,CAAC,CAGnH;AAED,2EAA2E;AAC3E,wBAAgB,mBAAmB,CAAC,KAAK,EAAE,OAAO,GAAG,OAAO,CAE3D;AAED;;;;;;;;;;;GAWG;AACH,wBAAsB,YAAY,CAAC,CAAC,EAClC,SAAS,EAAE,gBAAgB,EAC3B,KAAK,EAAE,aAAa,GAAG,SAAS,EAChC,KAAK,EAAE,CAAC,OAAO,CAAC,EAAE,aAAa,KAAK,OAAO,CAAC,CAAC,CAAC,GAC7C,OAAO,CAAC,CAAC,CAAC,CAaZ"}

package/dist/storage/domains/workflows/index.d.ts

@@ -1,22 +1,35 @@
 import { WorkflowsStorage } from '@mastra/core/storage';
-import type { PruneOptions, PruneResult, RetentionTablesDescriptor, TableRetentionPolicy, WorkflowRun, WorkflowRuns, StorageListWorkflowRunsInput, UpdateWorkflowStateOptions } from '@mastra/core/storage';
+import type { ClaimRunOwnershipInput, ClaimRunOwnershipResult, PruneOptions, PruneResult, RenewRunOwnershipInput, RenewRunOwnershipResult, RetentionTablesDescriptor, RunFence, RunOwnershipRecord, TableRetentionPolicy, WorkflowRun, WorkflowRuns, StorageListWorkflowRunsInput, UpdateWorkflowStateOptions } from '@mastra/core/storage';
 import type { StepResult, WorkflowRunState } from '@mastra/core/workflows';
 import type { MongoDBDomainConfig, MongoDBIndexConfig } from '../../types.js';
 export declare class WorkflowsStorageMongoDB extends WorkflowsStorage {
     #private;
     /** Collections managed by this domain */
-    static readonly MANAGED_COLLECTIONS: readonly ["mastra_workflow_snapshot"];
+    static readonly MANAGED_COLLECTIONS: readonly ["mastra_workflow_snapshot", "mastra_workflow_run_owners"];
     /**
-     * Anchor is `updatedAt` (BSON date), so the policy reads as inactivity:
-     * a run is pruned only after its snapshot has not been touched for `maxAge`.
+     * Both anchors are `updatedAt` (BSON date), so the policies read as
+     * inactivity: a run is pruned only after its snapshot has not been touched
+     * for `maxAge`, and an ownership document only after no claim, renewal or
+     * release has touched it for `maxAge`.
      */
     static readonly retentionTables: RetentionTablesDescriptor;
     constructor(config: MongoDBDomainConfig);
-    /** Delete workflow snapshots idle for longer than the policy's `maxAge`, batched. */
+    /** Delete workflow snapshots and ownership documents idle for longer than each policy's `maxAge`, batched. */
     prune(policies: Record<string, TableRetentionPolicy>, options?: PruneOptions): Promise<PruneResult[]>;
     supportsConcurrentUpdates(): boolean;
+    /**
+     * Fenced writes need multi-document transactions, so only replica sets and
+     * sharded clusters fence. Rejects while the deployment can't be probed.
+     */
+    supportsRunFencing(): Promise<boolean>;
     private getCollection;
     init(): Promise<void>;
+    claimRunOwnership({ runId, ownerId, leaseMs, force, expectedGeneration, }: ClaimRunOwnershipInput): Promise<ClaimRunOwnershipResult>;
+    renewRunOwnership({ leaseMs, ...fence }: RenewRunOwnershipInput): Promise<RenewRunOwnershipResult>;
+    releaseRunOwnership(fence: RunFence): Promise<boolean>;
+    getRunOwnership({ runId }: {
+        runId: string;
+    }): Promise<RunOwnershipRecord | null>;
     /**
      * Returns default index definitions for the workflows domain collections.
      */
@@ -30,26 +43,29 @@ export declare class WorkflowsStorageMongoDB extends WorkflowsStorage {
      */
     createCustomIndexes(): Promise<void>;
     dangerouslyClearAll(): Promise<void>;
-    updateWorkflowResults({ workflowName, runId, stepId, result, requestContext, state, }: {
+    updateWorkflowResults({ workflowName, runId, stepId, result, requestContext, state, fence, }: {
         workflowName: string;
         runId: string;
         stepId: string;
         result: StepResult<any, any, any, any>;
         requestContext: Record<string, any>;
         state?: Record<string, any>;
+        fence?: RunFence;
     }): Promise<Record<string, StepResult<any, any, any, any>>>;
-    updateWorkflowState({ workflowName, runId, opts, }: {
+    updateWorkflowState({ workflowName, runId, opts, fence, }: {
         workflowName: string;
         runId: string;
         opts: UpdateWorkflowStateOptions;
+        fence?: RunFence;
     }): Promise<WorkflowRunState | undefined>;
-    persistWorkflowSnapshot({ workflowName, runId, resourceId, snapshot, createdAt, updatedAt, }: {
+    persistWorkflowSnapshot({ workflowName, runId, resourceId, snapshot, createdAt, updatedAt, fence, }: {
         workflowName: string;
         runId: string;
         resourceId?: string;
         snapshot: WorkflowRunState;
         createdAt?: Date;
         updatedAt?: Date;
+        fence?: RunFence;
     }): Promise<void>;
     loadWorkflowSnapshot({ workflowName, runId, }: {
         workflowName: string;
@@ -60,9 +76,10 @@ export declare class WorkflowsStorageMongoDB extends WorkflowsStorage {
         runId: string;
         workflowName?: string;
     }): Promise<WorkflowRun | null>;
-    deleteWorkflowRunById({ runId, workflowName }: {
+    deleteWorkflowRunById({ runId, workflowName, fence, }: {
         runId: string;
         workflowName: string;
+        fence?: RunFence;
     }): Promise<void>;
     private parseWorkflowRun;
 }

package/dist/storage/domains/workflows/index.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../../src/storage/domains/workflows/index.ts"],"names":[],"mappings":"AACA,OAAO,EAEL,gBAAgB,EAIjB,MAAM,sBAAsB,CAAC;AAC9B,OAAO,KAAK,EACV,YAAY,EACZ,WAAW,EACX,yBAAyB,EACzB,oBAAoB,EACpB,WAAW,EACX,YAAY,EACZ,4BAA4B,EAC5B,0BAA0B,EAC3B,MAAM,sBAAsB,CAAC;AAC9B,OAAO,KAAK,EAAE,UAAU,EAAE,gBAAgB,EAAE,MAAM,wBAAwB,CAAC;AAI3E,OAAO,KAAK,EAAE,mBAAmB,EAAE,kBAAkB,EAAE,MAAM,aAAa,CAAC;AAE3E,qBAAa,uBAAwB,SAAQ,gBAAgB;;IAK3D,yCAAyC;IACzC,MAAM,CAAC,QAAQ,CAAC,mBAAmB,wCAAsC;IAEzE;;;OAGG;IACH,gBAAyB,eAAe,EAAE,yBAAyB,CAEjE;IAEF,YAAY,MAAM,EAAE,mBAAmB,EAQtC;IAED,qFAAqF;IAC/E,KAAK,CAAC,QAAQ,EAAE,MAAM,CAAC,MAAM,EAAE,oBAAoB,CAAC,EAAE,OAAO,CAAC,EAAE,YAAY,GAAG,OAAO,CAAC,WAAW,EAAE,CAAC,CAO1G;IAED,yBAAyB,IAAI,OAAO,CAEnC;YAEa,aAAa;IAIrB,IAAI,IAAI,OAAO,CAAC,IAAI,CAAC,CAG1B;IAED;;OAEG;IACH,0BAA0B,IAAI,kBAAkB,EAAE,CASjD;IAED;;OAEG;IACG,oBAAoB,IAAI,OAAO,CAAC,IAAI,CAAC,CAc1C;IAED;;OAEG;IACG,mBAAmB,IAAI,OAAO,CAAC,IAAI,CAAC,CAczC;IAEK,mBAAmB,IAAI,OAAO,CAAC,IAAI,CAAC,CAGzC;IAEK,qBAAqB,CAAC,EAC1B,YAAY,EACZ,KAAK,EACL,MAAM,EACN,MAAM,EACN,cAAc,EACd,KAAK,GACN,EAAE;QACD,YAAY,EAAE,MAAM,CAAC;QACrB,KAAK,EAAE,MAAM,CAAC;QACd,MAAM,EAAE,MAAM,CAAC;QACf,MAAM,EAAE,UAAU,CAAC,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,CAAC,CAAC;QACvC,cAAc,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;QACpC,KAAK,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;KAC7B,GAAG,OAAO,CAAC,MAAM,CAAC,MAAM,EAAE,UAAU,CAAC,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,CAAC,CAAC,CAAC,CAgF1D;IACK,mBAAmB,CAAC,EACxB,YAAY,EACZ,KAAK,EACL,IAAI,GACL,EAAE;QACD,YAAY,EAAE,MAAM,CAAC;QACrB,KAAK,EAAE,MAAM,CAAC;QACd,IAAI,EAAE,0BAA0B,CAAC;KAClC,GAAG,OAAO,CAAC,gBAAgB,GAAG,SAAS,CAAC,CA8DxC;IAEK,uBAAuB,CAAC,EAC5B,YAAY,EACZ,KAAK,EACL,UAAU,EACV,QAAQ,EACR,SAAS,EACT,SAAS,GACV,EAAE;QACD,YAAY,EAAE,MAAM,CAAC;QACrB,KAAK,EAAE,MAAM,CAAC;QACd,UAAU,CAAC,EAAE,MAAM,CAAC;QACpB,QAAQ,EAAE,gBAAgB,CAAC;QAC3B,SAAS,CAAC,EAAE,IAAI,CAAC;QACjB,SAAS,CAAC,EAAE,IAAI,CAAC;KAClB,GAAG,OAAO,CAAC,IAAI,CAAC,CA+BhB;IAEK,oBAAoB,CAAC,EACzB,YAAY,EACZ,KAAK,GACN,EAAE;QACD,YAAY,EAAE,MAAM,CAAC;QACrB,KAAK,EAAE,MAAM,CAAC;KACf,GAAG,OAAO,CAAC,gBAAgB,GAAG,IAAI,CAAC,CAwBnC;IAEK,gBAAgB,CAAC,IAAI,CAAC,EAAE,4BAA4B,GAAG,OAAO,CAAC,YAAY,CAAC,CA2EjF;IAEK,kBAAkB,CAAC,IAAI,EAAE;QAAE,KAAK,EAAE,MAAM,CAAC;QAAC,YAAY,CAAC,EAAE,MAAM,CAAA;KAAE,GAAG,OAAO,CAAC,WAAW,GAAG,IAAI,CAAC,CA4BpG;IAEK,qBAAqB,CAAC,EAAE,KAAK,EAAE,YAAY,EAAE,EAAE;QAAE,KAAK,EAAE,MAAM,CAAC;QAAC,YAAY,EAAE,MAAM,CAAA;KAAE,GAAG,OAAO,CAAC,IAAI,CAAC,CAe3G;IAED,OAAO,CAAC,gBAAgB;CAoBzB"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../../src/storage/domains/workflows/index.ts"],"names":[],"mappings":"AACA,OAAO,EAIL,gBAAgB,EAKjB,MAAM,sBAAsB,CAAC;AAC9B,OAAO,KAAK,EACV,sBAAsB,EACtB,uBAAuB,EACvB,YAAY,EACZ,WAAW,EACX,sBAAsB,EACtB,uBAAuB,EACvB,yBAAyB,EACzB,QAAQ,EACR,kBAAkB,EAClB,oBAAoB,EACpB,WAAW,EACX,YAAY,EACZ,4BAA4B,EAC5B,0BAA0B,EAC3B,MAAM,sBAAsB,CAAC;AAC9B,OAAO,KAAK,EAAE,UAAU,EAAE,gBAAgB,EAAE,MAAM,wBAAwB,CAAC;AAK3E,OAAO,KAAK,EAAE,mBAAmB,EAAE,kBAAkB,EAAE,MAAM,aAAa,CAAC;AAyB3E,qBAAa,uBAAwB,SAAQ,gBAAgB;;IAK3D,yCAAyC;IACzC,MAAM,CAAC,QAAQ,CAAC,mBAAmB,sEAAiE;IAEpG;;;;;OAKG;IACH,gBAAyB,eAAe,EAAE,yBAAyB,CAGjE;IAEF,YAAY,MAAM,EAAE,mBAAmB,EAQtC;IAED,8GAA8G;IACxG,KAAK,CAAC,QAAQ,EAAE,MAAM,CAAC,MAAM,EAAE,oBAAoB,CAAC,EAAE,OAAO,CAAC,EAAE,YAAY,GAAG,OAAO,CAAC,WAAW,EAAE,CAAC,CAO1G;IAED,yBAAyB,IAAI,OAAO,CAEnC;IAED;;;OAGG;IACH,kBAAkB,IAAI,OAAO,CAAC,OAAO,CAAC,CAErC;YAEa,aAAa;IAIrB,IAAI,IAAI,OAAO,CAAC,IAAI,CAAC,CAG1B;IA+BK,iBAAiB,CAAC,EACtB,KAAK,EACL,OAAO,EACP,OAAO,EACP,KAAK,EACL,kBAAkB,GACnB,EAAE,sBAAsB,GAAG,OAAO,CAAC,uBAAuB,CAAC,CA8C3D;IAEK,iBAAiB,CAAC,EAAE,OAAO,EAAE,GAAG,KAAK,EAAE,EAAE,sBAAsB,GAAG,OAAO,CAAC,uBAAuB,CAAC,CAkBvG;IAEK,mBAAmB,CAAC,KAAK,EAAE,QAAQ,GAAG,OAAO,CAAC,OAAO,CAAC,CAW3D;IAEK,eAAe,CAAC,EAAE,KAAK,EAAE,EAAE;QAAE,KAAK,EAAE,MAAM,CAAA;KAAE,GAAG,OAAO,CAAC,kBAAkB,GAAG,IAAI,CAAC,CAMtF;IAED;;OAEG;IACH,0BAA0B,IAAI,kBAAkB,EAAE,CASjD;IAED;;OAEG;IACG,oBAAoB,IAAI,OAAO,CAAC,IAAI,CAAC,CAc1C;IAED;;OAEG;IACG,mBAAmB,IAAI,OAAO,CAAC,IAAI,CAAC,CAczC;IAEK,mBAAmB,IAAI,OAAO,CAAC,IAAI,CAAC,CAKzC;IAEK,qBAAqB,CAAC,EAC1B,YAAY,EACZ,KAAK,EACL,MAAM,EACN,MAAM,EACN,cAAc,EACd,KAAK,EACL,KAAK,GACN,EAAE;QACD,YAAY,EAAE,MAAM,CAAC;QACrB,KAAK,EAAE,MAAM,CAAC;QACd,MAAM,EAAE,MAAM,CAAC;QACf,MAAM,EAAE,UAAU,CAAC,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,CAAC,CAAC;QACvC,cAAc,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;QACpC,KAAK,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;QAC5B,KAAK,CAAC,EAAE,QAAQ,CAAC;KAClB,GAAG,OAAO,CAAC,MAAM,CAAC,MAAM,EAAE,UAAU,CAAC,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,CAAC,CAAC,CAAC,CAuF1D;IACK,mBAAmB,CAAC,EACxB,YAAY,EACZ,KAAK,EACL,IAAI,EACJ,KAAK,GACN,EAAE;QACD,YAAY,EAAE,MAAM,CAAC;QACrB,KAAK,EAAE,MAAM,CAAC;QACd,IAAI,EAAE,0BAA0B,CAAC;QACjC,KAAK,CAAC,EAAE,QAAQ,CAAC;KAClB,GAAG,OAAO,CAAC,gBAAgB,GAAG,SAAS,CAAC,CAiExC;IAEK,uBAAuB,CAAC,EAC5B,YAAY,EACZ,KAAK,EACL,UAAU,EACV,QAAQ,EACR,SAAS,EACT,SAAS,EACT,KAAK,GACN,EAAE;QACD,YAAY,EAAE,MAAM,CAAC;QACrB,KAAK,EAAE,MAAM,CAAC;QACd,UAAU,CAAC,EAAE,MAAM,CAAC;QACpB,QAAQ,EAAE,gBAAgB,CAAC;QAC3B,SAAS,CAAC,EAAE,IAAI,CAAC;QACjB,SAAS,CAAC,EAAE,IAAI,CAAC;QACjB,KAAK,CAAC,EAAE,QAAQ,CAAC;KAClB,GAAG,OAAO,CAAC,IAAI,CAAC,CAmChB;IAEK,oBAAoB,CAAC,EACzB,YAAY,EACZ,KAAK,GACN,EAAE;QACD,YAAY,EAAE,MAAM,CAAC;QACrB,KAAK,EAAE,MAAM,CAAC;KACf,GAAG,OAAO,CAAC,gBAAgB,GAAG,IAAI,CAAC,CAwBnC;IAEK,gBAAgB,CAAC,IAAI,CAAC,EAAE,4BAA4B,GAAG,OAAO,CAAC,YAAY,CAAC,CA2EjF;IAEK,kBAAkB,CAAC,IAAI,EAAE;QAAE,KAAK,EAAE,MAAM,CAAC;QAAC,YAAY,CAAC,EAAE,MAAM,CAAA;KAAE,GAAG,OAAO,CAAC,WAAW,GAAG,IAAI,CAAC,CA4BpG;IAEK,qBAAqB,CAAC,EAC1B,KAAK,EACL,YAAY,EACZ,KAAK,GACN,EAAE;QACD,KAAK,EAAE,MAAM,CAAC;QACd,YAAY,EAAE,MAAM,CAAC;QACrB,KAAK,CAAC,EAAE,QAAQ,CAAC;KAClB,GAAG,OAAO,CAAC,IAAI,CAAC,CAmBhB;IAED,OAAO,CAAC,gBAAgB;CAoBzB"}

package/dist/storage/retention.d.ts

@@ -1,4 +1,4 @@
-import type { TABLE_NAMES, PruneOptions, PruneResult, TableRetentionPolicy } from '@mastra/core/storage';
+import type { TABLE_MEMORY_RUN_FENCES, TABLE_NAMES, TABLE_WORKFLOW_RUN_OWNERS, PruneOptions, PruneResult, TableRetentionPolicy } from '@mastra/core/storage';
 import type { MongoDBConnector } from './connectors/MongoDBConnector.js';
 export declare const DEFAULT_PRUNE_BATCH_SIZE = 1000;
 /**
@@ -12,10 +12,12 @@ export declare const DEFAULT_PRUNE_BATCH_SIZE = 1000;
  * - `epoch-ms`: a raw millisecond number, e.g. `schedule_triggers.actual_fire_at`.
  */
 export type AnchorEncoding = 'date' | 'iso' | 'epoch-ms';
+/** Collections `prune()` can target: the core tables plus the run fencing collections. */
+export type PrunableTable = TABLE_NAMES | typeof TABLE_WORKFLOW_RUN_OWNERS | typeof TABLE_MEMORY_RUN_FENCES;
 /** One collection to prune, resolved from a policy + the domain's descriptor. */
 export interface PruneTarget {
     /** Physical collection name. */
-    table: TABLE_NAMES;
+    table: PrunableTable;
     /** Anchor field for the age comparison. */
     column: string;
     /** Stored representation of the anchor field (decides the cutoff's type). */
@@ -47,12 +49,14 @@ export declare function cutoffFor(policy: TableRetentionPolicy, encoding: Anchor
  * Delete up to `limit` documents whose anchor field is older than `cutoff`.
  *
  * `deleteMany` has no limit clause, so each batch first collects `_id`s with a
- * bounded `find`, then deletes exactly that set — the Mongo equivalent of the
- * SQL stores' `DELETE ... WHERE rowid/ctid IN (SELECT ... LIMIT n)` pattern.
+ * bounded `find`, then deletes that set — the Mongo equivalent of the SQL
+ * stores' `DELETE ... WHERE rowid/ctid IN (SELECT ... LIMIT n)` pattern. The
+ * delete re-checks the anchor, so a document updated after the `find` (a
+ * renewed run ownership, a touched snapshot) survives.
  */
 export declare function pruneCollectionBatch({ connector, table, column, cutoff, limit, }: {
     connector: MongoDBConnector;
-    table: TABLE_NAMES;
+    table: PrunableTable;
     column: string;
     cutoff: Date | string | number;
     limit: number;

package/dist/storage/retention.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"retention.d.ts","sourceRoot":"","sources":["../../src/storage/retention.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,WAAW,EAAE,YAAY,EAAE,WAAW,EAAE,oBAAoB,EAAE,MAAM,sBAAsB,CAAC;AAEzG,OAAO,KAAK,EAAE,gBAAgB,EAAE,MAAM,+BAA+B,CAAC;AAEtE,eAAO,MAAM,wBAAwB,OAAO,CAAC;AAE7C;;;;;;;;;GASG;AACH,MAAM,MAAM,cAAc,GAAG,MAAM,GAAG,KAAK,GAAG,UAAU,CAAC;AAEzD,iFAAiF;AACjF,MAAM,WAAW,WAAW;IAC1B,gCAAgC;IAChC,KAAK,EAAE,WAAW,CAAC;IACnB,2CAA2C;IAC3C,MAAM,EAAE,MAAM,CAAC;IACf,6EAA6E;IAC7E,QAAQ,EAAE,cAAc,CAAC;IACzB,6EAA6E;IAC7E,OAAO,EAAE,OAAO,CAAC;IACjB,sDAAsD;IACtD,MAAM,EAAE,oBAAoB,CAAC;CAC9B;AAED,KAAK,eAAe,GAAG;IAAE,IAAI,CAAC,EAAE,CAAC,GAAG,EAAE,MAAM,EAAE,GAAG,CAAC,EAAE,OAAO,KAAK,IAAI,CAAA;CAAE,CAAC;AAEvE;;;;;;;;GAQG;AACH,wBAAsB,iBAAiB,CACrC,SAAS,EAAE,gBAAgB,EAC3B,MAAM,EAAE,IAAI,CAAC,WAAW,EAAE,OAAO,GAAG,QAAQ,GAAG,SAAS,CAAC,EACzD,MAAM,CAAC,EAAE,eAAe,GACvB,OAAO,CAAC,IAAI,CAAC,CAQf;AAiBD;;;GAGG;AACH,wBAAgB,SAAS,CAAC,MAAM,EAAE,oBAAoB,EAAE,QAAQ,EAAE,cAAc,EAAE,GAAG,SAAa,0BAKjG;AAED;;;;;;GAMG;AACH,wBAAsB,oBAAoB,CAAC,EACzC,SAAS,EACT,KAAK,EACL,MAAM,EACN,MAAM,EACN,KAAK,GACN,EAAE;IACD,SAAS,EAAE,gBAAgB,CAAC;IAC5B,KAAK,EAAE,WAAW,CAAC;IACnB,MAAM,EAAE,MAAM,CAAC;IACf,MAAM,EAAE,IAAI,GAAG,MAAM,GAAG,MAAM,CAAC;IAC/B,KAAK,EAAE,MAAM,CAAC;CACf,GAAG,OAAO,CAAC,MAAM,CAAC,CAUlB;AAED;;;;;GAKG;AACH,wBAAsB,gBAAgB,CAAC,EACrC,WAAW,EACX,SAAS,EACT,OAAO,GACR,EAAE;IACD,WAAW,EAAE,CAAC,KAAK,EAAE,MAAM,KAAK,OAAO,CAAC,MAAM,CAAC,CAAC;IAChD,SAAS,EAAE,MAAM,CAAC;IAClB,OAAO,CAAC,EAAE,YAAY,CAAC;CACxB,GAAG,OAAO,CAAC;IAAE,OAAO,EAAE,MAAM,CAAC;IAAC,IAAI,EAAE,OAAO,CAAA;CAAE,CAAC,CAkC9C;AAED;;;;;;;;;;;;;;;;GAgBG;AACH,wBAAsB,QAAQ,CAAC,EAC7B,SAAS,EACT,MAAM,EACN,OAAO,EACP,OAAO,EACP,MAAM,GACP,EAAE;IACD,SAAS,EAAE,gBAAgB,CAAC;IAC5B,MAAM,EAAE,MAAM,CAAC;IACf,OAAO,EAAE,WAAW,EAAE,CAAC;IACvB,OAAO,CAAC,EAAE,YAAY,CAAC;IACvB,MAAM,CAAC,EAAE,eAAe,CAAC;CAC1B,GAAG,OAAO,CAAC,WAAW,EAAE,CAAC,CA0BzB;AAED;;;;;;;;;;GAUG;AACH,wBAAgB,cAAc,CAAC,EAC7B,QAAQ,EACR,UAAU,EACV,KAAK,EACL,SAAS,GACV,EAAE;IACD,QAAQ,EAAE,MAAM,CAAC,MAAM,EAAE,oBAAoB,CAAC,CAAC;IAC/C,UAAU,EAAE,MAAM,CAChB,MAAM,EACN;QAAE,KAAK,EAAE,MAAM,CAAC;QAAC,MAAM,EAAE,MAAM,CAAC;QAAC,UAAU,CAAC,EAAE,WAAW,GAAG,UAAU,CAAC;QAAC,OAAO,CAAC,EAAE,OAAO,CAAA;KAAE,CAC5F,CAAC;IACF,KAAK,EAAE,MAAM,EAAE,CAAC;IAChB,SAAS,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,cAAc,CAAC,CAAC;CAC5C,GAAG,WAAW,EAAE,CAehB"}
+{"version":3,"file":"retention.d.ts","sourceRoot":"","sources":["../../src/storage/retention.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EACV,uBAAuB,EACvB,WAAW,EACX,yBAAyB,EACzB,YAAY,EACZ,WAAW,EACX,oBAAoB,EACrB,MAAM,sBAAsB,CAAC;AAE9B,OAAO,KAAK,EAAE,gBAAgB,EAAE,MAAM,+BAA+B,CAAC;AAEtE,eAAO,MAAM,wBAAwB,OAAO,CAAC;AAE7C;;;;;;;;;GASG;AACH,MAAM,MAAM,cAAc,GAAG,MAAM,GAAG,KAAK,GAAG,UAAU,CAAC;AAEzD,0FAA0F;AAC1F,MAAM,MAAM,aAAa,GAAG,WAAW,GAAG,OAAO,yBAAyB,GAAG,OAAO,uBAAuB,CAAC;AAE5G,iFAAiF;AACjF,MAAM,WAAW,WAAW;IAC1B,gCAAgC;IAChC,KAAK,EAAE,aAAa,CAAC;IACrB,2CAA2C;IAC3C,MAAM,EAAE,MAAM,CAAC;IACf,6EAA6E;IAC7E,QAAQ,EAAE,cAAc,CAAC;IACzB,6EAA6E;IAC7E,OAAO,EAAE,OAAO,CAAC;IACjB,sDAAsD;IACtD,MAAM,EAAE,oBAAoB,CAAC;CAC9B;AAED,KAAK,eAAe,GAAG;IAAE,IAAI,CAAC,EAAE,CAAC,GAAG,EAAE,MAAM,EAAE,GAAG,CAAC,EAAE,OAAO,KAAK,IAAI,CAAA;CAAE,CAAC;AAEvE;;;;;;;;GAQG;AACH,wBAAsB,iBAAiB,CACrC,SAAS,EAAE,gBAAgB,EAC3B,MAAM,EAAE,IAAI,CAAC,WAAW,EAAE,OAAO,GAAG,QAAQ,GAAG,SAAS,CAAC,EACzD,MAAM,CAAC,EAAE,eAAe,GACvB,OAAO,CAAC,IAAI,CAAC,CAQf;AAiBD;;;GAGG;AACH,wBAAgB,SAAS,CAAC,MAAM,EAAE,oBAAoB,EAAE,QAAQ,EAAE,cAAc,EAAE,GAAG,SAAa,0BAKjG;AAED;;;;;;;;GAQG;AACH,wBAAsB,oBAAoB,CAAC,EACzC,SAAS,EACT,KAAK,EACL,MAAM,EACN,MAAM,EACN,KAAK,GACN,EAAE;IACD,SAAS,EAAE,gBAAgB,CAAC;IAC5B,KAAK,EAAE,aAAa,CAAC;IACrB,MAAM,EAAE,MAAM,CAAC;IACf,MAAM,EAAE,IAAI,GAAG,MAAM,GAAG,MAAM,CAAC;IAC/B,KAAK,EAAE,MAAM,CAAC;CACf,GAAG,OAAO,CAAC,MAAM,CAAC,CAUlB;AAED;;;;;GAKG;AACH,wBAAsB,gBAAgB,CAAC,EACrC,WAAW,EACX,SAAS,EACT,OAAO,GACR,EAAE;IACD,WAAW,EAAE,CAAC,KAAK,EAAE,MAAM,KAAK,OAAO,CAAC,MAAM,CAAC,CAAC;IAChD,SAAS,EAAE,MAAM,CAAC;IAClB,OAAO,CAAC,EAAE,YAAY,CAAC;CACxB,GAAG,OAAO,CAAC;IAAE,OAAO,EAAE,MAAM,CAAC;IAAC,IAAI,EAAE,OAAO,CAAA;CAAE,CAAC,CAkC9C;AAED;;;;;;;;;;;;;;;;GAgBG;AACH,wBAAsB,QAAQ,CAAC,EAC7B,SAAS,EACT,MAAM,EACN,OAAO,EACP,OAAO,EACP,MAAM,GACP,EAAE;IACD,SAAS,EAAE,gBAAgB,CAAC;IAC5B,MAAM,EAAE,MAAM,CAAC;IACf,OAAO,EAAE,WAAW,EAAE,CAAC;IACvB,OAAO,CAAC,EAAE,YAAY,CAAC;IACvB,MAAM,CAAC,EAAE,eAAe,CAAC;CAC1B,GAAG,OAAO,CAAC,WAAW,EAAE,CAAC,CA0BzB;AAED;;;;;;;;;;GAUG;AACH,wBAAgB,cAAc,CAAC,EAC7B,QAAQ,EACR,UAAU,EACV,KAAK,EACL,SAAS,GACV,EAAE;IACD,QAAQ,EAAE,MAAM,CAAC,MAAM,EAAE,oBAAoB,CAAC,CAAC;IAC/C,UAAU,EAAE,MAAM,CAChB,MAAM,EACN;QAAE,KAAK,EAAE,MAAM,CAAC;QAAC,MAAM,EAAE,MAAM,CAAC;QAAC,UAAU,CAAC,EAAE,WAAW,GAAG,UAAU,CAAC;QAAC,OAAO,CAAC,EAAE,OAAO,CAAA;KAAE,CAC5F,CAAC;IACF,KAAK,EAAE,MAAM,EAAE,CAAC;IAChB,SAAS,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,cAAc,CAAC,CAAC;CAC5C,GAAG,WAAW,EAAE,CAehB"}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@mastra/mongodb",
-  "version": "1.23.0",
+  "version": "1.23.1-alpha.1",
   "description": "MongoDB provider for Mastra - includes vector store capabilities",
   "type": "module",
   "main": "dist/index.js",
@@ -32,14 +32,15 @@
     "tsx": "^4.23.1",
     "typescript": "^7.0.2",
     "vitest": "4.1.11",
+    "@internal/core": "0.1.2",
     "@internal/lint": "0.0.142",
     "@internal/storage-test-utils": "0.0.138",
     "@internal/types-builder": "0.0.117",
-    "@mastra/core": "1.76.0",
-    "@mastra/memory": "1.37.0"
+    "@mastra/memory": "1.37.1-alpha.3",
+    "@mastra/core": "1.77.0-alpha.4"
   },
   "peerDependencies": {
-    "@mastra/core": ">=1.68.0-0 <2.0.0-0"
+    "@mastra/core": ">=1.76.0-0 <2.0.0-0"
   },
   "files": [
     "dist"
@@ -62,7 +63,7 @@
     "pretest": "docker compose up -d --wait || (echo '--- docker compose up failed; dumping diagnostics ---' && docker compose ps -a && docker compose logs && df -h && exit 1)",
     "test": "vitest run && pnpm test:fidelity",
     "pretest:fidelity": "docker compose --profile fidelity up -d --wait mongodb-fidelity",
-    "test:fidelity": "MONGODB_REPLICA_SET_URL='mongodb://localhost:27019/?directConnection=true' vitest run src/storage/dataset-fidelity.test.ts",
+    "test:fidelity": "MONGODB_REPLICA_SET_URL='mongodb://localhost:27019/?directConnection=true' vitest run src/storage/dataset-fidelity.test.ts src/storage/run-fencing.test.ts",
     "posttest:fidelity": "docker compose --profile fidelity stop mongodb-fidelity",
     "posttest": "docker compose down -v",
     "pretest:watch": "docker compose up -d",