Update utils isFlowValidForStream

Some chains/agents streaming are not working atm, disabling streaming for those chains/agents
This commit is contained in:
Henry Heng
2023-07-08 22:38:56 +01:00
committed by GitHub
parent 6a7e378baf
commit e66c07d336
+12 -6
View File
@@ -680,10 +680,16 @@ export const isFlowValidForStream = (reactFlowNodes: IReactFlowNode[], endingNod
} }
} }
return ( let isValidChainOrAgent = false
isChatOrLLMsExist && if (endingNodeData.category === 'Chains') {
(endingNodeData.category === 'Chains' || endingNodeData.name === 'openAIFunctionAgent') && // Chains that are not available to stream
!isVectorStoreFaiss(endingNodeData) && const blacklistChains = ['openApiChain']
process.env.EXECUTION_MODE !== 'child' 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'
} }