@mastra/chroma 0.2.7-alpha.7 → 0.2.7-alpha.8

package/.turbo/turbo-build.log

@@ -1,23 +1,23 @@
 
-> @mastra/chroma@0.2.7-alpha.7 build /home/runner/work/mastra/mastra/stores/chroma
+> @mastra/chroma@0.2.7-alpha.8 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 8990ms
+TSC ⚡️ Build success in 8184ms
 DTS Build start
 CLI Target: es2022
 Analysis will use the bundled TypeScript version 5.8.2
 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.2
 Writing package typings: /home/runner/work/mastra/mastra/stores/chroma/dist/_tsup-dts-rollup.d.cts
-DTS ⚡️ Build success in 9209ms
+DTS ⚡️ Build success in 8248ms
 CLI Cleaning output folder
 ESM Build start
 CJS Build start
 CJS dist/index.cjs 8.08 KB
-CJS ⚡️ Build success in 744ms
+CJS ⚡️ Build success in 618ms
 ESM dist/index.js 8.05 KB
-ESM ⚡️ Build success in 745ms
+ESM ⚡️ Build success in 624ms

package/CHANGELOG.md

@@ -1,5 +1,13 @@
 # @mastra/chroma
 
+## 0.2.7-alpha.8
+
+### Patch Changes
+
+- a3c7b70: Update chroma version and tests
+- Updated dependencies [8df4a77]
+  - @mastra/core@0.8.0-alpha.8
+
 ## 0.2.7-alpha.7
 
 ### Patch Changes

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@mastra/chroma",
-  "version": "0.2.7-alpha.7",
+  "version": "0.2.7-alpha.8",
   "description": "Chroma vector store provider for Mastra",
   "type": "module",
   "main": "dist/index.js",
@@ -19,8 +19,8 @@
     "./package.json": "./package.json"
   },
   "dependencies": {
-    "chromadb": "^1.10.5",
-    "@mastra/core": "^0.8.0-alpha.7"
+    "chromadb": "^2.2.0",
+    "@mastra/core": "^0.8.0-alpha.8"
   },
   "devDependencies": {
     "@microsoft/api-extractor": "^7.52.1",

package/src/vector/index.test.ts

@@ -2,7 +2,6 @@ import type { QueryResult, IndexStats } from '@mastra/core/vector';
 import { describe, expect, beforeEach, afterEach, it, beforeAll, afterAll, vi } from 'vitest';
 
 import { ChromaVector } from './';
-import { Collection } from 'chromadb';
 
 describe('ChromaVector Integration Tests', () => {
   let vectorDB = new ChromaVector({
@@ -196,8 +195,8 @@ describe('ChromaVector Integration Tests', () => {
       expect(results[0]?.vector).toEqual(newVector);
     });
 
-    it('should throw exception when no updates are given', () => {
-      expect(vectorDB.updateIndexById(testIndexName, 'id', {})).rejects.toThrow('No updates provided');
+    it('should throw exception when no updates are given', async () => {
+      await expect(vectorDB.updateIndexById(testIndexName, 'id', {})).rejects.toThrow('No updates provided');
     });
 
     it('should delete the vector by id', async () => {
@@ -759,22 +758,20 @@ describe('ChromaVector Integration Tests', () => {
         expect(results.length).toBeGreaterThan(0);
       });
 
-      it('requires multiple conditions in logical operators', async () => {
-        await expect(
-          vectorDB.query({
-            indexName: testIndexName2,
-            queryVector: [1, 0, 0],
-            filter: { $and: [{ category: 'electronics' }] },
-          }),
-        ).rejects.toThrow();
+      it('accepts single conditions in logical operators', async () => {
+        const results = await vectorDB.query({
+          indexName: testIndexName2,
+          queryVector: [1, 0, 0],
+          filter: { $and: [{ category: 'electronics' }] },
+        });
+        expect(results.length).toBeGreaterThan(0);
 
-        await expect(
-          vectorDB.query({
-            indexName: testIndexName2,
-            queryVector: [1, 0, 0],
-            filter: { $or: [{ price: { $gt: 900 } }] },
-          }),
-        ).rejects.toThrow();
+        const results2 = await vectorDB.query({
+          indexName: testIndexName2,
+          queryVector: [1, 0, 0],
+          filter: { $or: [{ price: { $gt: 900 } }] },
+        });
+        expect(results2.length).toBeGreaterThan(0);
       });
     });
 
@@ -1292,15 +1289,14 @@ describe('ChromaVector Integration Tests', () => {
     });
 
     describe('Edge Cases and Validation', () => {
-      it('should reject empty string in $contains', async () => {
-        await expect(
-          vectorDB.query({
-            indexName: testIndexName3,
-            queryVector: [1.0, 0.0, 0.0],
-            topK: 3,
-            documentFilter: { $contains: '' },
-          }),
-        ).rejects.toThrow('Expected where document operand value for operator $contains to be a non-empty str');
+      it('allows empty string in $contains', async () => {
+        const results = await vectorDB.query({
+          indexName: testIndexName3,
+          queryVector: [1.0, 0.0, 0.0],
+          topK: 3,
+          documentFilter: { $contains: '' },
+        });
+        expect(results).toHaveLength(3);
       });
 
       it('should be case sensitive', async () => {
@@ -1515,4 +1511,89 @@ describe('ChromaVector Integration Tests', () => {
       expect(upsertResults).toHaveLength(1);
     });
   });
+
+  describe('Performance and Concurrency', () => {
+    const perfTestIndex = 'perf-test-index';
+
+    beforeEach(async () => {
+      try {
+        await vectorDB.deleteIndex(perfTestIndex);
+      } catch {
+        // Ignore errors if index doesn't exist
+      }
+      await vectorDB.createIndex({ indexName: perfTestIndex, dimension });
+    }, 10000);
+
+    afterEach(async () => {
+      try {
+        await vectorDB.deleteIndex(perfTestIndex);
+      } catch {
+        // Ignore cleanup errors
+      }
+    }, 10000);
+
+    it('handles concurrent operations correctly', async () => {
+      const promises = Array(10)
+        .fill(0)
+        .map((_, i) =>
+          vectorDB.upsert({
+            indexName: perfTestIndex,
+            vectors: [[1, 0, 0]],
+            metadata: [{ test: 'concurrent', id: i }],
+            ids: [`concurrent-${i}`],
+          }),
+        );
+      await Promise.all(promises);
+
+      const results = await vectorDB.query({
+        indexName: perfTestIndex,
+        queryVector: [1, 0, 0],
+        filter: { test: 'concurrent' },
+      });
+      expect(results).toHaveLength(10);
+    }, 15000);
+
+    it('handles large batch operations', async () => {
+      const batchSize = 100; // Using 100 instead of 1000 to keep tests fast
+      const vectors = Array(batchSize)
+        .fill(0)
+        .map(() => [1, 0, 0]);
+      const metadata = vectors.map((_, i) => ({ index: i, test: 'batch' }));
+      const ids = vectors.map((_, i) => `batch-${i}`);
+
+      await vectorDB.upsert({
+        indexName: perfTestIndex,
+        vectors,
+        metadata,
+        ids,
+      });
+
+      // Verify all vectors were inserted
+      const stats = await vectorDB.describeIndex(perfTestIndex);
+      expect(stats.count).toBe(batchSize);
+
+      const results = await vectorDB.query({
+        indexName: perfTestIndex,
+        queryVector: [1, 0, 0],
+        filter: { test: 'batch' },
+        topK: batchSize,
+      });
+      expect(results).toHaveLength(batchSize);
+
+      // Test querying with pagination
+      const pageSize = 20;
+      const pages: QueryResult[][] = [];
+      for (let i = 0; i < batchSize; i += pageSize) {
+        const page = await vectorDB.query({
+          indexName: perfTestIndex,
+          queryVector: [1, 0, 0],
+          filter: { test: 'batch' },
+          topK: pageSize,
+        });
+        pages.push(page);
+        expect(page).toHaveLength(Math.min(pageSize, batchSize - i));
+      }
+      expect(pages).toHaveLength(Math.ceil(batchSize / pageSize));
+    }, 30000);
+  });
 });

package/src/vector/index.ts

@@ -10,11 +10,10 @@ import type {
   UpsertVectorArgs,
 } from '@mastra/core/vector';
 
-import { ChromaClient, Collection } from 'chromadb';
-import { ChromaFilterTranslator } from './filter';
-
-import type { UpdateRecordsParams } from 'chromadb';
 import type { VectorFilter } from '@mastra/core/vector/filter';
+import { ChromaClient } from 'chromadb';
+import type { UpdateRecordsParams, Collection } from 'chromadb';
+import { ChromaFilterTranslator } from './filter';
 
 interface ChromaUpsertVectorParams extends UpsertVectorParams {
   documents?: string[];