diff options
| -rw-r--r-- | plugin/vim-ai.vim | 12 |
1 files changed, 10 insertions, 2 deletions
diff --git a/plugin/vim-ai.vim b/plugin/vim-ai.vim index c9f48e1..501dc2c 100644 --- a/plugin/vim-ai.vim +++ b/plugin/vim-ai.vim @@ -126,10 +126,18 @@ endfunction function! AIRedoRun() execute "normal! u" if s:last_command == "complete" - '<,'>call AIRun(s:last_is_selection, s:last_instruction) + if s:last_is_selection + '<,'>call AIRun(s:last_is_selection, s:last_instruction) + else + call AIRun(s:last_is_selection, s:last_instruction) + endif endif if s:last_command == "edit" - '<,'>call AIEditRun(s:last_is_selection, s:last_instruction) + if s:last_is_selection + '<,'>call AIEditRun(s:last_is_selection, s:last_instruction) + else + call AIEditRun(s:last_is_selection, s:last_instruction) + endif endif if s:last_command == "chat" call AIChatRun(s:last_is_selection, s:last_instruction) |