@n8n/ai-utilities 0.4.2 → 0.5.1

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

@@ -0,0 +1,64 @@
+"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/esm/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":";;AAeA,oFA0FC;AAtGD,+CAA8F;AAG9F,8CAA4D;AAC5D,wDAAmD;AAQ5C,KAAK,UAAU,oCAAoC,CACzD,OAA0B,EAC1B,IAAuC,EACvC,UAAsB,EACtB,SAAiB;IAEjB,MAAM,MAAM,GAAG,IAAA,kCAAwB,EAAC,OAAO,EAAE,SAAS,CAAC,CAAC;IAC5D,MAAM,WAAW,GAAG,MAAM,IAAI,CAAC,oBAAoB,CAClD,OAAO,EAEP,SAAS,EACT,UAAU,EACV,SAAS,CACT,CAAC;IAEF,IAAI,CAAC;QAEJ,MAAM,SAAS,GAAG,OAAO,CAAC,YAAY,EAAE,CAAC;QACzC,MAAM,IAAI,GAAG,SAAS,CAAC,SAAS,CAAC,CAAC;QAClC,MAAM,KAAK,GAAG,OAAO,IAAI,CAAC,IAAI,CAAC,KAAK,KAAK,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,SAAS,CAAC;QAEhF,IAAI,CAAC,KAAK,IAAI,OAAO,KAAK,KAAK,QAAQ,EAAE,CAAC;YACzC,MAAM,IAAI,KAAK,CAAC,+DAA+D,CAAC,CAAC;QAClF,CAAC;QAED,MAAM,IAAI,GAAG,OAAO,CAAC,gBAAgB,CAAC,MAAM,EAAE,SAAS,EAAE,CAAC,CAAC,CAAC;QAC5D,IAAA,kCAAmB,EAAC,MAAM,EAAE,IAAI,EAAE,OAAO,CAAC,OAAO,EAAE,CAAC,CAAC;QACrD,MAAM,WAAW,GAAG,OAAO,CAAC,gBAAgB,CAAC,aAAa,EAAE,SAAS,EAAE,KAAK,CAAC,CAAC;QAC9E,IAAA,mCAAoB,EAAC,aAAa,EAAE,WAAW,EAAE,OAAO,CAAC,OAAO,EAAE,CAAC,CAAC;QAEpE,MAAM,uBAAuB,GAAG,OAAO,CAAC,gBAAgB,CACvD,yBAAyB,EACzB,SAAS,EACT,IAAI,CACJ,CAAC;QACF,IAAA,mCAAoB,EAAC,yBAAyB,EAAE,uBAAuB,EAAE,OAAO,CAAC,OAAO,EAAE,CAAC,CAAC;QAG5F,MAAM,aAAa,GAAG,MAAM,UAAU,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC;QAGzD,IAAI,IAAI,GAAG,MAAM,WAAW,CAAC,+BAA+B,CAAC,aAAa,EAAE,IAAI,EAAE,MAAM,CAAC,CAAC;QAG1F,IAAI,WAAW,IAAI,IAAI,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACpC,MAAM,QAAQ,GAAG,CAAC,MAAM,OAAO,CAAC,sBAAsB,CACrD,kCAAmB,CAAC,UAAU,EAC9B,CAAC,CACD,CAA2B,CAAC;YAC7B,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,GAAG,CAAC,EAAE,EAAE,CAAC,GAAG,CAAC,CAAC;YAE3C,MAAM,iBAAiB,GAAG,MAAM,QAAQ,CAAC,iBAAiB,CAAC,SAAS,EAAE,KAAK,CAAC,CAAC;YAC7E,IAAI,GAAG,iBAAiB,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE;gBACpC,MAAM,EAAE,cAAc,EAAE,GAAG,QAAQ,EAAE,GAAG,GAAG,CAAC,QAAQ,IAAI,EAAE,CAAC;gBAC3D,OAAO,CAAC,EAAE,GAAG,GAAG,EAAE,QAAQ,EAAE,EAAE,cAAc,IAAI,CAAC,CAAC,CAAC;YACpD,CAAC,CAAC,CAAC;QACJ,CAAC;QAGD,MAAM,cAAc,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,GAAG,CAAC,EAAE,EAAE;YACzC,IAAI,uBAAuB,EAAE,CAAC;gBAC7B,OAAO;oBACN,IAAI,EAAE,MAAM;oBACZ,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,EAAE,GAAG,GAAG,EAAE,CAAC;iBAChC,CAAC;YACH,CAAC;iBAAM,CAAC;gBACP,OAAO;oBACN,IAAI,EAAE,MAAM;oBACZ,WAAW,EAAE,IAAI,CAAC,SAAS,CAAC,EAAE,WAAW,EAAE,GAAG,CAAC,WAAW,EAAE,CAAC;iBAC7D,CAAC;YACH,CAAC;QACF,CAAC,CAAC,CAAC;QAGH,IAAA,yBAAU,EAAC,OAAO,EAAE,0BAA0B,EAAE,EAAE,KAAK,EAAE,KAAK,EAAE,CAAC,CAAC;QAElE,OAAO;YACN;gBACC,IAAI,EAAE;oBACL,QAAQ,EAAE,cAAc;iBACxB;gBACD,UAAU,EAAE;oBACX,IAAI,EAAE,SAAS;iBACf;aACD;SACD,CAAC;IACH,CAAC;YAAS,CAAC;QAEV,IAAI,CAAC,wBAAwB,EAAE,CAAC,WAAW,CAAC,CAAC;IAC9C,CAAC;AACF,CAAC"}

package/dist/esm/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/esm/utils/vector-store/createVectorStoreNode/operations/retrieveAsToolOperation.js

@@ -0,0 +1,60 @@
+"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/esm/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":";;AAeA,sEAgGC;AA5GD,+CAA+E;AAG/E,sEAAkE;AAClE,8CAA4D;AAC5D,sDAAkD;AAO3C,KAAK,UAAU,6BAA6B,CAClD,OAA6B,EAC7B,IAAuC,EACvC,UAAsB,EACtB,SAAiB;IAGjB,MAAM,eAAe,GAAG,OAAO,CAAC,gBAAgB,CAAC,iBAAiB,EAAE,SAAS,CAAW,CAAC;IAEzF,MAAM,IAAI,GAAG,OAAO,CAAC,OAAO,EAAE,CAAC;IAC/B,MAAM,EAAE,WAAW,EAAE,GAAG,IAAI,CAAC;IAC7B,MAAM,QAAQ,GACb,WAAW,GAAG,GAAG;QAChB,CAAC,CAAE,OAAO,CAAC,gBAAgB,CAAC,UAAU,EAAE,SAAS,CAAY;QAC7D,CAAC,CAAC,IAAA,iCAAkB,EAAC,IAAI,CAAC,CAAC;IAG7B,MAAM,eAAe,GAAG,IAAA,wCAAkB,EAAC,IAAI,EAAE;QAChD,IAAI,EAAE,QAAQ;QACd,WAAW,EAAE,eAAe;QAC5B,SAAS,EAAE,CAAC,EAAE,GAAG,EAAE,OAAO,EAAE,WAAW,EAAE,+BAA+B,EAAE,CAAC;QAC3E,IAAI,EAAE,KAAK,EAAE,KAAK,EAAE,EAAE;YACrB,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;YACzF,MAAM,uBAAuB,GAAG,OAAO,CAAC,gBAAgB,CACvD,yBAAyB,EACzB,SAAS,EACT,IAAI,CACO,CAAC;YAGb,MAAM,MAAM,GAAG,IAAA,kCAAwB,EAAC,OAAO,EAAE,SAAS,CAAC,CAAC;YAG5D,MAAM,WAAW,GAAG,OAAO,KAAK,KAAK,QAAQ,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,KAAK,CAAC;YACpE,IAAA,qBAAM,EAAC,OAAO,WAAW,KAAK,QAAQ,EAAE,8BAA8B,CAAC,CAAC;YAKxE,MAAM,WAAW,GAAG,MAAM,IAAI,CAAC,oBAAoB,CAClD,OAAO,EACP,SAAS,EACT,UAAU,EACV,SAAS,CACT,CAAC;YAEF,IAAI,CAAC;gBAEJ,MAAM,cAAc,GAAG,MAAM,UAAU,CAAC,UAAU,CAAC,WAAW,CAAC,CAAC;gBAGhE,IAAI,SAAS,GAAG,MAAM,WAAW,CAAC,+BAA+B,CAChE,cAAc,EACd,IAAI,EACJ,MAAM,CACN,CAAC;gBAGF,IAAI,WAAW,IAAI,SAAS,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;oBACzC,MAAM,QAAQ,GAAG,CAAC,MAAM,OAAO,CAAC,sBAAsB,CACrD,kCAAmB,CAAC,UAAU,EAC9B,CAAC,CACD,CAA2B,CAAC;oBAE7B,MAAM,IAAI,GAAG,SAAS,CAAC,GAAG,CAAC,CAAC,CAAC,GAAG,CAAC,EAAE,EAAE,CAAC,GAAG,CAAC,CAAC;oBAC3C,MAAM,iBAAiB,GAAG,MAAM,QAAQ,CAAC,iBAAiB,CAAC,IAAI,EAAE,WAAW,CAAC,CAAC;oBAC9E,SAAS,GAAG,iBAAiB,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE;wBACzC,MAAM,EAAE,cAAc,EAAE,GAAG,QAAQ,EAAE,GAAG,GAAG,CAAC,QAAQ,CAAC;wBACrD,OAAO,CAAC,EAAE,GAAG,GAAG,EAAE,QAAQ,EAAE,EAAE,cAAc,CAAC,CAAC;oBAC/C,CAAC,CAAC,CAAC;gBACJ,CAAC;gBAGD,OAAO,SAAS;qBACd,GAAG,CAAC,CAAC,QAAQ,EAAE,EAAE;oBACjB,IAAI,uBAAuB,EAAE,CAAC;wBAC7B,OAAO,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;oBAC5D,CAAC;oBACD,OAAO;wBACN,IAAI,EAAE,MAAM;wBACZ,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,EAAE,WAAW,EAAE,QAAQ,CAAC,CAAC,CAAC,CAAC,WAAW,EAAE,CAAC;qBAC9D,CAAC;gBACH,CAAC,CAAC;qBACD,MAAM,CAAC,CAAC,QAAQ,EAAE,EAAE,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC;YACpC,CAAC;oBAAS,CAAC;gBAEV,IAAI,CAAC,wBAAwB,EAAE,CAAC,WAAW,CAAC,CAAC;YAC9C,CAAC;QACF,CAAC;KACD,CAAC,CAAC;IAGH,OAAO;QACN,QAAQ,EAAE,IAAA,wBAAU,EAAC,eAAe,EAAE,OAAO,CAAC;KAC9C,CAAC;AACH,CAAC"}

package/dist/esm/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/esm/utils/vector-store/createVectorStoreNode/operations/retrieveOperation.js

@@ -0,0 +1,29 @@
+"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/esm/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":";;AAcA,0DAsCC;AAjDD,+CAAmD;AAGnD,8CAA4D;AAC5D,sDAAkD;AAO3C,KAAK,UAAU,uBAAuB,CAC5C,OAA6B,EAC7B,IAAuC,EACvC,UAAsB,EACtB,SAAiB;IAGjB,MAAM,MAAM,GAAG,IAAA,kCAAwB,EAAC,OAAO,EAAE,SAAS,CAAC,CAAC;IAC5D,MAAM,WAAW,GAAG,OAAO,CAAC,gBAAgB,CAAC,aAAa,EAAE,SAAS,EAAE,KAAK,CAAY,CAAC;IAGzF,MAAM,WAAW,GAAG,MAAM,IAAI,CAAC,oBAAoB,CAAC,OAAO,EAAE,MAAM,EAAE,UAAU,EAAE,SAAS,CAAC,CAAC;IAC5F,IAAI,QAAQ,GACX,WAAW,CAAC;IAEb,IAAI,WAAW,EAAE,CAAC;QACjB,MAAM,QAAQ,GAAG,CAAC,MAAM,OAAO,CAAC,sBAAsB,CACrD,kCAAmB,CAAC,UAAU,EAC9B,CAAC,CACD,CAA2B,CAAC;QAG7B,QAAQ,GAAG;YACV,QAAQ;YACR,WAAW,EAAE,IAAA,wBAAU,EAAC,WAAW,EAAE,OAAO,CAAC;SAC7C,CAAC;IACH,CAAC;SAAM,CAAC;QAEP,QAAQ,GAAG,IAAA,wBAAU,EAAC,WAAW,EAAE,OAAO,CAAC,CAAC;IAC7C,CAAC;IAED,OAAO;QACN,QAAQ;QACR,aAAa,EAAE,KAAK,IAAI,EAAE;YAEzB,IAAI,CAAC,wBAAwB,EAAE,CAAC,WAAW,CAAC,CAAC;QAC9C,CAAC;KACD,CAAC;AACH,CAAC"}

package/dist/esm/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/esm/utils/vector-store/createVectorStoreNode/operations/updateOperation.js

@@ -0,0 +1,39 @@
+"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/esm/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":";;AAeA,sDA8DC;AA3ED,+CAAkD;AAGlD,wDAAmD;AACnD,8DAAyD;AACzD,6DAAyD;AAEzD,oCAA6C;AAMtC,KAAK,UAAU,qBAAqB,CAC1C,OAA0B,EAC1B,IAAuC,EACvC,UAAsB;IAGtB,IAAI,CAAC,IAAA,yBAAiB,EAAC,IAAI,CAAC,EAAE,CAAC;QAC9B,MAAM,IAAI,iCAAkB,CAC3B,OAAO,CAAC,OAAO,EAAE,EACjB,2DAA2D,CAC3D,CAAC;IACH,CAAC;IAGD,MAAM,KAAK,GAAG,OAAO,CAAC,YAAY,EAAE,CAAC;IAErC,MAAM,MAAM,GAAG,IAAI,+BAAa,CAAC,OAAO,CAAC,CAAC;IAE1C,MAAM,UAAU,GAAyB,EAAE,CAAC;IAG5C,KAAK,IAAI,SAAS,GAAG,CAAC,EAAE,SAAS,GAAG,KAAK,CAAC,MAAM,EAAE,SAAS,EAAE,EAAE,CAAC;QAC/D,MAAM,QAAQ,GAAG,KAAK,CAAC,SAAS,CAAC,CAAC;QAGlC,MAAM,UAAU,GAAG,OAAO,CAAC,gBAAgB,CAAC,IAAI,EAAE,SAAS,EAAE,EAAE,EAAE;YAChE,YAAY,EAAE,IAAI;SAClB,CAAW,CAAC;QAGb,MAAM,WAAW,GAAG,MAAM,IAAI,CAAC,oBAAoB,CAAC,OAAO,EAAE,SAAS,EAAE,UAAU,EAAE,SAAS,CAAC,CAAC;QAE/F,IAAI,CAAC;YAEJ,MAAM,EAAE,kBAAkB,EAAE,mBAAmB,EAAE,GAAG,MAAM,IAAA,kCAAe,EACxE,MAAM,EACN,QAAQ,EACR,SAAS,CACT,CAAC;YAGF,IAAI,kBAAkB,EAAE,MAAM,KAAK,CAAC,EAAE,CAAC;gBACtC,MAAM,IAAI,iCAAkB,CAAC,OAAO,CAAC,OAAO,EAAE,EAAE,mCAAmC,CAAC,CAAC;YACtF,CAAC;YAGD,UAAU,CAAC,IAAI,CAAC,GAAG,mBAAmB,CAAC,CAAC;YAGxC,MAAM,WAAW,CAAC,YAAY,CAAC,kBAAkB,EAAE;gBAClD,GAAG,EAAE,CAAC,UAAU,CAAC;aACjB,CAAC,CAAC;YAGH,IAAA,yBAAU,EAAC,OAAO,EAAE,yBAAyB,CAAC,CAAC;QAChD,CAAC;gBAAS,CAAC;YAEV,IAAI,CAAC,wBAAwB,EAAE,CAAC,WAAW,CAAC,CAAC;QAC9C,CAAC;IACF,CAAC;IAED,OAAO,UAAU,CAAC;AACnB,CAAC"}

package/dist/esm/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/esm/utils/vector-store/createVectorStoreNode/types.js

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

package/dist/esm/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/esm/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/esm/utils/vector-store/createVectorStoreNode/utils.js

@@ -0,0 +1,20 @@
+"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/esm/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":";;AAUA,oFAQC;AAMD,8CAIC;AAMD,0DAQC;AAvCD,2CAAmF;AAOnF,SAAgB,oCAAoC,CACnD,MAAyB,EACzB,IAA6C;IAE7C,OAAO,MAAM,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC;QAC7B,GAAG,KAAK;QACR,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;KACvE,CAAC,CAAC,CAAC;AACL,CAAC;AAMD,SAAgB,iBAAiB,CAChC,IAAuC;IAEvC,OAAO,IAAI,CAAC,IAAI,CAAC,cAAc,EAAE,QAAQ,CAAC,QAAQ,CAAC,IAAI,KAAK,CAAC;AAC9D,CAAC;AAMD,SAAgB,uBAAuB,CACtC,IAAuC;IAEvC,MAAM,qBAAqB,GAAG,IAAI,CAAC,IAAI,CAAC,cAAc,IAAI,mCAAuB,CAAC;IAElF,OAAO,uCAA2B,CAAC,MAAM,CAAC,CAAC,EAAE,KAAK,EAAE,EAAE,EAAE,CACvD,qBAAqB,CAAC,QAAQ,CAAC,KAA0B,CAAC,CAC1D,CAAC;AACH,CAAC"}

package/dist/esm/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;
+        };
+    }[];
+}>;

package/dist/esm/utils/vector-store/processDocuments.js

@@ -0,0 +1,42 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.processDocuments = processDocuments;
+exports.processDocument = processDocument;
+const n8n_binary_loader_1 = require("../n8n-binary-loader");
+const n8n_json_loader_1 = require("../n8n-json-loader");
+async function processDocuments(documentInput, inputItems) {
+    let processedDocuments;
+    if (documentInput instanceof n8n_json_loader_1.N8nJsonLoader || documentInput instanceof n8n_binary_loader_1.N8nBinaryLoader) {
+        processedDocuments = await documentInput.processAll(inputItems);
+    }
+    else {
+        processedDocuments = documentInput;
+    }
+    const serializedDocuments = processedDocuments.map(({ metadata, pageContent }) => ({
+        json: { metadata, pageContent },
+    }));
+    return {
+        processedDocuments,
+        serializedDocuments,
+    };
+}
+async function processDocument(documentInput, inputItem, itemIndex) {
+    let processedDocuments;
+    if (documentInput instanceof n8n_json_loader_1.N8nJsonLoader || documentInput instanceof n8n_binary_loader_1.N8nBinaryLoader) {
+        processedDocuments = await documentInput.processItem(inputItem, itemIndex);
+    }
+    else {
+        processedDocuments = documentInput;
+    }
+    const serializedDocuments = processedDocuments.map(({ metadata, pageContent }) => ({
+        json: { metadata, pageContent },
+        pairedItem: {
+            item: itemIndex,
+        },
+    }));
+    return {
+        processedDocuments,
+        serializedDocuments,
+    };
+}
+//# sourceMappingURL=processDocuments.js.map

package/dist/esm/utils/vector-store/processDocuments.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"processDocuments.js","sourceRoot":"","sources":["../../../../src/utils/vector-store/processDocuments.ts"],"names":[],"mappings":";;AAMA,4CAoBC;AACD,0CAwBC;AAhDD,4DAAuD;AACvD,wDAAmD;AAE5C,KAAK,UAAU,gBAAgB,CACrC,aAAyF,EACzF,UAAgC;IAEhC,IAAI,kBAA8B,CAAC;IAEnC,IAAI,aAAa,YAAY,+BAAa,IAAI,aAAa,YAAY,mCAAe,EAAE,CAAC;QACxF,kBAAkB,GAAG,MAAM,aAAa,CAAC,UAAU,CAAC,UAAU,CAAC,CAAC;IACjE,CAAC;SAAM,CAAC;QACP,kBAAkB,GAAG,aAAa,CAAC;IACpC,CAAC;IAED,MAAM,mBAAmB,GAAG,kBAAkB,CAAC,GAAG,CAAC,CAAC,EAAE,QAAQ,EAAE,WAAW,EAAE,EAAE,EAAE,CAAC,CAAC;QAClF,IAAI,EAAE,EAAE,QAAQ,EAAE,WAAW,EAAE;KAC/B,CAAC,CAAC,CAAC;IAEJ,OAAO;QACN,kBAAkB;QAClB,mBAAmB;KACnB,CAAC;AACH,CAAC;AACM,KAAK,UAAU,eAAe,CACpC,aAAyF,EACzF,SAA6B,EAC7B,SAAiB;IAEjB,IAAI,kBAA8B,CAAC;IAEnC,IAAI,aAAa,YAAY,+BAAa,IAAI,aAAa,YAAY,mCAAe,EAAE,CAAC;QACxF,kBAAkB,GAAG,MAAM,aAAa,CAAC,WAAW,CAAC,SAAS,EAAE,SAAS,CAAC,CAAC;IAC5E,CAAC;SAAM,CAAC;QACP,kBAAkB,GAAG,aAAa,CAAC;IACpC,CAAC;IAED,MAAM,mBAAmB,GAAG,kBAAkB,CAAC,GAAG,CAAC,CAAC,EAAE,QAAQ,EAAE,WAAW,EAAE,EAAE,EAAE,CAAC,CAAC;QAClF,IAAI,EAAE,EAAE,QAAQ,EAAE,WAAW,EAAE;QAC/B,UAAU,EAAE;YACX,IAAI,EAAE,SAAS;SACf;KACD,CAAC,CAAC,CAAC;IAEJ,OAAO;QACN,kBAAkB;QAClB,mBAAmB;KACnB,CAAC;AACH,CAAC"}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@n8n/ai-utilities",
-  "version": "0.4.2",
+  "version": "0.5.1",
   "description": "Utilities for building AI nodes in n8n",
   "types": "dist/esm/index.d.ts",
   "module": "dist/esm/index.js",
@@ -15,15 +15,16 @@
   },
   "files": [
     "dist",
-    "LICENSE.md",
-    "LICENSE_EE.md"
+    "LICENSE_EE.md",
+    "LICENSE.md"
   ],
   "devDependencies": {
     "@types/json-schema": "^7.0.15",
     "jest-mock-extended": "^3.0.4",
     "@types/mime-types": "3.0.1",
     "tsx": "^4.19.3",
-    "axios": "1.13.5"
+    "axios": "1.13.5",
+    "n8n-workflow": "2.11.1"
   },
   "dependencies": {
     "zod": "3.25.67",
@@ -39,9 +40,11 @@
     "https-proxy-agent": "7.0.6",
     "proxy-from-env": "^1.1.0",
     "undici": "^6.21.0",
-    "@n8n/config": "2.9.0",
     "@n8n/typescript-config": "1.3.0",
-    "n8n-workflow": "2.10.2"
+    "@n8n/config": "2.10.1"
+  },
+  "peerDependencies": {
+    "n8n-workflow": "*"
   },
   "license": "SEE LICENSE IN LICENSE.md",
   "homepage": "https://n8n.io",