fix: escape original_content (#5)

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

View File

@ -353,6 +353,7 @@ local function call_openai_api_stream(question, code_lang, code_content, on_chun
end
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
call_openai_api_stream(question, code_lang, code_content, on_chunk, on_complete)
elseif M.config.provider == "claude" then

View File

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