@mastra/astra 0.1.8-alpha.3 → 0.2.0-alpha.5

package/.turbo/turbo-build.log

@@ -1,23 +1,23 @@
 
-> @mastra/astra@0.1.8-alpha.3 build /home/runner/work/mastra/mastra/stores/astra
+> @mastra/astra@0.2.0-alpha.5 build /home/runner/work/mastra/mastra/stores/astra
 > 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 6291ms
+TSC ⚡️ Build success in 6994ms
 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/astra/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/astra/dist/_tsup-dts-rollup.d.cts
-DTS ⚡️ Build success in 10897ms
+DTS ⚡️ Build success in 9048ms
 CLI Cleaning output folder
 ESM Build start
 CJS Build start
-ESM dist/index.js 6.39 KB
-ESM ⚡️ Build success in 502ms
-CJS dist/index.cjs 6.43 KB
-CJS ⚡️ Build success in 503ms
+CJS dist/index.cjs 7.16 KB
+CJS ⚡️ Build success in 555ms
+ESM dist/index.js 7.13 KB
+ESM ⚡️ Build success in 556ms

package/CHANGELOG.md

@@ -1,5 +1,37 @@
 # @mastra/astra
 
+## 0.2.0-alpha.5
+
+### Minor Changes
+
+- 5bfb4b6: Added new operations implementations for new MastraVector interface methods in astra store
+
+### Patch Changes
+
+- 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
+
+- Updated dependencies [d79aedf]
+  - @mastra/core@0.5.0-alpha.4
+
 ## 0.1.8-alpha.3
 
 ### Patch Changes

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

@@ -77,6 +77,11 @@ declare class AstraVector extends MastraVector {
      * @returns {Promise<void>} A promise that resolves when the collection is deleted.
      */
     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 { AstraVector }
 export { AstraVector as AstraVector_alias_1 }

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

@@ -77,6 +77,11 @@ declare class AstraVector extends MastraVector {
      * @returns {Promise<void>} A promise that resolves when the collection is deleted.
      */
     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 { AstraVector }
 export { AstraVector as AstraVector_alias_1 }

package/dist/index.cjs

@@ -171,6 +171,28 @@ var AstraVector = class extends vector.MastraVector {
     const collection = this.#db.collection(indexName);
     await collection.drop();
   }
+  async updateIndexById(indexName, id, update) {
+    if (!update.vector && !update.metadata) {
+      throw new Error("No updates provided");
+    }
+    const collection = this.#db.collection(indexName);
+    const updateDoc = {};
+    if (update.vector) {
+      updateDoc.$vector = update.vector;
+    }
+    if (update.metadata) {
+      updateDoc.metadata = update.metadata;
+    }
+    await collection.findOneAndUpdate({ _id: id }, { $set: updateDoc });
+  }
+  async deleteIndexById(indexName, id) {
+    try {
+      const collection = this.#db.collection(indexName);
+      await collection.deleteOne({ _id: id });
+    } catch (error) {
+      throw new Error(`Failed to delete index by id: ${id} for index name: ${indexName}: ${error.message}`);
+    }
+  }
 };
 
 exports.AstraVector = AstraVector;

package/dist/index.js

@@ -169,6 +169,28 @@ var AstraVector = class extends MastraVector {
     const collection = this.#db.collection(indexName);
     await collection.drop();
   }
+  async updateIndexById(indexName, id, update) {
+    if (!update.vector && !update.metadata) {
+      throw new Error("No updates provided");
+    }
+    const collection = this.#db.collection(indexName);
+    const updateDoc = {};
+    if (update.vector) {
+      updateDoc.$vector = update.vector;
+    }
+    if (update.metadata) {
+      updateDoc.metadata = update.metadata;
+    }
+    await collection.findOneAndUpdate({ _id: id }, { $set: updateDoc });
+  }
+  async deleteIndexById(indexName, id) {
+    try {
+      const collection = this.#db.collection(indexName);
+      await collection.deleteOne({ _id: id });
+    } catch (error) {
+      throw new Error(`Failed to delete index by id: ${id} for index name: ${indexName}: ${error.message}`);
+    }
+  }
 };
 
 export { AstraVector };

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@mastra/astra",
-  "version": "0.1.8-alpha.3",
+  "version": "0.2.0-alpha.5",
   "description": "Astra DB provider for Mastra - includes vector store capabilities",
   "type": "module",
   "main": "dist/index.js",
@@ -20,7 +20,7 @@
   },
   "dependencies": {
     "@datastax/astra-db-ts": "^1.5.0",
-    "@mastra/core": "^0.5.0-alpha.3"
+    "@mastra/core": "^0.5.0-alpha.5"
   },
   "devDependencies": {
     "@microsoft/api-extractor": "^7.49.2",

package/src/vector/index.test.ts

@@ -1153,4 +1153,115 @@ describe('AstraVector Integration Tests', () => {
       expect(upsertResults).toHaveLength(1);
     });
   });
+
+  describe('Basic vector operations', () => {
+    const testVectors = [
+      [1, 0, 0, 0],
+      [0, 1, 0, 0],
+      [0, 0, 1, 0],
+      [0, 0, 0, 1],
+    ];
+
+    it('should update the vector by id', async () => {
+      const ids = await vectorDB.upsert({ indexName: testIndexName, vectors: testVectors });
+      expect(ids).toHaveLength(4);
+
+      const idToBeUpdated = ids[0];
+      const newVector = [1, 2, 3, 4];
+      const newMetaData = {
+        test: 'updates',
+      };
+
+      const update = {
+        vector: newVector,
+        metadata: newMetaData,
+      };
+
+      await vectorDB.updateIndexById(testIndexName, idToBeUpdated, update);
+
+      const results = await vectorDB.query({
+        indexName: testIndexName,
+        queryVector: newVector,
+        topK: 2,
+        includeVector: true,
+      });
+
+      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(4);
+      await new Promise(resolve => setTimeout(resolve, 2000));
+
+      const idToBeUpdated = ids[0];
+      const newMetaData = {
+        test: 'updates',
+      };
+
+      const update = {
+        metadata: newMetaData,
+      };
+
+      await vectorDB.updateIndexById(testIndexName, idToBeUpdated, update);
+      await new Promise(resolve => setTimeout(resolve, 2000));
+
+      const results = await vectorDB.query({
+        indexName: testIndexName,
+        queryVector: testVectors[0],
+        topK: 2,
+        includeVector: true,
+      });
+
+      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(4);
+      await new Promise(resolve => setTimeout(resolve, 2000));
+
+      const idToBeUpdated = ids[0];
+      const newVector = [1, 2, 3, 4];
+
+      const update = {
+        vector: newVector,
+      };
+
+      await vectorDB.updateIndexById(testIndexName, idToBeUpdated, update);
+      await new Promise(resolve => setTimeout(resolve, 2000));
+
+      const results = await vectorDB.query({
+        indexName: testIndexName,
+        queryVector: newVector,
+        topK: 2,
+        includeVector: true,
+      });
+
+      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(4);
+
+      const idToBeDeleted = ids[0];
+      await vectorDB.deleteIndexById(testIndexName, idToBeDeleted);
+
+      const results = await vectorDB.query({
+        indexName: testIndexName,
+        queryVector: [1, 0, 0, 0],
+        topK: 2,
+      });
+
+      expect(results).toHaveLength(2);
+      expect(results.map(res => res.id)).not.toContain(idToBeDeleted);
+    });
+  });
 });

package/src/vector/index.ts

@@ -168,4 +168,36 @@ export class AstraVector extends MastraVector {
     const collection = this.#db.collection(indexName);
     await collection.drop();
   }
+
+  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 = this.#db.collection(indexName);
+    const updateDoc: Record<string, any> = {};
+
+    if (update.vector) {
+      updateDoc.$vector = update.vector;
+    }
+
+    if (update.metadata) {
+      updateDoc.metadata = update.metadata;
+    }
+
+    await collection.findOneAndUpdate({ _id: id }, { $set: updateDoc });
+  }
+
+  async deleteIndexById(indexName: string, id: string): Promise<void> {
+    try {
+      const collection = this.#db.collection(indexName);
+      await collection.deleteOne({ _id: id });
+    } catch (error: any) {
+      throw new Error(`Failed to delete index by id: ${id} for index name: ${indexName}: ${error.message}`);
+    }
+  }
 }