fix: Replace ai nodes enum strings with hard-coded values (no-changelog) (#14091)

This commit is contained in:
Alex Grozav
2025-03-21 09:24:05 +02:00
committed by GitHub
parent cb5a803f9e
commit 49872fbbac
3 changed files with 15 additions and 17 deletions

View File

@@ -46,14 +46,14 @@ function getInputs(
inputs: SpecialInput[],
): Array<NodeConnectionType | INodeInputConfiguration> => {
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,
};