From 004f26eb94a7d629223cb972634bbd36746f7c35 Mon Sep 17 00:00:00 2001
From: gongzhiyong <gongzhiyong@gongzhiyongdeMacBook-Air.local>
Date: Sun, 12 Apr 2026 16:45:36 +0800
Subject: [PATCH] fix: Stop button uses cancelling poll flow, shows status text
 during stop

---
 langgraph/src/main.tsx | 20 ++++++++++++++++++--
 1 file changed, 18 insertions(+), 2 deletions(-)

diff --git a/langgraph/src/main.tsx b/langgraph/src/main.tsx
index 1e0c391..8b45343 100644
--- a/langgraph/src/main.tsx
+++ b/langgraph/src/main.tsx
@@ -1112,12 +1112,28 @@ function App() {
               {thread.isLoading ? (
                 <button
                   type="button"
-                  onClick={() => thread.stop()}
+                  onClick={() => {
+                    setConcurrentStatus("stopping");
+                    wasStoppedRef.current = true;
+                    thread.stop();
+                    setTimeout(() => {
+                      setConcurrentStatus("cancelling");
+                      const waitForStop = () => {
+                        if (!thread.isLoading) {
+                          wasStoppedRef.current = false;
+                          setConcurrentStatus("idle");
+                        } else {
+                          setTimeout(waitForStop, 200);
+                        }
+                      };
+                      waitForStop();
+                    }, 150);
+                  }}
                   className="rounded-xl bg-destructive text-destructive-foreground px-4 py-2.5 text-sm font-medium hover:opacity-90 transition-opacity flex items-center gap-1.5"
                   title="点击停止生成"
                 >
                   <Square className="size-4 fill-current" />
-                  停止
+                  {concurrentStatus === "stopping" ? "停止中..." : concurrentStatus === "cancelling" ? "等待停止..." : "停止"}
                 </button>
               ) : (
                 <button