Просмотр исходного кода

fix(tui): abort in-progress generation on undo (#4744)

Tommy D. Rossi 2 месяцев назад
Родитель
Сommit
64caeeb12d
1 измененных файлов с 3 добавлено и 1 удалено
  1. 3 1
      packages/opencode/src/cli/cmd/tui/routes/session/index.tsx

+ 3 - 1
packages/opencode/src/cli/cmd/tui/routes/session/index.tsx

@@ -319,7 +319,9 @@ export function Session() {
       value: "session.undo",
       keybind: "messages_undo",
       category: "Session",
-      onSelect: (dialog) => {
+      onSelect: async (dialog) => {
+        const status = sync.data.session_status[route.sessionID]
+        if (status?.type !== "idle") await sdk.client.session.abort({ path: { id: route.sessionID } }).catch(() => {})
         const revert = session().revert?.messageID
         const message = messages().findLast((x) => (!revert || x.id < revert) && x.role === "user")
         if (!message) return