CarGDev/avante.nvim
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix: filter out uncomplete tool_use (#2088)

Browse Source
This commit is contained in:
yetone
2025-05-27 16:09:37 +08:00
committed by GitHub
parent edb78e5646
commit dbddc0408f
1 changed files with 8 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
lua/avante/llm.lua
View File

@@ -400,15 +400,18 @@ function M.generate_prompts(opts)
local final_history_messages = {}
if cleaned_history_messages then
if opts.disable_compact_history_messages then
vim.iter(cleaned_history_messages):each(function(msg)
if Utils.is_tool_use_message(msg) and not Utils.get_tool_result_message(msg, cleaned_history_messages) then
return
for i, msg in ipairs(cleaned_history_messages) do
if Utils.is_tool_use_message(msg) then
local next_msg = cleaned_history_messages[i + 1]
if not next_msg or not Utils.is_tool_result_message(next_msg) then goto continue end
if next_msg.message.content[1].tool_use_id ~= msg.message.content[1].id then goto continue end
end
if Utils.is_tool_result_message(msg) and not Utils.get_tool_use_message(msg, cleaned_history_messages) then
return
goto continue
end
table.insert(final_history_messages, msg)
end)
::continue::
end
else
if Config.history.max_tokens > 0 then
remaining_tokens = math.min(Config.history.max_tokens, remaining_tokens)