@mastra/qdrant 0.10.3 → 0.11.0-alpha.1

package/dist/index.js

@@ -1,3 +1,4 @@
+import { MastraError, ErrorCategory, ErrorDomain } from '@mastra/core/error';
 import { MastraVector } from '@mastra/core/vector';
 import { QdrantClient } from '@qdrant/js-client-rest';
 import { BaseFilterTranslator } from '@mastra/core/vector/filter';
@@ -26,7 +27,7 @@ var QdrantFilterTranslator = class extends 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)) {
@@ -265,22 +266,46 @@ var QdrantVector = class extends 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) {
+      throw new MastraError(
+        {
+          id: "STORAGE_QDRANT_VECTOR_UPSERT_FAILED",
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.THIRD_PARTY,
+          details: { indexName, vectorCount: vectors.length }
+        },
+        error
+      );
     }
-    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 MastraError(
+        {
+          id: "STORAGE_QDRANT_VECTOR_CREATE_INDEX_INVALID_ARGS",
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.USER,
+          details: { indexName, dimension, metric }
+        },
+        validationError
+      );
     }
     try {
       await this.client.createCollection(indexName, {
@@ -295,6 +320,15 @@ var QdrantVector = class extends MastraVector {
         await this.validateExistingIndex(indexName, dimension, metric);
         return;
       }
+      throw new MastraError(
+        {
+          id: "STORAGE_QDRANT_VECTOR_CREATE_INDEX_FAILED",
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.THIRD_PARTY,
+          details: { indexName, dimension, metric }
+        },
+        error
+      );
     }
   }
   transformFilter(filter) {
@@ -309,33 +343,56 @@ var QdrantVector = class extends 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) {
+      throw new MastraError(
+        {
+          id: "STORAGE_QDRANT_VECTOR_QUERY_FAILED",
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.THIRD_PARTY,
+          details: { indexName, topK }
+        },
+        error
+      );
+    }
   }
   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) {
+      throw new MastraError(
+        {
+          id: "STORAGE_QDRANT_VECTOR_LIST_INDEXES_FAILED",
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.THIRD_PARTY
+        },
+        error
+      );
+    }
   }
   /**
    * Retrieves statistics about a vector index.
@@ -344,17 +401,41 @@ var QdrantVector = class extends 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) {
+      throw new MastraError(
+        {
+          id: "STORAGE_QDRANT_VECTOR_DESCRIBE_INDEX_FAILED",
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.THIRD_PARTY,
+          details: { indexName }
+        },
+        error
+      );
+    }
   }
   async deleteIndex({ indexName }) {
-    await this.client.deleteCollection(indexName);
+    try {
+      await this.client.deleteCollection(indexName);
+    } catch (error) {
+      throw new MastraError(
+        {
+          id: "STORAGE_QDRANT_VECTOR_DELETE_INDEX_FAILED",
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.THIRD_PARTY,
+          details: { indexName }
+        },
+        error
+      );
+    }
   }
   /**
    * Updates a vector by its ID with the provided vector and/or metadata.
@@ -367,8 +448,20 @@ var QdrantVector = class extends 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 MastraError(
+        {
+          id: "STORAGE_QDRANT_VECTOR_UPDATE_VECTOR_INVALID_ARGS",
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.USER,
+          details: { indexName, id }
+        },
+        validationError
+      );
     }
     const pointId = this.parsePointId(id);
     try {
@@ -399,8 +492,15 @@ var QdrantVector = class extends MastraVector {
         return;
       }
     } catch (error) {
-      console.error(`Failed to update vector by id: ${id} for index name: ${indexName}:`, error);
-      throw error;
+      throw new MastraError(
+        {
+          id: "STORAGE_QDRANT_VECTOR_UPDATE_VECTOR_FAILED",
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.THIRD_PARTY,
+          details: { indexName, id }
+        },
+        error
+      );
     }
   }
   /**
@@ -417,7 +517,15 @@ var QdrantVector = class extends MastraVector {
         points: [pointId]
       });
     } catch (error) {
-      throw new Error(`Failed to delete vector by id: ${id} for index name: ${indexName}: ${error.message}`);
+      throw new MastraError(
+        {
+          id: "STORAGE_QDRANT_VECTOR_DELETE_VECTOR_FAILED",
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.THIRD_PARTY,
+          details: { indexName, id }
+        },
+        error
+      );
     }
   }
   /**

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@mastra/qdrant",
-  "version": "0.10.3",
+  "version": "0.11.0-alpha.1",
   "description": "Qdrant vector store provider for Mastra",
   "type": "module",
   "main": "dist/index.js",
@@ -25,15 +25,15 @@
   "devDependencies": {
     "@microsoft/api-extractor": "^7.52.8",
     "@types/node": "^20.19.0",
-    "eslint": "^9.28.0",
+    "eslint": "^9.29.0",
     "tsup": "^8.5.0",
     "typescript": "^5.8.3",
     "vitest": "^3.2.3",
     "@internal/lint": "0.0.13",
-    "@mastra/core": "0.10.6"
+    "@mastra/core": "0.10.7-alpha.3"
   },
   "peerDependencies": {
-    "@mastra/core": ">=0.10.4-0 <0.11.0"
+    "@mastra/core": ">=0.10.7-0 <0.11.0-0"
   },
   "scripts": {
     "build": "tsup src/index.ts --format esm,cjs --experimental-dts --clean --treeshake=smallest --splitting",

package/src/vector/filter.test.ts

@@ -1,5 +1,5 @@
 import { describe, it, expect } from 'vitest';
-
+import type { QdrantVectorFilter } from './filter';
 import { QdrantFilterTranslator } from './filter';
 
 describe('QdrantFilterTranslator', () => {
@@ -7,7 +7,7 @@ describe('QdrantFilterTranslator', () => {
 
   describe('Basic Operators', () => {
     it('should translate direct value match', () => {
-      const filter = { field: 'value' };
+      const filter: QdrantVectorFilter = { field: 'value' };
       const expected = { must: [{ key: 'field', match: { value: 'value' } }] };
       expect(translator.translate(filter)).toEqual(expected);
     });
@@ -63,7 +63,7 @@ describe('QdrantFilterTranslator', () => {
     });
 
     it('should handle multiple comparison operators on same field', () => {
-      const filter = { field: { $gt: 10, $lt: 20 } };
+      const filter: QdrantVectorFilter = { field: { $gt: 10, $lt: 20 } };
       const expected = { must: [{ key: 'field', range: { gt: 10, lt: 20 } }] };
       expect(translator.translate(filter)).toEqual(expected);
     });
@@ -71,7 +71,7 @@ describe('QdrantFilterTranslator', () => {
 
   describe('Logical Operators', () => {
     it('should translate $and operator', () => {
-      const filter = {
+      const filter: QdrantVectorFilter = {
         $and: [{ field1: 'value1' }, { field2: { $gt: 100 } }],
       };
 
@@ -86,7 +86,7 @@ describe('QdrantFilterTranslator', () => {
     });
 
     it('should translate $or operator', () => {
-      const filter = {
+      const filter: QdrantVectorFilter = {
         $or: [{ field1: 'value1' }, { field2: { $lt: 100 } }],
       };
 
@@ -101,7 +101,7 @@ describe('QdrantFilterTranslator', () => {
     });
 
     it('should translate $not operator', () => {
-      const filter = {
+      const filter: QdrantVectorFilter = {
         $not: { field: 'value' },
       };
 
@@ -113,7 +113,7 @@ describe('QdrantFilterTranslator', () => {
     });
 
     it('should handle nested logical operators', () => {
-      const filter = {
+      const filter: QdrantVectorFilter = {
         $and: [
           { 'user.age': { $gte: 18 } },
           {
@@ -155,7 +155,7 @@ describe('QdrantFilterTranslator', () => {
     });
 
     it('should handle nested must_not operators', () => {
-      const filter = {
+      const filter: QdrantVectorFilter = {
         $not: {
           $not: { field: 'value' },
         },
@@ -171,7 +171,7 @@ describe('QdrantFilterTranslator', () => {
     });
 
     it('should handle complex logical combinations with ranges', () => {
-      const filter = {
+      const filter: QdrantVectorFilter = {
         $or: [
           { $and: [{ price: { $gte: 100, $lt: 200 } }, { stock: { $gt: 0 } }] },
           { $and: [{ price: { $lt: 100 } }, { featured: true }] },
@@ -199,13 +199,13 @@ describe('QdrantFilterTranslator', () => {
 
   describe('Custom Operators', () => {
     it('should translate $count operator', () => {
-      const filter = { field: { $count: { $gt: 5 } } };
+      const filter: QdrantVectorFilter = { field: { $count: { $gt: 5 } } };
       const expected = { must: [{ key: 'field', values_count: { gt: 5 } }] };
       expect(translator.translate(filter)).toEqual(expected);
     });
 
     it('should translate $geo operator with radius', () => {
-      const filter = {
+      const filter: QdrantVectorFilter = {
         location: {
           $geo: {
             type: 'radius',
@@ -229,7 +229,7 @@ describe('QdrantFilterTranslator', () => {
     });
 
     it('should translate $geo operator with bounding box', () => {
-      const filter = {
+      const filter: QdrantVectorFilter = {
         location: {
           $geo: {
             type: 'box',
@@ -255,7 +255,7 @@ describe('QdrantFilterTranslator', () => {
     });
 
     it('should translate $geo operator with polygon', () => {
-      const filter = {
+      const filter: QdrantVectorFilter = {
         location: {
           $geo: {
             type: 'polygon',
@@ -286,7 +286,7 @@ describe('QdrantFilterTranslator', () => {
     });
 
     it('should translate $nested operator', () => {
-      const filter = {
+      const filter: QdrantVectorFilter = {
         diet: {
           $nested: {
             food: 'meat',
@@ -322,7 +322,7 @@ describe('QdrantFilterTranslator', () => {
 
     it('should translate $datetime operator', () => {
       const now = new Date();
-      const filter = {
+      const filter: QdrantVectorFilter = {
         timestamp: {
           $datetime: {
             range: {
@@ -357,7 +357,7 @@ describe('QdrantFilterTranslator', () => {
     });
 
     it('should handle nested $count with multiple conditions', () => {
-      const filter = { 'array.items': { $count: { $gt: 5, $lt: 10 } } };
+      const filter: QdrantVectorFilter = { 'array.items': { $count: { $gt: 5, $lt: 10 } } };
       const expected = {
         must: [
           {
@@ -370,7 +370,7 @@ describe('QdrantFilterTranslator', () => {
     });
 
     it('should translate $nested operator with complex conditions', () => {
-      const filter = {
+      const filter: QdrantVectorFilter = {
         nested_field: {
           $nested: {
             inner_field: { $gt: 100 },
@@ -399,7 +399,7 @@ describe('QdrantFilterTranslator', () => {
 
     it('should translate $datetime operator with multiple range conditions', () => {
       const now = new Date();
-      const filter = {
+      const filter: QdrantVectorFilter = {
         timestamp: {
           $datetime: {
             range: {
@@ -430,7 +430,7 @@ describe('QdrantFilterTranslator', () => {
     });
 
     it('should translate $datetime operator with string dates', () => {
-      const filter = {
+      const filter: QdrantVectorFilter = {
         timestamp: {
           $datetime: {
             key: 'timestamp',
@@ -458,7 +458,7 @@ describe('QdrantFilterTranslator', () => {
     });
 
     it('should translate complex $nested operator', () => {
-      const filter = {
+      const filter: QdrantVectorFilter = {
         diet: {
           $nested: {
             food: { $in: ['meat', 'fish'] },
@@ -491,19 +491,19 @@ describe('QdrantFilterTranslator', () => {
 
   describe('Special Cases', () => {
     it('should handle nested paths', () => {
-      const filter = { 'obj.field': 'value' };
+      const filter: QdrantVectorFilter = { 'obj.field': 'value' };
       const expected = { must: [{ key: 'obj.field', match: { value: 'value' } }] };
       expect(translator.translate(filter)).toEqual(expected);
     });
 
     it('should handle deep nested paths', () => {
-      const filter = { 'a.b.c.d': { $gt: 100 } };
+      const filter: QdrantVectorFilter = { 'a.b.c.d': { $gt: 100 } };
       const expected = { must: [{ key: 'a.b.c.d', range: { gt: 100 } }] };
       expect(translator.translate(filter)).toEqual(expected);
     });
 
     it('should handle complex combinations', () => {
-      const filter = {
+      const filter: QdrantVectorFilter = {
         $and: [
           { 'user.age': { $gte: 18 } },
           {
@@ -534,7 +534,7 @@ describe('QdrantFilterTranslator', () => {
     });
 
     it('should handle multiple nested paths with same prefix', () => {
-      const filter = {
+      const filter: QdrantVectorFilter = {
         $and: [{ 'user.profile.age': { $gte: 18 } }, { 'user.profile.name': 'John' }],
       };
 
@@ -549,7 +549,7 @@ describe('QdrantFilterTranslator', () => {
     });
 
     it('should handle mixed array and object paths', () => {
-      const filter = {
+      const filter: QdrantVectorFilter = {
         'items[].category': { $in: ['A', 'B'] },
         'items[].details.price': { $gt: 100 },
       };
@@ -565,7 +565,7 @@ describe('QdrantFilterTranslator', () => {
     });
 
     it('should handle empty logical operators in combination', () => {
-      const filter = {
+      const filter: QdrantVectorFilter = {
         $and: [{ $or: [] }, { field: 'value' }],
       };
       const expected = {
@@ -575,7 +575,7 @@ describe('QdrantFilterTranslator', () => {
     });
 
     it('should handle deeply nested paths with array notation', () => {
-      const filter = {
+      const filter: QdrantVectorFilter = {
         'users[].addresses[].geo.location': {
           $geo: {
             type: 'radius',
@@ -599,7 +599,7 @@ describe('QdrantFilterTranslator', () => {
     });
 
     it('should handle array paths with multiple levels', () => {
-      const filter = {
+      const filter: QdrantVectorFilter = {
         'users[].addresses[].location[].coordinates': { $gt: 100 },
       };
       const expected = {
@@ -614,7 +614,7 @@ describe('QdrantFilterTranslator', () => {
     });
 
     it('should handle combination of array and object paths', () => {
-      const filter = {
+      const filter: QdrantVectorFilter = {
         'users[].profile.addresses[].location.coordinates': { $gt: 100 },
       };
       const expected = {
@@ -629,7 +629,7 @@ describe('QdrantFilterTranslator', () => {
     });
 
     it('should handle multiple conditions with same array path', () => {
-      const filter = {
+      const filter: QdrantVectorFilter = {
         $and: [
           { 'items[].price': { $gt: 100 } },
           { 'items[].quantity': { $gt: 0 } },
@@ -654,7 +654,7 @@ describe('QdrantFilterTranslator', () => {
     });
 
     it('should throw error for invalid geo filter type', () => {
-      const filter = {
+      const filter: QdrantVectorFilter = {
         location: {
           $geo: {
             type: 'invalid',
@@ -667,7 +667,7 @@ describe('QdrantFilterTranslator', () => {
     });
 
     it('should throw error for invalid custom operator', () => {
-      const filter = { $invalidCustom: 'value' };
+      const filter: QdrantVectorFilter = { $invalidCustom: 'value' };
       expect(() => translator.translate(filter)).toThrow();
     });
   });
@@ -686,7 +686,7 @@ describe('QdrantFilterTranslator', () => {
     });
 
     it('should validate array operator values', () => {
-      const invalidFilters = [
+      const invalidFilters: any = [
         { field: { $in: 'not-an-array' } }, // Should be array
         { field: { $nin: 123 } }, // Should be array
         { field: { $in: {} } }, // Should be array
@@ -697,14 +697,14 @@ describe('QdrantFilterTranslator', () => {
       });
     });
     it('throws error for non-logical operators at top level', () => {
-      const invalidFilters = [{ $gt: 100 }, { $in: ['value1', 'value2'] }, { $eq: true }];
+      const invalidFilters: any = [{ $gt: 100 }, { $in: ['value1', 'value2'] }, { $eq: true }];
 
       invalidFilters.forEach(filter => {
         expect(() => translator.translate(filter)).toThrow(/Invalid top-level operator/);
       });
     });
     it('allows logical operators at top level', () => {
-      const validFilters = [{ $and: [{ field: 'value' }] }, { $or: [{ field: 'value' }] }];
+      const validFilters: QdrantVectorFilter[] = [{ $and: [{ field: 'value' }] }, { $or: [{ field: 'value' }] }];
 
       validFilters.forEach(filter => {
         expect(() => translator.translate(filter)).not.toThrow();
@@ -732,7 +732,7 @@ describe('QdrantFilterTranslator', () => {
     });
 
     it('should wrap multiple field conditions in single must', () => {
-      const filter = {
+      const filter: QdrantVectorFilter = {
         field1: 'value1',
         field2: 'value2',
       };
@@ -746,7 +746,7 @@ describe('QdrantFilterTranslator', () => {
     });
 
     it('should wrap complex single field conditions', () => {
-      const filter = {
+      const filter: QdrantVectorFilter = {
         field: {
           $gt: 10,
           $lt: 20,
@@ -771,7 +771,7 @@ describe('QdrantFilterTranslator', () => {
 
   describe('No Must Wrapper Cases', () => {
     it('should not wrap nested logical operators', () => {
-      const filter = {
+      const filter: QdrantVectorFilter = {
         $and: [{ field1: 'value1' }, { field2: 'value2' }],
       };
       const expected = {
@@ -791,7 +791,7 @@ describe('QdrantFilterTranslator', () => {
     });
 
     it('should preserve existing logical structure', () => {
-      const filter = {
+      const filter: QdrantVectorFilter = {
         $or: [{ field1: 'value1' }, { $and: [{ field2: 'value2' }, { field3: 'value3' }] }],
       };
       const expected = {
@@ -826,7 +826,7 @@ describe('QdrantFilterTranslator', () => {
     });
 
     it('should handle regex in nested conditions', () => {
-      const filter = {
+      const filter: QdrantVectorFilter = {
         diet: {
           $nested: {
             food: { $regex: 'meat' },