vectra 0.12.2 → 0.12.3

package/lib/LocalIndex.spec.js

@@ -13,33 +13,166 @@ var __importDefault = (this && this.__importDefault) || function (mod) {
 };
 Object.defineProperty(exports, "__esModule", { value: true });
 const node_assert_1 = __importDefault(require("node:assert"));
+const sinon_1 = __importDefault(require("sinon"));
 const LocalIndex_1 = require("./LocalIndex");
 const promises_1 = __importDefault(require("fs/promises"));
 const path_1 = __importDefault(require("path"));
 describe('LocalIndex', () => {
     const testIndexDir = path_1.default.join(__dirname, 'test_index');
+    const basicIndexItems = [
+        { id: '1', vector: [1, 2, 3] },
+        { id: '2', vector: [2, 3, 4] },
+        { id: '3', vector: [3, 4, 5] }
+    ];
     beforeEach(() => __awaiter(void 0, void 0, void 0, function* () {
         yield promises_1.default.rm(testIndexDir, { recursive: true, force: true });
     }));
     afterEach(() => __awaiter(void 0, void 0, void 0, function* () {
         yield promises_1.default.rm(testIndexDir, { recursive: true, force: true });
+        sinon_1.default.restore();
     }));
     it('should create a new index', () => __awaiter(void 0, void 0, void 0, function* () {
         const index = new LocalIndex_1.LocalIndex(testIndexDir);
         yield index.createIndex();
         const created = yield index.isIndexCreated();
         node_assert_1.default.equal(created, true);
+        node_assert_1.default.equal(index.folderPath, testIndexDir);
     }));
+    it('blocks concurrent operations when lock is held', () => __awaiter(void 0, void 0, void 0, function* () {
+        const index = new LocalIndex_1.LocalIndex(testIndexDir);
+        yield index.createIndex();
+        yield index.beginUpdate(); // grab lock for a big update!
+        yield node_assert_1.default.rejects(() => __awaiter(void 0, void 0, void 0, function* () {
+            yield index.beginUpdate(); // try to grab lock again. should fail!
+        }), new Error('Update already in progress'));
+    }));
+    describe('createIndex', () => {
+        it('checks for existing index on creation', () => __awaiter(void 0, void 0, void 0, function* () {
+            const index = new LocalIndex_1.LocalIndex(testIndexDir);
+            yield index.createIndex(); // create first index.json
+            // create without deleteIfExists. Will reject
+            yield node_assert_1.default.rejects(() => __awaiter(void 0, void 0, void 0, function* () {
+                yield index.createIndex();
+            }), new Error('Index already exists'));
+            // create with deleteIfExists. Should remove old data
+            yield index.insertItem({ id: '1', vector: [1, 2, 3] });
+            const lengthBefore = (yield index.listItems()).length;
+            node_assert_1.default.equal(lengthBefore, 1);
+            yield index.createIndex({ deleteIfExists: true, version: 2, metadata_config: {} });
+            const lengthAfter = (yield index.listItems()).length;
+            node_assert_1.default.equal(lengthAfter, 0);
+        }));
+        it('delete index if file creation fails', () => __awaiter(void 0, void 0, void 0, function* () {
+            const index = new LocalIndex_1.LocalIndex(testIndexDir);
+            sinon_1.default.stub(promises_1.default, 'writeFile').rejects(new Error('fs error'));
+            yield node_assert_1.default.rejects(() => __awaiter(void 0, void 0, void 0, function* () {
+                yield index.createIndex();
+            }), new Error('Error creating index'));
+            yield node_assert_1.default.rejects(() => __awaiter(void 0, void 0, void 0, function* () {
+                yield index.listItems();
+            }));
+        }));
+    });
+    describe('deleteItem', () => {
+        it('does nothing when id not found', () => __awaiter(void 0, void 0, void 0, function* () {
+            const index = new LocalIndex_1.LocalIndex(testIndexDir);
+            yield index.createIndex();
+            yield index.beginUpdate();
+            yield index.insertItem(basicIndexItems[0]);
+            yield index.insertItem(basicIndexItems[1]);
+            yield index.insertItem(basicIndexItems[2]);
+            yield index.endUpdate();
+            yield node_assert_1.default.doesNotReject(() => __awaiter(void 0, void 0, void 0, function* () {
+                yield index.deleteItem('dne');
+            }));
+            node_assert_1.default.equal((yield index.listItems()).length, 3);
+        }));
+        it('leaves existing empty index when last el deleted', () => __awaiter(void 0, void 0, void 0, function* () {
+            var _a;
+            const index = new LocalIndex_1.LocalIndex(testIndexDir);
+            yield index.createIndex();
+            yield index.insertItem(basicIndexItems[0]);
+            yield index.deleteItem((_a = basicIndexItems[0].id) !== null && _a !== void 0 ? _a : '');
+            node_assert_1.default.equal(yield index.isIndexCreated(), true);
+            node_assert_1.default.equal((yield index.listItems()).length, 0);
+        }));
+        it('removes elements from any position', () => __awaiter(void 0, void 0, void 0, function* () {
+            const index = new LocalIndex_1.LocalIndex(testIndexDir);
+            yield index.createIndex();
+            yield index.batchInsertItems([
+                { id: '1', vector: [] },
+                { id: '2', vector: [] },
+                { id: '3', vector: [] },
+                { id: '4', vector: [] },
+                { id: '5', vector: [] },
+            ]);
+            yield index.beginUpdate();
+            yield index.deleteItem('1');
+            yield index.deleteItem('3');
+            yield index.deleteItem('5');
+            yield index.endUpdate();
+            node_assert_1.default.deepStrictEqual(yield index.listItems(), [{ id: '2', vector: [], metadata: {}, norm: 0 }, { id: '4', vector: [], metadata: {}, norm: 0 }]);
+        }));
+    });
+    describe('endUpdate', () => {
+        it('throws an error if no update has begun', () => __awaiter(void 0, void 0, void 0, function* () {
+            const index = new LocalIndex_1.LocalIndex(testIndexDir);
+            yield node_assert_1.default.rejects(() => __awaiter(void 0, void 0, void 0, function* () {
+                yield index.endUpdate();
+            }), new Error('No update in progress'));
+        }));
+        it('throws an error if the index could not be saved', () => __awaiter(void 0, void 0, void 0, function* () {
+            const index = new LocalIndex_1.LocalIndex(testIndexDir, 'index.json');
+            yield index.createIndex();
+            yield index.beginUpdate();
+            sinon_1.default.stub(promises_1.default, 'writeFile').rejects(new Error('fs error'));
+            yield node_assert_1.default.rejects(() => __awaiter(void 0, void 0, void 0, function* () {
+                yield index.endUpdate();
+            }), new Error('Error saving index: Error: fs error'));
+        }));
+    });
+    describe('getIndexStats', () => {
+        it('reports empty index correctly', () => __awaiter(void 0, void 0, void 0, function* () {
+            const index = new LocalIndex_1.LocalIndex(testIndexDir);
+            yield index.createIndex();
+            node_assert_1.default.deepStrictEqual(yield index.getIndexStats(), {
+                version: 1,
+                metadata_config: {},
+                items: 0
+            });
+        }));
+        it('correctly reports non-empty index stats', () => __awaiter(void 0, void 0, void 0, function* () {
+            const index = new LocalIndex_1.LocalIndex(testIndexDir);
+            yield index.createIndex({ version: 1, metadata_config: { indexed: [] } });
+            yield index.batchInsertItems(basicIndexItems);
+            node_assert_1.default.deepStrictEqual(yield index.getIndexStats(), {
+                version: 1,
+                metadata_config: { indexed: [] },
+                items: 3
+            });
+        }));
+    });
+    describe('getItem', () => {
+        it('returns undefined when item not found', () => __awaiter(void 0, void 0, void 0, function* () {
+            const index = new LocalIndex_1.LocalIndex(testIndexDir);
+            yield index.createIndex();
+            node_assert_1.default.equal(yield index.getItem('1'), undefined);
+        }));
+        it('returns requested item', () => __awaiter(void 0, void 0, void 0, function* () {
+            const index = new LocalIndex_1.LocalIndex(testIndexDir);
+            yield index.createIndex();
+            yield index.batchInsertItems(basicIndexItems);
+            const item2 = yield index.getItem('2');
+            node_assert_1.default.equal(item2 === null || item2 === void 0 ? void 0 : item2.id, basicIndexItems[1].id);
+            node_assert_1.default.equal(item2 === null || item2 === void 0 ? void 0 : item2.vector, basicIndexItems[1].vector);
+            node_assert_1.default.equal((yield index.listItems()).length, 3);
+        }));
+    });
     describe('batchInsertItems', () => {
-        const indexItems = [
-            { id: '1', vector: [1, 2, 3] },
-            { id: '2', vector: [2, 3, 4] },
-            { id: '3', vector: [3, 4, 5] }
-        ];
         it('should insert provided items', () => __awaiter(void 0, void 0, void 0, function* () {
             const index = new LocalIndex_1.LocalIndex(testIndexDir);
             yield index.createIndex();
-            const newItems = yield index.batchInsertItems(indexItems);
+            const newItems = yield index.batchInsertItems(basicIndexItems);
             node_assert_1.default.equal(newItems.length, 3);
             const retrievedItems = yield index.listItems();
             node_assert_1.default.equal(retrievedItems.length, 3);
@@ -50,7 +183,7 @@ describe('LocalIndex', () => {
             yield index.insertItem({ id: '2', vector: [9, 9, 9] });
             // ensures insert error is bubbled up to batchIndexItems caller
             yield node_assert_1.default.rejects(() => __awaiter(void 0, void 0, void 0, function* () {
-                yield index.batchInsertItems(indexItems);
+                yield index.batchInsertItems(basicIndexItems);
             }), {
                 name: 'Error',
                 message: 'Item with id 2 already exists'
@@ -60,5 +193,83 @@ describe('LocalIndex', () => {
             node_assert_1.default.equal(storedItems.length, 1);
         }));
     });
+    describe('listItemsByMetadata', () => {
+        it('returns items matching metadata filter', () => __awaiter(void 0, void 0, void 0, function* () {
+            const index = new LocalIndex_1.LocalIndex(testIndexDir);
+            yield index.createIndex();
+            yield index.batchInsertItems([
+                { id: '1', vector: [], metadata: { category: 'food' } },
+                { id: '2', vector: [], metadata: { category: 'food' } },
+                { id: '3', vector: [], metadata: { category: 'electronics' } },
+                { id: '4', vector: [], metadata: { category: 'drink' } },
+                { id: '5', vector: [], metadata: { category: 'food' } },
+            ]);
+            const foodItems = yield index.listItemsByMetadata({ category: { '$eq': 'food' } });
+            node_assert_1.default.deepStrictEqual(foodItems.map((item) => item.id), ["1", "2", "5"]);
+            const drinkItems = yield index.listItemsByMetadata({ category: { '$eq': 'drink' } });
+            node_assert_1.default.deepStrictEqual(drinkItems.map((item) => item.id), ["4"]);
+            const clothingItems = yield index.listItemsByMetadata({ category: { '$eq': 'clothes' } });
+            node_assert_1.default.deepStrictEqual(clothingItems, []);
+        }));
+        it('returns nothing when no items in index', () => __awaiter(void 0, void 0, void 0, function* () {
+            const index = new LocalIndex_1.LocalIndex(testIndexDir);
+            yield index.createIndex();
+            const items = yield index.listItemsByMetadata({});
+            node_assert_1.default.deepStrictEqual(items, []);
+        }));
+    });
+    describe("queryItems", () => {
+        it("returns empty array on empty index search", () => __awaiter(void 0, void 0, void 0, function* () {
+            const index = new LocalIndex_1.LocalIndex(testIndexDir);
+            yield index.createIndex();
+            const result = yield index.queryItems([1, 2, 3], "", 10);
+            node_assert_1.default.deepStrictEqual(result, []);
+        }));
+        it("returns bad match when no better match exists", () => __awaiter(void 0, void 0, void 0, function* () {
+            var _a, _b;
+            const index = new LocalIndex_1.LocalIndex(testIndexDir);
+            yield index.createIndex();
+            yield index.insertItem({ id: "1", vector: [0.9, 0, 0, 0, 0] });
+            const result = yield index.queryItems([0, 0, 0, 0, 0.1], "", 1);
+            node_assert_1.default.equal((_a = result[0]) === null || _a === void 0 ? void 0 : _a.score, 0);
+            node_assert_1.default.equal((_b = result[0]) === null || _b === void 0 ? void 0 : _b.item.id, "1");
+        }));
+        it("returns all vectors when fewer than topK exist", () => __awaiter(void 0, void 0, void 0, function* () {
+            const index = new LocalIndex_1.LocalIndex(testIndexDir);
+            yield index.createIndex();
+            yield index.batchInsertItems(basicIndexItems);
+            const result = yield index.queryItems([0, 0, 1], "", 10);
+            node_assert_1.default.equal(result.length, 3);
+            node_assert_1.default.deepStrictEqual(result.map(({ item }) => item.id), basicIndexItems.map((item) => item.id));
+        }));
+        it("filters by metadata when filter provided", () => __awaiter(void 0, void 0, void 0, function* () {
+            const index = new LocalIndex_1.LocalIndex(testIndexDir);
+            yield index.createIndex();
+            yield index.batchInsertItems([
+                { id: "1", vector: [1, 0, 0], metadata: { category: "food" } },
+                { id: "2", vector: [0, 0, 1], metadata: { category: "drink" } },
+            ]);
+            const bestGeneralMatch = yield index.queryItems([1, 0, 0], "", 1);
+            const bestDrinkMatch = yield index.queryItems([1, 0, 0], "", 1, {
+                category: { $eq: "drink" },
+            });
+            node_assert_1.default.equal(bestGeneralMatch[0].item.id, "1");
+            node_assert_1.default.equal(bestDrinkMatch[0].item.id, "2");
+        }));
+        it("reads item metadata file when provided", () => __awaiter(void 0, void 0, void 0, function* () {
+            const index = new LocalIndex_1.LocalIndex(testIndexDir);
+            yield index.createIndex({ version: 1, metadata_config: { indexed: ['category'] } });
+            yield index.batchInsertItems([
+                { id: "1", vector: [1, 0, 0] },
+                { id: "2", vector: [0, 0, 1], metadata: { category: 'drink' } },
+            ]);
+            sinon_1.default
+                .stub(promises_1.default, "readFile")
+                .resolves(JSON.stringify({ category: "drink" }));
+            const bestDrinkMatch = yield index.queryItems([1, 0, 0], "", 2, { category: { '$eq': 'drink' } });
+            node_assert_1.default.notEqual(bestDrinkMatch[0].item.metadataFile, undefined);
+            node_assert_1.default.equal(bestDrinkMatch[0].item.id, "2");
+        }));
+    });
 });
 //# sourceMappingURL=LocalIndex.spec.js.map

package/lib/LocalIndex.spec.js.map

@@ -1 +1 @@
-{"version":3,"file":"LocalIndex.spec.js","sourceRoot":"","sources":["../src/LocalIndex.spec.ts"],"names":[],"mappings":";;;;;;;;;;;;;;AAAA,8DAAgC;AAChC,6CAAyC;AAEzC,2DAA4B;AAC5B,gDAAuB;AAEvB,QAAQ,CAAC,YAAY,EAAE,GAAG,EAAE;IAC1B,MAAM,YAAY,GAAG,cAAI,CAAC,IAAI,CAAC,SAAS,EAAE,YAAY,CAAC,CAAC;IAExD,UAAU,CAAC,GAAS,EAAE;QACpB,MAAM,kBAAE,CAAC,EAAE,CAAC,YAAY,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;IAC9D,CAAC,CAAA,CAAC,CAAC;IAEH,SAAS,CAAC,GAAS,EAAE;QACnB,MAAM,kBAAE,CAAC,EAAE,CAAC,YAAY,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;IAC9D,CAAC,CAAA,CAAC,CAAC;IAEH,EAAE,CAAC,2BAA2B,EAAE,GAAS,EAAE;QACzC,MAAM,KAAK,GAAG,IAAI,uBAAU,CAAC,YAAY,CAAC,CAAC;QAC3C,MAAM,KAAK,CAAC,WAAW,EAAE,CAAC;QAC1B,MAAM,OAAO,GAAG,MAAM,KAAK,CAAC,cAAc,EAAE,CAAC;QAC7C,qBAAM,CAAC,KAAK,CAAC,OAAO,EAAE,IAAI,CAAC,CAAC;IAC9B,CAAC,CAAA,CAAC,CAAC;IAEH,QAAQ,CAAC,kBAAkB,EAAE,GAAG,EAAE;QAChC,MAAM,UAAU,GAAyB;YACvC,EAAE,EAAE,EAAE,GAAG,EAAE,MAAM,EAAE,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,EAAE;YAC9B,EAAE,EAAE,EAAE,GAAG,EAAE,MAAM,EAAE,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,EAAE;YAC9B,EAAE,EAAE,EAAE,GAAG,EAAE,MAAM,EAAE,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,EAAE;SAC/B,CAAC;QAEF,EAAE,CAAC,8BAA8B,EAAE,GAAS,EAAE;YAC5C,MAAM,KAAK,GAAG,IAAI,uBAAU,CAAC,YAAY,CAAC,CAAC;YAC3C,MAAM,KAAK,CAAC,WAAW,EAAE,CAAC;YAE1B,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,gBAAgB,CAAC,UAAU,CAAC,CAAC;YAE1D,qBAAM,CAAC,KAAK,CAAC,QAAQ,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC;YAEjC,MAAM,cAAc,GAAG,MAAM,KAAK,CAAC,SAAS,EAAE,CAAC;YAC/C,qBAAM,CAAC,KAAK,CAAC,cAAc,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC;QACzC,CAAC,CAAA,CAAC,CAAC;QAEH,EAAE,CAAC,yDAAyD,EAAE,GAAS,EAAE;YACvE,MAAM,KAAK,GAAG,IAAI,uBAAU,CAAC,YAAY,CAAC,CAAC;YAC3C,MAAM,KAAK,CAAC,WAAW,EAAE,CAAC;YAE1B,MAAM,KAAK,CAAC,UAAU,CAAC,EAAE,EAAE,EAAE,GAAG,EAAE,MAAM,EAAE,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC;YAEvD,+DAA+D;YAC/D,MAAM,qBAAM,CAAC,OAAO,CAClB,GAAS,EAAE;gBACT,MAAM,KAAK,CAAC,gBAAgB,CAAC,UAAU,CAAC,CAAC;YAC3C,CAAC,CAAA,EACD;gBACE,IAAI,EAAE,OAAO;gBACb,OAAO,EAAE,+BAA+B;aACzC,CACF,CAAC;YAEF,uCAAuC;YACvC,MAAM,WAAW,GAAG,MAAM,KAAK,CAAC,SAAS,EAAE,CAAC;YAC5C,qBAAM,CAAC,KAAK,CAAC,WAAW,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC;QACtC,CAAC,CAAA,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}
+{"version":3,"file":"LocalIndex.spec.js","sourceRoot":"","sources":["../src/LocalIndex.spec.ts"],"names":[],"mappings":";;;;;;;;;;;;;;AAAA,8DAAgC;AAChC,kDAAyB;AACzB,6CAAyC;AAEzC,2DAA4B;AAC5B,gDAAuB;AAEvB,QAAQ,CAAC,YAAY,EAAE,GAAG,EAAE;IAC1B,MAAM,YAAY,GAAG,cAAI,CAAC,IAAI,CAAC,SAAS,EAAE,YAAY,CAAC,CAAC;IAExD,MAAM,eAAe,GAAyB;QAC5C,EAAE,EAAE,EAAE,GAAG,EAAE,MAAM,EAAE,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,EAAE;QAC9B,EAAE,EAAE,EAAE,GAAG,EAAE,MAAM,EAAE,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,EAAE;QAC9B,EAAE,EAAE,EAAE,GAAG,EAAE,MAAM,EAAE,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,EAAE;KAC/B,CAAC;IAGF,UAAU,CAAC,GAAS,EAAE;QACpB,MAAM,kBAAE,CAAC,EAAE,CAAC,YAAY,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;IAC9D,CAAC,CAAA,CAAC,CAAC;IAEH,SAAS,CAAC,GAAS,EAAE;QACnB,MAAM,kBAAE,CAAC,EAAE,CAAC,YAAY,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;QAC5D,eAAK,CAAC,OAAO,EAAE,CAAC;IAClB,CAAC,CAAA,CAAC,CAAC;IAEH,EAAE,CAAC,2BAA2B,EAAE,GAAS,EAAE;QACzC,MAAM,KAAK,GAAG,IAAI,uBAAU,CAAC,YAAY,CAAC,CAAC;QAC3C,MAAM,KAAK,CAAC,WAAW,EAAE,CAAC;QAC1B,MAAM,OAAO,GAAG,MAAM,KAAK,CAAC,cAAc,EAAE,CAAC;QAC7C,qBAAM,CAAC,KAAK,CAAC,OAAO,EAAE,IAAI,CAAC,CAAC;QAC5B,qBAAM,CAAC,KAAK,CAAC,KAAK,CAAC,UAAU,EAAE,YAAY,CAAC,CAAC;IAC/C,CAAC,CAAA,CAAC,CAAC;IAEH,EAAE,CAAC,gDAAgD,EAAE,GAAS,EAAE;QAC9D,MAAM,KAAK,GAAG,IAAI,uBAAU,CAAC,YAAY,CAAC,CAAC;QAC3C,MAAM,KAAK,CAAC,WAAW,EAAE,CAAC;QAC1B,MAAM,KAAK,CAAC,WAAW,EAAE,CAAC,CAAC,8BAA8B;QACzD,MAAM,qBAAM,CAAC,OAAO,CAAC,GAAS,EAAE;YAC9B,MAAM,KAAK,CAAC,WAAW,EAAE,CAAC,CAAC,uCAAuC;QACpE,CAAC,CAAA,EAAE,IAAI,KAAK,CAAC,4BAA4B,CAAC,CAAC,CAAA;IAC7C,CAAC,CAAA,CAAC,CAAA;IAEF,QAAQ,CAAC,aAAa,EAAE,GAAG,EAAE;QAC3B,EAAE,CAAC,uCAAuC,EAAE,GAAS,EAAE;YACrD,MAAM,KAAK,GAAG,IAAI,uBAAU,CAAC,YAAY,CAAC,CAAC;YAC3C,MAAM,KAAK,CAAC,WAAW,EAAE,CAAC,CAAC,0BAA0B;YAErD,6CAA6C;YAC7C,MAAM,qBAAM,CAAC,OAAO,CAAC,GAAS,EAAE;gBAC9B,MAAM,KAAK,CAAC,WAAW,EAAE,CAAA;YAC3B,CAAC,CAAA,EAAE,IAAI,KAAK,CAAC,sBAAsB,CAAC,CAAC,CAAA;YAErC,qDAAqD;YACrD,MAAM,KAAK,CAAC,UAAU,CAAC,EAAC,EAAE,EAAC,GAAG,EAAE,MAAM,EAAE,CAAC,CAAC,EAAC,CAAC,EAAC,CAAC,CAAC,EAAC,CAAC,CAAA;YACjD,MAAM,YAAY,GAAG,CAAC,MAAM,KAAK,CAAC,SAAS,EAAE,CAAC,CAAC,MAAM,CAAA;YACrD,qBAAM,CAAC,KAAK,CAAC,YAAY,EAAE,CAAC,CAAC,CAAA;YAC7B,MAAM,KAAK,CAAC,WAAW,CAAC,EAAC,cAAc,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC,EAAE,eAAe,EAAE,EAAE,EAAC,CAAC,CAAA;YAChF,MAAM,WAAW,GAAG,CAAC,MAAM,KAAK,CAAC,SAAS,EAAE,CAAC,CAAC,MAAM,CAAA;YACpD,qBAAM,CAAC,KAAK,CAAC,WAAW,EAAE,CAAC,CAAC,CAAA;QAC9B,CAAC,CAAA,CAAC,CAAA;QAEF,EAAE,CAAC,qCAAqC,EAAE,GAAS,EAAE;YACnD,MAAM,KAAK,GAAG,IAAI,uBAAU,CAAC,YAAY,CAAC,CAAC;YAC3C,eAAK,CAAC,IAAI,CAAC,kBAAE,EAAE,WAAW,CAAC,CAAC,OAAO,CAAC,IAAI,KAAK,CAAC,UAAU,CAAC,CAAC,CAAA;YAE1D,MAAM,qBAAM,CAAC,OAAO,CAAC,GAAS,EAAE;gBAC9B,MAAM,KAAK,CAAC,WAAW,EAAE,CAAC;YAC5B,CAAC,CAAA,EAAE,IAAI,KAAK,CAAC,sBAAsB,CAAC,CAAC,CAAA;YAErC,MAAM,qBAAM,CAAC,OAAO,CAAC,GAAS,EAAE;gBAC9B,MAAM,KAAK,CAAC,SAAS,EAAE,CAAC;YAC1B,CAAC,CAAA,CAAC,CAAA;QACJ,CAAC,CAAA,CAAC,CAAA;IACJ,CAAC,CAAC,CAAA;IAEF,QAAQ,CAAC,YAAY,EAAE,GAAG,EAAE;QAC1B,EAAE,CAAC,gCAAgC,EAAE,GAAS,EAAE;YAC9C,MAAM,KAAK,GAAG,IAAI,uBAAU,CAAC,YAAY,CAAC,CAAC;YAC3C,MAAM,KAAK,CAAC,WAAW,EAAE,CAAC;YAC1B,MAAM,KAAK,CAAC,WAAW,EAAE,CAAC;YAC1B,MAAM,KAAK,CAAC,UAAU,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC,CAAA;YAC1C,MAAM,KAAK,CAAC,UAAU,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC,CAAA;YAC1C,MAAM,KAAK,CAAC,UAAU,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC,CAAA;YAC1C,MAAM,KAAK,CAAC,SAAS,EAAE,CAAC;YAExB,MAAM,qBAAM,CAAC,aAAa,CAAC,GAAS,EAAE;gBACpC,MAAM,KAAK,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC;YAChC,CAAC,CAAA,CAAC,CAAA;YACF,qBAAM,CAAC,KAAK,CAAC,CAAC,MAAM,KAAK,CAAC,SAAS,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,CAAC,CAAA;QACnD,CAAC,CAAA,CAAC,CAAA;QAEF,EAAE,CAAC,kDAAkD,EAAE,GAAS,EAAE;;YAChE,MAAM,KAAK,GAAG,IAAI,uBAAU,CAAC,YAAY,CAAC,CAAC;YAC3C,MAAM,KAAK,CAAC,WAAW,EAAE,CAAC;YAC1B,MAAM,KAAK,CAAC,UAAU,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC,CAAC;YAE3C,MAAM,KAAK,CAAC,UAAU,CAAC,MAAA,eAAe,CAAC,CAAC,CAAC,CAAC,EAAE,mCAAI,EAAE,CAAC,CAAC;YACpD,qBAAM,CAAC,KAAK,CAAC,MAAM,KAAK,CAAC,cAAc,EAAE,EAAE,IAAI,CAAC,CAAC;YACjD,qBAAM,CAAC,KAAK,CAAC,CAAC,MAAM,KAAK,CAAC,SAAS,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC;QACpD,CAAC,CAAA,CAAC,CAAA;QAEF,EAAE,CAAC,oCAAoC,EAAE,GAAS,EAAE;YAClD,MAAM,KAAK,GAAG,IAAI,uBAAU,CAAC,YAAY,CAAC,CAAC;YAC3C,MAAM,KAAK,CAAC,WAAW,EAAE,CAAC;YAC1B,MAAM,KAAK,CAAC,gBAAgB,CAAC;gBAC3B,EAAC,EAAE,EAAE,GAAG,EAAE,MAAM,EAAE,EAAE,EAAC;gBACrB,EAAC,EAAE,EAAE,GAAG,EAAE,MAAM,EAAE,EAAE,EAAC;gBACrB,EAAC,EAAE,EAAE,GAAG,EAAE,MAAM,EAAE,EAAE,EAAC;gBACrB,EAAC,EAAE,EAAE,GAAG,EAAE,MAAM,EAAE,EAAE,EAAC;gBACrB,EAAC,EAAE,EAAE,GAAG,EAAE,MAAM,EAAE,EAAE,EAAC;aACtB,CAAC,CAAC;YAEH,MAAM,KAAK,CAAC,WAAW,EAAE,CAAC;YAC1B,MAAM,KAAK,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC;YAC5B,MAAM,KAAK,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC;YAC5B,MAAM,KAAK,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC;YAC5B,MAAM,KAAK,CAAC,SAAS,EAAE,CAAC;YAExB,qBAAM,CAAC,eAAe,CAAC,MAAM,KAAK,CAAC,SAAS,EAAE,EAAE,CAAC,EAAC,EAAE,EAAE,GAAG,EAAE,MAAM,EAAE,EAAE,EAAE,QAAQ,EAAE,EAAE,EAAE,IAAI,EAAE,CAAC,EAAC,EAAE,EAAC,EAAE,EAAE,GAAG,EAAE,MAAM,EAAE,EAAE,EAAE,QAAQ,EAAE,EAAE,EAAE,IAAI,EAAE,CAAC,EAAC,CAAC,CAAC,CAAA;QAC/I,CAAC,CAAA,CAAC,CAAA;IACJ,CAAC,CAAC,CAAA;IAEF,QAAQ,CAAC,WAAW,EAAE,GAAG,EAAE;QACzB,EAAE,CAAC,wCAAwC,EAAE,GAAS,EAAE;YACtD,MAAM,KAAK,GAAG,IAAI,uBAAU,CAAC,YAAY,CAAC,CAAC;YAE3C,MAAM,qBAAM,CAAC,OAAO,CAAC,GAAS,EAAE;gBAC9B,MAAM,KAAK,CAAC,SAAS,EAAE,CAAC;YAC1B,CAAC,CAAA,EAAE,IAAI,KAAK,CAAC,uBAAuB,CAAC,CAAC,CAAC;QACzC,CAAC,CAAA,CAAC,CAAA;QAEF,EAAE,CAAC,iDAAiD,EAAE,GAAS,EAAE;YAC/D,MAAM,KAAK,GAAG,IAAI,uBAAU,CAAC,YAAY,EAAE,YAAY,CAAC,CAAC;YACzD,MAAM,KAAK,CAAC,WAAW,EAAE,CAAC;YAC1B,MAAM,KAAK,CAAC,WAAW,EAAE,CAAC;YAE1B,eAAK,CAAC,IAAI,CAAC,kBAAE,EAAE,WAAW,CAAC,CAAC,OAAO,CAAC,IAAI,KAAK,CAAC,UAAU,CAAC,CAAC,CAAA;YAE1D,MAAM,qBAAM,CAAC,OAAO,CAAC,GAAS,EAAE;gBAC9B,MAAM,KAAK,CAAC,SAAS,EAAE,CAAC;YAC1B,CAAC,CAAA,EAAE,IAAI,KAAK,CAAC,qCAAqC,CAAC,CAAC,CAAA;QACtD,CAAC,CAAA,CAAC,CAAA;IACJ,CAAC,CAAC,CAAA;IAEF,QAAQ,CAAC,eAAe,EAAE,GAAG,EAAE;QAC7B,EAAE,CAAC,+BAA+B,EAAE,GAAS,EAAE;YAC7C,MAAM,KAAK,GAAG,IAAI,uBAAU,CAAC,YAAY,CAAC,CAAC;YAC3C,MAAM,KAAK,CAAC,WAAW,EAAE,CAAC;YAE1B,qBAAM,CAAC,eAAe,CAAC,MAAM,KAAK,CAAC,aAAa,EAAE,EAAE;gBAClD,OAAO,EAAE,CAAC;gBACV,eAAe,EAAE,EAAE;gBACnB,KAAK,EAAE,CAAC;aACT,CAAC,CAAA;QACJ,CAAC,CAAA,CAAC,CAAA;QAEF,EAAE,CAAC,yCAAyC,EAAE,GAAS,EAAE;YACvD,MAAM,KAAK,GAAG,IAAI,uBAAU,CAAC,YAAY,CAAC,CAAA;YAC1C,MAAM,KAAK,CAAC,WAAW,CAAC,EAAC,OAAO,EAAE,CAAC,EAAE,eAAe,EAAE,EAAC,OAAO,EAAE,EAAE,EAAC,EAAC,CAAC,CAAA;YACrE,MAAM,KAAK,CAAC,gBAAgB,CAAC,eAAe,CAAC,CAAC;YAE9C,qBAAM,CAAC,eAAe,CAAC,MAAM,KAAK,CAAC,aAAa,EAAE,EAAE;gBAClD,OAAO,EAAE,CAAC;gBACV,eAAe,EAAE,EAAC,OAAO,EAAE,EAAE,EAAC;gBAC9B,KAAK,EAAE,CAAC;aACT,CAAC,CAAA;QACJ,CAAC,CAAA,CAAC,CAAA;IACJ,CAAC,CAAC,CAAA;IAEF,QAAQ,CAAC,SAAS,EAAE,GAAG,EAAE;QACvB,EAAE,CAAC,uCAAuC,EAAE,GAAS,EAAE;YACrD,MAAM,KAAK,GAAG,IAAI,uBAAU,CAAC,YAAY,CAAC,CAAC;YAC3C,MAAM,KAAK,CAAC,WAAW,EAAE,CAAC;YAE1B,qBAAM,CAAC,KAAK,CAAC,MAAM,KAAK,CAAC,OAAO,CAAC,GAAG,CAAC,EAAE,SAAS,CAAC,CAAA;QACnD,CAAC,CAAA,CAAC,CAAA;QAEF,EAAE,CAAC,wBAAwB,EAAE,GAAS,EAAE;YACtC,MAAM,KAAK,GAAG,IAAI,uBAAU,CAAC,YAAY,CAAC,CAAC;YAC3C,MAAM,KAAK,CAAC,WAAW,EAAE,CAAC;YAC1B,MAAM,KAAK,CAAC,gBAAgB,CAAC,eAAe,CAAC,CAAC;YAE9C,MAAM,KAAK,GAAG,MAAM,KAAK,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC;YACvC,qBAAM,CAAC,KAAK,CAAC,KAAK,aAAL,KAAK,uBAAL,KAAK,CAAE,EAAE,EAAE,eAAe,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAA;YAC9C,qBAAM,CAAC,KAAK,CAAC,KAAK,aAAL,KAAK,uBAAL,KAAK,CAAE,MAAM,EAAE,eAAe,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,CAAA;YACtD,qBAAM,CAAC,KAAK,CAAC,CAAC,MAAM,KAAK,CAAC,SAAS,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,CAAC,CAAA;QACnD,CAAC,CAAA,CAAC,CAAA;IACJ,CAAC,CAAC,CAAA;IAEF,QAAQ,CAAC,kBAAkB,EAAE,GAAG,EAAE;QAChC,EAAE,CAAC,8BAA8B,EAAE,GAAS,EAAE;YAC5C,MAAM,KAAK,GAAG,IAAI,uBAAU,CAAC,YAAY,CAAC,CAAC;YAC3C,MAAM,KAAK,CAAC,WAAW,EAAE,CAAC;YAE1B,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,gBAAgB,CAAC,eAAe,CAAC,CAAC;YAE/D,qBAAM,CAAC,KAAK,CAAC,QAAQ,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC;YAEjC,MAAM,cAAc,GAAG,MAAM,KAAK,CAAC,SAAS,EAAE,CAAC;YAC/C,qBAAM,CAAC,KAAK,CAAC,cAAc,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC;QACzC,CAAC,CAAA,CAAC,CAAC;QAEH,EAAE,CAAC,yDAAyD,EAAE,GAAS,EAAE;YACvE,MAAM,KAAK,GAAG,IAAI,uBAAU,CAAC,YAAY,CAAC,CAAC;YAC3C,MAAM,KAAK,CAAC,WAAW,EAAE,CAAC;YAE1B,MAAM,KAAK,CAAC,UAAU,CAAC,EAAE,EAAE,EAAE,GAAG,EAAE,MAAM,EAAE,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC;YAEvD,+DAA+D;YAC/D,MAAM,qBAAM,CAAC,OAAO,CAClB,GAAS,EAAE;gBACT,MAAM,KAAK,CAAC,gBAAgB,CAAC,eAAe,CAAC,CAAC;YAChD,CAAC,CAAA,EACD;gBACE,IAAI,EAAE,OAAO;gBACb,OAAO,EAAE,+BAA+B;aACzC,CACF,CAAC;YAEF,uCAAuC;YACvC,MAAM,WAAW,GAAG,MAAM,KAAK,CAAC,SAAS,EAAE,CAAC;YAC5C,qBAAM,CAAC,KAAK,CAAC,WAAW,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC;QACtC,CAAC,CAAA,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,qBAAqB,EAAE,GAAG,EAAE;QACnC,EAAE,CAAC,wCAAwC,EAAE,GAAS,EAAE;YACtD,MAAM,KAAK,GAAG,IAAI,uBAAU,CAAC,YAAY,CAAC,CAAC;YAC3C,MAAM,KAAK,CAAC,WAAW,EAAE,CAAC;YAC1B,MAAM,KAAK,CAAC,gBAAgB,CAAC;gBAC3B,EAAC,EAAE,EAAE,GAAG,EAAE,MAAM,EAAE,EAAE,EAAE,QAAQ,EAAE,EAAC,QAAQ,EAAE,MAAM,EAAC,EAAC;gBACnD,EAAC,EAAE,EAAE,GAAG,EAAE,MAAM,EAAE,EAAE,EAAE,QAAQ,EAAE,EAAC,QAAQ,EAAE,MAAM,EAAC,EAAC;gBACnD,EAAC,EAAE,EAAE,GAAG,EAAE,MAAM,EAAE,EAAE,EAAE,QAAQ,EAAE,EAAC,QAAQ,EAAE,aAAa,EAAC,EAAC;gBAC1D,EAAC,EAAE,EAAE,GAAG,EAAE,MAAM,EAAE,EAAE,EAAE,QAAQ,EAAE,EAAC,QAAQ,EAAE,OAAO,EAAC,EAAC;gBACpD,EAAC,EAAE,EAAE,GAAG,EAAE,MAAM,EAAE,EAAE,EAAE,QAAQ,EAAE,EAAC,QAAQ,EAAE,MAAM,EAAC,EAAC;aACpD,CAAC,CAAC;YAEH,MAAM,SAAS,GAAG,MAAM,KAAK,CAAC,mBAAmB,CAAC,EAAC,QAAQ,EAAE,EAAC,KAAK,EAAE,MAAM,EAAC,EAAC,CAAC,CAAA;YAC9E,qBAAM,CAAC,eAAe,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,EAAE,CAAC,EAAE,CAAC,GAAG,EAAE,GAAG,EAAE,GAAG,CAAC,CAAC,CAAA;YACzE,MAAM,UAAU,GAAG,MAAM,KAAK,CAAC,mBAAmB,CAAC,EAAC,QAAQ,EAAE,EAAC,KAAK,EAAE,OAAO,EAAC,EAAC,CAAC,CAAA;YAChF,qBAAM,CAAC,eAAe,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,EAAE,CAAC,EAAE,CAAC,GAAG,CAAC,CAAC,CAAA;YAChE,MAAM,aAAa,GAAG,MAAM,KAAK,CAAC,mBAAmB,CAAC,EAAC,QAAQ,EAAE,EAAC,KAAK,EAAE,SAAS,EAAC,EAAC,CAAC,CAAA;YACrF,qBAAM,CAAC,eAAe,CAAC,aAAa,EAAE,EAAE,CAAC,CAAA;QAC3C,CAAC,CAAA,CAAC,CAAA;QAEF,EAAE,CAAC,wCAAwC,EAAE,GAAS,EAAE;YACtD,MAAM,KAAK,GAAG,IAAI,uBAAU,CAAC,YAAY,CAAC,CAAC;YAC3C,MAAM,KAAK,CAAC,WAAW,EAAE,CAAC;YAE1B,MAAM,KAAK,GAAG,MAAM,KAAK,CAAC,mBAAmB,CAAC,EAAE,CAAC,CAAC;YAClD,qBAAM,CAAC,eAAe,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC;QACpC,CAAC,CAAA,CAAC,CAAA;IACJ,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,YAAY,EAAE,GAAG,EAAE;QAC1B,EAAE,CAAC,2CAA2C,EAAE,GAAS,EAAE;YACzD,MAAM,KAAK,GAAG,IAAI,uBAAU,CAAC,YAAY,CAAC,CAAC;YAC3C,MAAM,KAAK,CAAC,WAAW,EAAE,CAAC;YAE1B,MAAM,MAAM,GAAG,MAAM,KAAK,CAAC,UAAU,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,EAAE,EAAE,EAAE,EAAE,CAAC,CAAC;YACzD,qBAAM,CAAC,eAAe,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC;QACrC,CAAC,CAAA,CAAC,CAAC;QAEH,EAAE,CAAC,+CAA+C,EAAE,GAAS,EAAE;;YAC7D,MAAM,KAAK,GAAG,IAAI,uBAAU,CAAC,YAAY,CAAC,CAAC;YAC3C,MAAM,KAAK,CAAC,WAAW,EAAE,CAAC;YAC1B,MAAM,KAAK,CAAC,UAAU,CAAC,EAAE,EAAE,EAAE,GAAG,EAAE,MAAM,EAAE,CAAC,GAAG,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC;YAE/D,MAAM,MAAM,GAAG,MAAM,KAAK,CAAC,UAAU,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,GAAG,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC,CAAC;YAChE,qBAAM,CAAC,KAAK,CAAC,MAAA,MAAM,CAAC,CAAC,CAAC,0CAAE,KAAK,EAAE,CAAC,CAAC,CAAC;YAClC,qBAAM,CAAC,KAAK,CAAC,MAAA,MAAM,CAAC,CAAC,CAAC,0CAAE,IAAI,CAAC,EAAE,EAAE,GAAG,CAAC,CAAC;QACxC,CAAC,CAAA,CAAC,CAAC;QAEH,EAAE,CAAC,gDAAgD,EAAE,GAAS,EAAE;YAC9D,MAAM,KAAK,GAAG,IAAI,uBAAU,CAAC,YAAY,CAAC,CAAC;YAC3C,MAAM,KAAK,CAAC,WAAW,EAAE,CAAC;YAC1B,MAAM,KAAK,CAAC,gBAAgB,CAAC,eAAe,CAAC,CAAC;YAE9C,MAAM,MAAM,GAAG,MAAM,KAAK,CAAC,UAAU,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,EAAE,EAAE,EAAE,EAAE,CAAC,CAAC;YACzD,qBAAM,CAAC,KAAK,CAAC,MAAM,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC;YAC/B,qBAAM,CAAC,eAAe,CACpB,MAAM,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,EAAE,EAAE,EAAE,CAAC,IAAI,CAAC,EAAE,CAAC,EACjC,eAAe,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,EAAE,CAAC,CACvC,CAAC;QACJ,CAAC,CAAA,CAAC,CAAC;QAEH,EAAE,CAAC,0CAA0C,EAAE,GAAS,EAAE;YACxD,MAAM,KAAK,GAAG,IAAI,uBAAU,CAAC,YAAY,CAAC,CAAC;YAC3C,MAAM,KAAK,CAAC,WAAW,EAAE,CAAC;YAC1B,MAAM,KAAK,CAAC,gBAAgB,CAAC;gBAC3B,EAAE,EAAE,EAAE,GAAG,EAAE,MAAM,EAAE,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,EAAE,QAAQ,EAAE,EAAE,QAAQ,EAAE,MAAM,EAAE,EAAE;gBAC9D,EAAE,EAAE,EAAE,GAAG,EAAE,MAAM,EAAE,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,EAAE,QAAQ,EAAE,EAAE,QAAQ,EAAE,OAAO,EAAE,EAAE;aAChE,CAAC,CAAC;YAEH,MAAM,gBAAgB,GAAG,MAAM,KAAK,CAAC,UAAU,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC,CAAC;YAClE,MAAM,cAAc,GAAG,MAAM,KAAK,CAAC,UAAU,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,EAAE;gBAC9D,QAAQ,EAAE,EAAE,GAAG,EAAE,OAAO,EAAE;aAC3B,CAAC,CAAC;YAEH,qBAAM,CAAC,KAAK,CAAC,gBAAgB,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,EAAE,EAAE,GAAG,CAAC,CAAC;YAC/C,qBAAM,CAAC,KAAK,CAAC,cAAc,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,EAAE,EAAE,GAAG,CAAC,CAAC;QAC/C,CAAC,CAAA,CAAC,CAAC;QAEH,EAAE,CAAC,wCAAwC,EAAE,GAAS,EAAE;YACtD,MAAM,KAAK,GAAG,IAAI,uBAAU,CAAC,YAAY,CAAC,CAAC;YAC3C,MAAM,KAAK,CAAC,WAAW,CAAC,EAAC,OAAO,EAAE,CAAC,EAAE,eAAe,EAAE,EAAC,OAAO,EAAE,CAAC,UAAU,CAAC,EAAC,EAAC,CAAC,CAAC;YAChF,MAAM,KAAK,CAAC,gBAAgB,CAAC;gBAC3B,EAAE,EAAE,EAAE,GAAG,EAAE,MAAM,EAAE,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,EAAE;gBAC9B,EAAE,EAAE,EAAE,GAAG,EAAE,MAAM,EAAE,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,EAAE,QAAQ,EAAE,EAAC,QAAQ,EAAE,OAAO,EAAC,EAAE;aAC9D,CAAC,CAAC;YAEH,eAAK;iBACF,IAAI,CAAC,kBAAE,EAAE,UAAU,CAAC;iBACpB,QAAQ,CAAC,IAAI,CAAC,SAAS,CAAC,EAAE,QAAQ,EAAE,OAAO,EAAE,CAAC,CAAC,CAAC;YAEnD,MAAM,cAAc,GAAG,MAAM,KAAK,CAAC,UAAU,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,EAAE,EAAC,QAAQ,EAAE,EAAC,KAAK,EAAE,OAAO,EAAC,EAAC,CAAC,CAAC;YAE9F,qBAAM,CAAC,QAAQ,CAAC,cAAc,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,YAAY,EAAE,SAAS,CAAC,CAAC;YAChE,qBAAM,CAAC,KAAK,CAAC,cAAc,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,EAAE,EAAE,GAAG,CAAC,CAAC;QAC/C,CAAC,CAAA,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/lib/OpenAIEmbeddings.d.ts

@@ -0,0 +1,126 @@
+import { AxiosResponse, AxiosRequestConfig } from 'axios';
+import { EmbeddingsModel, EmbeddingsResponse } from "./types";
+import { CreateEmbeddingRequest, CreateEmbeddingResponse } from "./internals";
+export interface BaseOpenAIEmbeddingsOptions {
+    /**
+     * Optional. Number of embedding dimensions to return.
+     */
+    dimensions?: number;
+    /**
+     * Optional. Whether to log requests to the console.
+     * @remarks
+     * This is useful for debugging prompts and defaults to `false`.
+     */
+    logRequests?: boolean;
+    /**
+     * Optional. Maximum number of tokens that can be sent to the embedding model.
+     */
+    maxTokens?: number;
+    /**
+     * Optional. Retry policy to use when calling the OpenAI API.
+     * @remarks
+     * The default retry policy is `[2000, 5000]` which means that the first retry will be after
+     * 2 seconds and the second retry will be after 5 seconds.
+     */
+    retryPolicy?: number[];
+    /**
+     * Optional. Request options to use when calling the OpenAI API.
+     */
+    requestConfig?: AxiosRequestConfig;
+}
+/**
+ * Options for configuring an `OpenAIEmbeddings` to generate embeddings using an OSS hosted model.
+ */
+export interface OSSEmbeddingsOptions extends BaseOpenAIEmbeddingsOptions {
+    /**
+     * Model to use for completion.
+     */
+    ossModel: string;
+    /**
+     * Optional. Endpoint to use when calling the OpenAI API.
+     * @remarks
+     * For Azure OpenAI this is the deployment endpoint.
+     */
+    ossEndpoint: string;
+}
+/**
+ * Options for configuring an `OpenAIEmbeddings` to generate embeddings using an OpenAI hosted model.
+ */
+export interface OpenAIEmbeddingsOptions extends BaseOpenAIEmbeddingsOptions {
+    /**
+     * API key to use when calling the OpenAI API.
+     * @remarks
+     * A new API key can be created at https://platform.openai.com/account/api-keys.
+     */
+    apiKey: string;
+    /**
+     * Model to use for completion.
+     * @remarks
+     * For Azure OpenAI this is the name of the deployment to use.
+     */
+    model: string;
+    /**
+     * Optional. Organization to use when calling the OpenAI API.
+     */
+    organization?: string;
+    /**
+     * Optional. Endpoint to use when calling the OpenAI API.
+     */
+    endpoint?: string;
+}
+/**
+ * Options for configuring an `OpenAIEmbeddings` to generate embeddings using an Azure OpenAI hosted model.
+ */
+export interface AzureOpenAIEmbeddingsOptions extends BaseOpenAIEmbeddingsOptions {
+    /**
+     * API key to use when making requests to Azure OpenAI.
+     */
+    azureApiKey: string;
+    /**
+     * Deployment endpoint to use.
+     */
+    azureEndpoint: string;
+    /**
+     * Name of the Azure OpenAI deployment (model) to use.
+     */
+    azureDeployment: string;
+    /**
+     * Optional. Version of the API being called. Defaults to `2023-05-15`.
+     */
+    azureApiVersion?: string;
+}
+/**
+ * A `PromptCompletionModel` for calling OpenAI and Azure OpenAI hosted models.
+ * @remarks
+ */
+export declare class OpenAIEmbeddings implements EmbeddingsModel {
+    private readonly _httpClient;
+    private readonly _clientType;
+    private readonly UserAgent;
+    readonly maxTokens: number;
+    /**
+     * Options the client was configured with.
+     */
+    readonly options: OSSEmbeddingsOptions | OpenAIEmbeddingsOptions | AzureOpenAIEmbeddingsOptions;
+    /**
+     * Creates a new `OpenAIClient` instance.
+     * @param options Options for configuring an `OpenAIClient`.
+     */
+    constructor(options: OSSEmbeddingsOptions | OpenAIEmbeddingsOptions | AzureOpenAIEmbeddingsOptions);
+    /**
+     * Creates embeddings for the given inputs using the OpenAI API.
+     * @param model Name of the model to use (or deployment for Azure).
+     * @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>;
+    /**
+     * @private
+     */
+    protected createEmbeddingRequest(request: CreateEmbeddingRequest): Promise<AxiosResponse<CreateEmbeddingResponse>>;
+    /**
+     * @private
+     */
+    protected post<TData>(url: string, body: object, retryCount?: number): Promise<AxiosResponse<TData>>;
+}
+//# sourceMappingURL=OpenAIEmbeddings.d.ts.map

package/lib/OpenAIEmbeddings.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"OpenAIEmbeddings.d.ts","sourceRoot":"","sources":["../src/OpenAIEmbeddings.ts"],"names":[],"mappings":"AAAA,OAAc,EAAiB,aAAa,EAAE,kBAAkB,EAAE,MAAM,OAAO,CAAC;AAChF,OAAO,EAAE,eAAe,EAAE,kBAAkB,EAAE,MAAM,SAAS,CAAC;AAC9D,OAAO,EAAE,sBAAsB,EAAE,uBAAuB,EAAgC,MAAM,aAAa,CAAC;AAG5G,MAAM,WAAW,2BAA2B;IACxC;;OAEG;IACH,UAAU,CAAC,EAAE,MAAM,CAAC;IAEpB;;;;OAIG;IACH,WAAW,CAAC,EAAE,OAAO,CAAC;IAEtB;;OAEG;IACH,SAAS,CAAC,EAAE,MAAM,CAAC;IAEnB;;;;;OAKG;IACH,WAAW,CAAC,EAAE,MAAM,EAAE,CAAC;IAEvB;;OAEG;IACH,aAAa,CAAC,EAAE,kBAAkB,CAAC;CACtC;AAGD;;GAEG;AACH,MAAM,WAAW,oBAAqB,SAAQ,2BAA2B;IACrE;;OAEG;IACH,QAAQ,EAAE,MAAM,CAAC;IAEjB;;;;OAIG;IACH,WAAW,EAAE,MAAM,CAAC;CACvB;AAED;;GAEG;AACH,MAAM,WAAW,uBAAwB,SAAQ,2BAA2B;IACxE;;;;OAIG;IACH,MAAM,EAAE,MAAM,CAAC;IAEf;;;;OAIG;IACH,KAAK,EAAE,MAAM,CAAC;IAEd;;OAEG;IACH,YAAY,CAAC,EAAE,MAAM,CAAC;IAEtB;;OAEG;IACH,QAAQ,CAAC,EAAE,MAAM,CAAC;CACrB;AAED;;GAEG;AACH,MAAM,WAAW,4BAA6B,SAAQ,2BAA2B;IAC7E;;OAEG;IACH,WAAW,EAAE,MAAM,CAAC;IAEpB;;OAEG;IACH,aAAa,EAAE,MAAM,CAAC;IAEtB;;OAEG;IACH,eAAe,EAAE,MAAM,CAAC;IAExB;;OAEG;IACH,eAAe,CAAC,EAAE,MAAM,CAAC;CAC5B;AAED;;;GAGG;AACH,qBAAa,gBAAiB,YAAW,eAAe;IACpD,OAAO,CAAC,QAAQ,CAAC,WAAW,CAAgB;IAC5C,OAAO,CAAC,QAAQ,CAAC,WAAW,CAAa;IAEzC,OAAO,CAAC,QAAQ,CAAC,SAAS,CAAe;IAEzC,SAAgB,SAAS,SAAC;IAE1B;;OAEG;IACH,SAAgB,OAAO,EAAE,oBAAoB,GAAC,uBAAuB,GAAC,4BAA4B,CAAC;IAEnG;;;OAGG;gBACgB,OAAO,EAAE,oBAAoB,GAAC,uBAAuB,GAAC,4BAA4B;IAwCrG;;;;;OAKG;IACU,gBAAgB,CAAC,MAAM,EAAE,MAAM,GAAG,MAAM,EAAE,GAAG,OAAO,CAAC,kBAAkB,CAAC;IA6BrF;;OAEG;IACH,SAAS,CAAC,sBAAsB,CAAC,OAAO,EAAE,sBAAsB,GAAG,OAAO,CAAC,aAAa,CAAC,uBAAuB,CAAC,CAAC;IAqBlH;;OAEG;cACa,IAAI,CAAC,KAAK,EAAE,GAAG,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,EAAE,UAAU,SAAI,GAAG,OAAO,CAAC,aAAa,CAAC,KAAK,CAAC,CAAC;CAqCxG"}

package/lib/OpenAIEmbeddings.js

@@ -0,0 +1,174 @@
+"use strict";
+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());
+    });
+};
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.OpenAIEmbeddings = void 0;
+const axios_1 = __importDefault(require("axios"));
+const internals_1 = require("./internals");
+/**
+ * A `PromptCompletionModel` for calling OpenAI and Azure OpenAI hosted models.
+ * @remarks
+ */
+class OpenAIEmbeddings {
+    /**
+     * Creates a new `OpenAIClient` instance.
+     * @param options Options for configuring an `OpenAIClient`.
+     */
+    constructor(options) {
+        var _a;
+        this.UserAgent = 'AlphaWave';
+        this.maxTokens = (_a = options.maxTokens) !== null && _a !== void 0 ? _a : 500;
+        // Check for azure config
+        if (options.azureApiKey) {
+            this._clientType = ClientType.AzureOpenAI;
+            this.options = Object.assign({
+                retryPolicy: [2000, 5000],
+                azureApiVersion: '2023-05-15',
+            }, options);
+            // Cleanup and validate endpoint
+            let endpoint = this.options.azureEndpoint.trim();
+            if (endpoint.endsWith('/')) {
+                endpoint = endpoint.substring(0, endpoint.length - 1);
+            }
+            if (!endpoint.toLowerCase().startsWith('https://')) {
+                throw new Error(`Client created with an invalid endpoint of '${endpoint}'. The endpoint must be a valid HTTPS url.`);
+            }
+            this.options.azureEndpoint = endpoint;
+        }
+        else if (options.ossModel) {
+            this._clientType = ClientType.OSS;
+            this.options = Object.assign({
+                retryPolicy: [2000, 5000]
+            }, options);
+        }
+        else {
+            this._clientType = ClientType.OpenAI;
+            this.options = Object.assign({
+                retryPolicy: [2000, 5000]
+            }, options);
+        }
+        // Create client
+        this._httpClient = axios_1.default.create({
+            validateStatus: (status) => status < 400 || status == 429
+        });
+    }
+    /**
+     * Creates embeddings for the given inputs using the OpenAI API.
+     * @param model Name of the model to use (or deployment for Azure).
+     * @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) {
+        return __awaiter(this, void 0, void 0, function* () {
+            if (this.options.logRequests) {
+                console.log(internals_1.Colorize.title('EMBEDDINGS REQUEST:'));
+                console.log(internals_1.Colorize.output(inputs));
+            }
+            const startTime = Date.now();
+            const response = yield this.createEmbeddingRequest({
+                input: inputs,
+            });
+            if (this.options.logRequests) {
+                console.log(internals_1.Colorize.title('RESPONSE:'));
+                console.log(internals_1.Colorize.value('status', response.status));
+                console.log(internals_1.Colorize.value('duration', Date.now() - startTime, 'ms'));
+                console.log(internals_1.Colorize.output(response.data));
+            }
+            // Process response
+            if (response.status < 300) {
+                return { status: 'success', output: response.data.data.sort((a, b) => a.index - b.index).map((item) => item.embedding) };
+            }
+            else if (response.status == 429) {
+                return { status: 'rate_limited', message: `The embeddings API returned a rate limit error.` };
+            }
+            else {
+                return { status: 'error', message: `The embeddings API returned an error status of ${response.status}: ${response.statusText}` };
+            }
+        });
+    }
+    /**
+     * @private
+     */
+    createEmbeddingRequest(request) {
+        var _a;
+        if (this.options.dimensions) {
+            request.dimensions = this.options.dimensions;
+        }
+        if (this._clientType == ClientType.AzureOpenAI) {
+            const options = this.options;
+            const url = `${options.azureEndpoint}/openai/deployments/${options.azureDeployment}/embeddings?api-version=${options.azureApiVersion}`;
+            return this.post(url, request);
+        }
+        else if (this._clientType == ClientType.OSS) {
+            const options = this.options;
+            const url = `${options.ossEndpoint}/v1/embeddings`;
+            request.model = options.ossModel;
+            return this.post(url, request);
+        }
+        else {
+            const options = this.options;
+            const url = `${(_a = options.endpoint) !== null && _a !== void 0 ? _a : 'https://api.openai.com'}/v1/embeddings`;
+            request.model = options.model;
+            return this.post(url, request);
+        }
+    }
+    /**
+     * @private
+     */
+    post(url_1, body_1) {
+        return __awaiter(this, arguments, void 0, function* (url, body, retryCount = 0) {
+            // Initialize request config
+            const requestConfig = Object.assign({}, this.options.requestConfig);
+            // Initialize request headers
+            if (!requestConfig.headers) {
+                requestConfig.headers = {};
+            }
+            if (!requestConfig.headers['Content-Type']) {
+                requestConfig.headers['Content-Type'] = 'application/json';
+            }
+            if (!requestConfig.headers['User-Agent']) {
+                requestConfig.headers['User-Agent'] = this.UserAgent;
+            }
+            if (this._clientType == ClientType.AzureOpenAI) {
+                const options = this.options;
+                requestConfig.headers['api-key'] = options.azureApiKey;
+            }
+            else if (this._clientType == ClientType.OpenAI) {
+                const options = this.options;
+                requestConfig.headers['Authorization'] = `Bearer ${options.apiKey}`;
+                if (options.organization) {
+                    requestConfig.headers['OpenAI-Organization'] = options.organization;
+                }
+            }
+            // Send request
+            const response = yield this._httpClient.post(url, body, requestConfig);
+            // Check for rate limit error
+            if (response.status == 429 && Array.isArray(this.options.retryPolicy) && retryCount < this.options.retryPolicy.length) {
+                const delay = this.options.retryPolicy[retryCount];
+                yield new Promise((resolve) => setTimeout(resolve, delay));
+                return this.post(url, body, retryCount + 1);
+            }
+            else {
+                return response;
+            }
+        });
+    }
+}
+exports.OpenAIEmbeddings = OpenAIEmbeddings;
+var ClientType;
+(function (ClientType) {
+    ClientType[ClientType["OpenAI"] = 0] = "OpenAI";
+    ClientType[ClientType["AzureOpenAI"] = 1] = "AzureOpenAI";
+    ClientType[ClientType["OSS"] = 2] = "OSS";
+})(ClientType || (ClientType = {}));
+//# sourceMappingURL=OpenAIEmbeddings.js.map

package/lib/OpenAIEmbeddings.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"OpenAIEmbeddings.js","sourceRoot":"","sources":["../src/OpenAIEmbeddings.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;AAAA,kDAAgF;AAGhF,2CAAuC;AA0GvC;;;GAGG;AACH,MAAa,gBAAgB;IAazB;;;OAGG;IACH,YAAmB,OAAkF;;QAbpF,cAAS,GAAG,WAAW,CAAC;QAcrC,IAAI,CAAC,SAAS,GAAG,MAAA,OAAO,CAAC,SAAS,mCAAI,GAAG,CAAC;QAE1C,yBAAyB;QACzB,IAAK,OAAwC,CAAC,WAAW,EAAE,CAAC;YACxD,IAAI,CAAC,WAAW,GAAG,UAAU,CAAC,WAAW,CAAC;YAC1C,IAAI,CAAC,OAAO,GAAG,MAAM,CAAC,MAAM,CAAC;gBACzB,WAAW,EAAE,CAAC,IAAI,EAAE,IAAI,CAAC;gBACzB,eAAe,EAAE,YAAY;aAChC,EAAE,OAAO,CAAiC,CAAC;YAE5C,gCAAgC;YAChC,IAAI,QAAQ,GAAG,IAAI,CAAC,OAAO,CAAC,aAAa,CAAC,IAAI,EAAE,CAAC;YACjD,IAAI,QAAQ,CAAC,QAAQ,CAAC,GAAG,CAAC,EAAE,CAAC;gBACzB,QAAQ,GAAG,QAAQ,CAAC,SAAS,CAAC,CAAC,EAAE,QAAQ,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;YAC1D,CAAC;YAED,IAAI,CAAC,QAAQ,CAAC,WAAW,EAAE,CAAC,UAAU,CAAC,UAAU,CAAC,EAAE,CAAC;gBACjD,MAAM,IAAI,KAAK,CAAC,+CAA+C,QAAQ,4CAA4C,CAAC,CAAC;YACzH,CAAC;YAED,IAAI,CAAC,OAAO,CAAC,aAAa,GAAG,QAAQ,CAAC;QAC1C,CAAC;aAAM,IAAK,OAAgC,CAAC,QAAQ,EAAE,CAAC;YACpD,IAAI,CAAC,WAAW,GAAG,UAAU,CAAC,GAAG,CAAC;YAClC,IAAI,CAAC,OAAO,GAAG,MAAM,CAAC,MAAM,CAAC;gBACzB,WAAW,EAAE,CAAC,IAAI,EAAE,IAAI,CAAC;aAC5B,EAAE,OAAO,CAAyB,CAAC;QACxC,CAAC;aAAM,CAAC;YACJ,IAAI,CAAC,WAAW,GAAG,UAAU,CAAC,MAAM,CAAC;YACrC,IAAI,CAAC,OAAO,GAAG,MAAM,CAAC,MAAM,CAAC;gBACzB,WAAW,EAAE,CAAC,IAAI,EAAE,IAAI,CAAC;aAC5B,EAAE,OAAO,CAA4B,CAAC;QAC3C,CAAC;QAED,gBAAgB;QAChB,IAAI,CAAC,WAAW,GAAG,eAAK,CAAC,MAAM,CAAC;YAC5B,cAAc,EAAE,CAAC,MAAM,EAAE,EAAE,CAAC,MAAM,GAAG,GAAG,IAAI,MAAM,IAAI,GAAG;SAC5D,CAAC,CAAC;IACP,CAAC;IAED;;;;;OAKG;IACU,gBAAgB,CAAC,MAAyB;;YACnD,IAAI,IAAI,CAAC,OAAO,CAAC,WAAW,EAAE,CAAC;gBAC3B,OAAO,CAAC,GAAG,CAAC,oBAAQ,CAAC,KAAK,CAAC,qBAAqB,CAAC,CAAC,CAAC;gBACnD,OAAO,CAAC,GAAG,CAAC,oBAAQ,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC;YACzC,CAAC;YAED,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;YAC7B,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,sBAAsB,CAAC;gBAC/C,KAAK,EAAE,MAAM;aAChB,CAAC,CAAC;YAEH,IAAI,IAAI,CAAC,OAAO,CAAC,WAAW,EAAE,CAAC;gBAC3B,OAAO,CAAC,GAAG,CAAC,oBAAQ,CAAC,KAAK,CAAC,WAAW,CAAC,CAAC,CAAC;gBACzC,OAAO,CAAC,GAAG,CAAC,oBAAQ,CAAC,KAAK,CAAC,QAAQ,EAAE,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC;gBACvD,OAAO,CAAC,GAAG,CAAC,oBAAQ,CAAC,KAAK,CAAC,UAAU,EAAE,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS,EAAE,IAAI,CAAC,CAAC,CAAC;gBACtE,OAAO,CAAC,GAAG,CAAC,oBAAQ,CAAC,MAAM,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC;YAChD,CAAC;YAGD,mBAAmB;YACnB,IAAI,QAAQ,CAAC,MAAM,GAAG,GAAG,EAAE,CAAC;gBACxB,OAAO,EAAE,MAAM,EAAE,SAAS,EAAE,MAAM,EAAE,QAAQ,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,KAAK,GAAG,CAAC,CAAC,KAAK,CAAC,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,SAAS,CAAC,EAAE,CAAC;YAC7H,CAAC;iBAAM,IAAI,QAAQ,CAAC,MAAM,IAAI,GAAG,EAAE,CAAC;gBAChC,OAAO,EAAE,MAAM,EAAE,cAAc,EAAE,OAAO,EAAE,iDAAiD,EAAE,CAAA;YACjG,CAAC;iBAAM,CAAC;gBACJ,OAAO,EAAE,MAAM,EAAE,OAAO,EAAE,OAAO,EAAE,kDAAkD,QAAQ,CAAC,MAAM,KAAK,QAAQ,CAAC,UAAU,EAAE,EAAE,CAAC;YACrI,CAAC;QACL,CAAC;KAAA;IAED;;OAEG;IACO,sBAAsB,CAAC,OAA+B;;QAC5D,IAAI,IAAI,CAAC,OAAO,CAAC,UAAU,EAAE,CAAC;YAC1B,OAAO,CAAC,UAAU,GAAG,IAAI,CAAC,OAAO,CAAC,UAAU,CAAC;QACjD,CAAC;QACD,IAAI,IAAI,CAAC,WAAW,IAAI,UAAU,CAAC,WAAW,EAAE,CAAC;YAC7C,MAAM,OAAO,GAAG,IAAI,CAAC,OAAuC,CAAC;YAC7D,MAAM,GAAG,GAAG,GAAG,OAAO,CAAC,aAAa,uBAAuB,OAAO,CAAC,eAAe,2BAA2B,OAAO,CAAC,eAAgB,EAAE,CAAC;YACxI,OAAO,IAAI,CAAC,IAAI,CAAC,GAAG,EAAE,OAAO,CAAC,CAAC;QACnC,CAAC;aAAM,IAAI,IAAI,CAAC,WAAW,IAAI,UAAU,CAAC,GAAG,EAAE,CAAC;YAC5C,MAAM,OAAO,GAAG,IAAI,CAAC,OAA+B,CAAC;YACrD,MAAM,GAAG,GAAG,GAAG,OAAO,CAAC,WAAW,gBAAgB,CAAC;YAClD,OAAwC,CAAC,KAAK,GAAG,OAAO,CAAC,QAAQ,CAAC;YACnE,OAAO,IAAI,CAAC,IAAI,CAAC,GAAG,EAAE,OAAO,CAAC,CAAC;QACnC,CAAC;aAAM,CAAC;YACJ,MAAM,OAAO,GAAG,IAAI,CAAC,OAAkC,CAAC;YACxD,MAAM,GAAG,GAAG,GAAG,MAAA,OAAO,CAAC,QAAQ,mCAAI,wBAAwB,gBAAgB,CAAC;YAC3E,OAAwC,CAAC,KAAK,GAAG,OAAO,CAAC,KAAK,CAAC;YAChE,OAAO,IAAI,CAAC,IAAI,CAAC,GAAG,EAAE,OAAO,CAAC,CAAC;QACnC,CAAC;IACL,CAAC;IAED;;OAEG;IACa,IAAI;6DAAQ,GAAW,EAAE,IAAY,EAAE,UAAU,GAAG,CAAC;YACjE,4BAA4B;YAC5B,MAAM,aAAa,GAAuB,MAAM,CAAC,MAAM,CAAC,EAAE,EAAE,IAAI,CAAC,OAAO,CAAC,aAAa,CAAC,CAAC;YAExF,6BAA6B;YAC7B,IAAI,CAAC,aAAa,CAAC,OAAO,EAAE,CAAC;gBACzB,aAAa,CAAC,OAAO,GAAG,EAAE,CAAC;YAC/B,CAAC;YACD,IAAI,CAAC,aAAa,CAAC,OAAO,CAAC,cAAc,CAAC,EAAE,CAAC;gBACzC,aAAa,CAAC,OAAO,CAAC,cAAc,CAAC,GAAG,kBAAkB,CAAC;YAC/D,CAAC;YACD,IAAI,CAAC,aAAa,CAAC,OAAO,CAAC,YAAY,CAAC,EAAE,CAAC;gBACvC,aAAa,CAAC,OAAO,CAAC,YAAY,CAAC,GAAG,IAAI,CAAC,SAAS,CAAC;YACzD,CAAC;YACD,IAAI,IAAI,CAAC,WAAW,IAAI,UAAU,CAAC,WAAW,EAAE,CAAC;gBAC7C,MAAM,OAAO,GAAG,IAAI,CAAC,OAAuC,CAAC;gBAC7D,aAAa,CAAC,OAAO,CAAC,SAAS,CAAC,GAAG,OAAO,CAAC,WAAW,CAAC;YAC3D,CAAC;iBAAM,IAAI,IAAI,CAAC,WAAW,IAAI,UAAU,CAAC,MAAM,EAAE,CAAC;gBAC/C,MAAM,OAAO,GAAG,IAAI,CAAC,OAAkC,CAAC;gBACxD,aAAa,CAAC,OAAO,CAAC,eAAe,CAAC,GAAG,UAAU,OAAO,CAAC,MAAM,EAAE,CAAC;gBACpE,IAAI,OAAO,CAAC,YAAY,EAAE,CAAC;oBACvB,aAAa,CAAC,OAAO,CAAC,qBAAqB,CAAC,GAAG,OAAO,CAAC,YAAY,CAAC;gBACxE,CAAC;YACL,CAAC;YAED,eAAe;YACf,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,WAAW,CAAC,IAAI,CAAC,GAAG,EAAE,IAAI,EAAE,aAAa,CAAC,CAAC;YAEvE,6BAA6B;YAC7B,IAAI,QAAQ,CAAC,MAAM,IAAI,GAAG,IAAI,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,OAAO,CAAC,WAAW,CAAC,IAAI,UAAU,GAAG,IAAI,CAAC,OAAO,CAAC,WAAW,CAAC,MAAM,EAAE,CAAC;gBACpH,MAAM,KAAK,GAAG,IAAI,CAAC,OAAO,CAAC,WAAW,CAAC,UAAU,CAAC,CAAC;gBACnD,MAAM,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,KAAK,CAAC,CAAC,CAAC;gBAC3D,OAAO,IAAI,CAAC,IAAI,CAAC,GAAG,EAAE,IAAI,EAAE,UAAU,GAAG,CAAC,CAAC,CAAC;YAChD,CAAC;iBAAM,CAAC;gBACJ,OAAO,QAAQ,CAAC;YACpB,CAAC;QACL,CAAC;KAAA;CACJ;AA5JD,4CA4JC;AAED,IAAK,UAIJ;AAJD,WAAK,UAAU;IACX,+CAAM,CAAA;IACN,yDAAW,CAAA;IACX,yCAAG,CAAA;AACP,CAAC,EAJI,UAAU,KAAV,UAAU,QAId"}

package/lib/TextSplitter.d.ts

@@ -0,0 +1,19 @@
+import { TextChunk, Tokenizer } from "./types";
+export interface TextSplitterConfig {
+    separators: string[];
+    keepSeparators: boolean;
+    chunkSize: number;
+    chunkOverlap: number;
+    tokenizer: Tokenizer;
+    docType?: string;
+}
+export declare class TextSplitter {
+    private readonly _config;
+    constructor(config?: Partial<TextSplitterConfig>);
+    split(text: string): TextChunk[];
+    private recursiveSplit;
+    private combineChunks;
+    private splitBySpaces;
+    private getSeparators;
+}
+//# sourceMappingURL=TextSplitter.d.ts.map

package/lib/TextSplitter.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"TextSplitter.d.ts","sourceRoot":"","sources":["../src/TextSplitter.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,SAAS,EAAE,SAAS,EAAE,MAAM,SAAS,CAAC;AAI/C,MAAM,WAAW,kBAAkB;IAC/B,UAAU,EAAE,MAAM,EAAE,CAAC;IACrB,cAAc,EAAE,OAAO,CAAC;IACxB,SAAS,EAAE,MAAM,CAAC;IAClB,YAAY,EAAE,MAAM,CAAC;IACrB,SAAS,EAAE,SAAS,CAAC;IACrB,OAAO,CAAC,EAAE,MAAM,CAAC;CACpB;AAED,qBAAa,YAAY;IACrB,OAAO,CAAC,QAAQ,CAAC,OAAO,CAAqB;gBAE1B,MAAM,CAAC,EAAE,OAAO,CAAC,kBAAkB,CAAC;IA2BhD,KAAK,CAAC,IAAI,EAAE,MAAM,GAAG,SAAS,EAAE;IA4BvC,OAAO,CAAC,cAAc;IAoEtB,OAAO,CAAC,aAAa;IA8BrB,OAAO,CAAC,oBAAoB;IAS5B,OAAO,CAAC,aAAa;IAiBrB,OAAO,CAAC,aAAa;CA4WxB"}
+{"version":3,"file":"TextSplitter.d.ts","sourceRoot":"","sources":["../src/TextSplitter.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,SAAS,EAAE,SAAS,EAAE,MAAM,SAAS,CAAC;AAI/C,MAAM,WAAW,kBAAkB;IACjC,UAAU,EAAE,MAAM,EAAE,CAAC;IACrB,cAAc,EAAE,OAAO,CAAC;IACxB,SAAS,EAAE,MAAM,CAAC;IAClB,YAAY,EAAE,MAAM,CAAC;IACrB,SAAS,EAAE,SAAS,CAAC;IACrB,OAAO,CAAC,EAAE,MAAM,CAAC;CAClB;AAED,qBAAa,YAAY;IACvB,OAAO,CAAC,QAAQ,CAAC,OAAO,CAAqB;gBAE1B,MAAM,CAAC,EAAE,OAAO,CAAC,kBAAkB,CAAC;IA2BhD,KAAK,CAAC,IAAI,EAAE,MAAM,GAAG,SAAS,EAAE;IA+BvC,OAAO,CAAC,cAAc;IAuEtB,OAAO,CAAC,aAAa;IAsDrB,OAAO,CAAC,aAAa;IAkBrB,OAAO,CAAC,aAAa;CA8QtB"}