diff options
| author | Martin Bielik <mx.bielik@gmail.com> | 2023-03-21 18:36:38 +0100 |
|---|---|---|
| committer | Martin Bielik <mx.bielik@gmail.com> | 2023-03-21 18:36:38 +0100 |
| commit | 8125570fc4ca94b085857eadc563eafc2c0564a5 (patch) | |
| tree | c67323c90c13c3c06e1364cf662e4ed467b514b3 /plugin/vim-ai.vim | |
| parent | e025530d25cedd1dcbd9d2102cd06296f3707da9 (diff) | |
| download | vim-ai-8125570fc4ca94b085857eadc563eafc2c0564a5.tar.gz | |
openai configuration
Diffstat (limited to 'plugin/vim-ai.vim')
| -rw-r--r-- | plugin/vim-ai.vim | 28 |
1 files changed, 28 insertions, 0 deletions
diff --git a/plugin/vim-ai.vim b/plugin/vim-ai.vim index 9388623..b24c3ed 100644 --- a/plugin/vim-ai.vim +++ b/plugin/vim-ai.vim @@ -1,3 +1,28 @@ +let g:vim_ai_complete = { +\ "options": { +\ "model": "text-davinci-003", +\ "max_tokens": 1000, +\ "temperature": 0.1, +\ "request_timeout": 10, +\ }, +\} +let g:vim_ai_edit = { +\ "options": { +\ "model": "text-davinci-003", +\ "max_tokens": 1000, +\ "temperature": 0.1, +\ "request_timeout": 10, +\ }, +\} +let g:vim_ai_chat = { +\ "options": { +\ "model": "gpt-3.5-turbo", +\ "max_tokens": 1000, +\ "temperature": 1, +\ "request_timeout": 10, +\ }, +\} + let s:plugin_root = expand('<sfile>:p:h:h') let s:complete_py = s:plugin_root . "/py/complete.py" let s:chat_py = s:plugin_root . "/py/chat.py" @@ -22,6 +47,7 @@ endfunction function! AIRun(...) range let prompt = MakePrompt(getline(a:firstline, a:lastline), a:0 ? a:1 : "") + let options = g:vim_ai_complete['options'] set paste execute "normal! " . a:lastline . "Go" execute "py3file " . s:complete_py @@ -31,6 +57,7 @@ endfunction function! AIEditRun(...) range let prompt = MakePrompt(getline(a:firstline, a:lastline), a:0 ? a:1 : "") + let options = g:vim_ai_edit['options'] set paste execute "normal! " . a:firstline . "GV" . a:lastline . "Gc" execute "py3file " . s:complete_py @@ -47,6 +74,7 @@ function! AIChatRun(...) range execute "normal i>>> user\n\n" . prompt endif + let options = g:vim_ai_chat['options'] execute "py3file " . s:chat_py set nopaste endfunction |