@mastra/astra 1.1.1 → 1.1.2-alpha.1

package/dist/docs/SKILL.md

@@ -3,7 +3,7 @@ name: mastra-astra
 description: Documentation for @mastra/astra. Use when working with @mastra/astra APIs, configuration, or implementation.
 metadata:
   package: "@mastra/astra"
-  version: "1.1.1"
+  version: "1.1.2-alpha.1"
 ---
 
 ## When to use

package/dist/docs/assets/SOURCE_MAP.json

@@ -1,5 +1,5 @@
 {
-  "version": "1.1.1",
+  "version": "1.1.2-alpha.1",
   "package": "@mastra/astra",
   "exports": {},
   "modules": {}

package/dist/docs/references/reference-rag-retrieval.md

@@ -50,23 +50,25 @@ console.log(results)
 
 The `topK` parameter specifies the maximum number of most similar results to return from the vector search.
 
-Results include both the text content and a similarity score:
+Each result includes an `id`, a similarity `score`, and the stored `metadata`. When you store chunk text in metadata, it's available as `metadata.text`:
 
 ```ts
 [
   {
-    text: 'Climate change poses significant challenges...',
+    id: 'chunk-1',
     score: 0.89,
-    metadata: { source: 'article1.txt' },
+    metadata: { text: 'Climate change poses significant challenges...', source: 'article1.txt' },
   },
   {
-    text: 'Rising temperatures affect crop yields...',
+    id: 'chunk-2',
     score: 0.82,
-    metadata: { source: 'article1.txt' },
+    metadata: { text: 'Rising temperatures affect crop yields...', source: 'article1.txt' },
   },
 ]
 ```
 
+Results can also include `vector` when you set `includeVector: true`, and `document` when the store returns the original document (Chroma only).
+
 ## Advanced retrieval options
 
 ### Metadata Filtering

package/dist/docs/references/reference-rag-vector-databases.md

@@ -54,6 +54,29 @@ const results = await store.hybridQuery({
 
 See the [MongoDB vector reference](https://mastra.ai/reference/vectors/mongodb) for details on `createSearchIndex()`, `textQuery()`, and `hybridQuery()`.
 
+### Automated Embedding
+
+MongoDB can generate the embeddings itself, so you don't need an embedding provider in your application. Create the index with `autoEmbed` and a Voyage AI model, write plain text, and search with a query string:
+
+```ts
+await store.createIndex({
+  indexName: 'myCollection',
+  autoEmbed: { model: 'voyage-4' },
+})
+await store.upsert({
+  indexName: 'myCollection',
+  documents: chunks.map(chunk => chunk.text),
+  metadata: chunks.map(chunk => ({ text: chunk.text })),
+})
+const results = await store.query({
+  indexName: 'myCollection',
+  queryText: 'search terms',
+  topK: 10,
+})
+```
+
+Automated Embedding is a MongoDB Preview feature and needs a deployment where it's available. See the [MongoDB vector reference](https://mastra.ai/reference/vectors/mongodb) for the supported options and requirements.
+
 **PgVector**:
 
 ```ts
@@ -207,7 +230,7 @@ await store.upsert({
 **libSQL**:
 
 ```ts
-import { LibSQLVector } from '@mastra/core/vector/libsql'
+import { LibSQLVector } from '@mastra/libsql'
 
 const store = new LibSQLVector({
   id: 'libsql-vector',
@@ -248,6 +271,34 @@ await store.upsert({
 })
 ```
 
+**Weaviate**:
+
+```ts
+import { WeaviateVector } from '@mastra/weaviate'
+
+const store = new WeaviateVector({
+  id: 'weaviate-vector',
+  httpHost: process.env.WEAVIATE_HOST,
+  httpPort: 443,
+  httpSecure: true,
+  grpcHost: process.env.WEAVIATE_GRPC_HOST,
+  grpcPort: 443,
+  grpcSecure: true,
+  apiKey: process.env.WEAVIATE_API_KEY,
+})
+
+await store.createIndex({
+  indexName: 'myCollection',
+  dimension: 1536,
+})
+
+await store.upsert({
+  indexName: 'myCollection',
+  vectors: embeddings,
+  metadata: chunks.map(chunk => ({ text: chunk.text })),
+})
+```
+
 **Cloudflare**:
 
 ```ts
@@ -317,6 +368,29 @@ await store.upsert({
 
 For detailed setup instructions and best practices, see the [official Elasticsearch documentation](https://www.elastic.co/docs/solutions/search/get-started).
 
+**Azure AI Search**:
+
+```ts
+import { AzureAISearchVector } from '@mastra/azure-ai-search'
+
+const store = new AzureAISearchVector({
+  id: 'azure-search-vectors',
+  endpoint: process.env.AZURE_AI_SEARCH_ENDPOINT!,
+  credential: process.env.AZURE_AI_SEARCH_CREDENTIAL!,
+})
+
+await store.createIndex({
+  indexName: 'my-collection',
+  dimension: 1536,
+})
+
+await store.upsert({
+  indexName: 'my-collection',
+  vectors: embeddings,
+  metadata: chunks.map(chunk => ({ text: chunk.text })),
+})
+```
+
 **Couchbase**:
 
 ```ts
@@ -534,6 +608,17 @@ Namespace names must:
 
 - Example: `_namespace` isn't valid (starts with underscore)
 
+**Weaviate**:
+
+Index names map to Weaviate collections, which:
+
+- Are capitalized by Weaviate (the first letter is upper-cased)
+- Should contain only letters, numbers, and the `_` character
+- Must start with a letter, since only the first character is upper-cased (a leading digit or symbol stays invalid)
+- Preserve the original Mastra index name in the collection description, so `listIndexes()` and `describeIndex()` return the name you supplied
+- Example: `my_collection` is stored as `My_collection` and returned as `my_collection`
+- Example: `123_collection` isn't valid (starts with a digit)
+
 **Cloudflare**:
 
 Index names must:
@@ -587,6 +672,17 @@ Index names must:
 - Example: `myindex-` isn't valid (ends with hyphen)
 - Example: `MyIndex` isn't valid (contains uppercase letters)
 
+**Azure AI Search**:
+
+Index names must:
+
+- Use only lowercase letters, numbers, dashes (`-`), and underscore (`_`) characters
+- Not start or end with a dash
+- Be between 2 and 128 characters long
+- Example: `my-index-123` and `my_index` are valid
+- Example: `MyIndex` isn't valid (contains uppercase letters)
+- Example: `my-index-` isn't valid (ends with a dash)
+
 ### Upserting Embeddings
 
 After creating an index, you can store embeddings along with their basic metadata:
@@ -650,7 +746,7 @@ Key metadata considerations:
 
 ## Deleting vectors
 
-When building RAG applications, you often need to clean up stale vectors when documents are deleted or updated. Mastra provides the `deleteVectors` method that supports deleting vectors by metadata filters, making it straightforward to remove all embeddings associated with a specific document.
+Use `deleteVectors` with a metadata filter to remove embeddings associated with a document. This is useful for cleaning up stale vectors after a document is deleted or updated.
 
 ### Delete by Metadata Filter
 

package/dist/docs/references/reference-vectors-astra.md

@@ -60,7 +60,7 @@ Returns:
 interface IndexStats {
   dimension: number
   count: number
-  metric: 'cosine' | 'euclidean' | 'dotproduct'
+  metric?: 'cosine' | 'euclidean' | 'dotproduct'
 }
 ```
 
@@ -90,24 +90,29 @@ Query results are returned in this format:
 interface QueryResult {
   id: string
   score: number
-  metadata: Record<string, any>
+  metadata?: Record<string, any>
   vector?: number[] // Only included if includeVector is true
+  document?: string // Only returned by stores that keep document content
 }
 ```
 
 ## Error handling
 
-The store throws typed errors that can be caught:
+The store throws `MastraError` instances from `@mastra/core/error`:
 
 ```typescript
+import { MastraError } from '@mastra/core/error'
+
 try {
   await store.query({
     indexName: 'index_name',
     queryVector: queryVector,
   })
 } catch (error) {
-  if (error instanceof VectorStoreError) {
-    console.log(error.code) // 'connection_failed' | 'invalid_dimension' | etc
+  if (error instanceof MastraError) {
+    console.log(error.id) // 'MASTRA_VECTOR_ASTRA_QUERY_DB_ERROR'
+    console.log(error.domain) // 'MASTRA_VECTOR'
+    console.log(error.category) // 'USER' | 'THIRD_PARTY'
     console.log(error.details) // Additional error context
   }
 }

package/dist/index.cjs

@@ -151,7 +151,7 @@ var AstraVector = class extends _mastra_core_vector.MastraVector {
 				sort: { $vector: queryVector },
 				limit: topK,
 				includeSimilarity: true,
-				projection: { $vector: includeVector ? true : false }
+				projection: { $vector: !!includeVector }
 			}).toArray()).map((result) => ({
 				id: result.id,
 				score: result.$similarity,

package/dist/index.cjs.map

@@ -1 +1 @@
-{"version":3,"file":"index.cjs","names":["BaseFilterTranslator","MastraVector","#db","DataAPIClient","MastraError","ErrorDomain","ErrorCategory","UUID"],"sources":["../src/vector/filter.ts","../src/vector/index.ts","../src/vector/prompt.ts"],"sourcesContent":["import { BaseFilterTranslator } from '@mastra/core/vector/filter';\nimport type {\n  VectorFilter,\n  OperatorSupport,\n  QueryOperator,\n  OperatorValueMap,\n  LogicalOperatorValueMap,\n  BlacklistedRootOperators,\n} from '@mastra/core/vector/filter';\n\ntype AstraOperatorValueMap = Omit<OperatorValueMap, '$elemMatch' | '$regex' | '$options'> & {\n  $size: number; // Astra-specific\n};\n\ntype AstraLogicalOperatorValueMap = Omit<LogicalOperatorValueMap, '$nor'>;\n\ntype AstraBlacklisted = BlacklistedRootOperators | '$nor' | '$size';\n\nexport type AstraVectorFilter = VectorFilter<\n  keyof AstraOperatorValueMap,\n  AstraOperatorValueMap,\n  AstraLogicalOperatorValueMap,\n  AstraBlacklisted\n>;\n\n/**\n * Translator for Astra DB filter queries.\n * Maintains MongoDB-compatible syntax while ensuring proper validation\n * and normalization of values.\n */\nexport class AstraFilterTranslator extends BaseFilterTranslator<AstraVectorFilter> {\n  protected override getSupportedOperators(): OperatorSupport {\n    return {\n      ...BaseFilterTranslator.DEFAULT_OPERATORS,\n      array: ['$all', '$in', '$nin'],\n      logical: ['$and', '$or', '$not'],\n      regex: [],\n      custom: ['$size'],\n    };\n  }\n\n  translate(filter?: AstraVectorFilter): AstraVectorFilter {\n    if (this.isEmpty(filter)) return filter;\n    this.validateFilter(filter);\n\n    return this.translateNode(filter);\n  }\n\n  private translateNode(node: AstraVectorFilter): any {\n    // Handle primitive values and arrays\n    if (this.isRegex(node)) {\n      throw new Error('Regex is not supported in Astra DB');\n    }\n    if (this.isPrimitive(node) || Array.isArray(node)) {\n      return node;\n    }\n\n    const entries = Object.entries(node as Record<string, any>);\n    const translatedEntries = entries.map(([key, value]) => {\n      // Handle operators\n      if (this.isOperator(key)) {\n        return [key, this.translateOperatorValue(key, value)];\n      }\n\n      // Handle nested paths and objects\n      return [key, this.translateNode(value)];\n    });\n\n    return Object.fromEntries(translatedEntries);\n  }\n\n  private translateOperatorValue(operator: QueryOperator, value: any): any {\n    if (this.isBasicOperator(operator) || this.isNumericOperator(operator)) {\n      return this.normalizeComparisonValue(value);\n    }\n\n    if (this.isArrayOperator(operator) && Array.isArray(value)) {\n      return this.normalizeArrayValues(value);\n    }\n\n    return this.translateNode(value);\n  }\n}\n","import type { Db } from '@datastax/astra-db-ts';\nimport { DataAPIClient, UUID } from '@datastax/astra-db-ts';\nimport { MastraError, ErrorDomain, ErrorCategory } from '@mastra/core/error';\nimport { createVectorErrorId } from '@mastra/core/storage';\nimport { MastraVector } from '@mastra/core/vector';\nimport type {\n  QueryResult,\n  IndexStats,\n  CreateIndexParams,\n  UpsertVectorParams,\n  QueryVectorParams,\n  DescribeIndexParams,\n  DeleteIndexParams,\n  DeleteVectorParams,\n  UpdateVectorParams,\n  DeleteVectorsParams,\n} from '@mastra/core/vector';\nimport type { AstraVectorFilter } from './filter';\nimport { AstraFilterTranslator } from './filter';\n\n// Mastra and Astra DB agree on cosine and euclidean, but Astra DB uses dot_product instead of dotproduct.\nconst metricMap = {\n  cosine: 'cosine',\n  euclidean: 'euclidean',\n  dotproduct: 'dot_product',\n} as const;\n\nexport interface AstraDbOptions {\n  token: string;\n  endpoint: string;\n  keyspace?: string;\n}\n\ntype AstraQueryVectorParams = QueryVectorParams<AstraVectorFilter>;\n\nexport class AstraVector extends MastraVector<AstraVectorFilter> {\n  readonly #db: Db;\n\n  constructor({ token, endpoint, keyspace, id }: AstraDbOptions & { id: string }) {\n    super({ id });\n    const client = new DataAPIClient(token);\n    this.#db = client.db(endpoint, { keyspace });\n  }\n\n  /**\n   * Creates a new collection with the specified configuration.\n   *\n   * @param {string} indexName - The name of the collection to create.\n   * @param {number} dimension - The dimension of the vectors to be stored in the collection.\n   * @param {'cosine' | 'euclidean' | 'dotproduct'} [metric=cosine] - The metric to use to sort vectors in the collection.\n   * @returns {Promise<void>} A promise that resolves when the collection is created.\n   */\n  async createIndex({ indexName, dimension, metric = 'cosine' }: CreateIndexParams): Promise<void> {\n    if (!Number.isInteger(dimension) || dimension <= 0) {\n      throw new MastraError({\n        id: createVectorErrorId('ASTRA', 'CREATE_INDEX', 'INVALID_DIMENSION'),\n        text: 'Dimension must be a positive integer',\n        domain: ErrorDomain.MASTRA_VECTOR,\n        category: ErrorCategory.USER,\n      });\n    }\n    try {\n      await this.#db.createCollection(indexName, {\n        vector: {\n          dimension,\n          metric: metricMap[metric],\n        },\n        checkExists: false,\n      });\n    } catch (error: any) {\n      new MastraError(\n        {\n          id: createVectorErrorId('ASTRA', 'CREATE_INDEX', 'DB_ERROR'),\n          domain: ErrorDomain.MASTRA_VECTOR,\n          category: ErrorCategory.THIRD_PARTY,\n          details: { indexName },\n        },\n        error,\n      );\n    }\n  }\n\n  /**\n   * Inserts or updates vectors in the specified collection.\n   *\n   * @param {string} indexName - The name of the collection to upsert into.\n   * @param {number[][]} vectors - An array of vectors to upsert.\n   * @param {Record<string, any>[]} [metadata] - An optional array of metadata objects corresponding to each vector.\n   * @param {string[]} [ids] - An optional array of IDs corresponding to each vector. If not provided, new IDs will be generated.\n   * @returns {Promise<string[]>} A promise that resolves to an array of IDs of the upserted vectors.\n   */\n  async upsert({ indexName, vectors, metadata, ids }: UpsertVectorParams): Promise<string[]> {\n    const collection = this.#db.collection(indexName);\n\n    // Generate IDs if not provided\n    const vectorIds = ids || vectors.map(() => UUID.v7().toString());\n\n    const records = vectors.map((vector, i) => ({\n      id: vectorIds[i],\n      $vector: vector,\n      metadata: metadata?.[i] || {},\n    }));\n\n    try {\n      await collection.insertMany(records);\n    } catch (error: any) {\n      throw new MastraError(\n        {\n          id: createVectorErrorId('ASTRA', 'UPSERT', 'DB_ERROR'),\n          domain: ErrorDomain.MASTRA_VECTOR,\n          category: ErrorCategory.THIRD_PARTY,\n          details: { indexName },\n        },\n        error,\n      );\n    }\n    return vectorIds;\n  }\n\n  transformFilter(filter?: AstraVectorFilter) {\n    const translator = new AstraFilterTranslator();\n    return translator.translate(filter);\n  }\n\n  /**\n   * Queries the specified collection using a vector and optional filter.\n   *\n   * @param {string} indexName - The name of the collection to query.\n   * @param {number[]} queryVector - The vector to query with.\n   * @param {number} [topK] - The maximum number of results to return.\n   * @param {Record<string, any>} [filter] - An optional filter to apply to the query. For more on filters in Astra DB, see the filtering reference: https://docs.datastax.com/en/astra-db-serverless/api-reference/documents.html#operators\n   * @param {boolean} [includeVectors=false] - Whether to include the vectors in the response.\n   * @returns {Promise<QueryResult[]>} A promise that resolves to an array of query results.\n   */\n  async query({\n    indexName,\n    queryVector,\n    topK = 10,\n    filter,\n    includeVector = false,\n  }: AstraQueryVectorParams): Promise<QueryResult[]> {\n    if (!queryVector) {\n      throw new MastraError({\n        id: createVectorErrorId('ASTRA', 'QUERY', 'MISSING_VECTOR'),\n        text: 'queryVector is required for Astra queries. Metadata-only queries are not supported by this vector store.',\n        domain: ErrorDomain.MASTRA_VECTOR,\n        category: ErrorCategory.USER,\n        details: { indexName },\n      });\n    }\n\n    const collection = this.#db.collection(indexName);\n\n    const translatedFilter = this.transformFilter(filter);\n\n    try {\n      const cursor = collection.find(translatedFilter ?? {}, {\n        sort: { $vector: queryVector },\n        limit: topK,\n        includeSimilarity: true,\n        projection: {\n          $vector: includeVector ? true : false,\n        },\n      });\n\n      const results = await cursor.toArray();\n\n      return results.map(result => ({\n        id: result.id,\n        score: result.$similarity,\n        metadata: result.metadata,\n        ...(includeVector && { vector: result.$vector }),\n      }));\n    } catch (error: any) {\n      throw new MastraError(\n        {\n          id: createVectorErrorId('ASTRA', 'QUERY', 'DB_ERROR'),\n          domain: ErrorDomain.MASTRA_VECTOR,\n          category: ErrorCategory.THIRD_PARTY,\n          details: { indexName },\n        },\n        error,\n      );\n    }\n  }\n\n  /**\n   * Lists all collections in the database.\n   *\n   * @returns {Promise<string[]>} A promise that resolves to an array of collection names.\n   */\n  async listIndexes(): Promise<string[]> {\n    try {\n      return await this.#db.listCollections({ nameOnly: true });\n    } catch (error: any) {\n      throw new MastraError(\n        {\n          id: createVectorErrorId('ASTRA', 'LIST_INDEXES', 'DB_ERROR'),\n          domain: ErrorDomain.MASTRA_VECTOR,\n          category: ErrorCategory.THIRD_PARTY,\n        },\n        error,\n      );\n    }\n  }\n\n  /**\n   * Retrieves statistics about a vector index.\n   *\n   * @param {string} indexName - The name of the index to describe\n   * @returns A promise that resolves to the index statistics including dimension, count and metric\n   */\n  async describeIndex({ indexName }: DescribeIndexParams): Promise<IndexStats> {\n    const collection = this.#db.collection(indexName);\n    try {\n      const optionsPromise = collection.options();\n      const countPromise = collection.countDocuments({}, 100);\n      const [options, count] = await Promise.all([optionsPromise, countPromise]);\n\n      const keys = Object.keys(metricMap) as (keyof typeof metricMap)[];\n      const metric = keys.find(key => metricMap[key] === options.vector?.metric);\n      return {\n        dimension: options.vector?.dimension!,\n        metric,\n        count: count,\n      };\n    } catch (error: any) {\n      if (error instanceof MastraError) throw error;\n      throw new MastraError(\n        {\n          id: createVectorErrorId('ASTRA', 'DESCRIBE_INDEX', 'DB_ERROR'),\n          domain: ErrorDomain.MASTRA_VECTOR,\n          category: ErrorCategory.THIRD_PARTY,\n          details: { indexName },\n        },\n        error,\n      );\n    }\n  }\n\n  /**\n   * Deletes the specified collection.\n   *\n   * @param {string} indexName - The name of the collection to delete.\n   * @returns {Promise<void>} A promise that resolves when the collection is deleted.\n   */\n  async deleteIndex({ indexName }: DeleteIndexParams): Promise<void> {\n    const collection = this.#db.collection(indexName);\n    try {\n      await collection.drop();\n    } catch (error: any) {\n      throw new MastraError(\n        {\n          id: createVectorErrorId('ASTRA', 'DELETE_INDEX', 'DB_ERROR'),\n          domain: ErrorDomain.MASTRA_VECTOR,\n          category: ErrorCategory.THIRD_PARTY,\n          details: { indexName },\n        },\n        error,\n      );\n    }\n  }\n\n  /**\n   * Updates a vector by its ID with the provided vector and/or metadata.\n   * @param indexName - The name of the index containing the vector.\n   * @param id - The ID of the vector to update.\n   * @param update - An object containing the vector and/or metadata to update.\n   * @param update.vector - An optional array of numbers representing the new vector.\n   * @param update.metadata - An optional record containing the new metadata.\n   * @returns A promise that resolves when the update is complete.\n   * @throws Will throw an error if no updates are provided or if the update operation fails.\n   */\n  async updateVector({ indexName, id, update }: UpdateVectorParams): Promise<void> {\n    if (!id) {\n      throw new MastraError({\n        id: createVectorErrorId('ASTRA', 'UPDATE_VECTOR', 'NO_ID'),\n        text: 'id is required for Astra updateVector',\n        domain: ErrorDomain.MASTRA_VECTOR,\n        category: ErrorCategory.USER,\n        details: { indexName },\n      });\n    }\n\n    if (!update.vector && !update.metadata) {\n      throw new MastraError({\n        id: createVectorErrorId('ASTRA', 'UPDATE_VECTOR', 'NO_PAYLOAD'),\n        text: 'No updates provided for vector',\n        domain: ErrorDomain.MASTRA_VECTOR,\n        category: ErrorCategory.USER,\n        details: {\n          indexName,\n          id,\n        },\n      });\n    }\n\n    try {\n      const collection = this.#db.collection(indexName);\n      const updateDoc: Record<string, any> = {};\n\n      if (update.vector) {\n        updateDoc.$vector = update.vector;\n      }\n\n      if (update.metadata) {\n        updateDoc.metadata = update.metadata;\n      }\n\n      await collection.findOneAndUpdate({ id }, { $set: updateDoc });\n    } catch (error: any) {\n      if (error instanceof MastraError) throw error;\n      throw new MastraError(\n        {\n          id: createVectorErrorId('ASTRA', 'UPDATE_VECTOR', 'FAILED_UNHANDLED'),\n          domain: ErrorDomain.MASTRA_VECTOR,\n          category: ErrorCategory.THIRD_PARTY,\n          details: {\n            indexName,\n            ...(id && { id }),\n          },\n        },\n        error,\n      );\n    }\n  }\n\n  /**\n   * Deletes a vector by its ID.\n   * @param indexName - The name of the index containing the vector.\n   * @param id - The ID of the vector to delete.\n   * @returns A promise that resolves when the deletion is complete.\n   * @throws Will throw an error if the deletion operation fails.\n   */\n  async deleteVector({ indexName, id }: DeleteVectorParams): Promise<void> {\n    try {\n      const collection = this.#db.collection(indexName);\n      await collection.deleteOne({ id });\n    } catch (error: any) {\n      if (error instanceof MastraError) throw error;\n      throw new MastraError(\n        {\n          id: createVectorErrorId('ASTRA', 'DELETE_VECTOR', 'FAILED'),\n          domain: ErrorDomain.MASTRA_VECTOR,\n          category: ErrorCategory.THIRD_PARTY,\n          details: {\n            indexName,\n            ...(id && { id }),\n          },\n        },\n        error,\n      );\n    }\n  }\n\n  async deleteVectors({ indexName, filter, ids }: DeleteVectorsParams): Promise<void> {\n    throw new MastraError({\n      id: createVectorErrorId('ASTRA', 'DELETE_VECTORS', 'NOT_SUPPORTED'),\n      text: 'deleteVectors is not yet implemented for Astra vector store',\n      domain: ErrorDomain.MASTRA_VECTOR,\n      category: ErrorCategory.SYSTEM,\n      details: {\n        indexName,\n        ...(filter && { filter: JSON.stringify(filter) }),\n        ...(ids && { idsCount: ids.length }),\n      },\n    });\n  }\n}\n","/**\n * Vector store specific prompt that details supported operators and examples.\n * This prompt helps users construct valid filters for Astra Vector.\n */\nexport const ASTRA_PROMPT = `When querying Astra, you can ONLY use the operators listed below. Any other operators will be rejected.\nImportant: Don't explain how to construct the filter - use the specified operators and fields to search the content and return relevant results.\nIf a user tries to give an explicit operator that is not supported, reject the filter entirely and let them know that the operator is not supported.\n\nBasic Comparison Operators:\n- $eq: Exact match (default when using field: value)\n  Example: { \"category\": \"electronics\" }\n- $ne: Not equal\n  Example: { \"category\": { \"$ne\": \"electronics\" } }\n- $gt: Greater than\n  Example: { \"price\": { \"$gt\": 100 } }\n- $gte: Greater than or equal\n  Example: { \"price\": { \"$gte\": 100 } }\n- $lt: Less than\n  Example: { \"price\": { \"$lt\": 100 } }\n- $lte: Less than or equal\n  Example: { \"price\": { \"$lte\": 100 } }\n\nArray Operators:\n- $in: Match any value in array\n  Example: { \"category\": { \"$in\": [\"electronics\", \"books\"] } }\n- $nin: Does not match any value in array\n  Example: { \"category\": { \"$nin\": [\"electronics\", \"books\"] } }\n- $all: Match all values in array\n  Example: { \"tags\": { \"$all\": [\"premium\", \"sale\"] } }\n\nLogical Operators:\n- $and: Logical AND (can be implicit or explicit)\n  Implicit Example: { \"price\": { \"$gt\": 100 }, \"category\": \"electronics\" }\n  Explicit Example: { \"$and\": [{ \"price\": { \"$gt\": 100 } }, { \"category\": \"electronics\" }] }\n- $or: Logical OR\n  Example: { \"$or\": [{ \"price\": { \"$lt\": 50 } }, { \"category\": \"books\" }] }\n- $not: Logical NOT\n  Example: { \"$not\": { \"category\": \"electronics\" } }\n\nElement Operators:\n- $exists: Check if field exists\n  Example: { \"rating\": { \"$exists\": true } }\n\nSpecial Operators:\n- $size: Array length check\n  Example: { \"tags\": { \"$size\": 2 } }\n\nRestrictions:\n- Regex patterns are not supported\n- Only $and, $or, and $not logical operators are supported\n- Nested fields are supported using dot notation\n- Multiple conditions on the same field are supported with both implicit and explicit $and\n- Empty arrays in $in/$nin will return no results\n- A non-empty array is required for $all operator\n- Only logical operators ($and, $or, $not) can be used at the top level\n- All other operators must be used within a field condition\n  Valid: { \"field\": { \"$gt\": 100 } }\n  Valid: { \"$and\": [...] }\n  Invalid: { \"$gt\": 100 }\n- Logical operators must contain field conditions, not direct operators\n  Valid: { \"$and\": [{ \"field\": { \"$gt\": 100 } }] }\n  Invalid: { \"$and\": [{ \"$gt\": 100 }] }\n- $not operator:\n  - Must be an object\n  - Cannot be empty\n  - Can be used at field level or top level\n  - Valid: { \"$not\": { \"field\": \"value\" } }\n  - Valid: { \"field\": { \"$not\": { \"$eq\": \"value\" } } }\n- Other logical operators ($and, $or):\n  - Can only be used at top level or nested within other logical operators\n  - Can not be used on a field level, or be nested inside a field\n  - Can not be used inside an operator\n  - Valid: { \"$and\": [{ \"field\": { \"$gt\": 100 } }] }\n  - Valid: { \"$or\": [{ \"$and\": [{ \"field\": { \"$gt\": 100 } }] }] }\n  - Invalid: { \"field\": { \"$and\": [{ \"$gt\": 100 }] } }\n  - Invalid: { \"field\": { \"$or\": [{ \"$gt\": 100 }] } }\n  - Invalid: { \"field\": { \"$gt\": { \"$and\": [{...}] } } }\n\nExample Complex Query:\n{\n  \"$and\": [\n    { \"category\": { \"$in\": [\"electronics\", \"computers\"] } },\n    { \"price\": { \"$gte\": 100, \"$lte\": 1000 } },\n    { \"tags\": { \"$all\": [\"premium\"] } },\n    { \"rating\": { \"$exists\": true, \"$gt\": 4 } },\n    { \"$or\": [\n      { \"stock\": { \"$gt\": 0 } },\n      { \"preorder\": true }\n    ]}\n  ]\n}`;\n"],"mappings":";;;;;;;;;;;;AA8BA,IAAa,wBAAb,cAA2CA,2BAAAA,qBAAwC;CACjF,wBAA4D;EAC1D,OAAO;GACL,GAAGA,2BAAAA,qBAAqB;GACxB,OAAO;IAAC;IAAQ;IAAO;GAAM;GAC7B,SAAS;IAAC;IAAQ;IAAO;GAAM;GAC/B,OAAO,CAAC;GACR,QAAQ,CAAC,OAAO;EAClB;CACF;CAEA,UAAU,QAA+C;EACvD,IAAI,KAAK,QAAQ,MAAM,GAAG,OAAO;EACjC,KAAK,eAAe,MAAM;EAE1B,OAAO,KAAK,cAAc,MAAM;CAClC;CAEA,cAAsB,MAA8B;EAElD,IAAI,KAAK,QAAQ,IAAI,GACnB,MAAM,IAAI,MAAM,oCAAoC;EAEtD,IAAI,KAAK,YAAY,IAAI,KAAK,MAAM,QAAQ,IAAI,GAC9C,OAAO;EAIT,MAAM,oBADU,OAAO,QAAQ,IACC,CAAC,CAAC,KAAK,CAAC,KAAK,WAAW;GAEtD,IAAI,KAAK,WAAW,GAAG,GACrB,OAAO,CAAC,KAAK,KAAK,uBAAuB,KAAK,KAAK,CAAC;GAItD,OAAO,CAAC,KAAK,KAAK,cAAc,KAAK,CAAC;EACxC,CAAC;EAED,OAAO,OAAO,YAAY,iBAAiB;CAC7C;CAEA,uBAA+B,UAAyB,OAAiB;EACvE,IAAI,KAAK,gBAAgB,QAAQ,KAAK,KAAK,kBAAkB,QAAQ,GACnE,OAAO,KAAK,yBAAyB,KAAK;EAG5C,IAAI,KAAK,gBAAgB,QAAQ,KAAK,MAAM,QAAQ,KAAK,GACvD,OAAO,KAAK,qBAAqB,KAAK;EAGxC,OAAO,KAAK,cAAc,KAAK;CACjC;AACF;;;AC7DA,MAAM,YAAY;CAChB,QAAQ;CACR,WAAW;CACX,YAAY;AACd;AAUA,IAAa,cAAb,cAAiCC,oBAAAA,aAAgC;CAC/D;CAEA,YAAY,EAAE,OAAO,UAAU,UAAU,MAAuC;EAC9E,MAAM,EAAE,GAAG,CAAC;EACZ,MAAM,SAAS,IAAIE,sBAAAA,cAAc,KAAK;EACtC,KAAKD,MAAM,OAAO,GAAG,UAAU,EAAE,SAAS,CAAC;CAC7C;;;;;;;;;CAUA,MAAM,YAAY,EAAE,WAAW,WAAW,SAAS,YAA8C;EAC/F,IAAI,CAAC,OAAO,UAAU,SAAS,KAAK,aAAa,GAC/C,MAAM,IAAIE,mBAAAA,YAAY;GACpB,KAAA,GAAA,qBAAA,oBAAA,CAAwB,SAAS,gBAAgB,mBAAmB;GACpE,MAAM;GACN,QAAQC,mBAAAA,YAAY;GACpB,UAAUC,mBAAAA,cAAc;EAC1B,CAAC;EAEH,IAAI;GACF,MAAM,KAAKJ,IAAI,iBAAiB,WAAW;IACzC,QAAQ;KACN;KACA,QAAQ,UAAU;IACpB;IACA,aAAa;GACf,CAAC;EACH,SAAS,OAAY;GACnB,IAAIE,mBAAAA,YACF;IACE,KAAA,GAAA,qBAAA,oBAAA,CAAwB,SAAS,gBAAgB,UAAU;IAC3D,QAAQC,mBAAAA,YAAY;IACpB,UAAUC,mBAAAA,cAAc;IACxB,SAAS,EAAE,UAAU;GACvB,GACA,KACF;EACF;CACF;;;;;;;;;;CAWA,MAAM,OAAO,EAAE,WAAW,SAAS,UAAU,OAA8C;EACzF,MAAM,aAAa,KAAKJ,IAAI,WAAW,SAAS;EAGhD,MAAM,YAAY,OAAO,QAAQ,UAAUK,sBAAAA,KAAK,GAAG,CAAC,CAAC,SAAS,CAAC;EAE/D,MAAM,UAAU,QAAQ,KAAK,QAAQ,OAAO;GAC1C,IAAI,UAAU;GACd,SAAS;GACT,UAAU,WAAW,MAAM,CAAC;EAC9B,EAAE;EAEF,IAAI;GACF,MAAM,WAAW,WAAW,OAAO;EACrC,SAAS,OAAY;GACnB,MAAM,IAAIH,mBAAAA,YACR;IACE,KAAA,GAAA,qBAAA,oBAAA,CAAwB,SAAS,UAAU,UAAU;IACrD,QAAQC,mBAAAA,YAAY;IACpB,UAAUC,mBAAAA,cAAc;IACxB,SAAS,EAAE,UAAU;GACvB,GACA,KACF;EACF;EACA,OAAO;CACT;CAEA,gBAAgB,QAA4B;EAE1C,OAAO,IADgB,sBACP,CAAC,CAAC,UAAU,MAAM;CACpC;;;;;;;;;;;CAYA,MAAM,MAAM,EACV,WACA,aACA,OAAO,IACP,QACA,gBAAgB,SACiC;EACjD,IAAI,CAAC,aACH,MAAM,IAAIF,mBAAAA,YAAY;GACpB,KAAA,GAAA,qBAAA,oBAAA,CAAwB,SAAS,SAAS,gBAAgB;GAC1D,MAAM;GACN,QAAQC,mBAAAA,YAAY;GACpB,UAAUC,mBAAAA,cAAc;GACxB,SAAS,EAAE,UAAU;EACvB,CAAC;EAGH,MAAM,aAAa,KAAKJ,IAAI,WAAW,SAAS;EAEhD,MAAM,mBAAmB,KAAK,gBAAgB,MAAM;EAEpD,IAAI;GAYF,QAAO,MAXQ,WAAW,KAAK,oBAAoB,CAAC,GAAG;IACrD,MAAM,EAAE,SAAS,YAAY;IAC7B,OAAO;IACP,mBAAmB;IACnB,YAAY,EACV,SAAS,gBAAgB,OAAO,MAClC;GACF,CAE2B,CAAC,CAAC,QAAQ,EAAA,CAEtB,KAAI,YAAW;IAC5B,IAAI,OAAO;IACX,OAAO,OAAO;IACd,UAAU,OAAO;IACjB,GAAI,iBAAiB,EAAE,QAAQ,OAAO,QAAQ;GAChD,EAAE;EACJ,SAAS,OAAY;GACnB,MAAM,IAAIE,mBAAAA,YACR;IACE,KAAA,GAAA,qBAAA,oBAAA,CAAwB,SAAS,SAAS,UAAU;IACpD,QAAQC,mBAAAA,YAAY;IACpB,UAAUC,mBAAAA,cAAc;IACxB,SAAS,EAAE,UAAU;GACvB,GACA,KACF;EACF;CACF;;;;;;CAOA,MAAM,cAAiC;EACrC,IAAI;GACF,OAAO,MAAM,KAAKJ,IAAI,gBAAgB,EAAE,UAAU,KAAK,CAAC;EAC1D,SAAS,OAAY;GACnB,MAAM,IAAIE,mBAAAA,YACR;IACE,KAAA,GAAA,qBAAA,oBAAA,CAAwB,SAAS,gBAAgB,UAAU;IAC3D,QAAQC,mBAAAA,YAAY;IACpB,UAAUC,mBAAAA,cAAc;GAC1B,GACA,KACF;EACF;CACF;;;;;;;CAQA,MAAM,cAAc,EAAE,aAAuD;EAC3E,MAAM,aAAa,KAAKJ,IAAI,WAAW,SAAS;EAChD,IAAI;GACF,MAAM,iBAAiB,WAAW,QAAQ;GAC1C,MAAM,eAAe,WAAW,eAAe,CAAC,GAAG,GAAG;GACtD,MAAM,CAAC,SAAS,SAAS,MAAM,QAAQ,IAAI,CAAC,gBAAgB,YAAY,CAAC;GAGzE,MAAM,SADO,OAAO,KAAK,SACP,CAAC,CAAC,MAAK,QAAO,UAAU,SAAS,QAAQ,QAAQ,MAAM;GACzE,OAAO;IACL,WAAW,QAAQ,QAAQ;IAC3B;IACO;GACT;EACF,SAAS,OAAY;GACnB,IAAI,iBAAiBE,mBAAAA,aAAa,MAAM;GACxC,MAAM,IAAIA,mBAAAA,YACR;IACE,KAAA,GAAA,qBAAA,oBAAA,CAAwB,SAAS,kBAAkB,UAAU;IAC7D,QAAQC,mBAAAA,YAAY;IACpB,UAAUC,mBAAAA,cAAc;IACxB,SAAS,EAAE,UAAU;GACvB,GACA,KACF;EACF;CACF;;;;;;;CAQA,MAAM,YAAY,EAAE,aAA+C;EACjE,MAAM,aAAa,KAAKJ,IAAI,WAAW,SAAS;EAChD,IAAI;GACF,MAAM,WAAW,KAAK;EACxB,SAAS,OAAY;GACnB,MAAM,IAAIE,mBAAAA,YACR;IACE,KAAA,GAAA,qBAAA,oBAAA,CAAwB,SAAS,gBAAgB,UAAU;IAC3D,QAAQC,mBAAAA,YAAY;IACpB,UAAUC,mBAAAA,cAAc;IACxB,SAAS,EAAE,UAAU;GACvB,GACA,KACF;EACF;CACF;;;;;;;;;;;CAYA,MAAM,aAAa,EAAE,WAAW,IAAI,UAA6C;EAC/E,IAAI,CAAC,IACH,MAAM,IAAIF,mBAAAA,YAAY;GACpB,KAAA,GAAA,qBAAA,oBAAA,CAAwB,SAAS,iBAAiB,OAAO;GACzD,MAAM;GACN,QAAQC,mBAAAA,YAAY;GACpB,UAAUC,mBAAAA,cAAc;GACxB,SAAS,EAAE,UAAU;EACvB,CAAC;EAGH,IAAI,CAAC,OAAO,UAAU,CAAC,OAAO,UAC5B,MAAM,IAAIF,mBAAAA,YAAY;GACpB,KAAA,GAAA,qBAAA,oBAAA,CAAwB,SAAS,iBAAiB,YAAY;GAC9D,MAAM;GACN,QAAQC,mBAAAA,YAAY;GACpB,UAAUC,mBAAAA,cAAc;GACxB,SAAS;IACP;IACA;GACF;EACF,CAAC;EAGH,IAAI;GACF,MAAM,aAAa,KAAKJ,IAAI,WAAW,SAAS;GAChD,MAAM,YAAiC,CAAC;GAExC,IAAI,OAAO,QACT,UAAU,UAAU,OAAO;GAG7B,IAAI,OAAO,UACT,UAAU,WAAW,OAAO;GAG9B,MAAM,WAAW,iBAAiB,EAAE,GAAG,GAAG,EAAE,MAAM,UAAU,CAAC;EAC/D,SAAS,OAAY;GACnB,IAAI,iBAAiBE,mBAAAA,aAAa,MAAM;GACxC,MAAM,IAAIA,mBAAAA,YACR;IACE,KAAA,GAAA,qBAAA,oBAAA,CAAwB,SAAS,iBAAiB,kBAAkB;IACpE,QAAQC,mBAAAA,YAAY;IACpB,UAAUC,mBAAAA,cAAc;IACxB,SAAS;KACP;KACA,GAAI,MAAM,EAAE,GAAG;IACjB;GACF,GACA,KACF;EACF;CACF;;;;;;;;CASA,MAAM,aAAa,EAAE,WAAW,MAAyC;EACvE,IAAI;GAEF,MADmB,KAAKJ,IAAI,WAAW,SACxB,CAAC,CAAC,UAAU,EAAE,GAAG,CAAC;EACnC,SAAS,OAAY;GACnB,IAAI,iBAAiBE,mBAAAA,aAAa,MAAM;GACxC,MAAM,IAAIA,mBAAAA,YACR;IACE,KAAA,GAAA,qBAAA,oBAAA,CAAwB,SAAS,iBAAiB,QAAQ;IAC1D,QAAQC,mBAAAA,YAAY;IACpB,UAAUC,mBAAAA,cAAc;IACxB,SAAS;KACP;KACA,GAAI,MAAM,EAAE,GAAG;IACjB;GACF,GACA,KACF;EACF;CACF;CAEA,MAAM,cAAc,EAAE,WAAW,QAAQ,OAA2C;EAClF,MAAM,IAAIF,mBAAAA,YAAY;GACpB,KAAA,GAAA,qBAAA,oBAAA,CAAwB,SAAS,kBAAkB,eAAe;GAClE,MAAM;GACN,QAAQC,mBAAAA,YAAY;GACpB,UAAUC,mBAAAA,cAAc;GACxB,SAAS;IACP;IACA,GAAI,UAAU,EAAE,QAAQ,KAAK,UAAU,MAAM,EAAE;IAC/C,GAAI,OAAO,EAAE,UAAU,IAAI,OAAO;GACpC;EACF,CAAC;CACH;AACF;;;;;;;AC5WA,MAAa,eAAe"}
+{"version":3,"file":"index.cjs","names":["BaseFilterTranslator","MastraVector","#db","DataAPIClient","MastraError","ErrorDomain","ErrorCategory","UUID"],"sources":["../src/vector/filter.ts","../src/vector/index.ts","../src/vector/prompt.ts"],"sourcesContent":["import { BaseFilterTranslator } from '@mastra/core/vector/filter';\nimport type {\n  VectorFilter,\n  OperatorSupport,\n  QueryOperator,\n  OperatorValueMap,\n  LogicalOperatorValueMap,\n  BlacklistedRootOperators,\n} from '@mastra/core/vector/filter';\n\ntype AstraOperatorValueMap = Omit<OperatorValueMap, '$elemMatch' | '$regex' | '$options'> & {\n  $size: number; // Astra-specific\n};\n\ntype AstraLogicalOperatorValueMap = Omit<LogicalOperatorValueMap, '$nor'>;\n\ntype AstraBlacklisted = BlacklistedRootOperators | '$nor' | '$size';\n\nexport type AstraVectorFilter = VectorFilter<\n  keyof AstraOperatorValueMap,\n  AstraOperatorValueMap,\n  AstraLogicalOperatorValueMap,\n  AstraBlacklisted\n>;\n\n/**\n * Translator for Astra DB filter queries.\n * Maintains MongoDB-compatible syntax while ensuring proper validation\n * and normalization of values.\n */\nexport class AstraFilterTranslator extends BaseFilterTranslator<AstraVectorFilter> {\n  protected override getSupportedOperators(): OperatorSupport {\n    return {\n      ...BaseFilterTranslator.DEFAULT_OPERATORS,\n      array: ['$all', '$in', '$nin'],\n      logical: ['$and', '$or', '$not'],\n      regex: [],\n      custom: ['$size'],\n    };\n  }\n\n  translate(filter?: AstraVectorFilter): AstraVectorFilter {\n    if (this.isEmpty(filter)) return filter;\n    this.validateFilter(filter);\n\n    return this.translateNode(filter);\n  }\n\n  private translateNode(node: AstraVectorFilter): any {\n    // Handle primitive values and arrays\n    if (this.isRegex(node)) {\n      throw new Error('Regex is not supported in Astra DB');\n    }\n    if (this.isPrimitive(node) || Array.isArray(node)) {\n      return node;\n    }\n\n    const entries = Object.entries(node as Record<string, any>);\n    const translatedEntries = entries.map(([key, value]) => {\n      // Handle operators\n      if (this.isOperator(key)) {\n        return [key, this.translateOperatorValue(key, value)];\n      }\n\n      // Handle nested paths and objects\n      return [key, this.translateNode(value)];\n    });\n\n    return Object.fromEntries(translatedEntries);\n  }\n\n  private translateOperatorValue(operator: QueryOperator, value: any): any {\n    if (this.isBasicOperator(operator) || this.isNumericOperator(operator)) {\n      return this.normalizeComparisonValue(value);\n    }\n\n    if (this.isArrayOperator(operator) && Array.isArray(value)) {\n      return this.normalizeArrayValues(value);\n    }\n\n    return this.translateNode(value);\n  }\n}\n","import type { Db } from '@datastax/astra-db-ts';\nimport { DataAPIClient, UUID } from '@datastax/astra-db-ts';\nimport { MastraError, ErrorDomain, ErrorCategory } from '@mastra/core/error';\nimport { createVectorErrorId } from '@mastra/core/storage';\nimport { MastraVector } from '@mastra/core/vector';\nimport type {\n  QueryResult,\n  IndexStats,\n  CreateIndexParams,\n  UpsertVectorParams,\n  QueryVectorParams,\n  DescribeIndexParams,\n  DeleteIndexParams,\n  DeleteVectorParams,\n  UpdateVectorParams,\n  DeleteVectorsParams,\n} from '@mastra/core/vector';\nimport type { AstraVectorFilter } from './filter';\nimport { AstraFilterTranslator } from './filter';\n\n// Mastra and Astra DB agree on cosine and euclidean, but Astra DB uses dot_product instead of dotproduct.\nconst metricMap = {\n  cosine: 'cosine',\n  euclidean: 'euclidean',\n  dotproduct: 'dot_product',\n} as const;\n\nexport interface AstraDbOptions {\n  token: string;\n  endpoint: string;\n  keyspace?: string;\n}\n\ntype AstraQueryVectorParams = QueryVectorParams<AstraVectorFilter>;\n\nexport class AstraVector extends MastraVector<AstraVectorFilter> {\n  readonly #db: Db;\n\n  constructor({ token, endpoint, keyspace, id }: AstraDbOptions & { id: string }) {\n    super({ id });\n    const client = new DataAPIClient(token);\n    this.#db = client.db(endpoint, { keyspace });\n  }\n\n  /**\n   * Creates a new collection with the specified configuration.\n   *\n   * @param {string} indexName - The name of the collection to create.\n   * @param {number} dimension - The dimension of the vectors to be stored in the collection.\n   * @param {'cosine' | 'euclidean' | 'dotproduct'} [metric=cosine] - The metric to use to sort vectors in the collection.\n   * @returns {Promise<void>} A promise that resolves when the collection is created.\n   */\n  async createIndex({ indexName, dimension, metric = 'cosine' }: CreateIndexParams): Promise<void> {\n    if (!Number.isInteger(dimension) || dimension <= 0) {\n      throw new MastraError({\n        id: createVectorErrorId('ASTRA', 'CREATE_INDEX', 'INVALID_DIMENSION'),\n        text: 'Dimension must be a positive integer',\n        domain: ErrorDomain.MASTRA_VECTOR,\n        category: ErrorCategory.USER,\n      });\n    }\n    try {\n      await this.#db.createCollection(indexName, {\n        vector: {\n          dimension,\n          metric: metricMap[metric],\n        },\n        checkExists: false,\n      });\n    } catch (error: any) {\n      new MastraError(\n        {\n          id: createVectorErrorId('ASTRA', 'CREATE_INDEX', 'DB_ERROR'),\n          domain: ErrorDomain.MASTRA_VECTOR,\n          category: ErrorCategory.THIRD_PARTY,\n          details: { indexName },\n        },\n        error,\n      );\n    }\n  }\n\n  /**\n   * Inserts or updates vectors in the specified collection.\n   *\n   * @param {string} indexName - The name of the collection to upsert into.\n   * @param {number[][]} vectors - An array of vectors to upsert.\n   * @param {Record<string, any>[]} [metadata] - An optional array of metadata objects corresponding to each vector.\n   * @param {string[]} [ids] - An optional array of IDs corresponding to each vector. If not provided, new IDs will be generated.\n   * @returns {Promise<string[]>} A promise that resolves to an array of IDs of the upserted vectors.\n   */\n  async upsert({ indexName, vectors, metadata, ids }: UpsertVectorParams): Promise<string[]> {\n    const collection = this.#db.collection(indexName);\n\n    // Generate IDs if not provided\n    const vectorIds = ids || vectors.map(() => UUID.v7().toString());\n\n    const records = vectors.map((vector, i) => ({\n      id: vectorIds[i],\n      $vector: vector,\n      metadata: metadata?.[i] || {},\n    }));\n\n    try {\n      await collection.insertMany(records);\n    } catch (error: any) {\n      throw new MastraError(\n        {\n          id: createVectorErrorId('ASTRA', 'UPSERT', 'DB_ERROR'),\n          domain: ErrorDomain.MASTRA_VECTOR,\n          category: ErrorCategory.THIRD_PARTY,\n          details: { indexName },\n        },\n        error,\n      );\n    }\n    return vectorIds;\n  }\n\n  transformFilter(filter?: AstraVectorFilter) {\n    const translator = new AstraFilterTranslator();\n    return translator.translate(filter);\n  }\n\n  /**\n   * Queries the specified collection using a vector and optional filter.\n   *\n   * @param {string} indexName - The name of the collection to query.\n   * @param {number[]} queryVector - The vector to query with.\n   * @param {number} [topK] - The maximum number of results to return.\n   * @param {Record<string, any>} [filter] - An optional filter to apply to the query. For more on filters in Astra DB, see the filtering reference: https://docs.datastax.com/en/astra-db-serverless/api-reference/documents.html#operators\n   * @param {boolean} [includeVectors=false] - Whether to include the vectors in the response.\n   * @returns {Promise<QueryResult[]>} A promise that resolves to an array of query results.\n   */\n  async query({\n    indexName,\n    queryVector,\n    topK = 10,\n    filter,\n    includeVector = false,\n  }: AstraQueryVectorParams): Promise<QueryResult[]> {\n    if (!queryVector) {\n      throw new MastraError({\n        id: createVectorErrorId('ASTRA', 'QUERY', 'MISSING_VECTOR'),\n        text: 'queryVector is required for Astra queries. Metadata-only queries are not supported by this vector store.',\n        domain: ErrorDomain.MASTRA_VECTOR,\n        category: ErrorCategory.USER,\n        details: { indexName },\n      });\n    }\n\n    const collection = this.#db.collection(indexName);\n\n    const translatedFilter = this.transformFilter(filter);\n\n    try {\n      const cursor = collection.find(translatedFilter ?? {}, {\n        sort: { $vector: queryVector },\n        limit: topK,\n        includeSimilarity: true,\n        projection: {\n          $vector: !!includeVector,\n        },\n      });\n\n      const results = await cursor.toArray();\n\n      return results.map(result => ({\n        id: result.id,\n        score: result.$similarity,\n        metadata: result.metadata,\n        ...(includeVector && { vector: result.$vector }),\n      }));\n    } catch (error: any) {\n      throw new MastraError(\n        {\n          id: createVectorErrorId('ASTRA', 'QUERY', 'DB_ERROR'),\n          domain: ErrorDomain.MASTRA_VECTOR,\n          category: ErrorCategory.THIRD_PARTY,\n          details: { indexName },\n        },\n        error,\n      );\n    }\n  }\n\n  /**\n   * Lists all collections in the database.\n   *\n   * @returns {Promise<string[]>} A promise that resolves to an array of collection names.\n   */\n  async listIndexes(): Promise<string[]> {\n    try {\n      return await this.#db.listCollections({ nameOnly: true });\n    } catch (error: any) {\n      throw new MastraError(\n        {\n          id: createVectorErrorId('ASTRA', 'LIST_INDEXES', 'DB_ERROR'),\n          domain: ErrorDomain.MASTRA_VECTOR,\n          category: ErrorCategory.THIRD_PARTY,\n        },\n        error,\n      );\n    }\n  }\n\n  /**\n   * Retrieves statistics about a vector index.\n   *\n   * @param {string} indexName - The name of the index to describe\n   * @returns A promise that resolves to the index statistics including dimension, count and metric\n   */\n  async describeIndex({ indexName }: DescribeIndexParams): Promise<IndexStats> {\n    const collection = this.#db.collection(indexName);\n    try {\n      const optionsPromise = collection.options();\n      const countPromise = collection.countDocuments({}, 100);\n      const [options, count] = await Promise.all([optionsPromise, countPromise]);\n\n      const keys = Object.keys(metricMap) as (keyof typeof metricMap)[];\n      const metric = keys.find(key => metricMap[key] === options.vector?.metric);\n      return {\n        dimension: options.vector?.dimension!,\n        metric,\n        count: count,\n      };\n    } catch (error: any) {\n      if (error instanceof MastraError) throw error;\n      throw new MastraError(\n        {\n          id: createVectorErrorId('ASTRA', 'DESCRIBE_INDEX', 'DB_ERROR'),\n          domain: ErrorDomain.MASTRA_VECTOR,\n          category: ErrorCategory.THIRD_PARTY,\n          details: { indexName },\n        },\n        error,\n      );\n    }\n  }\n\n  /**\n   * Deletes the specified collection.\n   *\n   * @param {string} indexName - The name of the collection to delete.\n   * @returns {Promise<void>} A promise that resolves when the collection is deleted.\n   */\n  async deleteIndex({ indexName }: DeleteIndexParams): Promise<void> {\n    const collection = this.#db.collection(indexName);\n    try {\n      await collection.drop();\n    } catch (error: any) {\n      throw new MastraError(\n        {\n          id: createVectorErrorId('ASTRA', 'DELETE_INDEX', 'DB_ERROR'),\n          domain: ErrorDomain.MASTRA_VECTOR,\n          category: ErrorCategory.THIRD_PARTY,\n          details: { indexName },\n        },\n        error,\n      );\n    }\n  }\n\n  /**\n   * Updates a vector by its ID with the provided vector and/or metadata.\n   * @param indexName - The name of the index containing the vector.\n   * @param id - The ID of the vector to update.\n   * @param update - An object containing the vector and/or metadata to update.\n   * @param update.vector - An optional array of numbers representing the new vector.\n   * @param update.metadata - An optional record containing the new metadata.\n   * @returns A promise that resolves when the update is complete.\n   * @throws Will throw an error if no updates are provided or if the update operation fails.\n   */\n  async updateVector({ indexName, id, update }: UpdateVectorParams): Promise<void> {\n    if (!id) {\n      throw new MastraError({\n        id: createVectorErrorId('ASTRA', 'UPDATE_VECTOR', 'NO_ID'),\n        text: 'id is required for Astra updateVector',\n        domain: ErrorDomain.MASTRA_VECTOR,\n        category: ErrorCategory.USER,\n        details: { indexName },\n      });\n    }\n\n    if (!update.vector && !update.metadata) {\n      throw new MastraError({\n        id: createVectorErrorId('ASTRA', 'UPDATE_VECTOR', 'NO_PAYLOAD'),\n        text: 'No updates provided for vector',\n        domain: ErrorDomain.MASTRA_VECTOR,\n        category: ErrorCategory.USER,\n        details: {\n          indexName,\n          id,\n        },\n      });\n    }\n\n    try {\n      const collection = this.#db.collection(indexName);\n      const updateDoc: Record<string, any> = {};\n\n      if (update.vector) {\n        updateDoc.$vector = update.vector;\n      }\n\n      if (update.metadata) {\n        updateDoc.metadata = update.metadata;\n      }\n\n      await collection.findOneAndUpdate({ id }, { $set: updateDoc });\n    } catch (error: any) {\n      if (error instanceof MastraError) throw error;\n      throw new MastraError(\n        {\n          id: createVectorErrorId('ASTRA', 'UPDATE_VECTOR', 'FAILED_UNHANDLED'),\n          domain: ErrorDomain.MASTRA_VECTOR,\n          category: ErrorCategory.THIRD_PARTY,\n          details: {\n            indexName,\n            ...(id && { id }),\n          },\n        },\n        error,\n      );\n    }\n  }\n\n  /**\n   * Deletes a vector by its ID.\n   * @param indexName - The name of the index containing the vector.\n   * @param id - The ID of the vector to delete.\n   * @returns A promise that resolves when the deletion is complete.\n   * @throws Will throw an error if the deletion operation fails.\n   */\n  async deleteVector({ indexName, id }: DeleteVectorParams): Promise<void> {\n    try {\n      const collection = this.#db.collection(indexName);\n      await collection.deleteOne({ id });\n    } catch (error: any) {\n      if (error instanceof MastraError) throw error;\n      throw new MastraError(\n        {\n          id: createVectorErrorId('ASTRA', 'DELETE_VECTOR', 'FAILED'),\n          domain: ErrorDomain.MASTRA_VECTOR,\n          category: ErrorCategory.THIRD_PARTY,\n          details: {\n            indexName,\n            ...(id && { id }),\n          },\n        },\n        error,\n      );\n    }\n  }\n\n  async deleteVectors({ indexName, filter, ids }: DeleteVectorsParams): Promise<void> {\n    throw new MastraError({\n      id: createVectorErrorId('ASTRA', 'DELETE_VECTORS', 'NOT_SUPPORTED'),\n      text: 'deleteVectors is not yet implemented for Astra vector store',\n      domain: ErrorDomain.MASTRA_VECTOR,\n      category: ErrorCategory.SYSTEM,\n      details: {\n        indexName,\n        ...(filter && { filter: JSON.stringify(filter) }),\n        ...(ids && { idsCount: ids.length }),\n      },\n    });\n  }\n}\n","/**\n * Vector store specific prompt that details supported operators and examples.\n * This prompt helps users construct valid filters for Astra Vector.\n */\nexport const ASTRA_PROMPT = `When querying Astra, you can ONLY use the operators listed below. Any other operators will be rejected.\nImportant: Don't explain how to construct the filter - use the specified operators and fields to search the content and return relevant results.\nIf a user tries to give an explicit operator that is not supported, reject the filter entirely and let them know that the operator is not supported.\n\nBasic Comparison Operators:\n- $eq: Exact match (default when using field: value)\n  Example: { \"category\": \"electronics\" }\n- $ne: Not equal\n  Example: { \"category\": { \"$ne\": \"electronics\" } }\n- $gt: Greater than\n  Example: { \"price\": { \"$gt\": 100 } }\n- $gte: Greater than or equal\n  Example: { \"price\": { \"$gte\": 100 } }\n- $lt: Less than\n  Example: { \"price\": { \"$lt\": 100 } }\n- $lte: Less than or equal\n  Example: { \"price\": { \"$lte\": 100 } }\n\nArray Operators:\n- $in: Match any value in array\n  Example: { \"category\": { \"$in\": [\"electronics\", \"books\"] } }\n- $nin: Does not match any value in array\n  Example: { \"category\": { \"$nin\": [\"electronics\", \"books\"] } }\n- $all: Match all values in array\n  Example: { \"tags\": { \"$all\": [\"premium\", \"sale\"] } }\n\nLogical Operators:\n- $and: Logical AND (can be implicit or explicit)\n  Implicit Example: { \"price\": { \"$gt\": 100 }, \"category\": \"electronics\" }\n  Explicit Example: { \"$and\": [{ \"price\": { \"$gt\": 100 } }, { \"category\": \"electronics\" }] }\n- $or: Logical OR\n  Example: { \"$or\": [{ \"price\": { \"$lt\": 50 } }, { \"category\": \"books\" }] }\n- $not: Logical NOT\n  Example: { \"$not\": { \"category\": \"electronics\" } }\n\nElement Operators:\n- $exists: Check if field exists\n  Example: { \"rating\": { \"$exists\": true } }\n\nSpecial Operators:\n- $size: Array length check\n  Example: { \"tags\": { \"$size\": 2 } }\n\nRestrictions:\n- Regex patterns are not supported\n- Only $and, $or, and $not logical operators are supported\n- Nested fields are supported using dot notation\n- Multiple conditions on the same field are supported with both implicit and explicit $and\n- Empty arrays in $in/$nin will return no results\n- A non-empty array is required for $all operator\n- Only logical operators ($and, $or, $not) can be used at the top level\n- All other operators must be used within a field condition\n  Valid: { \"field\": { \"$gt\": 100 } }\n  Valid: { \"$and\": [...] }\n  Invalid: { \"$gt\": 100 }\n- Logical operators must contain field conditions, not direct operators\n  Valid: { \"$and\": [{ \"field\": { \"$gt\": 100 } }] }\n  Invalid: { \"$and\": [{ \"$gt\": 100 }] }\n- $not operator:\n  - Must be an object\n  - Cannot be empty\n  - Can be used at field level or top level\n  - Valid: { \"$not\": { \"field\": \"value\" } }\n  - Valid: { \"field\": { \"$not\": { \"$eq\": \"value\" } } }\n- Other logical operators ($and, $or):\n  - Can only be used at top level or nested within other logical operators\n  - Can not be used on a field level, or be nested inside a field\n  - Can not be used inside an operator\n  - Valid: { \"$and\": [{ \"field\": { \"$gt\": 100 } }] }\n  - Valid: { \"$or\": [{ \"$and\": [{ \"field\": { \"$gt\": 100 } }] }] }\n  - Invalid: { \"field\": { \"$and\": [{ \"$gt\": 100 }] } }\n  - Invalid: { \"field\": { \"$or\": [{ \"$gt\": 100 }] } }\n  - Invalid: { \"field\": { \"$gt\": { \"$and\": [{...}] } } }\n\nExample Complex Query:\n{\n  \"$and\": [\n    { \"category\": { \"$in\": [\"electronics\", \"computers\"] } },\n    { \"price\": { \"$gte\": 100, \"$lte\": 1000 } },\n    { \"tags\": { \"$all\": [\"premium\"] } },\n    { \"rating\": { \"$exists\": true, \"$gt\": 4 } },\n    { \"$or\": [\n      { \"stock\": { \"$gt\": 0 } },\n      { \"preorder\": true }\n    ]}\n  ]\n}`;\n"],"mappings":";;;;;;;;;;;;AA8BA,IAAa,wBAAb,cAA2CA,2BAAAA,qBAAwC;CACjF,wBAA4D;EAC1D,OAAO;GACL,GAAGA,2BAAAA,qBAAqB;GACxB,OAAO;IAAC;IAAQ;IAAO;GAAM;GAC7B,SAAS;IAAC;IAAQ;IAAO;GAAM;GAC/B,OAAO,CAAC;GACR,QAAQ,CAAC,OAAO;EAClB;CACF;CAEA,UAAU,QAA+C;EACvD,IAAI,KAAK,QAAQ,MAAM,GAAG,OAAO;EACjC,KAAK,eAAe,MAAM;EAE1B,OAAO,KAAK,cAAc,MAAM;CAClC;CAEA,cAAsB,MAA8B;EAElD,IAAI,KAAK,QAAQ,IAAI,GACnB,MAAM,IAAI,MAAM,oCAAoC;EAEtD,IAAI,KAAK,YAAY,IAAI,KAAK,MAAM,QAAQ,IAAI,GAC9C,OAAO;EAIT,MAAM,oBADU,OAAO,QAAQ,IACC,CAAC,CAAC,KAAK,CAAC,KAAK,WAAW;GAEtD,IAAI,KAAK,WAAW,GAAG,GACrB,OAAO,CAAC,KAAK,KAAK,uBAAuB,KAAK,KAAK,CAAC;GAItD,OAAO,CAAC,KAAK,KAAK,cAAc,KAAK,CAAC;EACxC,CAAC;EAED,OAAO,OAAO,YAAY,iBAAiB;CAC7C;CAEA,uBAA+B,UAAyB,OAAiB;EACvE,IAAI,KAAK,gBAAgB,QAAQ,KAAK,KAAK,kBAAkB,QAAQ,GACnE,OAAO,KAAK,yBAAyB,KAAK;EAG5C,IAAI,KAAK,gBAAgB,QAAQ,KAAK,MAAM,QAAQ,KAAK,GACvD,OAAO,KAAK,qBAAqB,KAAK;EAGxC,OAAO,KAAK,cAAc,KAAK;CACjC;AACF;;;AC7DA,MAAM,YAAY;CAChB,QAAQ;CACR,WAAW;CACX,YAAY;AACd;AAUA,IAAa,cAAb,cAAiCC,oBAAAA,aAAgC;CAC/D;CAEA,YAAY,EAAE,OAAO,UAAU,UAAU,MAAuC;EAC9E,MAAM,EAAE,GAAG,CAAC;EACZ,MAAM,SAAS,IAAIE,sBAAAA,cAAc,KAAK;EACtC,KAAKD,MAAM,OAAO,GAAG,UAAU,EAAE,SAAS,CAAC;CAC7C;;;;;;;;;CAUA,MAAM,YAAY,EAAE,WAAW,WAAW,SAAS,YAA8C;EAC/F,IAAI,CAAC,OAAO,UAAU,SAAS,KAAK,aAAa,GAC/C,MAAM,IAAIE,mBAAAA,YAAY;GACpB,KAAA,GAAA,qBAAA,oBAAA,CAAwB,SAAS,gBAAgB,mBAAmB;GACpE,MAAM;GACN,QAAQC,mBAAAA,YAAY;GACpB,UAAUC,mBAAAA,cAAc;EAC1B,CAAC;EAEH,IAAI;GACF,MAAM,KAAKJ,IAAI,iBAAiB,WAAW;IACzC,QAAQ;KACN;KACA,QAAQ,UAAU;IACpB;IACA,aAAa;GACf,CAAC;EACH,SAAS,OAAY;GACnB,IAAIE,mBAAAA,YACF;IACE,KAAA,GAAA,qBAAA,oBAAA,CAAwB,SAAS,gBAAgB,UAAU;IAC3D,QAAQC,mBAAAA,YAAY;IACpB,UAAUC,mBAAAA,cAAc;IACxB,SAAS,EAAE,UAAU;GACvB,GACA,KACF;EACF;CACF;;;;;;;;;;CAWA,MAAM,OAAO,EAAE,WAAW,SAAS,UAAU,OAA8C;EACzF,MAAM,aAAa,KAAKJ,IAAI,WAAW,SAAS;EAGhD,MAAM,YAAY,OAAO,QAAQ,UAAUK,sBAAAA,KAAK,GAAG,CAAC,CAAC,SAAS,CAAC;EAE/D,MAAM,UAAU,QAAQ,KAAK,QAAQ,OAAO;GAC1C,IAAI,UAAU;GACd,SAAS;GACT,UAAU,WAAW,MAAM,CAAC;EAC9B,EAAE;EAEF,IAAI;GACF,MAAM,WAAW,WAAW,OAAO;EACrC,SAAS,OAAY;GACnB,MAAM,IAAIH,mBAAAA,YACR;IACE,KAAA,GAAA,qBAAA,oBAAA,CAAwB,SAAS,UAAU,UAAU;IACrD,QAAQC,mBAAAA,YAAY;IACpB,UAAUC,mBAAAA,cAAc;IACxB,SAAS,EAAE,UAAU;GACvB,GACA,KACF;EACF;EACA,OAAO;CACT;CAEA,gBAAgB,QAA4B;EAE1C,OAAO,IADgB,sBACP,CAAC,CAAC,UAAU,MAAM;CACpC;;;;;;;;;;;CAYA,MAAM,MAAM,EACV,WACA,aACA,OAAO,IACP,QACA,gBAAgB,SACiC;EACjD,IAAI,CAAC,aACH,MAAM,IAAIF,mBAAAA,YAAY;GACpB,KAAA,GAAA,qBAAA,oBAAA,CAAwB,SAAS,SAAS,gBAAgB;GAC1D,MAAM;GACN,QAAQC,mBAAAA,YAAY;GACpB,UAAUC,mBAAAA,cAAc;GACxB,SAAS,EAAE,UAAU;EACvB,CAAC;EAGH,MAAM,aAAa,KAAKJ,IAAI,WAAW,SAAS;EAEhD,MAAM,mBAAmB,KAAK,gBAAgB,MAAM;EAEpD,IAAI;GAYF,QAAO,MAXQ,WAAW,KAAK,oBAAoB,CAAC,GAAG;IACrD,MAAM,EAAE,SAAS,YAAY;IAC7B,OAAO;IACP,mBAAmB;IACnB,YAAY,EACV,SAAS,CAAC,CAAC,cACb;GACF,CAE2B,CAAC,CAAC,QAAQ,EAAA,CAEtB,KAAI,YAAW;IAC5B,IAAI,OAAO;IACX,OAAO,OAAO;IACd,UAAU,OAAO;IACjB,GAAI,iBAAiB,EAAE,QAAQ,OAAO,QAAQ;GAChD,EAAE;EACJ,SAAS,OAAY;GACnB,MAAM,IAAIE,mBAAAA,YACR;IACE,KAAA,GAAA,qBAAA,oBAAA,CAAwB,SAAS,SAAS,UAAU;IACpD,QAAQC,mBAAAA,YAAY;IACpB,UAAUC,mBAAAA,cAAc;IACxB,SAAS,EAAE,UAAU;GACvB,GACA,KACF;EACF;CACF;;;;;;CAOA,MAAM,cAAiC;EACrC,IAAI;GACF,OAAO,MAAM,KAAKJ,IAAI,gBAAgB,EAAE,UAAU,KAAK,CAAC;EAC1D,SAAS,OAAY;GACnB,MAAM,IAAIE,mBAAAA,YACR;IACE,KAAA,GAAA,qBAAA,oBAAA,CAAwB,SAAS,gBAAgB,UAAU;IAC3D,QAAQC,mBAAAA,YAAY;IACpB,UAAUC,mBAAAA,cAAc;GAC1B,GACA,KACF;EACF;CACF;;;;;;;CAQA,MAAM,cAAc,EAAE,aAAuD;EAC3E,MAAM,aAAa,KAAKJ,IAAI,WAAW,SAAS;EAChD,IAAI;GACF,MAAM,iBAAiB,WAAW,QAAQ;GAC1C,MAAM,eAAe,WAAW,eAAe,CAAC,GAAG,GAAG;GACtD,MAAM,CAAC,SAAS,SAAS,MAAM,QAAQ,IAAI,CAAC,gBAAgB,YAAY,CAAC;GAGzE,MAAM,SADO,OAAO,KAAK,SACP,CAAC,CAAC,MAAK,QAAO,UAAU,SAAS,QAAQ,QAAQ,MAAM;GACzE,OAAO;IACL,WAAW,QAAQ,QAAQ;IAC3B;IACO;GACT;EACF,SAAS,OAAY;GACnB,IAAI,iBAAiBE,mBAAAA,aAAa,MAAM;GACxC,MAAM,IAAIA,mBAAAA,YACR;IACE,KAAA,GAAA,qBAAA,oBAAA,CAAwB,SAAS,kBAAkB,UAAU;IAC7D,QAAQC,mBAAAA,YAAY;IACpB,UAAUC,mBAAAA,cAAc;IACxB,SAAS,EAAE,UAAU;GACvB,GACA,KACF;EACF;CACF;;;;;;;CAQA,MAAM,YAAY,EAAE,aAA+C;EACjE,MAAM,aAAa,KAAKJ,IAAI,WAAW,SAAS;EAChD,IAAI;GACF,MAAM,WAAW,KAAK;EACxB,SAAS,OAAY;GACnB,MAAM,IAAIE,mBAAAA,YACR;IACE,KAAA,GAAA,qBAAA,oBAAA,CAAwB,SAAS,gBAAgB,UAAU;IAC3D,QAAQC,mBAAAA,YAAY;IACpB,UAAUC,mBAAAA,cAAc;IACxB,SAAS,EAAE,UAAU;GACvB,GACA,KACF;EACF;CACF;;;;;;;;;;;CAYA,MAAM,aAAa,EAAE,WAAW,IAAI,UAA6C;EAC/E,IAAI,CAAC,IACH,MAAM,IAAIF,mBAAAA,YAAY;GACpB,KAAA,GAAA,qBAAA,oBAAA,CAAwB,SAAS,iBAAiB,OAAO;GACzD,MAAM;GACN,QAAQC,mBAAAA,YAAY;GACpB,UAAUC,mBAAAA,cAAc;GACxB,SAAS,EAAE,UAAU;EACvB,CAAC;EAGH,IAAI,CAAC,OAAO,UAAU,CAAC,OAAO,UAC5B,MAAM,IAAIF,mBAAAA,YAAY;GACpB,KAAA,GAAA,qBAAA,oBAAA,CAAwB,SAAS,iBAAiB,YAAY;GAC9D,MAAM;GACN,QAAQC,mBAAAA,YAAY;GACpB,UAAUC,mBAAAA,cAAc;GACxB,SAAS;IACP;IACA;GACF;EACF,CAAC;EAGH,IAAI;GACF,MAAM,aAAa,KAAKJ,IAAI,WAAW,SAAS;GAChD,MAAM,YAAiC,CAAC;GAExC,IAAI,OAAO,QACT,UAAU,UAAU,OAAO;GAG7B,IAAI,OAAO,UACT,UAAU,WAAW,OAAO;GAG9B,MAAM,WAAW,iBAAiB,EAAE,GAAG,GAAG,EAAE,MAAM,UAAU,CAAC;EAC/D,SAAS,OAAY;GACnB,IAAI,iBAAiBE,mBAAAA,aAAa,MAAM;GACxC,MAAM,IAAIA,mBAAAA,YACR;IACE,KAAA,GAAA,qBAAA,oBAAA,CAAwB,SAAS,iBAAiB,kBAAkB;IACpE,QAAQC,mBAAAA,YAAY;IACpB,UAAUC,mBAAAA,cAAc;IACxB,SAAS;KACP;KACA,GAAI,MAAM,EAAE,GAAG;IACjB;GACF,GACA,KACF;EACF;CACF;;;;;;;;CASA,MAAM,aAAa,EAAE,WAAW,MAAyC;EACvE,IAAI;GAEF,MADmB,KAAKJ,IAAI,WAAW,SACxB,CAAC,CAAC,UAAU,EAAE,GAAG,CAAC;EACnC,SAAS,OAAY;GACnB,IAAI,iBAAiBE,mBAAAA,aAAa,MAAM;GACxC,MAAM,IAAIA,mBAAAA,YACR;IACE,KAAA,GAAA,qBAAA,oBAAA,CAAwB,SAAS,iBAAiB,QAAQ;IAC1D,QAAQC,mBAAAA,YAAY;IACpB,UAAUC,mBAAAA,cAAc;IACxB,SAAS;KACP;KACA,GAAI,MAAM,EAAE,GAAG;IACjB;GACF,GACA,KACF;EACF;CACF;CAEA,MAAM,cAAc,EAAE,WAAW,QAAQ,OAA2C;EAClF,MAAM,IAAIF,mBAAAA,YAAY;GACpB,KAAA,GAAA,qBAAA,oBAAA,CAAwB,SAAS,kBAAkB,eAAe;GAClE,MAAM;GACN,QAAQC,mBAAAA,YAAY;GACpB,UAAUC,mBAAAA,cAAc;GACxB,SAAS;IACP;IACA,GAAI,UAAU,EAAE,QAAQ,KAAK,UAAU,MAAM,EAAE;IAC/C,GAAI,OAAO,EAAE,UAAU,IAAI,OAAO;GACpC;EACF,CAAC;CACH;AACF;;;;;;;AC5WA,MAAa,eAAe"}

package/dist/index.js

@@ -150,7 +150,7 @@ var AstraVector = class extends MastraVector {
 				sort: { $vector: queryVector },
 				limit: topK,
 				includeSimilarity: true,
-				projection: { $vector: includeVector ? true : false }
+				projection: { $vector: !!includeVector }
 			}).toArray()).map((result) => ({
 				id: result.id,
 				score: result.$similarity,

package/dist/index.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.js","names":["#db"],"sources":["../src/vector/filter.ts","../src/vector/index.ts","../src/vector/prompt.ts"],"sourcesContent":["import { BaseFilterTranslator } from '@mastra/core/vector/filter';\nimport type {\n  VectorFilter,\n  OperatorSupport,\n  QueryOperator,\n  OperatorValueMap,\n  LogicalOperatorValueMap,\n  BlacklistedRootOperators,\n} from '@mastra/core/vector/filter';\n\ntype AstraOperatorValueMap = Omit<OperatorValueMap, '$elemMatch' | '$regex' | '$options'> & {\n  $size: number; // Astra-specific\n};\n\ntype AstraLogicalOperatorValueMap = Omit<LogicalOperatorValueMap, '$nor'>;\n\ntype AstraBlacklisted = BlacklistedRootOperators | '$nor' | '$size';\n\nexport type AstraVectorFilter = VectorFilter<\n  keyof AstraOperatorValueMap,\n  AstraOperatorValueMap,\n  AstraLogicalOperatorValueMap,\n  AstraBlacklisted\n>;\n\n/**\n * Translator for Astra DB filter queries.\n * Maintains MongoDB-compatible syntax while ensuring proper validation\n * and normalization of values.\n */\nexport class AstraFilterTranslator extends BaseFilterTranslator<AstraVectorFilter> {\n  protected override getSupportedOperators(): OperatorSupport {\n    return {\n      ...BaseFilterTranslator.DEFAULT_OPERATORS,\n      array: ['$all', '$in', '$nin'],\n      logical: ['$and', '$or', '$not'],\n      regex: [],\n      custom: ['$size'],\n    };\n  }\n\n  translate(filter?: AstraVectorFilter): AstraVectorFilter {\n    if (this.isEmpty(filter)) return filter;\n    this.validateFilter(filter);\n\n    return this.translateNode(filter);\n  }\n\n  private translateNode(node: AstraVectorFilter): any {\n    // Handle primitive values and arrays\n    if (this.isRegex(node)) {\n      throw new Error('Regex is not supported in Astra DB');\n    }\n    if (this.isPrimitive(node) || Array.isArray(node)) {\n      return node;\n    }\n\n    const entries = Object.entries(node as Record<string, any>);\n    const translatedEntries = entries.map(([key, value]) => {\n      // Handle operators\n      if (this.isOperator(key)) {\n        return [key, this.translateOperatorValue(key, value)];\n      }\n\n      // Handle nested paths and objects\n      return [key, this.translateNode(value)];\n    });\n\n    return Object.fromEntries(translatedEntries);\n  }\n\n  private translateOperatorValue(operator: QueryOperator, value: any): any {\n    if (this.isBasicOperator(operator) || this.isNumericOperator(operator)) {\n      return this.normalizeComparisonValue(value);\n    }\n\n    if (this.isArrayOperator(operator) && Array.isArray(value)) {\n      return this.normalizeArrayValues(value);\n    }\n\n    return this.translateNode(value);\n  }\n}\n","import type { Db } from '@datastax/astra-db-ts';\nimport { DataAPIClient, UUID } from '@datastax/astra-db-ts';\nimport { MastraError, ErrorDomain, ErrorCategory } from '@mastra/core/error';\nimport { createVectorErrorId } from '@mastra/core/storage';\nimport { MastraVector } from '@mastra/core/vector';\nimport type {\n  QueryResult,\n  IndexStats,\n  CreateIndexParams,\n  UpsertVectorParams,\n  QueryVectorParams,\n  DescribeIndexParams,\n  DeleteIndexParams,\n  DeleteVectorParams,\n  UpdateVectorParams,\n  DeleteVectorsParams,\n} from '@mastra/core/vector';\nimport type { AstraVectorFilter } from './filter';\nimport { AstraFilterTranslator } from './filter';\n\n// Mastra and Astra DB agree on cosine and euclidean, but Astra DB uses dot_product instead of dotproduct.\nconst metricMap = {\n  cosine: 'cosine',\n  euclidean: 'euclidean',\n  dotproduct: 'dot_product',\n} as const;\n\nexport interface AstraDbOptions {\n  token: string;\n  endpoint: string;\n  keyspace?: string;\n}\n\ntype AstraQueryVectorParams = QueryVectorParams<AstraVectorFilter>;\n\nexport class AstraVector extends MastraVector<AstraVectorFilter> {\n  readonly #db: Db;\n\n  constructor({ token, endpoint, keyspace, id }: AstraDbOptions & { id: string }) {\n    super({ id });\n    const client = new DataAPIClient(token);\n    this.#db = client.db(endpoint, { keyspace });\n  }\n\n  /**\n   * Creates a new collection with the specified configuration.\n   *\n   * @param {string} indexName - The name of the collection to create.\n   * @param {number} dimension - The dimension of the vectors to be stored in the collection.\n   * @param {'cosine' | 'euclidean' | 'dotproduct'} [metric=cosine] - The metric to use to sort vectors in the collection.\n   * @returns {Promise<void>} A promise that resolves when the collection is created.\n   */\n  async createIndex({ indexName, dimension, metric = 'cosine' }: CreateIndexParams): Promise<void> {\n    if (!Number.isInteger(dimension) || dimension <= 0) {\n      throw new MastraError({\n        id: createVectorErrorId('ASTRA', 'CREATE_INDEX', 'INVALID_DIMENSION'),\n        text: 'Dimension must be a positive integer',\n        domain: ErrorDomain.MASTRA_VECTOR,\n        category: ErrorCategory.USER,\n      });\n    }\n    try {\n      await this.#db.createCollection(indexName, {\n        vector: {\n          dimension,\n          metric: metricMap[metric],\n        },\n        checkExists: false,\n      });\n    } catch (error: any) {\n      new MastraError(\n        {\n          id: createVectorErrorId('ASTRA', 'CREATE_INDEX', 'DB_ERROR'),\n          domain: ErrorDomain.MASTRA_VECTOR,\n          category: ErrorCategory.THIRD_PARTY,\n          details: { indexName },\n        },\n        error,\n      );\n    }\n  }\n\n  /**\n   * Inserts or updates vectors in the specified collection.\n   *\n   * @param {string} indexName - The name of the collection to upsert into.\n   * @param {number[][]} vectors - An array of vectors to upsert.\n   * @param {Record<string, any>[]} [metadata] - An optional array of metadata objects corresponding to each vector.\n   * @param {string[]} [ids] - An optional array of IDs corresponding to each vector. If not provided, new IDs will be generated.\n   * @returns {Promise<string[]>} A promise that resolves to an array of IDs of the upserted vectors.\n   */\n  async upsert({ indexName, vectors, metadata, ids }: UpsertVectorParams): Promise<string[]> {\n    const collection = this.#db.collection(indexName);\n\n    // Generate IDs if not provided\n    const vectorIds = ids || vectors.map(() => UUID.v7().toString());\n\n    const records = vectors.map((vector, i) => ({\n      id: vectorIds[i],\n      $vector: vector,\n      metadata: metadata?.[i] || {},\n    }));\n\n    try {\n      await collection.insertMany(records);\n    } catch (error: any) {\n      throw new MastraError(\n        {\n          id: createVectorErrorId('ASTRA', 'UPSERT', 'DB_ERROR'),\n          domain: ErrorDomain.MASTRA_VECTOR,\n          category: ErrorCategory.THIRD_PARTY,\n          details: { indexName },\n        },\n        error,\n      );\n    }\n    return vectorIds;\n  }\n\n  transformFilter(filter?: AstraVectorFilter) {\n    const translator = new AstraFilterTranslator();\n    return translator.translate(filter);\n  }\n\n  /**\n   * Queries the specified collection using a vector and optional filter.\n   *\n   * @param {string} indexName - The name of the collection to query.\n   * @param {number[]} queryVector - The vector to query with.\n   * @param {number} [topK] - The maximum number of results to return.\n   * @param {Record<string, any>} [filter] - An optional filter to apply to the query. For more on filters in Astra DB, see the filtering reference: https://docs.datastax.com/en/astra-db-serverless/api-reference/documents.html#operators\n   * @param {boolean} [includeVectors=false] - Whether to include the vectors in the response.\n   * @returns {Promise<QueryResult[]>} A promise that resolves to an array of query results.\n   */\n  async query({\n    indexName,\n    queryVector,\n    topK = 10,\n    filter,\n    includeVector = false,\n  }: AstraQueryVectorParams): Promise<QueryResult[]> {\n    if (!queryVector) {\n      throw new MastraError({\n        id: createVectorErrorId('ASTRA', 'QUERY', 'MISSING_VECTOR'),\n        text: 'queryVector is required for Astra queries. Metadata-only queries are not supported by this vector store.',\n        domain: ErrorDomain.MASTRA_VECTOR,\n        category: ErrorCategory.USER,\n        details: { indexName },\n      });\n    }\n\n    const collection = this.#db.collection(indexName);\n\n    const translatedFilter = this.transformFilter(filter);\n\n    try {\n      const cursor = collection.find(translatedFilter ?? {}, {\n        sort: { $vector: queryVector },\n        limit: topK,\n        includeSimilarity: true,\n        projection: {\n          $vector: includeVector ? true : false,\n        },\n      });\n\n      const results = await cursor.toArray();\n\n      return results.map(result => ({\n        id: result.id,\n        score: result.$similarity,\n        metadata: result.metadata,\n        ...(includeVector && { vector: result.$vector }),\n      }));\n    } catch (error: any) {\n      throw new MastraError(\n        {\n          id: createVectorErrorId('ASTRA', 'QUERY', 'DB_ERROR'),\n          domain: ErrorDomain.MASTRA_VECTOR,\n          category: ErrorCategory.THIRD_PARTY,\n          details: { indexName },\n        },\n        error,\n      );\n    }\n  }\n\n  /**\n   * Lists all collections in the database.\n   *\n   * @returns {Promise<string[]>} A promise that resolves to an array of collection names.\n   */\n  async listIndexes(): Promise<string[]> {\n    try {\n      return await this.#db.listCollections({ nameOnly: true });\n    } catch (error: any) {\n      throw new MastraError(\n        {\n          id: createVectorErrorId('ASTRA', 'LIST_INDEXES', 'DB_ERROR'),\n          domain: ErrorDomain.MASTRA_VECTOR,\n          category: ErrorCategory.THIRD_PARTY,\n        },\n        error,\n      );\n    }\n  }\n\n  /**\n   * Retrieves statistics about a vector index.\n   *\n   * @param {string} indexName - The name of the index to describe\n   * @returns A promise that resolves to the index statistics including dimension, count and metric\n   */\n  async describeIndex({ indexName }: DescribeIndexParams): Promise<IndexStats> {\n    const collection = this.#db.collection(indexName);\n    try {\n      const optionsPromise = collection.options();\n      const countPromise = collection.countDocuments({}, 100);\n      const [options, count] = await Promise.all([optionsPromise, countPromise]);\n\n      const keys = Object.keys(metricMap) as (keyof typeof metricMap)[];\n      const metric = keys.find(key => metricMap[key] === options.vector?.metric);\n      return {\n        dimension: options.vector?.dimension!,\n        metric,\n        count: count,\n      };\n    } catch (error: any) {\n      if (error instanceof MastraError) throw error;\n      throw new MastraError(\n        {\n          id: createVectorErrorId('ASTRA', 'DESCRIBE_INDEX', 'DB_ERROR'),\n          domain: ErrorDomain.MASTRA_VECTOR,\n          category: ErrorCategory.THIRD_PARTY,\n          details: { indexName },\n        },\n        error,\n      );\n    }\n  }\n\n  /**\n   * Deletes the specified collection.\n   *\n   * @param {string} indexName - The name of the collection to delete.\n   * @returns {Promise<void>} A promise that resolves when the collection is deleted.\n   */\n  async deleteIndex({ indexName }: DeleteIndexParams): Promise<void> {\n    const collection = this.#db.collection(indexName);\n    try {\n      await collection.drop();\n    } catch (error: any) {\n      throw new MastraError(\n        {\n          id: createVectorErrorId('ASTRA', 'DELETE_INDEX', 'DB_ERROR'),\n          domain: ErrorDomain.MASTRA_VECTOR,\n          category: ErrorCategory.THIRD_PARTY,\n          details: { indexName },\n        },\n        error,\n      );\n    }\n  }\n\n  /**\n   * Updates a vector by its ID with the provided vector and/or metadata.\n   * @param indexName - The name of the index containing the vector.\n   * @param id - The ID of the vector to update.\n   * @param update - An object containing the vector and/or metadata to update.\n   * @param update.vector - An optional array of numbers representing the new vector.\n   * @param update.metadata - An optional record containing the new metadata.\n   * @returns A promise that resolves when the update is complete.\n   * @throws Will throw an error if no updates are provided or if the update operation fails.\n   */\n  async updateVector({ indexName, id, update }: UpdateVectorParams): Promise<void> {\n    if (!id) {\n      throw new MastraError({\n        id: createVectorErrorId('ASTRA', 'UPDATE_VECTOR', 'NO_ID'),\n        text: 'id is required for Astra updateVector',\n        domain: ErrorDomain.MASTRA_VECTOR,\n        category: ErrorCategory.USER,\n        details: { indexName },\n      });\n    }\n\n    if (!update.vector && !update.metadata) {\n      throw new MastraError({\n        id: createVectorErrorId('ASTRA', 'UPDATE_VECTOR', 'NO_PAYLOAD'),\n        text: 'No updates provided for vector',\n        domain: ErrorDomain.MASTRA_VECTOR,\n        category: ErrorCategory.USER,\n        details: {\n          indexName,\n          id,\n        },\n      });\n    }\n\n    try {\n      const collection = this.#db.collection(indexName);\n      const updateDoc: Record<string, any> = {};\n\n      if (update.vector) {\n        updateDoc.$vector = update.vector;\n      }\n\n      if (update.metadata) {\n        updateDoc.metadata = update.metadata;\n      }\n\n      await collection.findOneAndUpdate({ id }, { $set: updateDoc });\n    } catch (error: any) {\n      if (error instanceof MastraError) throw error;\n      throw new MastraError(\n        {\n          id: createVectorErrorId('ASTRA', 'UPDATE_VECTOR', 'FAILED_UNHANDLED'),\n          domain: ErrorDomain.MASTRA_VECTOR,\n          category: ErrorCategory.THIRD_PARTY,\n          details: {\n            indexName,\n            ...(id && { id }),\n          },\n        },\n        error,\n      );\n    }\n  }\n\n  /**\n   * Deletes a vector by its ID.\n   * @param indexName - The name of the index containing the vector.\n   * @param id - The ID of the vector to delete.\n   * @returns A promise that resolves when the deletion is complete.\n   * @throws Will throw an error if the deletion operation fails.\n   */\n  async deleteVector({ indexName, id }: DeleteVectorParams): Promise<void> {\n    try {\n      const collection = this.#db.collection(indexName);\n      await collection.deleteOne({ id });\n    } catch (error: any) {\n      if (error instanceof MastraError) throw error;\n      throw new MastraError(\n        {\n          id: createVectorErrorId('ASTRA', 'DELETE_VECTOR', 'FAILED'),\n          domain: ErrorDomain.MASTRA_VECTOR,\n          category: ErrorCategory.THIRD_PARTY,\n          details: {\n            indexName,\n            ...(id && { id }),\n          },\n        },\n        error,\n      );\n    }\n  }\n\n  async deleteVectors({ indexName, filter, ids }: DeleteVectorsParams): Promise<void> {\n    throw new MastraError({\n      id: createVectorErrorId('ASTRA', 'DELETE_VECTORS', 'NOT_SUPPORTED'),\n      text: 'deleteVectors is not yet implemented for Astra vector store',\n      domain: ErrorDomain.MASTRA_VECTOR,\n      category: ErrorCategory.SYSTEM,\n      details: {\n        indexName,\n        ...(filter && { filter: JSON.stringify(filter) }),\n        ...(ids && { idsCount: ids.length }),\n      },\n    });\n  }\n}\n","/**\n * Vector store specific prompt that details supported operators and examples.\n * This prompt helps users construct valid filters for Astra Vector.\n */\nexport const ASTRA_PROMPT = `When querying Astra, you can ONLY use the operators listed below. Any other operators will be rejected.\nImportant: Don't explain how to construct the filter - use the specified operators and fields to search the content and return relevant results.\nIf a user tries to give an explicit operator that is not supported, reject the filter entirely and let them know that the operator is not supported.\n\nBasic Comparison Operators:\n- $eq: Exact match (default when using field: value)\n  Example: { \"category\": \"electronics\" }\n- $ne: Not equal\n  Example: { \"category\": { \"$ne\": \"electronics\" } }\n- $gt: Greater than\n  Example: { \"price\": { \"$gt\": 100 } }\n- $gte: Greater than or equal\n  Example: { \"price\": { \"$gte\": 100 } }\n- $lt: Less than\n  Example: { \"price\": { \"$lt\": 100 } }\n- $lte: Less than or equal\n  Example: { \"price\": { \"$lte\": 100 } }\n\nArray Operators:\n- $in: Match any value in array\n  Example: { \"category\": { \"$in\": [\"electronics\", \"books\"] } }\n- $nin: Does not match any value in array\n  Example: { \"category\": { \"$nin\": [\"electronics\", \"books\"] } }\n- $all: Match all values in array\n  Example: { \"tags\": { \"$all\": [\"premium\", \"sale\"] } }\n\nLogical Operators:\n- $and: Logical AND (can be implicit or explicit)\n  Implicit Example: { \"price\": { \"$gt\": 100 }, \"category\": \"electronics\" }\n  Explicit Example: { \"$and\": [{ \"price\": { \"$gt\": 100 } }, { \"category\": \"electronics\" }] }\n- $or: Logical OR\n  Example: { \"$or\": [{ \"price\": { \"$lt\": 50 } }, { \"category\": \"books\" }] }\n- $not: Logical NOT\n  Example: { \"$not\": { \"category\": \"electronics\" } }\n\nElement Operators:\n- $exists: Check if field exists\n  Example: { \"rating\": { \"$exists\": true } }\n\nSpecial Operators:\n- $size: Array length check\n  Example: { \"tags\": { \"$size\": 2 } }\n\nRestrictions:\n- Regex patterns are not supported\n- Only $and, $or, and $not logical operators are supported\n- Nested fields are supported using dot notation\n- Multiple conditions on the same field are supported with both implicit and explicit $and\n- Empty arrays in $in/$nin will return no results\n- A non-empty array is required for $all operator\n- Only logical operators ($and, $or, $not) can be used at the top level\n- All other operators must be used within a field condition\n  Valid: { \"field\": { \"$gt\": 100 } }\n  Valid: { \"$and\": [...] }\n  Invalid: { \"$gt\": 100 }\n- Logical operators must contain field conditions, not direct operators\n  Valid: { \"$and\": [{ \"field\": { \"$gt\": 100 } }] }\n  Invalid: { \"$and\": [{ \"$gt\": 100 }] }\n- $not operator:\n  - Must be an object\n  - Cannot be empty\n  - Can be used at field level or top level\n  - Valid: { \"$not\": { \"field\": \"value\" } }\n  - Valid: { \"field\": { \"$not\": { \"$eq\": \"value\" } } }\n- Other logical operators ($and, $or):\n  - Can only be used at top level or nested within other logical operators\n  - Can not be used on a field level, or be nested inside a field\n  - Can not be used inside an operator\n  - Valid: { \"$and\": [{ \"field\": { \"$gt\": 100 } }] }\n  - Valid: { \"$or\": [{ \"$and\": [{ \"field\": { \"$gt\": 100 } }] }] }\n  - Invalid: { \"field\": { \"$and\": [{ \"$gt\": 100 }] } }\n  - Invalid: { \"field\": { \"$or\": [{ \"$gt\": 100 }] } }\n  - Invalid: { \"field\": { \"$gt\": { \"$and\": [{...}] } } }\n\nExample Complex Query:\n{\n  \"$and\": [\n    { \"category\": { \"$in\": [\"electronics\", \"computers\"] } },\n    { \"price\": { \"$gte\": 100, \"$lte\": 1000 } },\n    { \"tags\": { \"$all\": [\"premium\"] } },\n    { \"rating\": { \"$exists\": true, \"$gt\": 4 } },\n    { \"$or\": [\n      { \"stock\": { \"$gt\": 0 } },\n      { \"preorder\": true }\n    ]}\n  ]\n}`;\n"],"mappings":";;;;;;;;;;;AA8BA,IAAa,wBAAb,cAA2C,qBAAwC;CACjF,wBAA4D;EAC1D,OAAO;GACL,GAAG,qBAAqB;GACxB,OAAO;IAAC;IAAQ;IAAO;GAAM;GAC7B,SAAS;IAAC;IAAQ;IAAO;GAAM;GAC/B,OAAO,CAAC;GACR,QAAQ,CAAC,OAAO;EAClB;CACF;CAEA,UAAU,QAA+C;EACvD,IAAI,KAAK,QAAQ,MAAM,GAAG,OAAO;EACjC,KAAK,eAAe,MAAM;EAE1B,OAAO,KAAK,cAAc,MAAM;CAClC;CAEA,cAAsB,MAA8B;EAElD,IAAI,KAAK,QAAQ,IAAI,GACnB,MAAM,IAAI,MAAM,oCAAoC;EAEtD,IAAI,KAAK,YAAY,IAAI,KAAK,MAAM,QAAQ,IAAI,GAC9C,OAAO;EAIT,MAAM,oBADU,OAAO,QAAQ,IACC,CAAC,CAAC,KAAK,CAAC,KAAK,WAAW;GAEtD,IAAI,KAAK,WAAW,GAAG,GACrB,OAAO,CAAC,KAAK,KAAK,uBAAuB,KAAK,KAAK,CAAC;GAItD,OAAO,CAAC,KAAK,KAAK,cAAc,KAAK,CAAC;EACxC,CAAC;EAED,OAAO,OAAO,YAAY,iBAAiB;CAC7C;CAEA,uBAA+B,UAAyB,OAAiB;EACvE,IAAI,KAAK,gBAAgB,QAAQ,KAAK,KAAK,kBAAkB,QAAQ,GACnE,OAAO,KAAK,yBAAyB,KAAK;EAG5C,IAAI,KAAK,gBAAgB,QAAQ,KAAK,MAAM,QAAQ,KAAK,GACvD,OAAO,KAAK,qBAAqB,KAAK;EAGxC,OAAO,KAAK,cAAc,KAAK;CACjC;AACF;;;AC7DA,MAAM,YAAY;CAChB,QAAQ;CACR,WAAW;CACX,YAAY;AACd;AAUA,IAAa,cAAb,cAAiC,aAAgC;CAC/D;CAEA,YAAY,EAAE,OAAO,UAAU,UAAU,MAAuC;EAC9E,MAAM,EAAE,GAAG,CAAC;EACZ,MAAM,SAAS,IAAI,cAAc,KAAK;EACtC,KAAKA,MAAM,OAAO,GAAG,UAAU,EAAE,SAAS,CAAC;CAC7C;;;;;;;;;CAUA,MAAM,YAAY,EAAE,WAAW,WAAW,SAAS,YAA8C;EAC/F,IAAI,CAAC,OAAO,UAAU,SAAS,KAAK,aAAa,GAC/C,MAAM,IAAI,YAAY;GACpB,IAAI,oBAAoB,SAAS,gBAAgB,mBAAmB;GACpE,MAAM;GACN,QAAQ,YAAY;GACpB,UAAU,cAAc;EAC1B,CAAC;EAEH,IAAI;GACF,MAAM,KAAKA,IAAI,iBAAiB,WAAW;IACzC,QAAQ;KACN;KACA,QAAQ,UAAU;IACpB;IACA,aAAa;GACf,CAAC;EACH,SAAS,OAAY;GACnB,IAAI,YACF;IACE,IAAI,oBAAoB,SAAS,gBAAgB,UAAU;IAC3D,QAAQ,YAAY;IACpB,UAAU,cAAc;IACxB,SAAS,EAAE,UAAU;GACvB,GACA,KACF;EACF;CACF;;;;;;;;;;CAWA,MAAM,OAAO,EAAE,WAAW,SAAS,UAAU,OAA8C;EACzF,MAAM,aAAa,KAAKA,IAAI,WAAW,SAAS;EAGhD,MAAM,YAAY,OAAO,QAAQ,UAAU,KAAK,GAAG,CAAC,CAAC,SAAS,CAAC;EAE/D,MAAM,UAAU,QAAQ,KAAK,QAAQ,OAAO;GAC1C,IAAI,UAAU;GACd,SAAS;GACT,UAAU,WAAW,MAAM,CAAC;EAC9B,EAAE;EAEF,IAAI;GACF,MAAM,WAAW,WAAW,OAAO;EACrC,SAAS,OAAY;GACnB,MAAM,IAAI,YACR;IACE,IAAI,oBAAoB,SAAS,UAAU,UAAU;IACrD,QAAQ,YAAY;IACpB,UAAU,cAAc;IACxB,SAAS,EAAE,UAAU;GACvB,GACA,KACF;EACF;EACA,OAAO;CACT;CAEA,gBAAgB,QAA4B;EAE1C,OAAO,IADgB,sBACP,CAAC,CAAC,UAAU,MAAM;CACpC;;;;;;;;;;;CAYA,MAAM,MAAM,EACV,WACA,aACA,OAAO,IACP,QACA,gBAAgB,SACiC;EACjD,IAAI,CAAC,aACH,MAAM,IAAI,YAAY;GACpB,IAAI,oBAAoB,SAAS,SAAS,gBAAgB;GAC1D,MAAM;GACN,QAAQ,YAAY;GACpB,UAAU,cAAc;GACxB,SAAS,EAAE,UAAU;EACvB,CAAC;EAGH,MAAM,aAAa,KAAKA,IAAI,WAAW,SAAS;EAEhD,MAAM,mBAAmB,KAAK,gBAAgB,MAAM;EAEpD,IAAI;GAYF,QAAO,MAXQ,WAAW,KAAK,oBAAoB,CAAC,GAAG;IACrD,MAAM,EAAE,SAAS,YAAY;IAC7B,OAAO;IACP,mBAAmB;IACnB,YAAY,EACV,SAAS,gBAAgB,OAAO,MAClC;GACF,CAE2B,CAAC,CAAC,QAAQ,EAAA,CAEtB,KAAI,YAAW;IAC5B,IAAI,OAAO;IACX,OAAO,OAAO;IACd,UAAU,OAAO;IACjB,GAAI,iBAAiB,EAAE,QAAQ,OAAO,QAAQ;GAChD,EAAE;EACJ,SAAS,OAAY;GACnB,MAAM,IAAI,YACR;IACE,IAAI,oBAAoB,SAAS,SAAS,UAAU;IACpD,QAAQ,YAAY;IACpB,UAAU,cAAc;IACxB,SAAS,EAAE,UAAU;GACvB,GACA,KACF;EACF;CACF;;;;;;CAOA,MAAM,cAAiC;EACrC,IAAI;GACF,OAAO,MAAM,KAAKA,IAAI,gBAAgB,EAAE,UAAU,KAAK,CAAC;EAC1D,SAAS,OAAY;GACnB,MAAM,IAAI,YACR;IACE,IAAI,oBAAoB,SAAS,gBAAgB,UAAU;IAC3D,QAAQ,YAAY;IACpB,UAAU,cAAc;GAC1B,GACA,KACF;EACF;CACF;;;;;;;CAQA,MAAM,cAAc,EAAE,aAAuD;EAC3E,MAAM,aAAa,KAAKA,IAAI,WAAW,SAAS;EAChD,IAAI;GACF,MAAM,iBAAiB,WAAW,QAAQ;GAC1C,MAAM,eAAe,WAAW,eAAe,CAAC,GAAG,GAAG;GACtD,MAAM,CAAC,SAAS,SAAS,MAAM,QAAQ,IAAI,CAAC,gBAAgB,YAAY,CAAC;GAGzE,MAAM,SADO,OAAO,KAAK,SACP,CAAC,CAAC,MAAK,QAAO,UAAU,SAAS,QAAQ,QAAQ,MAAM;GACzE,OAAO;IACL,WAAW,QAAQ,QAAQ;IAC3B;IACO;GACT;EACF,SAAS,OAAY;GACnB,IAAI,iBAAiB,aAAa,MAAM;GACxC,MAAM,IAAI,YACR;IACE,IAAI,oBAAoB,SAAS,kBAAkB,UAAU;IAC7D,QAAQ,YAAY;IACpB,UAAU,cAAc;IACxB,SAAS,EAAE,UAAU;GACvB,GACA,KACF;EACF;CACF;;;;;;;CAQA,MAAM,YAAY,EAAE,aAA+C;EACjE,MAAM,aAAa,KAAKA,IAAI,WAAW,SAAS;EAChD,IAAI;GACF,MAAM,WAAW,KAAK;EACxB,SAAS,OAAY;GACnB,MAAM,IAAI,YACR;IACE,IAAI,oBAAoB,SAAS,gBAAgB,UAAU;IAC3D,QAAQ,YAAY;IACpB,UAAU,cAAc;IACxB,SAAS,EAAE,UAAU;GACvB,GACA,KACF;EACF;CACF;;;;;;;;;;;CAYA,MAAM,aAAa,EAAE,WAAW,IAAI,UAA6C;EAC/E,IAAI,CAAC,IACH,MAAM,IAAI,YAAY;GACpB,IAAI,oBAAoB,SAAS,iBAAiB,OAAO;GACzD,MAAM;GACN,QAAQ,YAAY;GACpB,UAAU,cAAc;GACxB,SAAS,EAAE,UAAU;EACvB,CAAC;EAGH,IAAI,CAAC,OAAO,UAAU,CAAC,OAAO,UAC5B,MAAM,IAAI,YAAY;GACpB,IAAI,oBAAoB,SAAS,iBAAiB,YAAY;GAC9D,MAAM;GACN,QAAQ,YAAY;GACpB,UAAU,cAAc;GACxB,SAAS;IACP;IACA;GACF;EACF,CAAC;EAGH,IAAI;GACF,MAAM,aAAa,KAAKA,IAAI,WAAW,SAAS;GAChD,MAAM,YAAiC,CAAC;GAExC,IAAI,OAAO,QACT,UAAU,UAAU,OAAO;GAG7B,IAAI,OAAO,UACT,UAAU,WAAW,OAAO;GAG9B,MAAM,WAAW,iBAAiB,EAAE,GAAG,GAAG,EAAE,MAAM,UAAU,CAAC;EAC/D,SAAS,OAAY;GACnB,IAAI,iBAAiB,aAAa,MAAM;GACxC,MAAM,IAAI,YACR;IACE,IAAI,oBAAoB,SAAS,iBAAiB,kBAAkB;IACpE,QAAQ,YAAY;IACpB,UAAU,cAAc;IACxB,SAAS;KACP;KACA,GAAI,MAAM,EAAE,GAAG;IACjB;GACF,GACA,KACF;EACF;CACF;;;;;;;;CASA,MAAM,aAAa,EAAE,WAAW,MAAyC;EACvE,IAAI;GAEF,MADmB,KAAKA,IAAI,WAAW,SACxB,CAAC,CAAC,UAAU,EAAE,GAAG,CAAC;EACnC,SAAS,OAAY;GACnB,IAAI,iBAAiB,aAAa,MAAM;GACxC,MAAM,IAAI,YACR;IACE,IAAI,oBAAoB,SAAS,iBAAiB,QAAQ;IAC1D,QAAQ,YAAY;IACpB,UAAU,cAAc;IACxB,SAAS;KACP;KACA,GAAI,MAAM,EAAE,GAAG;IACjB;GACF,GACA,KACF;EACF;CACF;CAEA,MAAM,cAAc,EAAE,WAAW,QAAQ,OAA2C;EAClF,MAAM,IAAI,YAAY;GACpB,IAAI,oBAAoB,SAAS,kBAAkB,eAAe;GAClE,MAAM;GACN,QAAQ,YAAY;GACpB,UAAU,cAAc;GACxB,SAAS;IACP;IACA,GAAI,UAAU,EAAE,QAAQ,KAAK,UAAU,MAAM,EAAE;IAC/C,GAAI,OAAO,EAAE,UAAU,IAAI,OAAO;GACpC;EACF,CAAC;CACH;AACF;;;;;;;AC5WA,MAAa,eAAe"}
+{"version":3,"file":"index.js","names":["#db"],"sources":["../src/vector/filter.ts","../src/vector/index.ts","../src/vector/prompt.ts"],"sourcesContent":["import { BaseFilterTranslator } from '@mastra/core/vector/filter';\nimport type {\n  VectorFilter,\n  OperatorSupport,\n  QueryOperator,\n  OperatorValueMap,\n  LogicalOperatorValueMap,\n  BlacklistedRootOperators,\n} from '@mastra/core/vector/filter';\n\ntype AstraOperatorValueMap = Omit<OperatorValueMap, '$elemMatch' | '$regex' | '$options'> & {\n  $size: number; // Astra-specific\n};\n\ntype AstraLogicalOperatorValueMap = Omit<LogicalOperatorValueMap, '$nor'>;\n\ntype AstraBlacklisted = BlacklistedRootOperators | '$nor' | '$size';\n\nexport type AstraVectorFilter = VectorFilter<\n  keyof AstraOperatorValueMap,\n  AstraOperatorValueMap,\n  AstraLogicalOperatorValueMap,\n  AstraBlacklisted\n>;\n\n/**\n * Translator for Astra DB filter queries.\n * Maintains MongoDB-compatible syntax while ensuring proper validation\n * and normalization of values.\n */\nexport class AstraFilterTranslator extends BaseFilterTranslator<AstraVectorFilter> {\n  protected override getSupportedOperators(): OperatorSupport {\n    return {\n      ...BaseFilterTranslator.DEFAULT_OPERATORS,\n      array: ['$all', '$in', '$nin'],\n      logical: ['$and', '$or', '$not'],\n      regex: [],\n      custom: ['$size'],\n    };\n  }\n\n  translate(filter?: AstraVectorFilter): AstraVectorFilter {\n    if (this.isEmpty(filter)) return filter;\n    this.validateFilter(filter);\n\n    return this.translateNode(filter);\n  }\n\n  private translateNode(node: AstraVectorFilter): any {\n    // Handle primitive values and arrays\n    if (this.isRegex(node)) {\n      throw new Error('Regex is not supported in Astra DB');\n    }\n    if (this.isPrimitive(node) || Array.isArray(node)) {\n      return node;\n    }\n\n    const entries = Object.entries(node as Record<string, any>);\n    const translatedEntries = entries.map(([key, value]) => {\n      // Handle operators\n      if (this.isOperator(key)) {\n        return [key, this.translateOperatorValue(key, value)];\n      }\n\n      // Handle nested paths and objects\n      return [key, this.translateNode(value)];\n    });\n\n    return Object.fromEntries(translatedEntries);\n  }\n\n  private translateOperatorValue(operator: QueryOperator, value: any): any {\n    if (this.isBasicOperator(operator) || this.isNumericOperator(operator)) {\n      return this.normalizeComparisonValue(value);\n    }\n\n    if (this.isArrayOperator(operator) && Array.isArray(value)) {\n      return this.normalizeArrayValues(value);\n    }\n\n    return this.translateNode(value);\n  }\n}\n","import type { Db } from '@datastax/astra-db-ts';\nimport { DataAPIClient, UUID } from '@datastax/astra-db-ts';\nimport { MastraError, ErrorDomain, ErrorCategory } from '@mastra/core/error';\nimport { createVectorErrorId } from '@mastra/core/storage';\nimport { MastraVector } from '@mastra/core/vector';\nimport type {\n  QueryResult,\n  IndexStats,\n  CreateIndexParams,\n  UpsertVectorParams,\n  QueryVectorParams,\n  DescribeIndexParams,\n  DeleteIndexParams,\n  DeleteVectorParams,\n  UpdateVectorParams,\n  DeleteVectorsParams,\n} from '@mastra/core/vector';\nimport type { AstraVectorFilter } from './filter';\nimport { AstraFilterTranslator } from './filter';\n\n// Mastra and Astra DB agree on cosine and euclidean, but Astra DB uses dot_product instead of dotproduct.\nconst metricMap = {\n  cosine: 'cosine',\n  euclidean: 'euclidean',\n  dotproduct: 'dot_product',\n} as const;\n\nexport interface AstraDbOptions {\n  token: string;\n  endpoint: string;\n  keyspace?: string;\n}\n\ntype AstraQueryVectorParams = QueryVectorParams<AstraVectorFilter>;\n\nexport class AstraVector extends MastraVector<AstraVectorFilter> {\n  readonly #db: Db;\n\n  constructor({ token, endpoint, keyspace, id }: AstraDbOptions & { id: string }) {\n    super({ id });\n    const client = new DataAPIClient(token);\n    this.#db = client.db(endpoint, { keyspace });\n  }\n\n  /**\n   * Creates a new collection with the specified configuration.\n   *\n   * @param {string} indexName - The name of the collection to create.\n   * @param {number} dimension - The dimension of the vectors to be stored in the collection.\n   * @param {'cosine' | 'euclidean' | 'dotproduct'} [metric=cosine] - The metric to use to sort vectors in the collection.\n   * @returns {Promise<void>} A promise that resolves when the collection is created.\n   */\n  async createIndex({ indexName, dimension, metric = 'cosine' }: CreateIndexParams): Promise<void> {\n    if (!Number.isInteger(dimension) || dimension <= 0) {\n      throw new MastraError({\n        id: createVectorErrorId('ASTRA', 'CREATE_INDEX', 'INVALID_DIMENSION'),\n        text: 'Dimension must be a positive integer',\n        domain: ErrorDomain.MASTRA_VECTOR,\n        category: ErrorCategory.USER,\n      });\n    }\n    try {\n      await this.#db.createCollection(indexName, {\n        vector: {\n          dimension,\n          metric: metricMap[metric],\n        },\n        checkExists: false,\n      });\n    } catch (error: any) {\n      new MastraError(\n        {\n          id: createVectorErrorId('ASTRA', 'CREATE_INDEX', 'DB_ERROR'),\n          domain: ErrorDomain.MASTRA_VECTOR,\n          category: ErrorCategory.THIRD_PARTY,\n          details: { indexName },\n        },\n        error,\n      );\n    }\n  }\n\n  /**\n   * Inserts or updates vectors in the specified collection.\n   *\n   * @param {string} indexName - The name of the collection to upsert into.\n   * @param {number[][]} vectors - An array of vectors to upsert.\n   * @param {Record<string, any>[]} [metadata] - An optional array of metadata objects corresponding to each vector.\n   * @param {string[]} [ids] - An optional array of IDs corresponding to each vector. If not provided, new IDs will be generated.\n   * @returns {Promise<string[]>} A promise that resolves to an array of IDs of the upserted vectors.\n   */\n  async upsert({ indexName, vectors, metadata, ids }: UpsertVectorParams): Promise<string[]> {\n    const collection = this.#db.collection(indexName);\n\n    // Generate IDs if not provided\n    const vectorIds = ids || vectors.map(() => UUID.v7().toString());\n\n    const records = vectors.map((vector, i) => ({\n      id: vectorIds[i],\n      $vector: vector,\n      metadata: metadata?.[i] || {},\n    }));\n\n    try {\n      await collection.insertMany(records);\n    } catch (error: any) {\n      throw new MastraError(\n        {\n          id: createVectorErrorId('ASTRA', 'UPSERT', 'DB_ERROR'),\n          domain: ErrorDomain.MASTRA_VECTOR,\n          category: ErrorCategory.THIRD_PARTY,\n          details: { indexName },\n        },\n        error,\n      );\n    }\n    return vectorIds;\n  }\n\n  transformFilter(filter?: AstraVectorFilter) {\n    const translator = new AstraFilterTranslator();\n    return translator.translate(filter);\n  }\n\n  /**\n   * Queries the specified collection using a vector and optional filter.\n   *\n   * @param {string} indexName - The name of the collection to query.\n   * @param {number[]} queryVector - The vector to query with.\n   * @param {number} [topK] - The maximum number of results to return.\n   * @param {Record<string, any>} [filter] - An optional filter to apply to the query. For more on filters in Astra DB, see the filtering reference: https://docs.datastax.com/en/astra-db-serverless/api-reference/documents.html#operators\n   * @param {boolean} [includeVectors=false] - Whether to include the vectors in the response.\n   * @returns {Promise<QueryResult[]>} A promise that resolves to an array of query results.\n   */\n  async query({\n    indexName,\n    queryVector,\n    topK = 10,\n    filter,\n    includeVector = false,\n  }: AstraQueryVectorParams): Promise<QueryResult[]> {\n    if (!queryVector) {\n      throw new MastraError({\n        id: createVectorErrorId('ASTRA', 'QUERY', 'MISSING_VECTOR'),\n        text: 'queryVector is required for Astra queries. Metadata-only queries are not supported by this vector store.',\n        domain: ErrorDomain.MASTRA_VECTOR,\n        category: ErrorCategory.USER,\n        details: { indexName },\n      });\n    }\n\n    const collection = this.#db.collection(indexName);\n\n    const translatedFilter = this.transformFilter(filter);\n\n    try {\n      const cursor = collection.find(translatedFilter ?? {}, {\n        sort: { $vector: queryVector },\n        limit: topK,\n        includeSimilarity: true,\n        projection: {\n          $vector: !!includeVector,\n        },\n      });\n\n      const results = await cursor.toArray();\n\n      return results.map(result => ({\n        id: result.id,\n        score: result.$similarity,\n        metadata: result.metadata,\n        ...(includeVector && { vector: result.$vector }),\n      }));\n    } catch (error: any) {\n      throw new MastraError(\n        {\n          id: createVectorErrorId('ASTRA', 'QUERY', 'DB_ERROR'),\n          domain: ErrorDomain.MASTRA_VECTOR,\n          category: ErrorCategory.THIRD_PARTY,\n          details: { indexName },\n        },\n        error,\n      );\n    }\n  }\n\n  /**\n   * Lists all collections in the database.\n   *\n   * @returns {Promise<string[]>} A promise that resolves to an array of collection names.\n   */\n  async listIndexes(): Promise<string[]> {\n    try {\n      return await this.#db.listCollections({ nameOnly: true });\n    } catch (error: any) {\n      throw new MastraError(\n        {\n          id: createVectorErrorId('ASTRA', 'LIST_INDEXES', 'DB_ERROR'),\n          domain: ErrorDomain.MASTRA_VECTOR,\n          category: ErrorCategory.THIRD_PARTY,\n        },\n        error,\n      );\n    }\n  }\n\n  /**\n   * Retrieves statistics about a vector index.\n   *\n   * @param {string} indexName - The name of the index to describe\n   * @returns A promise that resolves to the index statistics including dimension, count and metric\n   */\n  async describeIndex({ indexName }: DescribeIndexParams): Promise<IndexStats> {\n    const collection = this.#db.collection(indexName);\n    try {\n      const optionsPromise = collection.options();\n      const countPromise = collection.countDocuments({}, 100);\n      const [options, count] = await Promise.all([optionsPromise, countPromise]);\n\n      const keys = Object.keys(metricMap) as (keyof typeof metricMap)[];\n      const metric = keys.find(key => metricMap[key] === options.vector?.metric);\n      return {\n        dimension: options.vector?.dimension!,\n        metric,\n        count: count,\n      };\n    } catch (error: any) {\n      if (error instanceof MastraError) throw error;\n      throw new MastraError(\n        {\n          id: createVectorErrorId('ASTRA', 'DESCRIBE_INDEX', 'DB_ERROR'),\n          domain: ErrorDomain.MASTRA_VECTOR,\n          category: ErrorCategory.THIRD_PARTY,\n          details: { indexName },\n        },\n        error,\n      );\n    }\n  }\n\n  /**\n   * Deletes the specified collection.\n   *\n   * @param {string} indexName - The name of the collection to delete.\n   * @returns {Promise<void>} A promise that resolves when the collection is deleted.\n   */\n  async deleteIndex({ indexName }: DeleteIndexParams): Promise<void> {\n    const collection = this.#db.collection(indexName);\n    try {\n      await collection.drop();\n    } catch (error: any) {\n      throw new MastraError(\n        {\n          id: createVectorErrorId('ASTRA', 'DELETE_INDEX', 'DB_ERROR'),\n          domain: ErrorDomain.MASTRA_VECTOR,\n          category: ErrorCategory.THIRD_PARTY,\n          details: { indexName },\n        },\n        error,\n      );\n    }\n  }\n\n  /**\n   * Updates a vector by its ID with the provided vector and/or metadata.\n   * @param indexName - The name of the index containing the vector.\n   * @param id - The ID of the vector to update.\n   * @param update - An object containing the vector and/or metadata to update.\n   * @param update.vector - An optional array of numbers representing the new vector.\n   * @param update.metadata - An optional record containing the new metadata.\n   * @returns A promise that resolves when the update is complete.\n   * @throws Will throw an error if no updates are provided or if the update operation fails.\n   */\n  async updateVector({ indexName, id, update }: UpdateVectorParams): Promise<void> {\n    if (!id) {\n      throw new MastraError({\n        id: createVectorErrorId('ASTRA', 'UPDATE_VECTOR', 'NO_ID'),\n        text: 'id is required for Astra updateVector',\n        domain: ErrorDomain.MASTRA_VECTOR,\n        category: ErrorCategory.USER,\n        details: { indexName },\n      });\n    }\n\n    if (!update.vector && !update.metadata) {\n      throw new MastraError({\n        id: createVectorErrorId('ASTRA', 'UPDATE_VECTOR', 'NO_PAYLOAD'),\n        text: 'No updates provided for vector',\n        domain: ErrorDomain.MASTRA_VECTOR,\n        category: ErrorCategory.USER,\n        details: {\n          indexName,\n          id,\n        },\n      });\n    }\n\n    try {\n      const collection = this.#db.collection(indexName);\n      const updateDoc: Record<string, any> = {};\n\n      if (update.vector) {\n        updateDoc.$vector = update.vector;\n      }\n\n      if (update.metadata) {\n        updateDoc.metadata = update.metadata;\n      }\n\n      await collection.findOneAndUpdate({ id }, { $set: updateDoc });\n    } catch (error: any) {\n      if (error instanceof MastraError) throw error;\n      throw new MastraError(\n        {\n          id: createVectorErrorId('ASTRA', 'UPDATE_VECTOR', 'FAILED_UNHANDLED'),\n          domain: ErrorDomain.MASTRA_VECTOR,\n          category: ErrorCategory.THIRD_PARTY,\n          details: {\n            indexName,\n            ...(id && { id }),\n          },\n        },\n        error,\n      );\n    }\n  }\n\n  /**\n   * Deletes a vector by its ID.\n   * @param indexName - The name of the index containing the vector.\n   * @param id - The ID of the vector to delete.\n   * @returns A promise that resolves when the deletion is complete.\n   * @throws Will throw an error if the deletion operation fails.\n   */\n  async deleteVector({ indexName, id }: DeleteVectorParams): Promise<void> {\n    try {\n      const collection = this.#db.collection(indexName);\n      await collection.deleteOne({ id });\n    } catch (error: any) {\n      if (error instanceof MastraError) throw error;\n      throw new MastraError(\n        {\n          id: createVectorErrorId('ASTRA', 'DELETE_VECTOR', 'FAILED'),\n          domain: ErrorDomain.MASTRA_VECTOR,\n          category: ErrorCategory.THIRD_PARTY,\n          details: {\n            indexName,\n            ...(id && { id }),\n          },\n        },\n        error,\n      );\n    }\n  }\n\n  async deleteVectors({ indexName, filter, ids }: DeleteVectorsParams): Promise<void> {\n    throw new MastraError({\n      id: createVectorErrorId('ASTRA', 'DELETE_VECTORS', 'NOT_SUPPORTED'),\n      text: 'deleteVectors is not yet implemented for Astra vector store',\n      domain: ErrorDomain.MASTRA_VECTOR,\n      category: ErrorCategory.SYSTEM,\n      details: {\n        indexName,\n        ...(filter && { filter: JSON.stringify(filter) }),\n        ...(ids && { idsCount: ids.length }),\n      },\n    });\n  }\n}\n","/**\n * Vector store specific prompt that details supported operators and examples.\n * This prompt helps users construct valid filters for Astra Vector.\n */\nexport const ASTRA_PROMPT = `When querying Astra, you can ONLY use the operators listed below. Any other operators will be rejected.\nImportant: Don't explain how to construct the filter - use the specified operators and fields to search the content and return relevant results.\nIf a user tries to give an explicit operator that is not supported, reject the filter entirely and let them know that the operator is not supported.\n\nBasic Comparison Operators:\n- $eq: Exact match (default when using field: value)\n  Example: { \"category\": \"electronics\" }\n- $ne: Not equal\n  Example: { \"category\": { \"$ne\": \"electronics\" } }\n- $gt: Greater than\n  Example: { \"price\": { \"$gt\": 100 } }\n- $gte: Greater than or equal\n  Example: { \"price\": { \"$gte\": 100 } }\n- $lt: Less than\n  Example: { \"price\": { \"$lt\": 100 } }\n- $lte: Less than or equal\n  Example: { \"price\": { \"$lte\": 100 } }\n\nArray Operators:\n- $in: Match any value in array\n  Example: { \"category\": { \"$in\": [\"electronics\", \"books\"] } }\n- $nin: Does not match any value in array\n  Example: { \"category\": { \"$nin\": [\"electronics\", \"books\"] } }\n- $all: Match all values in array\n  Example: { \"tags\": { \"$all\": [\"premium\", \"sale\"] } }\n\nLogical Operators:\n- $and: Logical AND (can be implicit or explicit)\n  Implicit Example: { \"price\": { \"$gt\": 100 }, \"category\": \"electronics\" }\n  Explicit Example: { \"$and\": [{ \"price\": { \"$gt\": 100 } }, { \"category\": \"electronics\" }] }\n- $or: Logical OR\n  Example: { \"$or\": [{ \"price\": { \"$lt\": 50 } }, { \"category\": \"books\" }] }\n- $not: Logical NOT\n  Example: { \"$not\": { \"category\": \"electronics\" } }\n\nElement Operators:\n- $exists: Check if field exists\n  Example: { \"rating\": { \"$exists\": true } }\n\nSpecial Operators:\n- $size: Array length check\n  Example: { \"tags\": { \"$size\": 2 } }\n\nRestrictions:\n- Regex patterns are not supported\n- Only $and, $or, and $not logical operators are supported\n- Nested fields are supported using dot notation\n- Multiple conditions on the same field are supported with both implicit and explicit $and\n- Empty arrays in $in/$nin will return no results\n- A non-empty array is required for $all operator\n- Only logical operators ($and, $or, $not) can be used at the top level\n- All other operators must be used within a field condition\n  Valid: { \"field\": { \"$gt\": 100 } }\n  Valid: { \"$and\": [...] }\n  Invalid: { \"$gt\": 100 }\n- Logical operators must contain field conditions, not direct operators\n  Valid: { \"$and\": [{ \"field\": { \"$gt\": 100 } }] }\n  Invalid: { \"$and\": [{ \"$gt\": 100 }] }\n- $not operator:\n  - Must be an object\n  - Cannot be empty\n  - Can be used at field level or top level\n  - Valid: { \"$not\": { \"field\": \"value\" } }\n  - Valid: { \"field\": { \"$not\": { \"$eq\": \"value\" } } }\n- Other logical operators ($and, $or):\n  - Can only be used at top level or nested within other logical operators\n  - Can not be used on a field level, or be nested inside a field\n  - Can not be used inside an operator\n  - Valid: { \"$and\": [{ \"field\": { \"$gt\": 100 } }] }\n  - Valid: { \"$or\": [{ \"$and\": [{ \"field\": { \"$gt\": 100 } }] }] }\n  - Invalid: { \"field\": { \"$and\": [{ \"$gt\": 100 }] } }\n  - Invalid: { \"field\": { \"$or\": [{ \"$gt\": 100 }] } }\n  - Invalid: { \"field\": { \"$gt\": { \"$and\": [{...}] } } }\n\nExample Complex Query:\n{\n  \"$and\": [\n    { \"category\": { \"$in\": [\"electronics\", \"computers\"] } },\n    { \"price\": { \"$gte\": 100, \"$lte\": 1000 } },\n    { \"tags\": { \"$all\": [\"premium\"] } },\n    { \"rating\": { \"$exists\": true, \"$gt\": 4 } },\n    { \"$or\": [\n      { \"stock\": { \"$gt\": 0 } },\n      { \"preorder\": true }\n    ]}\n  ]\n}`;\n"],"mappings":";;;;;;;;;;;AA8BA,IAAa,wBAAb,cAA2C,qBAAwC;CACjF,wBAA4D;EAC1D,OAAO;GACL,GAAG,qBAAqB;GACxB,OAAO;IAAC;IAAQ;IAAO;GAAM;GAC7B,SAAS;IAAC;IAAQ;IAAO;GAAM;GAC/B,OAAO,CAAC;GACR,QAAQ,CAAC,OAAO;EAClB;CACF;CAEA,UAAU,QAA+C;EACvD,IAAI,KAAK,QAAQ,MAAM,GAAG,OAAO;EACjC,KAAK,eAAe,MAAM;EAE1B,OAAO,KAAK,cAAc,MAAM;CAClC;CAEA,cAAsB,MAA8B;EAElD,IAAI,KAAK,QAAQ,IAAI,GACnB,MAAM,IAAI,MAAM,oCAAoC;EAEtD,IAAI,KAAK,YAAY,IAAI,KAAK,MAAM,QAAQ,IAAI,GAC9C,OAAO;EAIT,MAAM,oBADU,OAAO,QAAQ,IACC,CAAC,CAAC,KAAK,CAAC,KAAK,WAAW;GAEtD,IAAI,KAAK,WAAW,GAAG,GACrB,OAAO,CAAC,KAAK,KAAK,uBAAuB,KAAK,KAAK,CAAC;GAItD,OAAO,CAAC,KAAK,KAAK,cAAc,KAAK,CAAC;EACxC,CAAC;EAED,OAAO,OAAO,YAAY,iBAAiB;CAC7C;CAEA,uBAA+B,UAAyB,OAAiB;EACvE,IAAI,KAAK,gBAAgB,QAAQ,KAAK,KAAK,kBAAkB,QAAQ,GACnE,OAAO,KAAK,yBAAyB,KAAK;EAG5C,IAAI,KAAK,gBAAgB,QAAQ,KAAK,MAAM,QAAQ,KAAK,GACvD,OAAO,KAAK,qBAAqB,KAAK;EAGxC,OAAO,KAAK,cAAc,KAAK;CACjC;AACF;;;AC7DA,MAAM,YAAY;CAChB,QAAQ;CACR,WAAW;CACX,YAAY;AACd;AAUA,IAAa,cAAb,cAAiC,aAAgC;CAC/D;CAEA,YAAY,EAAE,OAAO,UAAU,UAAU,MAAuC;EAC9E,MAAM,EAAE,GAAG,CAAC;EACZ,MAAM,SAAS,IAAI,cAAc,KAAK;EACtC,KAAKA,MAAM,OAAO,GAAG,UAAU,EAAE,SAAS,CAAC;CAC7C;;;;;;;;;CAUA,MAAM,YAAY,EAAE,WAAW,WAAW,SAAS,YAA8C;EAC/F,IAAI,CAAC,OAAO,UAAU,SAAS,KAAK,aAAa,GAC/C,MAAM,IAAI,YAAY;GACpB,IAAI,oBAAoB,SAAS,gBAAgB,mBAAmB;GACpE,MAAM;GACN,QAAQ,YAAY;GACpB,UAAU,cAAc;EAC1B,CAAC;EAEH,IAAI;GACF,MAAM,KAAKA,IAAI,iBAAiB,WAAW;IACzC,QAAQ;KACN;KACA,QAAQ,UAAU;IACpB;IACA,aAAa;GACf,CAAC;EACH,SAAS,OAAY;GACnB,IAAI,YACF;IACE,IAAI,oBAAoB,SAAS,gBAAgB,UAAU;IAC3D,QAAQ,YAAY;IACpB,UAAU,cAAc;IACxB,SAAS,EAAE,UAAU;GACvB,GACA,KACF;EACF;CACF;;;;;;;;;;CAWA,MAAM,OAAO,EAAE,WAAW,SAAS,UAAU,OAA8C;EACzF,MAAM,aAAa,KAAKA,IAAI,WAAW,SAAS;EAGhD,MAAM,YAAY,OAAO,QAAQ,UAAU,KAAK,GAAG,CAAC,CAAC,SAAS,CAAC;EAE/D,MAAM,UAAU,QAAQ,KAAK,QAAQ,OAAO;GAC1C,IAAI,UAAU;GACd,SAAS;GACT,UAAU,WAAW,MAAM,CAAC;EAC9B,EAAE;EAEF,IAAI;GACF,MAAM,WAAW,WAAW,OAAO;EACrC,SAAS,OAAY;GACnB,MAAM,IAAI,YACR;IACE,IAAI,oBAAoB,SAAS,UAAU,UAAU;IACrD,QAAQ,YAAY;IACpB,UAAU,cAAc;IACxB,SAAS,EAAE,UAAU;GACvB,GACA,KACF;EACF;EACA,OAAO;CACT;CAEA,gBAAgB,QAA4B;EAE1C,OAAO,IADgB,sBACP,CAAC,CAAC,UAAU,MAAM;CACpC;;;;;;;;;;;CAYA,MAAM,MAAM,EACV,WACA,aACA,OAAO,IACP,QACA,gBAAgB,SACiC;EACjD,IAAI,CAAC,aACH,MAAM,IAAI,YAAY;GACpB,IAAI,oBAAoB,SAAS,SAAS,gBAAgB;GAC1D,MAAM;GACN,QAAQ,YAAY;GACpB,UAAU,cAAc;GACxB,SAAS,EAAE,UAAU;EACvB,CAAC;EAGH,MAAM,aAAa,KAAKA,IAAI,WAAW,SAAS;EAEhD,MAAM,mBAAmB,KAAK,gBAAgB,MAAM;EAEpD,IAAI;GAYF,QAAO,MAXQ,WAAW,KAAK,oBAAoB,CAAC,GAAG;IACrD,MAAM,EAAE,SAAS,YAAY;IAC7B,OAAO;IACP,mBAAmB;IACnB,YAAY,EACV,SAAS,CAAC,CAAC,cACb;GACF,CAE2B,CAAC,CAAC,QAAQ,EAAA,CAEtB,KAAI,YAAW;IAC5B,IAAI,OAAO;IACX,OAAO,OAAO;IACd,UAAU,OAAO;IACjB,GAAI,iBAAiB,EAAE,QAAQ,OAAO,QAAQ;GAChD,EAAE;EACJ,SAAS,OAAY;GACnB,MAAM,IAAI,YACR;IACE,IAAI,oBAAoB,SAAS,SAAS,UAAU;IACpD,QAAQ,YAAY;IACpB,UAAU,cAAc;IACxB,SAAS,EAAE,UAAU;GACvB,GACA,KACF;EACF;CACF;;;;;;CAOA,MAAM,cAAiC;EACrC,IAAI;GACF,OAAO,MAAM,KAAKA,IAAI,gBAAgB,EAAE,UAAU,KAAK,CAAC;EAC1D,SAAS,OAAY;GACnB,MAAM,IAAI,YACR;IACE,IAAI,oBAAoB,SAAS,gBAAgB,UAAU;IAC3D,QAAQ,YAAY;IACpB,UAAU,cAAc;GAC1B,GACA,KACF;EACF;CACF;;;;;;;CAQA,MAAM,cAAc,EAAE,aAAuD;EAC3E,MAAM,aAAa,KAAKA,IAAI,WAAW,SAAS;EAChD,IAAI;GACF,MAAM,iBAAiB,WAAW,QAAQ;GAC1C,MAAM,eAAe,WAAW,eAAe,CAAC,GAAG,GAAG;GACtD,MAAM,CAAC,SAAS,SAAS,MAAM,QAAQ,IAAI,CAAC,gBAAgB,YAAY,CAAC;GAGzE,MAAM,SADO,OAAO,KAAK,SACP,CAAC,CAAC,MAAK,QAAO,UAAU,SAAS,QAAQ,QAAQ,MAAM;GACzE,OAAO;IACL,WAAW,QAAQ,QAAQ;IAC3B;IACO;GACT;EACF,SAAS,OAAY;GACnB,IAAI,iBAAiB,aAAa,MAAM;GACxC,MAAM,IAAI,YACR;IACE,IAAI,oBAAoB,SAAS,kBAAkB,UAAU;IAC7D,QAAQ,YAAY;IACpB,UAAU,cAAc;IACxB,SAAS,EAAE,UAAU;GACvB,GACA,KACF;EACF;CACF;;;;;;;CAQA,MAAM,YAAY,EAAE,aAA+C;EACjE,MAAM,aAAa,KAAKA,IAAI,WAAW,SAAS;EAChD,IAAI;GACF,MAAM,WAAW,KAAK;EACxB,SAAS,OAAY;GACnB,MAAM,IAAI,YACR;IACE,IAAI,oBAAoB,SAAS,gBAAgB,UAAU;IAC3D,QAAQ,YAAY;IACpB,UAAU,cAAc;IACxB,SAAS,EAAE,UAAU;GACvB,GACA,KACF;EACF;CACF;;;;;;;;;;;CAYA,MAAM,aAAa,EAAE,WAAW,IAAI,UAA6C;EAC/E,IAAI,CAAC,IACH,MAAM,IAAI,YAAY;GACpB,IAAI,oBAAoB,SAAS,iBAAiB,OAAO;GACzD,MAAM;GACN,QAAQ,YAAY;GACpB,UAAU,cAAc;GACxB,SAAS,EAAE,UAAU;EACvB,CAAC;EAGH,IAAI,CAAC,OAAO,UAAU,CAAC,OAAO,UAC5B,MAAM,IAAI,YAAY;GACpB,IAAI,oBAAoB,SAAS,iBAAiB,YAAY;GAC9D,MAAM;GACN,QAAQ,YAAY;GACpB,UAAU,cAAc;GACxB,SAAS;IACP;IACA;GACF;EACF,CAAC;EAGH,IAAI;GACF,MAAM,aAAa,KAAKA,IAAI,WAAW,SAAS;GAChD,MAAM,YAAiC,CAAC;GAExC,IAAI,OAAO,QACT,UAAU,UAAU,OAAO;GAG7B,IAAI,OAAO,UACT,UAAU,WAAW,OAAO;GAG9B,MAAM,WAAW,iBAAiB,EAAE,GAAG,GAAG,EAAE,MAAM,UAAU,CAAC;EAC/D,SAAS,OAAY;GACnB,IAAI,iBAAiB,aAAa,MAAM;GACxC,MAAM,IAAI,YACR;IACE,IAAI,oBAAoB,SAAS,iBAAiB,kBAAkB;IACpE,QAAQ,YAAY;IACpB,UAAU,cAAc;IACxB,SAAS;KACP;KACA,GAAI,MAAM,EAAE,GAAG;IACjB;GACF,GACA,KACF;EACF;CACF;;;;;;;;CASA,MAAM,aAAa,EAAE,WAAW,MAAyC;EACvE,IAAI;GAEF,MADmB,KAAKA,IAAI,WAAW,SACxB,CAAC,CAAC,UAAU,EAAE,GAAG,CAAC;EACnC,SAAS,OAAY;GACnB,IAAI,iBAAiB,aAAa,MAAM;GACxC,MAAM,IAAI,YACR;IACE,IAAI,oBAAoB,SAAS,iBAAiB,QAAQ;IAC1D,QAAQ,YAAY;IACpB,UAAU,cAAc;IACxB,SAAS;KACP;KACA,GAAI,MAAM,EAAE,GAAG;IACjB;GACF,GACA,KACF;EACF;CACF;CAEA,MAAM,cAAc,EAAE,WAAW,QAAQ,OAA2C;EAClF,MAAM,IAAI,YAAY;GACpB,IAAI,oBAAoB,SAAS,kBAAkB,eAAe;GAClE,MAAM;GACN,QAAQ,YAAY;GACpB,UAAU,cAAc;GACxB,SAAS;IACP;IACA,GAAI,UAAU,EAAE,QAAQ,KAAK,UAAU,MAAM,EAAE;IAC/C,GAAI,OAAO,EAAE,UAAU,IAAI,OAAO;GACpC;EACF,CAAC;CACH;AACF;;;;;;;AC5WA,MAAa,eAAe"}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@mastra/astra",
-  "version": "1.1.1",
+  "version": "1.1.2-alpha.1",
   "description": "Astra DB provider for Mastra - includes vector store capabilities",
   "type": "module",
   "main": "dist/index.js",
@@ -27,16 +27,15 @@
   },
   "devDependencies": {
     "@types/node": "22.20.1",
-    "@vitest/coverage-v8": "4.1.10",
-    "@vitest/ui": "4.1.10",
-    "eslint": "^10.7.0",
+    "@vitest/coverage-v8": "4.1.11",
+    "@vitest/ui": "4.1.11",
     "tsdown": "0.22.9",
     "tsx": "^4.23.1",
     "typescript": "^7.0.2",
-    "vitest": "4.1.10",
-    "@internal/lint": "0.0.130",
-    "@internal/types-builder": "0.0.105",
-    "@mastra/core": "1.64.0"
+    "vitest": "4.1.11",
+    "@internal/lint": "0.0.144",
+    "@internal/types-builder": "0.0.119",
+    "@mastra/core": "1.79.0-alpha.9"
   },
   "peerDependencies": {
     "@mastra/core": ">=1.0.0-0 <2.0.0-0"
@@ -57,7 +56,7 @@
     "build:lib": "tsdown --silent --config tsdown.config.ts",
     "build:watch": "tsdown --watch --silent --config tsdown.config.ts",
     "test": "vitest run",
-    "lint": "oxlint . && eslint .",
-    "lint:fix": "oxlint --fix . && eslint --fix ."
+    "lint": "oxlint --type-aware .",
+    "lint:fix": "oxlint --type-aware --fix ."
   }
 }