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

fix: compatible with the tool use format preferred by gemini (#2160)

Browse Source
This commit is contained in:
yetone
2025-06-05 03:12:07 +08:00
committed by GitHub
parent 54d2790fe0
commit b7a5184219
1 changed files with 20 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

21
lua/avante/providers/openai.lua
View File

@@ -243,8 +243,27 @@ function M:add_text_message(ctx, text, state, opts)
}) })
ctx.content_uuid = msg.uuid ctx.content_uuid = msg.uuid
local msgs = { msg } local msgs = { msg }
local xml_content = ctx.content
local xml_lines = vim.split(xml_content, "\n")
local cleaned_xml_lines = {}
local prev_tool_name = nil
for _, line in ipairs(xml_lines) do
if line:match("<tool_name>") then
local tool_name = line:match("<tool_name>(.*)</tool_name>")
if tool_name then prev_tool_name = tool_name end
elseif line:match("<parameters>") then
if prev_tool_name then table.insert(cleaned_xml_lines, "<" .. prev_tool_name .. ">") end
goto continue
elseif line:match("</parameters>") then
if prev_tool_name then table.insert(cleaned_xml_lines, "</" .. prev_tool_name .. ">") end
goto continue
end
table.insert(cleaned_xml_lines, line)
::continue::
end
local cleaned_xml_content = table.concat(cleaned_xml_lines, "\n")
local stream_parser = XMLParser.createStreamParser() local stream_parser = XMLParser.createStreamParser()
stream_parser:addData(ctx.content) stream_parser:addData(cleaned_xml_content)
local has_tool_use = false local has_tool_use = false
local xml = stream_parser:getAllElements() local xml = stream_parser:getAllElements()
if xml then if xml then