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

fix: escape original_content (#5)

Browse Source
This commit is contained in:
Hanchin Hsieh 2024-08-15 13:41:54 +08:00 committed by GitHub
parent 0b6a85ee64
commit 1634abb9cd
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 5 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
lua/avante/init.lua
View File

@ -353,6 +353,7 @@ local function call_openai_api_stream(question, code_lang, code_content, on_chun
end end
local function call_ai_api_stream(question, code_lang, code_content, on_chunk, on_complete) local function call_ai_api_stream(question, code_lang, code_content, on_chunk, on_complete)
code_content = utils.escape(code_content)
if M.config.provider == "openai" or M.config.provider == "azure" then if M.config.provider == "openai" or M.config.provider == "azure" then
call_openai_api_stream(question, code_lang, code_content, on_chunk, on_complete) call_openai_api_stream(question, code_lang, code_content, on_chunk, on_complete)
elseif M.config.provider == "claude" then elseif M.config.provider == "claude" then

4
lua/avante/utils.lua
View File

@ -4,4 +4,8 @@ function M.trim_suffix(str, suffix)
return string.gsub(str, suffix .. "$", "") return string.gsub(str, suffix .. "$", "")
end end
function M.escape(str)
return string.gsub(str, "([%(%)%.%%%+%-%*%?%[%^%$%]])", "%%%1")
end
return M return M