From 48dbd394160d8005d16f15f27e270483fe0e0916 Mon Sep 17 00:00:00 2001
From: dolphin <shanghang@dataelem.com>
Date: Wed, 17 Sep 2025 20:47:58 +0800
Subject: [PATCH] =?UTF-8?q?feat:=20=E4=BF=AE=E6=94=B9=E9=87=8D=E6=96=B0?=
 =?UTF-8?q?=E8=BF=90=E8=A1=8C=E6=8C=89=E9=92=AE=E5=B1=95=E7=A4=BA=E9=80=BB?=
 =?UTF-8?q?=E8=BE=91?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/frontend/client/src/@types/chat.ts               |  2 ++
 src/frontend/client/src/pages/appChat/ChatInput.tsx  |  5 +++--
 src/frontend/client/src/pages/appChat/index.tsx      |  3 ++-
 .../client/src/pages/appChat/useChatHelpers.ts       | 12 ++++++++++++
 .../client/src/pages/appChat/useWebsocket.ts         |  3 +++
 5 files changed, 22 insertions(+), 3 deletions(-)

diff --git a/src/frontend/client/src/@types/chat.ts b/src/frontend/client/src/@types/chat.ts
index 357df191c..c64f1c3b4 100644
--- a/src/frontend/client/src/@types/chat.ts
+++ b/src/frontend/client/src/@types/chat.ts
@@ -91,6 +91,8 @@ export interface RunningStatus {
   showStop: boolean;
   /** 引导词 */
   guideWord?: string[];
+  /** 重新运行展示 */
+  showReRun?: boolean;
 }
 
 // 毕昇配置
diff --git a/src/frontend/client/src/pages/appChat/ChatInput.tsx b/src/frontend/client/src/pages/appChat/ChatInput.tsx
index dc04f7514..875425940 100644
--- a/src/frontend/client/src/pages/appChat/ChatInput.tsx
+++ b/src/frontend/client/src/pages/appChat/ChatInput.tsx
@@ -7,7 +7,7 @@ import { useAreaText } from "./useAreaText";
 
 export default function ChatInput({ v }) {
     const [bishengConfig] = useRecoilState(bishengConfState)
-    const { inputDisabled, error: inputMsg, showUpload, showStop } = useRecoilValue(currentRunningState)
+    const { inputDisabled, error: inputMsg, showUpload, showStop, showReRun } = useRecoilValue(currentRunningState)
     const { accepts, chatState, inputRef, setChatFiles, handleInput, handleRestart, handleSendClick, handleStopClick } = useAreaText()
     const [fileUploading, setFileUploading] = useState(false)
 
@@ -57,7 +57,8 @@ export default function ChatInput({ v }) {
                 is工作流 & 未展示停止按钮 & 没有错误消息
             */}
             <div className="absolute w-full flex justify-center left-0 -top-14">
-                {!showStop && chatState?.flow?.flow_type === 10 && !inputMsg && <Button
+                {/* {!showStop && chatState?.flow?.flow_type === 10 && !inputMsg  */}
+                {!showStop && showReRun && !inputMsg && <Button
                     className="rounded-full bg-primary/10 bg-blue-50 text-primary"
                     variant="ghost"
                     onClick={handleRestart}>
diff --git a/src/frontend/client/src/pages/appChat/index.tsx b/src/frontend/client/src/pages/appChat/index.tsx
index 7586c62e4..977cf68e8 100644
--- a/src/frontend/client/src/pages/appChat/index.tsx
+++ b/src/frontend/client/src/pages/appChat/index.tsx
@@ -128,7 +128,8 @@ export default function index() {
                     inputForm: numericType !== FLOW_TYPES.WORK_FLOW || null,
                     showUpload: numericType === FLOW_TYPES.WORK_FLOW,
                     showStop: false,
-                    guideWord: flowData?.guide_question
+                    guideWord: flowData?.guide_question,
+                    showReRun: false
                 }
             }
         })
diff --git a/src/frontend/client/src/pages/appChat/useChatHelpers.ts b/src/frontend/client/src/pages/appChat/useChatHelpers.ts
index dba33b5b9..67dfffdd8 100644
--- a/src/frontend/client/src/pages/appChat/useChatHelpers.ts
+++ b/src/frontend/client/src/pages/appChat/useChatHelpers.ts
@@ -78,6 +78,17 @@ export default function useChatHelpers() {
         }))
     }
 
+    // 显示重试按钮
+    const reRunShow = (show: boolean) => {
+        setRunningState((prev) => ({
+            ...prev,
+            [chatId]: {
+                ...prev[chatId],
+                showReRun: show,
+            },
+        }))
+    }
+
     // 唤起输入(表单输入，文本输入)
     const showInputForm = (inputSchema) => {
         const { tab, value } = inputSchema
@@ -365,6 +376,7 @@ export default function useChatHelpers() {
         message,
         flow: chatState?.flow,
         stopShow,
+        reRunShow,
         handleMsgError,
         clearError,
         showInputForm,
diff --git a/src/frontend/client/src/pages/appChat/useWebsocket.ts b/src/frontend/client/src/pages/appChat/useWebsocket.ts
index d421f4268..749af291b 100644
--- a/src/frontend/client/src/pages/appChat/useWebsocket.ts
+++ b/src/frontend/client/src/pages/appChat/useWebsocket.ts
@@ -112,6 +112,7 @@ export const useWebSocket = (helpers) => {
             // 工作流input会有再begin之前出现的情况
             // helpers.stopShow(true)
         } else if (data.type === 'close' && data.category === 'processing') {
+            helpers.flow.flow_type === 10 && helpers.reRunShow(true)
             helpers.stopShow(false)
         }
 
@@ -135,11 +136,13 @@ export const useWebSocket = (helpers) => {
             sessionInfoMap.set(helpers.chatId, { node_id, message_id: data.message_id })
             // 待用户输入
             helpers.showInputForm({ ...input_schema, node_id })
+            helpers.flow.flow_type === 10 && helpers.reRunShow(true)
             return
         } else if (data.category === 'guide_question') {
             return helpers.showGuideQuestion(helpers.chatId, data.message.guide_question.filter(q => q))
         } else if (data.category === 'stream_msg') {
             helpers.message.streamMsg(helpers.chatId, data)
+            helpers.flow.flow_type === 10 && helpers.reRunShow(true)
         } else if (data.category === 'end_cover' && data.type === 'end_cover') {
             // helpers.handleMsgError('')
             _ws.send(JSON.stringify({ action: 'stop' }))