@mastra/astra 0.2.6 → 0.2.7-alpha.2

package/.turbo/turbo-build.log

@@ -1,23 +1,23 @@
 
-> @mastra/astra@0.2.6-alpha.3 build /home/runner/work/mastra/mastra/stores/astra
+> @mastra/astra@0.2.7-alpha.2 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 6577ms
+TSC ⚡️ Build success in 5799ms
 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/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 8257ms
+DTS ⚡️ Build success in 9416ms
 CLI Cleaning output folder
 ESM Build start
 CJS Build start
-ESM dist/index.js 7.06 KB
-ESM ⚡️ Build success in 571ms
 CJS dist/index.cjs 7.09 KB
-CJS ⚡️ Build success in 571ms
+CJS ⚡️ Build success in 461ms
+ESM dist/index.js 7.06 KB
+ESM ⚡️ Build success in 462ms

package/CHANGELOG.md

@@ -1,5 +1,39 @@
 # @mastra/astra
 
+## 0.2.7-alpha.2
+
+### Patch Changes
+
+- Updated dependencies [56c31b7]
+- Updated dependencies [dbbbf80]
+- Updated dependencies [99d43b9]
+  - @mastra/core@0.8.0-alpha.2
+
+## 0.2.7-alpha.1
+
+### Patch Changes
+
+- 866da3c: Update astra tests to check collection is empty
+- Updated dependencies [619c39d]
+- Updated dependencies [fe56be0]
+- Updated dependencies [a0967a0]
+- Updated dependencies [fca3b21]
+- Updated dependencies [0118361]
+- Updated dependencies [619c39d]
+  - @mastra/core@0.8.0-alpha.1
+
+## 0.2.7-alpha.0
+
+### Patch Changes
+
+- Updated dependencies [107bcfe]
+- Updated dependencies [5b4e19f]
+- Updated dependencies [7599d77]
+- Updated dependencies [cafae83]
+- Updated dependencies [8076ecf]
+- Updated dependencies [304397c]
+  - @mastra/core@0.7.1-alpha.0
+
 ## 0.2.6
 
 ### Patch Changes

package/package.json

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

package/src/vector/index.test.ts

@@ -23,6 +23,41 @@ async function waitForCondition(
   return false;
 }
 
+async function createIndexAndWait(
+  vectorDB: AstraVector,
+  indexName: string,
+  dimension: number,
+  metric: 'cosine' | 'euclidean' | 'dotproduct',
+) {
+  await vectorDB.createIndex({ indexName, dimension, metric });
+  const created = await waitForCondition(
+    async () => {
+      const newCollections = await vectorDB.listIndexes();
+      return newCollections.includes(indexName);
+    },
+    30000,
+    2000,
+  );
+  if (!created) {
+    throw new Error('Timed out waiting for collection to be created');
+  }
+}
+
+async function deleteIndexAndWait(vectorDB: AstraVector, indexName: string) {
+  await vectorDB.deleteIndex(indexName);
+  const deleted = await waitForCondition(
+    async () => {
+      const newCollections = await vectorDB.listIndexes();
+      return !newCollections.includes(indexName);
+    },
+    30000,
+    2000,
+  );
+  if (!deleted) {
+    throw new Error('Timed out waiting for collection to be deleted');
+  }
+}
+
 describe('AstraVector Integration Tests', () => {
   let vectorDB: AstraVector;
   const testIndexName = 'testvectors1733728136118'; // Unique collection name
@@ -46,12 +81,24 @@ describe('AstraVector Integration Tests', () => {
     try {
       const collections = await vectorDB.listIndexes();
       await Promise.all(collections.map(c => vectorDB.deleteIndex(c)));
+      const deleted = await waitForCondition(
+        async () => {
+          const remainingCollections = await vectorDB.listIndexes();
+          return remainingCollections.length === 0;
+        },
+        30000,
+        2000,
+      );
+      if (!deleted) {
+        throw new Error('Timed out waiting for collections to be deleted');
+      }
     } catch (error) {
       console.error('Failed to delete test collections:', error);
+      throw error;
     }
 
-    await vectorDB.createIndex({ indexName: testIndexName, dimension: 4, metric: 'cosine' });
-    await vectorDB.createIndex({ indexName: testIndexName2, dimension: 4, metric: 'cosine' });
+    await createIndexAndWait(vectorDB, testIndexName, 4, 'cosine');
+    await createIndexAndWait(vectorDB, testIndexName2, 4, 'cosine');
   }, 500000);
 
   afterAll(async () => {
@@ -159,11 +206,7 @@ describe('AstraVector Integration Tests', () => {
 
     try {
       // Create index with higher dimensions
-      await vectorDB.createIndex({
-        indexName: highDimIndexName,
-        dimension: 1536,
-        metric: 'cosine',
-      });
+      await createIndexAndWait(vectorDB, highDimIndexName, 1536, 'cosine');
 
       // Insert high-dimensional vectors
       const vectors = [
@@ -202,7 +245,7 @@ describe('AstraVector Integration Tests', () => {
       expect(results?.[0]?.score).toBeCloseTo(1, 4);
     } finally {
       // Cleanup
-      await vectorDB.deleteIndex(highDimIndexName);
+      await deleteIndexAndWait(vectorDB, highDimIndexName);
     }
   });
 
@@ -214,11 +257,7 @@ describe('AstraVector Integration Tests', () => {
 
       try {
         // Create index with different metric
-        await vectorDB.createIndex({
-          indexName: metricIndexName,
-          dimension: 4,
-          metric,
-        });
+        await createIndexAndWait(vectorDB, metricIndexName, 4, metric);
 
         // Insert same vectors
         const vectors = [
@@ -247,7 +286,7 @@ describe('AstraVector Integration Tests', () => {
         expect(results?.[0]?.score).toBeGreaterThan(results?.[1]?.score!);
       } finally {
         // Cleanup
-        await vectorDB.deleteIndex(metricIndexName);
+        await deleteIndexAndWait(vectorDB, metricIndexName);
       }
     }
   }, 500000);
@@ -1062,12 +1101,12 @@ describe('AstraVector Integration Tests', () => {
     let warnSpy;
 
     beforeAll(async () => {
-      await vectorDB.createIndex({ indexName: indexName, dimension: 3 });
+      await createIndexAndWait(vectorDB, indexName, 3, 'cosine');
     });
 
     afterAll(async () => {
-      await vectorDB.deleteIndex(indexName);
-      await vectorDB.deleteIndex(indexName2);
+      await deleteIndexAndWait(vectorDB, indexName);
+      await deleteIndexAndWait(vectorDB, indexName2);
     });
 
     beforeEach(async () => {
@@ -1076,7 +1115,7 @@ describe('AstraVector Integration Tests', () => {
 
     afterEach(async () => {
       warnSpy.mockRestore();
-      await vectorDB.deleteIndex(indexName2);
+      await deleteIndexAndWait(vectorDB, indexName2);
     });
 
     it('should show deprecation warning when using individual args for createIndex', async () => {
@@ -1156,11 +1195,11 @@ describe('AstraVector Integration Tests', () => {
     const indexName = 'testbasicvectoroperations';
 
     beforeAll(async () => {
-      await vectorDB.createIndex({ indexName, dimension: 4 });
+      await createIndexAndWait(vectorDB, indexName, 4, 'cosine');
     });
 
     afterAll(async () => {
-      await vectorDB.deleteIndex(indexName);
+      await deleteIndexAndWait(vectorDB, indexName);
     });
 
     const testVectors = [