summaryrefslogtreecommitdiff
path: root/plugin/vim-ai.vim
diff options
context:
space:
mode:
Diffstat (limited to 'plugin/vim-ai.vim')
-rw-r--r--plugin/vim-ai.vim6
1 files changed, 4 insertions, 2 deletions
diff --git a/plugin/vim-ai.vim b/plugin/vim-ai.vim
index 6046b04..eeb913a 100644
--- a/plugin/vim-ai.vim
+++ b/plugin/vim-ai.vim
@@ -70,8 +70,10 @@ function! AIChatRun(is_selection, ...) range
let is_outside_of_chat_window = search('^>>> user$', 'nw') == 0
if is_outside_of_chat_window
call ScratchWindow()
- " use prompt only in visual mode
- let prompt = a:is_selection ? MakePrompt(a:is_selection, lines, a:0 ? a:1 : "") : ""
+ let prompt = ""
+ if a:0 || a:is_selection
+ let prompt = MakePrompt(a:is_selection, lines, a:0 ? a:1 : "")
+ endif
execute "normal i>>> user\n\n" . prompt
endif