From e66c07d336dcaac9695862e8eb946b40e72ad044 Mon Sep 17 00:00:00 2001
From: Henry Heng <henryheng@flowiseai.com>
Date: Sat, 8 Jul 2023 22:38:56 +0100
Subject: [PATCH] Update utils isFlowValidForStream

Some chains/agents streaming are not working atm, disabling streaming for those chains/agents
---
 packages/server/src/utils/index.ts | 18 ++++++++++++------
 1 file changed, 12 insertions(+), 6 deletions(-)

diff --git a/packages/server/src/utils/index.ts b/packages/server/src/utils/index.ts
index 005f4a4b..d32faa8b 100644
--- a/packages/server/src/utils/index.ts
+++ b/packages/server/src/utils/index.ts
@@ -680,10 +680,16 @@ export const isFlowValidForStream = (reactFlowNodes: IReactFlowNode[], endingNod
         }
     }
 
-    return (
-        isChatOrLLMsExist &&
-        (endingNodeData.category === 'Chains' || endingNodeData.name === 'openAIFunctionAgent') &&
-        !isVectorStoreFaiss(endingNodeData) &&
-        process.env.EXECUTION_MODE !== 'child'
-    )
+    let isValidChainOrAgent = false
+    if (endingNodeData.category === 'Chains') {
+        // Chains that are not available to stream
+        const blacklistChains = ['openApiChain']
+        isValidChainOrAgent = !blacklistChains.includes(endingNodeData.name)
+    } else if (endingNodeData.category === 'Agents') {
+        // Agent that are available to stream
+        const whitelistAgents = ['openAIFunctionAgent']
+        isValidChainOrAgent = whitelistAgents.includes(endingNodeData.name)
+    }
+
+    return isChatOrLLMsExist && isValidChainOrAgent && !isVectorStoreFaiss(endingNodeData) && process.env.EXECUTION_MODE !== 'child'
 }