vectra 0.12.2 → 0.14.0

package/lib/templates/typescript/VectraClient.js

@@ -0,0 +1,328 @@
+"use strict";
+/**
+ * Vectra gRPC client — thin idiomatic wrapper over generated stubs.
+ *
+ * Usage:
+ *   import { VectraClient } from './VectraClient';
+ *
+ *   const client = new VectraClient();
+ *   const results = await client.queryDocuments('my-index', 'search query');
+ *   client.close();
+ *
+ * Generate stubs first:
+ *   npm install @grpc/grpc-js @grpc/proto-loader
+ *   // Proto is loaded dynamically — no separate codegen step required.
+ */
+var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    var desc = Object.getOwnPropertyDescriptor(m, k);
+    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
+      desc = { enumerable: true, get: function() { return m[k]; } };
+    }
+    Object.defineProperty(o, k2, desc);
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
+    Object.defineProperty(o, "default", { enumerable: true, value: v });
+}) : function(o, v) {
+    o["default"] = v;
+});
+var __importStar = (this && this.__importStar) || (function () {
+    var ownKeys = function(o) {
+        ownKeys = Object.getOwnPropertyNames || function (o) {
+            var ar = [];
+            for (var k in o) if (Object.prototype.hasOwnProperty.call(o, k)) ar[ar.length] = k;
+            return ar;
+        };
+        return ownKeys(o);
+    };
+    return function (mod) {
+        if (mod && mod.__esModule) return mod;
+        var result = {};
+        if (mod != null) for (var k = ownKeys(mod), i = 0; i < k.length; i++) if (k[i] !== "default") __createBinding(result, mod, k[i]);
+        __setModuleDefault(result, mod);
+        return result;
+    };
+})();
+var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
+    function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
+    return new (P || (P = Promise))(function (resolve, reject) {
+        function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
+        function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
+        function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); }
+        step((generator = generator.apply(thisArg, _arguments || [])).next());
+    });
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.VectraClient = void 0;
+const grpc = __importStar(require("@grpc/grpc-js"));
+const protoLoader = __importStar(require("@grpc/proto-loader"));
+const path = __importStar(require("path"));
+// Load proto definition dynamically
+const PROTO_PATH = path.join(__dirname, 'vectra_service.proto');
+const packageDefinition = protoLoader.loadSync(PROTO_PATH, {
+    keepCase: false,
+    longs: Number,
+    enums: String,
+    defaults: true,
+    oneofs: true,
+});
+const protoDescriptor = grpc.loadPackageDefinition(packageDefinition);
+const vectra = protoDescriptor.vectra;
+// ── Client ───────────────────────────────────────────────
+class VectraClient {
+    constructor(host = '127.0.0.1', port = 50051) {
+        this._client = new vectra.VectraService(`${host}:${port}`, grpc.credentials.createInsecure());
+    }
+    close() {
+        this._client.close();
+    }
+    // ── Index Management ─────────────────────────────────
+    createIndex(name, options) {
+        var _a, _b, _c;
+        const req = { indexName: name, format: (_a = options === null || options === void 0 ? void 0 : options.format) !== null && _a !== void 0 ? _a : 'json' };
+        if (options === null || options === void 0 ? void 0 : options.isDocumentIndex) {
+            req.isDocumentIndex = true;
+            req.documentConfig = {
+                version: 1,
+                chunkSize: (_b = options.chunkSize) !== null && _b !== void 0 ? _b : 512,
+                chunkOverlap: (_c = options.chunkOverlap) !== null && _c !== void 0 ? _c : 0,
+            };
+        }
+        return this._unary('createIndex', req);
+    }
+    deleteIndex(name) {
+        return this._unary('deleteIndex', { indexName: name });
+    }
+    listIndexes() {
+        return __awaiter(this, void 0, void 0, function* () {
+            var _a;
+            const resp = yield this._unary('listIndexes', {});
+            return ((_a = resp.indexes) !== null && _a !== void 0 ? _a : []).map((idx) => ({
+                name: idx.name,
+                format: idx.format,
+                isDocumentIndex: idx.isDocumentIndex,
+            }));
+        });
+    }
+    // ── Item Operations ──────────────────────────────────
+    insertItem(index, options) {
+        return __awaiter(this, void 0, void 0, function* () {
+            var _a, _b;
+            const req = {
+                indexName: index,
+                text: (_a = options.text) !== null && _a !== void 0 ? _a : '',
+                id: (_b = options.id) !== null && _b !== void 0 ? _b : '',
+            };
+            if (options.vector)
+                req.vector = options.vector;
+            if (options.metadata)
+                req.metadata = toProtoMetadata(options.metadata);
+            const resp = yield this._unary('insertItem', req);
+            return resp.id;
+        });
+    }
+    upsertItem(index, id, options) {
+        return __awaiter(this, void 0, void 0, function* () {
+            var _a;
+            const req = {
+                indexName: index,
+                id,
+                text: (_a = options.text) !== null && _a !== void 0 ? _a : '',
+            };
+            if (options.vector)
+                req.vector = options.vector;
+            if (options.metadata)
+                req.metadata = toProtoMetadata(options.metadata);
+            const resp = yield this._unary('upsertItem', req);
+            return resp.id;
+        });
+    }
+    getItem(index, id) {
+        return __awaiter(this, void 0, void 0, function* () {
+            const resp = yield this._unary('getItem', { indexName: index, id });
+            return resp.item ? itemToResult(resp.item) : null;
+        });
+    }
+    deleteItem(index, id) {
+        return this._unary('deleteItem', { indexName: index, id });
+    }
+    listItems(index, filter) {
+        return __awaiter(this, void 0, void 0, function* () {
+            var _a;
+            const req = { indexName: index };
+            if (filter)
+                req.filter = { filterJson: JSON.stringify(filter) };
+            const resp = yield this._unary('listItems', req);
+            return ((_a = resp.items) !== null && _a !== void 0 ? _a : []).map(itemToResult);
+        });
+    }
+    // ── Query ────────────────────────────────────────────
+    queryItems(index, options) {
+        return __awaiter(this, void 0, void 0, function* () {
+            var _a, _b, _c;
+            const req = {
+                indexName: index,
+                text: (_a = options.text) !== null && _a !== void 0 ? _a : '',
+                topK: (_b = options.topK) !== null && _b !== void 0 ? _b : 10,
+            };
+            if (options.vector)
+                req.vector = options.vector;
+            if (options.filter)
+                req.filter = { filterJson: JSON.stringify(options.filter) };
+            const resp = yield this._unary('queryItems', req);
+            return ((_c = resp.results) !== null && _c !== void 0 ? _c : []).map(itemToResult);
+        });
+    }
+    queryDocuments(index, query, options) {
+        return __awaiter(this, void 0, void 0, function* () {
+            var _a, _b, _c, _d;
+            const req = {
+                indexName: index,
+                query,
+                maxDocuments: (_a = options === null || options === void 0 ? void 0 : options.maxDocuments) !== null && _a !== void 0 ? _a : 10,
+                maxChunks: (_b = options === null || options === void 0 ? void 0 : options.maxChunks) !== null && _b !== void 0 ? _b : 50,
+                useBm25: (_c = options === null || options === void 0 ? void 0 : options.useBm25) !== null && _c !== void 0 ? _c : false,
+            };
+            if (options === null || options === void 0 ? void 0 : options.filter)
+                req.filter = { filterJson: JSON.stringify(options.filter) };
+            const resp = yield this._unary('queryDocuments', req);
+            return ((_d = resp.results) !== null && _d !== void 0 ? _d : []).map((doc) => {
+                var _a;
+                return ({
+                    uri: doc.uri,
+                    documentId: doc.documentId,
+                    score: doc.score,
+                    chunks: ((_a = doc.chunks) !== null && _a !== void 0 ? _a : []).map((c) => ({
+                        text: c.text,
+                        score: c.score,
+                        tokenCount: c.tokenCount,
+                    })),
+                });
+            });
+        });
+    }
+    // ── Document Operations ──────────────────────────────
+    upsertDocument(index, uri, text, options) {
+        return __awaiter(this, void 0, void 0, function* () {
+            var _a;
+            const req = {
+                indexName: index,
+                uri,
+                text,
+                docType: (_a = options === null || options === void 0 ? void 0 : options.docType) !== null && _a !== void 0 ? _a : '',
+            };
+            if (options === null || options === void 0 ? void 0 : options.metadata)
+                req.metadata = toProtoMetadata(options.metadata);
+            const resp = yield this._unary('upsertDocument', req);
+            return resp.documentId;
+        });
+    }
+    deleteDocument(index, uri) {
+        return this._unary('deleteDocument', { indexName: index, uri });
+    }
+    listDocuments(index) {
+        return __awaiter(this, void 0, void 0, function* () {
+            var _a;
+            const resp = yield this._unary('listDocuments', { indexName: index });
+            return ((_a = resp.documents) !== null && _a !== void 0 ? _a : []).map((d) => ({
+                uri: d.uri,
+                documentId: d.documentId,
+            }));
+        });
+    }
+    // ── Stats ────────────────────────────────────────────
+    getIndexStats(index) {
+        return __awaiter(this, void 0, void 0, function* () {
+            const resp = yield this._unary('getIndexStats', { indexName: index });
+            return {
+                version: resp.version,
+                format: resp.format,
+                itemCount: resp.itemCount,
+                metadataConfigCount: resp.metadataConfigCount,
+            };
+        });
+    }
+    getCatalogStats(index) {
+        return __awaiter(this, void 0, void 0, function* () {
+            var _a;
+            const resp = yield this._unary('getCatalogStats', { indexName: index });
+            return {
+                version: resp.version,
+                documentCount: resp.documentCount,
+                chunkCount: resp.chunkCount,
+                metadataCounts: (_a = resp.metadataCounts) !== null && _a !== void 0 ? _a : {},
+            };
+        });
+    }
+    // ── Lifecycle ────────────────────────────────────────
+    healthcheck() {
+        return __awaiter(this, void 0, void 0, function* () {
+            var _a;
+            const resp = yield this._unary('healthcheck', {});
+            return {
+                status: resp.status,
+                uptimeSeconds: resp.uptimeSeconds,
+                loadedIndexes: (_a = resp.loadedIndexes) !== null && _a !== void 0 ? _a : [],
+            };
+        });
+    }
+    shutdown() {
+        return this._unary('shutdown', {});
+    }
+    // ── Internals ────────────────────────────────────────
+    _unary(method, request) {
+        return new Promise((resolve, reject) => {
+            this._client[method](request, (err, response) => {
+                if (err)
+                    reject(err);
+                else
+                    resolve(response);
+            });
+        });
+    }
+}
+exports.VectraClient = VectraClient;
+// ── Helpers ──────────────────────────────────────────────
+function toProtoMetadata(metadata) {
+    const result = {};
+    for (const [key, value] of Object.entries(metadata)) {
+        if (typeof value === 'boolean') {
+            result[key] = { boolValue: value };
+        }
+        else if (typeof value === 'number') {
+            result[key] = { numberValue: value };
+        }
+        else {
+            result[key] = { stringValue: String(value) };
+        }
+    }
+    return result;
+}
+function itemToResult(item) {
+    var _a;
+    const metadata = {};
+    if (item.metadata) {
+        for (const [key, val] of Object.entries(item.metadata)) {
+            if (val.stringValue !== undefined && val.stringValue !== '') {
+                metadata[key] = val.stringValue;
+            }
+            else if (val.numberValue !== undefined && val.numberValue !== 0) {
+                metadata[key] = val.numberValue;
+            }
+            else if (val.boolValue !== undefined) {
+                metadata[key] = val.boolValue;
+            }
+        }
+    }
+    return {
+        id: item.id,
+        metadata,
+        vector: (_a = item.vector) !== null && _a !== void 0 ? _a : [],
+        norm: item.norm,
+        score: item.score,
+    };
+}
+//# sourceMappingURL=VectraClient.js.map

package/lib/templates/typescript/VectraClient.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"VectraClient.js","sourceRoot":"","sources":["../../../src/templates/typescript/VectraClient.ts"],"names":[],"mappings":";AAAA;;;;;;;;;;;;;GAaG;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAEH,oDAAsC;AACtC,gEAAkD;AAClD,2CAA6B;AAE7B,oCAAoC;AACpC,MAAM,UAAU,GAAG,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,sBAAsB,CAAC,CAAC;AAChE,MAAM,iBAAiB,GAAG,WAAW,CAAC,QAAQ,CAAC,UAAU,EAAE;IACvD,QAAQ,EAAE,KAAK;IACf,KAAK,EAAE,MAAM;IACb,KAAK,EAAE,MAAM;IACb,QAAQ,EAAE,IAAI;IACd,MAAM,EAAE,IAAI;CACf,CAAC,CAAC;AACH,MAAM,eAAe,GAAG,IAAI,CAAC,qBAAqB,CAAC,iBAAiB,CAAQ,CAAC;AAC7E,MAAM,MAAM,GAAG,eAAe,CAAC,MAAM,CAAC;AAiEtC,4DAA4D;AAE5D,MAAa,YAAY;IAGrB,YAAY,OAAe,WAAW,EAAE,OAAe,KAAK;QACxD,IAAI,CAAC,OAAO,GAAG,IAAI,MAAM,CAAC,aAAa,CACnC,GAAG,IAAI,IAAI,IAAI,EAAE,EACjB,IAAI,CAAC,WAAW,CAAC,cAAc,EAAE,CACpC,CAAC;IACN,CAAC;IAEM,KAAK;QACR,IAAI,CAAC,OAAO,CAAC,KAAK,EAAE,CAAC;IACzB,CAAC;IAED,wDAAwD;IAEjD,WAAW,CACd,IAAY,EACZ,OAKC;;QAED,MAAM,GAAG,GAAQ,EAAE,SAAS,EAAE,IAAI,EAAE,MAAM,EAAE,MAAA,OAAO,aAAP,OAAO,uBAAP,OAAO,CAAE,MAAM,mCAAI,MAAM,EAAE,CAAC;QACxE,IAAI,OAAO,aAAP,OAAO,uBAAP,OAAO,CAAE,eAAe,EAAE,CAAC;YAC3B,GAAG,CAAC,eAAe,GAAG,IAAI,CAAC;YAC3B,GAAG,CAAC,cAAc,GAAG;gBACjB,OAAO,EAAE,CAAC;gBACV,SAAS,EAAE,MAAA,OAAO,CAAC,SAAS,mCAAI,GAAG;gBACnC,YAAY,EAAE,MAAA,OAAO,CAAC,YAAY,mCAAI,CAAC;aAC1C,CAAC;QACN,CAAC;QACD,OAAO,IAAI,CAAC,MAAM,CAAC,aAAa,EAAE,GAAG,CAAC,CAAC;IAC3C,CAAC;IAEM,WAAW,CAAC,IAAY;QAC3B,OAAO,IAAI,CAAC,MAAM,CAAC,aAAa,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;IAC3D,CAAC;IAEY,WAAW;;;YACpB,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,aAAa,EAAE,EAAE,CAAC,CAAC;YAClD,OAAO,CAAC,MAAA,IAAI,CAAC,OAAO,mCAAI,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,GAAQ,EAAE,EAAE,CAAC,CAAC;gBAC3C,IAAI,EAAE,GAAG,CAAC,IAAI;gBACd,MAAM,EAAE,GAAG,CAAC,MAAM;gBAClB,eAAe,EAAE,GAAG,CAAC,eAAe;aACvC,CAAC,CAAC,CAAC;QACR,CAAC;KAAA;IAED,wDAAwD;IAE3C,UAAU,CACnB,KAAa,EACb,OAKC;;;YAED,MAAM,GAAG,GAAQ;gBACb,SAAS,EAAE,KAAK;gBAChB,IAAI,EAAE,MAAA,OAAO,CAAC,IAAI,mCAAI,EAAE;gBACxB,EAAE,EAAE,MAAA,OAAO,CAAC,EAAE,mCAAI,EAAE;aACvB,CAAC;YACF,IAAI,OAAO,CAAC,MAAM;gBAAE,GAAG,CAAC,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC;YAChD,IAAI,OAAO,CAAC,QAAQ;gBAAE,GAAG,CAAC,QAAQ,GAAG,eAAe,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;YACvE,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,YAAY,EAAE,GAAG,CAAC,CAAC;YAClD,OAAO,IAAI,CAAC,EAAE,CAAC;QACnB,CAAC;KAAA;IAEY,UAAU,CACnB,KAAa,EACb,EAAU,EACV,OAIC;;;YAED,MAAM,GAAG,GAAQ;gBACb,SAAS,EAAE,KAAK;gBAChB,EAAE;gBACF,IAAI,EAAE,MAAA,OAAO,CAAC,IAAI,mCAAI,EAAE;aAC3B,CAAC;YACF,IAAI,OAAO,CAAC,MAAM;gBAAE,GAAG,CAAC,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC;YAChD,IAAI,OAAO,CAAC,QAAQ;gBAAE,GAAG,CAAC,QAAQ,GAAG,eAAe,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;YACvE,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,YAAY,EAAE,GAAG,CAAC,CAAC;YAClD,OAAO,IAAI,CAAC,EAAE,CAAC;QACnB,CAAC;KAAA;IAEY,OAAO,CAAC,KAAa,EAAE,EAAU;;YAC1C,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,SAAS,EAAE,EAAE,SAAS,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC,CAAC;YACpE,OAAO,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,YAAY,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC;QACtD,CAAC;KAAA;IAEM,UAAU,CAAC,KAAa,EAAE,EAAU;QACvC,OAAO,IAAI,CAAC,MAAM,CAAC,YAAY,EAAE,EAAE,SAAS,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC,CAAC;IAC/D,CAAC;IAEY,SAAS,CAClB,KAAa,EACb,MAA4B;;;YAE5B,MAAM,GAAG,GAAQ,EAAE,SAAS,EAAE,KAAK,EAAE,CAAC;YACtC,IAAI,MAAM;gBAAE,GAAG,CAAC,MAAM,GAAG,EAAE,UAAU,EAAE,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,EAAE,CAAC;YAChE,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,WAAW,EAAE,GAAG,CAAC,CAAC;YACjD,OAAO,CAAC,MAAA,IAAI,CAAC,KAAK,mCAAI,EAAE,CAAC,CAAC,GAAG,CAAC,YAAY,CAAC,CAAC;QAChD,CAAC;KAAA;IAED,wDAAwD;IAE3C,UAAU,CACnB,KAAa,EACb,OAA0B;;;YAE1B,MAAM,GAAG,GAAQ;gBACb,SAAS,EAAE,KAAK;gBAChB,IAAI,EAAE,MAAA,OAAO,CAAC,IAAI,mCAAI,EAAE;gBACxB,IAAI,EAAE,MAAA,OAAO,CAAC,IAAI,mCAAI,EAAE;aAC3B,CAAC;YACF,IAAI,OAAO,CAAC,MAAM;gBAAE,GAAG,CAAC,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC;YAChD,IAAI,OAAO,CAAC,MAAM;gBAAE,GAAG,CAAC,MAAM,GAAG,EAAE,UAAU,EAAE,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,MAAM,CAAC,EAAE,CAAC;YAChF,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,YAAY,EAAE,GAAG,CAAC,CAAC;YAClD,OAAO,CAAC,MAAA,IAAI,CAAC,OAAO,mCAAI,EAAE,CAAC,CAAC,GAAG,CAAC,YAAY,CAAC,CAAC;QAClD,CAAC;KAAA;IAEY,cAAc,CACvB,KAAa,EACb,KAAa,EACb,OAA+B;;;YAE/B,MAAM,GAAG,GAAQ;gBACb,SAAS,EAAE,KAAK;gBAChB,KAAK;gBACL,YAAY,EAAE,MAAA,OAAO,aAAP,OAAO,uBAAP,OAAO,CAAE,YAAY,mCAAI,EAAE;gBACzC,SAAS,EAAE,MAAA,OAAO,aAAP,OAAO,uBAAP,OAAO,CAAE,SAAS,mCAAI,EAAE;gBACnC,OAAO,EAAE,MAAA,OAAO,aAAP,OAAO,uBAAP,OAAO,CAAE,OAAO,mCAAI,KAAK;aACrC,CAAC;YACF,IAAI,OAAO,aAAP,OAAO,uBAAP,OAAO,CAAE,MAAM;gBAAE,GAAG,CAAC,MAAM,GAAG,EAAE,UAAU,EAAE,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,MAAM,CAAC,EAAE,CAAC;YACjF,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,gBAAgB,EAAE,GAAG,CAAC,CAAC;YACtD,OAAO,CAAC,MAAA,IAAI,CAAC,OAAO,mCAAI,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,GAAQ,EAAE,EAAE;;gBAAC,OAAA,CAAC;oBAC3C,GAAG,EAAE,GAAG,CAAC,GAAG;oBACZ,UAAU,EAAE,GAAG,CAAC,UAAU;oBAC1B,KAAK,EAAE,GAAG,CAAC,KAAK;oBAChB,MAAM,EAAE,CAAC,MAAA,GAAG,CAAC,MAAM,mCAAI,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,CAAM,EAAE,EAAE,CAAC,CAAC;wBACxC,IAAI,EAAE,CAAC,CAAC,IAAI;wBACZ,KAAK,EAAE,CAAC,CAAC,KAAK;wBACd,UAAU,EAAE,CAAC,CAAC,UAAU;qBAC3B,CAAC,CAAC;iBACN,CAAC,CAAA;aAAA,CAAC,CAAC;QACR,CAAC;KAAA;IAED,wDAAwD;IAE3C,cAAc,CACvB,KAAa,EACb,GAAW,EACX,IAAY,EACZ,OAGC;;;YAED,MAAM,GAAG,GAAQ;gBACb,SAAS,EAAE,KAAK;gBAChB,GAAG;gBACH,IAAI;gBACJ,OAAO,EAAE,MAAA,OAAO,aAAP,OAAO,uBAAP,OAAO,CAAE,OAAO,mCAAI,EAAE;aAClC,CAAC;YACF,IAAI,OAAO,aAAP,OAAO,uBAAP,OAAO,CAAE,QAAQ;gBAAE,GAAG,CAAC,QAAQ,GAAG,eAAe,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;YACxE,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,gBAAgB,EAAE,GAAG,CAAC,CAAC;YACtD,OAAO,IAAI,CAAC,UAAU,CAAC;QAC3B,CAAC;KAAA;IAEM,cAAc,CAAC,KAAa,EAAE,GAAW;QAC5C,OAAO,IAAI,CAAC,MAAM,CAAC,gBAAgB,EAAE,EAAE,SAAS,EAAE,KAAK,EAAE,GAAG,EAAE,CAAC,CAAC;IACpE,CAAC;IAEY,aAAa,CACtB,KAAa;;;YAEb,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,eAAe,EAAE,EAAE,SAAS,EAAE,KAAK,EAAE,CAAC,CAAC;YACtE,OAAO,CAAC,MAAA,IAAI,CAAC,SAAS,mCAAI,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,CAAM,EAAE,EAAE,CAAC,CAAC;gBAC3C,GAAG,EAAE,CAAC,CAAC,GAAG;gBACV,UAAU,EAAE,CAAC,CAAC,UAAU;aAC3B,CAAC,CAAC,CAAC;QACR,CAAC;KAAA;IAED,wDAAwD;IAE3C,aAAa,CAAC,KAAa;;YACpC,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,eAAe,EAAE,EAAE,SAAS,EAAE,KAAK,EAAE,CAAC,CAAC;YACtE,OAAO;gBACH,OAAO,EAAE,IAAI,CAAC,OAAO;gBACrB,MAAM,EAAE,IAAI,CAAC,MAAM;gBACnB,SAAS,EAAE,IAAI,CAAC,SAAS;gBACzB,mBAAmB,EAAE,IAAI,CAAC,mBAAmB;aAChD,CAAC;QACN,CAAC;KAAA;IAEY,eAAe,CAAC,KAAa;;;YACtC,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,iBAAiB,EAAE,EAAE,SAAS,EAAE,KAAK,EAAE,CAAC,CAAC;YACxE,OAAO;gBACH,OAAO,EAAE,IAAI,CAAC,OAAO;gBACrB,aAAa,EAAE,IAAI,CAAC,aAAa;gBACjC,UAAU,EAAE,IAAI,CAAC,UAAU;gBAC3B,cAAc,EAAE,MAAA,IAAI,CAAC,cAAc,mCAAI,EAAE;aAC5C,CAAC;QACN,CAAC;KAAA;IAED,wDAAwD;IAE3C,WAAW;;;YACpB,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,aAAa,EAAE,EAAE,CAAC,CAAC;YAClD,OAAO;gBACH,MAAM,EAAE,IAAI,CAAC,MAAM;gBACnB,aAAa,EAAE,IAAI,CAAC,aAAa;gBACjC,aAAa,EAAE,MAAA,IAAI,CAAC,aAAa,mCAAI,EAAE;aAC1C,CAAC;QACN,CAAC;KAAA;IAEM,QAAQ;QACX,OAAO,IAAI,CAAC,MAAM,CAAC,UAAU,EAAE,EAAE,CAAC,CAAC;IACvC,CAAC;IAED,wDAAwD;IAEhD,MAAM,CAAC,MAAc,EAAE,OAAY;QACvC,OAAO,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;YACnC,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,OAAO,EAAE,CAAC,GAA6B,EAAE,QAAa,EAAE,EAAE;gBAC3E,IAAI,GAAG;oBAAE,MAAM,CAAC,GAAG,CAAC,CAAC;;oBAChB,OAAO,CAAC,QAAQ,CAAC,CAAC;YAC3B,CAAC,CAAC,CAAC;QACP,CAAC,CAAC,CAAC;IACP,CAAC;CACJ;AA7OD,oCA6OC;AAED,4DAA4D;AAE5D,SAAS,eAAe,CACpB,QAAmD;IAEnD,MAAM,MAAM,GAAkC,EAAE,CAAC;IACjD,KAAK,MAAM,CAAC,GAAG,EAAE,KAAK,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,QAAQ,CAAC,EAAE,CAAC;QAClD,IAAI,OAAO,KAAK,KAAK,SAAS,EAAE,CAAC;YAC7B,MAAM,CAAC,GAAG,CAAC,GAAG,EAAE,SAAS,EAAE,KAAK,EAAE,CAAC;QACvC,CAAC;aAAM,IAAI,OAAO,KAAK,KAAK,QAAQ,EAAE,CAAC;YACnC,MAAM,CAAC,GAAG,CAAC,GAAG,EAAE,WAAW,EAAE,KAAK,EAAE,CAAC;QACzC,CAAC;aAAM,CAAC;YACJ,MAAM,CAAC,GAAG,CAAC,GAAG,EAAE,WAAW,EAAE,MAAM,CAAC,KAAK,CAAC,EAAE,CAAC;QACjD,CAAC;IACL,CAAC;IACD,OAAO,MAAM,CAAC;AAClB,CAAC;AAED,SAAS,YAAY,CAAC,IAAS;;IAC3B,MAAM,QAAQ,GAA8C,EAAE,CAAC;IAC/D,IAAI,IAAI,CAAC,QAAQ,EAAE,CAAC;QAChB,KAAK,MAAM,CAAC,GAAG,EAAE,GAAG,CAAC,IAAI,MAAM,CAAC,OAAO,CAAM,IAAI,CAAC,QAAQ,CAAC,EAAE,CAAC;YAC1D,IAAI,GAAG,CAAC,WAAW,KAAK,SAAS,IAAI,GAAG,CAAC,WAAW,KAAK,EAAE,EAAE,CAAC;gBAC1D,QAAQ,CAAC,GAAG,CAAC,GAAG,GAAG,CAAC,WAAW,CAAC;YACpC,CAAC;iBAAM,IAAI,GAAG,CAAC,WAAW,KAAK,SAAS,IAAI,GAAG,CAAC,WAAW,KAAK,CAAC,EAAE,CAAC;gBAChE,QAAQ,CAAC,GAAG,CAAC,GAAG,GAAG,CAAC,WAAW,CAAC;YACpC,CAAC;iBAAM,IAAI,GAAG,CAAC,SAAS,KAAK,SAAS,EAAE,CAAC;gBACrC,QAAQ,CAAC,GAAG,CAAC,GAAG,GAAG,CAAC,SAAS,CAAC;YAClC,CAAC;QACL,CAAC;IACL,CAAC;IACD,OAAO;QACH,EAAE,EAAE,IAAI,CAAC,EAAE;QACX,QAAQ;QACR,MAAM,EAAE,MAAA,IAAI,CAAC,MAAM,mCAAI,EAAE;QACzB,IAAI,EAAE,IAAI,CAAC,IAAI;QACf,KAAK,EAAE,IAAI,CAAC,KAAK;KACpB,CAAC;AACN,CAAC"}

package/lib/types.d.ts

@@ -0,0 +1,153 @@
+/**
+ * An AI model that can be used to create embeddings.
+ */
+export interface EmbeddingsModel {
+    /**
+     * Maximum number of tokens
+     */
+    readonly maxTokens: number;
+    /**
+     * Creates embeddings for the given inputs.
+     * @param inputs Text inputs to create embeddings for.
+     * @returns A `EmbeddingsResponse` with a status and the generated embeddings or a message when an error occurs.
+     */
+    createEmbeddings(inputs: string | string[]): Promise<EmbeddingsResponse>;
+}
+/**
+ * Status of the embeddings response.
+ * @remarks
+ * `success` - The embeddings were successfully created.
+ * `error` - An error occurred while creating the embeddings.
+ * `rate_limited` - The request was rate limited.
+ */
+export type EmbeddingsResponseStatus = 'success' | 'error' | 'rate_limited' | 'cancelled';
+/**
+ * Response returned by a `EmbeddingsClient`.
+ */
+export interface EmbeddingsResponse {
+    /**
+     * Status of the embeddings response.
+     */
+    status: EmbeddingsResponseStatus;
+    /**
+     * Optional. Embeddings for the given inputs.
+     */
+    output?: number[][];
+    /**
+     * Optional. Message when status is not equal to `success`.
+     */
+    message?: string;
+    /**
+     * Optional. Model used to create the embeddings.
+     */
+    model?: string;
+    /**
+     * Optional. Usage statistics for the request.
+     */
+    usage?: Record<string, any>;
+}
+export interface TextChunk {
+    text: string;
+    tokens: number[];
+    startPos: number;
+    endPos: number;
+    startOverlap: number[];
+    endOverlap: number[];
+}
+export interface TextFetcher {
+    fetch(uri: string, onDocument: (uri: string, text: string, docType?: string) => Promise<boolean>): Promise<boolean>;
+}
+export interface IndexStats {
+    version: number;
+    metadata_config: {
+        indexed?: string[];
+    };
+    items: number;
+}
+export interface IndexItem<TMetadata = Record<string, MetadataTypes>> {
+    id: string;
+    metadata: TMetadata;
+    vector: number[];
+    norm: number;
+    metadataFile?: string;
+}
+export interface MetadataFilter {
+    /**
+     * Equal to (number, string, boolean)
+     */
+    '$eq'?: number | string | boolean;
+    /**
+     * Not equal to (number, string, boolean)
+     */
+    '$ne'?: number | string | boolean;
+    /**
+     * Greater than (number)
+     */
+    '$gt'?: number;
+    /**
+     * Greater than or equal to (number)
+     */
+    '$gte'?: number;
+    /**
+     * Less than (number)
+     */
+    '$lt'?: number;
+    /**
+     * Less than or equal to (number)
+     */
+    '$lte'?: number;
+    /**
+     * In array (string or number)
+     */
+    '$in'?: (number | string)[];
+    /**
+     * Not in array (string or number)
+     */
+    '$nin'?: (number | string)[];
+    /**
+     * AND (MetadataFilter[])
+     */
+    '$and'?: MetadataFilter[];
+    /**
+     * OR (MetadataFilter[])
+     */
+    '$or'?: MetadataFilter[];
+    [key: string]: unknown;
+}
+export type MetadataTypes = number | string | boolean;
+export interface QueryResult<TMetadata = Record<string, MetadataTypes>> {
+    item: IndexItem<TMetadata>;
+    score: number;
+}
+export interface Tokenizer {
+    decode(tokens: number[]): string;
+    encode(text: string): number[];
+}
+export interface DocumentChunkMetadata {
+    documentId: string;
+    startPos: number;
+    endPos: number;
+    [key: string]: MetadataTypes;
+}
+export interface DocumentCatalogStats {
+    version: number;
+    documents: number;
+    chunks: number;
+    metadata_config: {
+        indexed?: string[];
+    };
+}
+export interface DocumentTextSection {
+    text: string;
+    tokenCount: number;
+    score: number;
+    isBm25: boolean;
+}
+export interface IndexData {
+    version: number;
+    metadata_config: {
+        indexed?: string[];
+    };
+    items: IndexItem[];
+}
+//# sourceMappingURL=types.d.ts.map

package/lib/types.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"types.d.ts","sourceRoot":"","sources":["../src/types.ts"],"names":[],"mappings":"AAEA;;GAEG;AACH,MAAM,WAAW,eAAe;IAC9B;;OAEG;IACH,QAAQ,CAAC,SAAS,EAAE,MAAM,CAAC;IAE3B;;;;OAIG;IACH,gBAAgB,CAAC,MAAM,EAAE,MAAM,GAAG,MAAM,EAAE,GAAG,OAAO,CAAC,kBAAkB,CAAC,CAAC;CAC1E;AAED;;;;;;GAMG;AACH,MAAM,MAAM,wBAAwB,GAAG,SAAS,GAAG,OAAO,GAAG,cAAc,GAAG,WAAW,CAAC;AAE1F;;GAEG;AACH,MAAM,WAAW,kBAAkB;IACjC;;OAEG;IACH,MAAM,EAAE,wBAAwB,CAAC;IAEjC;;OAEG;IACH,MAAM,CAAC,EAAE,MAAM,EAAE,EAAE,CAAC;IAEpB;;OAEG;IACH,OAAO,CAAC,EAAE,MAAM,CAAC;IAEjB;;OAEG;IACH,KAAK,CAAC,EAAE,MAAM,CAAC;IAEf;;OAEG;IACH,KAAK,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;CAC7B;AAED,MAAM,WAAW,SAAS;IACxB,IAAI,EAAE,MAAM,CAAC;IACb,MAAM,EAAE,MAAM,EAAE,CAAC;IACjB,QAAQ,EAAE,MAAM,CAAC;IACjB,MAAM,EAAE,MAAM,CAAC;IACf,YAAY,EAAE,MAAM,EAAE,CAAC;IACvB,UAAU,EAAE,MAAM,EAAE,CAAC;CACtB;AAED,MAAM,WAAW,WAAW;IAC1B,KAAK,CAAC,GAAG,EAAE,MAAM,EAAE,UAAU,EAAE,CAAC,GAAG,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,CAAC,EAAE,MAAM,KAAK,OAAO,CAAC,OAAO,CAAC,GAAG,OAAO,CAAC,OAAO,CAAC,CAAC;CACrH;AAED,MAAM,WAAW,UAAU;IACzB,OAAO,EAAE,MAAM,CAAC;IAChB,eAAe,EAAE;QACf,OAAO,CAAC,EAAE,MAAM,EAAE,CAAC;KACpB,CAAC;IACF,KAAK,EAAE,MAAM,CAAC;CACf;AAED,MAAM,WAAW,SAAS,CAAC,SAAS,GAAG,MAAM,CAAC,MAAM,EAAE,aAAa,CAAC;IAClE,EAAE,EAAE,MAAM,CAAC;IACX,QAAQ,EAAE,SAAS,CAAC;IACpB,MAAM,EAAE,MAAM,EAAE,CAAC;IACjB,IAAI,EAAE,MAAM,CAAC;IACb,YAAY,CAAC,EAAE,MAAM,CAAC;CACvB;AAED,MAAM,WAAW,cAAc;IAE7B;;OAEG;IACH,KAAK,CAAC,EAAE,MAAM,GAAG,MAAM,GAAG,OAAO,CAAC;IAElC;;OAEG;IACH,KAAK,CAAC,EAAE,MAAM,GAAG,MAAM,GAAG,OAAO,CAAC;IAElC;;OAEG;IACH,KAAK,CAAC,EAAE,MAAM,CAAC;IAEf;;OAEG;IACH,MAAM,CAAC,EAAE,MAAM,CAAC;IAEhB;;OAEG;IACH,KAAK,CAAC,EAAE,MAAM,CAAC;IAEf;;OAEG;IACH,MAAM,CAAC,EAAE,MAAM,CAAC;IAEhB;;OAEG;IACH,KAAK,CAAC,EAAE,CAAC,MAAM,GAAG,MAAM,CAAC,EAAE,CAAC;IAE5B;;OAEG;IACH,MAAM,CAAC,EAAE,CAAC,MAAM,GAAG,MAAM,CAAC,EAAE,CAAC;IAE7B;;OAEG;IACH,MAAM,CAAC,EAAE,cAAc,EAAE,CAAC;IAE1B;;OAEG;IACH,KAAK,CAAC,EAAE,cAAc,EAAE,CAAC;IAEzB,CAAC,GAAG,EAAE,MAAM,GAAG,OAAO,CAAC;CACxB;AAED,MAAM,MAAM,aAAa,GAAG,MAAM,GAAG,MAAM,GAAG,OAAO,CAAC;AAEtD,MAAM,WAAW,WAAW,CAAC,SAAS,GAAG,MAAM,CAAC,MAAM,EAAE,aAAa,CAAC;IACpE,IAAI,EAAE,SAAS,CAAC,SAAS,CAAC,CAAC;IAC3B,KAAK,EAAE,MAAM,CAAC;CACf;AAED,MAAM,WAAW,SAAS;IACxB,MAAM,CAAC,MAAM,EAAE,MAAM,EAAE,GAAG,MAAM,CAAC;IACjC,MAAM,CAAC,IAAI,EAAE,MAAM,GAAG,MAAM,EAAE,CAAC;CAChC;AAED,MAAM,WAAW,qBAAqB;IACpC,UAAU,EAAE,MAAM,CAAC;IACnB,QAAQ,EAAE,MAAM,CAAC;IACjB,MAAM,EAAE,MAAM,CAAC;IACf,CAAC,GAAG,EAAE,MAAM,GAAG,aAAa,CAAC;CAC9B;AAED,MAAM,WAAW,oBAAoB;IACnC,OAAO,EAAE,MAAM,CAAC;IAChB,SAAS,EAAE,MAAM,CAAC;IAClB,MAAM,EAAE,MAAM,CAAC;IACf,eAAe,EAAE;QACf,OAAO,CAAC,EAAE,MAAM,EAAE,CAAC;KACpB,CAAC;CACH;AAED,MAAM,WAAW,mBAAmB;IAClC,IAAI,EAAE,MAAM,CAAC;IACb,UAAU,EAAE,MAAM,CAAC;IACnB,KAAK,EAAE,MAAM,CAAC;IACd,MAAM,EAAE,OAAO,CAAC;CACjB;AAED,MAAM,WAAW,SAAS;IACtB,OAAO,EAAE,MAAM,CAAC;IAChB,eAAe,EAAE;QACb,OAAO,CAAC,EAAE,MAAM,EAAE,CAAC;KACtB,CAAC;IACF,KAAK,EAAE,SAAS,EAAE,CAAC;CACtB"}

package/lib/types.js

@@ -0,0 +1,3 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+//# sourceMappingURL=types.js.map

package/lib/types.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"types.js","sourceRoot":"","sources":["../src/types.ts"],"names":[],"mappings":""}

package/lib/utils/index.d.ts

@@ -0,0 +1,2 @@
+export * from './pathUtils';
+//# sourceMappingURL=index.d.ts.map

package/lib/utils/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/utils/index.ts"],"names":[],"mappings":"AAAA,cAAc,aAAa,CAAC"}

package/lib/utils/index.js

@@ -0,0 +1,18 @@
+"use strict";
+var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    var desc = Object.getOwnPropertyDescriptor(m, k);
+    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
+      desc = { enumerable: true, get: function() { return m[k]; } };
+    }
+    Object.defineProperty(o, k2, desc);
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+var __exportStar = (this && this.__exportStar) || function(m, exports) {
+    for (var p in m) if (p !== "default" && !Object.prototype.hasOwnProperty.call(exports, p)) __createBinding(exports, m, p);
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+__exportStar(require("./pathUtils"), exports);
+//# sourceMappingURL=index.js.map

package/lib/utils/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/utils/index.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;AAAA,8CAA4B"}

package/lib/utils/pathUtils.d.ts

@@ -0,0 +1,40 @@
+/**
+ * Universal path utilities that work in Node, Browser, and Electron.
+ * Replaces Node's 'path' module for cross-platform compatibility.
+ */
+export declare const pathUtils: {
+    /**
+     * Path separator. Always '/' for consistency across platforms.
+     */
+    sep: "/";
+    /**
+     * Join path segments together.
+     */
+    join(...parts: string[]): string;
+    /**
+     * Get the last portion of a path.
+     */
+    basename(filePath: string, ext?: string): string;
+    /**
+     * Get the directory name of a path.
+     */
+    dirname(filePath: string): string;
+    /**
+     * Get the extension of the path.
+     */
+    extname(filePath: string): string;
+    /**
+     * Normalize a path, resolving '..' and '.' segments.
+     */
+    normalize(filePath: string): string;
+    /**
+     * Determine if a path is absolute.
+     */
+    isAbsolute(filePath: string): boolean;
+    /**
+     * Get the relative path from one path to another.
+     */
+    relative(from: string, to: string): string;
+};
+export default pathUtils;
+//# sourceMappingURL=pathUtils.d.ts.map

package/lib/utils/pathUtils.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"pathUtils.d.ts","sourceRoot":"","sources":["../../src/utils/pathUtils.ts"],"names":[],"mappings":"AAAA;;;GAGG;AACH,eAAO,MAAM,SAAS;IAClB;;OAEG;;IAEH;;OAEG;mBACY,MAAM,EAAE,GAAG,MAAM;IAUhC;;OAEG;uBACgB,MAAM,QAAQ,MAAM,GAAG,MAAM;IAQhD;;OAEG;sBACe,MAAM,GAAG,MAAM;IAMjC;;OAEG;sBACe,MAAM,GAAG,MAAM;IAMjC;;OAEG;wBACiB,MAAM,GAAG,MAAM;IAgBnC;;OAEG;yBACkB,MAAM,GAAG,OAAO;IAIrC;;OAEG;mBACY,MAAM,MAAM,MAAM,GAAG,MAAM;CAuB7C,CAAC;AAEF,eAAe,SAAS,CAAC"}