diff options
| author | Martin Bielik <mx.bielik@gmail.com> | 2023-03-13 20:57:57 +0100 |
|---|---|---|
| committer | Martin Bielik <mx.bielik@gmail.com> | 2023-03-13 20:57:57 +0100 |
| commit | 9545d08a879380d47901a7398c0c3aa6972933bf (patch) | |
| tree | d090127a7c674039e0e2f38f5b7ecdd77ca2dfd3 /plugin | |
| parent | 60f123341288df77a466528f2f1875f81c0cc450 (diff) | |
| download | vim-ai-9545d08a879380d47901a7398c0c3aa6972933bf.tar.gz | |
fixed instruction argument
Diffstat (limited to 'plugin')
| -rw-r--r-- | plugin/vim-ai.vim | 10 |
1 files changed, 5 insertions, 5 deletions
diff --git a/plugin/vim-ai.vim b/plugin/vim-ai.vim index 0f644b8..9388623 100644 --- a/plugin/vim-ai.vim +++ b/plugin/vim-ai.vim @@ -10,10 +10,10 @@ function! ScratchWindow() setlocal ft=aichat endfunction -function! MakePrompt(lines, args) +function! MakePrompt(lines, instruction) let lines = trim(join(a:lines, "\n")) let is_selection = lines != "" && lines == trim(@*) - let instruction = trim(get(a:args, 0)) + let instruction = trim(a:instruction) let delimiter = instruction != "" && is_selection ? ":\n" : "" let selection = is_selection ? lines : "" let prompt = join([instruction, delimiter, selection], "") @@ -21,7 +21,7 @@ function! MakePrompt(lines, args) endfunction function! AIRun(...) range - let prompt = MakePrompt(getline(a:firstline, a:lastline), a:000) + let prompt = MakePrompt(getline(a:firstline, a:lastline), a:0 ? a:1 : "") set paste execute "normal! " . a:lastline . "Go" execute "py3file " . s:complete_py @@ -30,7 +30,7 @@ function! AIRun(...) range endfunction function! AIEditRun(...) range - let prompt = MakePrompt(getline(a:firstline, a:lastline), a:000) + let prompt = MakePrompt(getline(a:firstline, a:lastline), a:0 ? a:1 : "") set paste execute "normal! " . a:firstline . "GV" . a:lastline . "Gc" execute "py3file " . s:complete_py @@ -43,7 +43,7 @@ function! AIChatRun(...) range let is_outside_of_chat_window = search('^>>> user$', 'nw') == 0 if is_outside_of_chat_window call ScratchWindow() - let prompt = MakePrompt(lines, a:000) + let prompt = MakePrompt(lines, a:0 ? a:1 : "") execute "normal i>>> user\n\n" . prompt endif |