From 49872fbbac7efb282f5768e532f8eed3e9e98a0f Mon Sep 17 00:00:00 2001 From: Alex Grozav Date: Fri, 21 Mar 2025 09:24:05 +0200 Subject: [PATCH] fix: Replace ai nodes enum strings with hard-coded values (no-changelog) (#14091) --- .../nodes/agents/Agent/Agent.node.ts | 18 ++++++++---------- .../nodes/chains/ChainLLM/methods/config.ts | 6 +++--- .../V2/ChainSummarizationV2.node.ts | 8 ++++---- 3 files changed, 15 insertions(+), 17 deletions(-) diff --git a/packages/@n8n/nodes-langchain/nodes/agents/Agent/Agent.node.ts b/packages/@n8n/nodes-langchain/nodes/agents/Agent/Agent.node.ts index fbb48d07d7..f531d3ebad 100644 --- a/packages/@n8n/nodes-langchain/nodes/agents/Agent/Agent.node.ts +++ b/packages/@n8n/nodes-langchain/nodes/agents/Agent/Agent.node.ts @@ -46,14 +46,14 @@ function getInputs( inputs: SpecialInput[], ): Array => { const displayNames: { [key: string]: string } = { - [NodeConnectionType.AiLanguageModel]: 'Model', - [NodeConnectionType.AiMemory]: 'Memory', - [NodeConnectionType.AiTool]: 'Tool', - [NodeConnectionType.AiOutputParser]: 'Output Parser', + ai_languageModel: 'Model', + ai_memory: 'Memory', + ai_tool: 'Tool', + ai_outputParser: 'Output Parser', }; return inputs.map(({ type, filter }) => { - const isModelType = type === NodeConnectionType.AiLanguageModel; + const isModelType = type === ('ai_languageModel' as NodeConnectionType); let displayName = type in displayNames ? displayNames[type] : undefined; if ( isModelType && @@ -65,11 +65,9 @@ function getInputs( type, displayName, required: isModelType, - maxConnections: [ - NodeConnectionType.AiLanguageModel, - NodeConnectionType.AiMemory, - NodeConnectionType.AiOutputParser, - ].includes(type as NodeConnectionType) + maxConnections: ['ai_languageModel', 'ai_memory', 'ai_outputParser'].includes( + type as NodeConnectionType, + ) ? 1 : undefined, }; diff --git a/packages/@n8n/nodes-langchain/nodes/chains/ChainLLM/methods/config.ts b/packages/@n8n/nodes-langchain/nodes/chains/ChainLLM/methods/config.ts index 973034714f..15686918be 100644 --- a/packages/@n8n/nodes-langchain/nodes/chains/ChainLLM/methods/config.ts +++ b/packages/@n8n/nodes-langchain/nodes/chains/ChainLLM/methods/config.ts @@ -14,11 +14,11 @@ import { getTemplateNoticeField } from '@utils/sharedFields'; */ export function getInputs(parameters: IDataObject) { const inputs = [ - { displayName: '', type: NodeConnectionType.Main }, + { displayName: '', type: 'main' }, { displayName: 'Model', maxConnections: 1, - type: NodeConnectionType.AiLanguageModel, + type: 'ai_languageModel', required: true, }, ]; @@ -29,7 +29,7 @@ export function getInputs(parameters: IDataObject) { if (hasOutputParser === undefined || hasOutputParser === true) { inputs.push({ displayName: 'Output Parser', - type: NodeConnectionType.AiOutputParser, + type: 'ai_outputParser', maxConnections: 1, required: false, }); diff --git a/packages/@n8n/nodes-langchain/nodes/chains/ChainSummarization/V2/ChainSummarizationV2.node.ts b/packages/@n8n/nodes-langchain/nodes/chains/ChainSummarization/V2/ChainSummarizationV2.node.ts index 4e17adb18d..fa074edc3b 100644 --- a/packages/@n8n/nodes-langchain/nodes/chains/ChainSummarization/V2/ChainSummarizationV2.node.ts +++ b/packages/@n8n/nodes-langchain/nodes/chains/ChainSummarization/V2/ChainSummarizationV2.node.ts @@ -25,11 +25,11 @@ function getInputs(parameters: IDataObject) { const chunkingMode = parameters?.chunkingMode; const operationMode = parameters?.operationMode; const inputs = [ - { displayName: '', type: NodeConnectionType.Main }, + { displayName: '', type: 'main' }, { displayName: 'Model', maxConnections: 1, - type: NodeConnectionType.AiLanguageModel, + type: 'ai_languageModel', required: true, }, ]; @@ -37,7 +37,7 @@ function getInputs(parameters: IDataObject) { if (operationMode === 'documentLoader') { inputs.push({ displayName: 'Document', - type: NodeConnectionType.AiDocument, + type: 'ai_document', required: true, maxConnections: 1, }); @@ -47,7 +47,7 @@ function getInputs(parameters: IDataObject) { if (chunkingMode === 'advanced') { inputs.push({ displayName: 'Text Splitter', - type: NodeConnectionType.AiTextSplitter, + type: 'ai_textSplitter', required: false, maxConnections: 1, });