@ductape/sdk 0.0.4-v9 → 0.0.4-v91

package/dist/vector/adapters/qdrant.adapter.js

@@ -0,0 +1,442 @@
+"use strict";
+/**
+ * Qdrant Adapter
+ *
+ * Vector database adapter for Qdrant.
+ * https://qdrant.tech/
+ */
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.QdrantAdapter = void 0;
+const types_1 = require("../types");
+const base_adapter_1 = require("./base.adapter");
+const utils_1 = require("../utils");
+/**
+ * Qdrant adapter implementation
+ */
+class QdrantAdapter extends base_adapter_1.BaseVectorAdapter {
+    constructor() {
+        super(...arguments);
+        this.type = types_1.VectorDBType.QDRANT;
+        this.apiKey = '';
+        this.baseUrl = '';
+        this.collectionName = '';
+        this.timeout = 30000;
+    }
+    // ==================== CONNECTION ====================
+    async connect(options) {
+        var _a;
+        this.connectionOptions = options;
+        this.apiKey = options.apiKey || '';
+        this.baseUrl = options.endpoint.replace(/\/$/, ''); // Remove trailing slash
+        this.collectionName = options.index;
+        this.timeout = ((_a = options.options) === null || _a === void 0 ? void 0 : _a.timeout) || 30000;
+        try {
+            // First, test basic connectivity by listing collections
+            const canConnect = await this.testConnection();
+            if (!canConnect) {
+                this._connected = false;
+                return {
+                    connected: false,
+                    error: 'Failed to connect to Qdrant server',
+                };
+            }
+            // Try to get collection info if it exists
+            try {
+                const info = await this.describeIndex();
+                this._connected = true;
+                return {
+                    connected: true,
+                    index: this.collectionName,
+                    dimensions: info.dimensions,
+                    vectorCount: info.vectorCount,
+                };
+            }
+            catch (_b) {
+                // Collection might not exist yet, but we're connected to the server
+                this._connected = true;
+                return {
+                    connected: true,
+                    index: this.collectionName,
+                    dimensions: options.dimensions,
+                    vectorCount: 0,
+                };
+            }
+        }
+        catch (error) {
+            this._connected = false;
+            return {
+                connected: false,
+                error: error.message || 'Failed to connect to Qdrant',
+            };
+        }
+    }
+    async disconnect() {
+        this._connected = false;
+        this.connectionOptions = null;
+    }
+    async testConnection() {
+        try {
+            await this.request('GET', '/collections');
+            return true;
+        }
+        catch (_a) {
+            return false;
+        }
+    }
+    // ==================== VECTOR OPERATIONS ====================
+    async upsert(options) {
+        this.validateDimensions(options.vectors);
+        const points = options.vectors.map((v) => ({
+            id: v.id,
+            vector: v.values,
+            payload: v.metadata || {},
+        }));
+        const body = { points };
+        await this.request('PUT', `/collections/${this.collectionName}/points`, body);
+        return {
+            upsertedCount: points.length,
+            upsertedIds: options.vectors.map((v) => v.id),
+        };
+    }
+    async query(options) {
+        var _a, _b;
+        const startTime = Date.now();
+        const body = {
+            vector: options.vector,
+            limit: options.topK,
+            with_vector: (_a = options.includeValues) !== null && _a !== void 0 ? _a : false,
+            with_payload: (_b = options.includeMetadata) !== null && _b !== void 0 ? _b : true,
+        };
+        if (options.filter) {
+            body.filter = this.convertFilter(options.filter);
+        }
+        if (options.minScore) {
+            body.score_threshold = options.minScore;
+        }
+        const response = await this.request('POST', `/collections/${this.collectionName}/points/search`, body);
+        const matches = (response.result || []).map((m) => ({
+            id: String(m.id),
+            score: m.score,
+            values: m.vector,
+            metadata: m.payload,
+        }));
+        return {
+            matches,
+            queryTime: Date.now() - startTime,
+        };
+    }
+    async fetch(options) {
+        var _a;
+        const body = {
+            ids: options.ids,
+            with_vector: (_a = options.includeValues) !== null && _a !== void 0 ? _a : true,
+            with_payload: true,
+        };
+        const response = await this.request('POST', `/collections/${this.collectionName}/points`, body);
+        const vectors = {};
+        for (const point of response.result || []) {
+            vectors[String(point.id)] = {
+                id: String(point.id),
+                values: point.vector,
+                metadata: point.payload,
+            };
+        }
+        // Mark missing IDs as null
+        for (const id of options.ids) {
+            if (!vectors[id]) {
+                vectors[id] = null;
+            }
+        }
+        return { vectors };
+    }
+    async update(options) {
+        // Qdrant requires full point update or payload update separately
+        if (options.values) {
+            // Update vector values
+            await this.request('PUT', `/collections/${this.collectionName}/points`, {
+                points: [
+                    {
+                        id: options.id,
+                        vector: options.values,
+                        payload: options.setMetadata || options.mergeMetadata || {},
+                    },
+                ],
+            });
+        }
+        else if (options.setMetadata) {
+            // Set payload (replace)
+            await this.request('POST', `/collections/${this.collectionName}/points/payload`, {
+                payload: options.setMetadata,
+                points: [options.id],
+            });
+        }
+        else if (options.mergeMetadata) {
+            // Update payload (merge)
+            await this.request('POST', `/collections/${this.collectionName}/points/payload`, {
+                payload: options.mergeMetadata,
+                points: [options.id],
+            });
+        }
+        return {
+            success: true,
+            id: options.id,
+        };
+    }
+    async delete(options) {
+        let deletedCount = 0;
+        if (options.ids) {
+            await this.request('POST', `/collections/${this.collectionName}/points/delete`, { points: options.ids });
+            deletedCount = options.ids.length;
+        }
+        else if (options.filter) {
+            await this.request('POST', `/collections/${this.collectionName}/points/delete`, { filter: this.convertFilter(options.filter) });
+            deletedCount = -1; // Unknown count
+        }
+        else if (options.deleteAll) {
+            // Delete collection and recreate
+            const info = await this.describeIndex();
+            await this.deleteIndex({ name: this.collectionName });
+            await this.createIndex({
+                name: this.collectionName,
+                dimensions: info.dimensions,
+                metric: info.metric,
+            });
+            deletedCount = info.vectorCount;
+        }
+        return {
+            deletedCount,
+            deletedIds: options.ids,
+        };
+    }
+    async list(options) {
+        var _a, _b;
+        const body = {
+            limit: options.limit || 100,
+            with_vector: false,
+            with_payload: false,
+        };
+        if (options.cursor) {
+            body.offset = options.cursor;
+        }
+        const response = await this.request('POST', `/collections/${this.collectionName}/points/scroll`, body);
+        return {
+            ids: (((_a = response.result) === null || _a === void 0 ? void 0 : _a.points) || []).map((p) => String(p.id)),
+            nextCursor: (_b = response.result) === null || _b === void 0 ? void 0 : _b.next_page_offset,
+        };
+    }
+    // ==================== NAMESPACE OPERATIONS ====================
+    async listNamespaces() {
+        var _a;
+        // Qdrant doesn't have namespaces, collections are the top level
+        const response = await this.request('GET', '/collections');
+        const namespaces = await Promise.all((((_a = response.result) === null || _a === void 0 ? void 0 : _a.collections) || []).map(async (c) => {
+            var _a;
+            let vectorCount = 0;
+            try {
+                const collectionInfo = await this.request('GET', `/collections/${c.name}`);
+                vectorCount = ((_a = collectionInfo.result) === null || _a === void 0 ? void 0 : _a.points_count) || 0;
+            }
+            catch (_b) {
+                // Ignore errors for individual collections
+            }
+            return {
+                name: c.name,
+                vectorCount,
+            };
+        }));
+        return { namespaces };
+    }
+    async deleteNamespace(namespace) {
+        await this.deleteIndex({ name: namespace });
+    }
+    // ==================== INDEX OPERATIONS ====================
+    async describeIndex() {
+        var _a, _b, _c, _d, _e, _f, _g;
+        const response = await this.request('GET', `/collections/${this.collectionName}`);
+        const config = (_a = response.result) === null || _a === void 0 ? void 0 : _a.config;
+        const status = (_b = response.result) === null || _b === void 0 ? void 0 : _b.status;
+        return {
+            name: this.collectionName,
+            dimensions: ((_d = (_c = config === null || config === void 0 ? void 0 : config.params) === null || _c === void 0 ? void 0 : _c.vectors) === null || _d === void 0 ? void 0 : _d.size) || this.dimensions,
+            metric: this.mapQdrantMetric((_f = (_e = config === null || config === void 0 ? void 0 : config.params) === null || _e === void 0 ? void 0 : _e.vectors) === null || _f === void 0 ? void 0 : _f.distance),
+            vectorCount: ((_g = response.result) === null || _g === void 0 ? void 0 : _g.points_count) || 0,
+            status: status === 'green' ? 'ready' : 'initializing',
+        };
+    }
+    async getIndexStats() {
+        const info = await this.describeIndex();
+        return {
+            vectorCount: info.vectorCount,
+            dimensions: info.dimensions,
+        };
+    }
+    async createIndex(options) {
+        const body = {
+            vectors: {
+                size: options.dimensions,
+                distance: this.mapToQdrantMetric(options.metric || types_1.DistanceMetric.COSINE),
+            },
+        };
+        await this.request('PUT', `/collections/${options.name}`, body);
+        return {
+            success: true,
+            name: options.name,
+        };
+    }
+    async deleteIndex(options) {
+        await this.request('DELETE', `/collections/${options.name}`);
+        return { success: true };
+    }
+    async listIndexes() {
+        var _a;
+        const response = await this.request('GET', '/collections');
+        return Promise.all((((_a = response.result) === null || _a === void 0 ? void 0 : _a.collections) || []).map(async (c) => {
+            var _a, _b, _c, _d, _e, _f, _g;
+            let vectorCount = 0;
+            let dimensions = -1;
+            let metric = types_1.DistanceMetric.COSINE;
+            let status = 'ready';
+            try {
+                const collectionInfo = await this.request('GET', `/collections/${c.name}`);
+                const config = (_a = collectionInfo.result) === null || _a === void 0 ? void 0 : _a.config;
+                vectorCount = ((_b = collectionInfo.result) === null || _b === void 0 ? void 0 : _b.points_count) || 0;
+                dimensions = ((_d = (_c = config === null || config === void 0 ? void 0 : config.params) === null || _c === void 0 ? void 0 : _c.vectors) === null || _d === void 0 ? void 0 : _d.size) || -1;
+                metric = this.mapQdrantMetric((_f = (_e = config === null || config === void 0 ? void 0 : config.params) === null || _e === void 0 ? void 0 : _e.vectors) === null || _f === void 0 ? void 0 : _f.distance);
+                status = ((_g = collectionInfo.result) === null || _g === void 0 ? void 0 : _g.status) === 'green' ? 'ready' : 'initializing';
+            }
+            catch (_h) {
+                // Ignore errors for individual collections
+            }
+            return {
+                name: c.name,
+                dimensions,
+                metric,
+                vectorCount,
+                status,
+            };
+        }));
+    }
+    // ==================== FEATURE DETECTION ====================
+    supportsFeature(feature) {
+        switch (feature) {
+            case types_1.VectorFeature.METADATA_FILTERING:
+            case types_1.VectorFeature.BATCH_OPERATIONS:
+            case types_1.VectorFeature.VECTOR_UPDATES:
+            case types_1.VectorFeature.INDEX_MANAGEMENT:
+                return true;
+            case types_1.VectorFeature.NAMESPACES:
+            case types_1.VectorFeature.HYBRID_SEARCH:
+            case types_1.VectorFeature.SPARSE_VECTORS:
+            case types_1.VectorFeature.AGGREGATIONS:
+                return false;
+            default:
+                return false;
+        }
+    }
+    // ==================== PRIVATE METHODS ====================
+    /**
+     * Make a request to Qdrant
+     */
+    async request(method, path, body) {
+        const url = `${this.baseUrl}${path}`;
+        const headers = {
+            'Content-Type': 'application/json',
+        };
+        if (this.apiKey) {
+            headers['api-key'] = this.apiKey;
+        }
+        const options = {
+            method,
+            headers,
+            signal: AbortSignal.timeout(this.timeout),
+        };
+        if (body && method !== 'GET') {
+            options.body = JSON.stringify(body);
+        }
+        const response = await fetch(url, options);
+        if (!response.ok) {
+            const errorText = await response.text();
+            throw utils_1.VectorError.fromUnknown(new Error(`Qdrant API error: ${response.status} - ${errorText}`), types_1.VectorDBType.QDRANT);
+        }
+        return response.json();
+    }
+    /**
+     * Convert Ductape filter to Qdrant format
+     */
+    convertFilter(filter) {
+        if ('$and' in filter || '$or' in filter) {
+            const group = filter;
+            if (group.$and) {
+                return {
+                    must: group.$and.map((f) => this.convertFilter(f)),
+                };
+            }
+            if (group.$or) {
+                return {
+                    should: group.$or.map((f) => this.convertFilter(f)),
+                };
+            }
+        }
+        const f = filter;
+        return this.convertSingleFilter(f);
+    }
+    /**
+     * Convert single filter to Qdrant format
+     */
+    convertSingleFilter(filter) {
+        const { field, operator, value } = filter;
+        switch (operator) {
+            case '$eq':
+                return { key: field, match: { value } };
+            case '$ne':
+                return {
+                    must_not: [{ key: field, match: { value } }],
+                };
+            case '$gt':
+                return { key: field, range: { gt: value } };
+            case '$gte':
+                return { key: field, range: { gte: value } };
+            case '$lt':
+                return { key: field, range: { lt: value } };
+            case '$lte':
+                return { key: field, range: { lte: value } };
+            case '$in':
+                return { key: field, match: { any: value } };
+            default:
+                return { key: field, match: { value } };
+        }
+    }
+    /**
+     * Map Qdrant distance to our enum
+     */
+    mapQdrantMetric(distance) {
+        switch (distance === null || distance === void 0 ? void 0 : distance.toLowerCase()) {
+            case 'cosine':
+                return types_1.DistanceMetric.COSINE;
+            case 'euclid':
+                return types_1.DistanceMetric.EUCLIDEAN;
+            case 'dot':
+                return types_1.DistanceMetric.DOT_PRODUCT;
+            default:
+                return types_1.DistanceMetric.COSINE;
+        }
+    }
+    /**
+     * Map our enum to Qdrant distance
+     */
+    mapToQdrantMetric(metric) {
+        switch (metric) {
+            case types_1.DistanceMetric.COSINE:
+                return 'Cosine';
+            case types_1.DistanceMetric.EUCLIDEAN:
+                return 'Euclid';
+            case types_1.DistanceMetric.DOT_PRODUCT:
+                return 'Dot';
+            default:
+                return 'Cosine';
+        }
+    }
+}
+exports.QdrantAdapter = QdrantAdapter;
+exports.default = QdrantAdapter;
+//# sourceMappingURL=qdrant.adapter.js.map

package/dist/vector/adapters/qdrant.adapter.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"qdrant.adapter.js","sourceRoot":"","sources":["../../../src/vector/adapters/qdrant.adapter.ts"],"names":[],"mappings":";AAAA;;;;;GAKG;;;AAEH,oCA0BkB;AAClB,iDAA8E;AAC9E,oCAAuC;AAEvC;;GAEG;AACH,MAAa,aAAc,SAAQ,gCAAiB;IAApD;;QACkB,SAAI,GAAG,oBAAY,CAAC,MAAM,CAAC;QAEnC,WAAM,GAAW,EAAE,CAAC;QACpB,YAAO,GAAW,EAAE,CAAC;QACrB,mBAAc,GAAW,EAAE,CAAC;QAC5B,YAAO,GAAW,KAAK,CAAC;IA+flC,CAAC;IA7fC,uDAAuD;IAEvD,KAAK,CAAC,OAAO,CAAC,OAAkC;;QAC9C,IAAI,CAAC,iBAAiB,GAAG,OAAO,CAAC;QACjC,IAAI,CAAC,MAAM,GAAG,OAAO,CAAC,MAAM,IAAI,EAAE,CAAC;QACnC,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC,QAAQ,CAAC,OAAO,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC,CAAC,wBAAwB;QAC5E,IAAI,CAAC,cAAc,GAAG,OAAO,CAAC,KAAK,CAAC;QACpC,IAAI,CAAC,OAAO,GAAG,CAAC,MAAA,OAAO,CAAC,OAAO,0CAAE,OAAkB,KAAI,KAAK,CAAC;QAE7D,IAAI,CAAC;YACH,wDAAwD;YACxD,MAAM,UAAU,GAAG,MAAM,IAAI,CAAC,cAAc,EAAE,CAAC;YAC/C,IAAI,CAAC,UAAU,EAAE,CAAC;gBAChB,IAAI,CAAC,UAAU,GAAG,KAAK,CAAC;gBACxB,OAAO;oBACL,SAAS,EAAE,KAAK;oBAChB,KAAK,EAAE,oCAAoC;iBAC5C,CAAC;YACJ,CAAC;YAED,0CAA0C;YAC1C,IAAI,CAAC;gBACH,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,aAAa,EAAE,CAAC;gBACxC,IAAI,CAAC,UAAU,GAAG,IAAI,CAAC;gBACvB,OAAO;oBACL,SAAS,EAAE,IAAI;oBACf,KAAK,EAAE,IAAI,CAAC,cAAc;oBAC1B,UAAU,EAAE,IAAI,CAAC,UAAU;oBAC3B,WAAW,EAAE,IAAI,CAAC,WAAW;iBAC9B,CAAC;YACJ,CAAC;YAAC,WAAM,CAAC;gBACP,oEAAoE;gBACpE,IAAI,CAAC,UAAU,GAAG,IAAI,CAAC;gBACvB,OAAO;oBACL,SAAS,EAAE,IAAI;oBACf,KAAK,EAAE,IAAI,CAAC,cAAc;oBAC1B,UAAU,EAAE,OAAO,CAAC,UAAU;oBAC9B,WAAW,EAAE,CAAC;iBACf,CAAC;YACJ,CAAC;QACH,CAAC;QAAC,OAAO,KAAU,EAAE,CAAC;YACpB,IAAI,CAAC,UAAU,GAAG,KAAK,CAAC;YACxB,OAAO;gBACL,SAAS,EAAE,KAAK;gBAChB,KAAK,EAAE,KAAK,CAAC,OAAO,IAAI,6BAA6B;aACtD,CAAC;QACJ,CAAC;IACH,CAAC;IAED,KAAK,CAAC,UAAU;QACd,IAAI,CAAC,UAAU,GAAG,KAAK,CAAC;QACxB,IAAI,CAAC,iBAAiB,GAAG,IAAI,CAAC;IAChC,CAAC;IAED,KAAK,CAAC,cAAc;QAClB,IAAI,CAAC;YACH,MAAM,IAAI,CAAC,OAAO,CAAC,KAAK,EAAE,cAAc,CAAC,CAAC;YAC1C,OAAO,IAAI,CAAC;QACd,CAAC;QAAC,WAAM,CAAC;YACP,OAAO,KAAK,CAAC;QACf,CAAC;IACH,CAAC;IAED,8DAA8D;IAE9D,KAAK,CAAC,MAAM,CAAC,OAA8B;QACzC,IAAI,CAAC,kBAAkB,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;QAEzC,MAAM,MAAM,GAAG,OAAO,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;YACzC,EAAE,EAAE,CAAC,CAAC,EAAE;YACR,MAAM,EAAE,CAAC,CAAC,MAAM;YAChB,OAAO,EAAE,CAAC,CAAC,QAAQ,IAAI,EAAE;SAC1B,CAAC,CAAC,CAAC;QAEJ,MAAM,IAAI,GAA4B,EAAE,MAAM,EAAE,CAAC;QAEjD,MAAM,IAAI,CAAC,OAAO,CAAC,KAAK,EAAE,gBAAgB,IAAI,CAAC,cAAc,SAAS,EAAE,IAAI,CAAC,CAAC;QAE9E,OAAO;YACL,aAAa,EAAE,MAAM,CAAC,MAAM;YAC5B,WAAW,EAAE,OAAO,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;SAC9C,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,KAAK,CAAC,OAA6B;;QACvC,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAE7B,MAAM,IAAI,GAA4B;YACpC,MAAM,EAAE,OAAO,CAAC,MAAM;YACtB,KAAK,EAAE,OAAO,CAAC,IAAI;YACnB,WAAW,EAAE,MAAA,OAAO,CAAC,aAAa,mCAAI,KAAK;YAC3C,YAAY,EAAE,MAAA,OAAO,CAAC,eAAe,mCAAI,IAAI;SAC9C,CAAC;QAEF,IAAI,OAAO,CAAC,MAAM,EAAE,CAAC;YACnB,IAAI,CAAC,MAAM,GAAG,IAAI,CAAC,aAAa,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;QACnD,CAAC;QAED,IAAI,OAAO,CAAC,QAAQ,EAAE,CAAC;YACrB,IAAI,CAAC,eAAe,GAAG,OAAO,CAAC,QAAQ,CAAC;QAC1C,CAAC;QAED,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,OAAO,CACjC,MAAM,EACN,gBAAgB,IAAI,CAAC,cAAc,gBAAgB,EACnD,IAAI,CACL,CAAC;QAEF,MAAM,OAAO,GAAG,CAAC,QAAQ,CAAC,MAAM,IAAI,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,CAAM,EAAE,EAAE,CAAC,CAAC;YACvD,EAAE,EAAE,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC;YAChB,KAAK,EAAE,CAAC,CAAC,KAAK;YACd,MAAM,EAAE,CAAC,CAAC,MAAM;YAChB,QAAQ,EAAE,CAAC,CAAC,OAAO;SACpB,CAAC,CAAC,CAAC;QAEJ,OAAO;YACL,OAAO;YACP,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS;SAClC,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,KAAK,CAAC,OAA6B;;QACvC,MAAM,IAAI,GAAG;YACX,GAAG,EAAE,OAAO,CAAC,GAAG;YAChB,WAAW,EAAE,MAAA,OAAO,CAAC,aAAa,mCAAI,IAAI;YAC1C,YAAY,EAAE,IAAI;SACnB,CAAC;QAEF,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,OAAO,CACjC,MAAM,EACN,gBAAgB,IAAI,CAAC,cAAc,SAAS,EAC5C,IAAI,CACL,CAAC;QAEF,MAAM,OAAO,GAAwB,EAAE,CAAC;QACxC,KAAK,MAAM,KAAK,IAAI,QAAQ,CAAC,MAAM,IAAI,EAAE,EAAE,CAAC;YAC1C,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC,GAAG;gBAC1B,EAAE,EAAE,MAAM,CAAC,KAAK,CAAC,EAAE,CAAC;gBACpB,MAAM,EAAE,KAAK,CAAC,MAAM;gBACpB,QAAQ,EAAE,KAAK,CAAC,OAAO;aACxB,CAAC;QACJ,CAAC;QAED,2BAA2B;QAC3B,KAAK,MAAM,EAAE,IAAI,OAAO,CAAC,GAAG,EAAE,CAAC;YAC7B,IAAI,CAAC,OAAO,CAAC,EAAE,CAAC,EAAE,CAAC;gBACjB,OAAO,CAAC,EAAE,CAAC,GAAG,IAAI,CAAC;YACrB,CAAC;QACH,CAAC;QAED,OAAO,EAAE,OAAO,EAAE,CAAC;IACrB,CAAC;IAED,KAAK,CAAC,MAAM,CAAC,OAA6B;QACxC,iEAAiE;QACjE,IAAI,OAAO,CAAC,MAAM,EAAE,CAAC;YACnB,uBAAuB;YACvB,MAAM,IAAI,CAAC,OAAO,CAAC,KAAK,EAAE,gBAAgB,IAAI,CAAC,cAAc,SAAS,EAAE;gBACtE,MAAM,EAAE;oBACN;wBACE,EAAE,EAAE,OAAO,CAAC,EAAE;wBACd,MAAM,EAAE,OAAO,CAAC,MAAM;wBACtB,OAAO,EAAE,OAAO,CAAC,WAAW,IAAI,OAAO,CAAC,aAAa,IAAI,EAAE;qBAC5D;iBACF;aACF,CAAC,CAAC;QACL,CAAC;aAAM,IAAI,OAAO,CAAC,WAAW,EAAE,CAAC;YAC/B,wBAAwB;YACxB,MAAM,IAAI,CAAC,OAAO,CAChB,MAAM,EACN,gBAAgB,IAAI,CAAC,cAAc,iBAAiB,EACpD;gBACE,OAAO,EAAE,OAAO,CAAC,WAAW;gBAC5B,MAAM,EAAE,CAAC,OAAO,CAAC,EAAE,CAAC;aACrB,CACF,CAAC;QACJ,CAAC;aAAM,IAAI,OAAO,CAAC,aAAa,EAAE,CAAC;YACjC,yBAAyB;YACzB,MAAM,IAAI,CAAC,OAAO,CAChB,MAAM,EACN,gBAAgB,IAAI,CAAC,cAAc,iBAAiB,EACpD;gBACE,OAAO,EAAE,OAAO,CAAC,aAAa;gBAC9B,MAAM,EAAE,CAAC,OAAO,CAAC,EAAE,CAAC;aACrB,CACF,CAAC;QACJ,CAAC;QAED,OAAO;YACL,OAAO,EAAE,IAAI;YACb,EAAE,EAAE,OAAO,CAAC,EAAE;SACf,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,MAAM,CAAC,OAA8B;QACzC,IAAI,YAAY,GAAG,CAAC,CAAC;QAErB,IAAI,OAAO,CAAC,GAAG,EAAE,CAAC;YAChB,MAAM,IAAI,CAAC,OAAO,CAChB,MAAM,EACN,gBAAgB,IAAI,CAAC,cAAc,gBAAgB,EACnD,EAAE,MAAM,EAAE,OAAO,CAAC,GAAG,EAAE,CACxB,CAAC;YACF,YAAY,GAAG,OAAO,CAAC,GAAG,CAAC,MAAM,CAAC;QACpC,CAAC;aAAM,IAAI,OAAO,CAAC,MAAM,EAAE,CAAC;YAC1B,MAAM,IAAI,CAAC,OAAO,CAChB,MAAM,EACN,gBAAgB,IAAI,CAAC,cAAc,gBAAgB,EACnD,EAAE,MAAM,EAAE,IAAI,CAAC,aAAa,CAAC,OAAO,CAAC,MAAM,CAAC,EAAE,CAC/C,CAAC;YACF,YAAY,GAAG,CAAC,CAAC,CAAC,CAAC,gBAAgB;QACrC,CAAC;aAAM,IAAI,OAAO,CAAC,SAAS,EAAE,CAAC;YAC7B,iCAAiC;YACjC,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,aAAa,EAAE,CAAC;YACxC,MAAM,IAAI,CAAC,WAAW,CAAC,EAAE,IAAI,EAAE,IAAI,CAAC,cAAc,EAAE,CAAC,CAAC;YACtD,MAAM,IAAI,CAAC,WAAW,CAAC;gBACrB,IAAI,EAAE,IAAI,CAAC,cAAc;gBACzB,UAAU,EAAE,IAAI,CAAC,UAAU;gBAC3B,MAAM,EAAE,IAAI,CAAC,MAAM;aACpB,CAAC,CAAC;YACH,YAAY,GAAG,IAAI,CAAC,WAAW,CAAC;QAClC,CAAC;QAED,OAAO;YACL,YAAY;YACZ,UAAU,EAAE,OAAO,CAAC,GAAG;SACxB,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,IAAI,CAAC,OAA4B;;QACrC,MAAM,IAAI,GAA4B;YACpC,KAAK,EAAE,OAAO,CAAC,KAAK,IAAI,GAAG;YAC3B,WAAW,EAAE,KAAK;YAClB,YAAY,EAAE,KAAK;SACpB,CAAC;QAEF,IAAI,OAAO,CAAC,MAAM,EAAE,CAAC;YACnB,IAAI,CAAC,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC;QAC/B,CAAC;QAED,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,OAAO,CACjC,MAAM,EACN,gBAAgB,IAAI,CAAC,cAAc,gBAAgB,EACnD,IAAI,CACL,CAAC;QAEF,OAAO;YACL,GAAG,EAAE,CAAC,CAAA,MAAA,QAAQ,CAAC,MAAM,0CAAE,MAAM,KAAI,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,CAAM,EAAE,EAAE,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;YAClE,UAAU,EAAE,MAAA,QAAQ,CAAC,MAAM,0CAAE,gBAAgB;SAC9C,CAAC;IACJ,CAAC;IAED,iEAAiE;IAEjE,KAAK,CAAC,cAAc;;QAClB,gEAAgE;QAChE,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,KAAK,EAAE,cAAc,CAAC,CAAC;QAE3D,MAAM,UAAU,GAAG,MAAM,OAAO,CAAC,GAAG,CAClC,CAAC,CAAA,MAAA,QAAQ,CAAC,MAAM,0CAAE,WAAW,KAAI,EAAE,CAAC,CAAC,GAAG,CAAC,KAAK,EAAE,CAAM,EAAE,EAAE;;YACxD,IAAI,WAAW,GAAG,CAAC,CAAC;YACpB,IAAI,CAAC;gBACH,MAAM,cAAc,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,KAAK,EAAE,gBAAgB,CAAC,CAAC,IAAI,EAAE,CAAC,CAAC;gBAC3E,WAAW,GAAG,CAAA,MAAA,cAAc,CAAC,MAAM,0CAAE,YAAY,KAAI,CAAC,CAAC;YACzD,CAAC;YAAC,WAAM,CAAC;gBACP,2CAA2C;YAC7C,CAAC;YACD,OAAO;gBACL,IAAI,EAAE,CAAC,CAAC,IAAI;gBACZ,WAAW;aACZ,CAAC;QACJ,CAAC,CAAC,CACH,CAAC;QAEF,OAAO,EAAE,UAAU,EAAE,CAAC;IACxB,CAAC;IAED,KAAK,CAAC,eAAe,CAAC,SAAiB;QACrC,MAAM,IAAI,CAAC,WAAW,CAAC,EAAE,IAAI,EAAE,SAAS,EAAE,CAAC,CAAC;IAC9C,CAAC;IAED,6DAA6D;IAE7D,KAAK,CAAC,aAAa;;QACjB,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,OAAO,CACjC,KAAK,EACL,gBAAgB,IAAI,CAAC,cAAc,EAAE,CACtC,CAAC;QAEF,MAAM,MAAM,GAAG,MAAA,QAAQ,CAAC,MAAM,0CAAE,MAAM,CAAC;QACvC,MAAM,MAAM,GAAG,MAAA,QAAQ,CAAC,MAAM,0CAAE,MAAM,CAAC;QAEvC,OAAO;YACL,IAAI,EAAE,IAAI,CAAC,cAAc;YACzB,UAAU,EAAE,CAAA,MAAA,MAAA,MAAM,aAAN,MAAM,uBAAN,MAAM,CAAE,MAAM,0CAAE,OAAO,0CAAE,IAAI,KAAI,IAAI,CAAC,UAAU;YAC5D,MAAM,EAAE,IAAI,CAAC,eAAe,CAAC,MAAA,MAAA,MAAM,aAAN,MAAM,uBAAN,MAAM,CAAE,MAAM,0CAAE,OAAO,0CAAE,QAAQ,CAAC;YAC/D,WAAW,EAAE,CAAA,MAAA,QAAQ,CAAC,MAAM,0CAAE,YAAY,KAAI,CAAC;YAC/C,MAAM,EAAE,MAAM,KAAK,OAAO,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,cAAc;SACtD,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,aAAa;QACjB,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,aAAa,EAAE,CAAC;QAExC,OAAO;YACL,WAAW,EAAE,IAAI,CAAC,WAAW;YAC7B,UAAU,EAAE,IAAI,CAAC,UAAU;SAC5B,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,WAAW,CAAC,OAA4B;QAC5C,MAAM,IAAI,GAAG;YACX,OAAO,EAAE;gBACP,IAAI,EAAE,OAAO,CAAC,UAAU;gBACxB,QAAQ,EAAE,IAAI,CAAC,iBAAiB,CAAC,OAAO,CAAC,MAAM,IAAI,sBAAc,CAAC,MAAM,CAAC;aAC1E;SACF,CAAC;QAEF,MAAM,IAAI,CAAC,OAAO,CAAC,KAAK,EAAE,gBAAgB,OAAO,CAAC,IAAI,EAAE,EAAE,IAAI,CAAC,CAAC;QAEhE,OAAO;YACL,OAAO,EAAE,IAAI;YACb,IAAI,EAAE,OAAO,CAAC,IAAI;SACnB,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,WAAW,CAAC,OAA4B;QAC5C,MAAM,IAAI,CAAC,OAAO,CAAC,QAAQ,EAAE,gBAAgB,OAAO,CAAC,IAAI,EAAE,CAAC,CAAC;QAE7D,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC;IAC3B,CAAC;IAED,KAAK,CAAC,WAAW;;QACf,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,KAAK,EAAE,cAAc,CAAC,CAAC;QAE3D,OAAO,OAAO,CAAC,GAAG,CAChB,CAAC,CAAA,MAAA,QAAQ,CAAC,MAAM,0CAAE,WAAW,KAAI,EAAE,CAAC,CAAC,GAAG,CAAC,KAAK,EAAE,CAAM,EAAE,EAAE;;YACxD,IAAI,WAAW,GAAG,CAAC,CAAC;YACpB,IAAI,UAAU,GAAG,CAAC,CAAC,CAAC;YACpB,IAAI,MAAM,GAAG,sBAAc,CAAC,MAAM,CAAC;YACnC,IAAI,MAAM,GAA6B,OAAO,CAAC;YAC/C,IAAI,CAAC;gBACH,MAAM,cAAc,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,KAAK,EAAE,gBAAgB,CAAC,CAAC,IAAI,EAAE,CAAC,CAAC;gBAC3E,MAAM,MAAM,GAAG,MAAA,cAAc,CAAC,MAAM,0CAAE,MAAM,CAAC;gBAC7C,WAAW,GAAG,CAAA,MAAA,cAAc,CAAC,MAAM,0CAAE,YAAY,KAAI,CAAC,CAAC;gBACvD,UAAU,GAAG,CAAA,MAAA,MAAA,MAAM,aAAN,MAAM,uBAAN,MAAM,CAAE,MAAM,0CAAE,OAAO,0CAAE,IAAI,KAAI,CAAC,CAAC,CAAC;gBACjD,MAAM,GAAG,IAAI,CAAC,eAAe,CAAC,MAAA,MAAA,MAAM,aAAN,MAAM,uBAAN,MAAM,CAAE,MAAM,0CAAE,OAAO,0CAAE,QAAQ,CAAC,CAAC;gBACjE,MAAM,GAAG,CAAA,MAAA,cAAc,CAAC,MAAM,0CAAE,MAAM,MAAK,OAAO,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,cAAc,CAAC;YAChF,CAAC;YAAC,WAAM,CAAC;gBACP,2CAA2C;YAC7C,CAAC;YACD,OAAO;gBACL,IAAI,EAAE,CAAC,CAAC,IAAI;gBACZ,UAAU;gBACV,MAAM;gBACN,WAAW;gBACX,MAAM;aACP,CAAC;QACJ,CAAC,CAAC,CACH,CAAC;IACJ,CAAC;IAED,8DAA8D;IAE9D,eAAe,CAAC,OAAsB;QACpC,QAAQ,OAAO,EAAE,CAAC;YAChB,KAAK,qBAAa,CAAC,kBAAkB,CAAC;YACtC,KAAK,qBAAa,CAAC,gBAAgB,CAAC;YACpC,KAAK,qBAAa,CAAC,cAAc,CAAC;YAClC,KAAK,qBAAa,CAAC,gBAAgB;gBACjC,OAAO,IAAI,CAAC;YACd,KAAK,qBAAa,CAAC,UAAU,CAAC;YAC9B,KAAK,qBAAa,CAAC,aAAa,CAAC;YACjC,KAAK,qBAAa,CAAC,cAAc,CAAC;YAClC,KAAK,qBAAa,CAAC,YAAY;gBAC7B,OAAO,KAAK,CAAC;YACf;gBACE,OAAO,KAAK,CAAC;QACjB,CAAC;IACH,CAAC;IAED,4DAA4D;IAE5D;;OAEG;IACK,KAAK,CAAC,OAAO,CACnB,MAAc,EACd,IAAY,EACZ,IAA8B;QAE9B,MAAM,GAAG,GAAG,GAAG,IAAI,CAAC,OAAO,GAAG,IAAI,EAAE,CAAC;QAErC,MAAM,OAAO,GAA2B;YACtC,cAAc,EAAE,kBAAkB;SACnC,CAAC;QAEF,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC;YAChB,OAAO,CAAC,SAAS,CAAC,GAAG,IAAI,CAAC,MAAM,CAAC;QACnC,CAAC;QAED,MAAM,OAAO,GAAgB;YAC3B,MAAM;YACN,OAAO;YACP,MAAM,EAAE,WAAW,CAAC,OAAO,CAAC,IAAI,CAAC,OAAO,CAAC;SAC1C,CAAC;QAEF,IAAI,IAAI,IAAI,MAAM,KAAK,KAAK,EAAE,CAAC;YAC7B,OAAO,CAAC,IAAI,GAAG,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC;QACtC,CAAC;QAED,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,GAAG,EAAE,OAAO,CAAC,CAAC;QAE3C,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;YACjB,MAAM,SAAS,GAAG,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC;YACxC,MAAM,mBAAW,CAAC,WAAW,CAC3B,IAAI,KAAK,CAAC,qBAAqB,QAAQ,CAAC,MAAM,MAAM,SAAS,EAAE,CAAC,EAChE,oBAAY,CAAC,MAAM,CACpB,CAAC;QACJ,CAAC;QAED,OAAO,QAAQ,CAAC,IAAI,EAAE,CAAC;IACzB,CAAC;IAED;;OAEG;IACK,aAAa,CACnB,MAA8C;QAE9C,IAAI,MAAM,IAAI,MAAM,IAAI,KAAK,IAAI,MAAM,EAAE,CAAC;YACxC,MAAM,KAAK,GAAG,MAA8B,CAAC;YAE7C,IAAI,KAAK,CAAC,IAAI,EAAE,CAAC;gBACf,OAAO;oBACL,IAAI,EAAE,KAAK,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC,CAAC,CAAC;iBACnD,CAAC;YACJ,CAAC;YAED,IAAI,KAAK,CAAC,GAAG,EAAE,CAAC;gBACd,OAAO;oBACL,MAAM,EAAE,KAAK,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC,CAAC,CAAC;iBACpD,CAAC;YACJ,CAAC;QACH,CAAC;QAED,MAAM,CAAC,GAAG,MAAyB,CAAC;QACpC,OAAO,IAAI,CAAC,mBAAmB,CAAC,CAAC,CAAC,CAAC;IACrC,CAAC;IAED;;OAEG;IACK,mBAAmB,CAAC,MAAuB;QACjD,MAAM,EAAE,KAAK,EAAE,QAAQ,EAAE,KAAK,EAAE,GAAG,MAAM,CAAC;QAE1C,QAAQ,QAAQ,EAAE,CAAC;YACjB,KAAK,KAAK;gBACR,OAAO,EAAE,GAAG,EAAE,KAAK,EAAE,KAAK,EAAE,EAAE,KAAK,EAAE,EAAE,CAAC;YAC1C,KAAK,KAAK;gBACR,OAAO;oBACL,QAAQ,EAAE,CAAC,EAAE,GAAG,EAAE,KAAK,EAAE,KAAK,EAAE,EAAE,KAAK,EAAE,EAAE,CAAC;iBAC7C,CAAC;YACJ,KAAK,KAAK;gBACR,OAAO,EAAE,GAAG,EAAE,KAAK,EAAE,KAAK,EAAE,EAAE,EAAE,EAAE,KAAK,EAAE,EAAE,CAAC;YAC9C,KAAK,MAAM;gBACT,OAAO,EAAE,GAAG,EAAE,KAAK,EAAE,KAAK,EAAE,EAAE,GAAG,EAAE,KAAK,EAAE,EAAE,CAAC;YAC/C,KAAK,KAAK;gBACR,OAAO,EAAE,GAAG,EAAE,KAAK,EAAE,KAAK,EAAE,EAAE,EAAE,EAAE,KAAK,EAAE,EAAE,CAAC;YAC9C,KAAK,MAAM;gBACT,OAAO,EAAE,GAAG,EAAE,KAAK,EAAE,KAAK,EAAE,EAAE,GAAG,EAAE,KAAK,EAAE,EAAE,CAAC;YAC/C,KAAK,KAAK;gBACR,OAAO,EAAE,GAAG,EAAE,KAAK,EAAE,KAAK,EAAE,EAAE,GAAG,EAAE,KAAK,EAAE,EAAE,CAAC;YAC/C;gBACE,OAAO,EAAE,GAAG,EAAE,KAAK,EAAE,KAAK,EAAE,EAAE,KAAK,EAAE,EAAE,CAAC;QAC5C,CAAC;IACH,CAAC;IAED;;OAEG;IACK,eAAe,CAAC,QAAgB;QACtC,QAAQ,QAAQ,aAAR,QAAQ,uBAAR,QAAQ,CAAE,WAAW,EAAE,EAAE,CAAC;YAChC,KAAK,QAAQ;gBACX,OAAO,sBAAc,CAAC,MAAM,CAAC;YAC/B,KAAK,QAAQ;gBACX,OAAO,sBAAc,CAAC,SAAS,CAAC;YAClC,KAAK,KAAK;gBACR,OAAO,sBAAc,CAAC,WAAW,CAAC;YACpC;gBACE,OAAO,sBAAc,CAAC,MAAM,CAAC;QACjC,CAAC;IACH,CAAC;IAED;;OAEG;IACK,iBAAiB,CAAC,MAAsB;QAC9C,QAAQ,MAAM,EAAE,CAAC;YACf,KAAK,sBAAc,CAAC,MAAM;gBACxB,OAAO,QAAQ,CAAC;YAClB,KAAK,sBAAc,CAAC,SAAS;gBAC3B,OAAO,QAAQ,CAAC;YAClB,KAAK,sBAAc,CAAC,WAAW;gBAC7B,OAAO,KAAK,CAAC;YACf;gBACE,OAAO,QAAQ,CAAC;QACpB,CAAC;IACH,CAAC;CACF;AArgBD,sCAqgBC;AAED,kBAAe,aAAa,CAAC"}

package/dist/vector/adapters/weaviate.adapter.d.ts

@@ -0,0 +1,68 @@
+/**
+ * Weaviate Adapter
+ *
+ * Vector database adapter for Weaviate.
+ * https://weaviate.io/
+ */
+import { VectorDBType, VectorFeature, IVectorConnectionResult, IUpsertVectorsOptions, IUpsertVectorsResult, IQueryVectorsOptions, IQueryVectorsResult, IFetchVectorsOptions, IFetchVectorsResult, IUpdateVectorOptions, IUpdateVectorResult, IDeleteVectorsOptions, IDeleteVectorsResult, IListVectorsOptions, IListVectorsResult, IListNamespacesResult, IVectorIndexInfo, ICreateIndexOptions, ICreateIndexResult, IDeleteIndexOptions, IDeleteIndexResult, IVectorIndexStats } from '../types';
+import { BaseVectorAdapter, IAdapterConnectionOptions } from './base.adapter';
+/**
+ * Weaviate adapter implementation
+ */
+export declare class WeaviateAdapter extends BaseVectorAdapter {
+    readonly type = VectorDBType.WEAVIATE;
+    private apiKey;
+    private baseUrl;
+    private className;
+    private timeout;
+    connect(options: IAdapterConnectionOptions): Promise<IVectorConnectionResult>;
+    disconnect(): Promise<void>;
+    testConnection(): Promise<boolean>;
+    upsert(options: IUpsertVectorsOptions): Promise<IUpsertVectorsResult>;
+    query(options: IQueryVectorsOptions): Promise<IQueryVectorsResult>;
+    fetch(options: IFetchVectorsOptions): Promise<IFetchVectorsResult>;
+    update(options: IUpdateVectorOptions): Promise<IUpdateVectorResult>;
+    delete(options: IDeleteVectorsOptions): Promise<IDeleteVectorsResult>;
+    list(options: IListVectorsOptions): Promise<IListVectorsResult>;
+    listNamespaces(): Promise<IListNamespacesResult>;
+    deleteNamespace(namespace: string): Promise<void>;
+    describeIndex(): Promise<IVectorIndexInfo>;
+    getIndexStats(): Promise<IVectorIndexStats>;
+    createIndex(options: ICreateIndexOptions): Promise<ICreateIndexResult>;
+    deleteIndex(options: IDeleteIndexOptions): Promise<IDeleteIndexResult>;
+    listIndexes(): Promise<IVectorIndexInfo[]>;
+    supportsFeature(feature: VectorFeature): boolean;
+    /**
+     * Ensure the Weaviate class (schema) exists, creating it if necessary
+     */
+    private ensureClassExists;
+    /**
+     * Make REST request
+     */
+    private request;
+    /**
+     * Make GraphQL request
+     */
+    private graphql;
+    /**
+     * Convert filter to Weaviate format
+     */
+    private convertFilter;
+    /**
+     * Convert single filter
+     */
+    private convertSingleFilter;
+    /**
+     * Extract properties from Weaviate result (excluding _additional)
+     */
+    private extractProperties;
+    /**
+     * Map Weaviate distance to our enum
+     */
+    private mapWeaviateMetric;
+    /**
+     * Map our enum to Weaviate distance
+     */
+    private mapToWeaviateMetric;
+}
+export default WeaviateAdapter;