diff --git a/webview-ui/src/components/chat/ChatView.tsx b/webview-ui/src/components/chat/ChatView.tsx index 9b8c96dea1..b3935f7841 100644 --- a/webview-ui/src/components/chat/ChatView.tsx +++ b/webview-ui/src/components/chat/ChatView.tsx @@ -1942,25 +1942,46 @@ const ChatViewComponent: React.ForwardRefRenderFunction )} - {(secondaryButtonText || isStreaming) && ( + {(secondaryButtonText || isStreaming) && !isStreaming && ( handleSecondaryButtonClick(inputValue, selectedImages)}> - {isStreaming ? t("chat:cancel.title") : secondaryButtonText} + {secondaryButtonText} + + + )} + {isStreaming && ( + + handleSecondaryButtonClick(inputValue, selectedImages)}> + {t("chat:cancel.title")} + + + )} + {/* Add stop/close button when not streaming but showing approval buttons */} + {!isStreaming && enableButtons && primaryButtonText && ( + + startNewTask()}> + )}