diff options
Diffstat (limited to '')
| -rw-r--r-- | plugin/vim-ai.vim | 20 |
1 files changed, 16 insertions, 4 deletions
diff --git a/plugin/vim-ai.vim b/plugin/vim-ai.vim index eeb913a..7570f69 100644 --- a/plugin/vim-ai.vim +++ b/plugin/vim-ai.vim @@ -1,4 +1,4 @@ -let g:vim_ai_complete = { +let g:vim_ai_complete_default = { \ "options": { \ "model": "text-davinci-003", \ "max_tokens": 1000, @@ -6,7 +6,7 @@ let g:vim_ai_complete = { \ "request_timeout": 10, \ }, \} -let g:vim_ai_edit = { +let g:vim_ai_edit_default = { \ "options": { \ "model": "text-davinci-003", \ "max_tokens": 1000, @@ -14,7 +14,7 @@ let g:vim_ai_edit = { \ "request_timeout": 10, \ }, \} -let g:vim_ai_chat = { +let g:vim_ai_chat_default = { \ "options": { \ "model": "gpt-3.5-turbo", \ "max_tokens": 1000, @@ -22,6 +22,16 @@ let g:vim_ai_chat = { \ "request_timeout": 10, \ }, \} +if !exists('g:vim_ai_complete') + let g:vim_ai_complete = {"options":{}} +endif +if !exists('g:vim_ai_edit') + let g:vim_ai_edit = {"options":{}} +endif +if !exists('g:vim_ai_chat') + let g:vim_ai_chat = {"options":{}} +endif + let s:plugin_root = expand('<sfile>:p:h:h') let s:complete_py = s:plugin_root . "/py/complete.py" @@ -46,6 +56,7 @@ endfunction function! AIRun(is_selection, ...) range let prompt = MakePrompt(a:is_selection, getline(a:firstline, a:lastline), a:0 ? a:1 : "") + let options_default = g:vim_ai_complete_default['options'] let options = g:vim_ai_complete['options'] set paste execute "normal! " . a:lastline . "Go" @@ -56,7 +67,7 @@ endfunction function! AIEditRun(is_selection, ...) range let prompt = MakePrompt(a:is_selection, getline(a:firstline, a:lastline), a:0 ? a:1 : "") - echo prompt + let options_default = g:vim_ai_edit_default['options'] let options = g:vim_ai_edit['options'] set paste execute "normal! " . a:firstline . "GV" . a:lastline . "Gc" @@ -77,6 +88,7 @@ function! AIChatRun(is_selection, ...) range execute "normal i>>> user\n\n" . prompt endif + let options_default = g:vim_ai_chat_default['options'] let options = g:vim_ai_chat['options'] execute "py3file " . s:chat_py set nopaste |