@mastra/core 0.16.1-alpha.0 → 0.16.1-alpha.1

package/dist/storage/index.cjs

@@ -2,7 +2,7 @@
 
 var chunk33GDW3SI_cjs = require('../chunk-33GDW3SI.cjs');
 var chunkXUAWHRUO_cjs = require('../chunk-XUAWHRUO.cjs');
-var chunkKT4WNDGP_cjs = require('../chunk-KT4WNDGP.cjs');
+var chunkD3UG2YLU_cjs = require('../chunk-D3UG2YLU.cjs');
 var chunkC73WLCY3_cjs = require('../chunk-C73WLCY3.cjs');
 var chunkKXCUCBEI_cjs = require('../chunk-KXCUCBEI.cjs');
 
@@ -311,7 +311,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 chunkKT4WNDGP_cjs.MessageList().add(rawMessages.map(this.parseStoredMessage), "memory");
+    const list = new chunkD3UG2YLU_cjs.MessageList().add(rawMessages.map(this.parseStoredMessage), "memory");
     if (format === "v1") return list.get.all.v1();
     return list.get.all.v2();
   }
@@ -341,7 +341,7 @@ var InMemoryMemory = class extends MemoryStorage {
       };
       this.collection.messages.set(key, storageMessage);
     }
-    const list = new chunkKT4WNDGP_cjs.MessageList().add(messages, "memory");
+    const list = new chunkD3UG2YLU_cjs.MessageList().add(messages, "memory");
     if (format === `v2`) return list.get.all.v2();
     return list.get.all.v1();
   }
@@ -816,6 +816,8 @@ var ObservabilityInMemory = class extends ObservabilityStorage {
     return spans.filter((span) => {
       if (filter?.name && span.name !== filter.name) return false;
       if (filter?.spanType && span.spanType !== filter.spanType) return false;
+      if (filter?.entityType === "agent" && filter.entityId !== span.attributes?.agentId) return false;
+      if (filter?.entityType === "workflow" && filter.entityId !== span.attributes?.workflowId) return false;
       return true;
     });
   }