Skip to content

feat: Preserve object return type in Agentflow Custom Function #5895

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
beliga44 wants to merge 2 commits into
base: main
Choose a base branch
from
+8 −8
Open

feat: Preserve object return type in Agentflow Custom Function #5895

Changes from all commits
Commits
Show all changes
2 commits
Select commit Hold shift + click to select a range
a5219f8
feat(agentflow): allow CustomFunction node to return JSON objects dir…
beliga44 Mar 4, 2026
4dba42f
update(CustomFunction): handle null values for finalOutput
beliga44 Mar 4, 2026
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
16 changes: 8 additions & 8 deletions packages/components/nodes/agentflow/CustomFunction/CustomFunction.ts
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -21,7 +21,7 @@ const exampleFunc = `/*
* You can use properties specified in Input Variables with the prefix $. For example: $foo
* You can get default flow config: $flow.sessionId, $flow.chatId, $flow.chatflowId, $flow.input, $flow.state
* You can get global variables: $vars.<variable-name>
* Must return a string value at the end of function
* You can return a string value or a JSON object at the end of function
*/

const fetch = require('node-fetch');
Expand All @@ -32,6 +32,7 @@ const options = {
'Content-Type': 'application/json'
}
};

try {
const response = await fetch(url, options);
const text = await response.text();
Expand Down Expand Up @@ -60,7 +61,7 @@ class CustomFunction_Agentflow implements INode {
constructor() {
this.label = 'Custom Function'
this.name = 'customFunctionAgentflow'
this.version = 1.1
this.version = 1.2
this.type = 'CustomFunction'
this.category = 'Agent Flows'
this.description = 'Execute custom function'
Expand Down Expand Up @@ -107,7 +108,8 @@ class CustomFunction_Agentflow implements INode {
label: 'Key',
name: 'key',
type: 'asyncOptions',
loadMethod: 'listRuntimeStateKeys'
loadMethod: 'listRuntimeStateKeys',
freeSolo: true
},
{
label: 'Value',
Expand Down Expand Up @@ -183,18 +185,16 @@ class CustomFunction_Agentflow implements INode {
streamOutput
})

let finalOutput = response
if (typeof response === 'object') {
finalOutput = JSON.stringify(response, null, 2)
}
const finalOutput = response

// Update flow state if needed
let newState = { ...state }
if (_customFunctionUpdateState && Array.isArray(_customFunctionUpdateState) && _customFunctionUpdateState.length > 0) {
newState = updateFlowState(state, _customFunctionUpdateState)
}

newState = processTemplateVariables(newState, finalOutput)
const outputForState = typeof finalOutput === 'object' && finalOutput !== null ? JSON.stringify(finalOutput, null, 2) : finalOutput
newState = processTemplateVariables(newState, outputForState)

const returnOutput = {
id: nodeData.id,
Expand Down