@mastra/mongodb 0.12.2 → 0.12.3-alpha.1

package/.turbo/turbo-build.log

@@ -1,23 +1,23 @@
 
-> @mastra/mongodb@0.12.1 build /home/runner/work/mastra/mastra/stores/mongodb
+> @mastra/mongodb@0.12.3-alpha.1 build /home/runner/work/mastra/mastra/stores/mongodb
 > tsup src/index.ts --format esm,cjs --experimental-dts --clean --treeshake=smallest --splitting
 
 CLI Building entry: src/index.ts
 CLI Using tsconfig: tsconfig.json
 CLI tsup v8.5.0
 TSC Build start
-TSC ⚡️ Build success in 10902ms
+TSC ⚡️ Build success in 11313ms
 DTS Build start
 CLI Target: es2022
 Analysis will use the bundled TypeScript version 5.8.3
 Writing package typings: /home/runner/work/mastra/mastra/stores/mongodb/dist/_tsup-dts-rollup.d.ts
 Analysis will use the bundled TypeScript version 5.8.3
 Writing package typings: /home/runner/work/mastra/mastra/stores/mongodb/dist/_tsup-dts-rollup.d.cts
-DTS ⚡️ Build success in 13195ms
+DTS ⚡️ Build success in 14299ms
 CLI Cleaning output folder
 ESM Build start
 CJS Build start
-ESM dist/index.js 73.11 KB
-ESM ⚡️ Build success in 1823ms
-CJS dist/index.cjs 74.40 KB
-CJS ⚡️ Build success in 1832ms
+CJS dist/index.cjs 74.73 KB
+CJS ⚡️ Build success in 1938ms
+ESM dist/index.js 73.39 KB
+ESM ⚡️ Build success in 1947ms

package/CHANGELOG.md

@@ -1,5 +1,42 @@
 # @mastra/mongodb
 
+## 0.12.3-alpha.1
+
+### Patch Changes
+
+- 9881232: dependencies updates:
+  - Updated dependency [`cloudflare@^4.5.0` ↗︎](https://www.npmjs.com/package/cloudflare/v/4.5.0) (from `^4.4.1`, in `dependencies`)
+- eba48c8: Sort getThreads by updatedAt.
+- 14456f3: fix: [MongoDB] batchInsert with process json
+- Updated dependencies [27cc97a]
+- Updated dependencies [41daa63]
+- Updated dependencies [254a36b]
+- Updated dependencies [0b89602]
+- Updated dependencies [4d37822]
+- Updated dependencies [ff9c125]
+- Updated dependencies [b8efbb9]
+- Updated dependencies [71466e7]
+- Updated dependencies [0c99fbe]
+  - @mastra/core@0.12.0-alpha.2
+
+## 0.12.3-alpha.0
+
+### Patch Changes
+
+- 25f2987: fix: filter.\_id.\$in cannot be empty
+- b0f1e2f: fix: mongodb save scores as an object
+- Updated dependencies [510e2c8]
+- Updated dependencies [2f72fb2]
+- Updated dependencies [3f89307]
+- Updated dependencies [9eda7d4]
+- Updated dependencies [9d49408]
+- Updated dependencies [2ecf658]
+- Updated dependencies [7a7754f]
+- Updated dependencies [fc92d80]
+- Updated dependencies [23a6a7c]
+- Updated dependencies [09bca64]
+  - @mastra/core@0.12.0-alpha.0
+
 ## 0.12.2
 
 ### Patch Changes

package/dist/_tsup-dts-rollup.d.cts

@@ -48,15 +48,12 @@ import type { WorkflowRuns } from '@mastra/core/storage';
 import type { WorkflowRunState } from '@mastra/core/workflows';
 import { WorkflowsStorage } from '@mastra/core/storage';
 
-export declare interface ConnectorHandler {
-    getCollection(collectionName: string): Promise<Collection>;
-    close(): Promise<void>;
-}
-
-export declare interface ConnectorHandler_alias_1 {
+declare interface ConnectorHandler {
     getCollection(collectionName: string): Promise<Collection>;
     close(): Promise<void>;
 }
+export { ConnectorHandler }
+export { ConnectorHandler as ConnectorHandler_alias_1 }
 
 export declare function createExecuteOperationWithRetry({ logger, maxRetries, initialBackoffMs, }: {
     logger: IMastraLogger;
@@ -173,7 +170,7 @@ export { MONGODB_PROMPT as MONGODB_PROMPT_alias_1 }
 declare type MongoDBBlacklisted = BlacklistedRootOperators | '$size';
 
 export declare type MongoDBConfig = DatabaseConfig | {
-    connectorHandler: ConnectorHandler_alias_1;
+    connectorHandler: ConnectorHandler;
 };
 
 export declare class MongoDBConnector {
@@ -190,7 +187,7 @@ export declare class MongoDBConnector_alias_1 {
     #private;
     constructor(options: MongoDBConnectorOptions_2);
     static fromDatabaseConfig(config: DatabaseConfig): MongoDBConnector_alias_1;
-    static fromConnectionHandler(handler: ConnectorHandler_alias_1): MongoDBConnector_alias_1;
+    static fromConnectionHandler(handler: ConnectorHandler): MongoDBConnector_alias_1;
     private getConnection;
     getCollection(collectionName: string): Promise<Collection<Document>>;
     close(): Promise<void>;
@@ -213,7 +210,7 @@ declare type MongoDBConnectorOptions_2 = {
 } | {
     client: undefined;
     dbName: undefined;
-    handler: ConnectorHandler_alias_1;
+    handler: ConnectorHandler;
 };
 
 /**
@@ -238,7 +235,7 @@ export { MongoDBIndexReadyParams }
 export { MongoDBIndexReadyParams as MongoDBIndexReadyParams_alias_1 }
 
 export declare interface MongoDBOperationsConfig {
-    connector: ConnectorHandler_alias_1;
+    connector: ConnectorHandler;
 }
 
 declare type MongoDBOperatorValueMap = Omit<OperatorValueMap, '$options'> & {
@@ -257,6 +254,7 @@ declare class MongoDBStore extends MastraStorage {
         resourceWorkingMemory: boolean;
         hasColumn: boolean;
         createTable: boolean;
+        deleteMessages: boolean;
     };
     constructor(config: MongoDBConfig);
     createTable({ tableName, schema, }: {
@@ -546,6 +544,7 @@ export declare class StoreOperationsMongoDB extends StoreOperations {
     dropTable({ tableName }: {
         tableName: TABLE_NAMES;
     }): Promise<void>;
+    private processJsonbFields;
     insert({ tableName, record }: {
         tableName: TABLE_NAMES;
         record: Record<string, any>;

package/dist/_tsup-dts-rollup.d.ts

@@ -48,15 +48,12 @@ import type { WorkflowRuns } from '@mastra/core/storage';
 import type { WorkflowRunState } from '@mastra/core/workflows';
 import { WorkflowsStorage } from '@mastra/core/storage';
 
-export declare interface ConnectorHandler {
-    getCollection(collectionName: string): Promise<Collection>;
-    close(): Promise<void>;
-}
-
-export declare interface ConnectorHandler_alias_1 {
+declare interface ConnectorHandler {
     getCollection(collectionName: string): Promise<Collection>;
     close(): Promise<void>;
 }
+export { ConnectorHandler }
+export { ConnectorHandler as ConnectorHandler_alias_1 }
 
 export declare function createExecuteOperationWithRetry({ logger, maxRetries, initialBackoffMs, }: {
     logger: IMastraLogger;
@@ -173,7 +170,7 @@ export { MONGODB_PROMPT as MONGODB_PROMPT_alias_1 }
 declare type MongoDBBlacklisted = BlacklistedRootOperators | '$size';
 
 export declare type MongoDBConfig = DatabaseConfig | {
-    connectorHandler: ConnectorHandler_alias_1;
+    connectorHandler: ConnectorHandler;
 };
 
 export declare class MongoDBConnector {
@@ -190,7 +187,7 @@ export declare class MongoDBConnector_alias_1 {
     #private;
     constructor(options: MongoDBConnectorOptions_2);
     static fromDatabaseConfig(config: DatabaseConfig): MongoDBConnector_alias_1;
-    static fromConnectionHandler(handler: ConnectorHandler_alias_1): MongoDBConnector_alias_1;
+    static fromConnectionHandler(handler: ConnectorHandler): MongoDBConnector_alias_1;
     private getConnection;
     getCollection(collectionName: string): Promise<Collection<Document>>;
     close(): Promise<void>;
@@ -213,7 +210,7 @@ declare type MongoDBConnectorOptions_2 = {
 } | {
     client: undefined;
     dbName: undefined;
-    handler: ConnectorHandler_alias_1;
+    handler: ConnectorHandler;
 };
 
 /**
@@ -238,7 +235,7 @@ export { MongoDBIndexReadyParams }
 export { MongoDBIndexReadyParams as MongoDBIndexReadyParams_alias_1 }
 
 export declare interface MongoDBOperationsConfig {
-    connector: ConnectorHandler_alias_1;
+    connector: ConnectorHandler;
 }
 
 declare type MongoDBOperatorValueMap = Omit<OperatorValueMap, '$options'> & {
@@ -257,6 +254,7 @@ declare class MongoDBStore extends MastraStorage {
         resourceWorkingMemory: boolean;
         hasColumn: boolean;
         createTable: boolean;
+        deleteMessages: boolean;
     };
     constructor(config: MongoDBConfig);
     createTable({ tableName, schema, }: {
@@ -546,6 +544,7 @@ export declare class StoreOperationsMongoDB extends StoreOperations {
     dropTable({ tableName }: {
         tableName: TABLE_NAMES;
     }): Promise<void>;
+    private processJsonbFields;
     insert({ tableName, record }: {
         tableName: TABLE_NAMES;
         record: Record<string, any>;

package/dist/index.cjs

@@ -343,7 +343,9 @@ var MongoDBVector = class extends vector.MastraVector {
       };
       if (Object.keys(combinedFilter).length > 0) {
         const candidateIds = await collection.aggregate([{ $match: combinedFilter }, { $project: { _id: 1 } }]).map((doc) => doc._id).toArray();
-        vectorSearch.filter = { _id: { $in: candidateIds } };
+        if (candidateIds.length > 0) {
+          vectorSearch.filter = { _id: { $in: candidateIds } };
+        }
       }
       const pipeline = [
         {
@@ -1211,7 +1213,7 @@ var MemoryStorageMongoDB = class extends storage.MemoryStorage {
   async getThreadsByResourceId({ resourceId }) {
     try {
       const collection = await this.operations.getCollection(storage.TABLE_THREADS);
-      const results = await collection.find({ resourceId }).toArray();
+      const results = await collection.find({ resourceId }).sort({ updatedAt: -1 }).toArray();
       if (!results.length) {
         return [];
       }
@@ -1411,18 +1413,21 @@ var StoreOperationsMongoDB = class extends storage.StoreOperations {
       );
     }
   }
+  processJsonbFields(tableName, record) {
+    const schema = storage.TABLE_SCHEMAS[tableName];
+    return Object.fromEntries(
+      Object.entries(schema).map(([key, value]) => {
+        if (value.type === "jsonb" && record[key] && typeof record[key] === "string") {
+          return [key, storage.safelyParseJSON(record[key])];
+        }
+        return [key, record[key]];
+      })
+    );
+  }
   async insert({ tableName, record }) {
     try {
       const collection = await this.getCollection(tableName);
-      const schema = storage.TABLE_SCHEMAS[tableName];
-      const recordToInsert = Object.fromEntries(
-        Object.entries(schema).map(([key, value]) => {
-          if (value.type === "jsonb" && record[key] && typeof record[key] === "string") {
-            return [key, storage.safelyParseJSON(record[key])];
-          }
-          return [key, record[key]];
-        })
-      );
+      const recordToInsert = this.processJsonbFields(tableName, record);
       await collection.insertOne(recordToInsert);
     } catch (error$1) {
       if (error$1 instanceof Error) {
@@ -1446,7 +1451,8 @@ var StoreOperationsMongoDB = class extends storage.StoreOperations {
     }
     try {
       const collection = await this.getCollection(tableName);
-      await collection.insertMany(records);
+      const processedRecords = records.map((record) => this.processJsonbFields(tableName, record));
+      await collection.insertMany(processedRecords);
     } catch (error$1) {
       throw new error.MastraError(
         {
@@ -1598,19 +1604,19 @@ var ScoresStorageMongoDB = class extends storage.ScoresStorage {
         scorerId: score.scorerId,
         traceId: score.traceId || "",
         runId: score.runId,
-        scorer: typeof score.scorer === "string" ? score.scorer : JSON.stringify(score.scorer),
-        extractStepResult: typeof score.extractStepResult === "string" ? score.extractStepResult : JSON.stringify(score.extractStepResult),
-        analyzeStepResult: typeof score.analyzeStepResult === "string" ? score.analyzeStepResult : JSON.stringify(score.analyzeStepResult),
+        scorer: typeof score.scorer === "string" ? storage.safelyParseJSON(score.scorer) : score.scorer,
+        extractStepResult: typeof score.extractStepResult === "string" ? storage.safelyParseJSON(score.extractStepResult) : score.extractStepResult,
+        analyzeStepResult: typeof score.analyzeStepResult === "string" ? storage.safelyParseJSON(score.analyzeStepResult) : score.analyzeStepResult,
         score: score.score,
         reason: score.reason,
         extractPrompt: score.extractPrompt,
         analyzePrompt: score.analyzePrompt,
         reasonPrompt: score.reasonPrompt,
-        input: typeof score.input === "string" ? score.input : JSON.stringify(score.input),
-        output: typeof score.output === "string" ? score.output : JSON.stringify(score.output),
+        input: typeof score.input === "string" ? storage.safelyParseJSON(score.input) : score.input,
+        output: typeof score.output === "string" ? storage.safelyParseJSON(score.output) : score.output,
         additionalContext: score.additionalContext,
-        runtimeContext: typeof score.runtimeContext === "string" ? score.runtimeContext : JSON.stringify(score.runtimeContext),
-        entity: typeof score.entity === "string" ? score.entity : JSON.stringify(score.entity),
+        runtimeContext: typeof score.runtimeContext === "string" ? storage.safelyParseJSON(score.runtimeContext) : score.runtimeContext,
+        entity: typeof score.entity === "string" ? storage.safelyParseJSON(score.entity) : score.entity,
         source: score.source,
         resourceId: score.resourceId || "",
         threadId: score.threadId || "",
@@ -2054,6 +2060,40 @@ var WorkflowsStorageMongoDB = class extends storage.WorkflowsStorage {
 };
 
 // src/storage/index.ts
+var loadConnector = (config) => {
+  try {
+    if ("connectorHandler" in config) {
+      return MongoDBConnector.fromConnectionHandler(config.connectorHandler);
+    }
+  } catch (error$1) {
+    throw new error.MastraError(
+      {
+        id: "STORAGE_MONGODB_STORE_CONSTRUCTOR_FAILED",
+        domain: error.ErrorDomain.STORAGE,
+        category: error.ErrorCategory.USER,
+        details: { connectionHandler: true }
+      },
+      error$1
+    );
+  }
+  try {
+    return MongoDBConnector.fromDatabaseConfig({
+      options: config.options,
+      url: config.url,
+      dbName: config.dbName
+    });
+  } catch (error$1) {
+    throw new error.MastraError(
+      {
+        id: "STORAGE_MONGODB_STORE_CONSTRUCTOR_FAILED",
+        domain: error.ErrorDomain.STORAGE,
+        category: error.ErrorCategory.USER,
+        details: { url: config?.url, dbName: config?.dbName }
+      },
+      error$1
+    );
+  }
+};
 var MongoDBStore = class extends storage.MastraStorage {
   #connector;
   stores;
@@ -2062,45 +2102,14 @@ var MongoDBStore = class extends storage.MastraStorage {
       selectByIncludeResourceScope: true,
       resourceWorkingMemory: true,
       hasColumn: false,
-      createTable: false
+      createTable: false,
+      deleteMessages: false
     };
   }
   constructor(config) {
     super({ name: "MongoDBStore" });
     this.stores = {};
-    try {
-      if ("connectorHandler" in config) {
-        this.#connector = MongoDBConnector.fromConnectionHandler(config.connectorHandler);
-        return;
-      }
-    } catch (error$1) {
-      throw new error.MastraError(
-        {
-          id: "STORAGE_MONGODB_STORE_CONSTRUCTOR_FAILED",
-          domain: error.ErrorDomain.STORAGE,
-          category: error.ErrorCategory.USER,
-          details: { connectionHandler: true }
-        },
-        error$1
-      );
-    }
-    try {
-      this.#connector = MongoDBConnector.fromDatabaseConfig({
-        options: config.options,
-        url: config.url,
-        dbName: config.dbName
-      });
-    } catch (error$1) {
-      throw new error.MastraError(
-        {
-          id: "STORAGE_MONGODB_STORE_CONSTRUCTOR_FAILED",
-          domain: error.ErrorDomain.STORAGE,
-          category: error.ErrorCategory.USER,
-          details: { url: config?.url, dbName: config?.dbName }
-        },
-        error$1
-      );
-    }
+    this.#connector = loadConnector(config);
     const operations = new StoreOperationsMongoDB({
       connector: this.#connector
     });

package/dist/index.d.cts

@@ -3,3 +3,4 @@ export { MongoDBUpsertVectorParams } from './_tsup-dts-rollup.cjs';
 export { MongoDBIndexReadyParams } from './_tsup-dts-rollup.cjs';
 export { MongoDBVector } from './_tsup-dts-rollup.cjs';
 export { MongoDBStore } from './_tsup-dts-rollup.cjs';
+export { ConnectorHandler } from './_tsup-dts-rollup.cjs';

package/dist/index.d.ts

@@ -3,3 +3,4 @@ export { MongoDBUpsertVectorParams } from './_tsup-dts-rollup.js';
 export { MongoDBIndexReadyParams } from './_tsup-dts-rollup.js';
 export { MongoDBVector } from './_tsup-dts-rollup.js';
 export { MongoDBStore } from './_tsup-dts-rollup.js';
+export { ConnectorHandler } from './_tsup-dts-rollup.js';

package/dist/index.js

@@ -341,7 +341,9 @@ var MongoDBVector = class extends MastraVector {
       };
       if (Object.keys(combinedFilter).length > 0) {
         const candidateIds = await collection.aggregate([{ $match: combinedFilter }, { $project: { _id: 1 } }]).map((doc) => doc._id).toArray();
-        vectorSearch.filter = { _id: { $in: candidateIds } };
+        if (candidateIds.length > 0) {
+          vectorSearch.filter = { _id: { $in: candidateIds } };
+        }
       }
       const pipeline = [
         {
@@ -1209,7 +1211,7 @@ var MemoryStorageMongoDB = class extends MemoryStorage {
   async getThreadsByResourceId({ resourceId }) {
     try {
       const collection = await this.operations.getCollection(TABLE_THREADS);
-      const results = await collection.find({ resourceId }).toArray();
+      const results = await collection.find({ resourceId }).sort({ updatedAt: -1 }).toArray();
       if (!results.length) {
         return [];
       }
@@ -1409,18 +1411,21 @@ var StoreOperationsMongoDB = class extends StoreOperations {
       );
     }
   }
+  processJsonbFields(tableName, record) {
+    const schema = TABLE_SCHEMAS[tableName];
+    return Object.fromEntries(
+      Object.entries(schema).map(([key, value]) => {
+        if (value.type === "jsonb" && record[key] && typeof record[key] === "string") {
+          return [key, safelyParseJSON(record[key])];
+        }
+        return [key, record[key]];
+      })
+    );
+  }
   async insert({ tableName, record }) {
     try {
       const collection = await this.getCollection(tableName);
-      const schema = TABLE_SCHEMAS[tableName];
-      const recordToInsert = Object.fromEntries(
-        Object.entries(schema).map(([key, value]) => {
-          if (value.type === "jsonb" && record[key] && typeof record[key] === "string") {
-            return [key, safelyParseJSON(record[key])];
-          }
-          return [key, record[key]];
-        })
-      );
+      const recordToInsert = this.processJsonbFields(tableName, record);
       await collection.insertOne(recordToInsert);
     } catch (error) {
       if (error instanceof Error) {
@@ -1444,7 +1449,8 @@ var StoreOperationsMongoDB = class extends StoreOperations {
     }
     try {
       const collection = await this.getCollection(tableName);
-      await collection.insertMany(records);
+      const processedRecords = records.map((record) => this.processJsonbFields(tableName, record));
+      await collection.insertMany(processedRecords);
     } catch (error) {
       throw new MastraError(
         {
@@ -1596,19 +1602,19 @@ var ScoresStorageMongoDB = class extends ScoresStorage {
         scorerId: score.scorerId,
         traceId: score.traceId || "",
         runId: score.runId,
-        scorer: typeof score.scorer === "string" ? score.scorer : JSON.stringify(score.scorer),
-        extractStepResult: typeof score.extractStepResult === "string" ? score.extractStepResult : JSON.stringify(score.extractStepResult),
-        analyzeStepResult: typeof score.analyzeStepResult === "string" ? score.analyzeStepResult : JSON.stringify(score.analyzeStepResult),
+        scorer: typeof score.scorer === "string" ? safelyParseJSON(score.scorer) : score.scorer,
+        extractStepResult: typeof score.extractStepResult === "string" ? safelyParseJSON(score.extractStepResult) : score.extractStepResult,
+        analyzeStepResult: typeof score.analyzeStepResult === "string" ? safelyParseJSON(score.analyzeStepResult) : score.analyzeStepResult,
         score: score.score,
         reason: score.reason,
         extractPrompt: score.extractPrompt,
         analyzePrompt: score.analyzePrompt,
         reasonPrompt: score.reasonPrompt,
-        input: typeof score.input === "string" ? score.input : JSON.stringify(score.input),
-        output: typeof score.output === "string" ? score.output : JSON.stringify(score.output),
+        input: typeof score.input === "string" ? safelyParseJSON(score.input) : score.input,
+        output: typeof score.output === "string" ? safelyParseJSON(score.output) : score.output,
         additionalContext: score.additionalContext,
-        runtimeContext: typeof score.runtimeContext === "string" ? score.runtimeContext : JSON.stringify(score.runtimeContext),
-        entity: typeof score.entity === "string" ? score.entity : JSON.stringify(score.entity),
+        runtimeContext: typeof score.runtimeContext === "string" ? safelyParseJSON(score.runtimeContext) : score.runtimeContext,
+        entity: typeof score.entity === "string" ? safelyParseJSON(score.entity) : score.entity,
         source: score.source,
         resourceId: score.resourceId || "",
         threadId: score.threadId || "",
@@ -2052,6 +2058,40 @@ var WorkflowsStorageMongoDB = class extends WorkflowsStorage {
 };
 
 // src/storage/index.ts
+var loadConnector = (config) => {
+  try {
+    if ("connectorHandler" in config) {
+      return MongoDBConnector.fromConnectionHandler(config.connectorHandler);
+    }
+  } catch (error) {
+    throw new MastraError(
+      {
+        id: "STORAGE_MONGODB_STORE_CONSTRUCTOR_FAILED",
+        domain: ErrorDomain.STORAGE,
+        category: ErrorCategory.USER,
+        details: { connectionHandler: true }
+      },
+      error
+    );
+  }
+  try {
+    return MongoDBConnector.fromDatabaseConfig({
+      options: config.options,
+      url: config.url,
+      dbName: config.dbName
+    });
+  } catch (error) {
+    throw new MastraError(
+      {
+        id: "STORAGE_MONGODB_STORE_CONSTRUCTOR_FAILED",
+        domain: ErrorDomain.STORAGE,
+        category: ErrorCategory.USER,
+        details: { url: config?.url, dbName: config?.dbName }
+      },
+      error
+    );
+  }
+};
 var MongoDBStore = class extends MastraStorage {
   #connector;
   stores;
@@ -2060,45 +2100,14 @@ var MongoDBStore = class extends MastraStorage {
       selectByIncludeResourceScope: true,
       resourceWorkingMemory: true,
       hasColumn: false,
-      createTable: false
+      createTable: false,
+      deleteMessages: false
     };
   }
   constructor(config) {
     super({ name: "MongoDBStore" });
     this.stores = {};
-    try {
-      if ("connectorHandler" in config) {
-        this.#connector = MongoDBConnector.fromConnectionHandler(config.connectorHandler);
-        return;
-      }
-    } catch (error) {
-      throw new MastraError(
-        {
-          id: "STORAGE_MONGODB_STORE_CONSTRUCTOR_FAILED",
-          domain: ErrorDomain.STORAGE,
-          category: ErrorCategory.USER,
-          details: { connectionHandler: true }
-        },
-        error
-      );
-    }
-    try {
-      this.#connector = MongoDBConnector.fromDatabaseConfig({
-        options: config.options,
-        url: config.url,
-        dbName: config.dbName
-      });
-    } catch (error) {
-      throw new MastraError(
-        {
-          id: "STORAGE_MONGODB_STORE_CONSTRUCTOR_FAILED",
-          domain: ErrorDomain.STORAGE,
-          category: ErrorCategory.USER,
-          details: { url: config?.url, dbName: config?.dbName }
-        },
-        error
-      );
-    }
+    this.#connector = loadConnector(config);
     const operations = new StoreOperationsMongoDB({
       connector: this.#connector
     });

package/docker-compose.yaml

@@ -1,6 +1,6 @@
 services:
   mongodb-storage:
-    image: mongo:8.0.11
+    image: mongo:8.0.12
     container_name: 'mongodb-storage-test-db'
     ports:
       - '27017:27017'

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@mastra/mongodb",
-  "version": "0.12.2",
+  "version": "0.12.3-alpha.1",
   "description": "MongoDB provider for Mastra - includes vector store capabilities",
   "type": "module",
   "main": "dist/index.js",
@@ -20,7 +20,7 @@
   },
   "dependencies": {
     "@types/mongodb": "^4.0.7",
-    "cloudflare": "^4.4.1",
+    "cloudflare": "^4.5.0",
     "mongodb": "^6.17.0",
     "uuid": "^11.1.0"
   },
@@ -31,9 +31,9 @@
     "tsup": "^8.5.0",
     "typescript": "^5.8.3",
     "vitest": "^3.2.4",
-    "@internal/lint": "0.0.22",
-    "@internal/storage-test-utils": "0.0.18",
-    "@mastra/core": "0.11.1"
+    "@internal/lint": "0.0.23",
+    "@internal/storage-test-utils": "0.0.19",
+    "@mastra/core": "0.12.0-alpha.2"
   },
   "peerDependencies": {
     "@mastra/core": ">=0.10.7-0 <0.12.0-0"

package/src/index.ts

@@ -1,3 +1,4 @@
 export * from './vector';
 export * from './storage';
+export * from './storage/connectors/base';
 export { MONGODB_PROMPT } from './vector/prompt';

package/src/storage/MongoDBConnector.ts

@@ -1,6 +1,6 @@
 import { MongoClient } from 'mongodb';
 import type { Db } from 'mongodb';
-import type { ConnectorHandler } from './ConnectorHandler';
+import type { ConnectorHandler } from './connectors/base';
 import type { DatabaseConfig } from './types';
 
 type MongoDBConnectorOptions =

package/src/storage/domains/memory/index.ts

@@ -568,7 +568,7 @@ export class MemoryStorageMongoDB extends MemoryStorage {
   async getThreadsByResourceId({ resourceId }: { resourceId: string }): Promise<StorageThreadType[]> {
     try {
       const collection = await this.operations.getCollection(TABLE_THREADS);
-      const results = await collection.find<any>({ resourceId }).toArray();
+      const results = await collection.find<any>({ resourceId }).sort({ updatedAt: -1 }).toArray();
       if (!results.length) {
         return [];
       }

package/src/storage/domains/operations/index.ts

@@ -78,21 +78,23 @@ export class StoreOperationsMongoDB extends StoreOperations {
     }
   }
 
+  private processJsonbFields(tableName: TABLE_NAMES, record: Record<string, any>): Record<string, any> {
+    const schema = TABLE_SCHEMAS[tableName];
+
+    return Object.fromEntries(
+      Object.entries(schema).map(([key, value]) => {
+        if (value.type === 'jsonb' && record[key] && typeof record[key] === 'string') {
+          return [key, safelyParseJSON(record[key])];
+        }
+        return [key, record[key]];
+      }),
+    );
+  }
+
   async insert({ tableName, record }: { tableName: TABLE_NAMES; record: Record<string, any> }): Promise<void> {
     try {
       const collection = await this.getCollection(tableName);
-
-      const schema = TABLE_SCHEMAS[tableName];
-
-      const recordToInsert = Object.fromEntries(
-        Object.entries(schema).map(([key, value]) => {
-          if (value.type === 'jsonb' && record[key] && typeof record[key] === 'string') {
-            return [key, safelyParseJSON(record[key])];
-          }
-          return [key, record[key]];
-        }),
-      );
-
+      const recordToInsert = this.processJsonbFields(tableName, record);
       await collection.insertOne(recordToInsert);
     } catch (error) {
       if (error instanceof Error) {
@@ -118,7 +120,8 @@ export class StoreOperationsMongoDB extends StoreOperations {
 
     try {
       const collection = await this.getCollection(tableName);
-      await collection.insertMany(records);
+      const processedRecords = records.map(record => this.processJsonbFields(tableName, record));
+      await collection.insertMany(processedRecords);
     } catch (error) {
       throw new MastraError(
         {

package/src/storage/domains/scores/index.ts

@@ -142,26 +142,26 @@ export class ScoresStorageMongoDB extends ScoresStorage {
         scorerId: score.scorerId,
         traceId: score.traceId || '',
         runId: score.runId,
-        scorer: typeof score.scorer === 'string' ? score.scorer : JSON.stringify(score.scorer),
+        scorer: typeof score.scorer === 'string' ? safelyParseJSON(score.scorer) : score.scorer,
         extractStepResult:
           typeof score.extractStepResult === 'string'
-            ? score.extractStepResult
-            : JSON.stringify(score.extractStepResult),
+            ? safelyParseJSON(score.extractStepResult)
+            : score.extractStepResult,
         analyzeStepResult:
           typeof score.analyzeStepResult === 'string'
-            ? score.analyzeStepResult
-            : JSON.stringify(score.analyzeStepResult),
+            ? safelyParseJSON(score.analyzeStepResult)
+            : score.analyzeStepResult,
         score: score.score,
         reason: score.reason,
         extractPrompt: score.extractPrompt,
         analyzePrompt: score.analyzePrompt,
         reasonPrompt: score.reasonPrompt,
-        input: typeof score.input === 'string' ? score.input : JSON.stringify(score.input),
-        output: typeof score.output === 'string' ? score.output : JSON.stringify(score.output),
+        input: typeof score.input === 'string' ? safelyParseJSON(score.input) : score.input,
+        output: typeof score.output === 'string' ? safelyParseJSON(score.output) : score.output,
         additionalContext: score.additionalContext,
         runtimeContext:
-          typeof score.runtimeContext === 'string' ? score.runtimeContext : JSON.stringify(score.runtimeContext),
-        entity: typeof score.entity === 'string' ? score.entity : JSON.stringify(score.entity),
+          typeof score.runtimeContext === 'string' ? safelyParseJSON(score.runtimeContext) : score.runtimeContext,
+        entity: typeof score.entity === 'string' ? safelyParseJSON(score.entity) : score.entity,
         source: score.source,
         resourceId: score.resourceId || '',
         threadId: score.threadId || '',

package/src/storage/index.test.ts

@@ -51,6 +51,11 @@ describe('Validation', () => {
     it('does not throw on valid config', () => {
       expect(() => new MongoDBStore(validWithConnectionHandlerConfig)).not.toThrow();
     });
+
+    it('should initialize the stores correctly', () => {
+      const store = new MongoDBStore(validWithConnectionHandlerConfig);
+      expect(Object.keys(store.stores)).not.toHaveLength(0);
+    });
   });
 });
 

package/src/storage/index.ts

@@ -1,5 +1,5 @@
 import type { MastraMessageContentV2 } from '@mastra/core/agent';
-import { ErrorDomain, ErrorCategory, MastraError } from '@mastra/core/error';
+import { ErrorCategory, ErrorDomain, MastraError } from '@mastra/core/error';
 import type { MastraMessageV1, MastraMessageV2, StorageThreadType } from '@mastra/core/memory';
 import type { ScoreRowData } from '@mastra/core/scores';
 import type {
@@ -7,15 +7,15 @@ import type {
   PaginationArgs,
   PaginationInfo,
   StorageColumn,
+  StorageDomains,
   StorageGetMessagesArg,
   StorageGetTracesArg,
+  StorageGetTracesPaginatedArg,
+  StoragePagination,
+  StorageResourceType,
   TABLE_NAMES,
   WorkflowRun,
   WorkflowRuns,
-  StorageResourceType,
-  StorageDomains,
-  StoragePagination,
-  StorageGetTracesPaginatedArg,
 } from '@mastra/core/storage';
 import { MastraStorage } from '@mastra/core/storage';
 import type { Trace } from '@mastra/core/telemetry';
@@ -29,6 +29,42 @@ import { TracesStorageMongoDB } from './domains/traces';
 import { WorkflowsStorageMongoDB } from './domains/workflows';
 import type { MongoDBConfig } from './types';
 
+const loadConnector = (config: MongoDBConfig): MongoDBConnector => {
+  try {
+    if ('connectorHandler' in config) {
+      return MongoDBConnector.fromConnectionHandler(config.connectorHandler);
+    }
+  } catch (error) {
+    throw new MastraError(
+      {
+        id: 'STORAGE_MONGODB_STORE_CONSTRUCTOR_FAILED',
+        domain: ErrorDomain.STORAGE,
+        category: ErrorCategory.USER,
+        details: { connectionHandler: true },
+      },
+      error,
+    );
+  }
+
+  try {
+    return MongoDBConnector.fromDatabaseConfig({
+      options: config.options,
+      url: config.url,
+      dbName: config.dbName,
+    });
+  } catch (error) {
+    throw new MastraError(
+      {
+        id: 'STORAGE_MONGODB_STORE_CONSTRUCTOR_FAILED',
+        domain: ErrorDomain.STORAGE,
+        category: ErrorCategory.USER,
+        details: { url: config?.url, dbName: config?.dbName },
+      },
+      error,
+    );
+  }
+};
+
 export class MongoDBStore extends MastraStorage {
   #connector: MongoDBConnector;
 
@@ -39,12 +75,14 @@ export class MongoDBStore extends MastraStorage {
     resourceWorkingMemory: boolean;
     hasColumn: boolean;
     createTable: boolean;
+    deleteMessages: boolean;
   } {
     return {
       selectByIncludeResourceScope: true,
       resourceWorkingMemory: true,
       hasColumn: false,
       createTable: false,
+      deleteMessages: false,
     };
   }
 
@@ -53,40 +91,7 @@ export class MongoDBStore extends MastraStorage {
 
     this.stores = {} as StorageDomains;
 
-    try {
-      if ('connectorHandler' in config) {
-        this.#connector = MongoDBConnector.fromConnectionHandler(config.connectorHandler);
-        return;
-      }
-    } catch (error) {
-      throw new MastraError(
-        {
-          id: 'STORAGE_MONGODB_STORE_CONSTRUCTOR_FAILED',
-          domain: ErrorDomain.STORAGE,
-          category: ErrorCategory.USER,
-          details: { connectionHandler: true },
-        },
-        error,
-      );
-    }
-
-    try {
-      this.#connector = MongoDBConnector.fromDatabaseConfig({
-        options: config.options,
-        url: config.url,
-        dbName: config.dbName,
-      });
-    } catch (error) {
-      throw new MastraError(
-        {
-          id: 'STORAGE_MONGODB_STORE_CONSTRUCTOR_FAILED',
-          domain: ErrorDomain.STORAGE,
-          category: ErrorCategory.USER,
-          details: { url: config?.url, dbName: config?.dbName },
-        },
-        error,
-      );
-    }
+    this.#connector = loadConnector(config);
 
     const operations = new StoreOperationsMongoDB({
       connector: this.#connector,

package/src/vector/index.ts

@@ -327,7 +327,9 @@ export class MongoDBVector extends MastraVector<MongoDBVectorFilter> {
           .map(doc => doc._id)
           .toArray();
 
-        vectorSearch.filter = { _id: { $in: candidateIds } };
+        if (candidateIds.length > 0) {
+          vectorSearch.filter = { _id: { $in: candidateIds } };
+        }
       }
 
       // Build the aggregation pipeline

package/src/storage/ConnectorHandler.ts

@@ -1,7 +0,0 @@
-import type { Collection } from 'mongodb';
-
-export interface ConnectorHandler {
-  getCollection(collectionName: string): Promise<Collection>;
-
-  close(): Promise<void>;
-}