@kyuda/n8n-nodes-databricks 1.0.2

package/dist/nodes/vector_store/shared/createVectorStoreNode/createVectorStoreNode.test.js

@@ -0,0 +1,131 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+const jest_mock_extended_1 = require("jest-mock-extended");
+const createVectorStoreNode_1 = require("./createVectorStoreNode");
+jest.mock('@utils/logWrapper', () => ({
+    logWrapper: jest.fn().mockImplementation((val) => ({ logWrapped: val })),
+}));
+const DEFAULT_PARAMETERS = {
+    options: {},
+    topK: 1,
+};
+const MOCK_DOCUMENTS = [
+    [
+        {
+            pageContent: 'first page',
+            metadata: {
+                id: 123,
+            },
+        },
+        0,
+    ],
+    [
+        {
+            pageContent: 'second page',
+            metadata: {
+                id: 567,
+            },
+        },
+        0,
+    ],
+];
+const MOCK_SEARCH_VALUE = 'search value';
+const MOCK_EMBEDDED_SEARCH_VALUE = [1, 2, 3];
+describe('createVectorStoreNode', () => {
+    const vectorStore = (0, jest_mock_extended_1.mock)({
+        similaritySearchVectorWithScore: jest.fn().mockResolvedValue(MOCK_DOCUMENTS),
+    });
+    const vectorStoreNodeArgs = (0, jest_mock_extended_1.mock)({
+        sharedFields: [],
+        insertFields: [],
+        loadFields: [
+            {
+                name: 'loadField',
+            },
+        ],
+        retrieveFields: [],
+        updateFields: [],
+        getVectorStoreClient: jest.fn().mockReturnValue(vectorStore),
+    });
+    const embeddings = (0, jest_mock_extended_1.mock)({
+        embedQuery: jest.fn().mockResolvedValue(MOCK_EMBEDDED_SEARCH_VALUE),
+    });
+    const context = (0, jest_mock_extended_1.mock)({
+        getNodeParameter: jest.fn(),
+        getInputConnectionData: jest.fn().mockReturnValue(embeddings),
+    });
+    describe('retrieve mode', () => {
+        it('supplies vector store as data', async () => {
+            const parameters = {
+                ...DEFAULT_PARAMETERS,
+                mode: 'retrieve',
+            };
+            context.getNodeParameter.mockImplementation((parameterName, itemIndex, fallbackValue, options) => {
+                var _a;
+                return (_a = parameters[parameterName]) !== null && _a !== void 0 ? _a : fallbackValue;
+            });
+            const VectorStoreNodeType = (0, createVectorStoreNode_1.createVectorStoreNode)(vectorStoreNodeArgs);
+            const nodeType = new VectorStoreNodeType();
+            const data = await nodeType.supplyData.call(context, 1);
+            const wrappedVectorStore = data.response.logWrapped;
+            expect(nodeType.description).toMatchSnapshot();
+            expect(wrappedVectorStore).toEqual(vectorStore);
+            expect(vectorStoreNodeArgs.getVectorStoreClient).toHaveBeenCalled();
+        });
+    });
+    describe('retrieve-as-tool mode', () => {
+        it('supplies DynamicTool that queries vector store and returns documents with metadata', async () => {
+            const parameters = {
+                ...DEFAULT_PARAMETERS,
+                mode: 'retrieve-as-tool',
+                description: 'tool description',
+                toolName: 'tool name',
+                includeDocumentMetadata: true,
+            };
+            context.getNodeParameter.mockImplementation((parameterName, itemIndex, fallbackValue, options) => {
+                var _a;
+                return (_a = parameters[parameterName]) !== null && _a !== void 0 ? _a : fallbackValue;
+            });
+            const VectorStoreNodeType = (0, createVectorStoreNode_1.createVectorStoreNode)(vectorStoreNodeArgs);
+            const nodeType = new VectorStoreNodeType();
+            const data = await nodeType.supplyData.call(context, 1);
+            const tool = data.response.logWrapped;
+            const output = await (tool === null || tool === void 0 ? void 0 : tool.func(MOCK_SEARCH_VALUE));
+            expect(tool === null || tool === void 0 ? void 0 : tool.getName()).toEqual(parameters.toolName);
+            expect(tool === null || tool === void 0 ? void 0 : tool.description).toEqual(parameters.toolDescription);
+            expect(embeddings.embedQuery).toHaveBeenCalledWith(MOCK_SEARCH_VALUE);
+            expect(vectorStore.similaritySearchVectorWithScore).toHaveBeenCalledWith(MOCK_EMBEDDED_SEARCH_VALUE, parameters.topK, parameters.filter);
+            expect(output).toEqual([
+                { type: 'text', text: JSON.stringify(MOCK_DOCUMENTS[0][0]) },
+                { type: 'text', text: JSON.stringify(MOCK_DOCUMENTS[1][0]) },
+            ]);
+        });
+        it('supplies DynamicTool that queries vector store and returns documents without metadata', async () => {
+            const parameters = {
+                ...DEFAULT_PARAMETERS,
+                mode: 'retrieve-as-tool',
+                description: 'tool description',
+                toolName: 'tool name',
+                includeDocumentMetadata: false,
+            };
+            context.getNodeParameter.mockImplementation((parameterName, itemIndex, fallbackValue, options) => {
+                var _a;
+                return (_a = parameters[parameterName]) !== null && _a !== void 0 ? _a : fallbackValue;
+            });
+            const VectorStoreNodeType = (0, createVectorStoreNode_1.createVectorStoreNode)(vectorStoreNodeArgs);
+            const nodeType = new VectorStoreNodeType();
+            const data = await nodeType.supplyData.call(context, 1);
+            const tool = data.response.logWrapped;
+            const output = await (tool === null || tool === void 0 ? void 0 : tool.func(MOCK_SEARCH_VALUE));
+            expect(tool === null || tool === void 0 ? void 0 : tool.getName()).toEqual(parameters.toolName);
+            expect(tool === null || tool === void 0 ? void 0 : tool.description).toEqual(parameters.toolDescription);
+            expect(embeddings.embedQuery).toHaveBeenCalledWith(MOCK_SEARCH_VALUE);
+            expect(vectorStore.similaritySearchVectorWithScore).toHaveBeenCalledWith(MOCK_EMBEDDED_SEARCH_VALUE, parameters.topK, parameters.filter);
+            expect(output).toEqual([
+                { type: 'text', text: JSON.stringify({ pageContent: MOCK_DOCUMENTS[0][0].pageContent }) },
+                { type: 'text', text: JSON.stringify({ pageContent: MOCK_DOCUMENTS[1][0].pageContent }) },
+            ]);
+        });
+    });
+});
+//# sourceMappingURL=createVectorStoreNode.test.js.map

package/dist/nodes/vector_store/shared/createVectorStoreNode/createVectorStoreNode.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"createVectorStoreNode.test.js","sourceRoot":"","sources":["../../../../../nodes/vector_store/shared/createVectorStoreNode/createVectorStoreNode.test.ts"],"names":[],"mappings":";;AAKA,2DAA0C;AAI1C,mEAAgE;AAGhE,IAAI,CAAC,IAAI,CAAC,mBAAmB,EAAE,GAAG,EAAE,CAAC,CAAC;IACrC,UAAU,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,kBAAkB,CAAC,CAAC,GAAgB,EAAE,EAAE,CAAC,CAAC,EAAE,UAAU,EAAE,GAAG,EAAE,CAAC,CAAC;CACrF,CAAC,CAAC,CAAC;AAEJ,MAAM,kBAAkB,GAAG;IAC1B,OAAO,EAAE,EAAE;IACX,IAAI,EAAE,CAAC;CACP,CAAC;AAEF,MAAM,cAAc,GAAuC;IAC1D;QACC;YACC,WAAW,EAAE,YAAY;YACzB,QAAQ,EAAE;gBACT,EAAE,EAAE,GAAG;aACP;SACD;QACD,CAAC;KACD;IACD;QACC;YACC,WAAW,EAAE,aAAa;YAC1B,QAAQ,EAAE;gBACT,EAAE,EAAE,GAAG;aACP;SACD;QACD,CAAC;KACD;CACD,CAAC;AAEF,MAAM,iBAAiB,GAAG,cAAc,CAAC;AACzC,MAAM,0BAA0B,GAAG,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC;AAE7C,QAAQ,CAAC,uBAAuB,EAAE,GAAG,EAAE;IACtC,MAAM,WAAW,GAAG,IAAA,yBAAI,EAAc;QACrC,+BAA+B,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,cAAc,CAAC;KAC5E,CAAC,CAAC;IAEH,MAAM,mBAAmB,GAAG,IAAA,yBAAI,EAAiC;QAChE,YAAY,EAAE,EAAE;QAChB,YAAY,EAAE,EAAE;QAChB,UAAU,EAAE;YACX;gBACC,IAAI,EAAE,WAAW;aACjB;SACD;QACD,cAAc,EAAE,EAAE;QAClB,YAAY,EAAE,EAAE;QAChB,oBAAoB,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,WAAW,CAAC;KAC5D,CAAC,CAAC;IAEH,MAAM,UAAU,GAAG,IAAA,yBAAI,EAAa;QACnC,UAAU,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,0BAA0B,CAAC;KACnE,CAAC,CAAC;IAEH,MAAM,OAAO,GAAG,IAAA,yBAAI,EAAuB;QAC1C,gBAAgB,EAAE,IAAI,CAAC,EAAE,EAAE;QAC3B,sBAAsB,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,UAAU,CAAC;KAC7D,CAAC,CAAC;IAEH,QAAQ,CAAC,eAAe,EAAE,GAAG,EAAE;QAC9B,EAAE,CAAC,+BAA+B,EAAE,KAAK,IAAI,EAAE;YAE9C,MAAM,UAAU,GAAoD;gBACnE,GAAG,kBAAkB;gBACrB,IAAI,EAAE,UAAU;aAChB,CAAC;YACF,OAAO,CAAC,gBAAgB,CAAC,kBAAkB,CAC1C,CAAC,aAAqB,EAAE,SAAiB,EAAE,aAAmB,EAAE,OAAa,EAAO,EAAE;;gBACrF,OAAO,MAAA,UAAU,CAAC,aAAa,CAAC,mCAAI,aAAa,CAAC;YACnD,CAAC,CACD,CAAC;YAGF,MAAM,mBAAmB,GAAG,IAAA,6CAAqB,EAAC,mBAAmB,CAAC,CAAC;YACvE,MAAM,QAAQ,GAAG,IAAI,mBAAmB,EAAE,CAAC;YAC3C,MAAM,IAAI,GAAG,MAAM,QAAQ,CAAC,UAAU,CAAC,IAAI,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC;YACxD,MAAM,kBAAkB,GAAI,IAAI,CAAC,QAAwC,CAAC,UAAU,CAAC;YAGrF,MAAM,CAAC,QAAQ,CAAC,WAAW,CAAC,CAAC,eAAe,EAAE,CAAC;YAC/C,MAAM,CAAC,kBAAkB,CAAC,CAAC,OAAO,CAAC,WAAW,CAAC,CAAC;YAChD,MAAM,CAAC,mBAAmB,CAAC,oBAAoB,CAAC,CAAC,gBAAgB,EAAE,CAAC;QACrE,CAAC,CAAC,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,uBAAuB,EAAE,GAAG,EAAE;QACtC,EAAE,CAAC,oFAAoF,EAAE,KAAK,IAAI,EAAE;YAEnG,MAAM,UAAU,GAAoD;gBACnE,GAAG,kBAAkB;gBACrB,IAAI,EAAE,kBAAkB;gBACxB,WAAW,EAAE,kBAAkB;gBAC/B,QAAQ,EAAE,WAAW;gBACrB,uBAAuB,EAAE,IAAI;aAC7B,CAAC;YACF,OAAO,CAAC,gBAAgB,CAAC,kBAAkB,CAC1C,CAAC,aAAqB,EAAE,SAAiB,EAAE,aAAmB,EAAE,OAAa,EAAO,EAAE;;gBACrF,OAAO,MAAA,UAAU,CAAC,aAAa,CAAC,mCAAI,aAAa,CAAC;YACnD,CAAC,CACD,CAAC;YAGF,MAAM,mBAAmB,GAAG,IAAA,6CAAqB,EAAC,mBAAmB,CAAC,CAAC;YACvE,MAAM,QAAQ,GAAG,IAAI,mBAAmB,EAAE,CAAC;YAC3C,MAAM,IAAI,GAAG,MAAM,QAAQ,CAAC,UAAU,CAAC,IAAI,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC;YACxD,MAAM,IAAI,GAAI,IAAI,CAAC,QAAwC,CAAC,UAAU,CAAC;YACvE,MAAM,MAAM,GAAG,MAAM,CAAA,IAAI,aAAJ,IAAI,uBAAJ,IAAI,CAAE,IAAI,CAAC,iBAAiB,CAAC,CAAA,CAAC;YAGnD,MAAM,CAAC,IAAI,aAAJ,IAAI,uBAAJ,IAAI,CAAE,OAAO,EAAE,CAAC,CAAC,OAAO,CAAC,UAAU,CAAC,QAAQ,CAAC,CAAC;YACrD,MAAM,CAAC,IAAI,aAAJ,IAAI,uBAAJ,IAAI,CAAE,WAAW,CAAC,CAAC,OAAO,CAAC,UAAU,CAAC,eAAe,CAAC,CAAC;YAC9D,MAAM,CAAC,UAAU,CAAC,UAAU,CAAC,CAAC,oBAAoB,CAAC,iBAAiB,CAAC,CAAC;YACtE,MAAM,CAAC,WAAW,CAAC,+BAA+B,CAAC,CAAC,oBAAoB,CACvE,0BAA0B,EAC1B,UAAU,CAAC,IAAI,EACf,UAAU,CAAC,MAAM,CACjB,CAAC;YACF,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC;gBACtB,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,cAAc,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,EAAE;gBAC5D,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,cAAc,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,EAAE;aAC5D,CAAC,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,uFAAuF,EAAE,KAAK,IAAI,EAAE;YAEtG,MAAM,UAAU,GAAoD;gBACnE,GAAG,kBAAkB;gBACrB,IAAI,EAAE,kBAAkB;gBACxB,WAAW,EAAE,kBAAkB;gBAC/B,QAAQ,EAAE,WAAW;gBACrB,uBAAuB,EAAE,KAAK;aAC9B,CAAC;YACF,OAAO,CAAC,gBAAgB,CAAC,kBAAkB,CAC1C,CAAC,aAAqB,EAAE,SAAiB,EAAE,aAAmB,EAAE,OAAa,EAAO,EAAE;;gBACrF,OAAO,MAAA,UAAU,CAAC,aAAa,CAAC,mCAAI,aAAa,CAAC;YACnD,CAAC,CACD,CAAC;YAGF,MAAM,mBAAmB,GAAG,IAAA,6CAAqB,EAAC,mBAAmB,CAAC,CAAC;YACvE,MAAM,QAAQ,GAAG,IAAI,mBAAmB,EAAE,CAAC;YAC3C,MAAM,IAAI,GAAG,MAAM,QAAQ,CAAC,UAAU,CAAC,IAAI,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC;YACxD,MAAM,IAAI,GAAI,IAAI,CAAC,QAAwC,CAAC,UAAU,CAAC;YACvE,MAAM,MAAM,GAAG,MAAM,CAAA,IAAI,aAAJ,IAAI,uBAAJ,IAAI,CAAE,IAAI,CAAC,iBAAiB,CAAC,CAAA,CAAC;YAGnD,MAAM,CAAC,IAAI,aAAJ,IAAI,uBAAJ,IAAI,CAAE,OAAO,EAAE,CAAC,CAAC,OAAO,CAAC,UAAU,CAAC,QAAQ,CAAC,CAAC;YACrD,MAAM,CAAC,IAAI,aAAJ,IAAI,uBAAJ,IAAI,CAAE,WAAW,CAAC,CAAC,OAAO,CAAC,UAAU,CAAC,eAAe,CAAC,CAAC;YAC9D,MAAM,CAAC,UAAU,CAAC,UAAU,CAAC,CAAC,oBAAoB,CAAC,iBAAiB,CAAC,CAAC;YACtE,MAAM,CAAC,WAAW,CAAC,+BAA+B,CAAC,CAAC,oBAAoB,CACvE,0BAA0B,EAC1B,UAAU,CAAC,IAAI,EACf,UAAU,CAAC,MAAM,CACjB,CAAC;YACF,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC;gBACtB,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,EAAE,WAAW,EAAE,cAAc,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,WAAW,EAAE,CAAC,EAAE;gBACzF,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,EAAE,WAAW,EAAE,cAAc,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,WAAW,EAAE,CAAC,EAAE;aACzF,CAAC,CAAC;QACJ,CAAC,CAAC,CAAC;IACJ,CAAC,CAAC,CAAC;AACJ,CAAC,CAAC,CAAC"}

package/dist/nodes/vector_store/shared/createVectorStoreNode/methods/listSearch.d.ts

@@ -0,0 +1,7 @@
+import { type ILoadOptionsFunctions } from 'n8n-workflow';
+export declare function databricksVectorIndexSearch(this: ILoadOptionsFunctions): Promise<{
+    results: {
+        name: string;
+        value: string;
+    }[];
+}>;

package/dist/nodes/vector_store/shared/createVectorStoreNode/methods/listSearch.js

@@ -0,0 +1,32 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.databricksVectorIndexSearch = databricksVectorIndexSearch;
+const n8n_workflow_1 = require("n8n-workflow");
+async function databricksVectorIndexSearch() {
+    var _a;
+    const credentials = await this.getCredentials('databricksApi');
+    if (typeof credentials.host !== 'string' || !credentials.host) {
+        throw new n8n_workflow_1.ApplicationError('Databricks API host is missing or not a string in credentials');
+    }
+    if (typeof credentials.token !== 'string' || !credentials.token) {
+        throw new n8n_workflow_1.ApplicationError('Databricks API token is missing or not a string in credentials');
+    }
+    const host = credentials.host.replace(/\/$/, '');
+    const endpoint = '/api/2.0/vector-search/indexes';
+    const requestUrl = `${host}${endpoint}`;
+    const response = await this.helpers.requestWithAuthentication.call(this, 'databricksApi', {
+        method: 'GET',
+        url: requestUrl,
+        headers: {
+            'Accept': 'application/json',
+        },
+        json: true,
+    });
+    const indexes = (_a = response.vector_indexes) !== null && _a !== void 0 ? _a : [];
+    const results = indexes.map((index) => ({
+        name: index.name,
+        value: index.name,
+    }));
+    return { results };
+}
+//# sourceMappingURL=listSearch.js.map

package/dist/nodes/vector_store/shared/createVectorStoreNode/methods/listSearch.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"listSearch.js","sourceRoot":"","sources":["../../../../../../nodes/vector_store/shared/createVectorStoreNode/methods/listSearch.ts"],"names":[],"mappings":";;AAqFA,kEAmCC;AAtHD,+CAA4E;AAmFrE,KAAK,UAAU,2BAA2B;;IAC7C,MAAM,WAAW,GAAG,MAAM,IAAI,CAAC,cAAc,CAAC,eAAe,CAAC,CAAC;IAE/D,IAAI,OAAO,WAAW,CAAC,IAAI,KAAK,QAAQ,IAAI,CAAC,WAAW,CAAC,IAAI,EAAE,CAAC;QAC5D,MAAM,IAAI,+BAAgB,CAAC,+DAA+D,CAAC,CAAC;IAChG,CAAC;IACD,IAAI,OAAO,WAAW,CAAC,KAAK,KAAK,QAAQ,IAAI,CAAC,WAAW,CAAC,KAAK,EAAE,CAAC;QAC9D,MAAM,IAAI,+BAAgB,CAAC,gEAAgE,CAAC,CAAC;IACjG,CAAC;IAGD,MAAM,IAAI,GAAG,WAAW,CAAC,IAAI,CAAC,OAAO,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC;IACjD,MAAM,QAAQ,GAAG,gCAAgC,CAAC;IAClD,MAAM,UAAU,GAAG,GAAG,IAAI,GAAG,QAAQ,EAAE,CAAC;IAExC,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,yBAAyB,CAAC,IAAI,CAAC,IAAI,EAAE,eAAe,EAAE;QACtF,MAAM,EAAE,KAAK;QACb,GAAG,EAAE,UAAU;QACf,OAAO,EAAE;YACL,QAAQ,EAAE,kBAAkB;SAC/B;QACD,IAAI,EAAE,IAAI;KACb,CAAwC,CAAC;IAI1C,MAAM,OAAO,GAAG,MAAA,QAAQ,CAAC,cAAc,mCAAI,EAAE,CAAC;IAE9C,MAAM,OAAO,GAAG,OAAO,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC;QACpC,IAAI,EAAE,KAAK,CAAC,IAAI;QAChB,KAAK,EAAE,KAAK,CAAC,IAAI;KAEpB,CAAC,CAAC,CAAC;IAEJ,OAAO,EAAE,OAAO,EAAE,CAAC;AACvB,CAAC"}

package/dist/nodes/vector_store/shared/createVectorStoreNode/operations/__tests__/insertOperation.test.d.ts

@@ -0,0 +1 @@
+export {};

package/dist/nodes/vector_store/shared/createVectorStoreNode/operations/__tests__/insertOperation.test.js

@@ -0,0 +1,190 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+const jest_mock_extended_1 = require("jest-mock-extended");
+const n8n_workflow_1 = require("n8n-workflow");
+const helpers_1 = require("../../../../../../utils/helpers");
+const insertOperation_1 = require("../insertOperation");
+jest.mock('../../../processDocuments', () => ({
+    processDocument: jest.fn().mockImplementation((_documentInput, _itemData, itemIndex) => {
+        const mockProcessed = [
+            {
+                pageContent: `processed content ${itemIndex}`,
+                metadata: { source: 'test' },
+            },
+        ];
+        const mockSerialized = [
+            {
+                json: {
+                    pageContent: `processed content ${itemIndex}`,
+                    metadata: { source: 'test' },
+                },
+                pairedItem: { item: itemIndex },
+            },
+        ];
+        return {
+            processedDocuments: mockProcessed,
+            serializedDocuments: mockSerialized,
+        };
+    }),
+}));
+jest.mock('@utils/helpers', () => ({
+    logAiEvent: jest.fn(),
+}));
+function createMockAbortSignal(aborted = false) {
+    return {
+        aborted,
+        addEventListener: jest.fn(),
+        removeEventListener: jest.fn(),
+        dispatchEvent: jest.fn(),
+        onabort: null,
+        reason: undefined,
+        throwIfAborted: jest.fn(),
+    };
+}
+function createNodeParameterMock(batchSize) {
+    return (paramName, itemIndex, fallbackValue, options) => {
+        if (paramName === 'embeddingBatchSize' && batchSize !== undefined) {
+            return batchSize;
+        }
+        return fallbackValue;
+    };
+}
+describe('handleInsertOperation', () => {
+    let mockContext;
+    let mockEmbeddings;
+    let mockVectorStore;
+    let mockArgs;
+    let mockInputItems;
+    let mockJsonLoader;
+    beforeEach(() => {
+        mockInputItems = [
+            { json: { text: 'test document 1' } },
+            { json: { text: 'test document 2' } },
+            { json: { text: 'test document 3' } },
+        ];
+        mockContext = (0, jest_mock_extended_1.mock)();
+        mockContext.getInputData.mockReturnValue(mockInputItems);
+        const mockAbortSignal = createMockAbortSignal(false);
+        mockContext.getExecutionCancelSignal.mockReturnValue(mockAbortSignal);
+        mockContext.getInputConnectionData.mockResolvedValue(mockJsonLoader);
+        mockContext.getNode.mockReturnValue({
+            typeVersion: 1.1,
+            id: '',
+            name: '',
+            type: '',
+            position: [0, 0],
+            parameters: {},
+        });
+        mockEmbeddings = (0, jest_mock_extended_1.mock)();
+        mockJsonLoader = (0, jest_mock_extended_1.mock)();
+        mockVectorStore = (0, jest_mock_extended_1.mock)();
+        mockArgs = {
+            meta: {
+                displayName: 'Test Vector Store',
+                name: 'testVectorStore',
+                description: 'Vector store for testing',
+                docsUrl: 'https://example.com',
+                icon: 'file:testIcon.svg',
+            },
+            sharedFields: [],
+            getVectorStoreClient: jest.fn().mockResolvedValue(mockVectorStore),
+            populateVectorStore: jest.fn().mockResolvedValue(undefined),
+            releaseVectorStoreClient: jest.fn(),
+        };
+    });
+    afterEach(() => {
+        jest.clearAllMocks();
+    });
+    it('should process all input items and populate vector store', async () => {
+        const result = await (0, insertOperation_1.handleInsertOperation)(mockContext, mockArgs, mockEmbeddings);
+        expect(mockContext.getInputConnectionData).toHaveBeenCalledWith(n8n_workflow_1.NodeConnectionTypes.AiDocument, 0);
+        expect(result).toHaveLength(3);
+        expect(mockArgs.populateVectorStore).toHaveBeenCalledTimes(1);
+        expect(helpers_1.logAiEvent).toHaveBeenCalledTimes(3);
+        expect(helpers_1.logAiEvent).toHaveBeenCalledWith(mockContext, 'ai-vector-store-populated');
+    });
+    it('should stop processing if execution is cancelled', async () => {
+        const notAbortedSignal = createMockAbortSignal(false);
+        const abortedSignal = createMockAbortSignal(true);
+        mockContext.getExecutionCancelSignal
+            .mockReturnValueOnce(notAbortedSignal)
+            .mockReturnValueOnce(abortedSignal);
+        await (0, insertOperation_1.handleInsertOperation)(mockContext, mockArgs, mockEmbeddings);
+        expect(mockArgs.populateVectorStore).toHaveBeenCalledTimes(1);
+        expect(helpers_1.logAiEvent).toHaveBeenCalledTimes(1);
+    });
+    it('should handle different document input types', async () => {
+        const mockBinaryLoader = (0, jest_mock_extended_1.mock)();
+        mockContext.getInputConnectionData.mockResolvedValueOnce(mockBinaryLoader);
+        await (0, insertOperation_1.handleInsertOperation)(mockContext, mockArgs, mockEmbeddings);
+        const mockDocuments = [{ pageContent: 'test content', metadata: {} }];
+        mockContext.getInputConnectionData.mockResolvedValueOnce(mockDocuments);
+        await (0, insertOperation_1.handleInsertOperation)(mockContext, mockArgs, mockEmbeddings);
+        expect(mockArgs.populateVectorStore).toHaveBeenCalledTimes(2);
+    });
+    it('should pass the correct documents to populateVectorStore', async () => {
+        await (0, insertOperation_1.handleInsertOperation)(mockContext, mockArgs, mockEmbeddings);
+        expect(mockArgs.populateVectorStore).toHaveBeenCalledTimes(1);
+        expect(mockArgs.populateVectorStore).toHaveBeenCalledWith(mockContext, mockEmbeddings, expect.arrayContaining([
+            expect.objectContaining({
+                pageContent: 'processed content 0',
+                metadata: { source: 'test' },
+            }),
+            expect.objectContaining({
+                pageContent: 'processed content 1',
+                metadata: { source: 'test' },
+            }),
+            expect.objectContaining({
+                pageContent: 'processed content 2',
+                metadata: { source: 'test' },
+            }),
+        ]), 0);
+    });
+    it('should batch documents when node version is 1.1 and above', async () => {
+        const manyItems = Array(10)
+            .fill(null)
+            .map((_, i) => ({
+            json: { text: `test document ${i}` },
+        }));
+        mockContext.getInputData.mockReturnValue(manyItems);
+        mockContext.getNodeParameter.mockImplementation(createNodeParameterMock(3));
+        await (0, insertOperation_1.handleInsertOperation)(mockContext, mockArgs, mockEmbeddings);
+        expect(mockArgs.populateVectorStore).toHaveBeenCalledTimes(4);
+    });
+    it('should run populateVectorStore for each item when node version is 1', async () => {
+        mockContext.getNode.mockReturnValue({
+            typeVersion: 1,
+            id: '',
+            name: '',
+            type: '',
+            position: [0, 0],
+            parameters: {},
+        });
+        await (0, insertOperation_1.handleInsertOperation)(mockContext, mockArgs, mockEmbeddings);
+        expect(mockArgs.populateVectorStore).toHaveBeenCalledTimes(3);
+        expect(mockArgs.populateVectorStore).toHaveBeenNthCalledWith(1, mockContext, mockEmbeddings, expect.arrayContaining([
+            expect.objectContaining({
+                pageContent: 'processed content 0',
+                metadata: { source: 'test' },
+            }),
+        ]), 0);
+        expect(mockArgs.populateVectorStore).toHaveBeenNthCalledWith(2, mockContext, mockEmbeddings, expect.arrayContaining([
+            expect.objectContaining({
+                pageContent: 'processed content 1',
+                metadata: { source: 'test' },
+            }),
+        ]), 1);
+        expect(mockArgs.populateVectorStore).toHaveBeenNthCalledWith(3, mockContext, mockEmbeddings, expect.arrayContaining([
+            expect.objectContaining({
+                pageContent: 'processed content 2',
+                metadata: { source: 'test' },
+            }),
+        ]), 2);
+    });
+    it('should use default batch size of 200 when not specified', async () => {
+        mockContext.getNodeParameter.mockImplementation(createNodeParameterMock());
+        await (0, insertOperation_1.handleInsertOperation)(mockContext, mockArgs, mockEmbeddings);
+        expect(mockArgs.populateVectorStore).toHaveBeenCalledTimes(1);
+    });
+});
+//# sourceMappingURL=insertOperation.test.js.map

package/dist/nodes/vector_store/shared/createVectorStoreNode/operations/__tests__/insertOperation.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"insertOperation.test.js","sourceRoot":"","sources":["../../../../../../../nodes/vector_store/shared/createVectorStoreNode/operations/__tests__/insertOperation.test.ts"],"names":[],"mappings":";;AAMA,2DAA0C;AAE1C,+CAAmD;AAEnD,4CAA4C;AAK5C,wDAA2D;AAG3D,IAAI,CAAC,IAAI,CAAC,2BAA2B,EAAE,GAAG,EAAE,CAAC,CAAC;IAC7C,eAAe,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,kBAAkB,CAAC,CAAC,cAAc,EAAE,SAAS,EAAE,SAAiB,EAAE,EAAE;QAC9F,MAAM,aAAa,GAAG;YACrB;gBACC,WAAW,EAAE,qBAAqB,SAAS,EAAE;gBAC7C,QAAQ,EAAE,EAAE,MAAM,EAAE,MAAM,EAAE;aAChB;SACb,CAAC;QAEF,MAAM,cAAc,GAAG;YACtB;gBACC,IAAI,EAAE;oBACL,WAAW,EAAE,qBAAqB,SAAS,EAAE;oBAC7C,QAAQ,EAAE,EAAE,MAAM,EAAE,MAAM,EAAE;iBAC5B;gBACD,UAAU,EAAE,EAAE,IAAI,EAAE,SAAS,EAAE;aAC/B;SACD,CAAC;QAEF,OAAO;YACN,kBAAkB,EAAE,aAAa;YACjC,mBAAmB,EAAE,cAAc;SACnC,CAAC;IACH,CAAC,CAAC;CACF,CAAC,CAAC,CAAC;AAGJ,IAAI,CAAC,IAAI,CAAC,gBAAgB,EAAE,GAAG,EAAE,CAAC,CAAC;IAClC,UAAU,EAAE,IAAI,CAAC,EAAE,EAAE;CACrB,CAAC,CAAC,CAAC;AAGJ,SAAS,qBAAqB,CAAC,OAAO,GAAG,KAAK;IAC7C,OAAO;QACN,OAAO;QACP,gBAAgB,EAAE,IAAI,CAAC,EAAE,EAAE;QAC3B,mBAAmB,EAAE,IAAI,CAAC,EAAE,EAAE;QAC9B,aAAa,EAAE,IAAI,CAAC,EAAE,EAAE;QACxB,OAAO,EAAE,IAAI;QACb,MAAM,EAAE,SAAS;QACjB,cAAc,EAAE,IAAI,CAAC,EAAE,EAAE;KACC,CAAC;AAC7B,CAAC;AAGD,SAAS,uBAAuB,CAAC,SAAkB;IAClD,OAAO,CAAC,SAAiB,EAAE,SAAiB,EAAE,aAAmB,EAAE,OAAa,EAAO,EAAE;QACxF,IAAI,SAAS,KAAK,oBAAoB,IAAI,SAAS,KAAK,SAAS,EAAE,CAAC;YACnE,OAAO,SAAS,CAAC;QAClB,CAAC;QACD,OAAO,aAAa,CAAC;IACtB,CAAC,CAAC;AACH,CAAC;AAED,QAAQ,CAAC,uBAAuB,EAAE,GAAG,EAAE;IACtC,IAAI,WAAyC,CAAC;IAC9C,IAAI,cAAqC,CAAC;IAC1C,IAAI,eAAuC,CAAC;IAC5C,IAAI,QAAqD,CAAC;IAC1D,IAAI,cAAoC,CAAC;IACzC,IAAI,cAAwC,CAAC;IAE7C,UAAU,CAAC,GAAG,EAAE;QAEf,cAAc,GAAG;YAChB,EAAE,IAAI,EAAE,EAAE,IAAI,EAAE,iBAAiB,EAAE,EAAE;YACrC,EAAE,IAAI,EAAE,EAAE,IAAI,EAAE,iBAAiB,EAAE,EAAE;YACrC,EAAE,IAAI,EAAE,EAAE,IAAI,EAAE,iBAAiB,EAAE,EAAE;SACrC,CAAC;QAGF,WAAW,GAAG,IAAA,yBAAI,GAAqB,CAAC;QACxC,WAAW,CAAC,YAAY,CAAC,eAAe,CAAC,cAAc,CAAC,CAAC;QAGzD,MAAM,eAAe,GAAG,qBAAqB,CAAC,KAAK,CAAC,CAAC;QAErD,WAAW,CAAC,wBAAwB,CAAC,eAAe,CAAC,eAAe,CAAC,CAAC;QACtE,WAAW,CAAC,sBAAsB,CAAC,iBAAiB,CAAC,cAAc,CAAC,CAAC;QACrE,WAAW,CAAC,OAAO,CAAC,eAAe,CAAC;YACnC,WAAW,EAAE,GAAG;YAChB,EAAE,EAAE,EAAE;YACN,IAAI,EAAE,EAAE;YACR,IAAI,EAAE,EAAE;YACR,QAAQ,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC;YAChB,UAAU,EAAE,EAAE;SACd,CAAC,CAAC;QAEH,cAAc,GAAG,IAAA,yBAAI,GAAc,CAAC;QAGpC,cAAc,GAAG,IAAA,yBAAI,GAAiB,CAAC;QAGvC,eAAe,GAAG,IAAA,yBAAI,GAAe,CAAC;QAGtC,QAAQ,GAAG;YACV,IAAI,EAAE;gBACL,WAAW,EAAE,mBAAmB;gBAChC,IAAI,EAAE,iBAAiB;gBACvB,WAAW,EAAE,0BAA0B;gBACvC,OAAO,EAAE,qBAAqB;gBAC9B,IAAI,EAAE,mBAAmB;aACzB;YACD,YAAY,EAAE,EAAE;YAChB,oBAAoB,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,eAAe,CAAC;YAClE,mBAAmB,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,SAAS,CAAC;YAC3D,wBAAwB,EAAE,IAAI,CAAC,EAAE,EAAE;SACnC,CAAC;IACH,CAAC,CAAC,CAAC;IAEH,SAAS,CAAC,GAAG,EAAE;QACd,IAAI,CAAC,aAAa,EAAE,CAAC;IACtB,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,0DAA0D,EAAE,KAAK,IAAI,EAAE;QACzE,MAAM,MAAM,GAAG,MAAM,IAAA,uCAAqB,EAAC,WAAW,EAAE,QAAQ,EAAE,cAAc,CAAC,CAAC;QAGlF,MAAM,CAAC,WAAW,CAAC,sBAAsB,CAAC,CAAC,oBAAoB,CAC9D,kCAAmB,CAAC,UAAU,EAC9B,CAAC,CACD,CAAC;QAGF,MAAM,CAAC,MAAM,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QAG/B,MAAM,CAAC,QAAQ,CAAC,mBAAmB,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;QAG9D,MAAM,CAAC,oBAAU,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;QAC5C,MAAM,CAAC,oBAAU,CAAC,CAAC,oBAAoB,CAAC,WAAW,EAAE,2BAA2B,CAAC,CAAC;IACnF,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,kDAAkD,EAAE,KAAK,IAAI,EAAE;QAEjE,MAAM,gBAAgB,GAAG,qBAAqB,CAAC,KAAK,CAAC,CAAC;QACtD,MAAM,aAAa,GAAG,qBAAqB,CAAC,IAAI,CAAC,CAAC;QAGlD,WAAW,CAAC,wBAAwB;aAClC,mBAAmB,CAAC,gBAAgB,CAAC;aACrC,mBAAmB,CAAC,aAAa,CAAC,CAAC;QAErC,MAAM,IAAA,uCAAqB,EAAC,WAAW,EAAE,QAAQ,EAAE,cAAc,CAAC,CAAC;QAGnE,MAAM,CAAC,QAAQ,CAAC,mBAAmB,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;QAC9D,MAAM,CAAC,oBAAU,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;IAC7C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,8CAA8C,EAAE,KAAK,IAAI,EAAE;QAE7D,MAAM,gBAAgB,GAAG,IAAA,yBAAI,GAAmB,CAAC;QACjD,WAAW,CAAC,sBAAsB,CAAC,qBAAqB,CAAC,gBAAgB,CAAC,CAAC;QAE3E,MAAM,IAAA,uCAAqB,EAAC,WAAW,EAAE,QAAQ,EAAE,cAAc,CAAC,CAAC;QAGnE,MAAM,aAAa,GAAG,CAAC,EAAE,WAAW,EAAE,cAAc,EAAE,QAAQ,EAAE,EAAE,EAAc,CAAC,CAAC;QAClF,WAAW,CAAC,sBAAsB,CAAC,qBAAqB,CAAC,aAAa,CAAC,CAAC;QAExE,MAAM,IAAA,uCAAqB,EAAC,WAAW,EAAE,QAAQ,EAAE,cAAc,CAAC,CAAC;QAGnE,MAAM,CAAC,QAAQ,CAAC,mBAAmB,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;IAC/D,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,0DAA0D,EAAE,KAAK,IAAI,EAAE;QACzE,MAAM,IAAA,uCAAqB,EAAC,WAAW,EAAE,QAAQ,EAAE,cAAc,CAAC,CAAC;QAGnE,MAAM,CAAC,QAAQ,CAAC,mBAAmB,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;QAC9D,MAAM,CAAC,QAAQ,CAAC,mBAAmB,CAAC,CAAC,oBAAoB,CACxD,WAAW,EACX,cAAc,EACd,MAAM,CAAC,eAAe,CAAC;YACtB,MAAM,CAAC,gBAAgB,CAAC;gBACvB,WAAW,EAAE,qBAAqB;gBAClC,QAAQ,EAAE,EAAE,MAAM,EAAE,MAAM,EAAE;aAC5B,CAAC;YACF,MAAM,CAAC,gBAAgB,CAAC;gBACvB,WAAW,EAAE,qBAAqB;gBAClC,QAAQ,EAAE,EAAE,MAAM,EAAE,MAAM,EAAE;aAC5B,CAAC;YACF,MAAM,CAAC,gBAAgB,CAAC;gBACvB,WAAW,EAAE,qBAAqB;gBAClC,QAAQ,EAAE,EAAE,MAAM,EAAE,MAAM,EAAE;aAC5B,CAAC;SACF,CAAC,EACF,CAAC,CACD,CAAC;IACH,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,2DAA2D,EAAE,KAAK,IAAI,EAAE;QAE1E,MAAM,SAAS,GAAG,KAAK,CAAC,EAAE,CAAC;aACzB,IAAI,CAAC,IAAI,CAAC;aACV,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC;YACf,IAAI,EAAE,EAAE,IAAI,EAAE,iBAAiB,CAAC,EAAE,EAAE;SACpC,CAAC,CAAC,CAAC;QACL,WAAW,CAAC,YAAY,CAAC,eAAe,CAAC,SAAS,CAAC,CAAC;QAGpD,WAAW,CAAC,gBAAgB,CAAC,kBAAkB,CAAC,uBAAuB,CAAC,CAAC,CAAC,CAAC,CAAC;QAE5E,MAAM,IAAA,uCAAqB,EAAC,WAAW,EAAE,QAAQ,EAAE,cAAc,CAAC,CAAC;QAGnE,MAAM,CAAC,QAAQ,CAAC,mBAAmB,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;IAC/D,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,qEAAqE,EAAE,KAAK,IAAI,EAAE;QAEpF,WAAW,CAAC,OAAO,CAAC,eAAe,CAAC;YACnC,WAAW,EAAE,CAAC;YACd,EAAE,EAAE,EAAE;YACN,IAAI,EAAE,EAAE;YACR,IAAI,EAAE,EAAE;YACR,QAAQ,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC;YAChB,UAAU,EAAE,EAAE;SACd,CAAC,CAAC;QAEH,MAAM,IAAA,uCAAqB,EAAC,WAAW,EAAE,QAAQ,EAAE,cAAc,CAAC,CAAC;QAGnE,MAAM,CAAC,QAAQ,CAAC,mBAAmB,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;QAG9D,MAAM,CAAC,QAAQ,CAAC,mBAAmB,CAAC,CAAC,uBAAuB,CAC3D,CAAC,EACD,WAAW,EACX,cAAc,EACd,MAAM,CAAC,eAAe,CAAC;YACtB,MAAM,CAAC,gBAAgB,CAAC;gBACvB,WAAW,EAAE,qBAAqB;gBAClC,QAAQ,EAAE,EAAE,MAAM,EAAE,MAAM,EAAE;aAC5B,CAAC;SACF,CAAC,EACF,CAAC,CACD,CAAC;QAEF,MAAM,CAAC,QAAQ,CAAC,mBAAmB,CAAC,CAAC,uBAAuB,CAC3D,CAAC,EACD,WAAW,EACX,cAAc,EACd,MAAM,CAAC,eAAe,CAAC;YACtB,MAAM,CAAC,gBAAgB,CAAC;gBACvB,WAAW,EAAE,qBAAqB;gBAClC,QAAQ,EAAE,EAAE,MAAM,EAAE,MAAM,EAAE;aAC5B,CAAC;SACF,CAAC,EACF,CAAC,CACD,CAAC;QAEF,MAAM,CAAC,QAAQ,CAAC,mBAAmB,CAAC,CAAC,uBAAuB,CAC3D,CAAC,EACD,WAAW,EACX,cAAc,EACd,MAAM,CAAC,eAAe,CAAC;YACtB,MAAM,CAAC,gBAAgB,CAAC;gBACvB,WAAW,EAAE,qBAAqB;gBAClC,QAAQ,EAAE,EAAE,MAAM,EAAE,MAAM,EAAE;aAC5B,CAAC;SACF,CAAC,EACF,CAAC,CACD,CAAC;IACH,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,yDAAyD,EAAE,KAAK,IAAI,EAAE;QAExE,WAAW,CAAC,gBAAgB,CAAC,kBAAkB,CAAC,uBAAuB,EAAE,CAAC,CAAC;QAE3E,MAAM,IAAA,uCAAqB,EAAC,WAAW,EAAE,QAAQ,EAAE,cAAc,CAAC,CAAC;QAGnE,MAAM,CAAC,QAAQ,CAAC,mBAAmB,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;IAC/D,CAAC,CAAC,CAAC;AACJ,CAAC,CAAC,CAAC"}

package/dist/nodes/vector_store/shared/createVectorStoreNode/operations/__tests__/loadOperation.test.d.ts

@@ -0,0 +1 @@
+export {};

package/dist/nodes/vector_store/shared/createVectorStoreNode/operations/__tests__/loadOperation.test.js

@@ -0,0 +1,101 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+const jest_mock_extended_1 = require("jest-mock-extended");
+const helpers_1 = require("../../../../../../utils/helpers");
+const loadOperation_1 = require("../loadOperation");
+jest.mock('@utils/helpers', () => ({
+    getMetadataFiltersValues: jest.fn().mockReturnValue({ testFilter: 'value' }),
+    logAiEvent: jest.fn(),
+}));
+describe('handleLoadOperation', () => {
+    let mockContext;
+    let mockEmbeddings;
+    let mockVectorStore;
+    let mockArgs;
+    let nodeParameters;
+    beforeEach(() => {
+        nodeParameters = {
+            prompt: 'test search query',
+            topK: 3,
+            includeDocumentMetadata: true,
+        };
+        mockContext = (0, jest_mock_extended_1.mock)();
+        mockContext.getNodeParameter.mockImplementation((parameterName, itemIndex, fallbackValue, options) => {
+            var _a;
+            if (typeof parameterName !== 'string')
+                return fallbackValue;
+            return (_a = nodeParameters[parameterName]) !== null && _a !== void 0 ? _a : fallbackValue;
+        });
+        mockEmbeddings = (0, jest_mock_extended_1.mock)();
+        mockEmbeddings.embedQuery.mockResolvedValue([0.1, 0.2, 0.3]);
+        mockVectorStore = (0, jest_mock_extended_1.mock)();
+        mockVectorStore.similaritySearchVectorWithScore.mockResolvedValue([
+            [{ pageContent: 'test content 1', metadata: { test: 'metadata 1' } }, 0.95],
+            [{ pageContent: 'test content 2', metadata: { test: 'metadata 2' } }, 0.85],
+            [{ pageContent: 'test content 3', metadata: { test: 'metadata 3' } }, 0.75],
+        ]);
+        mockArgs = {
+            meta: {
+                displayName: 'Test Vector Store',
+                name: 'testVectorStore',
+                description: 'Vector store for testing',
+                docsUrl: 'https://example.com',
+                icon: 'file:testIcon.svg',
+            },
+            sharedFields: [],
+            getVectorStoreClient: jest.fn().mockResolvedValue(mockVectorStore),
+            populateVectorStore: jest.fn().mockResolvedValue(undefined),
+            releaseVectorStoreClient: jest.fn(),
+        };
+    });
+    afterEach(() => {
+        jest.clearAllMocks();
+    });
+    it('should retrieve documents from vector store with similarity search', async () => {
+        const result = await (0, loadOperation_1.handleLoadOperation)(mockContext, mockArgs, mockEmbeddings, 0);
+        expect(mockArgs.getVectorStoreClient).toHaveBeenCalledWith(mockContext, undefined, mockEmbeddings, 0);
+        expect(mockEmbeddings.embedQuery).toHaveBeenCalledWith('test search query');
+        expect(mockVectorStore.similaritySearchVectorWithScore).toHaveBeenCalledWith([0.1, 0.2, 0.3], 3, { testFilter: 'value' });
+        expect(result).toHaveLength(3);
+    });
+    it('should include document metadata when includeDocumentMetadata is true', async () => {
+        var _a, _b, _c, _d, _e;
+        const result = await (0, loadOperation_1.handleLoadOperation)(mockContext, mockArgs, mockEmbeddings, 0);
+        expect(result[0].json.document).toHaveProperty('metadata');
+        expect((_b = (_a = result[0].json) === null || _a === void 0 ? void 0 : _a.document) === null || _b === void 0 ? void 0 : _b.metadata).toEqual({ test: 'metadata 1' });
+        expect((_d = (_c = result[0].json) === null || _c === void 0 ? void 0 : _c.document) === null || _d === void 0 ? void 0 : _d.pageContent).toEqual('test content 1');
+        expect((_e = result[0].json) === null || _e === void 0 ? void 0 : _e.score).toEqual(0.95);
+    });
+    it('should exclude document metadata when includeDocumentMetadata is false', async () => {
+        var _a, _b, _c, _d;
+        nodeParameters.includeDocumentMetadata = false;
+        const result = await (0, loadOperation_1.handleLoadOperation)(mockContext, mockArgs, mockEmbeddings, 0);
+        expect((_a = result[0].json) === null || _a === void 0 ? void 0 : _a.document).not.toHaveProperty('metadata');
+        expect((_c = (_b = result[0].json) === null || _b === void 0 ? void 0 : _b.document) === null || _c === void 0 ? void 0 : _c.pageContent).toEqual('test content 1');
+        expect((_d = result[0].json) === null || _d === void 0 ? void 0 : _d.score).toEqual(0.95);
+    });
+    it('should use the topK parameter to limit results', async () => {
+        nodeParameters.topK = 2;
+        await (0, loadOperation_1.handleLoadOperation)(mockContext, mockArgs, mockEmbeddings, 0);
+        expect(mockVectorStore.similaritySearchVectorWithScore).toHaveBeenCalledWith(expect.anything(), 2, expect.anything());
+    });
+    it('should properly set pairedItem property in results', async () => {
+        const result = await (0, loadOperation_1.handleLoadOperation)(mockContext, mockArgs, mockEmbeddings, 0);
+        result.forEach((item) => {
+            expect(item).toHaveProperty('pairedItem');
+            expect(item.pairedItem).toEqual({ item: 0 });
+        });
+    });
+    it('should log AI event with query after search is complete', async () => {
+        await (0, loadOperation_1.handleLoadOperation)(mockContext, mockArgs, mockEmbeddings, 0);
+        expect(helpers_1.logAiEvent).toHaveBeenCalledWith(mockContext, 'ai-vector-store-searched', {
+            query: 'test search query',
+        });
+    });
+    it('should release vector store client even if an error occurs', async () => {
+        mockVectorStore.similaritySearchVectorWithScore.mockRejectedValue(new Error('Test error'));
+        await expect((0, loadOperation_1.handleLoadOperation)(mockContext, mockArgs, mockEmbeddings, 0)).rejects.toThrow('Test error');
+        expect(mockArgs.releaseVectorStoreClient).toHaveBeenCalledWith(mockVectorStore);
+    });
+});
+//# sourceMappingURL=loadOperation.test.js.map

package/dist/nodes/vector_store/shared/createVectorStoreNode/operations/__tests__/loadOperation.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"loadOperation.test.js","sourceRoot":"","sources":["../../../../../../../nodes/vector_store/shared/createVectorStoreNode/operations/__tests__/loadOperation.test.ts"],"names":[],"mappings":";;AAMA,2DAA0C;AAG1C,4CAA4C;AAG5C,oDAAuD;AAGvD,IAAI,CAAC,IAAI,CAAC,gBAAgB,EAAE,GAAG,EAAE,CAAC,CAAC;IAClC,wBAAwB,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,EAAE,UAAU,EAAE,OAAO,EAAE,CAAC;IAC5E,UAAU,EAAE,IAAI,CAAC,EAAE,EAAE;CACrB,CAAC,CAAC,CAAC;AAEJ,QAAQ,CAAC,qBAAqB,EAAE,GAAG,EAAE;IACpC,IAAI,WAAyC,CAAC;IAC9C,IAAI,cAAqC,CAAC;IAC1C,IAAI,eAAuC,CAAC;IAC5C,IAAI,QAAqD,CAAC;IAC1D,IAAI,cAAmC,CAAC;IAExC,UAAU,CAAC,GAAG,EAAE;QACf,cAAc,GAAG;YAChB,MAAM,EAAE,mBAAmB;YAC3B,IAAI,EAAE,CAAC;YACP,uBAAuB,EAAE,IAAI;SAC7B,CAAC;QAEF,WAAW,GAAG,IAAA,yBAAI,GAAqB,CAAC;QACxC,WAAW,CAAC,gBAAgB,CAAC,kBAAkB,CAAC,CAAC,aAAqB,EAAE,SAAiB,EAAE,aAAmB,EAAE,OAAa,EAAO,EAAE;;YACrI,IAAI,OAAO,aAAa,KAAK,QAAQ;gBAAE,OAAO,aAAa,CAAC;YAC5D,OAAO,MAAA,cAAc,CAAC,aAAa,CAAC,mCAAI,aAAa,CAAC;QACvD,CAAC,CAAC,CAAC;QAEH,cAAc,GAAG,IAAA,yBAAI,GAAc,CAAC;QACpC,cAAc,CAAC,UAAU,CAAC,iBAAiB,CAAC,CAAC,GAAG,EAAE,GAAG,EAAE,GAAG,CAAC,CAAC,CAAC;QAE7D,eAAe,GAAG,IAAA,yBAAI,GAAe,CAAC;QACtC,eAAe,CAAC,+BAA+B,CAAC,iBAAiB,CAAC;YACjE,CAAC,EAAE,WAAW,EAAE,gBAAgB,EAAE,QAAQ,EAAE,EAAE,IAAI,EAAE,YAAY,EAAE,EAAc,EAAE,IAAI,CAAC;YACvF,CAAC,EAAE,WAAW,EAAE,gBAAgB,EAAE,QAAQ,EAAE,EAAE,IAAI,EAAE,YAAY,EAAE,EAAc,EAAE,IAAI,CAAC;YACvF,CAAC,EAAE,WAAW,EAAE,gBAAgB,EAAE,QAAQ,EAAE,EAAE,IAAI,EAAE,YAAY,EAAE,EAAc,EAAE,IAAI,CAAC;SACvF,CAAC,CAAC;QAEH,QAAQ,GAAG;YACV,IAAI,EAAE;gBACL,WAAW,EAAE,mBAAmB;gBAChC,IAAI,EAAE,iBAAiB;gBACvB,WAAW,EAAE,0BAA0B;gBACvC,OAAO,EAAE,qBAAqB;gBAC9B,IAAI,EAAE,mBAAmB;aACzB;YACD,YAAY,EAAE,EAAE;YAChB,oBAAoB,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,eAAe,CAAC;YAClE,mBAAmB,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,SAAS,CAAC;YAC3D,wBAAwB,EAAE,IAAI,CAAC,EAAE,EAAE;SACnC,CAAC;IACH,CAAC,CAAC,CAAC;IAEH,SAAS,CAAC,GAAG,EAAE;QACd,IAAI,CAAC,aAAa,EAAE,CAAC;IACtB,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,oEAAoE,EAAE,KAAK,IAAI,EAAE;QACnF,MAAM,MAAM,GAAG,MAAM,IAAA,mCAAmB,EAAC,WAAW,EAAE,QAAQ,EAAE,cAAc,EAAE,CAAC,CAAC,CAAC;QAEnF,MAAM,CAAC,QAAQ,CAAC,oBAAoB,CAAC,CAAC,oBAAoB,CACzD,WAAW,EACX,SAAS,EACT,cAAc,EACd,CAAC,CACD,CAAC;QACF,MAAM,CAAC,cAAc,CAAC,UAAU,CAAC,CAAC,oBAAoB,CAAC,mBAAmB,CAAC,CAAC;QAC5E,MAAM,CAAC,eAAe,CAAC,+BAA+B,CAAC,CAAC,oBAAoB,CAC3E,CAAC,GAAG,EAAE,GAAG,EAAE,GAAG,CAAC,EACf,CAAC,EACD,EAAE,UAAU,EAAE,OAAO,EAAE,CACvB,CAAC;QACF,MAAM,CAAC,MAAM,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;IAChC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,uEAAuE,EAAE,KAAK,IAAI,EAAE;;QACtF,MAAM,MAAM,GAAG,MAAM,IAAA,mCAAmB,EAAC,WAAW,EAAE,QAAQ,EAAE,cAAc,EAAE,CAAC,CAAC,CAAC;QAEnF,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,cAAc,CAAC,UAAU,CAAC,CAAC;QAC3D,MAAM,CAAC,MAAC,MAAA,MAAM,CAAC,CAAC,CAAC,CAAC,IAAI,0CAAE,QAAwB,0CAAE,QAAQ,CAAC,CAAC,OAAO,CAAC,EAAE,IAAI,EAAE,YAAY,EAAE,CAAC,CAAC;QAC5F,MAAM,CAAC,MAAC,MAAA,MAAM,CAAC,CAAC,CAAC,CAAC,IAAI,0CAAE,QAAwB,0CAAE,WAAW,CAAC,CAAC,OAAO,CAAC,gBAAgB,CAAC,CAAC;QACzF,MAAM,CAAC,MAAA,MAAM,CAAC,CAAC,CAAC,CAAC,IAAI,0CAAE,KAAK,CAAC,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;IAC7C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,wEAAwE,EAAE,KAAK,IAAI,EAAE;;QACvF,cAAc,CAAC,uBAAuB,GAAG,KAAK,CAAC;QAE/C,MAAM,MAAM,GAAG,MAAM,IAAA,mCAAmB,EAAC,WAAW,EAAE,QAAQ,EAAE,cAAc,EAAE,CAAC,CAAC,CAAC;QAEnF,MAAM,CAAC,MAAA,MAAM,CAAC,CAAC,CAAC,CAAC,IAAI,0CAAE,QAAQ,CAAC,CAAC,GAAG,CAAC,cAAc,CAAC,UAAU,CAAC,CAAC;QAChE,MAAM,CAAC,MAAC,MAAA,MAAM,CAAC,CAAC,CAAC,CAAC,IAAI,0CAAE,QAAwB,0CAAE,WAAW,CAAC,CAAC,OAAO,CAAC,gBAAgB,CAAC,CAAC;QACzF,MAAM,CAAC,MAAA,MAAM,CAAC,CAAC,CAAC,CAAC,IAAI,0CAAE,KAAK,CAAC,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;IAC7C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,gDAAgD,EAAE,KAAK,IAAI,EAAE;QAC/D,cAAc,CAAC,IAAI,GAAG,CAAC,CAAC;QAExB,MAAM,IAAA,mCAAmB,EAAC,WAAW,EAAE,QAAQ,EAAE,cAAc,EAAE,CAAC,CAAC,CAAC;QAEpE,MAAM,CAAC,eAAe,CAAC,+BAA+B,CAAC,CAAC,oBAAoB,CAC3E,MAAM,CAAC,QAAQ,EAAE,EACjB,CAAC,EACD,MAAM,CAAC,QAAQ,EAAE,CACjB,CAAC;IACH,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,oDAAoD,EAAE,KAAK,IAAI,EAAE;QACnE,MAAM,MAAM,GAAG,MAAM,IAAA,mCAAmB,EAAC,WAAW,EAAE,QAAQ,EAAE,cAAc,EAAE,CAAC,CAAC,CAAC;QAEnF,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,EAAE,EAAE;YACvB,MAAM,CAAC,IAAI,CAAC,CAAC,cAAc,CAAC,YAAY,CAAC,CAAC;YAC1C,MAAM,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC,OAAO,CAAC,EAAE,IAAI,EAAE,CAAC,EAAE,CAAC,CAAC;QAC9C,CAAC,CAAC,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,yDAAyD,EAAE,KAAK,IAAI,EAAE;QACxE,MAAM,IAAA,mCAAmB,EAAC,WAAW,EAAE,QAAQ,EAAE,cAAc,EAAE,CAAC,CAAC,CAAC;QAEpE,MAAM,CAAC,oBAAU,CAAC,CAAC,oBAAoB,CAAC,WAAW,EAAE,0BAA0B,EAAE;YAChF,KAAK,EAAE,mBAAmB;SAC1B,CAAC,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,4DAA4D,EAAE,KAAK,IAAI,EAAE;QAC3E,eAAe,CAAC,+BAA+B,CAAC,iBAAiB,CAAC,IAAI,KAAK,CAAC,YAAY,CAAC,CAAC,CAAC;QAE3F,MAAM,MAAM,CAAC,IAAA,mCAAmB,EAAC,WAAW,EAAE,QAAQ,EAAE,cAAc,EAAE,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CAC1F,YAAY,CACZ,CAAC;QAEF,MAAM,CAAC,QAAQ,CAAC,wBAAwB,CAAC,CAAC,oBAAoB,CAAC,eAAe,CAAC,CAAC;IACjF,CAAC,CAAC,CAAC;AACJ,CAAC,CAAC,CAAC"}

package/dist/nodes/vector_store/shared/createVectorStoreNode/operations/__tests__/operationHandlers.test.d.ts

@@ -0,0 +1 @@
+export {};

package/dist/nodes/vector_store/shared/createVectorStoreNode/operations/__tests__/operationHandlers.test.js

@@ -0,0 +1,93 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+const jest_mock_extended_1 = require("jest-mock-extended");
+const n8n_workflow_1 = require("n8n-workflow");
+const loadOperation_1 = require("../loadOperation");
+const retrieveAsToolOperation_1 = require("../retrieveAsToolOperation");
+const retrieveOperation_1 = require("../retrieveOperation");
+const updateOperation_1 = require("../updateOperation");
+describe('Vector Store Operation Handlers', () => {
+    let mockContext;
+    let mockEmbeddings;
+    let mockVectorStore;
+    let mockArgs;
+    let nodeParameters;
+    beforeEach(() => {
+        nodeParameters = {
+            mode: 'load',
+            prompt: 'test query',
+            topK: 3,
+            includeDocumentMetadata: true,
+            toolName: 'test_tool',
+            toolDescription: 'Test tool description',
+        };
+        mockContext = (0, jest_mock_extended_1.mock)();
+        mockContext.getNodeParameter.mockImplementation((parameterName, itemIndex, fallbackValue, options) => {
+            var _a;
+            if (typeof parameterName !== 'string')
+                return fallbackValue;
+            return (_a = nodeParameters[parameterName]) !== null && _a !== void 0 ? _a : fallbackValue;
+        });
+        mockContext.getInputData.mockReturnValue([{ json: { test: 'data' } }]);
+        mockEmbeddings = (0, jest_mock_extended_1.mock)();
+        mockEmbeddings.embedQuery.mockResolvedValue([0.1, 0.2, 0.3]);
+        mockVectorStore = (0, jest_mock_extended_1.mock)();
+        mockVectorStore.similaritySearchVectorWithScore.mockResolvedValue([
+            [{ pageContent: 'test content', metadata: { test: 'metadata' } }, 0.95],
+            [{ pageContent: 'test content 2', metadata: { test: 'metadata 2' } }, 0.85],
+        ]);
+        mockArgs = {
+            meta: {
+                displayName: 'Test Vector Store',
+                name: 'testVectorStore',
+                description: 'Vector store for testing',
+                docsUrl: 'https://example.com',
+                icon: 'file:testIcon.svg',
+                operationModes: ['load', 'insert', 'retrieve', 'retrieve-as-tool', 'update'],
+            },
+            sharedFields: [],
+            getVectorStoreClient: jest.fn().mockResolvedValue(mockVectorStore),
+            populateVectorStore: jest.fn().mockResolvedValue(undefined),
+            releaseVectorStoreClient: jest.fn(),
+        };
+    });
+    describe('handleLoadOperation', () => {
+        it('should properly process load operation', async () => {
+            const result = await (0, loadOperation_1.handleLoadOperation)(mockContext, mockArgs, mockEmbeddings, 0);
+            expect(mockArgs.getVectorStoreClient).toHaveBeenCalledTimes(1);
+            expect(mockEmbeddings.embedQuery).toHaveBeenCalledWith('test query');
+            expect(mockVectorStore.similaritySearchVectorWithScore).toHaveBeenCalled();
+            expect(result).toHaveLength(2);
+            expect(result[0].json).toHaveProperty('document');
+            expect(result[0].json).toHaveProperty('score');
+            expect(mockArgs.releaseVectorStoreClient).toHaveBeenCalledWith(mockVectorStore);
+        });
+        it('should exclude metadata when includeDocumentMetadata is false', async () => {
+            nodeParameters.includeDocumentMetadata = false;
+            const result = await (0, loadOperation_1.handleLoadOperation)(mockContext, mockArgs, mockEmbeddings, 0);
+            expect(result[0].json.document).not.toHaveProperty('metadata');
+        });
+    });
+    describe('handleUpdateOperation', () => {
+        it('should throw error when update is not supported', async () => {
+            mockArgs.meta.operationModes = ['load', 'insert'];
+            await expect((0, updateOperation_1.handleUpdateOperation)(mockContext, mockArgs, mockEmbeddings)).rejects.toThrow(n8n_workflow_1.NodeOperationError);
+        });
+    });
+    describe('handleRetrieveOperation', () => {
+        it('should return vector store with log wrapper and close function', async () => {
+            const result = await (0, retrieveOperation_1.handleRetrieveOperation)(mockContext, mockArgs, mockEmbeddings, 0);
+            expect(result).toHaveProperty('response');
+            expect(result).toHaveProperty('closeFunction');
+        });
+    });
+    describe('handleRetrieveAsToolOperation', () => {
+        it('should return a tool with the correct name and description', async () => {
+            const result = await (0, retrieveAsToolOperation_1.handleRetrieveAsToolOperation)(mockContext, mockArgs, mockEmbeddings, 0);
+            expect(result).toHaveProperty('response');
+            expect(result.response).toHaveProperty('name', 'test_tool');
+            expect(result.response).toHaveProperty('description', 'Test tool description');
+        });
+    });
+});
+//# sourceMappingURL=operationHandlers.test.js.map

package/dist/nodes/vector_store/shared/createVectorStoreNode/operations/__tests__/operationHandlers.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"operationHandlers.test.js","sourceRoot":"","sources":["../../../../../../../nodes/vector_store/shared/createVectorStoreNode/operations/__tests__/operationHandlers.test.ts"],"names":[],"mappings":";;AAIA,2DAA0C;AAE1C,+CAAkD;AAGlD,oDAAuD;AACvD,wEAA2E;AAC3E,4DAA+D;AAC/D,wDAA2D;AAE3D,QAAQ,CAAC,iCAAiC,EAAE,GAAG,EAAE;IAChD,IAAI,WAAgE,CAAC;IACrE,IAAI,cAAqC,CAAC;IAC1C,IAAI,eAAuC,CAAC;IAC5C,IAAI,QAAqD,CAAC;IAC1D,IAAI,cAAmC,CAAC;IAExC,UAAU,CAAC,GAAG,EAAE;QACf,cAAc,GAAG;YAChB,IAAI,EAAE,MAAM;YACZ,MAAM,EAAE,YAAY;YACpB,IAAI,EAAE,CAAC;YACP,uBAAuB,EAAE,IAAI;YAC7B,QAAQ,EAAE,WAAW;YACrB,eAAe,EAAE,uBAAuB;SACxC,CAAC;QAEF,WAAW,GAAG,IAAA,yBAAI,GAA4C,CAAC;QAC/D,WAAW,CAAC,gBAAgB,CAAC,kBAAkB,CAAC,CAAC,aAAqB,EAAE,SAAiB,EAAE,aAAmB,EAAE,OAAa,EAAO,EAAE;;YACrI,IAAI,OAAO,aAAa,KAAK,QAAQ;gBAAE,OAAO,aAAa,CAAC;YAC5D,OAAO,MAAA,cAAc,CAAC,aAAa,CAAC,mCAAI,aAAa,CAAC;QACvD,CAAC,CAAC,CAAC;QACH,WAAW,CAAC,YAAY,CAAC,eAAe,CAAC,CAAC,EAAE,IAAI,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE,EAAE,CAAC,CAAC,CAAC;QAEvE,cAAc,GAAG,IAAA,yBAAI,GAAc,CAAC;QACpC,cAAc,CAAC,UAAU,CAAC,iBAAiB,CAAC,CAAC,GAAG,EAAE,GAAG,EAAE,GAAG,CAAC,CAAC,CAAC;QAE7D,eAAe,GAAG,IAAA,yBAAI,GAAe,CAAC;QACtC,eAAe,CAAC,+BAA+B,CAAC,iBAAiB,CAAC;YACjE,CAAC,EAAE,WAAW,EAAE,cAAc,EAAE,QAAQ,EAAE,EAAE,IAAI,EAAE,UAAU,EAAE,EAAc,EAAE,IAAI,CAAC;YACnF,CAAC,EAAE,WAAW,EAAE,gBAAgB,EAAE,QAAQ,EAAE,EAAE,IAAI,EAAE,YAAY,EAAE,EAAc,EAAE,IAAI,CAAC;SACvF,CAAC,CAAC;QAEH,QAAQ,GAAG;YACV,IAAI,EAAE;gBACL,WAAW,EAAE,mBAAmB;gBAChC,IAAI,EAAE,iBAAiB;gBACvB,WAAW,EAAE,0BAA0B;gBACvC,OAAO,EAAE,qBAAqB;gBAC9B,IAAI,EAAE,mBAAmB;gBACzB,cAAc,EAAE,CAAC,MAAM,EAAE,QAAQ,EAAE,UAAU,EAAE,kBAAkB,EAAE,QAAQ,CAAC;aAC5E;YACD,YAAY,EAAE,EAAE;YAChB,oBAAoB,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,eAAe,CAAC;YAClE,mBAAmB,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,SAAS,CAAC;YAC3D,wBAAwB,EAAE,IAAI,CAAC,EAAE,EAAE;SACnC,CAAC;IACH,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,qBAAqB,EAAE,GAAG,EAAE;QACpC,EAAE,CAAC,wCAAwC,EAAE,KAAK,IAAI,EAAE;YACvD,MAAM,MAAM,GAAG,MAAM,IAAA,mCAAmB,EAAC,WAAW,EAAE,QAAQ,EAAE,cAAc,EAAE,CAAC,CAAC,CAAC;YAEnF,MAAM,CAAC,QAAQ,CAAC,oBAAoB,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;YAC/D,MAAM,CAAC,cAAc,CAAC,UAAU,CAAC,CAAC,oBAAoB,CAAC,YAAY,CAAC,CAAC;YACrE,MAAM,CAAC,eAAe,CAAC,+BAA+B,CAAC,CAAC,gBAAgB,EAAE,CAAC;YAC3E,MAAM,CAAC,MAAM,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;YAC/B,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,cAAc,CAAC,UAAU,CAAC,CAAC;YAClD,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,cAAc,CAAC,OAAO,CAAC,CAAC;YAC/C,MAAM,CAAC,QAAQ,CAAC,wBAAwB,CAAC,CAAC,oBAAoB,CAAC,eAAe,CAAC,CAAC;QACjF,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,+DAA+D,EAAE,KAAK,IAAI,EAAE;YAC9E,cAAc,CAAC,uBAAuB,GAAG,KAAK,CAAC;YAE/C,MAAM,MAAM,GAAG,MAAM,IAAA,mCAAmB,EAAC,WAAW,EAAE,QAAQ,EAAE,cAAc,EAAE,CAAC,CAAC,CAAC;YAEnF,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,GAAG,CAAC,cAAc,CAAC,UAAU,CAAC,CAAC;QAChE,CAAC,CAAC,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,uBAAuB,EAAE,GAAG,EAAE;QACtC,EAAE,CAAC,iDAAiD,EAAE,KAAK,IAAI,EAAE;YAChE,QAAQ,CAAC,IAAI,CAAC,cAAc,GAAG,CAAC,MAAM,EAAE,QAAQ,CAAC,CAAC;YAElD,MAAM,MAAM,CAAC,IAAA,uCAAqB,EAAC,WAAW,EAAE,QAAQ,EAAE,cAAc,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CACzF,iCAAkB,CAClB,CAAC;QACH,CAAC,CAAC,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,yBAAyB,EAAE,GAAG,EAAE;QACxC,EAAE,CAAC,gEAAgE,EAAE,KAAK,IAAI,EAAE;YAC/E,MAAM,MAAM,GAAG,MAAM,IAAA,2CAAuB,EAAC,WAAW,EAAE,QAAQ,EAAE,cAAc,EAAE,CAAC,CAAC,CAAC;YAEvF,MAAM,CAAC,MAAM,CAAC,CAAC,cAAc,CAAC,UAAU,CAAC,CAAC;YAC1C,MAAM,CAAC,MAAM,CAAC,CAAC,cAAc,CAAC,eAAe,CAAC,CAAC;QAChD,CAAC,CAAC,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,+BAA+B,EAAE,GAAG,EAAE;QAC9C,EAAE,CAAC,4DAA4D,EAAE,KAAK,IAAI,EAAE;YAC3E,MAAM,MAAM,GAAG,MAAM,IAAA,uDAA6B,EAAC,WAAW,EAAE,QAAQ,EAAE,cAAc,EAAE,CAAC,CAAC,CAAC;YAE7F,MAAM,CAAC,MAAM,CAAC,CAAC,cAAc,CAAC,UAAU,CAAC,CAAC;YAC1C,MAAM,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,cAAc,CAAC,MAAM,EAAE,WAAW,CAAC,CAAC;YAC5D,MAAM,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,cAAc,CAAC,aAAa,EAAE,uBAAuB,CAAC,CAAC;QAChF,CAAC,CAAC,CAAC;IACJ,CAAC,CAAC,CAAC;AACJ,CAAC,CAAC,CAAC"}

package/dist/nodes/vector_store/shared/createVectorStoreNode/operations/__tests__/retrieveAsToolOperation.test.d.ts

@@ -0,0 +1 @@
+export {};