@mastra/couchbase 0.0.0-trigger-playground-ui-package-20250506151043 → 0.0.0-vector-query-sources-20250516172905

package/dist/index.js

@@ -1,5 +1,5 @@
 import { MastraVector } from '@mastra/core/vector';
-import { connect, SearchRequest, VectorSearch, VectorQuery } from 'couchbase';
+import { connect, SearchRequest, VectorSearch, VectorQuery, MutateInSpec } from 'couchbase';
 
 // src/vector/index.ts
 var DISTANCE_MAPPING = {
@@ -17,11 +17,32 @@ var CouchbaseVector = class extends MastraVector {
   bucket;
   scope;
   vector_dimension;
-  constructor(cnn_string, username, password, bucketName, scopeName, collectionName) {
+  constructor(paramsOrConnectionString, username, password, bucketName, scopeName, collectionName) {
+    let connectionString_, username_, password_, bucketName_, scopeName_, collectionName_;
+    if (typeof paramsOrConnectionString === "object" && paramsOrConnectionString !== null && "connectionString" in paramsOrConnectionString) {
+      connectionString_ = paramsOrConnectionString.connectionString;
+      username_ = paramsOrConnectionString.username;
+      password_ = paramsOrConnectionString.password;
+      bucketName_ = paramsOrConnectionString.bucketName;
+      scopeName_ = paramsOrConnectionString.scopeName;
+      collectionName_ = paramsOrConnectionString.collectionName;
+    } else {
+      if (arguments.length > 1) {
+        console.warn(
+          "Deprecation Warning: CouchbaseVector constructor positional arguments are deprecated. Please use a single object parameter instead. This signature will be removed on May 20th, 2025."
+        );
+      }
+      connectionString_ = paramsOrConnectionString;
+      username_ = username;
+      password_ = password;
+      bucketName_ = bucketName;
+      scopeName_ = scopeName;
+      collectionName_ = collectionName;
+    }
     super();
-    const baseClusterPromise = connect(cnn_string, {
-      username,
-      password,
+    const baseClusterPromise = connect(connectionString_, {
+      username: username_,
+      password: password_,
       configProfile: "wanDevelopment"
     });
     const telemetry = this.__getTelemetry();
@@ -32,9 +53,9 @@ var CouchbaseVector = class extends MastraVector {
       }
     }) ?? baseClusterPromise;
     this.cluster = null;
-    this.bucketName = bucketName;
-    this.collectionName = collectionName;
-    this.scopeName = scopeName;
+    this.bucketName = bucketName_;
+    this.collectionName = collectionName_;
+    this.scopeName = scopeName_;
     this.collection = null;
     this.bucket = null;
     this.scope = null;
@@ -57,86 +78,95 @@ var CouchbaseVector = class extends MastraVector {
     if (!Number.isInteger(dimension) || dimension <= 0) {
       throw new Error("Dimension must be a positive integer");
     }
-    await this.scope.searchIndexes().upsertIndex({
-      name: indexName,
-      sourceName: this.bucketName,
-      type: "fulltext-index",
-      params: {
-        doc_config: {
-          docid_prefix_delim: "",
-          docid_regexp: "",
-          mode: "scope.collection.type_field",
-          type_field: "type"
-        },
-        mapping: {
-          default_analyzer: "standard",
-          default_datetime_parser: "dateTimeOptional",
-          default_field: "_all",
-          default_mapping: {
-            dynamic: true,
-            enabled: false
+    try {
+      await this.scope.searchIndexes().upsertIndex({
+        name: indexName,
+        sourceName: this.bucketName,
+        type: "fulltext-index",
+        params: {
+          doc_config: {
+            docid_prefix_delim: "",
+            docid_regexp: "",
+            mode: "scope.collection.type_field",
+            type_field: "type"
           },
-          default_type: "_default",
-          docvalues_dynamic: true,
-          // [Doc](https://docs.couchbase.com/server/current/search/search-index-params.html#params) mentions this attribute is required for vector search to return the indexed field
-          index_dynamic: true,
-          store_dynamic: true,
-          // [Doc](https://docs.couchbase.com/server/current/search/search-index-params.html#params) mentions this attribute is required for vector search to return the indexed field
-          type_field: "_type",
-          types: {
-            [`${this.scopeName}.${this.collectionName}`]: {
+          mapping: {
+            default_analyzer: "standard",
+            default_datetime_parser: "dateTimeOptional",
+            default_field: "_all",
+            default_mapping: {
               dynamic: true,
-              enabled: true,
-              properties: {
-                embedding: {
-                  enabled: true,
-                  fields: [
-                    {
-                      dims: dimension,
-                      index: true,
-                      name: "embedding",
-                      similarity: DISTANCE_MAPPING[metric],
-                      type: "vector",
-                      vector_index_optimized_for: "recall",
-                      store: true,
-                      // CHANGED due to https://docs.couchbase.com/server/current/search/search-index-params.html#fields
-                      docvalues: true,
-                      // CHANGED due to https://docs.couchbase.com/server/current/search/search-index-params.html#fields
-                      include_term_vectors: true
-                      // CHANGED due to https://docs.couchbase.com/server/current/search/search-index-params.html#fields
-                    }
-                  ]
-                },
-                content: {
-                  enabled: true,
-                  fields: [
-                    {
-                      index: true,
-                      name: "content",
-                      store: true,
-                      type: "text"
-                    }
-                  ]
+              enabled: false
+            },
+            default_type: "_default",
+            docvalues_dynamic: true,
+            // [Doc](https://docs.couchbase.com/server/current/search/search-index-params.html#params) mentions this attribute is required for vector search to return the indexed field
+            index_dynamic: true,
+            store_dynamic: true,
+            // [Doc](https://docs.couchbase.com/server/current/search/search-index-params.html#params) mentions this attribute is required for vector search to return the indexed field
+            type_field: "_type",
+            types: {
+              [`${this.scopeName}.${this.collectionName}`]: {
+                dynamic: true,
+                enabled: true,
+                properties: {
+                  embedding: {
+                    enabled: true,
+                    fields: [
+                      {
+                        dims: dimension,
+                        index: true,
+                        name: "embedding",
+                        similarity: DISTANCE_MAPPING[metric],
+                        type: "vector",
+                        vector_index_optimized_for: "recall",
+                        store: true,
+                        // CHANGED due to https://docs.couchbase.com/server/current/search/search-index-params.html#fields
+                        docvalues: true,
+                        // CHANGED due to https://docs.couchbase.com/server/current/search/search-index-params.html#fields
+                        include_term_vectors: true
+                        // CHANGED due to https://docs.couchbase.com/server/current/search/search-index-params.html#fields
+                      }
+                    ]
+                  },
+                  content: {
+                    enabled: true,
+                    fields: [
+                      {
+                        index: true,
+                        name: "content",
+                        store: true,
+                        type: "text"
+                      }
+                    ]
+                  }
                 }
               }
             }
+          },
+          store: {
+            indexType: "scorch",
+            segmentVersion: 16
           }
         },
-        store: {
-          indexType: "scorch",
-          segmentVersion: 16
+        sourceUuid: "",
+        sourceParams: {},
+        sourceType: "gocbcore",
+        planParams: {
+          maxPartitionsPerPIndex: 64,
+          indexPartitions: 16,
+          numReplicas: 0
         }
-      },
-      sourceUuid: "",
-      sourceParams: {},
-      sourceType: "gocbcore",
-      planParams: {
-        maxPartitionsPerPIndex: 64,
-        indexPartitions: 16,
-        numReplicas: 0
+      });
+      this.vector_dimension = dimension;
+    } catch (error) {
+      const message = error?.message || error?.toString();
+      if (message && message.toLowerCase().includes("index exists")) {
+        await this.validateExistingIndex(indexName, dimension, metric);
+        return;
       }
-    });
-    this.vector_dimension = dimension;
+      throw error;
+    }
   }
   async upsert(params) {
     const { vectors, metadata, ids } = params;
@@ -210,7 +240,16 @@ var CouchbaseVector = class extends MastraVector {
     const indexes = await this.scope.searchIndexes().getAllIndexes();
     return indexes?.map((index) => index.name) || [];
   }
-  async describeIndex(indexName) {
+  /**
+   * Retrieves statistics about a vector index.
+   *
+   * @param params - The parameters for describing an index
+   * @param params.indexName - The name of the index to describe
+   * @returns A promise that resolves to the index statistics including dimension, count and metric
+   */
+  async describeIndex(...args) {
+    const params = this.normalizeArgs("describeIndex", args);
+    const { indexName } = params;
     await this.getCollection();
     if (!(await this.listIndexes()).includes(indexName)) {
       throw new Error(`Index ${indexName} does not exist`);
@@ -227,7 +266,9 @@ var CouchbaseVector = class extends MastraVector {
       )
     };
   }
-  async deleteIndex(indexName) {
+  async deleteIndex(...args) {
+    const params = this.normalizeArgs("deleteIndex", args);
+    const { indexName } = params;
     await this.getCollection();
     if (!(await this.listIndexes()).includes(indexName)) {
       throw new Error(`Index ${indexName} does not exist`);
@@ -235,6 +276,60 @@ var CouchbaseVector = class extends MastraVector {
     await this.scope.searchIndexes().dropIndex(indexName);
     this.vector_dimension = null;
   }
+  /**
+   * 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(...args) {
+    const params = this.normalizeArgs("updateVector", args);
+    const { id, update } = params;
+    if (!update.vector && !update.metadata) {
+      throw new Error("No updates provided");
+    }
+    if (update.vector && this.vector_dimension && update.vector.length !== this.vector_dimension) {
+      throw new Error("Vector dimension mismatch");
+    }
+    const collection = await this.getCollection();
+    try {
+      await collection.get(id);
+    } catch (err) {
+      if (err.code === 13 || err.message?.includes("document not found")) {
+        throw new Error(`Vector with id ${id} does not exist`);
+      }
+      throw err;
+    }
+    const specs = [];
+    if (update.vector) specs.push(MutateInSpec.replace("embedding", update.vector));
+    if (update.metadata) specs.push(MutateInSpec.replace("metadata", update.metadata));
+    await collection.mutateIn(id, specs);
+  }
+  /**
+   * 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 deleteVector(...args) {
+    const params = this.normalizeArgs("deleteVector", args);
+    const { id } = params;
+    const collection = await this.getCollection();
+    try {
+      await collection.get(id);
+    } catch (err) {
+      if (err.code === 13 || err.message?.includes("document not found")) {
+        throw new Error(`Vector with id ${id} does not exist`);
+      }
+      throw err;
+    }
+    await collection.remove(id);
+  }
 };
 
 export { CouchbaseVector, DISTANCE_MAPPING };

package/docker-compose.yaml

@@ -2,20 +2,20 @@ services:
   mastra_couchbase_testing:
     image: couchbase:enterprise-7.6.4
     ports:
-      - "8091-8095:8091-8095"
-      - "11210:11210"
-      - "9102:9102"
+      - '8091-8095:8091-8095'
+      - '11210:11210'
+      - '9102:9102'
     expose:
-      - "8091"
-      - "8092"
-      - "8093"
-      - "8094"
-      - "8095"
-      - "9102"
-      - "11210"
+      - '8091'
+      - '8092'
+      - '8093'
+      - '8094'
+      - '8095'
+      - '9102'
+      - '11210'
     healthcheck: # checks couchbase server is up
-      test: ["CMD", "curl", "-v", "http://localhost:8091/pools"]
+      test: ['CMD', 'curl', '-v', 'http://localhost:8091/pools']
       interval: 20s
       timeout: 20s
       retries: 5
-    container_name: mastra_couchbase_testing
+    container_name: mastra_couchbase_testing

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@mastra/couchbase",
-  "version": "0.0.0-trigger-playground-ui-package-20250506151043",
+  "version": "0.0.0-vector-query-sources-20250516172905",
   "description": "Couchbase vector store provider for Mastra",
   "type": "module",
   "main": "dist/index.js",
@@ -20,7 +20,7 @@
   },
   "dependencies": {
     "couchbase": "^4.4.5",
-    "@mastra/core": "0.0.0-trigger-playground-ui-package-20250506151043"
+    "@mastra/core": "0.0.0-vector-query-sources-20250516172905"
   },
   "devDependencies": {
     "@microsoft/api-extractor": "^7.52.1",
@@ -32,7 +32,7 @@
     "tsup": "^8.4.0",
     "typescript": "^5.8.2",
     "vitest": "^3.0.9",
-    "@internal/lint": "0.0.0-trigger-playground-ui-package-20250506151043"
+    "@internal/lint": "0.0.0-vector-query-sources-20250516172905"
   },
   "scripts": {
     "build": "tsup src/index.ts --format esm,cjs --experimental-dts --clean --treeshake=smallest --splitting",

package/src/vector/index.integration.test.ts

@@ -3,10 +3,11 @@
 // The tests will automatically start and configure the required Couchbase container.
 
 import { execSync } from 'child_process';
+import { randomUUID } from 'crypto';
 import axios from 'axios';
 import type { Cluster, Bucket, Scope, Collection } from 'couchbase';
 import { connect } from 'couchbase';
-import { describe, it, expect, beforeAll, afterAll } from 'vitest';
+import { describe, it, expect, beforeAll, afterAll, vi } from 'vitest';
 import { CouchbaseVector, DISTANCE_MAPPING } from './index';
 
 const containerName = 'mastra_couchbase_testing';
@@ -172,14 +173,14 @@ describe('Integration Testing CouchbaseVector', async () => {
 
   describe('Connection', () => {
     it('should connect to couchbase', async () => {
-      couchbase_client = new CouchbaseVector(
+      couchbase_client = new CouchbaseVector({
         connectionString,
         username,
         password,
-        test_bucketName,
-        test_scopeName,
-        test_collectionName,
-      );
+        bucketName: test_bucketName,
+        scopeName: test_scopeName,
+        collectionName: test_collectionName,
+      });
       expect(couchbase_client).toBeDefined();
       const collection = await couchbase_client.getCollection();
       expect(collection).toBeDefined();
@@ -210,14 +211,14 @@ describe('Integration Testing CouchbaseVector', async () => {
     }, 50000);
 
     it('should describe index', async () => {
-      const stats = await couchbase_client.describeIndex(test_indexName);
+      const stats = await couchbase_client.describeIndex({ indexName: test_indexName });
       expect(stats.dimension).toBe(dimension);
       expect(stats.metric).toBe('euclidean'); // similiarity(=="l2_norm") is mapped to euclidean in couchbase
       expect(typeof stats.count).toBe('number');
     }, 50000);
 
     it('should delete index', async () => {
-      await couchbase_client.deleteIndex(test_indexName);
+      await couchbase_client.deleteIndex({ indexName: test_indexName });
       await new Promise(resolve => setTimeout(resolve, 5000));
       await expect(scope.searchIndexes().getIndex(test_indexName)).rejects.toThrowError();
     }, 50000);
@@ -245,7 +246,7 @@ describe('Integration Testing CouchbaseVector', async () => {
     }, 50000);
 
     afterAll(async () => {
-      await couchbase_client.deleteIndex(test_indexName);
+      await couchbase_client.deleteIndex({ indexName: test_indexName });
       await new Promise(resolve => setTimeout(resolve, 5000));
     }, 50000);
 
@@ -379,6 +380,120 @@ describe('Integration Testing CouchbaseVector', async () => {
         }),
       ).rejects.toThrow('Including vectors in search results is not yet supported by the Couchbase vector store');
     }, 50000);
+
+    it('should upsert vectors with generated ids', async () => {
+      const ids = await couchbase_client.upsert({ indexName: test_indexName, vectors: testVectors });
+      expect(ids).toHaveLength(testVectors.length);
+      ids.forEach(id => expect(typeof id).toBe('string'));
+
+      // Count is not supported by Couchbase
+      const stats = await couchbase_client.describeIndex({ indexName: test_indexName });
+      expect(stats.count).toBe(-1);
+    });
+
+    it('should update existing vectors', async () => {
+      // Initial upsert
+      await couchbase_client.upsert({
+        indexName: test_indexName,
+        vectors: testVectors,
+        metadata: testMetadata,
+        ids: testVectorIds,
+      });
+
+      // Update first vector
+      const updatedVector = [[0.5, 0.5, 0.0]];
+      const updatedMetadata = [{ label: 'updated-x-axis' }];
+      await couchbase_client.upsert({
+        indexName: test_indexName,
+        vectors: updatedVector,
+        metadata: updatedMetadata,
+        ids: [testVectorIds?.[0]!],
+      });
+
+      // Verify update
+      const result = await collection.get(testVectorIds?.[0]!);
+      expect(result.content.embedding).toEqual(updatedVector[0]);
+      expect(result.content.metadata).toEqual(updatedMetadata[0]);
+    });
+
+    it('should update the vector by id', async () => {
+      const ids = await couchbase_client.upsert({ indexName: test_indexName, vectors: testVectors });
+      expect(ids).toHaveLength(3);
+
+      const idToBeUpdated = ids[0];
+      const newVector = [1, 2, 3];
+      const newMetaData = {
+        test: 'updates',
+      };
+
+      const update = {
+        vector: newVector,
+        metadata: newMetaData,
+      };
+
+      await couchbase_client.updateVector({ indexName: test_indexName, id: idToBeUpdated, update });
+
+      const result = await collection.get(idToBeUpdated);
+      expect(result.content.embedding).toEqual(newVector);
+      expect(result.content.metadata).toEqual(newMetaData);
+    });
+
+    it('should only update the metadata by id', async () => {
+      const ids = await couchbase_client.upsert({ indexName: test_indexName, vectors: testVectors });
+      expect(ids).toHaveLength(3);
+
+      const idToBeUpdated = ids[0];
+      const newMetaData = {
+        test: 'updates',
+      };
+
+      const update = {
+        metadata: newMetaData,
+      };
+
+      await couchbase_client.updateVector({ indexName: test_indexName, id: idToBeUpdated, update });
+
+      const result = await collection.get(idToBeUpdated);
+      expect(result.content.embedding).toEqual(testVectors[0]);
+      expect(result.content.metadata).toEqual(newMetaData);
+    });
+
+    it('should only update vector embeddings by id', async () => {
+      const ids = await couchbase_client.upsert({ indexName: test_indexName, vectors: testVectors });
+      expect(ids).toHaveLength(3);
+
+      const idToBeUpdated = ids[0];
+      const newVector = [1, 2, 3];
+
+      const update = {
+        vector: newVector,
+      };
+
+      await couchbase_client.updateVector({ indexName: test_indexName, id: idToBeUpdated, update });
+
+      const result = await collection.get(idToBeUpdated);
+      expect(result.content.embedding).toEqual(newVector);
+    });
+
+    it('should throw exception when no updates are given', async () => {
+      await expect(couchbase_client.updateVector({ indexName: test_indexName, id: 'id', update: {} })).rejects.toThrow(
+        'No updates provided',
+      );
+    });
+
+    it('should delete the vector by id', async () => {
+      const ids = await couchbase_client.upsert({ indexName: test_indexName, vectors: testVectors });
+      expect(ids).toHaveLength(3);
+      const idToBeDeleted = ids[0];
+
+      await couchbase_client.deleteVector({ indexName: test_indexName, id: idToBeDeleted });
+
+      try {
+        await collection.get(idToBeDeleted);
+      } catch (error) {
+        expect(error).toBeInstanceOf(Error);
+      }
+    });
   });
 
   describe('Error Cases and Edge Cases', () => {
@@ -408,7 +523,7 @@ describe('Integration Testing CouchbaseVector', async () => {
       expect(allIndexes.find(idx => idx.name === nonExistentIndex)).toBeUndefined();
 
       // Now test the couchbase_client method
-      await expect(couchbase_client.describeIndex(nonExistentIndex)).rejects.toThrow();
+      await expect(couchbase_client.describeIndex({ indexName: nonExistentIndex })).rejects.toThrow();
     }, 50000);
 
     it('should throw error when deleting a non-existent index', async () => {
@@ -419,7 +534,7 @@ describe('Integration Testing CouchbaseVector', async () => {
       expect(allIndexes.find(idx => idx.name === nonExistentIndex)).toBeUndefined();
 
       // Now test the couchbase_client method
-      await expect(couchbase_client.deleteIndex(nonExistentIndex)).rejects.toThrow();
+      await expect(couchbase_client.deleteIndex({ indexName: nonExistentIndex })).rejects.toThrow();
     }, 50000);
 
     it('should throw error for empty vectors array in upsert', async () => {
@@ -431,6 +546,66 @@ describe('Integration Testing CouchbaseVector', async () => {
         }),
       ).rejects.toThrow('No vectors provided');
     }, 50000);
+
+    it('should handle non-existent index queries', async () => {
+      await expect(
+        couchbase_client.query({ indexName: 'non-existent-index', queryVector: [1, 2, 3] }),
+      ).rejects.toThrow();
+    }, 50000);
+
+    it('should handle duplicate index creation gracefully', async () => {
+      const duplicateIndexName = `duplicate-test-${randomUUID()}`;
+      const dimension = 768;
+      const infoSpy = vi.spyOn(couchbase_client['logger'], 'info');
+      const warnSpy = vi.spyOn(couchbase_client['logger'], 'warn');
+
+      try {
+        // Create index first time
+        await couchbase_client.createIndex({
+          indexName: duplicateIndexName,
+          dimension,
+          metric: 'cosine',
+        });
+
+        // Try to create with same dimensions - should not throw
+        await expect(
+          couchbase_client.createIndex({
+            indexName: duplicateIndexName,
+            dimension,
+            metric: 'cosine',
+          }),
+        ).resolves.not.toThrow();
+
+        expect(infoSpy).toHaveBeenCalledWith(expect.stringContaining('already exists with'));
+
+        // Try to create with same dimensions and different metric - should not throw
+        await expect(
+          couchbase_client.createIndex({
+            indexName: duplicateIndexName,
+            dimension,
+            metric: 'euclidean',
+          }),
+        ).resolves.not.toThrow();
+
+        expect(warnSpy).toHaveBeenCalledWith(expect.stringContaining('Attempted to create index with metric'));
+
+        // Try to create with different dimensions - should throw
+        await expect(
+          couchbase_client.createIndex({
+            indexName: duplicateIndexName,
+            dimension: dimension + 1,
+            metric: 'cosine',
+          }),
+        ).rejects.toThrow(
+          `Index "${duplicateIndexName}" already exists with ${dimension} dimensions, but ${dimension + 1} dimensions were requested`,
+        );
+      } finally {
+        infoSpy.mockRestore();
+        warnSpy.mockRestore();
+        // Cleanup
+        await couchbase_client.deleteIndex({ indexName: duplicateIndexName });
+      }
+    }, 50000);
   });
 
   describe('Vector Dimension Tracking', () => {
@@ -438,7 +613,7 @@ describe('Integration Testing CouchbaseVector', async () => {
       const indexes = await couchbase_client.listIndexes();
       if (indexes.length > 0) {
         for (const index of indexes) {
-          await couchbase_client.deleteIndex(index);
+          await couchbase_client.deleteIndex({ indexName: index });
           await new Promise(resolve => setTimeout(resolve, 5000));
         }
       }
@@ -501,7 +676,7 @@ describe('Integration Testing CouchbaseVector', async () => {
       expect((couchbase_client as any).vector_dimension).toBe(testDimension);
 
       // Delete the index
-      await couchbase_client.deleteIndex(testIndexName);
+      await couchbase_client.deleteIndex({ indexName: testIndexName });
       await new Promise(resolve => setTimeout(resolve, 5000));
 
       // Verify dimension is reset
@@ -517,7 +692,7 @@ describe('Integration Testing CouchbaseVector', async () => {
       const indexes = await couchbase_client.listIndexes();
       if (indexes.length > 0) {
         for (const index of indexes) {
-          await couchbase_client.deleteIndex(index);
+          await couchbase_client.deleteIndex({ indexName: index });
           await new Promise(resolve => setTimeout(resolve, 5000));
         }
       }
@@ -546,11 +721,11 @@ describe('Integration Testing CouchbaseVector', async () => {
         expect(similarityParam).toBe(couchbaseMetric);
 
         // Verify through our API
-        const stats = await couchbase_client.describeIndex(testIndexName);
+        const stats = await couchbase_client.describeIndex({ indexName: testIndexName });
         expect(stats.metric).toBe(mastraMetric);
 
         // Clean up
-        await couchbase_client.deleteIndex(testIndexName);
+        await couchbase_client.deleteIndex({ indexName: testIndexName });
         await new Promise(resolve => setTimeout(resolve, 5000));
       }
     }, 50000);