@mastra/astra 0.10.2 → 0.10.3-alpha.1

package/.turbo/turbo-build.log

@@ -1,23 +1,23 @@
 
-> @mastra/astra@0.10.2-alpha.0 build /home/runner/work/mastra/mastra/stores/astra
+> @mastra/astra@0.10.3-alpha.1 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.5.0
 TSC Build start
-TSC ⚡️ Build success in 6874ms
+TSC ⚡️ Build success in 5956ms
 DTS Build start
 CLI Target: es2022
 Analysis will use the bundled TypeScript version 5.8.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.8.3
 Writing package typings: /home/runner/work/mastra/mastra/stores/astra/dist/_tsup-dts-rollup.d.cts
-DTS ⚡️ Build success in 9957ms
+DTS ⚡️ Build success in 8264ms
 CLI Cleaning output folder
 ESM Build start
 CJS Build start
-CJS dist/index.cjs 11.63 KB
-CJS ⚡️ Build success in 733ms
-ESM dist/index.js 11.57 KB
-ESM ⚡️ Build success in 734ms
+CJS dist/index.cjs 14.42 KB
+CJS ⚡️ Build success in 539ms
+ESM dist/index.js 14.16 KB
+ESM ⚡️ Build success in 540ms

package/CHANGELOG.md

@@ -1,5 +1,29 @@
 # @mastra/astra
 
+## 0.10.3-alpha.1
+
+### Patch Changes
+
+- 144eb0b: [MASTRA-3669] Metadata Filter Types
+- Updated dependencies [15e9d26]
+- Updated dependencies [07d6d88]
+- Updated dependencies [5d74aab]
+- Updated dependencies [144eb0b]
+  - @mastra/core@0.10.7-alpha.2
+
+## 0.10.3-alpha.0
+
+### Patch Changes
+
+- 0e17048: Throw mastra errors in storage packages
+- Updated dependencies [d1baedb]
+- Updated dependencies [4d21bf2]
+- Updated dependencies [2097952]
+- Updated dependencies [4fb0cc2]
+- Updated dependencies [d2a7a31]
+- Updated dependencies [0e17048]
+  - @mastra/core@0.10.7-alpha.1
+
 ## 0.10.2
 
 ### Patch Changes

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

@@ -1,11 +1,14 @@
 import { BaseFilterTranslator } from '@mastra/core/vector/filter';
+import type { BlacklistedRootOperators } from '@mastra/core/vector/filter';
 import type { CreateIndexParams } from '@mastra/core/vector';
 import type { DeleteIndexParams } from '@mastra/core/vector';
 import type { DeleteVectorParams } from '@mastra/core/vector';
 import type { DescribeIndexParams } from '@mastra/core/vector';
 import type { IndexStats } from '@mastra/core/vector';
+import type { LogicalOperatorValueMap } from '@mastra/core/vector/filter';
 import { MastraVector } from '@mastra/core/vector';
 import type { OperatorSupport } from '@mastra/core/vector/filter';
+import type { OperatorValueMap } from '@mastra/core/vector/filter';
 import type { QueryResult } from '@mastra/core/vector';
 import type { QueryVectorParams } from '@mastra/core/vector';
 import type { UpdateVectorParams } from '@mastra/core/vector';
@@ -20,6 +23,8 @@ declare const ASTRA_PROMPT = "When querying Astra, you can ONLY use the operator
 export { ASTRA_PROMPT }
 export { ASTRA_PROMPT as ASTRA_PROMPT_alias_1 }
 
+declare type AstraBlacklisted = BlacklistedRootOperators | '$nor' | '$size';
+
 declare interface AstraDbOptions {
     token: string;
     endpoint: string;
@@ -33,14 +38,22 @@ export { AstraDbOptions as AstraDbOptions_alias_1 }
  * Maintains MongoDB-compatible syntax while ensuring proper validation
  * and normalization of values.
  */
-export declare class AstraFilterTranslator extends BaseFilterTranslator {
+export declare class AstraFilterTranslator extends BaseFilterTranslator<AstraVectorFilter> {
     protected getSupportedOperators(): OperatorSupport;
-    translate(filter?: VectorFilter): VectorFilter;
+    translate(filter?: AstraVectorFilter): AstraVectorFilter;
     private translateNode;
     private translateOperatorValue;
 }
 
-declare class AstraVector extends MastraVector {
+declare type AstraLogicalOperatorValueMap = Omit<LogicalOperatorValueMap, '$nor'>;
+
+declare type AstraOperatorValueMap = Omit<OperatorValueMap, '$elemMatch' | '$regex' | '$options'> & {
+    $size: number;
+};
+
+declare type AstraQueryVectorParams = QueryVectorParams<AstraVectorFilter>;
+
+declare class AstraVector extends MastraVector<AstraVectorFilter> {
     #private;
     constructor({ token, endpoint, keyspace }: AstraDbOptions);
     /**
@@ -62,7 +75,7 @@ declare class AstraVector extends MastraVector {
      * @returns {Promise<string[]>} A promise that resolves to an array of IDs of the upserted vectors.
      */
     upsert({ indexName, vectors, metadata, ids }: UpsertVectorParams): Promise<string[]>;
-    transformFilter(filter?: VectorFilter): VectorFilter;
+    transformFilter(filter?: AstraVectorFilter): AstraVectorFilter;
     /**
      * Queries the specified collection using a vector and optional filter.
      *
@@ -73,7 +86,7 @@ declare class AstraVector extends MastraVector {
      * @param {boolean} [includeVectors=false] - Whether to include the vectors in the response.
      * @returns {Promise<QueryResult[]>} A promise that resolves to an array of query results.
      */
-    query({ indexName, queryVector, topK, filter, includeVector, }: QueryVectorParams): Promise<QueryResult[]>;
+    query({ indexName, queryVector, topK, filter, includeVector, }: AstraQueryVectorParams): Promise<QueryResult[]>;
     /**
      * Lists all collections in the database.
      *
@@ -117,4 +130,6 @@ declare class AstraVector extends MastraVector {
 export { AstraVector }
 export { AstraVector as AstraVector_alias_1 }
 
+export declare type AstraVectorFilter = VectorFilter<keyof AstraOperatorValueMap, AstraOperatorValueMap, AstraLogicalOperatorValueMap, AstraBlacklisted>;
+
 export { }

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

@@ -1,11 +1,14 @@
 import { BaseFilterTranslator } from '@mastra/core/vector/filter';
+import type { BlacklistedRootOperators } from '@mastra/core/vector/filter';
 import type { CreateIndexParams } from '@mastra/core/vector';
 import type { DeleteIndexParams } from '@mastra/core/vector';
 import type { DeleteVectorParams } from '@mastra/core/vector';
 import type { DescribeIndexParams } from '@mastra/core/vector';
 import type { IndexStats } from '@mastra/core/vector';
+import type { LogicalOperatorValueMap } from '@mastra/core/vector/filter';
 import { MastraVector } from '@mastra/core/vector';
 import type { OperatorSupport } from '@mastra/core/vector/filter';
+import type { OperatorValueMap } from '@mastra/core/vector/filter';
 import type { QueryResult } from '@mastra/core/vector';
 import type { QueryVectorParams } from '@mastra/core/vector';
 import type { UpdateVectorParams } from '@mastra/core/vector';
@@ -20,6 +23,8 @@ declare const ASTRA_PROMPT = "When querying Astra, you can ONLY use the operator
 export { ASTRA_PROMPT }
 export { ASTRA_PROMPT as ASTRA_PROMPT_alias_1 }
 
+declare type AstraBlacklisted = BlacklistedRootOperators | '$nor' | '$size';
+
 declare interface AstraDbOptions {
     token: string;
     endpoint: string;
@@ -33,14 +38,22 @@ export { AstraDbOptions as AstraDbOptions_alias_1 }
  * Maintains MongoDB-compatible syntax while ensuring proper validation
  * and normalization of values.
  */
-export declare class AstraFilterTranslator extends BaseFilterTranslator {
+export declare class AstraFilterTranslator extends BaseFilterTranslator<AstraVectorFilter> {
     protected getSupportedOperators(): OperatorSupport;
-    translate(filter?: VectorFilter): VectorFilter;
+    translate(filter?: AstraVectorFilter): AstraVectorFilter;
     private translateNode;
     private translateOperatorValue;
 }
 
-declare class AstraVector extends MastraVector {
+declare type AstraLogicalOperatorValueMap = Omit<LogicalOperatorValueMap, '$nor'>;
+
+declare type AstraOperatorValueMap = Omit<OperatorValueMap, '$elemMatch' | '$regex' | '$options'> & {
+    $size: number;
+};
+
+declare type AstraQueryVectorParams = QueryVectorParams<AstraVectorFilter>;
+
+declare class AstraVector extends MastraVector<AstraVectorFilter> {
     #private;
     constructor({ token, endpoint, keyspace }: AstraDbOptions);
     /**
@@ -62,7 +75,7 @@ declare class AstraVector extends MastraVector {
      * @returns {Promise<string[]>} A promise that resolves to an array of IDs of the upserted vectors.
      */
     upsert({ indexName, vectors, metadata, ids }: UpsertVectorParams): Promise<string[]>;
-    transformFilter(filter?: VectorFilter): VectorFilter;
+    transformFilter(filter?: AstraVectorFilter): AstraVectorFilter;
     /**
      * Queries the specified collection using a vector and optional filter.
      *
@@ -73,7 +86,7 @@ declare class AstraVector extends MastraVector {
      * @param {boolean} [includeVectors=false] - Whether to include the vectors in the response.
      * @returns {Promise<QueryResult[]>} A promise that resolves to an array of query results.
      */
-    query({ indexName, queryVector, topK, filter, includeVector, }: QueryVectorParams): Promise<QueryResult[]>;
+    query({ indexName, queryVector, topK, filter, includeVector, }: AstraQueryVectorParams): Promise<QueryResult[]>;
     /**
      * Lists all collections in the database.
      *
@@ -117,4 +130,6 @@ declare class AstraVector extends MastraVector {
 export { AstraVector }
 export { AstraVector as AstraVector_alias_1 }
 
+export declare type AstraVectorFilter = VectorFilter<keyof AstraOperatorValueMap, AstraOperatorValueMap, AstraLogicalOperatorValueMap, AstraBlacklisted>;
+
 export { }

package/dist/index.cjs

@@ -1,6 +1,7 @@
 'use strict';
 
 var astraDbTs = require('@datastax/astra-db-ts');
+var error = require('@mastra/core/error');
 var vector = require('@mastra/core/vector');
 var filter = require('@mastra/core/vector/filter');
 
@@ -70,15 +71,32 @@ var AstraVector = class extends vector.MastraVector {
    */
   async createIndex({ indexName, dimension, metric = "cosine" }) {
     if (!Number.isInteger(dimension) || dimension <= 0) {
-      throw new Error("Dimension must be a positive integer");
+      throw new error.MastraError({
+        id: "ASTRA_VECTOR_CREATE_INDEX_INVALID_DIMENSION",
+        text: "Dimension must be a positive integer",
+        domain: error.ErrorDomain.MASTRA_VECTOR,
+        category: error.ErrorCategory.USER
+      });
+    }
+    try {
+      await this.#db.createCollection(indexName, {
+        vector: {
+          dimension,
+          metric: metricMap[metric]
+        },
+        checkExists: false
+      });
+    } catch (error$1) {
+      new error.MastraError(
+        {
+          id: "ASTRA_VECTOR_CREATE_INDEX_DB_ERROR",
+          domain: error.ErrorDomain.MASTRA_VECTOR,
+          category: error.ErrorCategory.THIRD_PARTY,
+          details: { indexName }
+        },
+        error$1
+      );
     }
-    await this.#db.createCollection(indexName, {
-      vector: {
-        dimension,
-        metric: metricMap[metric]
-      },
-      checkExists: false
-    });
   }
   /**
    * Inserts or updates vectors in the specified collection.
@@ -97,7 +115,19 @@ var AstraVector = class extends vector.MastraVector {
       $vector: vector,
       metadata: metadata?.[i] || {}
     }));
-    await collection.insertMany(records);
+    try {
+      await collection.insertMany(records);
+    } catch (error$1) {
+      throw new error.MastraError(
+        {
+          id: "ASTRA_VECTOR_UPSERT_DB_ERROR",
+          domain: error.ErrorDomain.MASTRA_VECTOR,
+          category: error.ErrorCategory.THIRD_PARTY,
+          details: { indexName }
+        },
+        error$1
+      );
+    }
     return vectorIds;
   }
   transformFilter(filter) {
@@ -123,29 +153,52 @@ var AstraVector = class extends vector.MastraVector {
   }) {
     const collection = this.#db.collection(indexName);
     const translatedFilter = this.transformFilter(filter);
-    const cursor = collection.find(translatedFilter ?? {}, {
-      sort: { $vector: queryVector },
-      limit: topK,
-      includeSimilarity: true,
-      projection: {
-        $vector: includeVector ? true : false
-      }
-    });
-    const results = await cursor.toArray();
-    return results.map((result) => ({
-      id: result.id,
-      score: result.$similarity,
-      metadata: result.metadata,
-      ...includeVector && { vector: result.$vector }
-    }));
+    try {
+      const cursor = collection.find(translatedFilter ?? {}, {
+        sort: { $vector: queryVector },
+        limit: topK,
+        includeSimilarity: true,
+        projection: {
+          $vector: includeVector ? true : false
+        }
+      });
+      const results = await cursor.toArray();
+      return results.map((result) => ({
+        id: result.id,
+        score: result.$similarity,
+        metadata: result.metadata,
+        ...includeVector && { vector: result.$vector }
+      }));
+    } catch (error$1) {
+      throw new error.MastraError(
+        {
+          id: "ASTRA_VECTOR_QUERY_DB_ERROR",
+          domain: error.ErrorDomain.MASTRA_VECTOR,
+          category: error.ErrorCategory.THIRD_PARTY,
+          details: { indexName }
+        },
+        error$1
+      );
+    }
   }
   /**
    * Lists all collections in the database.
    *
    * @returns {Promise<string[]>} A promise that resolves to an array of collection names.
    */
-  listIndexes() {
-    return this.#db.listCollections({ nameOnly: true });
+  async listIndexes() {
+    try {
+      return await this.#db.listCollections({ nameOnly: true });
+    } catch (error$1) {
+      throw new error.MastraError(
+        {
+          id: "ASTRA_VECTOR_LIST_INDEXES_DB_ERROR",
+          domain: error.ErrorDomain.MASTRA_VECTOR,
+          category: error.ErrorCategory.THIRD_PARTY
+        },
+        error$1
+      );
+    }
   }
   /**
    * Retrieves statistics about a vector index.
@@ -155,16 +208,29 @@ var AstraVector = class extends vector.MastraVector {
    */
   async describeIndex({ indexName }) {
     const collection = this.#db.collection(indexName);
-    const optionsPromise = collection.options();
-    const countPromise = collection.countDocuments({}, 100);
-    const [options, count] = await Promise.all([optionsPromise, countPromise]);
-    const keys = Object.keys(metricMap);
-    const metric = keys.find((key) => metricMap[key] === options.vector?.metric);
-    return {
-      dimension: options.vector?.dimension,
-      metric,
-      count
-    };
+    try {
+      const optionsPromise = collection.options();
+      const countPromise = collection.countDocuments({}, 100);
+      const [options, count] = await Promise.all([optionsPromise, countPromise]);
+      const keys = Object.keys(metricMap);
+      const metric = keys.find((key) => metricMap[key] === options.vector?.metric);
+      return {
+        dimension: options.vector?.dimension,
+        metric,
+        count
+      };
+    } catch (error$1) {
+      if (error$1 instanceof error.MastraError) throw error$1;
+      throw new error.MastraError(
+        {
+          id: "ASTRA_VECTOR_DESCRIBE_INDEX_DB_ERROR",
+          domain: error.ErrorDomain.MASTRA_VECTOR,
+          category: error.ErrorCategory.THIRD_PARTY,
+          details: { indexName }
+        },
+        error$1
+      );
+    }
   }
   /**
    * Deletes the specified collection.
@@ -174,7 +240,19 @@ var AstraVector = class extends vector.MastraVector {
    */
   async deleteIndex({ indexName }) {
     const collection = this.#db.collection(indexName);
-    await collection.drop();
+    try {
+      await collection.drop();
+    } catch (error$1) {
+      throw new error.MastraError(
+        {
+          id: "ASTRA_VECTOR_DELETE_INDEX_DB_ERROR",
+          domain: error.ErrorDomain.MASTRA_VECTOR,
+          category: error.ErrorCategory.THIRD_PARTY,
+          details: { indexName }
+        },
+        error$1
+      );
+    }
   }
   /**
    * Updates a vector by its ID with the provided vector and/or metadata.
@@ -187,10 +265,16 @@ var AstraVector = class extends vector.MastraVector {
    * @throws Will throw an error if no updates are provided or if the update operation fails.
    */
   async updateVector({ indexName, id, update }) {
+    if (!update.vector && !update.metadata) {
+      throw new error.MastraError({
+        id: "ASTRA_VECTOR_UPDATE_NO_PAYLOAD",
+        text: "No updates provided for vector",
+        domain: error.ErrorDomain.MASTRA_VECTOR,
+        category: error.ErrorCategory.USER,
+        details: { indexName, id }
+      });
+    }
     try {
-      if (!update.vector && !update.metadata) {
-        throw new Error("No updates provided");
-      }
       const collection = this.#db.collection(indexName);
       const updateDoc = {};
       if (update.vector) {
@@ -200,8 +284,17 @@ var AstraVector = class extends vector.MastraVector {
         updateDoc.metadata = update.metadata;
       }
       await collection.findOneAndUpdate({ id }, { $set: updateDoc });
-    } catch (error) {
-      throw new Error(`Failed to update vector by id: ${id} for index name: ${indexName}: ${error.message}`);
+    } catch (error$1) {
+      if (error$1 instanceof error.MastraError) throw error$1;
+      throw new error.MastraError(
+        {
+          id: "ASTRA_VECTOR_UPDATE_FAILED_UNHANDLED",
+          domain: error.ErrorDomain.MASTRA_VECTOR,
+          category: error.ErrorCategory.THIRD_PARTY,
+          details: { indexName, id }
+        },
+        error$1
+      );
     }
   }
   /**
@@ -215,8 +308,17 @@ var AstraVector = class extends vector.MastraVector {
     try {
       const collection = this.#db.collection(indexName);
       await collection.deleteOne({ id });
-    } catch (error) {
-      throw new Error(`Failed to delete vector by id: ${id} for index name: ${indexName}: ${error.message}`);
+    } catch (error$1) {
+      if (error$1 instanceof error.MastraError) throw error$1;
+      throw new error.MastraError(
+        {
+          id: "ASTRA_VECTOR_DELETE_FAILED",
+          domain: error.ErrorDomain.MASTRA_VECTOR,
+          category: error.ErrorCategory.THIRD_PARTY,
+          details: { indexName, id }
+        },
+        error$1
+      );
     }
   }
 };