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

fix: try to fix suggestions (#1209)

Browse Source
This commit is contained in:
yetone 2025-02-07 16:26:35 +08:00 committed by GitHub
parent e8b0f33f18
commit faa6728d4b
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 6 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
lua/avante/suggestion.lua
View File

@ -154,10 +154,13 @@ L5: pass
local cursor_row, cursor_col = Utils.get_cursor_pos() local cursor_row, cursor_col = Utils.get_cursor_pos()
if cursor_row ~= doc.position.row or cursor_col ~= doc.position.col then return end if cursor_row ~= doc.position.row or cursor_col ~= doc.position.col then return end
-- Clean up markdown code blocks -- Clean up markdown code blocks
full_response = Utils.trim_think_content(full_response)
full_response = full_response:gsub("^```%w*\n(.-)\n```$", "%1") full_response = full_response:gsub("^```%w*\n(.-)\n```$", "%1")
full_response = full_response:gsub("(.-)\n```\n?$", "%1") full_response = full_response:gsub("(.-)\n```\n?$", "%1")
-- Remove everything before the first '[' to ensure we get just the JSON array -- Remove everything before the first '[' to ensure we get just the JSON array
full_response = full_response:gsub("^.-(%[.*)", "%1") full_response = full_response:gsub("^.-(%[.*)", "%1")
-- Remove everything after the last ']' to ensure we get just the JSON array
full_response = full_response:gsub("(.*%]).-$", "%1")
local ok, suggestions_list = pcall(vim.json.decode, full_response) local ok, suggestions_list = pcall(vim.json.decode, full_response)
if not ok then if not ok then
Utils.error("Error while decoding suggestions: " .. full_response, { once = true, title = "Avante" }) Utils.error("Error while decoding suggestions: " .. full_response, { once = true, title = "Avante" })
@ -167,6 +170,9 @@ L5: pass
Utils.info("No suggestions found", { once = true, title = "Avante" }) Utils.info("No suggestions found", { once = true, title = "Avante" })
return return
end end
if #suggestions_list ~= 0 and not vim.islist(suggestions_list[1]) then
suggestions_list = { suggestions_list }
end
local current_lines = Utils.get_buf_lines(0, -1, bufnr) local current_lines = Utils.get_buf_lines(0, -1, bufnr)
suggestions_list = vim suggestions_list = vim
.iter(suggestions_list) .iter(suggestions_list)