@mastra/qdrant 0.1.9-alpha.0 → 0.2.0

package/.turbo/turbo-build.log

@@ -1,23 +1,23 @@
 
-> @mastra/qdrant@0.1.9-alpha.0 build /home/runner/work/mastra/mastra/stores/qdrant
+> @mastra/qdrant@0.2.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.4.0
 TSC Build start
-TSC ⚡️ Build success in 8010ms
+TSC ⚡️ Build success in 7393ms
 DTS Build start
 CLI Target: es2022
 Analysis will use the bundled TypeScript version 5.7.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.7.3
 Writing package typings: /home/runner/work/mastra/mastra/stores/qdrant/dist/_tsup-dts-rollup.d.cts
-DTS ⚡️ Build success in 7750ms
+DTS ⚡️ Build success in 7651ms
 CLI Cleaning output folder
 ESM Build start
 CJS Build start
-CJS dist/index.cjs 10.62 KB
-CJS ⚡️ Build success in 692ms
-ESM dist/index.js 10.58 KB
-ESM ⚡️ Build success in 692ms
+CJS dist/index.cjs 13.17 KB
+CJS ⚡️ Build success in 556ms
+ESM dist/index.js 13.14 KB
+ESM ⚡️ Build success in 564ms

package/CHANGELOG.md

@@ -1,5 +1,34 @@
 # @mastra/qdrant
 
+## 0.2.0
+
+### Minor Changes
+
+- 6cbf6bb: Added new operation implementations for MastraVector in qdrant store
+
+### Patch Changes
+
+- Updated dependencies [16b98d9]
+- Updated dependencies [1c8cda4]
+- Updated dependencies [95b4144]
+- Updated dependencies [3729dbd]
+- Updated dependencies [c2144f4]
+  - @mastra/core@0.6.0
+
+## 0.2.0-alpha.1
+
+### Minor Changes
+
+- 6cbf6bb: Added new operation implementations for MastraVector in qdrant store
+
+### Patch Changes
+
+- Updated dependencies [16b98d9]
+- Updated dependencies [1c8cda4]
+- Updated dependencies [95b4144]
+- Updated dependencies [c2144f4]
+  - @mastra/core@0.6.0-alpha.1
+
 ## 0.1.9-alpha.0
 
 ### Patch Changes

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

@@ -55,6 +55,41 @@ declare class QdrantVector extends MastraVector {
     listIndexes(): Promise<string[]>;
     describeIndex(indexName: string): Promise<IndexStats>;
     deleteIndex(indexName: string): Promise<void>;
+    updateIndexById(indexName: string, id: string, update: {
+        vector?: number[];
+        metadata?: Record<string, any>;
+    }): Promise<void>;
+    deleteIndexById(indexName: string, id: string): Promise<void>;
+    /**
+     * Parses and converts a string ID to the appropriate type (string or number) for Qdrant point operations.
+     *
+     * Qdrant supports both numeric and string IDs. This helper method ensures IDs are in the correct format
+     * before sending them to the Qdrant client API.
+     *
+     * @param id - The ID string to parse
+     * @returns The parsed ID as either a number (if string contains only digits) or the original string
+     *
+     * @example
+     * // Numeric ID strings are converted to numbers
+     * parsePointId("123") => 123
+     * parsePointId("42") => 42
+     * parsePointId("0") => 0
+     *
+     * // String IDs containing any non-digit characters remain as strings
+     * parsePointId("doc-123") => "doc-123"
+     * parsePointId("user_42") => "user_42"
+     * parsePointId("abc123") => "abc123"
+     * parsePointId("123abc") => "123abc"
+     * parsePointId("") => ""
+     * parsePointId("uuid-5678-xyz") => "uuid-5678-xyz"
+     *
+     * @remarks
+     * - This conversion is important because Qdrant treats numeric and string IDs differently
+     * - Only positive integers are converted to numbers (negative numbers with minus signs remain strings)
+     * - The method uses base-10 parsing, so leading zeros will be dropped in numeric conversions
+     * - reference: https://qdrant.tech/documentation/concepts/points/?q=qdrant+point+id#point-ids
+     */
+    private parsePointId;
 }
 export { QdrantVector }
 export { QdrantVector as QdrantVector_alias_1 }

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

@@ -55,6 +55,41 @@ declare class QdrantVector extends MastraVector {
     listIndexes(): Promise<string[]>;
     describeIndex(indexName: string): Promise<IndexStats>;
     deleteIndex(indexName: string): Promise<void>;
+    updateIndexById(indexName: string, id: string, update: {
+        vector?: number[];
+        metadata?: Record<string, any>;
+    }): Promise<void>;
+    deleteIndexById(indexName: string, id: string): Promise<void>;
+    /**
+     * Parses and converts a string ID to the appropriate type (string or number) for Qdrant point operations.
+     *
+     * Qdrant supports both numeric and string IDs. This helper method ensures IDs are in the correct format
+     * before sending them to the Qdrant client API.
+     *
+     * @param id - The ID string to parse
+     * @returns The parsed ID as either a number (if string contains only digits) or the original string
+     *
+     * @example
+     * // Numeric ID strings are converted to numbers
+     * parsePointId("123") => 123
+     * parsePointId("42") => 42
+     * parsePointId("0") => 0
+     *
+     * // String IDs containing any non-digit characters remain as strings
+     * parsePointId("doc-123") => "doc-123"
+     * parsePointId("user_42") => "user_42"
+     * parsePointId("abc123") => "abc123"
+     * parsePointId("123abc") => "123abc"
+     * parsePointId("") => ""
+     * parsePointId("uuid-5678-xyz") => "uuid-5678-xyz"
+     *
+     * @remarks
+     * - This conversion is important because Qdrant treats numeric and string IDs differently
+     * - Only positive integers are converted to numbers (negative numbers with minus signs remain strings)
+     * - The method uses base-10 parsing, so leading zeros will be dropped in numeric conversions
+     * - reference: https://qdrant.tech/documentation/concepts/points/?q=qdrant+point+id#point-ids
+     */
+    private parsePointId;
 }
 export { QdrantVector }
 export { QdrantVector as QdrantVector_alias_1 }

package/dist/index.cjs

@@ -337,6 +337,84 @@ var QdrantVector = class extends vector.MastraVector {
   async deleteIndex(indexName) {
     await this.client.deleteCollection(indexName);
   }
+  async updateIndexById(indexName, id, update) {
+    if (!update.vector && !update.metadata) {
+      throw new Error("No updates provided");
+    }
+    const pointId = this.parsePointId(id);
+    try {
+      if (update.metadata && !update.vector) {
+        await this.client.setPayload(indexName, { payload: update.metadata, points: [pointId] });
+        return;
+      }
+      if (update.vector && !update.metadata) {
+        await this.client.updateVectors(indexName, {
+          points: [
+            {
+              id: pointId,
+              vector: update.vector
+            }
+          ]
+        });
+        return;
+      }
+      if (update.vector && update.metadata) {
+        const point = {
+          id: pointId,
+          vector: update.vector,
+          payload: update.metadata
+        };
+        await this.client.upsert(indexName, {
+          points: [point]
+        });
+        return;
+      }
+    } catch (error) {
+      console.error("Error updating point in Qdrant:", error);
+      throw error;
+    }
+  }
+  async deleteIndexById(indexName, id) {
+    const pointId = this.parsePointId(id);
+    await this.client.delete(indexName, {
+      points: [pointId]
+    });
+  }
+  /**
+   * Parses and converts a string ID to the appropriate type (string or number) for Qdrant point operations.
+   *
+   * Qdrant supports both numeric and string IDs. This helper method ensures IDs are in the correct format
+   * before sending them to the Qdrant client API.
+   *
+   * @param id - The ID string to parse
+   * @returns The parsed ID as either a number (if string contains only digits) or the original string
+   *
+   * @example
+   * // Numeric ID strings are converted to numbers
+   * parsePointId("123") => 123
+   * parsePointId("42") => 42
+   * parsePointId("0") => 0
+   *
+   * // String IDs containing any non-digit characters remain as strings
+   * parsePointId("doc-123") => "doc-123"
+   * parsePointId("user_42") => "user_42"
+   * parsePointId("abc123") => "abc123"
+   * parsePointId("123abc") => "123abc"
+   * parsePointId("") => ""
+   * parsePointId("uuid-5678-xyz") => "uuid-5678-xyz"
+   *
+   * @remarks
+   * - This conversion is important because Qdrant treats numeric and string IDs differently
+   * - Only positive integers are converted to numbers (negative numbers with minus signs remain strings)
+   * - The method uses base-10 parsing, so leading zeros will be dropped in numeric conversions
+   * - reference: https://qdrant.tech/documentation/concepts/points/?q=qdrant+point+id#point-ids
+   */
+  parsePointId(id) {
+    if (/^\d+$/.test(id)) {
+      return parseInt(id, 10);
+    }
+    return id;
+  }
 };
 
 exports.QdrantVector = QdrantVector;

package/dist/index.js

@@ -335,6 +335,84 @@ var QdrantVector = class extends MastraVector {
   async deleteIndex(indexName) {
     await this.client.deleteCollection(indexName);
   }
+  async updateIndexById(indexName, id, update) {
+    if (!update.vector && !update.metadata) {
+      throw new Error("No updates provided");
+    }
+    const pointId = this.parsePointId(id);
+    try {
+      if (update.metadata && !update.vector) {
+        await this.client.setPayload(indexName, { payload: update.metadata, points: [pointId] });
+        return;
+      }
+      if (update.vector && !update.metadata) {
+        await this.client.updateVectors(indexName, {
+          points: [
+            {
+              id: pointId,
+              vector: update.vector
+            }
+          ]
+        });
+        return;
+      }
+      if (update.vector && update.metadata) {
+        const point = {
+          id: pointId,
+          vector: update.vector,
+          payload: update.metadata
+        };
+        await this.client.upsert(indexName, {
+          points: [point]
+        });
+        return;
+      }
+    } catch (error) {
+      console.error("Error updating point in Qdrant:", error);
+      throw error;
+    }
+  }
+  async deleteIndexById(indexName, id) {
+    const pointId = this.parsePointId(id);
+    await this.client.delete(indexName, {
+      points: [pointId]
+    });
+  }
+  /**
+   * Parses and converts a string ID to the appropriate type (string or number) for Qdrant point operations.
+   *
+   * Qdrant supports both numeric and string IDs. This helper method ensures IDs are in the correct format
+   * before sending them to the Qdrant client API.
+   *
+   * @param id - The ID string to parse
+   * @returns The parsed ID as either a number (if string contains only digits) or the original string
+   *
+   * @example
+   * // Numeric ID strings are converted to numbers
+   * parsePointId("123") => 123
+   * parsePointId("42") => 42
+   * parsePointId("0") => 0
+   *
+   * // String IDs containing any non-digit characters remain as strings
+   * parsePointId("doc-123") => "doc-123"
+   * parsePointId("user_42") => "user_42"
+   * parsePointId("abc123") => "abc123"
+   * parsePointId("123abc") => "123abc"
+   * parsePointId("") => ""
+   * parsePointId("uuid-5678-xyz") => "uuid-5678-xyz"
+   *
+   * @remarks
+   * - This conversion is important because Qdrant treats numeric and string IDs differently
+   * - Only positive integers are converted to numbers (negative numbers with minus signs remain strings)
+   * - The method uses base-10 parsing, so leading zeros will be dropped in numeric conversions
+   * - reference: https://qdrant.tech/documentation/concepts/points/?q=qdrant+point+id#point-ids
+   */
+  parsePointId(id) {
+    if (/^\d+$/.test(id)) {
+      return parseInt(id, 10);
+    }
+    return id;
+  }
 };
 
 export { QdrantVector };

package/docker-compose.yaml

@@ -0,0 +1,17 @@
+version: '3.8'
+
+services:
+  qdrant:
+    image: qdrant/qdrant:latest
+    ports:
+      - '6333:6333' # REST API port
+      - '6334:6334' # gRPC port
+    volumes:
+      - qdrant_data:/qdrant/storage
+    environment:
+      - QDRANT_LOG_LEVEL=INFO
+    restart: always
+
+volumes:
+  qdrant_data:
+    driver: local

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@mastra/qdrant",
-  "version": "0.1.9-alpha.0",
+  "version": "0.2.0",
   "description": "Qdrant vector store provider for Mastra",
   "type": "module",
   "main": "dist/index.js",
@@ -20,7 +20,7 @@
   },
   "dependencies": {
     "@qdrant/js-client-rest": "^1.13.0",
-    "@mastra/core": "^0.5.1-alpha.0"
+    "@mastra/core": "^0.6.0"
   },
   "devDependencies": {
     "@microsoft/api-extractor": "^7.52.1",

package/src/vector/index.test.ts

@@ -3,6 +3,7 @@
 import { describe, it, expect, beforeAll, afterAll, afterEach, vi, beforeEach } from 'vitest';
 
 import { QdrantVector } from './index';
+import type { QueryResult } from '@mastra/core';
 
 const dimension = 3;
 
@@ -87,6 +88,157 @@ describe('QdrantVector', () => {
     }, 50000);
   });
 
+  describe('Vector update operations', () => {
+    const testVectors = [
+      [1, 2, 3],
+      [4, 5, 6],
+      [7, 8, 9],
+    ];
+
+    beforeEach(async () => {
+      await qdrant.createIndex({ indexName: testCollectionName, dimension: 3 });
+    });
+
+    afterEach(async () => {
+      await qdrant.deleteIndex(testCollectionName);
+    });
+
+    it('should update the vector by id', async () => {
+      const ids = await qdrant.upsert({ indexName: testCollectionName, vectors: testVectors });
+      expect(ids).toHaveLength(3);
+
+      const idToBeUpdated = ids[0];
+      const newVector = [1, 2, 3];
+      const newMetaData = {
+        test: 'updates',
+      };
+
+      const update = {
+        vector: newVector,
+        metadata: newMetaData,
+      };
+
+      await qdrant.updateIndexById(testCollectionName, idToBeUpdated, update);
+
+      const results: QueryResult[] = await qdrant.query({
+        indexName: testCollectionName,
+        queryVector: newVector,
+        topK: 2,
+        includeVector: true,
+      });
+      console.log(results);
+      expect(results[0]?.id).toBe(idToBeUpdated);
+      // not matching the vector in results list because, the stored vector is stored in a normalized form inside qdrant
+      // expect(results[0]?.vector).toEqual(newVector);
+      expect(results[0]?.metadata).toEqual(newMetaData);
+    });
+
+    it('should only update the metadata by id', async () => {
+      const ids = await qdrant.upsert({ indexName: testCollectionName, vectors: testVectors });
+      expect(ids).toHaveLength(3);
+
+      const idToBeUpdated = ids[0];
+      const newMetaData = {
+        test: 'updates',
+      };
+
+      const update = {
+        metadata: newMetaData,
+      };
+
+      await qdrant.updateIndexById(testCollectionName, idToBeUpdated, update);
+
+      const results: QueryResult[] = await qdrant.query({
+        indexName: testCollectionName,
+        queryVector: testVectors[0],
+        topK: 2,
+        includeVector: true,
+      });
+      expect(results[0]?.id).toBe(idToBeUpdated);
+      // not matching the vector in results list because, the stored vector is stored in a normalized form inside qdrant
+      // expect(results[0]?.vector).toEqual(testVectors[0]);
+      expect(results[0]?.metadata).toEqual(newMetaData);
+    });
+
+    it('should only update vector embeddings by id', async () => {
+      const ids = await qdrant.upsert({ indexName: testCollectionName, vectors: testVectors });
+      expect(ids).toHaveLength(3);
+
+      const idToBeUpdated = ids[0];
+      const newVector = [1, 2, 3];
+
+      const update = {
+        vector: newVector,
+      };
+
+      await qdrant.updateIndexById(testCollectionName, idToBeUpdated, update);
+
+      const results: QueryResult[] = await qdrant.query({
+        indexName: testCollectionName,
+        queryVector: newVector,
+        topK: 2,
+        includeVector: true,
+      });
+      expect(results[0]?.id).toBe(idToBeUpdated);
+      // not matching the vector in results list because, the stored vector is stored in a normalized form inside qdrant
+      // expect(results[0]?.vector).toEqual(newVector);
+    });
+
+    it('should throw exception when no updates are given', () => {
+      expect(qdrant.updateIndexById(testCollectionName, 'id', {})).rejects.toThrow('No updates provided');
+    });
+
+    it('should throw error for non-existent index', async () => {
+      const nonExistentIndex = 'non-existent-index';
+      await expect(qdrant.updateIndexById(nonExistentIndex, 'test-id', { vector: [1, 2, 3] })).rejects.toThrow();
+    });
+
+    it('should throw error for invalid vector dimension', async () => {
+      const [id] = await qdrant.upsert({
+        indexName: testCollectionName,
+        vectors: [[1, 2, 3]],
+        metadata: [{ test: 'initial' }],
+      });
+
+      await expect(
+        qdrant.updateIndexById(testCollectionName, id, { vector: [1, 2] }), // Wrong dimension
+      ).rejects.toThrow();
+    });
+  });
+
+  describe('Vector delete operations', () => {
+    const testVectors = [
+      [1, 2, 3],
+      [4, 5, 6],
+      [7, 8, 9],
+    ];
+
+    beforeEach(async () => {
+      await qdrant.createIndex({ indexName: testCollectionName, dimension: 3 });
+    });
+
+    afterEach(async () => {
+      await qdrant.deleteIndex(testCollectionName);
+    });
+
+    it('should delete the vector by id', async () => {
+      const ids = await qdrant.upsert({ indexName: testCollectionName, vectors: testVectors });
+      expect(ids).toHaveLength(3);
+      const idToBeDeleted = ids[0];
+
+      await qdrant.deleteIndexById(testCollectionName, idToBeDeleted);
+
+      const results: QueryResult[] = await qdrant.query({
+        indexName: testCollectionName,
+        queryVector: [1.0, 0.0, 0.0],
+        topK: 2,
+      });
+
+      expect(results).toHaveLength(2);
+      expect(results.map(res => res.id)).not.toContain(idToBeDeleted);
+    });
+  });
+
   describe('Filter Queries', () => {
     const filterTestVectors = Array(10)
       .fill(null)

package/src/vector/index.ts

@@ -148,4 +148,105 @@ export class QdrantVector extends MastraVector {
   async deleteIndex(indexName: string): Promise<void> {
     await this.client.deleteCollection(indexName);
   }
+
+  async updateIndexById(
+    indexName: string,
+    id: string,
+    update: {
+      vector?: number[];
+      metadata?: Record<string, any>;
+    },
+  ): Promise<void> {
+    if (!update.vector && !update.metadata) {
+      throw new Error('No updates provided');
+    }
+
+    const pointId = this.parsePointId(id);
+
+    try {
+      // Handle metadata-only update
+      if (update.metadata && !update.vector) {
+        // For metadata-only updates, use the setPayload method
+        await this.client.setPayload(indexName, { payload: update.metadata, points: [pointId] });
+        return;
+      }
+
+      // Handle vector-only update
+      if (update.vector && !update.metadata) {
+        await this.client.updateVectors(indexName, {
+          points: [
+            {
+              id: pointId,
+              vector: update.vector,
+            },
+          ],
+        });
+        return;
+      }
+
+      // Handle both vector and metadata update
+      if (update.vector && update.metadata) {
+        const point = {
+          id: pointId,
+          vector: update.vector,
+          payload: update.metadata,
+        };
+
+        await this.client.upsert(indexName, {
+          points: [point],
+        });
+        return;
+      }
+    } catch (error) {
+      console.error('Error updating point in Qdrant:', error);
+      throw error;
+    }
+  }
+
+  async deleteIndexById(indexName: string, id: string): Promise<void> {
+    // Parse the ID - Qdrant supports both string and numeric IDs
+    const pointId = this.parsePointId(id);
+
+    // Use the Qdrant client to delete the point from the collection
+    await this.client.delete(indexName, {
+      points: [pointId],
+    });
+  }
+
+  /**
+   * Parses and converts a string ID to the appropriate type (string or number) for Qdrant point operations.
+   *
+   * Qdrant supports both numeric and string IDs. This helper method ensures IDs are in the correct format
+   * before sending them to the Qdrant client API.
+   *
+   * @param id - The ID string to parse
+   * @returns The parsed ID as either a number (if string contains only digits) or the original string
+   *
+   * @example
+   * // Numeric ID strings are converted to numbers
+   * parsePointId("123") => 123
+   * parsePointId("42") => 42
+   * parsePointId("0") => 0
+   *
+   * // String IDs containing any non-digit characters remain as strings
+   * parsePointId("doc-123") => "doc-123"
+   * parsePointId("user_42") => "user_42"
+   * parsePointId("abc123") => "abc123"
+   * parsePointId("123abc") => "123abc"
+   * parsePointId("") => ""
+   * parsePointId("uuid-5678-xyz") => "uuid-5678-xyz"
+   *
+   * @remarks
+   * - This conversion is important because Qdrant treats numeric and string IDs differently
+   * - Only positive integers are converted to numbers (negative numbers with minus signs remain strings)
+   * - The method uses base-10 parsing, so leading zeros will be dropped in numeric conversions
+   * - reference: https://qdrant.tech/documentation/concepts/points/?q=qdrant+point+id#point-ids
+   */
+  private parsePointId(id: string): string | number {
+    // Try to parse as number if it looks like one
+    if (/^\d+$/.test(id)) {
+      return parseInt(id, 10);
+    }
+    return id;
+  }
 }