From cd81fbf8afb35f5ee880a775f311bfacd2bc70eb Mon Sep 17 00:00:00 2001
From: yetone <yetoneful@gmail.com>
Date: Thu, 17 Jul 2025 14:28:35 +0800
Subject: [PATCH] fix: tool_use_id is nil

---
 lua/avante/llm.lua            | 2 ++
 lua/avante/llm_tools/init.lua | 2 ++
 2 files changed, 4 insertions(+)

diff --git a/lua/avante/llm.lua b/lua/avante/llm.lua
index a9dd1ac..17b2f69 100644
--- a/lua/avante/llm.lua
+++ b/lua/avante/llm.lua
@@ -154,6 +154,7 @@ function M.generate_todos(user_input, cb)
               local result = LLMTools.process_tool_use(tools, pending_tool, {
                 session_ctx = {},
                 on_complete = function() cb() end,
+                tool_use_id = pending_tool.id,
               })
               if result ~= nil then cb() end
             end
@@ -850,6 +851,7 @@ function M._stream(opts)
           on_log = opts.on_tool_log,
           set_tool_use_store = opts.set_tool_use_store,
           on_complete = handle_tool_result,
+          tool_use_id = partial_tool_use.id,
         })
         if result ~= nil or error ~= nil then return handle_tool_result(result, error) end
       end
diff --git a/lua/avante/llm_tools/init.lua b/lua/avante/llm_tools/init.lua
index 8d5ffcd..b8eca6b 100644
--- a/lua/avante/llm_tools/init.lua
+++ b/lua/avante/llm_tools/init.lua
@@ -1324,6 +1324,8 @@ function M.process_tool_use(tools, tool_use, opts)
       end
       on_complete(result, err)
     end,
+    streaming = opts.streaming,
+    tool_use_id = opts.tool_use_id,
   })
 
   -- Result and error being nil means that the tool was executed asynchronously