fix: cancel error (#1965)

2025-05-02 02:06:10 +08:00
parent 0dc8ee4771
commit 555b2e615c
3 changed files with 17 additions and 5 deletions
--- a/lua/avante/llm_tools/init.lua
+++ b/lua/avante/llm_tools/init.lua
@@ -1148,6 +1148,7 @@ function M.process_tool_use(tools, tool_use, on_log, on_complete, session_ctx)
              cancel_timer:stop()
              cancel_timer:close()
            end
+            Helpers.is_cancelled = false
            on_complete(nil, Helpers.CANCEL_TOKEN)
          end
        end)
@@ -1191,6 +1192,7 @@ function M.process_tool_use(tools, tool_use, on_log, on_complete, session_ctx)
  end, function(result, err)
    -- Check for cancellation before completing
    if Helpers.is_cancelled then
+      Helpers.is_cancelled = false
      if on_complete then on_complete(nil, Helpers.CANCEL_TOKEN) end
      return
    end