@mastra/chroma 0.2.13 → 0.2.14-alpha.1

package/.turbo/turbo-build.log

@@ -1,23 +1,23 @@
 
-> @mastra/chroma@0.2.13-alpha.6 build /home/runner/work/mastra/mastra/stores/chroma
+> @mastra/chroma@0.2.14-alpha.1 build /home/runner/work/mastra/mastra/stores/chroma
 > 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 8507ms
+TSC ⚡️ Build success in 8360ms
 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/chroma/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/chroma/dist/_tsup-dts-rollup.d.cts
-DTS ⚡️ Build success in 9564ms
+DTS ⚡️ Build success in 9498ms
 CLI Cleaning output folder
 ESM Build start
 CJS Build start
-CJS dist/index.cjs 10.90 KB
-CJS ⚡️ Build success in 707ms
-ESM dist/index.js 10.85 KB
-ESM ⚡️ Build success in 708ms
+ESM dist/index.js 11.12 KB
+ESM ⚡️ Build success in 520ms
+CJS dist/index.cjs 11.18 KB
+CJS ⚡️ Build success in 520ms

package/CHANGELOG.md

@@ -1,5 +1,26 @@
 # @mastra/chroma
 
+## 0.2.14-alpha.1
+
+### Patch Changes
+
+- 9cd1a46: [MASTRA-3338] update naming scheme for embedding index based on vector store rules and added duplicate index checks
+- Updated dependencies [e450778]
+- Updated dependencies [8902157]
+- Updated dependencies [ca0dc88]
+- Updated dependencies [9cd1a46]
+- Updated dependencies [70dbf51]
+  - @mastra/core@0.9.3-alpha.1
+
+## 0.2.14-alpha.0
+
+### Patch Changes
+
+- Updated dependencies [526c570]
+- Updated dependencies [b5d2de0]
+- Updated dependencies [644f8ad]
+  - @mastra/core@0.9.3-alpha.0
+
 ## 0.2.13
 
 ### Patch Changes

package/dist/index.cjs

@@ -157,13 +157,22 @@ var ChromaVector = class extends vector.MastraVector {
     if (!["cosine", "l2", "ip"].includes(hnswSpace)) {
       throw new Error(`Invalid metric: "${metric}". Must be one of: cosine, euclidean, dotproduct`);
     }
-    await this.client.createCollection({
-      name: indexName,
-      metadata: {
-        dimension,
-        "hnsw:space": this.HnswSpaceMap[metric]
+    try {
+      await this.client.createCollection({
+        name: indexName,
+        metadata: {
+          dimension,
+          "hnsw:space": hnswSpace
+        }
+      });
+    } catch (error) {
+      const message = error?.message || error?.toString();
+      if (message && message.toLowerCase().includes("already exists")) {
+        await this.validateExistingIndex(indexName, dimension, metric);
+        return;
       }
-    });
+      throw error;
+    }
   }
   transformFilter(filter) {
     const translator = new ChromaFilterTranslator();

package/dist/index.js

@@ -155,13 +155,22 @@ var ChromaVector = class extends MastraVector {
     if (!["cosine", "l2", "ip"].includes(hnswSpace)) {
       throw new Error(`Invalid metric: "${metric}". Must be one of: cosine, euclidean, dotproduct`);
     }
-    await this.client.createCollection({
-      name: indexName,
-      metadata: {
-        dimension,
-        "hnsw:space": this.HnswSpaceMap[metric]
+    try {
+      await this.client.createCollection({
+        name: indexName,
+        metadata: {
+          dimension,
+          "hnsw:space": hnswSpace
+        }
+      });
+    } catch (error) {
+      const message = error?.message || error?.toString();
+      if (message && message.toLowerCase().includes("already exists")) {
+        await this.validateExistingIndex(indexName, dimension, metric);
+        return;
       }
-    });
+      throw error;
+    }
   }
   transformFilter(filter) {
     const translator = new ChromaFilterTranslator();

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@mastra/chroma",
-  "version": "0.2.13",
+  "version": "0.2.14-alpha.1",
   "description": "Chroma vector store provider for Mastra",
   "type": "module",
   "main": "dist/index.js",
@@ -21,7 +21,7 @@
   "license": "MIT",
   "dependencies": {
     "chromadb": "^2.2.0",
-    "@mastra/core": "^0.9.2"
+    "@mastra/core": "^0.9.3-alpha.1"
   },
   "devDependencies": {
     "@microsoft/api-extractor": "^7.52.5",

package/src/vector/index.test.ts

@@ -273,8 +273,17 @@ describe('ChromaVector Integration Tests', () => {
   });
 
   describe('Error Handling', () => {
+    const testIndexName = 'test_index_error';
+    beforeAll(async () => {
+      await vectorDB.createIndex({ indexName: testIndexName, dimension: 3 });
+    });
+
+    afterAll(async () => {
+      await vectorDB.deleteIndex(testIndexName);
+    });
+
     it('should handle non-existent index queries', async () => {
-      await expect(vectorDB.query({ indexName: 'non-existent-index-yu', queryVector: [1, 2, 3] })).rejects.toThrow();
+      await expect(vectorDB.query({ indexName: 'non-existent-index', queryVector: [1, 2, 3] })).rejects.toThrow();
     });
 
     it('should handle invalid dimension vectors', async () => {
@@ -282,10 +291,59 @@ describe('ChromaVector Integration Tests', () => {
       await expect(vectorDB.upsert({ indexName: testIndexName, vectors: [invalidVector] })).rejects.toThrow();
     });
 
-    it('should handle mismatched metadata and vectors length', async () => {
-      const vectors = [[1, 2, 3]];
-      const metadata = [{}, {}]; // More metadata than vectors
-      await expect(vectorDB.upsert({ indexName: testIndexName, vectors, metadata })).rejects.toThrow();
+    it('should handle duplicate index creation gracefully', async () => {
+      const infoSpy = vi.spyOn(vectorDB['logger'], 'info');
+      const warnSpy = vi.spyOn(vectorDB['logger'], 'warn');
+
+      const duplicateIndexName = `duplicate-test`;
+      const dimension = 768;
+
+      try {
+        // Create index first time
+        await vectorDB.createIndex({
+          indexName: duplicateIndexName,
+          dimension,
+          metric: 'cosine',
+        });
+
+        // Try to create with same dimensions - should not throw
+        await expect(
+          vectorDB.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(
+          vectorDB.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(
+          vectorDB.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 vectorDB.deleteIndex(duplicateIndexName);
+      }
     });
   });
 

package/src/vector/index.ts

@@ -111,7 +111,6 @@ export class ChromaVector extends MastraVector {
 
   async createIndex(...args: ParamsToArgs<CreateIndexParams>): Promise<void> {
     const params = this.normalizeArgs<CreateIndexParams>('createIndex', args);
-
     const { indexName, dimension, metric = 'cosine' } = params;
 
     if (!Number.isInteger(dimension) || dimension <= 0) {
@@ -121,13 +120,24 @@ export class ChromaVector extends MastraVector {
     if (!['cosine', 'l2', 'ip'].includes(hnswSpace)) {
       throw new Error(`Invalid metric: "${metric}". Must be one of: cosine, euclidean, dotproduct`);
     }
-    await this.client.createCollection({
-      name: indexName,
-      metadata: {
-        dimension,
-        'hnsw:space': this.HnswSpaceMap[metric],
-      },
-    });
+    try {
+      await this.client.createCollection({
+        name: indexName,
+        metadata: {
+          dimension,
+          'hnsw:space': hnswSpace,
+        },
+      });
+    } catch (error: any) {
+      // Check for 'already exists' error
+      const message = error?.message || error?.toString();
+      if (message && message.toLowerCase().includes('already exists')) {
+        // Fetch collection info and check dimension
+        await this.validateExistingIndex(indexName, dimension, metric);
+        return;
+      }
+      throw error;
+    }
   }
 
   transformFilter(filter?: VectorFilter) {