From d6030f8e9c5abc1b41ce1ab8591b84487d912961 Mon Sep 17 00:00:00 2001 From: Henry Date: Fri, 26 Jan 2024 19:24:48 +0000 Subject: [PATCH] fix custom function ending node --- packages/server/src/index.ts | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/packages/server/src/index.ts b/packages/server/src/index.ts index 76e48e58..b8ebd5f7 100644 --- a/packages/server/src/index.ts +++ b/packages/server/src/index.ts @@ -473,7 +473,7 @@ export class App { } } - isStreaming = isFlowValidForStream(nodes, endingNodeData) + isStreaming = isEndingNode ? false : isFlowValidForStream(nodes, endingNodeData) } const obj = { isStreaming } @@ -1676,7 +1676,11 @@ export class App { return res.status(500).send(`Ending node must be either a Chain or Agent`) } - if (!Object.values(endingNodeData.outputs ?? {}).includes(endingNodeData.name)) { + if ( + endingNodeData.outputs && + Object.keys(endingNodeData.outputs).length && + !Object.values(endingNodeData.outputs ?? {}).includes(endingNodeData.name) + ) { return res .status(500) .send(