@mastra/chroma 0.1.8-alpha.4 → 0.2.0-alpha.6

package/.turbo/turbo-build.log

@@ -1,23 +1,23 @@
 
-> @mastra/chroma@0.1.8-alpha.4 build /home/runner/work/mastra/mastra/stores/chroma
+> @mastra/chroma@0.2.0-alpha.6 build /home/runner/work/mastra/mastra/stores/chroma
 > tsup src/index.ts --format esm,cjs --experimental-dts --clean --treeshake
 
 CLI Building entry: src/index.ts
 CLI Using tsconfig: tsconfig.json
 CLI tsup v8.3.6
 TSC Build start
-TSC ⚡️ Build success in 7924ms
+TSC ⚡️ Build success in 6673ms
 DTS Build start
 CLI Target: es2022
 Analysis will use the bundled TypeScript version 5.7.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.7.3
 Writing package typings: /home/runner/work/mastra/mastra/stores/chroma/dist/_tsup-dts-rollup.d.cts
-DTS ⚡️ Build success in 9330ms
+DTS ⚡️ Build success in 8364ms
 CLI Cleaning output folder
 ESM Build start
 CJS Build start
-ESM dist/index.js 7.28 KB
-ESM ⚡️ Build success in 592ms
-CJS dist/index.cjs 7.31 KB
-CJS ⚡️ Build success in 592ms
+ESM dist/index.js 8.05 KB
+ESM ⚡️ Build success in 705ms
+CJS dist/index.cjs 8.08 KB
+CJS ⚡️ Build success in 707ms

package/CHANGELOG.md

@@ -1,5 +1,38 @@
 # @mastra/chroma
 
+## 0.2.0-alpha.6
+
+### Patch Changes
+
+- Updated dependencies [f6678e4]
+  - @mastra/core@0.5.0-alpha.6
+
+## 0.2.0-alpha.5
+
+### Minor Changes
+
+- 8bee68a: Added implementations to support new operations for MastraVector interface for chroma store
+
+### Patch Changes
+
+- 9e81f35: Fix query filter for vector search and rerank
+- Updated dependencies [22643eb]
+- Updated dependencies [6feb23f]
+- Updated dependencies [f2d6727]
+- Updated dependencies [301e4ee]
+- Updated dependencies [dfbe4e9]
+- Updated dependencies [9e81f35]
+- Updated dependencies [caefaa2]
+- Updated dependencies [c151ae6]
+- Updated dependencies [52e0418]
+- Updated dependencies [03236ec]
+- Updated dependencies [3764e71]
+- Updated dependencies [df982db]
+- Updated dependencies [0461849]
+- Updated dependencies [2259379]
+- Updated dependencies [358f069]
+  - @mastra/core@0.5.0-alpha.5
+
 ## 0.1.8-alpha.4
 
 ### Patch Changes

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

@@ -55,6 +55,11 @@ declare class ChromaVector extends MastraVector {
     listIndexes(): Promise<string[]>;
     describeIndex(indexName: string): Promise<IndexStats>;
     deleteIndex(indexName: string): Promise<void>;
+    updateIndexById(indexName: string, id: string, update: {
+        vector?: number[];
+        metadata?: Record<string, any>;
+    }): Promise<void>;
+    deleteIndexById(indexName: string, id: string): Promise<void>;
 }
 export { ChromaVector }
 export { ChromaVector as ChromaVector_alias_1 }

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

@@ -55,6 +55,11 @@ declare class ChromaVector extends MastraVector {
     listIndexes(): Promise<string[]>;
     describeIndex(indexName: string): Promise<IndexStats>;
     deleteIndex(indexName: string): Promise<void>;
+    updateIndexById(indexName: string, id: string, update: {
+        vector?: number[];
+        metadata?: Record<string, any>;
+    }): Promise<void>;
+    deleteIndexById(indexName: string, id: string): Promise<void>;
 }
 export { ChromaVector }
 export { ChromaVector as ChromaVector_alias_1 }

package/dist/index.cjs

@@ -209,6 +209,28 @@ var ChromaVector = class extends vector.MastraVector {
     await this.client.deleteCollection({ name: indexName });
     this.collections.delete(indexName);
   }
+  async updateIndexById(indexName, id, update) {
+    if (!update.vector && !update.metadata) {
+      throw new Error("No updates provided");
+    }
+    const collection = await this.getCollection(indexName, true);
+    const updateOptions = { ids: [id] };
+    if (update?.vector) {
+      updateOptions.embeddings = [update.vector];
+    }
+    if (update?.metadata) {
+      updateOptions.metadatas = [update.metadata];
+    }
+    return await collection.update(updateOptions);
+  }
+  async deleteIndexById(indexName, id) {
+    try {
+      const collection = await this.getCollection(indexName, true);
+      await collection.delete({ ids: [id] });
+    } catch (error) {
+      throw new Error(`Failed to delete index by id: ${id} for index name: ${indexName}: ${error.message}`);
+    }
+  }
 };
 
 exports.ChromaVector = ChromaVector;

package/dist/index.js

@@ -207,6 +207,28 @@ var ChromaVector = class extends MastraVector {
     await this.client.deleteCollection({ name: indexName });
     this.collections.delete(indexName);
   }
+  async updateIndexById(indexName, id, update) {
+    if (!update.vector && !update.metadata) {
+      throw new Error("No updates provided");
+    }
+    const collection = await this.getCollection(indexName, true);
+    const updateOptions = { ids: [id] };
+    if (update?.vector) {
+      updateOptions.embeddings = [update.vector];
+    }
+    if (update?.metadata) {
+      updateOptions.metadatas = [update.metadata];
+    }
+    return await collection.update(updateOptions);
+  }
+  async deleteIndexById(indexName, id) {
+    try {
+      const collection = await this.getCollection(indexName, true);
+      await collection.delete({ ids: [id] });
+    } catch (error) {
+      throw new Error(`Failed to delete index by id: ${id} for index name: ${indexName}: ${error.message}`);
+    }
+  }
 };
 
 export { ChromaVector };

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@mastra/chroma",
-  "version": "0.1.8-alpha.4",
+  "version": "0.2.0-alpha.6",
   "description": "Chroma vector store provider for Mastra",
   "type": "module",
   "main": "dist/index.js",
@@ -20,7 +20,7 @@
   },
   "dependencies": {
     "chromadb": "^1.9.4",
-    "@mastra/core": "^0.5.0-alpha.4"
+    "@mastra/core": "^0.5.0-alpha.6"
   },
   "devDependencies": {
     "@microsoft/api-extractor": "^7.49.2",

package/src/vector/index.test.ts

@@ -2,6 +2,7 @@ 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({
@@ -117,6 +118,104 @@ describe('ChromaVector Integration Tests', () => {
       expect(results?.[0]?.id).toBe(testIds[0]);
       expect(results?.[0]?.metadata).toEqual(updatedMetadata[0]);
     });
+
+    it('should update the vector by id', async () => {
+      const ids = await vectorDB.upsert({ indexName: testIndexName, 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 vectorDB.updateIndexById(testIndexName, idToBeUpdated, update);
+
+      const results: QueryResult[] = await vectorDB.query({
+        indexName: testIndexName,
+        queryVector: newVector,
+        topK: 2,
+        includeVector: true,
+      });
+      expect(results[0]?.id).toBe(idToBeUpdated);
+      expect(results[0]?.vector).toEqual(newVector);
+      expect(results[0]?.metadata).toEqual(newMetaData);
+    });
+
+    it('should only update the metadata by id', async () => {
+      const ids = await vectorDB.upsert({ indexName: testIndexName, vectors: testVectors });
+      expect(ids).toHaveLength(3);
+
+      const idToBeUpdated = ids[0];
+      const newMetaData = {
+        test: 'updates',
+      };
+
+      const update = {
+        metadata: newMetaData,
+      };
+
+      await vectorDB.updateIndexById(testIndexName, idToBeUpdated, update);
+
+      const results: QueryResult[] = await vectorDB.query({
+        indexName: testIndexName,
+        queryVector: testVectors[0],
+        topK: 2,
+        includeVector: true,
+      });
+      expect(results[0]?.id).toBe(idToBeUpdated);
+      expect(results[0]?.vector).toEqual(testVectors[0]);
+      expect(results[0]?.metadata).toEqual(newMetaData);
+    });
+
+    it('should only update vector embeddings by id', async () => {
+      const ids = await vectorDB.upsert({ indexName: testIndexName, vectors: testVectors });
+      expect(ids).toHaveLength(3);
+
+      const idToBeUpdated = ids[0];
+      const newVector = [1, 2, 3];
+
+      const update = {
+        vector: newVector,
+      };
+
+      await vectorDB.updateIndexById(testIndexName, idToBeUpdated, update);
+
+      const results: QueryResult[] = await vectorDB.query({
+        indexName: testIndexName,
+        queryVector: newVector,
+        topK: 2,
+        includeVector: true,
+      });
+      expect(results[0]?.id).toBe(idToBeUpdated);
+      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 delete the vector by id', async () => {
+      const ids = await vectorDB.upsert({ indexName: testIndexName, vectors: testVectors });
+      expect(ids).toHaveLength(3);
+      const idToBeDeleted = ids[0];
+
+      await vectorDB.deleteIndexById(testIndexName, idToBeDeleted);
+
+      const results: QueryResult[] = await vectorDB.query({
+        indexName: testIndexName,
+        queryVector: [1.0, 0.0, 0.0],
+        topK: 2,
+      });
+
+      expect(results).toHaveLength(2);
+      expect(results.map(res => res.id)).not.toContain(idToBeDeleted);
+    });
   });
 
   describe('Query Operations', () => {
@@ -932,7 +1031,6 @@ describe('ChromaVector Integration Tests', () => {
         const results2 = await vectorDB.query({
           indexName: testIndexName2,
           queryVector: [1, 0, 0],
-          filter: undefined,
         });
         expect(results1).toEqual(results2);
         expect(results1.length).toBeGreaterThan(0);
@@ -947,7 +1045,6 @@ describe('ChromaVector Integration Tests', () => {
         const results2 = await vectorDB.query({
           indexName: testIndexName2,
           queryVector: [1, 0, 0],
-          filter: undefined,
         });
         expect(results).toEqual(results2);
         expect(results.length).toBeGreaterThan(0);
@@ -1270,6 +1367,43 @@ describe('ChromaVector Integration Tests', () => {
           }
         });
       });
+      it('should handle undefined document filter', async () => {
+        const results1 = await vectorDB.query({
+          indexName: testIndexName3,
+          queryVector: [1, 0, 0],
+          documentFilter: undefined,
+        });
+        const results2 = await vectorDB.query({
+          indexName: testIndexName3,
+          queryVector: [1, 0, 0],
+        });
+        expect(results1).toEqual(results2);
+        expect(results1.length).toBeGreaterThan(0);
+      });
+
+      it('should handle empty object document filter', async () => {
+        await expect(
+          vectorDB.query({
+            indexName: testIndexName3,
+            queryVector: [1, 0, 0],
+            documentFilter: {},
+          }),
+        ).rejects.toThrow();
+      });
+
+      it('should handle null filter', async () => {
+        const results = await vectorDB.query({
+          indexName: testIndexName3,
+          queryVector: [1, 0, 0],
+          documentFilter: null,
+        });
+        const results2 = await vectorDB.query({
+          indexName: testIndexName3,
+          queryVector: [1, 0, 0],
+        });
+        expect(results).toEqual(results2);
+        expect(results.length).toBeGreaterThan(0);
+      });
     });
   });
   describe('Deprecation Warnings', () => {

package/src/vector/index.ts

@@ -9,11 +9,13 @@ import type {
   QueryVectorArgs,
   UpsertVectorArgs,
 } from '@mastra/core/vector';
-import type { VectorFilter } from '@mastra/core/vector/filter';
-import { ChromaClient } from 'chromadb';
 
+import { ChromaClient, Collection } from 'chromadb';
 import { ChromaFilterTranslator } from './filter';
 
+import type { UpdateRecordsParams } from 'chromadb';
+import type { VectorFilter } from '@mastra/core/vector/filter';
+
 interface ChromaUpsertVectorParams extends UpsertVectorParams {
   documents?: string[];
 }
@@ -184,4 +186,37 @@ export class ChromaVector extends MastraVector {
     await this.client.deleteCollection({ name: indexName });
     this.collections.delete(indexName);
   }
+
+  async updateIndexById(
+    indexName: string,
+    id: string,
+    update: { vector?: number[]; metadata?: Record<string, any> },
+  ): Promise<void> {
+    if (!update.vector && !update.metadata) {
+      throw new Error('No updates provided');
+    }
+
+    const collection: Collection = await this.getCollection(indexName, true);
+
+    const updateOptions: UpdateRecordsParams = { ids: [id] };
+
+    if (update?.vector) {
+      updateOptions.embeddings = [update.vector];
+    }
+
+    if (update?.metadata) {
+      updateOptions.metadatas = [update.metadata];
+    }
+
+    return await collection.update(updateOptions);
+  }
+
+  async deleteIndexById(indexName: string, id: string): Promise<void> {
+    try {
+      const collection: Collection = await this.getCollection(indexName, true);
+      await collection.delete({ ids: [id] });
+    } catch (error: any) {
+      throw new Error(`Failed to delete index by id: ${id} for index name: ${indexName}: ${error.message}`);
+    }
+  }
 }