70 lines
2.1 KiB
TypeScript
70 lines
2.1 KiB
TypeScript
import { ICommonObject, INode, INodeData, INodeParams } from '../../../src/Interface'
|
|
|
|
class Iteration_Agentflow implements INode {
|
|
label: string
|
|
name: string
|
|
version: number
|
|
description: string
|
|
type: string
|
|
icon: string
|
|
category: string
|
|
color: string
|
|
baseClasses: string[]
|
|
documentation?: string
|
|
credential: INodeParams
|
|
inputs: INodeParams[]
|
|
|
|
constructor() {
|
|
this.label = 'Iteration'
|
|
this.name = 'iterationAgentflow'
|
|
this.version = 1.0
|
|
this.type = 'Iteration'
|
|
this.category = 'Agent Flows'
|
|
this.description = 'Execute the nodes within the iteration block through N iterations'
|
|
this.baseClasses = [this.type]
|
|
this.color = '#9C89B8'
|
|
this.inputs = [
|
|
{
|
|
label: 'Array Input',
|
|
name: 'iterationInput',
|
|
type: 'string',
|
|
description: 'The input array to iterate over',
|
|
acceptVariable: true,
|
|
rows: 4
|
|
}
|
|
]
|
|
}
|
|
|
|
async run(nodeData: INodeData, _: string, options: ICommonObject): Promise<any> {
|
|
const iterationInput = nodeData.inputs?.iterationInput
|
|
|
|
// Helper function to clean JSON strings with redundant backslashes
|
|
const cleanJsonString = (str: string): string => {
|
|
return str.replace(/\\(["'[\]{}])/g, '$1')
|
|
}
|
|
|
|
const iterationInputArray =
|
|
typeof iterationInput === 'string' && iterationInput !== '' ? JSON.parse(cleanJsonString(iterationInput)) : iterationInput
|
|
|
|
if (!iterationInputArray || !Array.isArray(iterationInputArray)) {
|
|
throw new Error('Invalid input array')
|
|
}
|
|
|
|
const state = options.agentflowRuntime?.state as ICommonObject
|
|
|
|
const returnOutput = {
|
|
id: nodeData.id,
|
|
name: this.name,
|
|
input: {
|
|
iterationInput: iterationInputArray
|
|
},
|
|
output: {},
|
|
state
|
|
}
|
|
|
|
return returnOutput
|
|
}
|
|
}
|
|
|
|
module.exports = { nodeClass: Iteration_Agentflow }
|