@mastra/mongodb 1.14.0 → 1.15.0-alpha.0

package/dist/index.js

@@ -1,6 +1,6 @@
 import { v4 } from '@lukeed/uuid';
 import { MastraError, ErrorCategory, ErrorDomain } from '@mastra/core/error';
-import { createVectorErrorId, AgentsStorage, TABLE_AGENTS, TABLE_AGENT_VERSIONS, createStorageErrorId, normalizePerPage, calculatePagination, BackgroundTasksStorage, TABLE_BACKGROUND_TASKS, BlobStore, TABLE_SKILL_BLOBS, DatasetsStorage, TABLE_DATASETS, TABLE_DATASET_ITEMS, TABLE_DATASET_VERSIONS, ensureDate, safelyParseJSON, hasErrorCode, TABLE_EXPERIMENTS, TABLE_EXPERIMENT_RESULTS, ExperimentsStorage, MCPClientsStorage, TABLE_MCP_CLIENTS, TABLE_MCP_CLIENT_VERSIONS, MCPServersStorage, TABLE_MCP_SERVERS, TABLE_MCP_SERVER_VERSIONS, MemoryStorage, TABLE_THREADS, TABLE_MESSAGES, TABLE_RESOURCES, NotificationsStorage, TABLE_NOTIFICATIONS, ObservabilityStorage, TABLE_SPANS, listTracesArgsSchema, toTraceSpans, PromptBlocksStorage, TABLE_PROMPT_BLOCKS, TABLE_PROMPT_BLOCK_VERSIONS, SchedulesStorage, TABLE_SCHEDULES, TABLE_SCHEDULE_TRIGGERS, ScorerDefinitionsStorage, TABLE_SCORER_DEFINITIONS, TABLE_SCORER_DEFINITION_VERSIONS, ScoresStorage, TABLE_SCORERS, SkillsStorage, TABLE_SKILLS, TABLE_SKILL_VERSIONS, WorkflowsStorage, TABLE_WORKFLOW_SNAPSHOT, WorkspacesStorage, TABLE_WORKSPACES, TABLE_WORKSPACE_VERSIONS, MastraCompositeStore, TraceStatus, parseDuration, normalizeScheduleTarget, transformScoreRow as transformScoreRow$1 } from '@mastra/core/storage';
+import { createVectorErrorId, AgentsStorage, TABLE_AGENTS, TABLE_AGENT_VERSIONS, createStorageErrorId, normalizePerPage, calculatePagination, BackgroundTasksStorage, TABLE_BACKGROUND_TASKS, BlobStore, TABLE_SKILL_BLOBS, DatasetsStorage, TABLE_DATASETS, TABLE_DATASET_ITEMS, TABLE_DATASET_VERSIONS, ensureDate, safelyParseJSON, hasErrorCode, TABLE_EXPERIMENTS, TABLE_EXPERIMENT_RESULTS, ExperimentsStorage, MCPClientsStorage, TABLE_MCP_CLIENTS, TABLE_MCP_CLIENT_VERSIONS, MCPServersStorage, TABLE_MCP_SERVERS, TABLE_MCP_SERVER_VERSIONS, MemoryStorage, TABLE_THREADS, TABLE_MESSAGES, TABLE_RESOURCES, validateStorageMetadataFilter, storageMessageMatchesMetadataFilter, NotificationsStorage, TABLE_NOTIFICATIONS, ObservabilityStorage, TABLE_SPANS, listTracesArgsSchema, toTraceSpans, PromptBlocksStorage, TABLE_PROMPT_BLOCKS, TABLE_PROMPT_BLOCK_VERSIONS, SchedulesStorage, TABLE_SCHEDULES, TABLE_SCHEDULE_TRIGGERS, ScorerDefinitionsStorage, TABLE_SCORER_DEFINITIONS, TABLE_SCORER_DEFINITION_VERSIONS, ScoresStorage, TABLE_SCORERS, SkillsStorage, TABLE_SKILLS, TABLE_SKILL_VERSIONS, WorkflowsStorage, TABLE_WORKFLOW_SNAPSHOT, WorkspacesStorage, TABLE_WORKSPACES, TABLE_WORKSPACE_VERSIONS, MastraCompositeStore, TraceStatus, parseDuration, normalizeScheduleTarget, transformScoreRow as transformScoreRow$1 } from '@mastra/core/storage';
 import { MastraVector, validateUpsertInput, validateVectorValues } from '@mastra/core/vector';
 import { MongoClient, MongoBulkWriteError } from 'mongodb';
 import { BaseFilterTranslator } from '@mastra/core/vector/filter';
@@ -13,7 +13,7 @@ import { skillSnapshotFieldValuesEqual } from '@mastra/core/storage/domains/skil
 
 // package.json
 var package_default = {
-  version: "1.14.0"};
+  version: "1.15.0-alpha.0"};
 var MongoDBFilterTranslator = class extends BaseFilterTranslator {
   getSupportedOperators() {
     return {
@@ -5271,6 +5271,7 @@ var MemoryStorageMongoDB = class _MemoryStorageMongoDB extends MemoryStorage {
   }
   async listMessages(args) {
     const { threadId, resourceId, include, filter, perPage: perPageInput, page = 0, orderBy } = args;
+    const metadataFilter = validateStorageMetadataFilter(filter?.metadata);
     const threadIds = Array.isArray(threadId) ? threadId : [threadId];
     if (threadIds.length === 0 || threadIds.some((id) => !id.trim())) {
       throw new MastraError(
@@ -5326,16 +5327,27 @@ var MemoryStorageMongoDB = class _MemoryStorageMongoDB extends MemoryStorage {
           hasMore: false
         };
       }
-      const total = await collection.countDocuments(query);
       const messages = [];
+      let total = 0;
       if (perPage !== 0) {
         const sortObj = { [field]: sortOrder };
-        let cursor = collection.find(query).sort(sortObj).skip(offset);
-        if (perPageInput !== false) {
-          cursor = cursor.limit(perPage);
+        if (metadataFilter) {
+          const candidates = (await collection.find(query).sort(sortObj).toArray()).map((row) => this.parseRow(row)).filter((message) => storageMessageMatchesMetadataFilter(message.content, metadataFilter));
+          total = candidates.length;
+          messages.push(...perPageInput === false ? candidates : candidates.slice(offset, offset + perPage));
+        } else {
+          total = await collection.countDocuments(query);
+          let cursor = collection.find(query).sort(sortObj).skip(offset);
+          if (perPageInput !== false) {
+            cursor = cursor.limit(perPage);
+          }
+          const dataResult = await cursor.toArray();
+          messages.push(...dataResult.map((row) => this.parseRow(row)));
         }
-        const dataResult = await cursor.toArray();
-        messages.push(...dataResult.map((row) => this.parseRow(row)));
+      } else if (metadataFilter) {
+        total = (await collection.find(query).toArray()).map((row) => this.parseRow(row)).filter((message) => storageMessageMatchesMetadataFilter(message.content, metadataFilter)).length;
+      } else {
+        total = await collection.countDocuments(query);
       }
       if (total === 0 && messages.length === 0 && (!include || include.length === 0)) {
         return {
@@ -5365,7 +5377,7 @@ var MemoryStorageMongoDB = class _MemoryStorageMongoDB extends MemoryStorage {
         finalMessages.filter((m) => m.threadId && threadIdSet.has(m.threadId)).map((m) => m.id)
       );
       const allThreadMessagesReturned = returnedThreadMessageIds.size >= total;
-      const hasMore = perPageInput !== false && !allThreadMessagesReturned && offset + perPage < total;
+      const hasMore = metadataFilter ? perPageInput !== false && offset + perPage < total : perPageInput !== false && !allThreadMessagesReturned && offset + perPage < total;
       return {
         messages: finalMessages,
         total,
@@ -5399,6 +5411,7 @@ var MemoryStorageMongoDB = class _MemoryStorageMongoDB extends MemoryStorage {
   }
   async listMessagesByResourceId(args) {
     const { resourceId, include, filter, perPage: perPageInput, page = 0, orderBy } = args;
+    const metadataFilter = validateStorageMetadataFilter(filter?.metadata);
     if (!resourceId || typeof resourceId !== "string" || resourceId.trim().length === 0) {
       throw new MastraError(
         {
@@ -5454,16 +5467,27 @@ var MemoryStorageMongoDB = class _MemoryStorageMongoDB extends MemoryStorage {
           hasMore: false
         };
       }
-      const total = await collection.countDocuments(query);
       const messages = [];
+      let total = 0;
       if (perPage !== 0) {
         const sortObj = { [field]: sortOrder };
-        let cursor = collection.find(query).sort(sortObj).skip(offset);
-        if (perPageInput !== false) {
-          cursor = cursor.limit(perPage);
+        if (metadataFilter) {
+          const candidates = (await collection.find(query).sort(sortObj).toArray()).map((row) => this.parseRow(row)).filter((message) => storageMessageMatchesMetadataFilter(message.content, metadataFilter));
+          total = candidates.length;
+          messages.push(...perPageInput === false ? candidates : candidates.slice(offset, offset + perPage));
+        } else {
+          total = await collection.countDocuments(query);
+          let cursor = collection.find(query).sort(sortObj).skip(offset);
+          if (perPageInput !== false) {
+            cursor = cursor.limit(perPage);
+          }
+          const dataResult = await cursor.toArray();
+          messages.push(...dataResult.map((row) => this.parseRow(row)));
         }
-        const dataResult = await cursor.toArray();
-        messages.push(...dataResult.map((row) => this.parseRow(row)));
+      } else if (metadataFilter) {
+        total = (await collection.find(query).toArray()).map((row) => this.parseRow(row)).filter((message) => storageMessageMatchesMetadataFilter(message.content, metadataFilter)).length;
+      } else {
+        total = await collection.countDocuments(query);
       }
       if (total === 0 && messages.length === 0 && (!include || include.length === 0)) {
         return {