diff --git a/web/app/components/workflow/panel/debug-and-preview/chat-wrapper.tsx b/web/app/components/workflow/panel/debug-and-preview/chat-wrapper.tsx index 101125c290..7e4a7a5f9e 100644 --- a/web/app/components/workflow/panel/debug-and-preview/chat-wrapper.tsx +++ b/web/app/components/workflow/panel/debug-and-preview/chat-wrapper.tsx @@ -2,6 +2,7 @@ import { forwardRef, memo, useCallback, + useEffect, useImperativeHandle, useMemo, } from 'react' @@ -30,9 +31,15 @@ type ChatWrapperProps = { showConversationVariableModal: boolean onConversationModalHide: () => void showInputsFieldsPanel: boolean + onHide: () => void } -const ChatWrapper = forwardRef(({ showConversationVariableModal, onConversationModalHide, showInputsFieldsPanel }, ref) => { +const ChatWrapper = forwardRef(({ + showConversationVariableModal, + onConversationModalHide, + showInputsFieldsPanel, + onHide, +}, ref) => { const nodes = useNodes() const startNode = nodes.find(node => node.data.type === BlockEnum.Start) const startVariables = startNode?.data.variables @@ -92,6 +99,11 @@ const ChatWrapper = forwardRef(({ showConv } }, [handleRestart]) + useEffect(() => { + if (isResponding) + onHide() + }, [isResponding, onHide]) + return ( <> { showConversationVariableModal={showConversationVariableModal} onConversationModalHide={() => setShowConversationVariableModal(false)} showInputsFieldsPanel={expanded} + onHide={() => setExpanded(false)} />