From 86e8563125d88af5887409d6cab90f927f762c4b Mon Sep 17 00:00:00 2001 From: Dmitry Torokhov Date: Thu, 17 Jul 2025 22:32:13 -0700 Subject: [PATCH] refactor(llm_tools): use History.Helpers.get_tool_result_data() Avoid poking into internals of avante.HistoryMessage, and use appropriate helper instead. --- lua/avante/llm_tools/dispatch_agent.lua | 19 +++++++++++-------- 1 file changed, 11 insertions(+), 8 deletions(-) diff --git a/lua/avante/llm_tools/dispatch_agent.lua b/lua/avante/llm_tools/dispatch_agent.lua index ae99ffe..06c7a98 100644 --- a/lua/avante/llm_tools/dispatch_agent.lua +++ b/lua/avante/llm_tools/dispatch_agent.lua @@ -148,14 +148,17 @@ function M.on_render(input, opts) local icon = Utils.icon("🔄 ") local hl = Highlights.AVANTE_TASK_RUNNING if result_message then - if result_message.message.content[1].is_error then - state = "failed" - icon = Utils.icon("❌ ") - hl = Highlights.AVANTE_TASK_FAILED - else - state = "completed" - icon = Utils.icon("✅ ") - hl = Highlights.AVANTE_TASK_COMPLETED + local result = History.Helpers.get_tool_result_data(result_message) + if result then + if result.is_error then + state = "failed" + icon = Utils.icon("❌ ") + hl = Highlights.AVANTE_TASK_FAILED + else + state = "completed" + icon = Utils.icon("✅ ") + hl = Highlights.AVANTE_TASK_COMPLETED + end end end local lines = {}