@ductape/sdk 0.1.11 → 0.1.13

package/dist/vector/adapters/azure-search.adapter.js

@@ -0,0 +1,217 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.AzureSearchAdapter = void 0;
+/**
+ * Azure AI Search vector adapter (REST API 2024-07-01).
+ */
+const axios_1 = __importDefault(require("axios"));
+const types_1 = require("../types");
+const base_adapter_1 = require("./base.adapter");
+const API_VERSION = '2024-07-01';
+const DEFAULT_VECTOR_FIELD = 'contentVector';
+class AzureSearchAdapter extends base_adapter_1.BaseVectorAdapter {
+    constructor() {
+        super(...arguments);
+        this.type = types_1.VectorDBType.AZURE_SEARCH;
+        this.client = null;
+        this.indexName = '';
+        this.vectorField = DEFAULT_VECTOR_FIELD;
+    }
+    async connect(options) {
+        var _a, _b;
+        if (!options.endpoint) {
+            return { connected: false, error: 'Azure Search endpoint is required' };
+        }
+        if (!options.apiKey) {
+            return { connected: false, error: 'Azure Search apiKey is required' };
+        }
+        this.connectionOptions = options;
+        this.indexName = options.index;
+        this.vectorField = String(((_a = options.options) === null || _a === void 0 ? void 0 : _a.vectorField) || DEFAULT_VECTOR_FIELD);
+        this.client = axios_1.default.create({
+            baseURL: options.endpoint.replace(/\/$/, ''),
+            headers: {
+                'Content-Type': 'application/json',
+                'api-key': options.apiKey,
+            },
+            timeout: ((_b = options.options) === null || _b === void 0 ? void 0 : _b.timeout) || 30000,
+        });
+        const ok = await this.testConnection();
+        this._connected = ok;
+        if (!ok) {
+            return { connected: false, error: 'Failed to connect to Azure AI Search' };
+        }
+        try {
+            const info = await this.describeIndex();
+            return {
+                connected: true,
+                index: this.indexName,
+                dimensions: info.dimensions,
+                vectorCount: info.vectorCount,
+            };
+        }
+        catch (_c) {
+            return {
+                connected: true,
+                index: this.indexName,
+                dimensions: options.dimensions,
+                vectorCount: 0,
+            };
+        }
+    }
+    async disconnect() {
+        this.client = null;
+        this._connected = false;
+        this.connectionOptions = null;
+    }
+    async testConnection() {
+        if (!this.client)
+            return false;
+        try {
+            await this.client.get(`/indexes?api-version=${API_VERSION}`);
+            return true;
+        }
+        catch (_a) {
+            return false;
+        }
+    }
+    async upsert(options) {
+        this.validateDimensions(options.vectors);
+        const value = options.vectors.map((v) => (Object.assign({ '@search.action': 'mergeOrUpload', id: v.id, [this.vectorField]: v.values }, (v.metadata || {}))));
+        await this.client.post(`/indexes/${this.indexName}/docs/index?api-version=${API_VERSION}`, { value });
+        return {
+            upsertedCount: options.vectors.length,
+            upsertedIds: options.vectors.map((v) => v.id),
+        };
+    }
+    async query(options) {
+        const start = Date.now();
+        const body = {
+            vectorQueries: [
+                {
+                    kind: 'vector',
+                    vector: options.vector,
+                    k: options.topK,
+                    fields: this.vectorField,
+                },
+            ],
+            top: options.topK,
+            select: 'id,*',
+        };
+        const { data } = await this.client.post(`/indexes/${this.indexName}/docs/search?api-version=${API_VERSION}`, body);
+        const hits = (data === null || data === void 0 ? void 0 : data.value) || [];
+        return {
+            matches: hits.map((hit) => {
+                var _a;
+                return ({
+                    id: String(hit.id),
+                    score: (_a = hit['@search.score']) !== null && _a !== void 0 ? _a : 0,
+                    values: options.includeValues ? hit[this.vectorField] : undefined,
+                    metadata: hit,
+                });
+            }),
+            queryTime: Date.now() - start,
+        };
+    }
+    async fetch(options) {
+        const vectors = {};
+        for (const id of options.ids) {
+            try {
+                const { data } = await this.client.get(`/indexes/${this.indexName}/docs('${encodeURIComponent(id)}')?api-version=${API_VERSION}`);
+                vectors[id] = {
+                    id,
+                    values: options.includeValues ? data[this.vectorField] : undefined,
+                    metadata: data,
+                };
+            }
+            catch (_a) {
+                vectors[id] = null;
+            }
+        }
+        return { vectors };
+    }
+    async update(options) {
+        const doc = Object.assign({ '@search.action': 'merge', id: options.id }, (options.setMetadata || options.mergeMetadata || {}));
+        if (options.values)
+            doc[this.vectorField] = options.values;
+        await this.client.post(`/indexes/${this.indexName}/docs/index?api-version=${API_VERSION}`, {
+            value: [doc],
+        });
+        return { success: true, id: options.id };
+    }
+    async delete(options) {
+        var _a;
+        if (!((_a = options.ids) === null || _a === void 0 ? void 0 : _a.length))
+            return { deletedCount: 0 };
+        const value = options.ids.map((id) => ({ '@search.action': 'delete', id }));
+        await this.client.post(`/indexes/${this.indexName}/docs/index?api-version=${API_VERSION}`, { value });
+        return { deletedCount: options.ids.length, deletedIds: options.ids };
+    }
+    async list(options) {
+        const { data } = await this.client.post(`/indexes/${this.indexName}/docs/search?api-version=${API_VERSION}`, { search: '*', top: options.limit || 100, select: 'id' });
+        const ids = ((data === null || data === void 0 ? void 0 : data.value) || []).map((h) => String(h.id));
+        return { ids, total: ids.length };
+    }
+    async listNamespaces() {
+        return { namespaces: [] };
+    }
+    async deleteNamespace() { }
+    async describeIndex() {
+        const stats = await this.getIndexStats();
+        return {
+            name: this.indexName,
+            dimensions: this.dimensions,
+            metric: this.metric,
+            vectorCount: stats.vectorCount,
+            status: 'ready',
+        };
+    }
+    async getIndexStats() {
+        const { data } = await this.client.get(`/indexes/${this.indexName}/docs/$count?api-version=${API_VERSION}`);
+        return { vectorCount: data !== null && data !== void 0 ? data : 0, dimensions: this.dimensions };
+    }
+    async createIndex(options) {
+        const metric = options.metric || types_1.DistanceMetric.COSINE;
+        const algorithm = metric === types_1.DistanceMetric.EUCLIDEAN ? 'euclidean' : metric === types_1.DistanceMetric.DOT_PRODUCT ? 'dotProduct' : 'cosine';
+        await this.client.put(`/indexes/${options.name}?api-version=${API_VERSION}`, {
+            name: options.name,
+            fields: [
+                { name: 'id', type: 'Edm.String', key: true, filterable: true },
+                {
+                    name: this.vectorField,
+                    type: 'Collection(Edm.Single)',
+                    searchable: true,
+                    vectorSearchDimensions: options.dimensions,
+                    vectorSearchProfileName: 'default-profile',
+                },
+            ],
+            vectorSearch: {
+                algorithms: [{ name: 'default-algo', kind: 'hnsw' }],
+                profiles: [{ name: 'default-profile', algorithmConfigurationName: 'default-algo' }],
+            },
+        });
+        return { success: true, name: options.name };
+    }
+    async deleteIndex(options) {
+        await this.client.delete(`/indexes/${options.name}?api-version=${API_VERSION}`);
+        return { success: true };
+    }
+    async listIndexes() {
+        const { data } = await this.client.get(`/indexes?api-version=${API_VERSION}`);
+        return ((data === null || data === void 0 ? void 0 : data.value) || []).map((idx) => ({
+            name: idx.name,
+            dimensions: this.dimensions,
+            metric: this.metric,
+            vectorCount: 0,
+            status: 'ready',
+        }));
+    }
+    supportsFeature(feature) {
+        return feature === types_1.VectorFeature.METADATA_FILTERING || feature === types_1.VectorFeature.HYBRID_SEARCH;
+    }
+}
+exports.AzureSearchAdapter = AzureSearchAdapter;
+//# sourceMappingURL=azure-search.adapter.js.map

package/dist/vector/adapters/azure-search.adapter.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"azure-search.adapter.js","sourceRoot":"","sources":["../../../src/vector/adapters/azure-search.adapter.ts"],"names":[],"mappings":";;;;;;AAAA;;GAEG;AACH,kDAA6C;AAC7C,oCAwBkB;AAClB,iDAA8E;AAE9E,MAAM,WAAW,GAAG,YAAY,CAAC;AACjC,MAAM,oBAAoB,GAAG,eAAe,CAAC;AAE7C,MAAa,kBAAmB,SAAQ,gCAAiB;IAAzD;;QACkB,SAAI,GAAG,oBAAY,CAAC,YAAY,CAAC;QAEzC,WAAM,GAAyB,IAAI,CAAC;QACpC,cAAS,GAAG,EAAE,CAAC;QACf,gBAAW,GAAG,oBAAoB,CAAC;IA2N7C,CAAC;IAzNC,KAAK,CAAC,OAAO,CAAC,OAAkC;;QAC9C,IAAI,CAAC,OAAO,CAAC,QAAQ,EAAE,CAAC;YACtB,OAAO,EAAE,SAAS,EAAE,KAAK,EAAE,KAAK,EAAE,mCAAmC,EAAE,CAAC;QAC1E,CAAC;QACD,IAAI,CAAC,OAAO,CAAC,MAAM,EAAE,CAAC;YACpB,OAAO,EAAE,SAAS,EAAE,KAAK,EAAE,KAAK,EAAE,iCAAiC,EAAE,CAAC;QACxE,CAAC;QACD,IAAI,CAAC,iBAAiB,GAAG,OAAO,CAAC;QACjC,IAAI,CAAC,SAAS,GAAG,OAAO,CAAC,KAAK,CAAC;QAC/B,IAAI,CAAC,WAAW,GAAG,MAAM,CAAC,CAAA,MAAA,OAAO,CAAC,OAAO,0CAAE,WAAW,KAAI,oBAAoB,CAAC,CAAC;QAChF,IAAI,CAAC,MAAM,GAAG,eAAK,CAAC,MAAM,CAAC;YACzB,OAAO,EAAE,OAAO,CAAC,QAAQ,CAAC,OAAO,CAAC,KAAK,EAAE,EAAE,CAAC;YAC5C,OAAO,EAAE;gBACP,cAAc,EAAE,kBAAkB;gBAClC,SAAS,EAAE,OAAO,CAAC,MAAM;aAC1B;YACD,OAAO,EAAE,CAAC,MAAA,OAAO,CAAC,OAAO,0CAAE,OAAkB,KAAI,KAAK;SACvD,CAAC,CAAC;QACH,MAAM,EAAE,GAAG,MAAM,IAAI,CAAC,cAAc,EAAE,CAAC;QACvC,IAAI,CAAC,UAAU,GAAG,EAAE,CAAC;QACrB,IAAI,CAAC,EAAE,EAAE,CAAC;YACR,OAAO,EAAE,SAAS,EAAE,KAAK,EAAE,KAAK,EAAE,sCAAsC,EAAE,CAAC;QAC7E,CAAC;QACD,IAAI,CAAC;YACH,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,aAAa,EAAE,CAAC;YACxC,OAAO;gBACL,SAAS,EAAE,IAAI;gBACf,KAAK,EAAE,IAAI,CAAC,SAAS;gBACrB,UAAU,EAAE,IAAI,CAAC,UAAU;gBAC3B,WAAW,EAAE,IAAI,CAAC,WAAW;aAC9B,CAAC;QACJ,CAAC;QAAC,WAAM,CAAC;YACP,OAAO;gBACL,SAAS,EAAE,IAAI;gBACf,KAAK,EAAE,IAAI,CAAC,SAAS;gBACrB,UAAU,EAAE,OAAO,CAAC,UAAU;gBAC9B,WAAW,EAAE,CAAC;aACf,CAAC;QACJ,CAAC;IACH,CAAC;IAED,KAAK,CAAC,UAAU;QACd,IAAI,CAAC,MAAM,GAAG,IAAI,CAAC;QACnB,IAAI,CAAC,UAAU,GAAG,KAAK,CAAC;QACxB,IAAI,CAAC,iBAAiB,GAAG,IAAI,CAAC;IAChC,CAAC;IAED,KAAK,CAAC,cAAc;QAClB,IAAI,CAAC,IAAI,CAAC,MAAM;YAAE,OAAO,KAAK,CAAC;QAC/B,IAAI,CAAC;YACH,MAAM,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,wBAAwB,WAAW,EAAE,CAAC,CAAC;YAC7D,OAAO,IAAI,CAAC;QACd,CAAC;QAAC,WAAM,CAAC;YACP,OAAO,KAAK,CAAC;QACf,CAAC;IACH,CAAC;IAED,KAAK,CAAC,MAAM,CAAC,OAA8B;QACzC,IAAI,CAAC,kBAAkB,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;QACzC,MAAM,KAAK,GAAG,OAAO,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,iBACvC,gBAAgB,EAAE,eAAe,EACjC,EAAE,EAAE,CAAC,CAAC,EAAE,EACR,CAAC,IAAI,CAAC,WAAW,CAAC,EAAE,CAAC,CAAC,MAAM,IACzB,CAAC,CAAC,CAAC,QAAQ,IAAI,EAAE,CAAC,EACrB,CAAC,CAAC;QACJ,MAAM,IAAI,CAAC,MAAO,CAAC,IAAI,CAAC,YAAY,IAAI,CAAC,SAAS,2BAA2B,WAAW,EAAE,EAAE,EAAE,KAAK,EAAE,CAAC,CAAC;QACvG,OAAO;YACL,aAAa,EAAE,OAAO,CAAC,OAAO,CAAC,MAAM;YACrC,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,KAAK,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QACzB,MAAM,IAAI,GAAG;YACX,aAAa,EAAE;gBACb;oBACE,IAAI,EAAE,QAAQ;oBACd,MAAM,EAAE,OAAO,CAAC,MAAM;oBACtB,CAAC,EAAE,OAAO,CAAC,IAAI;oBACf,MAAM,EAAE,IAAI,CAAC,WAAW;iBACzB;aACF;YACD,GAAG,EAAE,OAAO,CAAC,IAAI;YACjB,MAAM,EAAE,MAAM;SACf,CAAC;QACF,MAAM,EAAE,IAAI,EAAE,GAAG,MAAM,IAAI,CAAC,MAAO,CAAC,IAAI,CACtC,YAAY,IAAI,CAAC,SAAS,4BAA4B,WAAW,EAAE,EACnE,IAAI,CACL,CAAC;QACF,MAAM,IAAI,GAAG,CAAA,IAAI,aAAJ,IAAI,uBAAJ,IAAI,CAAE,KAAK,KAAI,EAAE,CAAC;QAC/B,OAAO;YACL,OAAO,EAAE,IAAI,CAAC,GAAG,CAAC,CAAC,GAAQ,EAAE,EAAE;;gBAAC,OAAA,CAAC;oBAC/B,EAAE,EAAE,MAAM,CAAC,GAAG,CAAC,EAAE,CAAC;oBAClB,KAAK,EAAE,MAAA,GAAG,CAAC,eAAe,CAAC,mCAAI,CAAC;oBAChC,MAAM,EAAE,OAAO,CAAC,aAAa,CAAC,CAAC,CAAC,GAAG,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC,SAAS;oBACjE,QAAQ,EAAE,GAAG;iBACd,CAAC,CAAA;aAAA,CAAC;YACH,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE,GAAG,KAAK;SAC9B,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,KAAK,CAAC,OAA6B;QACvC,MAAM,OAAO,GAAmC,EAAE,CAAC;QACnD,KAAK,MAAM,EAAE,IAAI,OAAO,CAAC,GAAG,EAAE,CAAC;YAC7B,IAAI,CAAC;gBACH,MAAM,EAAE,IAAI,EAAE,GAAG,MAAM,IAAI,CAAC,MAAO,CAAC,GAAG,CACrC,YAAY,IAAI,CAAC,SAAS,UAAU,kBAAkB,CAAC,EAAE,CAAC,kBAAkB,WAAW,EAAE,CAC1F,CAAC;gBACF,OAAO,CAAC,EAAE,CAAC,GAAG;oBACZ,EAAE;oBACF,MAAM,EAAE,OAAO,CAAC,aAAa,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC,SAAS;oBAClE,QAAQ,EAAE,IAAI;iBACf,CAAC;YACJ,CAAC;YAAC,WAAM,CAAC;gBACP,OAAO,CAAC,EAAE,CAAC,GAAG,IAAI,CAAC;YACrB,CAAC;QACH,CAAC;QACD,OAAO,EAAE,OAAO,EAAE,CAAC;IACrB,CAAC;IAED,KAAK,CAAC,MAAM,CAAC,OAA6B;QACxC,MAAM,GAAG,mBACP,gBAAgB,EAAE,OAAO,EACzB,EAAE,EAAE,OAAO,CAAC,EAAE,IACX,CAAC,OAAO,CAAC,WAAW,IAAI,OAAO,CAAC,aAAa,IAAI,EAAE,CAAC,CACxD,CAAC;QACF,IAAI,OAAO,CAAC,MAAM;YAAE,GAAG,CAAC,IAAI,CAAC,WAAW,CAAC,GAAG,OAAO,CAAC,MAAM,CAAC;QAC3D,MAAM,IAAI,CAAC,MAAO,CAAC,IAAI,CAAC,YAAY,IAAI,CAAC,SAAS,2BAA2B,WAAW,EAAE,EAAE;YAC1F,KAAK,EAAE,CAAC,GAAG,CAAC;SACb,CAAC,CAAC;QACH,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,EAAE,EAAE,OAAO,CAAC,EAAE,EAAE,CAAC;IAC3C,CAAC;IAED,KAAK,CAAC,MAAM,CAAC,OAA8B;;QACzC,IAAI,CAAC,CAAA,MAAA,OAAO,CAAC,GAAG,0CAAE,MAAM,CAAA;YAAE,OAAO,EAAE,YAAY,EAAE,CAAC,EAAE,CAAC;QACrD,MAAM,KAAK,GAAG,OAAO,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC,EAAE,gBAAgB,EAAE,QAAQ,EAAE,EAAE,EAAE,CAAC,CAAC,CAAC;QAC5E,MAAM,IAAI,CAAC,MAAO,CAAC,IAAI,CAAC,YAAY,IAAI,CAAC,SAAS,2BAA2B,WAAW,EAAE,EAAE,EAAE,KAAK,EAAE,CAAC,CAAC;QACvG,OAAO,EAAE,YAAY,EAAE,OAAO,CAAC,GAAG,CAAC,MAAM,EAAE,UAAU,EAAE,OAAO,CAAC,GAAG,EAAE,CAAC;IACvE,CAAC;IAED,KAAK,CAAC,IAAI,CAAC,OAA4B;QACrC,MAAM,EAAE,IAAI,EAAE,GAAG,MAAM,IAAI,CAAC,MAAO,CAAC,IAAI,CACtC,YAAY,IAAI,CAAC,SAAS,4BAA4B,WAAW,EAAE,EACnE,EAAE,MAAM,EAAE,GAAG,EAAE,GAAG,EAAE,OAAO,CAAC,KAAK,IAAI,GAAG,EAAE,MAAM,EAAE,IAAI,EAAE,CACzD,CAAC;QACF,MAAM,GAAG,GAAG,CAAC,CAAA,IAAI,aAAJ,IAAI,uBAAJ,IAAI,CAAE,KAAK,KAAI,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,CAAM,EAAE,EAAE,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;QAC9D,OAAO,EAAE,GAAG,EAAE,KAAK,EAAE,GAAG,CAAC,MAAM,EAAE,CAAC;IACpC,CAAC;IAED,KAAK,CAAC,cAAc;QAClB,OAAO,EAAE,UAAU,EAAE,EAAE,EAAE,CAAC;IAC5B,CAAC;IAED,KAAK,CAAC,eAAe,KAAmB,CAAC;IAEzC,KAAK,CAAC,aAAa;QACjB,MAAM,KAAK,GAAG,MAAM,IAAI,CAAC,aAAa,EAAE,CAAC;QACzC,OAAO;YACL,IAAI,EAAE,IAAI,CAAC,SAAS;YACpB,UAAU,EAAE,IAAI,CAAC,UAAU;YAC3B,MAAM,EAAE,IAAI,CAAC,MAAM;YACnB,WAAW,EAAE,KAAK,CAAC,WAAW;YAC9B,MAAM,EAAE,OAAO;SAChB,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,aAAa;QACjB,MAAM,EAAE,IAAI,EAAE,GAAG,MAAM,IAAI,CAAC,MAAO,CAAC,GAAG,CACrC,YAAY,IAAI,CAAC,SAAS,4BAA4B,WAAW,EAAE,CACpE,CAAC;QACF,OAAO,EAAE,WAAW,EAAE,IAAI,aAAJ,IAAI,cAAJ,IAAI,GAAI,CAAC,EAAE,UAAU,EAAE,IAAI,CAAC,UAAU,EAAE,CAAC;IACjE,CAAC;IAED,KAAK,CAAC,WAAW,CAAC,OAA4B;QAC5C,MAAM,MAAM,GAAG,OAAO,CAAC,MAAM,IAAI,sBAAc,CAAC,MAAM,CAAC;QACvD,MAAM,SAAS,GACb,MAAM,KAAK,sBAAc,CAAC,SAAS,CAAC,CAAC,CAAC,WAAW,CAAC,CAAC,CAAC,MAAM,KAAK,sBAAc,CAAC,WAAW,CAAC,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,QAAQ,CAAC;QACtH,MAAM,IAAI,CAAC,MAAO,CAAC,GAAG,CAAC,YAAY,OAAO,CAAC,IAAI,gBAAgB,WAAW,EAAE,EAAE;YAC5E,IAAI,EAAE,OAAO,CAAC,IAAI;YAClB,MAAM,EAAE;gBACN,EAAE,IAAI,EAAE,IAAI,EAAE,IAAI,EAAE,YAAY,EAAE,GAAG,EAAE,IAAI,EAAE,UAAU,EAAE,IAAI,EAAE;gBAC/D;oBACE,IAAI,EAAE,IAAI,CAAC,WAAW;oBACtB,IAAI,EAAE,wBAAwB;oBAC9B,UAAU,EAAE,IAAI;oBAChB,sBAAsB,EAAE,OAAO,CAAC,UAAU;oBAC1C,uBAAuB,EAAE,iBAAiB;iBAC3C;aACF;YACD,YAAY,EAAE;gBACZ,UAAU,EAAE,CAAC,EAAE,IAAI,EAAE,cAAc,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC;gBACpD,QAAQ,EAAE,CAAC,EAAE,IAAI,EAAE,iBAAiB,EAAE,0BAA0B,EAAE,cAAc,EAAE,CAAC;aACpF;SACF,CAAC,CAAC;QACH,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,IAAI,EAAE,OAAO,CAAC,IAAI,EAAE,CAAC;IAC/C,CAAC;IAED,KAAK,CAAC,WAAW,CAAC,OAA4B;QAC5C,MAAM,IAAI,CAAC,MAAO,CAAC,MAAM,CAAC,YAAY,OAAO,CAAC,IAAI,gBAAgB,WAAW,EAAE,CAAC,CAAC;QACjF,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC;IAC3B,CAAC;IAED,KAAK,CAAC,WAAW;QACf,MAAM,EAAE,IAAI,EAAE,GAAG,MAAM,IAAI,CAAC,MAAO,CAAC,GAAG,CAAC,wBAAwB,WAAW,EAAE,CAAC,CAAC;QAC/E,OAAO,CAAC,CAAA,IAAI,aAAJ,IAAI,uBAAJ,IAAI,CAAE,KAAK,KAAI,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,GAAQ,EAAE,EAAE,CAAC,CAAC;YAC5C,IAAI,EAAE,GAAG,CAAC,IAAI;YACd,UAAU,EAAE,IAAI,CAAC,UAAU;YAC3B,MAAM,EAAE,IAAI,CAAC,MAAM;YACnB,WAAW,EAAE,CAAC;YACd,MAAM,EAAE,OAAgB;SACzB,CAAC,CAAC,CAAC;IACN,CAAC;IAED,eAAe,CAAC,OAAsB;QACpC,OAAO,OAAO,KAAK,qBAAa,CAAC,kBAAkB,IAAI,OAAO,KAAK,qBAAa,CAAC,aAAa,CAAC;IACjG,CAAC;CACF;AAhOD,gDAgOC"}

package/dist/vector/adapters/index.d.ts

@@ -7,4 +7,7 @@ export { BaseVectorAdapter, type IAdapterConnectionOptions } from './base.adapte
 export { PineconeAdapter } from './pinecone.adapter';
 export { QdrantAdapter } from './qdrant.adapter';
 export { WeaviateAdapter } from './weaviate.adapter';
+export { OpenSearchAdapter } from './opensearch.adapter';
+export { AzureSearchAdapter } from './azure-search.adapter';
+export { VertexVectorSearchAdapter } from './vertex-vector-search.adapter';
 export { MemoryAdapter } from './memory.adapter';

package/dist/vector/adapters/index.js

@@ -5,7 +5,7 @@
  * Export all vector database adapters.
  */
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.MemoryAdapter = exports.WeaviateAdapter = exports.QdrantAdapter = exports.PineconeAdapter = exports.BaseVectorAdapter = void 0;
+exports.MemoryAdapter = exports.VertexVectorSearchAdapter = exports.AzureSearchAdapter = exports.OpenSearchAdapter = exports.WeaviateAdapter = exports.QdrantAdapter = exports.PineconeAdapter = exports.BaseVectorAdapter = void 0;
 var base_adapter_1 = require("./base.adapter");
 Object.defineProperty(exports, "BaseVectorAdapter", { enumerable: true, get: function () { return base_adapter_1.BaseVectorAdapter; } });
 var pinecone_adapter_1 = require("./pinecone.adapter");
@@ -14,6 +14,12 @@ var qdrant_adapter_1 = require("./qdrant.adapter");
 Object.defineProperty(exports, "QdrantAdapter", { enumerable: true, get: function () { return qdrant_adapter_1.QdrantAdapter; } });
 var weaviate_adapter_1 = require("./weaviate.adapter");
 Object.defineProperty(exports, "WeaviateAdapter", { enumerable: true, get: function () { return weaviate_adapter_1.WeaviateAdapter; } });
+var opensearch_adapter_1 = require("./opensearch.adapter");
+Object.defineProperty(exports, "OpenSearchAdapter", { enumerable: true, get: function () { return opensearch_adapter_1.OpenSearchAdapter; } });
+var azure_search_adapter_1 = require("./azure-search.adapter");
+Object.defineProperty(exports, "AzureSearchAdapter", { enumerable: true, get: function () { return azure_search_adapter_1.AzureSearchAdapter; } });
+var vertex_vector_search_adapter_1 = require("./vertex-vector-search.adapter");
+Object.defineProperty(exports, "VertexVectorSearchAdapter", { enumerable: true, get: function () { return vertex_vector_search_adapter_1.VertexVectorSearchAdapter; } });
 var memory_adapter_1 = require("./memory.adapter");
 Object.defineProperty(exports, "MemoryAdapter", { enumerable: true, get: function () { return memory_adapter_1.MemoryAdapter; } });
 //# sourceMappingURL=index.js.map

package/dist/vector/adapters/index.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../src/vector/adapters/index.ts"],"names":[],"mappings":";AAAA;;;;GAIG;;;AAEH,+CAAmF;AAA1E,iHAAA,iBAAiB,OAAA;AAC1B,uDAAqD;AAA5C,mHAAA,eAAe,OAAA;AACxB,mDAAiD;AAAxC,+GAAA,aAAa,OAAA;AACtB,uDAAqD;AAA5C,mHAAA,eAAe,OAAA;AACxB,mDAAiD;AAAxC,+GAAA,aAAa,OAAA"}
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../src/vector/adapters/index.ts"],"names":[],"mappings":";AAAA;;;;GAIG;;;AAEH,+CAAmF;AAA1E,iHAAA,iBAAiB,OAAA;AAC1B,uDAAqD;AAA5C,mHAAA,eAAe,OAAA;AACxB,mDAAiD;AAAxC,+GAAA,aAAa,OAAA;AACtB,uDAAqD;AAA5C,mHAAA,eAAe,OAAA;AACxB,2DAAyD;AAAhD,uHAAA,iBAAiB,OAAA;AAC1B,+DAA4D;AAAnD,0HAAA,kBAAkB,OAAA;AAC3B,+EAA2E;AAAlE,yIAAA,yBAAyB,OAAA;AAClC,mDAAiD;AAAxC,+GAAA,aAAa,OAAA"}

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

@@ -0,0 +1,25 @@
+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';
+export declare class OpenSearchAdapter extends BaseVectorAdapter {
+    readonly type = VectorDBType.OPENSEARCH;
+    private client;
+    private indexName;
+    private vectorField;
+    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(): 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;
+}

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

@@ -0,0 +1,224 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.OpenSearchAdapter = void 0;
+/**
+ * AWS OpenSearch / Elasticsearch-compatible vector adapter (k-NN REST API).
+ */
+const axios_1 = __importDefault(require("axios"));
+const types_1 = require("../types");
+const base_adapter_1 = require("./base.adapter");
+const DEFAULT_VECTOR_FIELD = 'embedding';
+class OpenSearchAdapter extends base_adapter_1.BaseVectorAdapter {
+    constructor() {
+        super(...arguments);
+        this.type = types_1.VectorDBType.OPENSEARCH;
+        this.client = null;
+        this.indexName = '';
+        this.vectorField = DEFAULT_VECTOR_FIELD;
+    }
+    async connect(options) {
+        var _a, _b;
+        if (!options.endpoint) {
+            return { connected: false, error: 'OpenSearch endpoint is required' };
+        }
+        this.connectionOptions = options;
+        this.indexName = options.index;
+        this.vectorField = String(((_a = options.options) === null || _a === void 0 ? void 0 : _a.vectorField) || DEFAULT_VECTOR_FIELD);
+        const headers = { 'Content-Type': 'application/json' };
+        if (options.apiKey) {
+            headers.Authorization = `Basic ${Buffer.from(`admin:${options.apiKey}`).toString('base64')}`;
+        }
+        this.client = axios_1.default.create({
+            baseURL: options.endpoint.replace(/\/$/, ''),
+            headers,
+            timeout: ((_b = options.options) === null || _b === void 0 ? void 0 : _b.timeout) || 30000,
+        });
+        const ok = await this.testConnection();
+        this._connected = ok;
+        if (!ok) {
+            return { connected: false, error: 'Failed to connect to OpenSearch' };
+        }
+        try {
+            const info = await this.describeIndex();
+            return {
+                connected: true,
+                index: this.indexName,
+                dimensions: info.dimensions,
+                vectorCount: info.vectorCount,
+            };
+        }
+        catch (_c) {
+            return {
+                connected: true,
+                index: this.indexName,
+                dimensions: options.dimensions,
+                vectorCount: 0,
+            };
+        }
+    }
+    async disconnect() {
+        this.client = null;
+        this._connected = false;
+        this.connectionOptions = null;
+    }
+    async testConnection() {
+        if (!this.client)
+            return false;
+        try {
+            await this.client.get('/');
+            return true;
+        }
+        catch (_a) {
+            return false;
+        }
+    }
+    async upsert(options) {
+        this.validateDimensions(options.vectors);
+        for (const vector of options.vectors) {
+            await this.client.put(`/${this.indexName}/_doc/${encodeURIComponent(vector.id)}`, Object.assign({ [this.vectorField]: vector.values }, (vector.metadata || {})));
+        }
+        return {
+            upsertedCount: options.vectors.length,
+            upsertedIds: options.vectors.map((v) => v.id),
+        };
+    }
+    async query(options) {
+        var _a;
+        const start = Date.now();
+        const body = {
+            size: options.topK,
+            query: {
+                knn: {
+                    [this.vectorField]: {
+                        vector: options.vector,
+                        k: options.topK,
+                    },
+                },
+            },
+        };
+        const { data } = await this.client.post(`/${this.indexName}/_search`, body);
+        const hits = ((_a = data === null || data === void 0 ? void 0 : data.hits) === null || _a === void 0 ? void 0 : _a.hits) || [];
+        return {
+            matches: hits.map((hit) => {
+                var _a, _b;
+                return ({
+                    id: String(hit._id),
+                    score: (_a = hit._score) !== null && _a !== void 0 ? _a : 0,
+                    values: options.includeValues ? (_b = hit._source) === null || _b === void 0 ? void 0 : _b[this.vectorField] : undefined,
+                    metadata: hit._source,
+                });
+            }),
+            queryTime: Date.now() - start,
+        };
+    }
+    async fetch(options) {
+        var _a;
+        const vectors = {};
+        for (const id of options.ids) {
+            try {
+                const { data } = await this.client.get(`/${this.indexName}/_doc/${encodeURIComponent(id)}`);
+                if (data.found) {
+                    vectors[id] = {
+                        id,
+                        values: options.includeValues ? (_a = data._source) === null || _a === void 0 ? void 0 : _a[this.vectorField] : undefined,
+                        metadata: data._source,
+                    };
+                }
+                else {
+                    vectors[id] = null;
+                }
+            }
+            catch (_b) {
+                vectors[id] = null;
+            }
+        }
+        return { vectors };
+    }
+    async update(options) {
+        const doc = Object.assign({}, (options.setMetadata || options.mergeMetadata || {}));
+        if (options.values)
+            doc[this.vectorField] = options.values;
+        await this.client.post(`/${this.indexName}/_update/${encodeURIComponent(options.id)}`, { doc });
+        return { success: true, id: options.id };
+    }
+    async delete(options) {
+        var _a;
+        if ((_a = options.ids) === null || _a === void 0 ? void 0 : _a.length) {
+            for (const id of options.ids) {
+                await this.client.delete(`/${this.indexName}/_doc/${encodeURIComponent(id)}`);
+            }
+            return { deletedCount: options.ids.length, deletedIds: options.ids };
+        }
+        if (options.deleteAll) {
+            await this.client.post(`/${this.indexName}/_delete_by_query`, { query: { match_all: {} } });
+            return { deletedCount: -1 };
+        }
+        return { deletedCount: 0 };
+    }
+    async list(options) {
+        var _a, _b, _c, _d;
+        const { data } = await this.client.post(`/${this.indexName}/_search`, {
+            size: options.limit || 100,
+            _source: false,
+        });
+        const ids = (((_a = data === null || data === void 0 ? void 0 : data.hits) === null || _a === void 0 ? void 0 : _a.hits) || []).map((h) => String(h._id));
+        return { ids, total: (_d = (_c = (_b = data === null || data === void 0 ? void 0 : data.hits) === null || _b === void 0 ? void 0 : _b.total) === null || _c === void 0 ? void 0 : _c.value) !== null && _d !== void 0 ? _d : ids.length };
+    }
+    async listNamespaces() {
+        return { namespaces: [] };
+    }
+    async deleteNamespace() { }
+    async describeIndex() {
+        const stats = await this.getIndexStats();
+        return {
+            name: this.indexName,
+            dimensions: this.dimensions,
+            metric: this.metric,
+            vectorCount: stats.vectorCount,
+            status: 'ready',
+        };
+    }
+    async getIndexStats() {
+        var _a;
+        const { data } = await this.client.get(`/${this.indexName}/_count`);
+        return { vectorCount: (_a = data === null || data === void 0 ? void 0 : data.count) !== null && _a !== void 0 ? _a : 0, dimensions: this.dimensions };
+    }
+    async createIndex(options) {
+        const metric = options.metric || types_1.DistanceMetric.COSINE;
+        const spaceType = metric === types_1.DistanceMetric.EUCLIDEAN ? 'l2' : metric === types_1.DistanceMetric.DOT_PRODUCT ? 'innerproduct' : 'cosinesimil';
+        await this.client.put(`/${options.name}`, {
+            mappings: {
+                properties: {
+                    [this.vectorField]: {
+                        type: 'knn_vector',
+                        dimension: options.dimensions,
+                        method: { name: 'hnsw', space_type: spaceType, engine: 'nmslib' },
+                    },
+                },
+            },
+        });
+        return { success: true, name: options.name };
+    }
+    async deleteIndex(options) {
+        await this.client.delete(`/${options.name}`);
+        return { success: true };
+    }
+    async listIndexes() {
+        const { data } = await this.client.get('/_cat/indices?format=json');
+        return (data || []).map((idx) => ({
+            name: idx.index,
+            dimensions: this.dimensions,
+            metric: this.metric,
+            vectorCount: Number(idx['docs.count'] || 0),
+            status: 'ready',
+        }));
+    }
+    supportsFeature(feature) {
+        return feature === types_1.VectorFeature.METADATA_FILTERING || feature === types_1.VectorFeature.VECTOR_UPDATES;
+    }
+}
+exports.OpenSearchAdapter = OpenSearchAdapter;
+//# sourceMappingURL=opensearch.adapter.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"opensearch.adapter.js","sourceRoot":"","sources":["../../../src/vector/adapters/opensearch.adapter.ts"],"names":[],"mappings":";;;;;;AAAA;;GAEG;AACH,kDAA6C;AAC7C,oCAwBkB;AAClB,iDAA8E;AAE9E,MAAM,oBAAoB,GAAG,WAAW,CAAC;AAEzC,MAAa,iBAAkB,SAAQ,gCAAiB;IAAxD;;QACkB,SAAI,GAAG,oBAAY,CAAC,UAAU,CAAC;QAEvC,WAAM,GAAyB,IAAI,CAAC;QACpC,cAAS,GAAG,EAAE,CAAC;QACf,gBAAW,GAAG,oBAAoB,CAAC;IA+M7C,CAAC;IA7MC,KAAK,CAAC,OAAO,CAAC,OAAkC;;QAC9C,IAAI,CAAC,OAAO,CAAC,QAAQ,EAAE,CAAC;YACtB,OAAO,EAAE,SAAS,EAAE,KAAK,EAAE,KAAK,EAAE,iCAAiC,EAAE,CAAC;QACxE,CAAC;QACD,IAAI,CAAC,iBAAiB,GAAG,OAAO,CAAC;QACjC,IAAI,CAAC,SAAS,GAAG,OAAO,CAAC,KAAK,CAAC;QAC/B,IAAI,CAAC,WAAW,GAAG,MAAM,CAAC,CAAA,MAAA,OAAO,CAAC,OAAO,0CAAE,WAAW,KAAI,oBAAoB,CAAC,CAAC;QAChF,MAAM,OAAO,GAA2B,EAAE,cAAc,EAAE,kBAAkB,EAAE,CAAC;QAC/E,IAAI,OAAO,CAAC,MAAM,EAAE,CAAC;YACnB,OAAO,CAAC,aAAa,GAAG,SAAS,MAAM,CAAC,IAAI,CAAC,SAAS,OAAO,CAAC,MAAM,EAAE,CAAC,CAAC,QAAQ,CAAC,QAAQ,CAAC,EAAE,CAAC;QAC/F,CAAC;QACD,IAAI,CAAC,MAAM,GAAG,eAAK,CAAC,MAAM,CAAC;YACzB,OAAO,EAAE,OAAO,CAAC,QAAQ,CAAC,OAAO,CAAC,KAAK,EAAE,EAAE,CAAC;YAC5C,OAAO;YACP,OAAO,EAAE,CAAC,MAAA,OAAO,CAAC,OAAO,0CAAE,OAAkB,KAAI,KAAK;SACvD,CAAC,CAAC;QACH,MAAM,EAAE,GAAG,MAAM,IAAI,CAAC,cAAc,EAAE,CAAC;QACvC,IAAI,CAAC,UAAU,GAAG,EAAE,CAAC;QACrB,IAAI,CAAC,EAAE,EAAE,CAAC;YACR,OAAO,EAAE,SAAS,EAAE,KAAK,EAAE,KAAK,EAAE,iCAAiC,EAAE,CAAC;QACxE,CAAC;QACD,IAAI,CAAC;YACH,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,aAAa,EAAE,CAAC;YACxC,OAAO;gBACL,SAAS,EAAE,IAAI;gBACf,KAAK,EAAE,IAAI,CAAC,SAAS;gBACrB,UAAU,EAAE,IAAI,CAAC,UAAU;gBAC3B,WAAW,EAAE,IAAI,CAAC,WAAW;aAC9B,CAAC;QACJ,CAAC;QAAC,WAAM,CAAC;YACP,OAAO;gBACL,SAAS,EAAE,IAAI;gBACf,KAAK,EAAE,IAAI,CAAC,SAAS;gBACrB,UAAU,EAAE,OAAO,CAAC,UAAU;gBAC9B,WAAW,EAAE,CAAC;aACf,CAAC;QACJ,CAAC;IACH,CAAC;IAED,KAAK,CAAC,UAAU;QACd,IAAI,CAAC,MAAM,GAAG,IAAI,CAAC;QACnB,IAAI,CAAC,UAAU,GAAG,KAAK,CAAC;QACxB,IAAI,CAAC,iBAAiB,GAAG,IAAI,CAAC;IAChC,CAAC;IAED,KAAK,CAAC,cAAc;QAClB,IAAI,CAAC,IAAI,CAAC,MAAM;YAAE,OAAO,KAAK,CAAC;QAC/B,IAAI,CAAC;YACH,MAAM,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;YAC3B,OAAO,IAAI,CAAC;QACd,CAAC;QAAC,WAAM,CAAC;YACP,OAAO,KAAK,CAAC;QACf,CAAC;IACH,CAAC;IAED,KAAK,CAAC,MAAM,CAAC,OAA8B;QACzC,IAAI,CAAC,kBAAkB,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;QACzC,KAAK,MAAM,MAAM,IAAI,OAAO,CAAC,OAAO,EAAE,CAAC;YACrC,MAAM,IAAI,CAAC,MAAO,CAAC,GAAG,CAAC,IAAI,IAAI,CAAC,SAAS,SAAS,kBAAkB,CAAC,MAAM,CAAC,EAAE,CAAC,EAAE,kBAC/E,CAAC,IAAI,CAAC,WAAW,CAAC,EAAE,MAAM,CAAC,MAAM,IAC9B,CAAC,MAAM,CAAC,QAAQ,IAAI,EAAE,CAAC,EAC1B,CAAC;QACL,CAAC;QACD,OAAO;YACL,aAAa,EAAE,OAAO,CAAC,OAAO,CAAC,MAAM;YACrC,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,KAAK,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QACzB,MAAM,IAAI,GAAG;YACX,IAAI,EAAE,OAAO,CAAC,IAAI;YAClB,KAAK,EAAE;gBACL,GAAG,EAAE;oBACH,CAAC,IAAI,CAAC,WAAW,CAAC,EAAE;wBAClB,MAAM,EAAE,OAAO,CAAC,MAAM;wBACtB,CAAC,EAAE,OAAO,CAAC,IAAI;qBAChB;iBACF;aACF;SACF,CAAC;QACF,MAAM,EAAE,IAAI,EAAE,GAAG,MAAM,IAAI,CAAC,MAAO,CAAC,IAAI,CAAC,IAAI,IAAI,CAAC,SAAS,UAAU,EAAE,IAAI,CAAC,CAAC;QAC7E,MAAM,IAAI,GAAG,CAAA,MAAA,IAAI,aAAJ,IAAI,uBAAJ,IAAI,CAAE,IAAI,0CAAE,IAAI,KAAI,EAAE,CAAC;QACpC,OAAO;YACL,OAAO,EAAE,IAAI,CAAC,GAAG,CAAC,CAAC,GAAQ,EAAE,EAAE;;gBAAC,OAAA,CAAC;oBAC/B,EAAE,EAAE,MAAM,CAAC,GAAG,CAAC,GAAG,CAAC;oBACnB,KAAK,EAAE,MAAA,GAAG,CAAC,MAAM,mCAAI,CAAC;oBACtB,MAAM,EAAE,OAAO,CAAC,aAAa,CAAC,CAAC,CAAC,MAAA,GAAG,CAAC,OAAO,0CAAG,IAAI,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC,SAAS;oBAC3E,QAAQ,EAAE,GAAG,CAAC,OAAO;iBACtB,CAAC,CAAA;aAAA,CAAC;YACH,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE,GAAG,KAAK;SAC9B,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,KAAK,CAAC,OAA6B;;QACvC,MAAM,OAAO,GAAmC,EAAE,CAAC;QACnD,KAAK,MAAM,EAAE,IAAI,OAAO,CAAC,GAAG,EAAE,CAAC;YAC7B,IAAI,CAAC;gBACH,MAAM,EAAE,IAAI,EAAE,GAAG,MAAM,IAAI,CAAC,MAAO,CAAC,GAAG,CAAC,IAAI,IAAI,CAAC,SAAS,SAAS,kBAAkB,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC;gBAC7F,IAAI,IAAI,CAAC,KAAK,EAAE,CAAC;oBACf,OAAO,CAAC,EAAE,CAAC,GAAG;wBACZ,EAAE;wBACF,MAAM,EAAE,OAAO,CAAC,aAAa,CAAC,CAAC,CAAC,MAAA,IAAI,CAAC,OAAO,0CAAG,IAAI,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC,SAAS;wBAC5E,QAAQ,EAAE,IAAI,CAAC,OAAO;qBACvB,CAAC;gBACJ,CAAC;qBAAM,CAAC;oBACN,OAAO,CAAC,EAAE,CAAC,GAAG,IAAI,CAAC;gBACrB,CAAC;YACH,CAAC;YAAC,WAAM,CAAC;gBACP,OAAO,CAAC,EAAE,CAAC,GAAG,IAAI,CAAC;YACrB,CAAC;QACH,CAAC;QACD,OAAO,EAAE,OAAO,EAAE,CAAC;IACrB,CAAC;IAED,KAAK,CAAC,MAAM,CAAC,OAA6B;QACxC,MAAM,GAAG,qBAAiC,CAAC,OAAO,CAAC,WAAW,IAAI,OAAO,CAAC,aAAa,IAAI,EAAE,CAAC,CAAE,CAAC;QACjG,IAAI,OAAO,CAAC,MAAM;YAAE,GAAG,CAAC,IAAI,CAAC,WAAW,CAAC,GAAG,OAAO,CAAC,MAAM,CAAC;QAC3D,MAAM,IAAI,CAAC,MAAO,CAAC,IAAI,CAAC,IAAI,IAAI,CAAC,SAAS,YAAY,kBAAkB,CAAC,OAAO,CAAC,EAAE,CAAC,EAAE,EAAE,EAAE,GAAG,EAAE,CAAC,CAAC;QACjG,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,EAAE,EAAE,OAAO,CAAC,EAAE,EAAE,CAAC;IAC3C,CAAC;IAED,KAAK,CAAC,MAAM,CAAC,OAA8B;;QACzC,IAAI,MAAA,OAAO,CAAC,GAAG,0CAAE,MAAM,EAAE,CAAC;YACxB,KAAK,MAAM,EAAE,IAAI,OAAO,CAAC,GAAG,EAAE,CAAC;gBAC7B,MAAM,IAAI,CAAC,MAAO,CAAC,MAAM,CAAC,IAAI,IAAI,CAAC,SAAS,SAAS,kBAAkB,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC;YACjF,CAAC;YACD,OAAO,EAAE,YAAY,EAAE,OAAO,CAAC,GAAG,CAAC,MAAM,EAAE,UAAU,EAAE,OAAO,CAAC,GAAG,EAAE,CAAC;QACvE,CAAC;QACD,IAAI,OAAO,CAAC,SAAS,EAAE,CAAC;YACtB,MAAM,IAAI,CAAC,MAAO,CAAC,IAAI,CAAC,IAAI,IAAI,CAAC,SAAS,mBAAmB,EAAE,EAAE,KAAK,EAAE,EAAE,SAAS,EAAE,EAAE,EAAE,EAAE,CAAC,CAAC;YAC7F,OAAO,EAAE,YAAY,EAAE,CAAC,CAAC,EAAE,CAAC;QAC9B,CAAC;QACD,OAAO,EAAE,YAAY,EAAE,CAAC,EAAE,CAAC;IAC7B,CAAC;IAED,KAAK,CAAC,IAAI,CAAC,OAA4B;;QACrC,MAAM,EAAE,IAAI,EAAE,GAAG,MAAM,IAAI,CAAC,MAAO,CAAC,IAAI,CAAC,IAAI,IAAI,CAAC,SAAS,UAAU,EAAE;YACrE,IAAI,EAAE,OAAO,CAAC,KAAK,IAAI,GAAG;YAC1B,OAAO,EAAE,KAAK;SACf,CAAC,CAAC;QACH,MAAM,GAAG,GAAG,CAAC,CAAA,MAAA,IAAI,aAAJ,IAAI,uBAAJ,IAAI,CAAE,IAAI,0CAAE,IAAI,KAAI,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,CAAM,EAAE,EAAE,CAAC,MAAM,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC;QACpE,OAAO,EAAE,GAAG,EAAE,KAAK,EAAE,MAAA,MAAA,MAAA,IAAI,aAAJ,IAAI,uBAAJ,IAAI,CAAE,IAAI,0CAAE,KAAK,0CAAE,KAAK,mCAAI,GAAG,CAAC,MAAM,EAAE,CAAC;IAChE,CAAC;IAED,KAAK,CAAC,cAAc;QAClB,OAAO,EAAE,UAAU,EAAE,EAAE,EAAE,CAAC;IAC5B,CAAC;IAED,KAAK,CAAC,eAAe,KAAmB,CAAC;IAEzC,KAAK,CAAC,aAAa;QACjB,MAAM,KAAK,GAAG,MAAM,IAAI,CAAC,aAAa,EAAE,CAAC;QACzC,OAAO;YACL,IAAI,EAAE,IAAI,CAAC,SAAS;YACpB,UAAU,EAAE,IAAI,CAAC,UAAU;YAC3B,MAAM,EAAE,IAAI,CAAC,MAAM;YACnB,WAAW,EAAE,KAAK,CAAC,WAAW;YAC9B,MAAM,EAAE,OAAO;SAChB,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,aAAa;;QACjB,MAAM,EAAE,IAAI,EAAE,GAAG,MAAM,IAAI,CAAC,MAAO,CAAC,GAAG,CAAC,IAAI,IAAI,CAAC,SAAS,SAAS,CAAC,CAAC;QACrE,OAAO,EAAE,WAAW,EAAE,MAAA,IAAI,aAAJ,IAAI,uBAAJ,IAAI,CAAE,KAAK,mCAAI,CAAC,EAAE,UAAU,EAAE,IAAI,CAAC,UAAU,EAAE,CAAC;IACxE,CAAC;IAED,KAAK,CAAC,WAAW,CAAC,OAA4B;QAC5C,MAAM,MAAM,GAAG,OAAO,CAAC,MAAM,IAAI,sBAAc,CAAC,MAAM,CAAC;QACvD,MAAM,SAAS,GACb,MAAM,KAAK,sBAAc,CAAC,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,MAAM,KAAK,sBAAc,CAAC,WAAW,CAAC,CAAC,CAAC,cAAc,CAAC,CAAC,CAAC,aAAa,CAAC;QACtH,MAAM,IAAI,CAAC,MAAO,CAAC,GAAG,CAAC,IAAI,OAAO,CAAC,IAAI,EAAE,EAAE;YACzC,QAAQ,EAAE;gBACR,UAAU,EAAE;oBACV,CAAC,IAAI,CAAC,WAAW,CAAC,EAAE;wBAClB,IAAI,EAAE,YAAY;wBAClB,SAAS,EAAE,OAAO,CAAC,UAAU;wBAC7B,MAAM,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE,UAAU,EAAE,SAAS,EAAE,MAAM,EAAE,QAAQ,EAAE;qBAClE;iBACF;aACF;SACF,CAAC,CAAC;QACH,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,IAAI,EAAE,OAAO,CAAC,IAAI,EAAE,CAAC;IAC/C,CAAC;IAED,KAAK,CAAC,WAAW,CAAC,OAA4B;QAC5C,MAAM,IAAI,CAAC,MAAO,CAAC,MAAM,CAAC,IAAI,OAAO,CAAC,IAAI,EAAE,CAAC,CAAC;QAC9C,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC;IAC3B,CAAC;IAED,KAAK,CAAC,WAAW;QACf,MAAM,EAAE,IAAI,EAAE,GAAG,MAAM,IAAI,CAAC,MAAO,CAAC,GAAG,CAAC,2BAA2B,CAAC,CAAC;QACrE,OAAO,CAAC,IAAI,IAAI,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,GAAQ,EAAE,EAAE,CAAC,CAAC;YACrC,IAAI,EAAE,GAAG,CAAC,KAAK;YACf,UAAU,EAAE,IAAI,CAAC,UAAU;YAC3B,MAAM,EAAE,IAAI,CAAC,MAAM;YACnB,WAAW,EAAE,MAAM,CAAC,GAAG,CAAC,YAAY,CAAC,IAAI,CAAC,CAAC;YAC3C,MAAM,EAAE,OAAgB;SACzB,CAAC,CAAC,CAAC;IACN,CAAC;IAED,eAAe,CAAC,OAAsB;QACpC,OAAO,OAAO,KAAK,qBAAa,CAAC,kBAAkB,IAAI,OAAO,KAAK,qBAAa,CAAC,cAAc,CAAC;IAClG,CAAC;CACF;AApND,8CAoNC"}

package/dist/vector/adapters/vertex-vector-search.adapter.d.ts

@@ -0,0 +1,29 @@
+import { VectorDBType, VectorFeature, IVectorConnectionResult, IUpsertVectorsOptions, IUpsertVectorsResult, IQueryVectorsOptions, IQueryVectorsResult, IFetchVectorsOptions, IFetchVectorsResult, IUpdateVectorResult, IDeleteVectorsResult, IListVectorsResult, IListNamespacesResult, IVectorIndexInfo, ICreateIndexResult, IDeleteIndexResult, IVectorIndexStats } from '../types';
+import { BaseVectorAdapter, IAdapterConnectionOptions } from './base.adapter';
+export declare class VertexVectorSearchAdapter extends BaseVectorAdapter {
+    readonly type = VectorDBType.VERTEX_VECTOR_SEARCH;
+    private client;
+    private auth;
+    private indexEndpoint;
+    private deployedIndexId;
+    private projectId;
+    private location;
+    connect(options: IAdapterConnectionOptions): Promise<IVectorConnectionResult>;
+    private authHeaders;
+    disconnect(): Promise<void>;
+    testConnection(): Promise<boolean>;
+    upsert(options: IUpsertVectorsOptions): Promise<IUpsertVectorsResult>;
+    query(options: IQueryVectorsOptions): Promise<IQueryVectorsResult>;
+    fetch(options: IFetchVectorsOptions): Promise<IFetchVectorsResult>;
+    update(): Promise<IUpdateVectorResult>;
+    delete(): Promise<IDeleteVectorsResult>;
+    list(): Promise<IListVectorsResult>;
+    listNamespaces(): Promise<IListNamespacesResult>;
+    deleteNamespace(): Promise<void>;
+    describeIndex(): Promise<IVectorIndexInfo>;
+    getIndexStats(): Promise<IVectorIndexStats>;
+    createIndex(): Promise<ICreateIndexResult>;
+    deleteIndex(): Promise<IDeleteIndexResult>;
+    listIndexes(): Promise<IVectorIndexInfo[]>;
+    supportsFeature(feature: VectorFeature): boolean;
+}