@n8n/ai-utilities 0.4.1 → 0.5.0

package/dist/cjs/utils/vector-store/createVectorStoreNode/operations/index.js

@@ -0,0 +1,33 @@
+var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    var desc = Object.getOwnPropertyDescriptor(m, k);
+    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
+      desc = { enumerable: true, get: function() { return m[k]; } };
+    }
+    Object.defineProperty(o, k2, desc);
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+var __exportStar = (this && this.__exportStar) || function(m, exports) {
+    for (var p in m) if (p !== "default" && !Object.prototype.hasOwnProperty.call(exports, p)) __createBinding(exports, m, p);
+};
+(function (factory) {
+    if (typeof module === "object" && typeof module.exports === "object") {
+        var v = factory(require, exports);
+        if (v !== undefined) module.exports = v;
+    }
+    else if (typeof define === "function" && define.amd) {
+        define(["require", "exports", "./loadOperation", "./insertOperation", "./updateOperation", "./retrieveOperation", "./retrieveAsToolOperation", "./retrieveAsToolExecuteOperation"], factory);
+    }
+})(function (require, exports) {
+    "use strict";
+    Object.defineProperty(exports, "__esModule", { value: true });
+    __exportStar(require("./loadOperation"), exports);
+    __exportStar(require("./insertOperation"), exports);
+    __exportStar(require("./updateOperation"), exports);
+    __exportStar(require("./retrieveOperation"), exports);
+    __exportStar(require("./retrieveAsToolOperation"), exports);
+    __exportStar(require("./retrieveAsToolExecuteOperation"), exports);
+});
+//# sourceMappingURL=index.js.map

package/dist/cjs/utils/vector-store/createVectorStoreNode/operations/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../../../../src/utils/vector-store/createVectorStoreNode/operations/index.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;IAAA,kDAAgC;IAChC,oDAAkC;IAClC,oDAAkC;IAClC,sDAAoC;IACpC,4DAA0C;IAC1C,mEAAiD"}

package/dist/cjs/utils/vector-store/createVectorStoreNode/operations/insertOperation.d.ts

@@ -0,0 +1,5 @@
+import type { Embeddings } from '@langchain/core/embeddings';
+import type { VectorStore } from '@langchain/core/vectorstores';
+import type { IExecuteFunctions, INodeExecutionData } from 'n8n-workflow';
+import type { VectorStoreNodeConstructorArgs } from '../types';
+export declare function handleInsertOperation<T extends VectorStore = VectorStore>(context: IExecuteFunctions, args: VectorStoreNodeConstructorArgs<T>, embeddings: Embeddings): Promise<INodeExecutionData[]>;

package/dist/cjs/utils/vector-store/createVectorStoreNode/operations/insertOperation.js

@@ -0,0 +1,48 @@
+(function (factory) {
+    if (typeof module === "object" && typeof module.exports === "object") {
+        var v = factory(require, exports);
+        if (v !== undefined) module.exports = v;
+    }
+    else if (typeof define === "function" && define.amd) {
+        define(["require", "exports", "n8n-workflow", "../../../log-ai-event", "../../processDocuments"], factory);
+    }
+})(function (require, exports) {
+    "use strict";
+    Object.defineProperty(exports, "__esModule", { value: true });
+    exports.handleInsertOperation = handleInsertOperation;
+    const n8n_workflow_1 = require("n8n-workflow");
+    const log_ai_event_1 = require("../../../log-ai-event");
+    const processDocuments_1 = require("../../processDocuments");
+    async function handleInsertOperation(context, args, embeddings) {
+        const nodeVersion = context.getNode().typeVersion;
+        const items = context.getInputData();
+        const documentInput = (await context.getInputConnectionData(n8n_workflow_1.NodeConnectionTypes.AiDocument, 0));
+        const resultData = [];
+        const documentsForEmbedding = [];
+        for (let itemIndex = 0; itemIndex < items.length; itemIndex++) {
+            if (context.getExecutionCancelSignal()?.aborted) {
+                break;
+            }
+            const itemData = items[itemIndex];
+            const processedDocuments = await (0, processDocuments_1.processDocument)(documentInput, itemData, itemIndex);
+            resultData.push(...processedDocuments.serializedDocuments);
+            documentsForEmbedding.push(...processedDocuments.processedDocuments);
+            if (nodeVersion === 1) {
+                await args.populateVectorStore(context, embeddings, processedDocuments.processedDocuments, itemIndex);
+            }
+            (0, log_ai_event_1.logAiEvent)(context, 'ai-vector-store-populated');
+        }
+        if (nodeVersion >= 1.1) {
+            if (args.beforeInsert) {
+                await args.beforeInsert(context, embeddings, 0);
+            }
+            const embeddingBatchSize = context.getNodeParameter('embeddingBatchSize', 0, 200) ?? 200;
+            for (let i = 0; i < documentsForEmbedding.length; i += embeddingBatchSize) {
+                const nextBatch = documentsForEmbedding.slice(i, i + embeddingBatchSize);
+                await args.populateVectorStore(context, embeddings, nextBatch, 0);
+            }
+        }
+        return resultData;
+    }
+});
+//# sourceMappingURL=insertOperation.js.map

package/dist/cjs/utils/vector-store/createVectorStoreNode/operations/insertOperation.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"insertOperation.js","sourceRoot":"","sources":["../../../../../../src/utils/vector-store/createVectorStoreNode/operations/insertOperation.ts"],"names":[],"mappings":";;;;;;;;;;;IAgBA,sDAgEC;IA7ED,+CAAmD;IAGnD,wDAAmD;IAGnD,6DAAyD;IAOlD,KAAK,UAAU,qBAAqB,CAC1C,OAA0B,EAC1B,IAAuC,EACvC,UAAsB;QAEtB,MAAM,WAAW,GAAG,OAAO,CAAC,OAAO,EAAE,CAAC,WAAW,CAAC;QAElD,MAAM,KAAK,GAAG,OAAO,CAAC,YAAY,EAAE,CAAC;QACrC,MAAM,aAAa,GAAG,CAAC,MAAM,OAAO,CAAC,sBAAsB,CAAC,kCAAmB,CAAC,UAAU,EAAE,CAAC,CAAC,CAGnD,CAAC;QAE5C,MAAM,UAAU,GAAyB,EAAE,CAAC;QAC5C,MAAM,qBAAqB,GAA6C,EAAE,CAAC;QAG3E,KAAK,IAAI,SAAS,GAAG,CAAC,EAAE,SAAS,GAAG,KAAK,CAAC,MAAM,EAAE,SAAS,EAAE,EAAE,CAAC;YAE/D,IAAI,OAAO,CAAC,wBAAwB,EAAE,EAAE,OAAO,EAAE,CAAC;gBACjD,MAAM;YACP,CAAC;YAED,MAAM,QAAQ,GAAG,KAAK,CAAC,SAAS,CAAC,CAAC;YAGlC,MAAM,kBAAkB,GAAG,MAAM,IAAA,kCAAe,EAAC,aAAa,EAAE,QAAQ,EAAE,SAAS,CAAC,CAAC;YAGrF,UAAU,CAAC,IAAI,CAAC,GAAG,kBAAkB,CAAC,mBAAmB,CAAC,CAAC;YAG3D,qBAAqB,CAAC,IAAI,CAAC,GAAG,kBAAkB,CAAC,kBAAkB,CAAC,CAAC;YAGrE,IAAI,WAAW,KAAK,CAAC,EAAE,CAAC;gBACvB,MAAM,IAAI,CAAC,mBAAmB,CAC7B,OAAO,EACP,UAAU,EACV,kBAAkB,CAAC,kBAAkB,EACrC,SAAS,CACT,CAAC;YACH,CAAC;YAED,IAAA,yBAAU,EAAC,OAAO,EAAE,2BAA2B,CAAC,CAAC;QAClD,CAAC;QAGD,IAAI,WAAW,IAAI,GAAG,EAAE,CAAC;YACxB,IAAI,IAAI,CAAC,YAAY,EAAE,CAAC;gBACvB,MAAM,IAAI,CAAC,YAAY,CAAC,OAAO,EAAE,UAAU,EAAE,CAAC,CAAC,CAAC;YACjD,CAAC;YAED,MAAM,kBAAkB,GACtB,OAAO,CAAC,gBAAgB,CAAC,oBAAoB,EAAE,CAAC,EAAE,GAAG,CAAY,IAAI,GAAG,CAAC;YAG3E,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,qBAAqB,CAAC,MAAM,EAAE,CAAC,IAAI,kBAAkB,EAAE,CAAC;gBAC3E,MAAM,SAAS,GAAG,qBAAqB,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,GAAG,kBAAkB,CAAC,CAAC;gBACzE,MAAM,IAAI,CAAC,mBAAmB,CAAC,OAAO,EAAE,UAAU,EAAE,SAAS,EAAE,CAAC,CAAC,CAAC;YACnE,CAAC;QACF,CAAC;QAED,OAAO,UAAU,CAAC;IACnB,CAAC"}

package/dist/cjs/utils/vector-store/createVectorStoreNode/operations/loadOperation.d.ts

@@ -0,0 +1,5 @@
+import type { Embeddings } from '@langchain/core/embeddings';
+import type { VectorStore } from '@langchain/core/vectorstores';
+import type { IExecuteFunctions, INodeExecutionData } from 'n8n-workflow';
+import type { VectorStoreNodeConstructorArgs } from '../types';
+export declare function handleLoadOperation<T extends VectorStore = VectorStore>(context: IExecuteFunctions, args: VectorStoreNodeConstructorArgs<T>, embeddings: Embeddings, itemIndex: number): Promise<INodeExecutionData[]>;

package/dist/cjs/utils/vector-store/createVectorStoreNode/operations/loadOperation.js

@@ -0,0 +1,55 @@
+(function (factory) {
+    if (typeof module === "object" && typeof module.exports === "object") {
+        var v = factory(require, exports);
+        if (v !== undefined) module.exports = v;
+    }
+    else if (typeof define === "function" && define.amd) {
+        define(["require", "exports", "n8n-workflow", "../../../helpers", "../../../log-ai-event"], factory);
+    }
+})(function (require, exports) {
+    "use strict";
+    Object.defineProperty(exports, "__esModule", { value: true });
+    exports.handleLoadOperation = handleLoadOperation;
+    const n8n_workflow_1 = require("n8n-workflow");
+    const helpers_1 = require("../../../helpers");
+    const log_ai_event_1 = require("../../../log-ai-event");
+    async function handleLoadOperation(context, args, embeddings, itemIndex) {
+        const filter = (0, helpers_1.getMetadataFiltersValues)(context, itemIndex);
+        const vectorStore = await args.getVectorStoreClient(context, undefined, embeddings, itemIndex);
+        try {
+            const prompt = context.getNodeParameter('prompt', itemIndex);
+            const topK = context.getNodeParameter('topK', itemIndex, 4);
+            const useReranker = context.getNodeParameter('useReranker', itemIndex, false);
+            const includeDocumentMetadata = context.getNodeParameter('includeDocumentMetadata', itemIndex, true);
+            const embeddedPrompt = await embeddings.embedQuery(prompt);
+            let docs = await vectorStore.similaritySearchVectorWithScore(embeddedPrompt, topK, filter);
+            if (useReranker && docs.length > 0) {
+                const reranker = (await context.getInputConnectionData(n8n_workflow_1.NodeConnectionTypes.AiReranker, 0));
+                const documents = docs.map(([doc]) => doc);
+                const rerankedDocuments = await reranker.compressDocuments(documents, prompt);
+                docs = rerankedDocuments.map((doc) => {
+                    const { relevanceScore, ...metadata } = doc.metadata || {};
+                    return [{ ...doc, metadata }, relevanceScore];
+                });
+            }
+            const serializedDocs = docs.map(([doc, score]) => {
+                const document = {
+                    pageContent: doc.pageContent,
+                    ...(includeDocumentMetadata ? { metadata: doc.metadata } : {}),
+                };
+                return {
+                    json: { document, score },
+                    pairedItem: {
+                        item: itemIndex,
+                    },
+                };
+            });
+            (0, log_ai_event_1.logAiEvent)(context, 'ai-vector-store-searched', { query: prompt });
+            return serializedDocs;
+        }
+        finally {
+            args.releaseVectorStoreClient?.(vectorStore);
+        }
+    }
+});
+//# sourceMappingURL=loadOperation.js.map

package/dist/cjs/utils/vector-store/createVectorStoreNode/operations/loadOperation.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"loadOperation.js","sourceRoot":"","sources":["../../../../../../src/utils/vector-store/createVectorStoreNode/operations/loadOperation.ts"],"names":[],"mappings":";;;;;;;;;;;IAcA,kDAuEC;IAlFD,+CAAmD;IAGnD,8CAA4D;IAC5D,wDAAmD;IAO5C,KAAK,UAAU,mBAAmB,CACxC,OAA0B,EAC1B,IAAuC,EACvC,UAAsB,EACtB,SAAiB;QAEjB,MAAM,MAAM,GAAG,IAAA,kCAAwB,EAAC,OAAO,EAAE,SAAS,CAAC,CAAC;QAC5D,MAAM,WAAW,GAAG,MAAM,IAAI,CAAC,oBAAoB,CAClD,OAAO,EAEP,SAAS,EACT,UAAU,EACV,SAAS,CACT,CAAC;QAEF,IAAI,CAAC;YAEJ,MAAM,MAAM,GAAG,OAAO,CAAC,gBAAgB,CAAC,QAAQ,EAAE,SAAS,CAAW,CAAC;YACvE,MAAM,IAAI,GAAG,OAAO,CAAC,gBAAgB,CAAC,MAAM,EAAE,SAAS,EAAE,CAAC,CAAW,CAAC;YACtE,MAAM,WAAW,GAAG,OAAO,CAAC,gBAAgB,CAAC,aAAa,EAAE,SAAS,EAAE,KAAK,CAAY,CAAC;YAEzF,MAAM,uBAAuB,GAAG,OAAO,CAAC,gBAAgB,CACvD,yBAAyB,EACzB,SAAS,EACT,IAAI,CACO,CAAC;YAGb,MAAM,cAAc,GAAG,MAAM,UAAU,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC;YAG3D,IAAI,IAAI,GAAG,MAAM,WAAW,CAAC,+BAA+B,CAAC,cAAc,EAAE,IAAI,EAAE,MAAM,CAAC,CAAC;YAG3F,IAAI,WAAW,IAAI,IAAI,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBACpC,MAAM,QAAQ,GAAG,CAAC,MAAM,OAAO,CAAC,sBAAsB,CACrD,kCAAmB,CAAC,UAAU,EAC9B,CAAC,CACD,CAA2B,CAAC;gBAC7B,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,GAAG,CAAC,EAAE,EAAE,CAAC,GAAG,CAAC,CAAC;gBAE3C,MAAM,iBAAiB,GAAG,MAAM,QAAQ,CAAC,iBAAiB,CAAC,SAAS,EAAE,MAAM,CAAC,CAAC;gBAC9E,IAAI,GAAG,iBAAiB,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE;oBACpC,MAAM,EAAE,cAAc,EAAE,GAAG,QAAQ,EAAE,GAAG,GAAG,CAAC,QAAQ,IAAI,EAAE,CAAC;oBAC3D,OAAO,CAAC,EAAE,GAAG,GAAG,EAAE,QAAQ,EAAE,EAAE,cAAc,CAAC,CAAC;gBAC/C,CAAC,CAAC,CAAC;YACJ,CAAC;YAGD,MAAM,cAAc,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,GAAG,EAAE,KAAK,CAAC,EAAE,EAAE;gBAChD,MAAM,QAAQ,GAAG;oBAChB,WAAW,EAAE,GAAG,CAAC,WAAW;oBAC5B,GAAG,CAAC,uBAAuB,CAAC,CAAC,CAAC,EAAE,QAAQ,EAAE,GAAG,CAAC,QAAQ,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;iBAC9D,CAAC;gBAEF,OAAO;oBACN,IAAI,EAAE,EAAE,QAAQ,EAAE,KAAK,EAAE;oBACzB,UAAU,EAAE;wBACX,IAAI,EAAE,SAAS;qBACf;iBACD,CAAC;YACH,CAAC,CAAC,CAAC;YAGH,IAAA,yBAAU,EAAC,OAAO,EAAE,0BAA0B,EAAE,EAAE,KAAK,EAAE,MAAM,EAAE,CAAC,CAAC;YAEnE,OAAO,cAAc,CAAC;QACvB,CAAC;gBAAS,CAAC;YAEV,IAAI,CAAC,wBAAwB,EAAE,CAAC,WAAW,CAAC,CAAC;QAC9C,CAAC;IACF,CAAC"}

package/dist/cjs/utils/vector-store/createVectorStoreNode/operations/retrieveAsToolExecuteOperation.d.ts

@@ -0,0 +1,5 @@
+import type { Embeddings } from '@langchain/core/embeddings';
+import type { VectorStore } from '@langchain/core/vectorstores';
+import type { IExecuteFunctions, INodeExecutionData } from 'n8n-workflow';
+import type { VectorStoreNodeConstructorArgs } from '../types';
+export declare function handleRetrieveAsToolExecuteOperation<T extends VectorStore = VectorStore>(context: IExecuteFunctions, args: VectorStoreNodeConstructorArgs<T>, embeddings: Embeddings, itemIndex: number): Promise<INodeExecutionData[]>;

package/dist/cjs/utils/vector-store/createVectorStoreNode/operations/retrieveAsToolExecuteOperation.js

@@ -0,0 +1,74 @@
+(function (factory) {
+    if (typeof module === "object" && typeof module.exports === "object") {
+        var v = factory(require, exports);
+        if (v !== undefined) module.exports = v;
+    }
+    else if (typeof define === "function" && define.amd) {
+        define(["require", "exports", "n8n-workflow", "../../../helpers", "../../../log-ai-event"], factory);
+    }
+})(function (require, exports) {
+    "use strict";
+    Object.defineProperty(exports, "__esModule", { value: true });
+    exports.handleRetrieveAsToolExecuteOperation = handleRetrieveAsToolExecuteOperation;
+    const n8n_workflow_1 = require("n8n-workflow");
+    const helpers_1 = require("../../../helpers");
+    const log_ai_event_1 = require("../../../log-ai-event");
+    async function handleRetrieveAsToolExecuteOperation(context, args, embeddings, itemIndex) {
+        const filter = (0, helpers_1.getMetadataFiltersValues)(context, itemIndex);
+        const vectorStore = await args.getVectorStoreClient(context, undefined, embeddings, itemIndex);
+        try {
+            const inputData = context.getInputData();
+            const item = inputData[itemIndex];
+            const query = typeof item.json.input === 'string' ? item.json.input : undefined;
+            if (!query || typeof query !== 'string') {
+                throw new Error('Input data must contain a "input" field with the search query');
+            }
+            const topK = context.getNodeParameter('topK', itemIndex, 4);
+            (0, n8n_workflow_1.assertParamIsNumber)('topK', topK, context.getNode());
+            const useReranker = context.getNodeParameter('useReranker', itemIndex, false);
+            (0, n8n_workflow_1.assertParamIsBoolean)('useReranker', useReranker, context.getNode());
+            const includeDocumentMetadata = context.getNodeParameter('includeDocumentMetadata', itemIndex, true);
+            (0, n8n_workflow_1.assertParamIsBoolean)('includeDocumentMetadata', includeDocumentMetadata, context.getNode());
+            const embeddedQuery = await embeddings.embedQuery(query);
+            let docs = await vectorStore.similaritySearchVectorWithScore(embeddedQuery, topK, filter);
+            if (useReranker && docs.length > 0) {
+                const reranker = (await context.getInputConnectionData(n8n_workflow_1.NodeConnectionTypes.AiReranker, 0));
+                const documents = docs.map(([doc]) => doc);
+                const rerankedDocuments = await reranker.compressDocuments(documents, query);
+                docs = rerankedDocuments.map((doc) => {
+                    const { relevanceScore, ...metadata } = doc.metadata || {};
+                    return [{ ...doc, metadata }, relevanceScore ?? 0];
+                });
+            }
+            const serializedDocs = docs.map(([doc]) => {
+                if (includeDocumentMetadata) {
+                    return {
+                        type: 'text',
+                        text: JSON.stringify({ ...doc }),
+                    };
+                }
+                else {
+                    return {
+                        type: 'text',
+                        pageContent: JSON.stringify({ pageContent: doc.pageContent }),
+                    };
+                }
+            });
+            (0, log_ai_event_1.logAiEvent)(context, 'ai-vector-store-searched', { input: query });
+            return [
+                {
+                    json: {
+                        response: serializedDocs,
+                    },
+                    pairedItem: {
+                        item: itemIndex,
+                    },
+                },
+            ];
+        }
+        finally {
+            args.releaseVectorStoreClient?.(vectorStore);
+        }
+    }
+});
+//# sourceMappingURL=retrieveAsToolExecuteOperation.js.map

package/dist/cjs/utils/vector-store/createVectorStoreNode/operations/retrieveAsToolExecuteOperation.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"retrieveAsToolExecuteOperation.js","sourceRoot":"","sources":["../../../../../../src/utils/vector-store/createVectorStoreNode/operations/retrieveAsToolExecuteOperation.ts"],"names":[],"mappings":";;;;;;;;;;;IAeA,oFA0FC;IAtGD,+CAA8F;IAG9F,8CAA4D;IAC5D,wDAAmD;IAQ5C,KAAK,UAAU,oCAAoC,CACzD,OAA0B,EAC1B,IAAuC,EACvC,UAAsB,EACtB,SAAiB;QAEjB,MAAM,MAAM,GAAG,IAAA,kCAAwB,EAAC,OAAO,EAAE,SAAS,CAAC,CAAC;QAC5D,MAAM,WAAW,GAAG,MAAM,IAAI,CAAC,oBAAoB,CAClD,OAAO,EAEP,SAAS,EACT,UAAU,EACV,SAAS,CACT,CAAC;QAEF,IAAI,CAAC;YAEJ,MAAM,SAAS,GAAG,OAAO,CAAC,YAAY,EAAE,CAAC;YACzC,MAAM,IAAI,GAAG,SAAS,CAAC,SAAS,CAAC,CAAC;YAClC,MAAM,KAAK,GAAG,OAAO,IAAI,CAAC,IAAI,CAAC,KAAK,KAAK,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,SAAS,CAAC;YAEhF,IAAI,CAAC,KAAK,IAAI,OAAO,KAAK,KAAK,QAAQ,EAAE,CAAC;gBACzC,MAAM,IAAI,KAAK,CAAC,+DAA+D,CAAC,CAAC;YAClF,CAAC;YAED,MAAM,IAAI,GAAG,OAAO,CAAC,gBAAgB,CAAC,MAAM,EAAE,SAAS,EAAE,CAAC,CAAC,CAAC;YAC5D,IAAA,kCAAmB,EAAC,MAAM,EAAE,IAAI,EAAE,OAAO,CAAC,OAAO,EAAE,CAAC,CAAC;YACrD,MAAM,WAAW,GAAG,OAAO,CAAC,gBAAgB,CAAC,aAAa,EAAE,SAAS,EAAE,KAAK,CAAC,CAAC;YAC9E,IAAA,mCAAoB,EAAC,aAAa,EAAE,WAAW,EAAE,OAAO,CAAC,OAAO,EAAE,CAAC,CAAC;YAEpE,MAAM,uBAAuB,GAAG,OAAO,CAAC,gBAAgB,CACvD,yBAAyB,EACzB,SAAS,EACT,IAAI,CACJ,CAAC;YACF,IAAA,mCAAoB,EAAC,yBAAyB,EAAE,uBAAuB,EAAE,OAAO,CAAC,OAAO,EAAE,CAAC,CAAC;YAG5F,MAAM,aAAa,GAAG,MAAM,UAAU,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC;YAGzD,IAAI,IAAI,GAAG,MAAM,WAAW,CAAC,+BAA+B,CAAC,aAAa,EAAE,IAAI,EAAE,MAAM,CAAC,CAAC;YAG1F,IAAI,WAAW,IAAI,IAAI,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBACpC,MAAM,QAAQ,GAAG,CAAC,MAAM,OAAO,CAAC,sBAAsB,CACrD,kCAAmB,CAAC,UAAU,EAC9B,CAAC,CACD,CAA2B,CAAC;gBAC7B,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,GAAG,CAAC,EAAE,EAAE,CAAC,GAAG,CAAC,CAAC;gBAE3C,MAAM,iBAAiB,GAAG,MAAM,QAAQ,CAAC,iBAAiB,CAAC,SAAS,EAAE,KAAK,CAAC,CAAC;gBAC7E,IAAI,GAAG,iBAAiB,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE;oBACpC,MAAM,EAAE,cAAc,EAAE,GAAG,QAAQ,EAAE,GAAG,GAAG,CAAC,QAAQ,IAAI,EAAE,CAAC;oBAC3D,OAAO,CAAC,EAAE,GAAG,GAAG,EAAE,QAAQ,EAAE,EAAE,cAAc,IAAI,CAAC,CAAC,CAAC;gBACpD,CAAC,CAAC,CAAC;YACJ,CAAC;YAGD,MAAM,cAAc,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,GAAG,CAAC,EAAE,EAAE;gBACzC,IAAI,uBAAuB,EAAE,CAAC;oBAC7B,OAAO;wBACN,IAAI,EAAE,MAAM;wBACZ,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,EAAE,GAAG,GAAG,EAAE,CAAC;qBAChC,CAAC;gBACH,CAAC;qBAAM,CAAC;oBACP,OAAO;wBACN,IAAI,EAAE,MAAM;wBACZ,WAAW,EAAE,IAAI,CAAC,SAAS,CAAC,EAAE,WAAW,EAAE,GAAG,CAAC,WAAW,EAAE,CAAC;qBAC7D,CAAC;gBACH,CAAC;YACF,CAAC,CAAC,CAAC;YAGH,IAAA,yBAAU,EAAC,OAAO,EAAE,0BAA0B,EAAE,EAAE,KAAK,EAAE,KAAK,EAAE,CAAC,CAAC;YAElE,OAAO;gBACN;oBACC,IAAI,EAAE;wBACL,QAAQ,EAAE,cAAc;qBACxB;oBACD,UAAU,EAAE;wBACX,IAAI,EAAE,SAAS;qBACf;iBACD;aACD,CAAC;QACH,CAAC;gBAAS,CAAC;YAEV,IAAI,CAAC,wBAAwB,EAAE,CAAC,WAAW,CAAC,CAAC;QAC9C,CAAC;IACF,CAAC"}

package/dist/cjs/utils/vector-store/createVectorStoreNode/operations/retrieveAsToolOperation.d.ts

@@ -0,0 +1,5 @@
+import type { Embeddings } from '@langchain/core/embeddings';
+import type { VectorStore } from '@langchain/core/vectorstores';
+import type { ISupplyDataFunctions, SupplyData } from 'n8n-workflow';
+import type { VectorStoreNodeConstructorArgs } from '../types';
+export declare function handleRetrieveAsToolOperation<T extends VectorStore = VectorStore>(context: ISupplyDataFunctions, args: VectorStoreNodeConstructorArgs<T>, embeddings: Embeddings, itemIndex: number): Promise<SupplyData>;

package/dist/cjs/utils/vector-store/createVectorStoreNode/operations/retrieveAsToolOperation.js

@@ -0,0 +1,70 @@
+(function (factory) {
+    if (typeof module === "object" && typeof module.exports === "object") {
+        var v = factory(require, exports);
+        if (v !== undefined) module.exports = v;
+    }
+    else if (typeof define === "function" && define.amd) {
+        define(["require", "exports", "n8n-workflow", "../../../fromai-tool-factory", "../../../helpers", "../../../log-wrapper"], factory);
+    }
+})(function (require, exports) {
+    "use strict";
+    Object.defineProperty(exports, "__esModule", { value: true });
+    exports.handleRetrieveAsToolOperation = handleRetrieveAsToolOperation;
+    const n8n_workflow_1 = require("n8n-workflow");
+    const fromai_tool_factory_1 = require("../../../fromai-tool-factory");
+    const helpers_1 = require("../../../helpers");
+    const log_wrapper_1 = require("../../../log-wrapper");
+    async function handleRetrieveAsToolOperation(context, args, embeddings, itemIndex) {
+        const toolDescription = context.getNodeParameter('toolDescription', itemIndex);
+        const node = context.getNode();
+        const { typeVersion } = node;
+        const toolName = typeVersion < 1.3
+            ? context.getNodeParameter('toolName', itemIndex)
+            : (0, n8n_workflow_1.nodeNameToToolName)(node);
+        const vectorStoreTool = (0, fromai_tool_factory_1.createToolFromNode)(node, {
+            name: toolName,
+            description: toolDescription,
+            extraArgs: [{ key: 'input', description: 'Query to search for. Required' }],
+            func: async (query) => {
+                const topK = context.getNodeParameter('topK', itemIndex, 4);
+                const useReranker = context.getNodeParameter('useReranker', itemIndex, false);
+                const includeDocumentMetadata = context.getNodeParameter('includeDocumentMetadata', itemIndex, true);
+                const filter = (0, helpers_1.getMetadataFiltersValues)(context, itemIndex);
+                const queryString = typeof query === 'string' ? query : query.input;
+                (0, n8n_workflow_1.assert)(typeof queryString === 'string', 'Query must be of type string');
+                const vectorStore = await args.getVectorStoreClient(context, undefined, embeddings, itemIndex);
+                try {
+                    const embeddedPrompt = await embeddings.embedQuery(queryString);
+                    let documents = await vectorStore.similaritySearchVectorWithScore(embeddedPrompt, topK, filter);
+                    if (useReranker && documents.length > 0) {
+                        const reranker = (await context.getInputConnectionData(n8n_workflow_1.NodeConnectionTypes.AiReranker, 0));
+                        const docs = documents.map(([doc]) => doc);
+                        const rerankedDocuments = await reranker.compressDocuments(docs, queryString);
+                        documents = rerankedDocuments.map((doc) => {
+                            const { relevanceScore, ...metadata } = doc.metadata;
+                            return [{ ...doc, metadata }, relevanceScore];
+                        });
+                    }
+                    return documents
+                        .map((document) => {
+                        if (includeDocumentMetadata) {
+                            return { type: 'text', text: JSON.stringify(document[0]) };
+                        }
+                        return {
+                            type: 'text',
+                            text: JSON.stringify({ pageContent: document[0].pageContent }),
+                        };
+                    })
+                        .filter((document) => !!document);
+                }
+                finally {
+                    args.releaseVectorStoreClient?.(vectorStore);
+                }
+            },
+        });
+        return {
+            response: (0, log_wrapper_1.logWrapper)(vectorStoreTool, context),
+        };
+    }
+});
+//# sourceMappingURL=retrieveAsToolOperation.js.map

package/dist/cjs/utils/vector-store/createVectorStoreNode/operations/retrieveAsToolOperation.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"retrieveAsToolOperation.js","sourceRoot":"","sources":["../../../../../../src/utils/vector-store/createVectorStoreNode/operations/retrieveAsToolOperation.ts"],"names":[],"mappings":";;;;;;;;;;;IAeA,sEAgGC;IA5GD,+CAA+E;IAG/E,sEAAkE;IAClE,8CAA4D;IAC5D,sDAAkD;IAO3C,KAAK,UAAU,6BAA6B,CAClD,OAA6B,EAC7B,IAAuC,EACvC,UAAsB,EACtB,SAAiB;QAGjB,MAAM,eAAe,GAAG,OAAO,CAAC,gBAAgB,CAAC,iBAAiB,EAAE,SAAS,CAAW,CAAC;QAEzF,MAAM,IAAI,GAAG,OAAO,CAAC,OAAO,EAAE,CAAC;QAC/B,MAAM,EAAE,WAAW,EAAE,GAAG,IAAI,CAAC;QAC7B,MAAM,QAAQ,GACb,WAAW,GAAG,GAAG;YAChB,CAAC,CAAE,OAAO,CAAC,gBAAgB,CAAC,UAAU,EAAE,SAAS,CAAY;YAC7D,CAAC,CAAC,IAAA,iCAAkB,EAAC,IAAI,CAAC,CAAC;QAG7B,MAAM,eAAe,GAAG,IAAA,wCAAkB,EAAC,IAAI,EAAE;YAChD,IAAI,EAAE,QAAQ;YACd,WAAW,EAAE,eAAe;YAC5B,SAAS,EAAE,CAAC,EAAE,GAAG,EAAE,OAAO,EAAE,WAAW,EAAE,+BAA+B,EAAE,CAAC;YAC3E,IAAI,EAAE,KAAK,EAAE,KAAK,EAAE,EAAE;gBACrB,MAAM,IAAI,GAAG,OAAO,CAAC,gBAAgB,CAAC,MAAM,EAAE,SAAS,EAAE,CAAC,CAAW,CAAC;gBACtE,MAAM,WAAW,GAAG,OAAO,CAAC,gBAAgB,CAAC,aAAa,EAAE,SAAS,EAAE,KAAK,CAAY,CAAC;gBACzF,MAAM,uBAAuB,GAAG,OAAO,CAAC,gBAAgB,CACvD,yBAAyB,EACzB,SAAS,EACT,IAAI,CACO,CAAC;gBAGb,MAAM,MAAM,GAAG,IAAA,kCAAwB,EAAC,OAAO,EAAE,SAAS,CAAC,CAAC;gBAG5D,MAAM,WAAW,GAAG,OAAO,KAAK,KAAK,QAAQ,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,KAAK,CAAC;gBACpE,IAAA,qBAAM,EAAC,OAAO,WAAW,KAAK,QAAQ,EAAE,8BAA8B,CAAC,CAAC;gBAKxE,MAAM,WAAW,GAAG,MAAM,IAAI,CAAC,oBAAoB,CAClD,OAAO,EACP,SAAS,EACT,UAAU,EACV,SAAS,CACT,CAAC;gBAEF,IAAI,CAAC;oBAEJ,MAAM,cAAc,GAAG,MAAM,UAAU,CAAC,UAAU,CAAC,WAAW,CAAC,CAAC;oBAGhE,IAAI,SAAS,GAAG,MAAM,WAAW,CAAC,+BAA+B,CAChE,cAAc,EACd,IAAI,EACJ,MAAM,CACN,CAAC;oBAGF,IAAI,WAAW,IAAI,SAAS,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;wBACzC,MAAM,QAAQ,GAAG,CAAC,MAAM,OAAO,CAAC,sBAAsB,CACrD,kCAAmB,CAAC,UAAU,EAC9B,CAAC,CACD,CAA2B,CAAC;wBAE7B,MAAM,IAAI,GAAG,SAAS,CAAC,GAAG,CAAC,CAAC,CAAC,GAAG,CAAC,EAAE,EAAE,CAAC,GAAG,CAAC,CAAC;wBAC3C,MAAM,iBAAiB,GAAG,MAAM,QAAQ,CAAC,iBAAiB,CAAC,IAAI,EAAE,WAAW,CAAC,CAAC;wBAC9E,SAAS,GAAG,iBAAiB,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE;4BACzC,MAAM,EAAE,cAAc,EAAE,GAAG,QAAQ,EAAE,GAAG,GAAG,CAAC,QAAQ,CAAC;4BACrD,OAAO,CAAC,EAAE,GAAG,GAAG,EAAE,QAAQ,EAAE,EAAE,cAAc,CAAC,CAAC;wBAC/C,CAAC,CAAC,CAAC;oBACJ,CAAC;oBAGD,OAAO,SAAS;yBACd,GAAG,CAAC,CAAC,QAAQ,EAAE,EAAE;wBACjB,IAAI,uBAAuB,EAAE,CAAC;4BAC7B,OAAO,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;wBAC5D,CAAC;wBACD,OAAO;4BACN,IAAI,EAAE,MAAM;4BACZ,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,EAAE,WAAW,EAAE,QAAQ,CAAC,CAAC,CAAC,CAAC,WAAW,EAAE,CAAC;yBAC9D,CAAC;oBACH,CAAC,CAAC;yBACD,MAAM,CAAC,CAAC,QAAQ,EAAE,EAAE,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC;gBACpC,CAAC;wBAAS,CAAC;oBAEV,IAAI,CAAC,wBAAwB,EAAE,CAAC,WAAW,CAAC,CAAC;gBAC9C,CAAC;YACF,CAAC;SACD,CAAC,CAAC;QAGH,OAAO;YACN,QAAQ,EAAE,IAAA,wBAAU,EAAC,eAAe,EAAE,OAAO,CAAC;SAC9C,CAAC;IACH,CAAC"}

package/dist/cjs/utils/vector-store/createVectorStoreNode/operations/retrieveOperation.d.ts

@@ -0,0 +1,5 @@
+import type { Embeddings } from '@langchain/core/embeddings';
+import type { VectorStore } from '@langchain/core/vectorstores';
+import type { ISupplyDataFunctions, SupplyData } from 'n8n-workflow';
+import type { VectorStoreNodeConstructorArgs } from '../types';
+export declare function handleRetrieveOperation<T extends VectorStore = VectorStore>(context: ISupplyDataFunctions, args: VectorStoreNodeConstructorArgs<T>, embeddings: Embeddings, itemIndex: number): Promise<SupplyData>;

package/dist/cjs/utils/vector-store/createVectorStoreNode/operations/retrieveOperation.js

@@ -0,0 +1,39 @@
+(function (factory) {
+    if (typeof module === "object" && typeof module.exports === "object") {
+        var v = factory(require, exports);
+        if (v !== undefined) module.exports = v;
+    }
+    else if (typeof define === "function" && define.amd) {
+        define(["require", "exports", "n8n-workflow", "../../../helpers", "../../../log-wrapper"], factory);
+    }
+})(function (require, exports) {
+    "use strict";
+    Object.defineProperty(exports, "__esModule", { value: true });
+    exports.handleRetrieveOperation = handleRetrieveOperation;
+    const n8n_workflow_1 = require("n8n-workflow");
+    const helpers_1 = require("../../../helpers");
+    const log_wrapper_1 = require("../../../log-wrapper");
+    async function handleRetrieveOperation(context, args, embeddings, itemIndex) {
+        const filter = (0, helpers_1.getMetadataFiltersValues)(context, itemIndex);
+        const useReranker = context.getNodeParameter('useReranker', itemIndex, false);
+        const vectorStore = await args.getVectorStoreClient(context, filter, embeddings, itemIndex);
+        let response = vectorStore;
+        if (useReranker) {
+            const reranker = (await context.getInputConnectionData(n8n_workflow_1.NodeConnectionTypes.AiReranker, 0));
+            response = {
+                reranker,
+                vectorStore: (0, log_wrapper_1.logWrapper)(vectorStore, context),
+            };
+        }
+        else {
+            response = (0, log_wrapper_1.logWrapper)(vectorStore, context);
+        }
+        return {
+            response,
+            closeFunction: async () => {
+                args.releaseVectorStoreClient?.(vectorStore);
+            },
+        };
+    }
+});
+//# sourceMappingURL=retrieveOperation.js.map

package/dist/cjs/utils/vector-store/createVectorStoreNode/operations/retrieveOperation.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"retrieveOperation.js","sourceRoot":"","sources":["../../../../../../src/utils/vector-store/createVectorStoreNode/operations/retrieveOperation.ts"],"names":[],"mappings":";;;;;;;;;;;IAcA,0DAsCC;IAjDD,+CAAmD;IAGnD,8CAA4D;IAC5D,sDAAkD;IAO3C,KAAK,UAAU,uBAAuB,CAC5C,OAA6B,EAC7B,IAAuC,EACvC,UAAsB,EACtB,SAAiB;QAGjB,MAAM,MAAM,GAAG,IAAA,kCAAwB,EAAC,OAAO,EAAE,SAAS,CAAC,CAAC;QAC5D,MAAM,WAAW,GAAG,OAAO,CAAC,gBAAgB,CAAC,aAAa,EAAE,SAAS,EAAE,KAAK,CAAY,CAAC;QAGzF,MAAM,WAAW,GAAG,MAAM,IAAI,CAAC,oBAAoB,CAAC,OAAO,EAAE,MAAM,EAAE,UAAU,EAAE,SAAS,CAAC,CAAC;QAC5F,IAAI,QAAQ,GACX,WAAW,CAAC;QAEb,IAAI,WAAW,EAAE,CAAC;YACjB,MAAM,QAAQ,GAAG,CAAC,MAAM,OAAO,CAAC,sBAAsB,CACrD,kCAAmB,CAAC,UAAU,EAC9B,CAAC,CACD,CAA2B,CAAC;YAG7B,QAAQ,GAAG;gBACV,QAAQ;gBACR,WAAW,EAAE,IAAA,wBAAU,EAAC,WAAW,EAAE,OAAO,CAAC;aAC7C,CAAC;QACH,CAAC;aAAM,CAAC;YAEP,QAAQ,GAAG,IAAA,wBAAU,EAAC,WAAW,EAAE,OAAO,CAAC,CAAC;QAC7C,CAAC;QAED,OAAO;YACN,QAAQ;YACR,aAAa,EAAE,KAAK,IAAI,EAAE;gBAEzB,IAAI,CAAC,wBAAwB,EAAE,CAAC,WAAW,CAAC,CAAC;YAC9C,CAAC;SACD,CAAC;IACH,CAAC"}

package/dist/cjs/utils/vector-store/createVectorStoreNode/operations/updateOperation.d.ts

@@ -0,0 +1,5 @@
+import type { Embeddings } from '@langchain/core/embeddings';
+import type { VectorStore } from '@langchain/core/vectorstores';
+import type { IExecuteFunctions, INodeExecutionData } from 'n8n-workflow';
+import type { VectorStoreNodeConstructorArgs } from '../types';
+export declare function handleUpdateOperation<T extends VectorStore = VectorStore>(context: IExecuteFunctions, args: VectorStoreNodeConstructorArgs<T>, embeddings: Embeddings): Promise<INodeExecutionData[]>;

package/dist/cjs/utils/vector-store/createVectorStoreNode/operations/updateOperation.js

@@ -0,0 +1,49 @@
+(function (factory) {
+    if (typeof module === "object" && typeof module.exports === "object") {
+        var v = factory(require, exports);
+        if (v !== undefined) module.exports = v;
+    }
+    else if (typeof define === "function" && define.amd) {
+        define(["require", "exports", "n8n-workflow", "../../../log-ai-event", "../../../n8n-json-loader", "../../processDocuments", "../utils"], factory);
+    }
+})(function (require, exports) {
+    "use strict";
+    Object.defineProperty(exports, "__esModule", { value: true });
+    exports.handleUpdateOperation = handleUpdateOperation;
+    const n8n_workflow_1 = require("n8n-workflow");
+    const log_ai_event_1 = require("../../../log-ai-event");
+    const n8n_json_loader_1 = require("../../../n8n-json-loader");
+    const processDocuments_1 = require("../../processDocuments");
+    const utils_1 = require("../utils");
+    async function handleUpdateOperation(context, args, embeddings) {
+        if (!(0, utils_1.isUpdateSupported)(args)) {
+            throw new n8n_workflow_1.NodeOperationError(context.getNode(), 'Update operation is not implemented for this Vector Store');
+        }
+        const items = context.getInputData();
+        const loader = new n8n_json_loader_1.N8nJsonLoader(context);
+        const resultData = [];
+        for (let itemIndex = 0; itemIndex < items.length; itemIndex++) {
+            const itemData = items[itemIndex];
+            const documentId = context.getNodeParameter('id', itemIndex, '', {
+                extractValue: true,
+            });
+            const vectorStore = await args.getVectorStoreClient(context, undefined, embeddings, itemIndex);
+            try {
+                const { processedDocuments, serializedDocuments } = await (0, processDocuments_1.processDocument)(loader, itemData, itemIndex);
+                if (processedDocuments?.length !== 1) {
+                    throw new n8n_workflow_1.NodeOperationError(context.getNode(), 'Single document per item expected');
+                }
+                resultData.push(...serializedDocuments);
+                await vectorStore.addDocuments(processedDocuments, {
+                    ids: [documentId],
+                });
+                (0, log_ai_event_1.logAiEvent)(context, 'ai-vector-store-updated');
+            }
+            finally {
+                args.releaseVectorStoreClient?.(vectorStore);
+            }
+        }
+        return resultData;
+    }
+});
+//# sourceMappingURL=updateOperation.js.map

package/dist/cjs/utils/vector-store/createVectorStoreNode/operations/updateOperation.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"updateOperation.js","sourceRoot":"","sources":["../../../../../../src/utils/vector-store/createVectorStoreNode/operations/updateOperation.ts"],"names":[],"mappings":";;;;;;;;;;;IAeA,sDA8DC;IA3ED,+CAAkD;IAGlD,wDAAmD;IACnD,8DAAyD;IACzD,6DAAyD;IAEzD,oCAA6C;IAMtC,KAAK,UAAU,qBAAqB,CAC1C,OAA0B,EAC1B,IAAuC,EACvC,UAAsB;QAGtB,IAAI,CAAC,IAAA,yBAAiB,EAAC,IAAI,CAAC,EAAE,CAAC;YAC9B,MAAM,IAAI,iCAAkB,CAC3B,OAAO,CAAC,OAAO,EAAE,EACjB,2DAA2D,CAC3D,CAAC;QACH,CAAC;QAGD,MAAM,KAAK,GAAG,OAAO,CAAC,YAAY,EAAE,CAAC;QAErC,MAAM,MAAM,GAAG,IAAI,+BAAa,CAAC,OAAO,CAAC,CAAC;QAE1C,MAAM,UAAU,GAAyB,EAAE,CAAC;QAG5C,KAAK,IAAI,SAAS,GAAG,CAAC,EAAE,SAAS,GAAG,KAAK,CAAC,MAAM,EAAE,SAAS,EAAE,EAAE,CAAC;YAC/D,MAAM,QAAQ,GAAG,KAAK,CAAC,SAAS,CAAC,CAAC;YAGlC,MAAM,UAAU,GAAG,OAAO,CAAC,gBAAgB,CAAC,IAAI,EAAE,SAAS,EAAE,EAAE,EAAE;gBAChE,YAAY,EAAE,IAAI;aAClB,CAAW,CAAC;YAGb,MAAM,WAAW,GAAG,MAAM,IAAI,CAAC,oBAAoB,CAAC,OAAO,EAAE,SAAS,EAAE,UAAU,EAAE,SAAS,CAAC,CAAC;YAE/F,IAAI,CAAC;gBAEJ,MAAM,EAAE,kBAAkB,EAAE,mBAAmB,EAAE,GAAG,MAAM,IAAA,kCAAe,EACxE,MAAM,EACN,QAAQ,EACR,SAAS,CACT,CAAC;gBAGF,IAAI,kBAAkB,EAAE,MAAM,KAAK,CAAC,EAAE,CAAC;oBACtC,MAAM,IAAI,iCAAkB,CAAC,OAAO,CAAC,OAAO,EAAE,EAAE,mCAAmC,CAAC,CAAC;gBACtF,CAAC;gBAGD,UAAU,CAAC,IAAI,CAAC,GAAG,mBAAmB,CAAC,CAAC;gBAGxC,MAAM,WAAW,CAAC,YAAY,CAAC,kBAAkB,EAAE;oBAClD,GAAG,EAAE,CAAC,UAAU,CAAC;iBACjB,CAAC,CAAC;gBAGH,IAAA,yBAAU,EAAC,OAAO,EAAE,yBAAyB,CAAC,CAAC;YAChD,CAAC;oBAAS,CAAC;gBAEV,IAAI,CAAC,wBAAwB,EAAE,CAAC,WAAW,CAAC,CAAC;YAC9C,CAAC;QACF,CAAC;QAED,OAAO,UAAU,CAAC;IACnB,CAAC"}

package/dist/cjs/utils/vector-store/createVectorStoreNode/types.d.ts

@@ -0,0 +1,39 @@
+import type { Document } from '@langchain/core/documents';
+import type { Embeddings } from '@langchain/core/embeddings';
+import type { VectorStore } from '@langchain/core/vectorstores';
+import type { IExecuteFunctions, INodeCredentialDescription, INodeProperties, ILoadOptionsFunctions, INodeListSearchResult, Icon, ISupplyDataFunctions, ThemeIconColor, IDataObject, NodeParameterValueType, IBuilderHint } from 'n8n-workflow';
+export type NodeOperationMode = 'insert' | 'load' | 'retrieve' | 'update' | 'retrieve-as-tool';
+export interface NodeMeta {
+    displayName: string;
+    name: string;
+    hidden?: boolean;
+    description: string;
+    docsUrl: string;
+    icon: Icon;
+    iconColor?: ThemeIconColor;
+    credentials?: INodeCredentialDescription[];
+    operationModes?: NodeOperationMode[];
+    categories?: string[];
+    subcategories?: Record<string, string[]>;
+    builderHint?: IBuilderHint;
+}
+export interface VectorStoreNodeConstructorArgs<T extends VectorStore = VectorStore> {
+    meta: NodeMeta;
+    methods?: {
+        listSearch?: {
+            [key: string]: (this: ILoadOptionsFunctions, filter?: string, paginationToken?: string) => Promise<INodeListSearchResult>;
+        };
+        actionHandler?: {
+            [functionName: string]: (this: ILoadOptionsFunctions, payload: IDataObject | string | undefined) => Promise<NodeParameterValueType>;
+        };
+    };
+    sharedFields: INodeProperties[];
+    insertFields?: INodeProperties[];
+    loadFields?: INodeProperties[];
+    retrieveFields?: INodeProperties[];
+    updateFields?: INodeProperties[];
+    beforeInsert?: (context: IExecuteFunctions | ISupplyDataFunctions, embeddings: Embeddings, itemIndex: number) => Promise<void>;
+    populateVectorStore: (context: IExecuteFunctions | ISupplyDataFunctions, embeddings: Embeddings, documents: Array<Document<Record<string, unknown>>>, itemIndex: number) => Promise<void>;
+    getVectorStoreClient: (context: IExecuteFunctions | ISupplyDataFunctions, filter: Record<string, never> | undefined, embeddings: Embeddings, itemIndex: number) => Promise<T>;
+    releaseVectorStoreClient?: (vectorStore: T) => void;
+}

package/dist/cjs/utils/vector-store/createVectorStoreNode/types.js

@@ -0,0 +1,13 @@
+(function (factory) {
+    if (typeof module === "object" && typeof module.exports === "object") {
+        var v = factory(require, exports);
+        if (v !== undefined) module.exports = v;
+    }
+    else if (typeof define === "function" && define.amd) {
+        define(["require", "exports"], factory);
+    }
+})(function (require, exports) {
+    "use strict";
+    Object.defineProperty(exports, "__esModule", { value: true });
+});
+//# sourceMappingURL=types.js.map

package/dist/cjs/utils/vector-store/createVectorStoreNode/types.js.map

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

package/dist/cjs/utils/vector-store/createVectorStoreNode/utils.d.ts

@@ -0,0 +1,6 @@
+import type { VectorStore } from '@langchain/core/vectorstores';
+import type { INodeProperties, INodePropertyOptions } from 'n8n-workflow';
+import type { NodeOperationMode, VectorStoreNodeConstructorArgs } from './types';
+export declare function transformDescriptionForOperationMode(fields: INodeProperties[], mode: NodeOperationMode | NodeOperationMode[]): INodeProperties[];
+export declare function isUpdateSupported<T extends VectorStore>(args: VectorStoreNodeConstructorArgs<T>): boolean;
+export declare function getOperationModeOptions<T extends VectorStore>(args: VectorStoreNodeConstructorArgs<T>): INodePropertyOptions[];

package/dist/cjs/utils/vector-store/createVectorStoreNode/utils.js

@@ -0,0 +1,30 @@
+(function (factory) {
+    if (typeof module === "object" && typeof module.exports === "object") {
+        var v = factory(require, exports);
+        if (v !== undefined) module.exports = v;
+    }
+    else if (typeof define === "function" && define.amd) {
+        define(["require", "exports", "./constants"], factory);
+    }
+})(function (require, exports) {
+    "use strict";
+    Object.defineProperty(exports, "__esModule", { value: true });
+    exports.transformDescriptionForOperationMode = transformDescriptionForOperationMode;
+    exports.isUpdateSupported = isUpdateSupported;
+    exports.getOperationModeOptions = getOperationModeOptions;
+    const constants_1 = require("./constants");
+    function transformDescriptionForOperationMode(fields, mode) {
+        return fields.map((field) => ({
+            ...field,
+            displayOptions: { show: { mode: Array.isArray(mode) ? mode : [mode] } },
+        }));
+    }
+    function isUpdateSupported(args) {
+        return args.meta.operationModes?.includes('update') ?? false;
+    }
+    function getOperationModeOptions(args) {
+        const enabledOperationModes = args.meta.operationModes ?? constants_1.DEFAULT_OPERATION_MODES;
+        return constants_1.OPERATION_MODE_DESCRIPTIONS.filter(({ value }) => enabledOperationModes.includes(value));
+    }
+});
+//# sourceMappingURL=utils.js.map

package/dist/cjs/utils/vector-store/createVectorStoreNode/utils.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"utils.js","sourceRoot":"","sources":["../../../../../src/utils/vector-store/createVectorStoreNode/utils.ts"],"names":[],"mappings":";;;;;;;;;;;IAUA,oFAQC;IAMD,8CAIC;IAMD,0DAQC;IAvCD,2CAAmF;IAOnF,SAAgB,oCAAoC,CACnD,MAAyB,EACzB,IAA6C;QAE7C,OAAO,MAAM,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC;YAC7B,GAAG,KAAK;YACR,cAAc,EAAE,EAAE,IAAI,EAAE,EAAE,IAAI,EAAE,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,EAAE,EAAE;SACvE,CAAC,CAAC,CAAC;IACL,CAAC;IAMD,SAAgB,iBAAiB,CAChC,IAAuC;QAEvC,OAAO,IAAI,CAAC,IAAI,CAAC,cAAc,EAAE,QAAQ,CAAC,QAAQ,CAAC,IAAI,KAAK,CAAC;IAC9D,CAAC;IAMD,SAAgB,uBAAuB,CACtC,IAAuC;QAEvC,MAAM,qBAAqB,GAAG,IAAI,CAAC,IAAI,CAAC,cAAc,IAAI,mCAAuB,CAAC;QAElF,OAAO,uCAA2B,CAAC,MAAM,CAAC,CAAC,EAAE,KAAK,EAAE,EAAE,EAAE,CACvD,qBAAqB,CAAC,QAAQ,CAAC,KAA0B,CAAC,CAC1D,CAAC;IACH,CAAC"}

package/dist/cjs/utils/vector-store/processDocuments.d.ts

@@ -0,0 +1,25 @@
+import type { Document } from '@langchain/core/documents';
+import type { INodeExecutionData } from 'n8n-workflow';
+import { N8nBinaryLoader } from '../n8n-binary-loader';
+import { N8nJsonLoader } from '../n8n-json-loader';
+export declare function processDocuments(documentInput: N8nJsonLoader | N8nBinaryLoader | Array<Document<Record<string, unknown>>>, inputItems: INodeExecutionData[]): Promise<{
+    processedDocuments: Document<Record<string, any>>[];
+    serializedDocuments: {
+        json: {
+            metadata: Record<string, any>;
+            pageContent: string;
+        };
+    }[];
+}>;
+export declare function processDocument(documentInput: N8nJsonLoader | N8nBinaryLoader | Array<Document<Record<string, unknown>>>, inputItem: INodeExecutionData, itemIndex: number): Promise<{
+    processedDocuments: Document<Record<string, any>>[];
+    serializedDocuments: {
+        json: {
+            metadata: Record<string, any>;
+            pageContent: string;
+        };
+        pairedItem: {
+            item: number;
+        };
+    }[];
+}>;