@mastra/astra 0.2.5 → 0.2.6-alpha.0

package/.turbo/turbo-build.log

@@ -1,12 +1,12 @@
 
-> @mastra/astra@0.2.5-alpha.1 build /home/runner/work/mastra/mastra/stores/astra
+> @mastra/astra@0.2.6-alpha.0 build /home/runner/work/mastra/mastra/stores/astra
 > 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 7826ms
+TSC ⚡️ Build success in 5677ms
 DTS Build start
 CLI Target: es2022
 CLI Cleaning output folder
@@ -16,8 +16,8 @@ Analysis will use the bundled TypeScript version 5.8.2
 Writing package typings: /home/runner/work/mastra/mastra/stores/astra/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/astra/dist/_tsup-dts-rollup.d.cts
-DTS ⚡️ Build success in 9392ms
-ESM dist/index.js 7.13 KB
-ESM ⚡️ Build success in 9751ms
-CJS dist/index.cjs 7.16 KB
-CJS ⚡️ Build success in 9742ms
+DTS ⚡️ Build success in 7633ms
+CJS dist/index.cjs 7.09 KB
+CJS ⚡️ Build success in 8059ms
+ESM dist/index.js 7.06 KB
+ESM ⚡️ Build success in 8059ms

package/CHANGELOG.md

@@ -1,5 +1,13 @@
 # @mastra/astra
 
+## 0.2.6-alpha.0
+
+### Patch Changes
+
+- 15702ff: Updated upsert, updateIndexById, deleteIndexById for astra and updated tests for astra and vectorize
+- Updated dependencies [b4fbc59]
+  - @mastra/core@0.6.5-alpha.0
+
 ## 0.2.5
 
 ### Patch Changes

package/dist/index.cjs

@@ -101,8 +101,8 @@ var AstraVector = class extends vector.MastraVector {
       $vector: vector,
       metadata: metadata?.[i] || {}
     }));
-    const result = await collection.insertMany(records);
-    return result.insertedIds.map((id) => (id || "").toString());
+    await collection.insertMany(records);
+    return vectorIds;
   }
   transformFilter(filter) {
     const translator = new AstraFilterTranslator();
@@ -183,12 +183,12 @@ var AstraVector = class extends vector.MastraVector {
     if (update.metadata) {
       updateDoc.metadata = update.metadata;
     }
-    await collection.findOneAndUpdate({ _id: id }, { $set: updateDoc });
+    await collection.findOneAndUpdate({ id }, { $set: updateDoc });
   }
   async deleteIndexById(indexName, id) {
     try {
       const collection = this.#db.collection(indexName);
-      await collection.deleteOne({ _id: id });
+      await collection.deleteOne({ id });
     } catch (error) {
       throw new Error(`Failed to delete index by id: ${id} for index name: ${indexName}: ${error.message}`);
     }

package/dist/index.js

@@ -99,8 +99,8 @@ var AstraVector = class extends MastraVector {
       $vector: vector,
       metadata: metadata?.[i] || {}
     }));
-    const result = await collection.insertMany(records);
-    return result.insertedIds.map((id) => (id || "").toString());
+    await collection.insertMany(records);
+    return vectorIds;
   }
   transformFilter(filter) {
     const translator = new AstraFilterTranslator();
@@ -181,12 +181,12 @@ var AstraVector = class extends MastraVector {
     if (update.metadata) {
       updateDoc.metadata = update.metadata;
     }
-    await collection.findOneAndUpdate({ _id: id }, { $set: updateDoc });
+    await collection.findOneAndUpdate({ id }, { $set: updateDoc });
   }
   async deleteIndexById(indexName, id) {
     try {
       const collection = this.#db.collection(indexName);
-      await collection.deleteOne({ _id: id });
+      await collection.deleteOne({ id });
     } catch (error) {
       throw new Error(`Failed to delete index by id: ${id} for index name: ${indexName}: ${error.message}`);
     }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@mastra/astra",
-  "version": "0.2.5",
+  "version": "0.2.6-alpha.0",
   "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.6.4"
+    "@mastra/core": "^0.6.5-alpha.0"
   },
   "devDependencies": {
     "@microsoft/api-extractor": "^7.52.1",

package/src/vector/index.test.ts

@@ -28,8 +28,6 @@ describe('AstraVector Integration Tests', () => {
   const testIndexName = 'testvectors1733728136118'; // Unique collection name
   const testIndexName2 = 'testvectors1733728136119'; // Unique collection name
 
-  console.log('testIndexName:', testIndexName);
-
   beforeAll(async () => {
     // Ensure required environment variables are set
     const token = process.env.ASTRA_DB_TOKEN;
@@ -1155,6 +1153,16 @@ describe('AstraVector Integration Tests', () => {
   });
 
   describe('Basic vector operations', () => {
+    const indexName = 'testbasicvectoroperations';
+
+    beforeAll(async () => {
+      await vectorDB.createIndex({ indexName, dimension: 4 });
+    });
+
+    afterAll(async () => {
+      await vectorDB.deleteIndex(indexName);
+    });
+
     const testVectors = [
       [1, 0, 0, 0],
       [0, 1, 0, 0],
@@ -1163,7 +1171,7 @@ describe('AstraVector Integration Tests', () => {
     ];
 
     it('should update the vector by id', async () => {
-      const ids = await vectorDB.upsert({ indexName: testIndexName, vectors: testVectors });
+      const ids = await vectorDB.upsert({ indexName, vectors: testVectors });
       expect(ids).toHaveLength(4);
 
       const idToBeUpdated = ids[0];
@@ -1177,23 +1185,28 @@ describe('AstraVector Integration Tests', () => {
         metadata: newMetaData,
       };
 
-      await vectorDB.updateIndexById(testIndexName, idToBeUpdated, update);
+      await vectorDB.updateIndexById(indexName, idToBeUpdated, update);
+
+      await new Promise(resolve => setTimeout(resolve, 2000));
 
       const results = await vectorDB.query({
-        indexName: testIndexName,
+        indexName: indexName,
         queryVector: newVector,
         topK: 2,
         includeVector: true,
       });
 
-      expect(results[0]?.vector).toEqual(newVector);
-      expect(results[0]?.metadata).toEqual(newMetaData);
+      expect(results).toHaveLength(2);
+      const updatedResult = results.find(result => result.id === idToBeUpdated);
+      expect(updatedResult).toBeDefined();
+      expect(updatedResult?.id).toEqual(idToBeUpdated);
+      expect(updatedResult?.vector).toEqual(newVector);
+      expect(updatedResult?.metadata).toEqual(newMetaData);
     });
 
     it('should only update the metadata by id', async () => {
-      const ids = await vectorDB.upsert({ indexName: testIndexName, vectors: testVectors });
+      const ids = await vectorDB.upsert({ indexName, vectors: testVectors });
       expect(ids).toHaveLength(4);
-      await new Promise(resolve => setTimeout(resolve, 2000));
 
       const idToBeUpdated = ids[0];
       const newMetaData = {
@@ -1204,24 +1217,27 @@ describe('AstraVector Integration Tests', () => {
         metadata: newMetaData,
       };
 
-      await vectorDB.updateIndexById(testIndexName, idToBeUpdated, update);
+      await vectorDB.updateIndexById(indexName, idToBeUpdated, update);
       await new Promise(resolve => setTimeout(resolve, 2000));
 
       const results = await vectorDB.query({
-        indexName: testIndexName,
+        indexName: indexName,
         queryVector: testVectors[0],
         topK: 2,
         includeVector: true,
       });
 
-      expect(results[0]?.vector).toEqual(testVectors[0]);
-      expect(results[0]?.metadata).toEqual(newMetaData);
+      expect(results).toHaveLength(2);
+      const updatedResult = results.find(result => result.id === idToBeUpdated);
+      expect(updatedResult).toBeDefined();
+      expect(updatedResult?.id).toEqual(idToBeUpdated);
+      expect(updatedResult?.vector).toEqual(testVectors[0]);
+      expect(updatedResult?.metadata).toEqual(newMetaData);
     });
 
     it('should only update vector embeddings by id', async () => {
-      const ids = await vectorDB.upsert({ indexName: testIndexName, vectors: testVectors });
+      const ids = await vectorDB.upsert({ indexName, vectors: testVectors });
       expect(ids).toHaveLength(4);
-      await new Promise(resolve => setTimeout(resolve, 2000));
 
       const idToBeUpdated = ids[0];
       const newVector = [1, 2, 3, 4];
@@ -1230,32 +1246,36 @@ describe('AstraVector Integration Tests', () => {
         vector: newVector,
       };
 
-      await vectorDB.updateIndexById(testIndexName, idToBeUpdated, update);
+      await vectorDB.updateIndexById(indexName, idToBeUpdated, update);
       await new Promise(resolve => setTimeout(resolve, 2000));
 
       const results = await vectorDB.query({
-        indexName: testIndexName,
+        indexName: indexName,
         queryVector: newVector,
         topK: 2,
         includeVector: true,
       });
 
-      expect(results[0]?.vector).toEqual(newVector);
+      expect(results).toHaveLength(2);
+      const updatedResult = results.find(result => result.id === idToBeUpdated);
+      expect(updatedResult).toBeDefined();
+      expect(updatedResult?.id).toEqual(idToBeUpdated);
+      expect(updatedResult?.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(indexName, 'id', {})).rejects.toThrow('No updates provided');
     });
 
     it('should delete the vector by id', async () => {
-      const ids = await vectorDB.upsert({ indexName: testIndexName, vectors: testVectors });
+      const ids = await vectorDB.upsert({ indexName, vectors: testVectors });
       expect(ids).toHaveLength(4);
 
       const idToBeDeleted = ids[0];
-      await vectorDB.deleteIndexById(testIndexName, idToBeDeleted);
+      await vectorDB.deleteIndexById(indexName, idToBeDeleted);
 
       const results = await vectorDB.query({
-        indexName: testIndexName,
+        indexName: indexName,
         queryVector: [1, 0, 0, 0],
         topK: 2,
       });

package/src/vector/index.ts

@@ -85,8 +85,8 @@ export class AstraVector extends MastraVector {
       metadata: metadata?.[i] || {},
     }));
 
-    const result = await collection.insertMany(records);
-    return result.insertedIds.map(id => (id || '').toString());
+    await collection.insertMany(records);
+    return vectorIds;
   }
 
   transformFilter(filter?: VectorFilter) {
@@ -189,13 +189,13 @@ export class AstraVector extends MastraVector {
       updateDoc.metadata = update.metadata;
     }
 
-    await collection.findOneAndUpdate({ _id: id }, { $set: updateDoc });
+    await collection.findOneAndUpdate({ id }, { $set: updateDoc });
   }
 
   async deleteIndexById(indexName: string, id: string): Promise<void> {
     try {
       const collection = this.#db.collection(indexName);
-      await collection.deleteOne({ _id: id });
+      await collection.deleteOne({ id });
     } catch (error: any) {
       throw new Error(`Failed to delete index by id: ${id} for index name: ${indexName}: ${error.message}`);
     }