@mastra/core 0.15.3-alpha.4 → 0.15.3-alpha.6

package/dist/storage/index.cjs

@@ -2,7 +2,7 @@
 
 var chunkK6UMYGK5_cjs = require('../chunk-K6UMYGK5.cjs');
 var chunkXUAWHRUO_cjs = require('../chunk-XUAWHRUO.cjs');
-var chunk5CJDO3UO_cjs = require('../chunk-5CJDO3UO.cjs');
+var chunkAAJF7WLJ_cjs = require('../chunk-AAJF7WLJ.cjs');
 var chunkC73WLCY3_cjs = require('../chunk-C73WLCY3.cjs');
 var chunkKXCUCBEI_cjs = require('../chunk-KXCUCBEI.cjs');
 
@@ -310,7 +310,7 @@ var InMemoryMemory = class extends MemoryStorage {
   }) {
     this.logger.debug(`MockStore: getMessagesById called`);
     const rawMessages = messageIds.map((id) => this.collection.messages.get(id)).filter((message) => !!message);
-    const list = new chunk5CJDO3UO_cjs.MessageList().add(rawMessages.map(this.parseStoredMessage), "memory");
+    const list = new chunkAAJF7WLJ_cjs.MessageList().add(rawMessages.map(this.parseStoredMessage), "memory");
     if (format === "v1") return list.get.all.v1();
     return list.get.all.v2();
   }
@@ -340,7 +340,7 @@ var InMemoryMemory = class extends MemoryStorage {
       };
       this.collection.messages.set(key, storageMessage);
     }
-    const list = new chunk5CJDO3UO_cjs.MessageList().add(messages, "memory");
+    const list = new chunkAAJF7WLJ_cjs.MessageList().add(messages, "memory");
     if (format === `v2`) return list.get.all.v2();
     return list.get.all.v1();
   }
@@ -937,7 +937,7 @@ var StoreOperationsInMemory = class extends StoreOperations {
     const table = this.data[tableName];
     let key = record.id;
     if ([chunkXUAWHRUO_cjs.TABLE_WORKFLOW_SNAPSHOT, chunkXUAWHRUO_cjs.TABLE_EVALS].includes(tableName) && !record.id && record.run_id) {
-      key = record.run_id;
+      key = record.workflow_name ? `${record.workflow_name}-${record.run_id}` : record.run_id;
       record.id = key;
     } else if (!record.id) {
       key = `auto-${Date.now()}-${Math.random()}`;
@@ -1199,7 +1199,7 @@ var WorkflowsInMemory = class extends WorkflowsStorage {
     runtimeContext
   }) {
     this.logger.debug(`MockStore: updateWorkflowResults called for ${workflowName} ${runId} ${stepId}`, result);
-    const run = this.collection.get(runId);
+    const run = this.collection.get(`${workflowName}-${runId}`);
     if (!run) {
       return {};
     }
@@ -1216,7 +1216,7 @@ var WorkflowsInMemory = class extends WorkflowsStorage {
         status: "pending",
         runId: run.run_id
       };
-      this.collection.set(runId, {
+      this.collection.set(`${workflowName}-${runId}`, {
         ...run,
         snapshot
       });
@@ -1227,17 +1227,18 @@ var WorkflowsInMemory = class extends WorkflowsStorage {
     }
     snapshot.context[stepId] = result;
     snapshot.runtimeContext = { ...snapshot.runtimeContext, ...runtimeContext };
-    this.collection.set(runId, {
+    this.collection.set(`${workflowName}-${runId}`, {
       ...run,
       snapshot
     });
     return JSON.parse(JSON.stringify(snapshot.context));
   }
   async updateWorkflowState({
+    workflowName,
     runId,
     opts
   }) {
-    const run = this.collection.get(runId);
+    const run = this.collection.get(`${workflowName}-${runId}`);
     if (!run) {
       return;
     }
@@ -1254,7 +1255,7 @@ var WorkflowsInMemory = class extends WorkflowsStorage {
         status: "pending",
         runId: run.run_id
       };
-      this.collection.set(runId, {
+      this.collection.set(`${workflowName}-${runId}`, {
         ...run,
         snapshot
       });
@@ -1265,7 +1266,7 @@ var WorkflowsInMemory = class extends WorkflowsStorage {
       throw new Error(`Snapshot not found for runId ${runId}`);
     }
     snapshot = { ...snapshot, ...opts };
-    this.collection.set(runId, {
+    this.collection.set(`${workflowName}-${runId}`, {
       ...run,
       snapshot
     });
@@ -1340,10 +1341,8 @@ var WorkflowsInMemory = class extends WorkflowsStorage {
     runId,
     workflowName
   }) {
-    let run = Array.from(this.collection.values()).find((r) => r.run_id === runId);
-    if (run && workflowName && run.workflow_name !== workflowName) {
-      run = void 0;
-    }
+    const runs = Array.from(this.collection.values()).filter((r) => r.run_id === runId);
+    let run = runs.find((r) => r.workflow_name === workflowName);
     if (!run) return null;
     const parsedRun = {
       ...run,