@mastra/mongodb 0.10.0 → 0.10.1-alpha.0

package/.turbo/turbo-build.log

@@ -1,23 +1,23 @@
 
-> @mastra/mongodb@0.10.0-alpha.1 build /home/runner/work/mastra/mastra/stores/mongodb
+> @mastra/mongodb@0.10.1-alpha.0 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 5643ms
+TSC ⚡️ Build success in 11126ms
 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 8219ms
+DTS ⚡️ Build success in 13746ms
 CLI Cleaning output folder
 ESM Build start
 CJS Build start
-CJS dist/index.cjs 18.11 KB
-CJS ⚡️ Build success in 706ms
-ESM dist/index.js 18.05 KB
-ESM ⚡️ Build success in 707ms
+ESM dist/index.js 32.81 KB
+ESM ⚡️ Build success in 1273ms
+CJS dist/index.cjs 32.93 KB
+CJS ⚡️ Build success in 1273ms

package/CHANGELOG.md

@@ -1,5 +1,14 @@
 # @mastra/mongodb
 
+## 0.10.1-alpha.0
+
+### Patch Changes
+
+- fcc915f: Support MongoDB database as store
+- Updated dependencies [6d16390]
+- Updated dependencies [1e4a421]
+  - @mastra/core@0.10.1-alpha.0
+
 ## 0.10.0
 
 ### Minor Changes

package/README.md

@@ -56,6 +56,40 @@ const results = await vectorDB.query({
 await vectorDB.disconnect();
 ```
 
+### Storage
+
+```typescript
+import { MongoDBStore } from '@mastra/mongodb';
+
+const store = new MongoDBStore({
+  uri: 'mongodb://mongodb:mongodb@localhost:27018/?authSource=admin&directConnection=true',
+  dbName: 'mastra',
+});
+
+// Create a thread
+await store.saveThread({
+  id: 'thread-123',
+  resourceId: 'resource-456',
+  title: 'My Thread',
+  metadata: { key: 'value' },
+});
+
+// Add messages to thread
+await store.saveMessages([
+  {
+    id: 'msg-789',
+    threadId: 'thread-123',
+    role: 'user',
+    type: 'text',
+    content: [{ type: 'text', text: 'Hello' }],
+  },
+]);
+
+// Query threads and messages
+const savedThread = await store.getThread('thread-123');
+const messages = await store.getMessages('thread-123');
+```
+
 ## Configuration
 
 The MongoDB vector store is initialized with:
@@ -83,6 +117,13 @@ const vectorDB = new MongoDBVector({
 - Collection (index) management: create, list, describe, delete
 - Atlas Search readiness checks for reliable testing
 
+### Storage Features
+
+- Thread and message storage with JSON support
+- Efficient batch operations
+- Rich metadata support
+- Timestamp tracking
+
 ## Supported Filter Operators
 
 - Comparison: `$eq`, `$ne`, `$gt`, `$gte`, `$lt`, `$lte`
@@ -121,6 +162,15 @@ The following distance metrics are supported:
 - `deleteIndex(indexName)`: Delete a collection
 - `disconnect()`: Close the MongoDB connection
 
+## Storage Methods
+
+- `saveThread(thread)`: Create or update a thread
+- `getThread(threadId)`: Get a thread by ID
+- `deleteThread(threadId)`: Delete a thread and its messages
+- `saveMessages(messages)`: Save multiple messages in a transaction
+- `getMessages(threadId)`: Get all messages for a thread
+- `deleteMessages(messageIds)`: Delete specific messages
+
 ## Query Response Format
 
 Each query result includes:

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

@@ -3,15 +3,23 @@ import type { CreateIndexParams } from '@mastra/core/vector';
 import type { DeleteIndexParams } from '@mastra/core/vector';
 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 { 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';
 import type { QueryVectorParams } from '@mastra/core/vector';
+import type { StorageGetMessagesArg } from '@mastra/core/storage';
+import type { StorageThreadType } from '@mastra/core/memory';
+import type { TABLE_NAMES } from '@mastra/core/storage';
 import type { UpdateVectorParams } from '@mastra/core/vector';
 import type { UpsertVectorParams } from '@mastra/core/vector';
 import type { VectorFilter } from '@mastra/core/vector/filter';
+import type { WorkflowRun } from '@mastra/core/storage';
+import type { WorkflowRunState } from '@mastra/core/workflows';
 
 /**
  * Vector store specific prompt that details supported operators and examples.
@@ -21,6 +29,13 @@ declare const MONGODB_PROMPT = "When querying MongoDB Vector, you can ONLY use t
 export { MONGODB_PROMPT }
 export { MONGODB_PROMPT as MONGODB_PROMPT_alias_1 }
 
+declare interface MongoDBConfig {
+    url: string;
+    dbName: string;
+}
+export { MongoDBConfig }
+export { MongoDBConfig as MongoDBConfig_alias_1 }
+
 /**
  * Translator for MongoDB filter queries.
  * Maintains MongoDB-compatible syntax while ensuring proper validation
@@ -48,6 +63,94 @@ declare interface MongoDBQueryVectorParams extends QueryVectorParams {
 export { MongoDBQueryVectorParams }
 export { MongoDBQueryVectorParams as MongoDBQueryVectorParams_alias_1 }
 
+declare class MongoDBStore extends MastraStorage {
+    #private;
+    constructor(config: MongoDBConfig);
+    private getConnection;
+    private getCollection;
+    createTable(): Promise<void>;
+    clearTable({ tableName }: {
+        tableName: TABLE_NAMES;
+    }): Promise<void>;
+    insert({ tableName, record }: {
+        tableName: TABLE_NAMES;
+        record: Record<string, any>;
+    }): Promise<void>;
+    batchInsert({ tableName, records }: {
+        tableName: TABLE_NAMES;
+        records: Record<string, any>[];
+    }): Promise<void>;
+    load<R>({ tableName, keys }: {
+        tableName: TABLE_NAMES;
+        keys: Record<string, string>;
+    }): Promise<R | null>;
+    getThreadById({ threadId }: {
+        threadId: string;
+    }): Promise<StorageThreadType | null>;
+    getThreadsByResourceId({ resourceId }: {
+        resourceId: string;
+    }): Promise<StorageThreadType[]>;
+    saveThread({ thread }: {
+        thread: StorageThreadType;
+    }): Promise<StorageThreadType>;
+    updateThread({ id, title, metadata, }: {
+        id: string;
+        title: string;
+        metadata: Record<string, unknown>;
+    }): Promise<StorageThreadType>;
+    deleteThread({ threadId }: {
+        threadId: string;
+    }): Promise<void>;
+    getMessages<T = unknown>({ threadId, selectBy }: StorageGetMessagesArg): Promise<T[]>;
+    saveMessages({ messages }: {
+        messages: MessageType[];
+    }): Promise<MessageType[]>;
+    getTraces({ name, scope, page, perPage, attributes, filters, }?: {
+        name?: string;
+        scope?: string;
+        page: number;
+        perPage: number;
+        attributes?: Record<string, string>;
+        filters?: Record<string, any>;
+    }): Promise<any[]>;
+    getWorkflowRuns({ workflowName, fromDate, toDate, limit, offset, }?: {
+        workflowName?: string;
+        fromDate?: Date;
+        toDate?: Date;
+        limit?: number;
+        offset?: number;
+    }): Promise<{
+        runs: Array<{
+            workflowName: string;
+            runId: string;
+            snapshot: WorkflowRunState | string;
+            createdAt: Date;
+            updatedAt: Date;
+        }>;
+        total: number;
+    }>;
+    getEvalsByAgentName(agentName: string, type?: 'test' | 'live'): Promise<EvalRow[]>;
+    persistWorkflowSnapshot({ workflowName, runId, snapshot, }: {
+        workflowName: string;
+        runId: string;
+        snapshot: WorkflowRunState;
+    }): Promise<void>;
+    loadWorkflowSnapshot({ workflowName, runId, }: {
+        workflowName: string;
+        runId: string;
+    }): Promise<WorkflowRunState | null>;
+    getWorkflowRunById({ runId, workflowName, }: {
+        runId: string;
+        workflowName?: string;
+    }): Promise<WorkflowRun | null>;
+    private parseWorkflowRun;
+    private parseRow;
+    private transformEvalRow;
+    close(): Promise<void>;
+}
+export { MongoDBStore }
+export { MongoDBStore as MongoDBStore_alias_1 }
+
 declare interface MongoDBUpsertVectorParams extends UpsertVectorParams {
     documents?: string[];
 }

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

@@ -3,15 +3,23 @@ import type { CreateIndexParams } from '@mastra/core/vector';
 import type { DeleteIndexParams } from '@mastra/core/vector';
 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 { 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';
 import type { QueryVectorParams } from '@mastra/core/vector';
+import type { StorageGetMessagesArg } from '@mastra/core/storage';
+import type { StorageThreadType } from '@mastra/core/memory';
+import type { TABLE_NAMES } from '@mastra/core/storage';
 import type { UpdateVectorParams } from '@mastra/core/vector';
 import type { UpsertVectorParams } from '@mastra/core/vector';
 import type { VectorFilter } from '@mastra/core/vector/filter';
+import type { WorkflowRun } from '@mastra/core/storage';
+import type { WorkflowRunState } from '@mastra/core/workflows';
 
 /**
  * Vector store specific prompt that details supported operators and examples.
@@ -21,6 +29,13 @@ declare const MONGODB_PROMPT = "When querying MongoDB Vector, you can ONLY use t
 export { MONGODB_PROMPT }
 export { MONGODB_PROMPT as MONGODB_PROMPT_alias_1 }
 
+declare interface MongoDBConfig {
+    url: string;
+    dbName: string;
+}
+export { MongoDBConfig }
+export { MongoDBConfig as MongoDBConfig_alias_1 }
+
 /**
  * Translator for MongoDB filter queries.
  * Maintains MongoDB-compatible syntax while ensuring proper validation
@@ -48,6 +63,94 @@ declare interface MongoDBQueryVectorParams extends QueryVectorParams {
 export { MongoDBQueryVectorParams }
 export { MongoDBQueryVectorParams as MongoDBQueryVectorParams_alias_1 }
 
+declare class MongoDBStore extends MastraStorage {
+    #private;
+    constructor(config: MongoDBConfig);
+    private getConnection;
+    private getCollection;
+    createTable(): Promise<void>;
+    clearTable({ tableName }: {
+        tableName: TABLE_NAMES;
+    }): Promise<void>;
+    insert({ tableName, record }: {
+        tableName: TABLE_NAMES;
+        record: Record<string, any>;
+    }): Promise<void>;
+    batchInsert({ tableName, records }: {
+        tableName: TABLE_NAMES;
+        records: Record<string, any>[];
+    }): Promise<void>;
+    load<R>({ tableName, keys }: {
+        tableName: TABLE_NAMES;
+        keys: Record<string, string>;
+    }): Promise<R | null>;
+    getThreadById({ threadId }: {
+        threadId: string;
+    }): Promise<StorageThreadType | null>;
+    getThreadsByResourceId({ resourceId }: {
+        resourceId: string;
+    }): Promise<StorageThreadType[]>;
+    saveThread({ thread }: {
+        thread: StorageThreadType;
+    }): Promise<StorageThreadType>;
+    updateThread({ id, title, metadata, }: {
+        id: string;
+        title: string;
+        metadata: Record<string, unknown>;
+    }): Promise<StorageThreadType>;
+    deleteThread({ threadId }: {
+        threadId: string;
+    }): Promise<void>;
+    getMessages<T = unknown>({ threadId, selectBy }: StorageGetMessagesArg): Promise<T[]>;
+    saveMessages({ messages }: {
+        messages: MessageType[];
+    }): Promise<MessageType[]>;
+    getTraces({ name, scope, page, perPage, attributes, filters, }?: {
+        name?: string;
+        scope?: string;
+        page: number;
+        perPage: number;
+        attributes?: Record<string, string>;
+        filters?: Record<string, any>;
+    }): Promise<any[]>;
+    getWorkflowRuns({ workflowName, fromDate, toDate, limit, offset, }?: {
+        workflowName?: string;
+        fromDate?: Date;
+        toDate?: Date;
+        limit?: number;
+        offset?: number;
+    }): Promise<{
+        runs: Array<{
+            workflowName: string;
+            runId: string;
+            snapshot: WorkflowRunState | string;
+            createdAt: Date;
+            updatedAt: Date;
+        }>;
+        total: number;
+    }>;
+    getEvalsByAgentName(agentName: string, type?: 'test' | 'live'): Promise<EvalRow[]>;
+    persistWorkflowSnapshot({ workflowName, runId, snapshot, }: {
+        workflowName: string;
+        runId: string;
+        snapshot: WorkflowRunState;
+    }): Promise<void>;
+    loadWorkflowSnapshot({ workflowName, runId, }: {
+        workflowName: string;
+        runId: string;
+    }): Promise<WorkflowRunState | null>;
+    getWorkflowRunById({ runId, workflowName, }: {
+        runId: string;
+        workflowName?: string;
+    }): Promise<WorkflowRun | null>;
+    private parseWorkflowRun;
+    private parseRow;
+    private transformEvalRow;
+    close(): Promise<void>;
+}
+export { MongoDBStore }
+export { MongoDBStore as MongoDBStore_alias_1 }
+
 declare interface MongoDBUpsertVectorParams extends UpsertVectorParams {
     documents?: string[];
 }