summaryrefslogtreecommitdiff
path: root/plugin/vim-ai.vim
diff options
context:
space:
mode:
authorMartin Bielik <mx.bielik@gmail.com>2023-04-04 22:18:40 +0200
committerMartin Bielik <mx.bielik@gmail.com>2023-04-04 22:18:40 +0200
commitc66a57be70e7c3ba48e74ca6e666d74cf00258d8 (patch)
treebe0366700a0dd4efb4e4d9b73bed6f4d1ee5f362 /plugin/vim-ai.vim
parente7cdbfb658e903452232f15c4f355bdefcbeacd9 (diff)
downloadvim-ai-c66a57be70e7c3ba48e74ca6e666d74cf00258d8.tar.gz
chat engine
Diffstat (limited to '')
-rw-r--r--plugin/vim-ai.vim7
1 files changed, 6 insertions, 1 deletions
diff --git a/plugin/vim-ai.vim b/plugin/vim-ai.vim
index b117598..c2b5faf 100644
--- a/plugin/vim-ai.vim
+++ b/plugin/vim-ai.vim
@@ -1,4 +1,5 @@
let g:vim_ai_complete_default = {
+\ "engine": "complete",
\ "options": {
\ "model": "text-davinci-003",
\ "max_tokens": 1000,
@@ -7,6 +8,7 @@ let g:vim_ai_complete_default = {
\ },
\}
let g:vim_ai_edit_default = {
+\ "engine": "complete",
\ "options": {
\ "model": "text-davinci-003",
\ "max_tokens": 1000,
@@ -26,9 +28,9 @@ Audience: Users of text editor and programmers that need to transform/generate t
<<< assistant
Okay
-
END
let g:vim_ai_chat_default = {
+\ "engine": "chat",
\ "options": {
\ "model": "gpt-3.5-turbo",
\ "max_tokens": 1000,
@@ -90,6 +92,7 @@ function! AIRun(is_selection, ...) range
let s:last_instruction = instruction
let s:last_is_selection = a:is_selection
+ let engine = g:vim_ai_complete['engine']
let options = g:vim_ai_complete['options']
let cursor_on_empty_line = trim(join(lines, "\n")) == ""
set paste
@@ -111,6 +114,7 @@ function! AIEditRun(is_selection, ...) range
let s:last_instruction = instruction
let s:last_is_selection = a:is_selection
+ let engine = g:vim_ai_edit['engine']
let options = g:vim_ai_edit['options']
set paste
execute "normal! " . a:firstline . "GV" . a:lastline . "Gc"
@@ -138,6 +142,7 @@ function! AIChatRun(is_selection, ...) range
let s:last_instruction = instruction
let s:last_is_selection = a:is_selection
+ let engine = g:vim_ai_chat['engine']
let options = g:vim_ai_chat['options']
execute "py3file " . s:chat_py
set nopaste