@mastra/qdrant 0.10.3 → 0.11.0-alpha.1

package/.turbo/turbo-build.log

@@ -1,23 +1,23 @@
 
-> @mastra/qdrant@0.10.3-alpha.0 build /home/runner/work/mastra/mastra/stores/qdrant
+> @mastra/qdrant@0.11.0-alpha.1 build /home/runner/work/mastra/mastra/stores/qdrant
 > tsup src/index.ts --format esm,cjs --experimental-dts --clean --treeshake=smallest --splitting
 
 CLI Building entry: src/index.ts
 CLI Using tsconfig: tsconfig.json
 CLI tsup v8.5.0
 TSC Build start
-TSC ⚡️ Build success in 7578ms
+TSC ⚡️ Build success in 7990ms
 DTS Build start
 CLI Target: es2022
 Analysis will use the bundled TypeScript version 5.8.3
 Writing package typings: /home/runner/work/mastra/mastra/stores/qdrant/dist/_tsup-dts-rollup.d.ts
 Analysis will use the bundled TypeScript version 5.8.3
 Writing package typings: /home/runner/work/mastra/mastra/stores/qdrant/dist/_tsup-dts-rollup.d.cts
-DTS ⚡️ Build success in 10357ms
+DTS ⚡️ Build success in 9199ms
 CLI Cleaning output folder
 ESM Build start
 CJS Build start
-ESM dist/index.js 18.24 KB
-ESM ⚡️ Build success in 687ms
-CJS dist/index.cjs 18.30 KB
-CJS ⚡️ Build success in 691ms
+CJS dist/index.cjs 21.23 KB
+CJS ⚡️ Build success in 978ms
+ESM dist/index.js 21.00 KB
+ESM ⚡️ Build success in 980ms

package/CHANGELOG.md

@@ -1,5 +1,31 @@
 # @mastra/qdrant
 
+## 0.11.0-alpha.1
+
+### Minor Changes
+
+- 8a3bfd2: Update peerdeps to latest core
+
+### Patch Changes
+
+- Updated dependencies [792c4c0]
+- Updated dependencies [502fe05]
+- Updated dependencies [4efcfa0]
+  - @mastra/core@0.10.7-alpha.3
+
+## 0.10.4-alpha.0
+
+### Patch Changes
+
+- 0e17048: Throw mastra errors in storage packages
+- Updated dependencies [d1baedb]
+- Updated dependencies [4d21bf2]
+- Updated dependencies [2097952]
+- Updated dependencies [4fb0cc2]
+- Updated dependencies [d2a7a31]
+- Updated dependencies [0e17048]
+  - @mastra/core@0.10.7-alpha.1
+
 ## 0.10.3
 
 ### Patch Changes

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

@@ -1,12 +1,15 @@
 import { BaseFilterTranslator } from '@mastra/core/vector/filter';
+import type { BlacklistedRootOperators } from '@mastra/core/vector/filter';
 import type { CreateIndexParams } from '@mastra/core/vector';
 import type { DeleteIndexParams } from '@mastra/core/vector';
 import type { DeleteVectorParams } from '@mastra/core/vector';
 import type { DescribeIndexParams } from '@mastra/core/vector';
 import type { IndexStats } from '@mastra/core/vector';
 import type { LogicalOperator } from '@mastra/core/vector/filter';
+import type { LogicalOperatorValueMap } from '@mastra/core/vector/filter';
 import { MastraVector } from '@mastra/core/vector';
 import type { OperatorSupport } from '@mastra/core/vector/filter';
+import type { OperatorValueMap } from '@mastra/core/vector/filter';
 import type { QueryResult } from '@mastra/core/vector';
 import type { QueryVectorParams } from '@mastra/core/vector';
 import type { UpdateVectorParams } from '@mastra/core/vector';
@@ -21,6 +24,8 @@ declare const QDRANT_PROMPT = "When querying Qdrant, you can ONLY use the operat
 export { QDRANT_PROMPT }
 export { QDRANT_PROMPT as QDRANT_PROMPT_alias_1 }
 
+declare type QdrantBlacklistedRootOperators = BlacklistedRootOperators | '$count' | '$geo' | '$nested' | '$datetime' | '$null' | '$empty';
+
 /**
  * Translates MongoDB-style filters to Qdrant compatible filters.
  *
@@ -40,10 +45,10 @@ export { QDRANT_PROMPT as QDRANT_PROMPT_alias_1 }
  * - $null -> is_null check
  * - $empty -> is_empty check
  */
-export declare class QdrantFilterTranslator extends BaseFilterTranslator {
+export declare class QdrantFilterTranslator extends BaseFilterTranslator<QdrantVectorFilter> {
     protected isLogicalOperator(key: string): key is LogicalOperator;
     protected getSupportedOperators(): OperatorSupport;
-    translate(filter?: VectorFilter): VectorFilter;
+    translate(filter?: QdrantVectorFilter): QdrantVectorFilter;
     private createCondition;
     private translateNode;
     private buildFinalConditions;
@@ -56,6 +61,75 @@ export declare class QdrantFilterTranslator extends BaseFilterTranslator {
     private normalizeDatetimeRange;
 }
 
+declare type QdrantLogicalOperatorValueMap = Omit<LogicalOperatorValueMap, '$nor'>;
+
+declare type QdrantOperatorValueMap = Omit<OperatorValueMap, '$options' | '$elemMatch' | '$all'> & {
+    /**
+     * $count: Filter by array length or value count.
+     * Example: { tags: { $count: { gt: 2 } } }
+     */
+    $count: {
+        $gt?: number;
+        $gte?: number;
+        $lt?: number;
+        $lte?: number;
+        $eq?: number;
+    };
+    /**
+     * $geo: Geospatial filter.
+     * Example: { location: { $geo: { type: 'geo_radius', center: [lon, lat], radius: 1000 } } }
+     */
+    $geo: {
+        type: string;
+        [key: string]: any;
+    };
+    /**
+     * $hasId: Filter by point IDs.
+     * Allowed at root level.
+     * Example: { $hasId: '123' } or { $hasId: ['123', '456'] }
+     */
+    $hasId: string | string[];
+    /**
+     * $nested: Nested object filter.
+     * Example: { metadata: { $nested: { key: 'foo', filter: { $eq: 'bar' } } } }
+     */
+    $nested: {
+        [key: string]: any;
+    };
+    /**
+     * $hasVector: Filter by vector existence or field.
+     * Allowed at root level.
+     * Example: { $hasVector: true } or { $hasVector: 'vector_field' }
+     */
+    $hasVector: boolean | string;
+    /**
+     * $datetime: RFC 3339 datetime range.
+     * Example: { createdAt: { $datetime: { gte: '2024-01-01T00:00:00Z' } } }
+     */
+    $datetime: {
+        key?: string;
+        range?: {
+            gt?: Date | string;
+            gte?: Date | string;
+            lt?: Date | string;
+            lte?: Date | string;
+            eq?: Date | string;
+        };
+    };
+    /**
+     * $null: Check if a field is null.
+     * Example: { metadata: { $null: true } }
+     */
+    $null: boolean;
+    /**
+     * $empty: Check if an array or object field is empty.
+     * Example: { tags: { $empty: true } }
+     */
+    $empty: boolean;
+};
+
+declare type QdrantQueryVectorParams = QueryVectorParams<QdrantVectorFilter>;
+
 declare class QdrantVector extends MastraVector {
     private client;
     /**
@@ -71,8 +145,8 @@ declare class QdrantVector extends MastraVector {
     });
     upsert({ indexName, vectors, metadata, ids }: UpsertVectorParams): Promise<string[]>;
     createIndex({ indexName, dimension, metric }: CreateIndexParams): Promise<void>;
-    transformFilter(filter?: VectorFilter): VectorFilter;
-    query({ indexName, queryVector, topK, filter, includeVector, }: QueryVectorParams): Promise<QueryResult[]>;
+    transformFilter(filter?: QdrantVectorFilter): QdrantVectorFilter;
+    query({ indexName, queryVector, topK, filter, includeVector, }: QdrantQueryVectorParams): Promise<QueryResult[]>;
     listIndexes(): Promise<string[]>;
     /**
      * Retrieves statistics about a vector index.
@@ -135,4 +209,6 @@ declare class QdrantVector extends MastraVector {
 export { QdrantVector }
 export { QdrantVector as QdrantVector_alias_1 }
 
+export declare type QdrantVectorFilter = VectorFilter<keyof QdrantOperatorValueMap, QdrantOperatorValueMap, QdrantLogicalOperatorValueMap, QdrantBlacklistedRootOperators>;
+
 export { }

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

@@ -1,12 +1,15 @@
 import { BaseFilterTranslator } from '@mastra/core/vector/filter';
+import type { BlacklistedRootOperators } from '@mastra/core/vector/filter';
 import type { CreateIndexParams } from '@mastra/core/vector';
 import type { DeleteIndexParams } from '@mastra/core/vector';
 import type { DeleteVectorParams } from '@mastra/core/vector';
 import type { DescribeIndexParams } from '@mastra/core/vector';
 import type { IndexStats } from '@mastra/core/vector';
 import type { LogicalOperator } from '@mastra/core/vector/filter';
+import type { LogicalOperatorValueMap } from '@mastra/core/vector/filter';
 import { MastraVector } from '@mastra/core/vector';
 import type { OperatorSupport } from '@mastra/core/vector/filter';
+import type { OperatorValueMap } from '@mastra/core/vector/filter';
 import type { QueryResult } from '@mastra/core/vector';
 import type { QueryVectorParams } from '@mastra/core/vector';
 import type { UpdateVectorParams } from '@mastra/core/vector';
@@ -21,6 +24,8 @@ declare const QDRANT_PROMPT = "When querying Qdrant, you can ONLY use the operat
 export { QDRANT_PROMPT }
 export { QDRANT_PROMPT as QDRANT_PROMPT_alias_1 }
 
+declare type QdrantBlacklistedRootOperators = BlacklistedRootOperators | '$count' | '$geo' | '$nested' | '$datetime' | '$null' | '$empty';
+
 /**
  * Translates MongoDB-style filters to Qdrant compatible filters.
  *
@@ -40,10 +45,10 @@ export { QDRANT_PROMPT as QDRANT_PROMPT_alias_1 }
  * - $null -> is_null check
  * - $empty -> is_empty check
  */
-export declare class QdrantFilterTranslator extends BaseFilterTranslator {
+export declare class QdrantFilterTranslator extends BaseFilterTranslator<QdrantVectorFilter> {
     protected isLogicalOperator(key: string): key is LogicalOperator;
     protected getSupportedOperators(): OperatorSupport;
-    translate(filter?: VectorFilter): VectorFilter;
+    translate(filter?: QdrantVectorFilter): QdrantVectorFilter;
     private createCondition;
     private translateNode;
     private buildFinalConditions;
@@ -56,6 +61,75 @@ export declare class QdrantFilterTranslator extends BaseFilterTranslator {
     private normalizeDatetimeRange;
 }
 
+declare type QdrantLogicalOperatorValueMap = Omit<LogicalOperatorValueMap, '$nor'>;
+
+declare type QdrantOperatorValueMap = Omit<OperatorValueMap, '$options' | '$elemMatch' | '$all'> & {
+    /**
+     * $count: Filter by array length or value count.
+     * Example: { tags: { $count: { gt: 2 } } }
+     */
+    $count: {
+        $gt?: number;
+        $gte?: number;
+        $lt?: number;
+        $lte?: number;
+        $eq?: number;
+    };
+    /**
+     * $geo: Geospatial filter.
+     * Example: { location: { $geo: { type: 'geo_radius', center: [lon, lat], radius: 1000 } } }
+     */
+    $geo: {
+        type: string;
+        [key: string]: any;
+    };
+    /**
+     * $hasId: Filter by point IDs.
+     * Allowed at root level.
+     * Example: { $hasId: '123' } or { $hasId: ['123', '456'] }
+     */
+    $hasId: string | string[];
+    /**
+     * $nested: Nested object filter.
+     * Example: { metadata: { $nested: { key: 'foo', filter: { $eq: 'bar' } } } }
+     */
+    $nested: {
+        [key: string]: any;
+    };
+    /**
+     * $hasVector: Filter by vector existence or field.
+     * Allowed at root level.
+     * Example: { $hasVector: true } or { $hasVector: 'vector_field' }
+     */
+    $hasVector: boolean | string;
+    /**
+     * $datetime: RFC 3339 datetime range.
+     * Example: { createdAt: { $datetime: { gte: '2024-01-01T00:00:00Z' } } }
+     */
+    $datetime: {
+        key?: string;
+        range?: {
+            gt?: Date | string;
+            gte?: Date | string;
+            lt?: Date | string;
+            lte?: Date | string;
+            eq?: Date | string;
+        };
+    };
+    /**
+     * $null: Check if a field is null.
+     * Example: { metadata: { $null: true } }
+     */
+    $null: boolean;
+    /**
+     * $empty: Check if an array or object field is empty.
+     * Example: { tags: { $empty: true } }
+     */
+    $empty: boolean;
+};
+
+declare type QdrantQueryVectorParams = QueryVectorParams<QdrantVectorFilter>;
+
 declare class QdrantVector extends MastraVector {
     private client;
     /**
@@ -71,8 +145,8 @@ declare class QdrantVector extends MastraVector {
     });
     upsert({ indexName, vectors, metadata, ids }: UpsertVectorParams): Promise<string[]>;
     createIndex({ indexName, dimension, metric }: CreateIndexParams): Promise<void>;
-    transformFilter(filter?: VectorFilter): VectorFilter;
-    query({ indexName, queryVector, topK, filter, includeVector, }: QueryVectorParams): Promise<QueryResult[]>;
+    transformFilter(filter?: QdrantVectorFilter): QdrantVectorFilter;
+    query({ indexName, queryVector, topK, filter, includeVector, }: QdrantQueryVectorParams): Promise<QueryResult[]>;
     listIndexes(): Promise<string[]>;
     /**
      * Retrieves statistics about a vector index.
@@ -135,4 +209,6 @@ declare class QdrantVector extends MastraVector {
 export { QdrantVector }
 export { QdrantVector as QdrantVector_alias_1 }
 
+export declare type QdrantVectorFilter = VectorFilter<keyof QdrantOperatorValueMap, QdrantOperatorValueMap, QdrantLogicalOperatorValueMap, QdrantBlacklistedRootOperators>;
+
 export { }

package/dist/index.cjs

@@ -1,5 +1,6 @@
 'use strict';
 
+var error = require('@mastra/core/error');
 var vector = require('@mastra/core/vector');
 var jsClientRest = require('@qdrant/js-client-rest');
 var filter = require('@mastra/core/vector/filter');
@@ -28,7 +29,7 @@ var QdrantFilterTranslator = class extends filter.BaseFilterTranslator {
     return fieldKey ? { key: fieldKey, ...condition } : condition;
   }
   translateNode(node, isNested = false, fieldKey) {
-    if (!this.isEmpty(node) && typeof node === "object" && "must" in node) {
+    if (!this.isEmpty(node) && !!node && typeof node === "object" && "must" in node) {
       return node;
     }
     if (this.isPrimitive(node)) {
@@ -267,22 +268,46 @@ var QdrantVector = class extends vector.MastraVector {
       vector,
       payload: metadata?.[i] || {}
     }));
-    for (let i = 0; i < records.length; i += BATCH_SIZE) {
-      const batch = records.slice(i, i + BATCH_SIZE);
-      await this.client.upsert(indexName, {
-        // @ts-expect-error
-        points: batch,
-        wait: true
-      });
+    try {
+      for (let i = 0; i < records.length; i += BATCH_SIZE) {
+        const batch = records.slice(i, i + BATCH_SIZE);
+        await this.client.upsert(indexName, {
+          // @ts-expect-error
+          points: batch,
+          wait: true
+        });
+      }
+      return pointIds;
+    } catch (error$1) {
+      throw new error.MastraError(
+        {
+          id: "STORAGE_QDRANT_VECTOR_UPSERT_FAILED",
+          domain: error.ErrorDomain.STORAGE,
+          category: error.ErrorCategory.THIRD_PARTY,
+          details: { indexName, vectorCount: vectors.length }
+        },
+        error$1
+      );
     }
-    return pointIds;
   }
   async createIndex({ indexName, dimension, metric = "cosine" }) {
-    if (!Number.isInteger(dimension) || dimension <= 0) {
-      throw new Error("Dimension must be a positive integer");
-    }
-    if (!DISTANCE_MAPPING[metric]) {
-      throw new Error(`Invalid metric: "${metric}". Must be one of: cosine, euclidean, dotproduct`);
+    try {
+      if (!Number.isInteger(dimension) || dimension <= 0) {
+        throw new Error("Dimension must be a positive integer");
+      }
+      if (!DISTANCE_MAPPING[metric]) {
+        throw new Error(`Invalid metric: "${metric}". Must be one of: cosine, euclidean, dotproduct`);
+      }
+    } catch (validationError) {
+      throw new error.MastraError(
+        {
+          id: "STORAGE_QDRANT_VECTOR_CREATE_INDEX_INVALID_ARGS",
+          domain: error.ErrorDomain.STORAGE,
+          category: error.ErrorCategory.USER,
+          details: { indexName, dimension, metric }
+        },
+        validationError
+      );
     }
     try {
       await this.client.createCollection(indexName, {
@@ -291,12 +316,21 @@ var QdrantVector = class extends vector.MastraVector {
           distance: DISTANCE_MAPPING[metric]
         }
       });
-    } catch (error) {
-      const message = error?.message || error?.toString();
-      if (error?.status === 409 || typeof message === "string" && message.toLowerCase().includes("exists")) {
+    } catch (error$1) {
+      const message = error$1?.message || error$1?.toString();
+      if (error$1?.status === 409 || typeof message === "string" && message.toLowerCase().includes("exists")) {
         await this.validateExistingIndex(indexName, dimension, metric);
         return;
       }
+      throw new error.MastraError(
+        {
+          id: "STORAGE_QDRANT_VECTOR_CREATE_INDEX_FAILED",
+          domain: error.ErrorDomain.STORAGE,
+          category: error.ErrorCategory.THIRD_PARTY,
+          details: { indexName, dimension, metric }
+        },
+        error$1
+      );
     }
   }
   transformFilter(filter) {
@@ -311,33 +345,56 @@ var QdrantVector = class extends vector.MastraVector {
     includeVector = false
   }) {
     const translatedFilter = this.transformFilter(filter) ?? {};
-    const results = (await this.client.query(indexName, {
-      query: queryVector,
-      limit: topK,
-      filter: translatedFilter,
-      with_payload: true,
-      with_vector: includeVector
-    })).points;
-    return results.map((match) => {
-      let vector = [];
-      if (includeVector) {
-        if (Array.isArray(match.vector)) {
-          vector = match.vector;
-        } else if (typeof match.vector === "object" && match.vector !== null) {
-          vector = Object.values(match.vector).filter((v) => typeof v === "number");
+    try {
+      const results = (await this.client.query(indexName, {
+        query: queryVector,
+        limit: topK,
+        filter: translatedFilter,
+        with_payload: true,
+        with_vector: includeVector
+      })).points;
+      return results.map((match) => {
+        let vector = [];
+        if (includeVector) {
+          if (Array.isArray(match.vector)) {
+            vector = match.vector;
+          } else if (typeof match.vector === "object" && match.vector !== null) {
+            vector = Object.values(match.vector).filter((v) => typeof v === "number");
+          }
         }
-      }
-      return {
-        id: match.id,
-        score: match.score || 0,
-        metadata: match.payload,
-        ...includeVector && { vector }
-      };
-    });
+        return {
+          id: match.id,
+          score: match.score || 0,
+          metadata: match.payload,
+          ...includeVector && { vector }
+        };
+      });
+    } catch (error$1) {
+      throw new error.MastraError(
+        {
+          id: "STORAGE_QDRANT_VECTOR_QUERY_FAILED",
+          domain: error.ErrorDomain.STORAGE,
+          category: error.ErrorCategory.THIRD_PARTY,
+          details: { indexName, topK }
+        },
+        error$1
+      );
+    }
   }
   async listIndexes() {
-    const response = await this.client.getCollections();
-    return response.collections.map((collection) => collection.name) || [];
+    try {
+      const response = await this.client.getCollections();
+      return response.collections.map((collection) => collection.name) || [];
+    } catch (error$1) {
+      throw new error.MastraError(
+        {
+          id: "STORAGE_QDRANT_VECTOR_LIST_INDEXES_FAILED",
+          domain: error.ErrorDomain.STORAGE,
+          category: error.ErrorCategory.THIRD_PARTY
+        },
+        error$1
+      );
+    }
   }
   /**
    * Retrieves statistics about a vector index.
@@ -346,17 +403,41 @@ var QdrantVector = class extends vector.MastraVector {
    * @returns A promise that resolves to the index statistics including dimension, count and metric
    */
   async describeIndex({ indexName }) {
-    const { config, points_count } = await this.client.getCollection(indexName);
-    const distance = config.params.vectors?.distance;
-    return {
-      dimension: config.params.vectors?.size,
-      count: points_count || 0,
-      // @ts-expect-error
-      metric: Object.keys(DISTANCE_MAPPING).find((key) => DISTANCE_MAPPING[key] === distance)
-    };
+    try {
+      const { config, points_count } = await this.client.getCollection(indexName);
+      const distance = config.params.vectors?.distance;
+      return {
+        dimension: config.params.vectors?.size,
+        count: points_count || 0,
+        // @ts-expect-error
+        metric: Object.keys(DISTANCE_MAPPING).find((key) => DISTANCE_MAPPING[key] === distance)
+      };
+    } catch (error$1) {
+      throw new error.MastraError(
+        {
+          id: "STORAGE_QDRANT_VECTOR_DESCRIBE_INDEX_FAILED",
+          domain: error.ErrorDomain.STORAGE,
+          category: error.ErrorCategory.THIRD_PARTY,
+          details: { indexName }
+        },
+        error$1
+      );
+    }
   }
   async deleteIndex({ indexName }) {
-    await this.client.deleteCollection(indexName);
+    try {
+      await this.client.deleteCollection(indexName);
+    } catch (error$1) {
+      throw new error.MastraError(
+        {
+          id: "STORAGE_QDRANT_VECTOR_DELETE_INDEX_FAILED",
+          domain: error.ErrorDomain.STORAGE,
+          category: error.ErrorCategory.THIRD_PARTY,
+          details: { indexName }
+        },
+        error$1
+      );
+    }
   }
   /**
    * Updates a vector by its ID with the provided vector and/or metadata.
@@ -369,8 +450,20 @@ var QdrantVector = class extends vector.MastraVector {
    * @throws Will throw an error if no updates are provided or if the update operation fails.
    */
   async updateVector({ indexName, id, update }) {
-    if (!update.vector && !update.metadata) {
-      throw new Error("No updates provided");
+    try {
+      if (!update.vector && !update.metadata) {
+        throw new Error("No updates provided");
+      }
+    } catch (validationError) {
+      throw new error.MastraError(
+        {
+          id: "STORAGE_QDRANT_VECTOR_UPDATE_VECTOR_INVALID_ARGS",
+          domain: error.ErrorDomain.STORAGE,
+          category: error.ErrorCategory.USER,
+          details: { indexName, id }
+        },
+        validationError
+      );
     }
     const pointId = this.parsePointId(id);
     try {
@@ -400,9 +493,16 @@ var QdrantVector = class extends vector.MastraVector {
         });
         return;
       }
-    } catch (error) {
-      console.error(`Failed to update vector by id: ${id} for index name: ${indexName}:`, error);
-      throw error;
+    } catch (error$1) {
+      throw new error.MastraError(
+        {
+          id: "STORAGE_QDRANT_VECTOR_UPDATE_VECTOR_FAILED",
+          domain: error.ErrorDomain.STORAGE,
+          category: error.ErrorCategory.THIRD_PARTY,
+          details: { indexName, id }
+        },
+        error$1
+      );
     }
   }
   /**
@@ -418,8 +518,16 @@ var QdrantVector = class extends vector.MastraVector {
       await this.client.delete(indexName, {
         points: [pointId]
       });
-    } catch (error) {
-      throw new Error(`Failed to delete vector by id: ${id} for index name: ${indexName}: ${error.message}`);
+    } catch (error$1) {
+      throw new error.MastraError(
+        {
+          id: "STORAGE_QDRANT_VECTOR_DELETE_VECTOR_FAILED",
+          domain: error.ErrorDomain.STORAGE,
+          category: error.ErrorCategory.THIRD_PARTY,
+          details: { indexName, id }
+        },
+        error$1
+      );
     }
   }
   /**