@mastra/mongodb 0.10.2-alpha.0 → 0.10.2-alpha.1

package/.turbo/turbo-build.log

@@ -1,23 +1,23 @@
 
-> @mastra/mongodb@0.10.2-alpha.0 build /home/runner/work/mastra/mastra/stores/mongodb
+> @mastra/mongodb@0.10.2-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.4.0
 TSC Build start
-TSC ⚡️ Build success in 9494ms
+TSC ⚡️ Build success in 8403ms
 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 12316ms
+DTS ⚡️ Build success in 12384ms
 CLI Cleaning output folder
 ESM Build start
 CJS Build start
-CJS dist/index.cjs 32.94 KB
-CJS ⚡️ Build success in 973ms
-ESM dist/index.js 32.83 KB
-ESM ⚡️ Build success in 973ms
+ESM dist/index.js 33.01 KB
+ESM ⚡️ Build success in 985ms
+CJS dist/index.cjs 33.13 KB
+CJS ⚡️ Build success in 986ms

package/CHANGELOG.md

@@ -1,5 +1,14 @@
 # @mastra/mongodb
 
+## 0.10.2-alpha.1
+
+### Patch Changes
+
+- c5bf1ce: Add backwards compat code for new MessageList in storage
+- Updated dependencies [c5bf1ce]
+- Updated dependencies [12b7002]
+  - @mastra/core@0.10.2-alpha.4
+
 ## 0.10.2-alpha.0
 
 ### Patch Changes

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

@@ -5,9 +5,10 @@ import type { DeleteVectorParams } from '@mastra/core/vector';
 import type { DescribeIndexParams } from '@mastra/core/vector';
 import type { EvalRow } from '@mastra/core/storage';
 import type { IndexStats } from '@mastra/core/vector';
+import type { MastraMessageV1 } from '@mastra/core/memory';
+import type { MastraMessageV2 } from '@mastra/core/memory';
 import { MastraStorage } from '@mastra/core/storage';
 import { MastraVector } from '@mastra/core/vector';
-import type { MessageType } from '@mastra/core/memory';
 import type { MongoClientOptions } from 'mongodb';
 import type { OperatorSupport } from '@mastra/core/vector/filter';
 import type { QueryResult } from '@mastra/core/vector';
@@ -103,9 +104,14 @@ declare class MongoDBStore extends MastraStorage {
         threadId: string;
     }): Promise<void>;
     getMessages<T = unknown>({ threadId, selectBy }: StorageGetMessagesArg): Promise<T[]>;
-    saveMessages({ messages }: {
-        messages: MessageType[];
-    }): Promise<MessageType[]>;
+    saveMessages(args: {
+        messages: MastraMessageV1[];
+        format?: undefined | 'v1';
+    }): Promise<MastraMessageV1[]>;
+    saveMessages(args: {
+        messages: MastraMessageV2[];
+        format: 'v2';
+    }): Promise<MastraMessageV2[]>;
     getTraces({ name, scope, page, perPage, attributes, filters, }?: {
         name?: string;
         scope?: string;

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

@@ -5,9 +5,10 @@ import type { DeleteVectorParams } from '@mastra/core/vector';
 import type { DescribeIndexParams } from '@mastra/core/vector';
 import type { EvalRow } from '@mastra/core/storage';
 import type { IndexStats } from '@mastra/core/vector';
+import type { MastraMessageV1 } from '@mastra/core/memory';
+import type { MastraMessageV2 } from '@mastra/core/memory';
 import { MastraStorage } from '@mastra/core/storage';
 import { MastraVector } from '@mastra/core/vector';
-import type { MessageType } from '@mastra/core/memory';
 import type { MongoClientOptions } from 'mongodb';
 import type { OperatorSupport } from '@mastra/core/vector/filter';
 import type { QueryResult } from '@mastra/core/vector';
@@ -103,9 +104,14 @@ declare class MongoDBStore extends MastraStorage {
         threadId: string;
     }): Promise<void>;
     getMessages<T = unknown>({ threadId, selectBy }: StorageGetMessagesArg): Promise<T[]>;
-    saveMessages({ messages }: {
-        messages: MessageType[];
-    }): Promise<MessageType[]>;
+    saveMessages(args: {
+        messages: MastraMessageV1[];
+        format?: undefined | 'v1';
+    }): Promise<MastraMessageV1[]>;
+    saveMessages(args: {
+        messages: MastraMessageV2[];
+        format: 'v2';
+    }): Promise<MastraMessageV2[]>;
     getTraces({ name, scope, page, perPage, attributes, filters, }?: {
         name?: string;
         scope?: string;

package/dist/index.cjs

@@ -4,6 +4,7 @@ var vector = require('@mastra/core/vector');
 var mongodb = require('mongodb');
 var uuid = require('uuid');
 var filter = require('@mastra/core/vector/filter');
+var agent = require('@mastra/core/agent');
 var storage = require('@mastra/core/storage');
 
 // src/vector/index.ts
@@ -628,7 +629,10 @@ var MongoDBStore = class extends storage.MastraStorage {
       throw error;
     }
   }
-  async saveMessages({ messages }) {
+  async saveMessages({
+    messages,
+    format
+  }) {
     if (!messages.length) {
       return messages;
     }
@@ -652,7 +656,9 @@ var MongoDBStore = class extends storage.MastraStorage {
       });
       const collection = await this.getCollection(storage.TABLE_MESSAGES);
       await collection.insertMany(messagesToInsert);
-      return messages;
+      const list = new agent.MessageList().add(messages, "memory");
+      if (format === `v2`) return list.get.all.v2();
+      return list.get.all.v1();
     } catch (error) {
       this.logger.error("Failed to save messages in database: " + error?.message);
       throw error;

package/dist/index.js

@@ -2,6 +2,7 @@ import { MastraVector } from '@mastra/core/vector';
 import { MongoClient } from 'mongodb';
 import { v4 } from 'uuid';
 import { BaseFilterTranslator } from '@mastra/core/vector/filter';
+import { MessageList } from '@mastra/core/agent';
 import { MastraStorage, TABLE_THREADS, TABLE_MESSAGES, TABLE_TRACES, TABLE_WORKFLOW_SNAPSHOT, TABLE_EVALS } from '@mastra/core/storage';
 
 // src/vector/index.ts
@@ -626,7 +627,10 @@ var MongoDBStore = class extends MastraStorage {
       throw error;
     }
   }
-  async saveMessages({ messages }) {
+  async saveMessages({
+    messages,
+    format
+  }) {
     if (!messages.length) {
       return messages;
     }
@@ -650,7 +654,9 @@ var MongoDBStore = class extends MastraStorage {
       });
       const collection = await this.getCollection(TABLE_MESSAGES);
       await collection.insertMany(messagesToInsert);
-      return messages;
+      const list = new MessageList().add(messages, "memory");
+      if (format === `v2`) return list.get.all.v2();
+      return list.get.all.v1();
     } catch (error) {
       this.logger.error("Failed to save messages in database: " + error?.message);
       throw error;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@mastra/mongodb",
-  "version": "0.10.2-alpha.0",
+  "version": "0.10.2-alpha.1",
   "description": "MongoDB provider for Mastra - includes vector store capabilities",
   "type": "module",
   "main": "dist/index.js",
@@ -32,7 +32,7 @@
     "typescript": "^5.8.2",
     "vitest": "^3.1.2",
     "@internal/lint": "0.0.7",
-    "@mastra/core": "0.10.2-alpha.2"
+    "@mastra/core": "0.10.2-alpha.4"
   },
   "peerDependencies": {
     "@mastra/core": "^0.10.0-alpha.0"

package/src/storage/index.test.ts

@@ -1,5 +1,5 @@
 import { randomUUID } from 'crypto';
-import type { MessageType, MetricResult, WorkflowRunState } from '@mastra/core';
+import type { MastraMessageV1, MetricResult, WorkflowRunState } from '@mastra/core';
 import { TABLE_EVALS, TABLE_MESSAGES, TABLE_THREADS, TABLE_WORKFLOW_SNAPSHOT } from '@mastra/core/storage';
 import { afterAll, beforeAll, describe, expect, it } from 'vitest';
 import type { MongoDBConfig } from './index';
@@ -40,7 +40,7 @@ class Test {
     };
   }
 
-  generateSampleMessage(threadId: string): MessageType {
+  generateSampleMessage(threadId: string): MastraMessageV1 {
     return {
       id: `msg-${randomUUID()}`,
       role: 'user',
@@ -88,6 +88,7 @@ class Test {
         },
         input: {},
       },
+      serializedStepGraph: [],
       activePaths: [],
       suspendedPaths: {},
       runId,
@@ -244,7 +245,10 @@ describe('MongoDBStore', () => {
       const thread = test.generateSampleThread();
       await store.saveThread({ thread });
 
-      const messages = [test.generateSampleMessage(thread.id), test.generateSampleMessage(thread.id)];
+      const messages = [
+        test.generateSampleMessage(thread.id),
+        { ...test.generateSampleMessage(thread.id), role: 'assistant' as const },
+      ];
 
       // Save messages
       const savedMessages = await store.saveMessages({ messages });
@@ -274,15 +278,15 @@ describe('MongoDBStore', () => {
       const messages = [
         {
           ...test.generateSampleMessage(thread.id),
-          content: [{ type: 'text', text: 'First' }] as MessageType['content'],
+          content: [{ type: 'text', text: 'First' }] as MastraMessageV1['content'],
         },
         {
           ...test.generateSampleMessage(thread.id),
-          content: [{ type: 'text', text: 'Second' }] as MessageType['content'],
+          content: [{ type: 'text', text: 'Second' }] as MastraMessageV1['content'],
         },
         {
           ...test.generateSampleMessage(thread.id),
-          content: [{ type: 'text', text: 'Third' }] as MessageType['content'],
+          content: [{ type: 'text', text: 'Third' }] as MastraMessageV1['content'],
         },
       ];
 
@@ -479,6 +483,7 @@ describe('MongoDBStore', () => {
             status: 'waiting',
           },
         ],
+        serializedStepGraph: [],
         runId: runId,
         timestamp: Date.now(),
       };

package/src/storage/index.ts

@@ -1,5 +1,6 @@
+import { MessageList } from '@mastra/core/agent';
 import type { MetricResult, TestInfo } from '@mastra/core/eval';
-import type { MessageType, StorageThreadType } from '@mastra/core/memory';
+import type { MastraMessageV1, MastraMessageV2, StorageThreadType } from '@mastra/core/memory';
 import type { EvalRow, StorageGetMessagesArg, TABLE_NAMES, WorkflowRun } from '@mastra/core/storage';
 import {
   MastraStorage,
@@ -235,8 +236,8 @@ export class MongoDBStore extends MastraStorage {
     try {
       const limit = typeof selectBy?.last === 'number' ? selectBy.last : 40;
       const include = selectBy?.include || [];
-      let messages: MessageType[] = [];
-      let allMessages: MessageType[] = [];
+      let messages: MastraMessageV2[] = [];
+      let allMessages: MastraMessageV2[] = [];
       const collection = await this.getCollection(TABLE_MESSAGES);
       // Get all messages from the thread ordered by creation date descending
       allMessages = (await collection.find({ thread_id: threadId }).sort({ createdAt: -1 }).toArray()).map((row: any) =>
@@ -270,7 +271,7 @@ export class MongoDBStore extends MastraStorage {
         messages.push(
           ...Array.from(selectedIndexes)
             .map(i => allMessages[i])
-            .filter((m): m is MessageType => !!m),
+            .filter((m): m is MastraMessageV2 => !!m),
         );
       }
 
@@ -293,7 +294,14 @@ export class MongoDBStore extends MastraStorage {
     }
   }
 
-  async saveMessages({ messages }: { messages: MessageType[] }): Promise<MessageType[]> {
+  async saveMessages(args: { messages: MastraMessageV1[]; format?: undefined | 'v1' }): Promise<MastraMessageV1[]>;
+  async saveMessages(args: { messages: MastraMessageV2[]; format: 'v2' }): Promise<MastraMessageV2[]>;
+  async saveMessages({
+    messages,
+    format,
+  }:
+    | { messages: MastraMessageV1[]; format?: undefined | 'v1' }
+    | { messages: MastraMessageV2[]; format: 'v2' }): Promise<MastraMessageV2[] | MastraMessageV1[]> {
     if (!messages.length) {
       return messages;
     }
@@ -321,7 +329,9 @@ export class MongoDBStore extends MastraStorage {
       // Execute all inserts in a single batch
       const collection = await this.getCollection(TABLE_MESSAGES);
       await collection.insertMany(messagesToInsert);
-      return messages;
+      const list = new MessageList().add(messages, 'memory');
+      if (format === `v2`) return list.get.all.v2();
+      return list.get.all.v1();
     } catch (error) {
       this.logger.error('Failed to save messages in database: ' + (error as { message: string })?.message);
       throw error;
@@ -628,7 +638,7 @@ export class MongoDBStore extends MastraStorage {
     };
   }
 
-  private parseRow(row: any): MessageType {
+  private parseRow(row: any): MastraMessageV2 {
     let content = row.content;
     try {
       content = JSON.parse(row.content);
@@ -642,7 +652,7 @@ export class MongoDBStore extends MastraStorage {
       type: row.type,
       createdAt: new Date(row.createdAt as string),
       threadId: row.thread_id,
-    } as MessageType;
+    } as MastraMessageV2;
   }
 
   private transformEvalRow(row: Record<string, any>): EvalRow {