@mastra/mongodb 0.0.5 → 0.1.0-alpha.1

package/.turbo/turbo-build.log

@@ -1,23 +1,23 @@
 
-> @mastra/mongodb@0.0.5-alpha.4 build /home/runner/work/mastra/mastra/stores/mongodb
+> @mastra/mongodb@0.1.0-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 7651ms
+TSC ⚡️ Build success in 7075ms
 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 9483ms
+DTS ⚡️ Build success in 8116ms
 CLI Cleaning output folder
 ESM Build start
 CJS Build start
-ESM dist/index.js 19.08 KB
-ESM ⚡️ Build success in 675ms
-CJS dist/index.cjs 19.13 KB
-CJS ⚡️ Build success in 681ms
+ESM dist/index.js 18.05 KB
+ESM ⚡️ Build success in 634ms
+CJS dist/index.cjs 18.11 KB
+CJS ⚡️ Build success in 629ms

package/CHANGELOG.md

@@ -1,5 +1,40 @@
 # @mastra/mongodb
 
+## 0.1.0-alpha.1
+
+### Minor Changes
+
+- 83da932: Move @mastra/core to peerdeps
+
+### Patch Changes
+
+- a7292b0: BREAKING(@mastra/core, all vector stores): Vector store breaking changes (remove deprecated functions and positional arguments)
+- Updated dependencies [b3a3d63]
+- Updated dependencies [344f453]
+- Updated dependencies [0a3ae6d]
+- Updated dependencies [95911be]
+- Updated dependencies [5eb5a99]
+- Updated dependencies [7e632c5]
+- Updated dependencies [1e9fbfa]
+- Updated dependencies [b2ae5aa]
+- Updated dependencies [a7292b0]
+- Updated dependencies [0dcb9f0]
+  - @mastra/core@0.10.0-alpha.1
+
+## 0.0.6-alpha.0
+
+### Patch Changes
+
+- d0ee3c6: Change all public functions and constructors in vector stores to use named args and prepare to phase out positional args
+- Updated dependencies [f53a6ac]
+- Updated dependencies [eabdcd9]
+- Updated dependencies [90be034]
+- Updated dependencies [99f050a]
+- Updated dependencies [d0ee3c6]
+- Updated dependencies [23f258c]
+- Updated dependencies [2672a05]
+  - @mastra/core@0.9.5-alpha.0
+
 ## 0.0.5
 
 ### Patch Changes

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

@@ -1,14 +1,15 @@
 import { BaseFilterTranslator } from '@mastra/core/vector/filter';
 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 { IndexStats } from '@mastra/core/vector';
 import { MastraVector } from '@mastra/core/vector';
 import type { MongoClientOptions } from 'mongodb';
 import type { OperatorSupport } from '@mastra/core/vector/filter';
-import type { ParamsToArgs } from '@mastra/core/vector';
 import type { QueryResult } from '@mastra/core/vector';
-import type { QueryVectorArgs } from '@mastra/core/vector';
 import type { QueryVectorParams } from '@mastra/core/vector';
-import type { UpsertVectorArgs } from '@mastra/core/vector';
+import type { UpdateVectorParams } from '@mastra/core/vector';
 import type { UpsertVectorParams } from '@mastra/core/vector';
 import type { VectorFilter } from '@mastra/core/vector/filter';
 
@@ -33,9 +34,13 @@ export declare class MongoDBFilterTranslator extends BaseFilterTranslator {
     isEmpty(filter: any): boolean;
 }
 
-declare type MongoDBQueryArgs = [...QueryVectorArgs, string?];
-export { MongoDBQueryArgs }
-export { MongoDBQueryArgs as MongoDBQueryArgs_alias_1 }
+declare interface MongoDBIndexReadyParams {
+    indexName: string;
+    timeoutMs?: number;
+    checkIntervalMs?: number;
+}
+export { MongoDBIndexReadyParams }
+export { MongoDBIndexReadyParams as MongoDBIndexReadyParams_alias_1 }
 
 declare interface MongoDBQueryVectorParams extends QueryVectorParams {
     documentFilter?: VectorFilter;
@@ -43,14 +48,6 @@ declare interface MongoDBQueryVectorParams extends QueryVectorParams {
 export { MongoDBQueryVectorParams }
 export { MongoDBQueryVectorParams as MongoDBQueryVectorParams_alias_1 }
 
-declare type MongoDBUpsertArgs = [...UpsertVectorArgs, string[]?];
-export { MongoDBUpsertArgs }
-export { MongoDBUpsertArgs as MongoDBUpsertArgs_alias_1 }
-
-declare type MongoDBUpsertParams = ParamsToArgs<MongoDBUpsertArgs>;
-export { MongoDBUpsertParams }
-export { MongoDBUpsertParams as MongoDBUpsertParams_alias_1 }
-
 declare interface MongoDBUpsertVectorParams extends UpsertVectorParams {
     documents?: string[];
 }
@@ -73,29 +70,27 @@ declare class MongoDBVector extends MastraVector {
     });
     connect(): Promise<void>;
     disconnect(): Promise<void>;
-    createIndex(params: CreateIndexParams): Promise<void>;
-    waitForIndexReady(indexName: string, timeoutMs?: number, checkIntervalMs?: number): Promise<void>;
-    upsert(params: MongoDBUpsertVectorParams): Promise<string[]>;
-    query(params: MongoDBQueryVectorParams): Promise<QueryResult[]>;
+    createIndex({ indexName, dimension, metric }: CreateIndexParams): Promise<void>;
+    /**
+     * Waits for the index to be ready.
+     *
+     * @param {string} indexName - The name of the index to wait for
+     * @param {number} timeoutMs - The maximum time in milliseconds to wait for the index to be ready (default: 60000)
+     * @param {number} checkIntervalMs - The interval in milliseconds at which to check if the index is ready (default: 2000)
+     * @returns A promise that resolves when the index is ready
+     */
+    waitForIndexReady({ indexName, timeoutMs, checkIntervalMs, }: MongoDBIndexReadyParams): Promise<void>;
+    upsert({ indexName, vectors, metadata, ids, documents }: MongoDBUpsertVectorParams): Promise<string[]>;
+    query({ indexName, queryVector, topK, filter, includeVector, documentFilter, }: MongoDBQueryVectorParams): Promise<QueryResult[]>;
     listIndexes(): Promise<string[]>;
-    describeIndex(indexName: string): Promise<IndexStats>;
-    deleteIndex(indexName: string): Promise<void>;
     /**
-     * @deprecated Use {@link updateVector} instead. This method will be removed on May 20th, 2025.
+     * Retrieves statistics about a vector index.
      *
-     * Updates a vector by its ID with the provided vector and/or metadata.
-     * @param indexName - The name of the index containing the vector.
-     * @param id - The ID of the vector to update.
-     * @param update - An object containing the vector and/or metadata to update.
-     * @param update.vector - An optional array of numbers representing the new vector.
-     * @param update.metadata - An optional record containing the new metadata.
-     * @returns A promise that resolves when the update is complete.
-     * @throws Will throw an error if no updates are provided or if the update operation fails.
+     * @param {string} indexName - The name of the index to describe
+     * @returns A promise that resolves to the index statistics including dimension, count and metric
      */
-    updateIndexById(indexName: string, id: string, update: {
-        vector?: number[];
-        metadata?: Record<string, any>;
-    }): Promise<void>;
+    describeIndex({ indexName }: DescribeIndexParams): Promise<IndexStats>;
+    deleteIndex({ indexName }: DeleteIndexParams): Promise<void>;
     /**
      * Updates a vector by its ID with the provided vector and/or metadata.
      * @param indexName - The name of the index containing the vector.
@@ -106,20 +101,7 @@ declare class MongoDBVector extends MastraVector {
      * @returns A promise that resolves when the update is complete.
      * @throws Will throw an error if no updates are provided or if the update operation fails.
      */
-    updateVector(indexName: string, id: string, update: {
-        vector?: number[];
-        metadata?: Record<string, any>;
-    }): Promise<void>;
-    /**
-     * @deprecated Use {@link deleteVector} instead. This method will be removed on May 20th, 2025.
-     *
-     * Deletes a vector by its ID.
-     * @param indexName - The name of the index containing the vector.
-     * @param id - The ID of the vector to delete.
-     * @returns A promise that resolves when the deletion is complete.
-     * @throws Will throw an error if the deletion operation fails.
-     */
-    deleteIndexById(indexName: string, id: string): Promise<void>;
+    updateVector({ indexName, id, update }: UpdateVectorParams): Promise<void>;
     /**
      * Deletes a vector by its ID.
      * @param indexName - The name of the index containing the vector.
@@ -127,7 +109,7 @@ declare class MongoDBVector extends MastraVector {
      * @returns A promise that resolves when the deletion is complete.
      * @throws Will throw an error if the deletion operation fails.
      */
-    deleteVector(indexName: string, id: string): Promise<void>;
+    deleteVector({ indexName, id }: DeleteVectorParams): Promise<void>;
     private getCollection;
     private validateVectorDimensions;
     private setIndexDimension;

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

@@ -1,14 +1,15 @@
 import { BaseFilterTranslator } from '@mastra/core/vector/filter';
 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 { IndexStats } from '@mastra/core/vector';
 import { MastraVector } from '@mastra/core/vector';
 import type { MongoClientOptions } from 'mongodb';
 import type { OperatorSupport } from '@mastra/core/vector/filter';
-import type { ParamsToArgs } from '@mastra/core/vector';
 import type { QueryResult } from '@mastra/core/vector';
-import type { QueryVectorArgs } from '@mastra/core/vector';
 import type { QueryVectorParams } from '@mastra/core/vector';
-import type { UpsertVectorArgs } from '@mastra/core/vector';
+import type { UpdateVectorParams } from '@mastra/core/vector';
 import type { UpsertVectorParams } from '@mastra/core/vector';
 import type { VectorFilter } from '@mastra/core/vector/filter';
 
@@ -33,9 +34,13 @@ export declare class MongoDBFilterTranslator extends BaseFilterTranslator {
     isEmpty(filter: any): boolean;
 }
 
-declare type MongoDBQueryArgs = [...QueryVectorArgs, string?];
-export { MongoDBQueryArgs }
-export { MongoDBQueryArgs as MongoDBQueryArgs_alias_1 }
+declare interface MongoDBIndexReadyParams {
+    indexName: string;
+    timeoutMs?: number;
+    checkIntervalMs?: number;
+}
+export { MongoDBIndexReadyParams }
+export { MongoDBIndexReadyParams as MongoDBIndexReadyParams_alias_1 }
 
 declare interface MongoDBQueryVectorParams extends QueryVectorParams {
     documentFilter?: VectorFilter;
@@ -43,14 +48,6 @@ declare interface MongoDBQueryVectorParams extends QueryVectorParams {
 export { MongoDBQueryVectorParams }
 export { MongoDBQueryVectorParams as MongoDBQueryVectorParams_alias_1 }
 
-declare type MongoDBUpsertArgs = [...UpsertVectorArgs, string[]?];
-export { MongoDBUpsertArgs }
-export { MongoDBUpsertArgs as MongoDBUpsertArgs_alias_1 }
-
-declare type MongoDBUpsertParams = ParamsToArgs<MongoDBUpsertArgs>;
-export { MongoDBUpsertParams }
-export { MongoDBUpsertParams as MongoDBUpsertParams_alias_1 }
-
 declare interface MongoDBUpsertVectorParams extends UpsertVectorParams {
     documents?: string[];
 }
@@ -73,29 +70,27 @@ declare class MongoDBVector extends MastraVector {
     });
     connect(): Promise<void>;
     disconnect(): Promise<void>;
-    createIndex(params: CreateIndexParams): Promise<void>;
-    waitForIndexReady(indexName: string, timeoutMs?: number, checkIntervalMs?: number): Promise<void>;
-    upsert(params: MongoDBUpsertVectorParams): Promise<string[]>;
-    query(params: MongoDBQueryVectorParams): Promise<QueryResult[]>;
+    createIndex({ indexName, dimension, metric }: CreateIndexParams): Promise<void>;
+    /**
+     * Waits for the index to be ready.
+     *
+     * @param {string} indexName - The name of the index to wait for
+     * @param {number} timeoutMs - The maximum time in milliseconds to wait for the index to be ready (default: 60000)
+     * @param {number} checkIntervalMs - The interval in milliseconds at which to check if the index is ready (default: 2000)
+     * @returns A promise that resolves when the index is ready
+     */
+    waitForIndexReady({ indexName, timeoutMs, checkIntervalMs, }: MongoDBIndexReadyParams): Promise<void>;
+    upsert({ indexName, vectors, metadata, ids, documents }: MongoDBUpsertVectorParams): Promise<string[]>;
+    query({ indexName, queryVector, topK, filter, includeVector, documentFilter, }: MongoDBQueryVectorParams): Promise<QueryResult[]>;
     listIndexes(): Promise<string[]>;
-    describeIndex(indexName: string): Promise<IndexStats>;
-    deleteIndex(indexName: string): Promise<void>;
     /**
-     * @deprecated Use {@link updateVector} instead. This method will be removed on May 20th, 2025.
+     * Retrieves statistics about a vector index.
      *
-     * Updates a vector by its ID with the provided vector and/or metadata.
-     * @param indexName - The name of the index containing the vector.
-     * @param id - The ID of the vector to update.
-     * @param update - An object containing the vector and/or metadata to update.
-     * @param update.vector - An optional array of numbers representing the new vector.
-     * @param update.metadata - An optional record containing the new metadata.
-     * @returns A promise that resolves when the update is complete.
-     * @throws Will throw an error if no updates are provided or if the update operation fails.
+     * @param {string} indexName - The name of the index to describe
+     * @returns A promise that resolves to the index statistics including dimension, count and metric
      */
-    updateIndexById(indexName: string, id: string, update: {
-        vector?: number[];
-        metadata?: Record<string, any>;
-    }): Promise<void>;
+    describeIndex({ indexName }: DescribeIndexParams): Promise<IndexStats>;
+    deleteIndex({ indexName }: DeleteIndexParams): Promise<void>;
     /**
      * Updates a vector by its ID with the provided vector and/or metadata.
      * @param indexName - The name of the index containing the vector.
@@ -106,20 +101,7 @@ declare class MongoDBVector extends MastraVector {
      * @returns A promise that resolves when the update is complete.
      * @throws Will throw an error if no updates are provided or if the update operation fails.
      */
-    updateVector(indexName: string, id: string, update: {
-        vector?: number[];
-        metadata?: Record<string, any>;
-    }): Promise<void>;
-    /**
-     * @deprecated Use {@link deleteVector} instead. This method will be removed on May 20th, 2025.
-     *
-     * Deletes a vector by its ID.
-     * @param indexName - The name of the index containing the vector.
-     * @param id - The ID of the vector to delete.
-     * @returns A promise that resolves when the deletion is complete.
-     * @throws Will throw an error if the deletion operation fails.
-     */
-    deleteIndexById(indexName: string, id: string): Promise<void>;
+    updateVector({ indexName, id, update }: UpdateVectorParams): Promise<void>;
     /**
      * Deletes a vector by its ID.
      * @param indexName - The name of the index containing the vector.
@@ -127,7 +109,7 @@ declare class MongoDBVector extends MastraVector {
      * @returns A promise that resolves when the deletion is complete.
      * @throws Will throw an error if the deletion operation fails.
      */
-    deleteVector(indexName: string, id: string): Promise<void>;
+    deleteVector({ indexName, id }: DeleteVectorParams): Promise<void>;
     private getCollection;
     private validateVectorDimensions;
     private setIndexDimension;

package/dist/index.cjs

@@ -121,8 +121,7 @@ var MongoDBVector = class extends vector.MastraVector {
   async disconnect() {
     await this.client.close();
   }
-  async createIndex(params) {
-    const { indexName, dimension, metric = "cosine" } = params;
+  async createIndex({ indexName, dimension, metric = "cosine" }) {
     if (!Number.isInteger(dimension) || dimension <= 0) {
       throw new Error("Dimension must be a positive integer");
     }
@@ -160,7 +159,19 @@ var MongoDBVector = class extends vector.MastraVector {
     }
     await collection.updateOne({ _id: "__index_metadata__" }, { $set: { dimension, metric } }, { upsert: true });
   }
-  async waitForIndexReady(indexName, timeoutMs = 6e4, checkIntervalMs = 2e3) {
+  /**
+   * Waits for the index to be ready.
+   *
+   * @param {string} indexName - The name of the index to wait for
+   * @param {number} timeoutMs - The maximum time in milliseconds to wait for the index to be ready (default: 60000)
+   * @param {number} checkIntervalMs - The interval in milliseconds at which to check if the index is ready (default: 2000)
+   * @returns A promise that resolves when the index is ready
+   */
+  async waitForIndexReady({
+    indexName,
+    timeoutMs = 6e4,
+    checkIntervalMs = 2e3
+  }) {
     const collection = await this.getCollection(indexName, true);
     const indexNameInternal = `${indexName}_vector_index`;
     const startTime = Date.now();
@@ -175,11 +186,10 @@ var MongoDBVector = class extends vector.MastraVector {
     }
     throw new Error(`Index "${indexNameInternal}" did not become ready within timeout`);
   }
-  async upsert(params) {
-    const { indexName, vectors, metadata, ids, documents } = params;
+  async upsert({ indexName, vectors, metadata, ids, documents }) {
     const collection = await this.getCollection(indexName);
     this.collectionForValidation = collection;
-    const stats = await this.describeIndex(indexName);
+    const stats = await this.describeIndex({ indexName });
     await this.validateVectorDimensions(vectors, stats.dimension);
     const generatedIds = ids || vectors.map(() => uuid.v4());
     const operations = vectors.map((vector, idx) => {
@@ -212,8 +222,14 @@ var MongoDBVector = class extends vector.MastraVector {
     await collection.bulkWrite(operations);
     return generatedIds;
   }
-  async query(params) {
-    const { indexName, queryVector, topK = 10, filter, includeVector = false, documentFilter } = params;
+  async query({
+    indexName,
+    queryVector,
+    topK = 10,
+    filter,
+    includeVector = false,
+    documentFilter
+  }) {
     const collection = await this.getCollection(indexName, true);
     const indexNameInternal = `${indexName}_vector_index`;
     const mongoFilter = this.transformFilter(filter);
@@ -269,7 +285,13 @@ var MongoDBVector = class extends vector.MastraVector {
     const collections = await this.db.listCollections().toArray();
     return collections.map((col) => col.name);
   }
-  async describeIndex(indexName) {
+  /**
+   * Retrieves statistics about a vector index.
+   *
+   * @param {string} indexName - The name of the index to describe
+   * @returns A promise that resolves to the index statistics including dimension, count and metric
+   */
+  async describeIndex({ indexName }) {
     const collection = await this.getCollection(indexName, true);
     const count = await collection.countDocuments({ _id: { $ne: "__index_metadata__" } });
     const metadataDoc = await collection.findOne({ _id: "__index_metadata__" });
@@ -281,7 +303,7 @@ var MongoDBVector = class extends vector.MastraVector {
       metric
     };
   }
-  async deleteIndex(indexName) {
+  async deleteIndex({ indexName }) {
     const collection = await this.getCollection(indexName, false);
     if (collection) {
       await collection.drop();
@@ -291,8 +313,6 @@ var MongoDBVector = class extends vector.MastraVector {
     }
   }
   /**
-   * @deprecated Use {@link updateVector} instead. This method will be removed on May 20th, 2025.
-   *
    * Updates a vector by its ID with the provided vector and/or metadata.
    * @param indexName - The name of the index containing the vector.
    * @param id - The ID of the vector to update.
@@ -302,25 +322,7 @@ var MongoDBVector = class extends vector.MastraVector {
    * @returns A promise that resolves when the update is complete.
    * @throws Will throw an error if no updates are provided or if the update operation fails.
    */
-  async updateIndexById(indexName, id, update) {
-    this.logger.warn(
-      `Deprecation Warning: updateIndexById() is deprecated. 
-      Please use updateVector() instead. 
-      updateIndexById() will be removed on May 20th, 2025.`
-    );
-    await this.updateVector(indexName, id, update);
-  }
-  /**
-   * Updates a vector by its ID with the provided vector and/or metadata.
-   * @param indexName - The name of the index containing the vector.
-   * @param id - The ID of the vector to update.
-   * @param update - An object containing the vector and/or metadata to update.
-   * @param update.vector - An optional array of numbers representing the new vector.
-   * @param update.metadata - An optional record containing the new metadata.
-   * @returns A promise that resolves when the update is complete.
-   * @throws Will throw an error if no updates are provided or if the update operation fails.
-   */
-  async updateVector(indexName, id, update) {
+  async updateVector({ indexName, id, update }) {
     try {
       if (!update.vector && !update.metadata) {
         throw new Error("No updates provided");
@@ -328,7 +330,7 @@ var MongoDBVector = class extends vector.MastraVector {
       const collection = await this.getCollection(indexName, true);
       const updateDoc = {};
       if (update.vector) {
-        const stats = await this.describeIndex(indexName);
+        const stats = await this.describeIndex({ indexName });
         await this.validateVectorDimensions([update.vector], stats.dimension);
         updateDoc[this.embeddingFieldName] = update.vector;
       }
@@ -347,23 +349,6 @@ var MongoDBVector = class extends vector.MastraVector {
       throw new Error(`Failed to update vector by id: ${id} for index name: ${indexName}: ${error.message}`);
     }
   }
-  /**
-   * @deprecated Use {@link deleteVector} instead. This method will be removed on May 20th, 2025.
-   *
-   * Deletes a vector by its ID.
-   * @param indexName - The name of the index containing the vector.
-   * @param id - The ID of the vector to delete.
-   * @returns A promise that resolves when the deletion is complete.
-   * @throws Will throw an error if the deletion operation fails.
-   */
-  async deleteIndexById(indexName, id) {
-    this.logger.warn(
-      `Deprecation Warning: deleteIndexById() is deprecated. 
-      Please use deleteVector() instead. 
-      deleteIndexById() will be removed on May 20th, 2025.`
-    );
-    await this.deleteVector(indexName, id);
-  }
   /**
    * Deletes a vector by its ID.
    * @param indexName - The name of the index containing the vector.
@@ -371,7 +356,7 @@ var MongoDBVector = class extends vector.MastraVector {
    * @returns A promise that resolves when the deletion is complete.
    * @throws Will throw an error if the deletion operation fails.
    */
-  async deleteVector(indexName, id) {
+  async deleteVector({ indexName, id }) {
     try {
       const collection = await this.getCollection(indexName, true);
       await collection.deleteOne({ _id: id });

package/dist/index.d.cts

@@ -1,7 +1,5 @@
 export { MONGODB_PROMPT } from './_tsup-dts-rollup.cjs';
-export { MongoDBUpsertArgs } from './_tsup-dts-rollup.cjs';
-export { MongoDBQueryArgs } from './_tsup-dts-rollup.cjs';
-export { MongoDBUpsertParams } from './_tsup-dts-rollup.cjs';
 export { MongoDBUpsertVectorParams } from './_tsup-dts-rollup.cjs';
 export { MongoDBQueryVectorParams } from './_tsup-dts-rollup.cjs';
+export { MongoDBIndexReadyParams } from './_tsup-dts-rollup.cjs';
 export { MongoDBVector } from './_tsup-dts-rollup.cjs';

package/dist/index.d.ts

@@ -1,7 +1,5 @@
 export { MONGODB_PROMPT } from './_tsup-dts-rollup.js';
-export { MongoDBUpsertArgs } from './_tsup-dts-rollup.js';
-export { MongoDBQueryArgs } from './_tsup-dts-rollup.js';
-export { MongoDBUpsertParams } from './_tsup-dts-rollup.js';
 export { MongoDBUpsertVectorParams } from './_tsup-dts-rollup.js';
 export { MongoDBQueryVectorParams } from './_tsup-dts-rollup.js';
+export { MongoDBIndexReadyParams } from './_tsup-dts-rollup.js';
 export { MongoDBVector } from './_tsup-dts-rollup.js';

package/dist/index.js

@@ -119,8 +119,7 @@ var MongoDBVector = class extends MastraVector {
   async disconnect() {
     await this.client.close();
   }
-  async createIndex(params) {
-    const { indexName, dimension, metric = "cosine" } = params;
+  async createIndex({ indexName, dimension, metric = "cosine" }) {
     if (!Number.isInteger(dimension) || dimension <= 0) {
       throw new Error("Dimension must be a positive integer");
     }
@@ -158,7 +157,19 @@ var MongoDBVector = class extends MastraVector {
     }
     await collection.updateOne({ _id: "__index_metadata__" }, { $set: { dimension, metric } }, { upsert: true });
   }
-  async waitForIndexReady(indexName, timeoutMs = 6e4, checkIntervalMs = 2e3) {
+  /**
+   * Waits for the index to be ready.
+   *
+   * @param {string} indexName - The name of the index to wait for
+   * @param {number} timeoutMs - The maximum time in milliseconds to wait for the index to be ready (default: 60000)
+   * @param {number} checkIntervalMs - The interval in milliseconds at which to check if the index is ready (default: 2000)
+   * @returns A promise that resolves when the index is ready
+   */
+  async waitForIndexReady({
+    indexName,
+    timeoutMs = 6e4,
+    checkIntervalMs = 2e3
+  }) {
     const collection = await this.getCollection(indexName, true);
     const indexNameInternal = `${indexName}_vector_index`;
     const startTime = Date.now();
@@ -173,11 +184,10 @@ var MongoDBVector = class extends MastraVector {
     }
     throw new Error(`Index "${indexNameInternal}" did not become ready within timeout`);
   }
-  async upsert(params) {
-    const { indexName, vectors, metadata, ids, documents } = params;
+  async upsert({ indexName, vectors, metadata, ids, documents }) {
     const collection = await this.getCollection(indexName);
     this.collectionForValidation = collection;
-    const stats = await this.describeIndex(indexName);
+    const stats = await this.describeIndex({ indexName });
     await this.validateVectorDimensions(vectors, stats.dimension);
     const generatedIds = ids || vectors.map(() => v4());
     const operations = vectors.map((vector, idx) => {
@@ -210,8 +220,14 @@ var MongoDBVector = class extends MastraVector {
     await collection.bulkWrite(operations);
     return generatedIds;
   }
-  async query(params) {
-    const { indexName, queryVector, topK = 10, filter, includeVector = false, documentFilter } = params;
+  async query({
+    indexName,
+    queryVector,
+    topK = 10,
+    filter,
+    includeVector = false,
+    documentFilter
+  }) {
     const collection = await this.getCollection(indexName, true);
     const indexNameInternal = `${indexName}_vector_index`;
     const mongoFilter = this.transformFilter(filter);
@@ -267,7 +283,13 @@ var MongoDBVector = class extends MastraVector {
     const collections = await this.db.listCollections().toArray();
     return collections.map((col) => col.name);
   }
-  async describeIndex(indexName) {
+  /**
+   * Retrieves statistics about a vector index.
+   *
+   * @param {string} indexName - The name of the index to describe
+   * @returns A promise that resolves to the index statistics including dimension, count and metric
+   */
+  async describeIndex({ indexName }) {
     const collection = await this.getCollection(indexName, true);
     const count = await collection.countDocuments({ _id: { $ne: "__index_metadata__" } });
     const metadataDoc = await collection.findOne({ _id: "__index_metadata__" });
@@ -279,7 +301,7 @@ var MongoDBVector = class extends MastraVector {
       metric
     };
   }
-  async deleteIndex(indexName) {
+  async deleteIndex({ indexName }) {
     const collection = await this.getCollection(indexName, false);
     if (collection) {
       await collection.drop();
@@ -289,8 +311,6 @@ var MongoDBVector = class extends MastraVector {
     }
   }
   /**
-   * @deprecated Use {@link updateVector} instead. This method will be removed on May 20th, 2025.
-   *
    * Updates a vector by its ID with the provided vector and/or metadata.
    * @param indexName - The name of the index containing the vector.
    * @param id - The ID of the vector to update.
@@ -300,25 +320,7 @@ var MongoDBVector = class extends MastraVector {
    * @returns A promise that resolves when the update is complete.
    * @throws Will throw an error if no updates are provided or if the update operation fails.
    */
-  async updateIndexById(indexName, id, update) {
-    this.logger.warn(
-      `Deprecation Warning: updateIndexById() is deprecated. 
-      Please use updateVector() instead. 
-      updateIndexById() will be removed on May 20th, 2025.`
-    );
-    await this.updateVector(indexName, id, update);
-  }
-  /**
-   * Updates a vector by its ID with the provided vector and/or metadata.
-   * @param indexName - The name of the index containing the vector.
-   * @param id - The ID of the vector to update.
-   * @param update - An object containing the vector and/or metadata to update.
-   * @param update.vector - An optional array of numbers representing the new vector.
-   * @param update.metadata - An optional record containing the new metadata.
-   * @returns A promise that resolves when the update is complete.
-   * @throws Will throw an error if no updates are provided or if the update operation fails.
-   */
-  async updateVector(indexName, id, update) {
+  async updateVector({ indexName, id, update }) {
     try {
       if (!update.vector && !update.metadata) {
         throw new Error("No updates provided");
@@ -326,7 +328,7 @@ var MongoDBVector = class extends MastraVector {
       const collection = await this.getCollection(indexName, true);
       const updateDoc = {};
       if (update.vector) {
-        const stats = await this.describeIndex(indexName);
+        const stats = await this.describeIndex({ indexName });
         await this.validateVectorDimensions([update.vector], stats.dimension);
         updateDoc[this.embeddingFieldName] = update.vector;
       }
@@ -345,23 +347,6 @@ var MongoDBVector = class extends MastraVector {
       throw new Error(`Failed to update vector by id: ${id} for index name: ${indexName}: ${error.message}`);
     }
   }
-  /**
-   * @deprecated Use {@link deleteVector} instead. This method will be removed on May 20th, 2025.
-   *
-   * Deletes a vector by its ID.
-   * @param indexName - The name of the index containing the vector.
-   * @param id - The ID of the vector to delete.
-   * @returns A promise that resolves when the deletion is complete.
-   * @throws Will throw an error if the deletion operation fails.
-   */
-  async deleteIndexById(indexName, id) {
-    this.logger.warn(
-      `Deprecation Warning: deleteIndexById() is deprecated. 
-      Please use deleteVector() instead. 
-      deleteIndexById() will be removed on May 20th, 2025.`
-    );
-    await this.deleteVector(indexName, id);
-  }
   /**
    * Deletes a vector by its ID.
    * @param indexName - The name of the index containing the vector.
@@ -369,7 +354,7 @@ var MongoDBVector = class extends MastraVector {
    * @returns A promise that resolves when the deletion is complete.
    * @throws Will throw an error if the deletion operation fails.
    */
-  async deleteVector(indexName, id) {
+  async deleteVector({ indexName, id }) {
     try {
       const collection = await this.getCollection(indexName, true);
       await collection.deleteOne({ _id: id });

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@mastra/mongodb",
-  "version": "0.0.5",
+  "version": "0.1.0-alpha.1",
   "description": "MongoDB provider for Mastra - includes vector store capabilities",
   "type": "module",
   "main": "dist/index.js",
@@ -22,8 +22,7 @@
     "@types/mongodb": "^4.0.7",
     "cloudflare": "^4.1.0",
     "mongodb": "^6.15.0",
-    "uuid": "^11.1.0",
-    "@mastra/core": "^0.9.4"
+    "uuid": "^11.1.0"
   },
   "devDependencies": {
     "@microsoft/api-extractor": "^7.52.5",
@@ -32,7 +31,11 @@
     "tsup": "^8.4.0",
     "typescript": "^5.8.2",
     "vitest": "^3.1.2",
-    "@internal/lint": "0.0.5"
+    "@internal/lint": "0.0.5",
+    "@mastra/core": "0.10.0-alpha.1"
+  },
+  "peerDependencies": {
+    "@mastra/core": "^0.9.4"
   },
   "scripts": {
     "build": "tsup src/index.ts --format esm,cjs --experimental-dts --clean --treeshake=smallest --splitting",

package/src/vector/index.test.ts

@@ -60,7 +60,7 @@ async function createIndexAndWait(
   metric: 'cosine' | 'euclidean' | 'dotproduct',
 ) {
   await vectorDB.createIndex({ indexName, dimension, metric });
-  await vectorDB.waitForIndexReady(indexName);
+  await vectorDB.waitForIndexReady({ indexName });
   const created = await waitForCondition(
     async () => {
       const cols = await vectorDB.listIndexes();
@@ -75,7 +75,7 @@ async function createIndexAndWait(
 // Delete index (collection) and wait until it is removed
 async function deleteIndexAndWait(vectorDB: MongoDBVector, indexName: string) {
   try {
-    await vectorDB.deleteIndex(indexName);
+    await vectorDB.deleteIndex({ indexName });
     const deleted = await waitForCondition(
       async () => {
         const cols = await vectorDB.listIndexes();
@@ -105,7 +105,7 @@ describe('MongoDBVector Integration Tests', () => {
     // Cleanup any existing collections
     try {
       const cols = await vectorDB.listIndexes();
-      await Promise.all(cols.map(c => vectorDB.deleteIndex(c)));
+      await Promise.all(cols.map(c => vectorDB.deleteIndex({ indexName: c })));
       const deleted = await waitForCondition(async () => (await vectorDB.listIndexes()).length === 0, 30000, 2000);
       if (!deleted) throw new Error('Timed out waiting for collections to be deleted');
     } catch (error) {
@@ -119,12 +119,12 @@ describe('MongoDBVector Integration Tests', () => {
 
   afterAll(async () => {
     try {
-      await vectorDB.deleteIndex(testIndexName);
+      await vectorDB.deleteIndex({ indexName: testIndexName });
     } catch (error) {
       console.error('Failed to delete test collection:', error);
     }
     try {
-      await vectorDB.deleteIndex(testIndexName2);
+      await vectorDB.deleteIndex({ indexName: testIndexName2 });
     } catch (error) {
       console.error('Failed to delete test collection:', error);
     }
@@ -137,7 +137,7 @@ describe('MongoDBVector Integration Tests', () => {
     expect(cols).toContain(testIndexName);
 
     // Check stats (should be zero docs initially)
-    const initialStats = await vectorDB.describeIndex(testIndexName);
+    const initialStats = await vectorDB.describeIndex({ indexName: testIndexName });
     expect(initialStats).toEqual({ dimension: 4, metric: 'cosine', count: 0 });
 
     // Upsert 4 vectors with metadata
@@ -153,7 +153,7 @@ describe('MongoDBVector Integration Tests', () => {
 
     // Wait for the document count to update (increased delay to 5000ms)
     await new Promise(resolve => setTimeout(resolve, 5000));
-    const updatedStats = await vectorDB.describeIndex(testIndexName);
+    const updatedStats = await vectorDB.describeIndex({ indexName: testIndexName });
     expect(updatedStats.count).toEqual(4);
 
     // Query for similar vectors (delay again to allow for index update)
@@ -175,7 +175,7 @@ describe('MongoDBVector Integration Tests', () => {
     expect(filteredResults[0]?.metadata).toEqual({ label: 'vector2' });
 
     // Final stats should show > 0 documents
-    const finalStats = await vectorDB.describeIndex(testIndexName);
+    const finalStats = await vectorDB.describeIndex({ indexName: testIndexName });
     expect(finalStats.count).toBeGreaterThan(0);
   });
 
@@ -379,7 +379,11 @@ describe('MongoDBVector Integration Tests', () => {
       const idToBeUpdated = ids[0];
       const newVector = [1, 2, 3, 4];
       const newMetaData = { test: 'updates' };
-      await vectorDB.updateVector(indexName, idToBeUpdated, { vector: newVector, metadata: newMetaData });
+      await vectorDB.updateVector({
+        indexName,
+        id: idToBeUpdated,
+        update: { vector: newVector, metadata: newMetaData },
+      });
       await new Promise(resolve => setTimeout(resolve, 5000));
       const results = await vectorDB.query({
         indexName,
@@ -399,7 +403,7 @@ describe('MongoDBVector Integration Tests', () => {
       expect(ids).toHaveLength(4);
       const idToBeUpdated = ids[0];
       const newMetaData = { test: 'metadata only update' };
-      await vectorDB.updateVector(indexName, idToBeUpdated, { metadata: newMetaData });
+      await vectorDB.updateVector({ indexName, id: idToBeUpdated, update: { metadata: newMetaData } });
       await new Promise(resolve => setTimeout(resolve, 5000));
       const results = await vectorDB.query({
         indexName,
@@ -419,7 +423,7 @@ describe('MongoDBVector Integration Tests', () => {
       expect(ids).toHaveLength(4);
       const idToBeUpdated = ids[0];
       const newVector = [1, 2, 3, 4];
-      await vectorDB.updateVector(indexName, idToBeUpdated, { vector: newVector });
+      await vectorDB.updateVector({ indexName, id: idToBeUpdated, update: { vector: newVector } });
       await new Promise(resolve => setTimeout(resolve, 5000));
       const results = await vectorDB.query({
         indexName,
@@ -434,21 +438,23 @@ describe('MongoDBVector Integration Tests', () => {
       expect(updatedResult?.vector).toEqual(newVector);
     });
     it('should throw exception when no updates are given', async () => {
-      await expect(vectorDB.updateVector(indexName, 'nonexistent-id', {})).rejects.toThrow('No updates provided');
+      await expect(vectorDB.updateVector({ indexName, id: 'nonexistent-id', update: {} })).rejects.toThrow(
+        'No updates provided',
+      );
     });
     it('should delete the vector by id', async () => {
       const ids = await vectorDB.upsert({ indexName, vectors: testVectors });
       expect(ids).toHaveLength(4);
       const idToBeDeleted = ids[0];
 
-      const initialStats = await vectorDB.describeIndex(indexName);
+      const initialStats = await vectorDB.describeIndex({ indexName });
 
-      await vectorDB.deleteVector(indexName, idToBeDeleted);
+      await vectorDB.deleteVector({ indexName, id: idToBeDeleted });
       const results = await vectorDB.query({ indexName, queryVector: [1, 0, 0, 0], topK: 2 });
       expect(results.map(res => res.id)).not.toContain(idToBeDeleted);
 
       await new Promise(resolve => setTimeout(resolve, 5000)); // Wait for count to update
-      const finalStats = await vectorDB.describeIndex(indexName);
+      const finalStats = await vectorDB.describeIndex({ indexName });
       expect(finalStats.count).toBe(initialStats.count - 1);
     });
   });

package/src/vector/index.ts

@@ -5,9 +5,10 @@ import type {
   CreateIndexParams,
   UpsertVectorParams,
   QueryVectorParams,
-  ParamsToArgs,
-  QueryVectorArgs,
-  UpsertVectorArgs,
+  DescribeIndexParams,
+  DeleteIndexParams,
+  DeleteVectorParams,
+  UpdateVectorParams,
 } from '@mastra/core/vector';
 import type { VectorFilter } from '@mastra/core/vector/filter';
 import { MongoClient } from 'mongodb';
@@ -17,10 +18,6 @@ import { v4 as uuidv4 } from 'uuid';
 import { MongoDBFilterTranslator } from './filter';
 
 // Define necessary types and interfaces
-export type MongoDBUpsertArgs = [...UpsertVectorArgs, string[]?];
-export type MongoDBQueryArgs = [...QueryVectorArgs, string?];
-export type MongoDBUpsertParams = ParamsToArgs<MongoDBUpsertArgs>;
-
 export interface MongoDBUpsertVectorParams extends UpsertVectorParams {
   documents?: string[];
 }
@@ -29,6 +26,12 @@ export interface MongoDBQueryVectorParams extends QueryVectorParams {
   documentFilter?: VectorFilter;
 }
 
+export interface MongoDBIndexReadyParams {
+  indexName: string;
+  timeoutMs?: number;
+  checkIntervalMs?: number;
+}
+
 // Define the document interface
 interface MongoDBDocument extends Document {
   _id: string; // Explicitly declare '_id' as string
@@ -68,9 +71,7 @@ export class MongoDBVector extends MastraVector {
     await this.client.close();
   }
 
-  async createIndex(params: CreateIndexParams): Promise<void> {
-    const { indexName, dimension, metric = 'cosine' } = params;
-
+  async createIndex({ indexName, dimension, metric = 'cosine' }: CreateIndexParams): Promise<void> {
     if (!Number.isInteger(dimension) || dimension <= 0) {
       throw new Error('Dimension must be a positive integer');
     }
@@ -118,7 +119,19 @@ export class MongoDBVector extends MastraVector {
     await collection.updateOne({ _id: '__index_metadata__' }, { $set: { dimension, metric } }, { upsert: true });
   }
 
-  async waitForIndexReady(indexName: string, timeoutMs: number = 60000, checkIntervalMs: number = 2000): Promise<void> {
+  /**
+   * Waits for the index to be ready.
+   *
+   * @param {string} indexName - The name of the index to wait for
+   * @param {number} timeoutMs - The maximum time in milliseconds to wait for the index to be ready (default: 60000)
+   * @param {number} checkIntervalMs - The interval in milliseconds at which to check if the index is ready (default: 2000)
+   * @returns A promise that resolves when the index is ready
+   */
+  async waitForIndexReady({
+    indexName,
+    timeoutMs = 60000,
+    checkIntervalMs = 2000,
+  }: MongoDBIndexReadyParams): Promise<void> {
     const collection = await this.getCollection(indexName, true);
     const indexNameInternal = `${indexName}_vector_index`;
 
@@ -135,15 +148,13 @@ export class MongoDBVector extends MastraVector {
     throw new Error(`Index "${indexNameInternal}" did not become ready within timeout`);
   }
 
-  async upsert(params: MongoDBUpsertVectorParams): Promise<string[]> {
-    const { indexName, vectors, metadata, ids, documents } = params;
-
+  async upsert({ indexName, vectors, metadata, ids, documents }: MongoDBUpsertVectorParams): Promise<string[]> {
     const collection = await this.getCollection(indexName);
 
     this.collectionForValidation = collection;
 
     // Get index stats to check dimension
-    const stats = await this.describeIndex(indexName);
+    const stats = await this.describeIndex({ indexName });
 
     // Validate vector dimensions
     await this.validateVectorDimensions(vectors, stats.dimension);
@@ -187,9 +198,14 @@ export class MongoDBVector extends MastraVector {
     return generatedIds;
   }
 
-  async query(params: MongoDBQueryVectorParams): Promise<QueryResult[]> {
-    const { indexName, queryVector, topK = 10, filter, includeVector = false, documentFilter } = params;
-
+  async query({
+    indexName,
+    queryVector,
+    topK = 10,
+    filter,
+    includeVector = false,
+    documentFilter,
+  }: MongoDBQueryVectorParams): Promise<QueryResult[]> {
     const collection = await this.getCollection(indexName, true);
     const indexNameInternal = `${indexName}_vector_index`;
 
@@ -255,7 +271,13 @@ export class MongoDBVector extends MastraVector {
     return collections.map(col => col.name);
   }
 
-  async describeIndex(indexName: string): Promise<IndexStats> {
+  /**
+   * Retrieves statistics about a vector index.
+   *
+   * @param {string} indexName - The name of the index to describe
+   * @returns A promise that resolves to the index statistics including dimension, count and metric
+   */
+  async describeIndex({ indexName }: DescribeIndexParams): Promise<IndexStats> {
     const collection = await this.getCollection(indexName, true);
 
     // Get the count of documents, excluding the metadata document
@@ -273,7 +295,7 @@ export class MongoDBVector extends MastraVector {
     };
   }
 
-  async deleteIndex(indexName: string): Promise<void> {
+  async deleteIndex({ indexName }: DeleteIndexParams): Promise<void> {
     const collection = await this.getCollection(indexName, false); // Do not throw error if collection doesn't exist
     if (collection) {
       await collection.drop();
@@ -285,8 +307,6 @@ export class MongoDBVector extends MastraVector {
   }
 
   /**
-   * @deprecated Use {@link updateVector} instead. This method will be removed on May 20th, 2025.
-   *
    * Updates a vector by its ID with the provided vector and/or metadata.
    * @param indexName - The name of the index containing the vector.
    * @param id - The ID of the vector to update.
@@ -296,34 +316,7 @@ export class MongoDBVector extends MastraVector {
    * @returns A promise that resolves when the update is complete.
    * @throws Will throw an error if no updates are provided or if the update operation fails.
    */
-  async updateIndexById(
-    indexName: string,
-    id: string,
-    update: { vector?: number[]; metadata?: Record<string, any> },
-  ): Promise<void> {
-    this.logger.warn(
-      `Deprecation Warning: updateIndexById() is deprecated. 
-      Please use updateVector() instead. 
-      updateIndexById() will be removed on May 20th, 2025.`,
-    );
-    await this.updateVector(indexName, id, update);
-  }
-
-  /**
-   * Updates a vector by its ID with the provided vector and/or metadata.
-   * @param indexName - The name of the index containing the vector.
-   * @param id - The ID of the vector to update.
-   * @param update - An object containing the vector and/or metadata to update.
-   * @param update.vector - An optional array of numbers representing the new vector.
-   * @param update.metadata - An optional record containing the new metadata.
-   * @returns A promise that resolves when the update is complete.
-   * @throws Will throw an error if no updates are provided or if the update operation fails.
-   */
-  async updateVector(
-    indexName: string,
-    id: string,
-    update: { vector?: number[]; metadata?: Record<string, any> },
-  ): Promise<void> {
+  async updateVector({ indexName, id, update }: UpdateVectorParams): Promise<void> {
     try {
       if (!update.vector && !update.metadata) {
         throw new Error('No updates provided');
@@ -333,7 +326,7 @@ export class MongoDBVector extends MastraVector {
       const updateDoc: Record<string, any> = {};
 
       if (update.vector) {
-        const stats = await this.describeIndex(indexName);
+        const stats = await this.describeIndex({ indexName });
         await this.validateVectorDimensions([update.vector], stats.dimension);
         updateDoc[this.embeddingFieldName] = update.vector;
       }
@@ -358,24 +351,6 @@ export class MongoDBVector extends MastraVector {
     }
   }
 
-  /**
-   * @deprecated Use {@link deleteVector} instead. This method will be removed on May 20th, 2025.
-   *
-   * Deletes a vector by its ID.
-   * @param indexName - The name of the index containing the vector.
-   * @param id - The ID of the vector to delete.
-   * @returns A promise that resolves when the deletion is complete.
-   * @throws Will throw an error if the deletion operation fails.
-   */
-  async deleteIndexById(indexName: string, id: string): Promise<void> {
-    this.logger.warn(
-      `Deprecation Warning: deleteIndexById() is deprecated. 
-      Please use deleteVector() instead. 
-      deleteIndexById() will be removed on May 20th, 2025.`,
-    );
-    await this.deleteVector(indexName, id);
-  }
-
   /**
    * Deletes a vector by its ID.
    * @param indexName - The name of the index containing the vector.
@@ -383,7 +358,7 @@ export class MongoDBVector extends MastraVector {
    * @returns A promise that resolves when the deletion is complete.
    * @throws Will throw an error if the deletion operation fails.
    */
-  async deleteVector(indexName: string, id: string): Promise<void> {
+  async deleteVector({ indexName, id }: DeleteVectorParams): Promise<void> {
     try {
       const collection = await this.getCollection(indexName, true);
       await collection.deleteOne({ _id: id });