diff --git a/lua/avante/utils/init.lua b/lua/avante/utils/init.lua
index f209440..000e750 100644
--- a/lua/avante/utils/init.lua
+++ b/lua/avante/utils/init.lua
@@ -576,7 +576,7 @@ end
 --- remove indentation from code: spaces or tabs
 function M.remove_indentation(code)
   if not code then return code end
-  return code:gsub("^%s*", "")
+  return code:gsub("^%s*", ""):gsub("%s*$", "")
 end
 
 function M.relative_path(absolute)