fix: uncompleted task (#2086)

2025-05-27 15:59:56 +08:00
parent 0e7d66333d
commit edb78e5646
8 changed files with 195 additions and 44 deletions
--- a/lua/avante/utils/init.lua
+++ b/lua/avante/utils/init.lua
@@ -1494,6 +1494,10 @@ function M.message_content_item_to_lines(item, message, messages)
      local lines = {}
      local state = "generating"
      local hl = "AvanteStateSpinnerToolCalling"
+      local ok, llm_tool = pcall(require, "avante.llm_tools." .. item.name)
+      if ok then
+        if llm_tool.on_render then return llm_tool.on_render(item.input, message.tool_use_logs) end
+      end
      local tool_result_message = M.get_tool_result_message(message, messages)
      if tool_result_message then
        local tool_result = tool_result_message.message.content[1]