From 9db190c977d1f76e2f574882e4edde062eecbb0d Mon Sep 17 00:00:00 2001 From: Martin Bielik Date: Sat, 9 Mar 2024 19:08:34 +0100 Subject: roles completion --- autoload/vim_ai.vim | 9 +++++++++ 1 file changed, 9 insertions(+) (limited to 'autoload') diff --git a/autoload/vim_ai.vim b/autoload/vim_ai.vim index 74515d4..302ef1d 100644 --- a/autoload/vim_ai.vim +++ b/autoload/vim_ai.vim @@ -1,8 +1,10 @@ call vim_ai_config#load() + let s:plugin_root = expand(':p:h:h') let s:complete_py = s:plugin_root . "/py/complete.py" let s:chat_py = s:plugin_root . "/py/chat.py" +let s:roles_py = s:plugin_root . "/py/roles.py" " remembers last command parameters to be used in AIRedoRun let s:last_is_selection = 0 @@ -249,3 +251,10 @@ function! vim_ai#AIRedoRun() call vim_ai#AIChatRun(0, s:last_config) endif endfunction + +function! vim_ai#RoleCompletion(A,L,P) abort + execute "py3file " . s:roles_py + call map(l:role_list, '"/" . v:val') + echom a:A + return filter(l:role_list, 'v:val =~ "^' . a:A . '"') +endfunction -- cgit v1.2.3 From f4130feb986760a8d956983cb5cfe8d7106e7d4a Mon Sep 17 00:00:00 2001 From: Martin Bielik Date: Sat, 9 Mar 2024 19:37:55 +0100 Subject: roles example file --- autoload/vim_ai.vim | 1 - autoload/vim_ai_config.vim | 5 +++++ 2 files changed, 5 insertions(+), 1 deletion(-) (limited to 'autoload') diff --git a/autoload/vim_ai.vim b/autoload/vim_ai.vim index 302ef1d..b6f4617 100644 --- a/autoload/vim_ai.vim +++ b/autoload/vim_ai.vim @@ -1,6 +1,5 @@ call vim_ai_config#load() - let s:plugin_root = expand(':p:h:h') let s:complete_py = s:plugin_root . "/py/complete.py" let s:chat_py = s:plugin_root . "/py/chat.py" diff --git a/autoload/vim_ai_config.vim b/autoload/vim_ai_config.vim index ef3c863..41932de 100644 --- a/autoload/vim_ai_config.vim +++ b/autoload/vim_ai_config.vim @@ -1,3 +1,5 @@ +let s:plugin_root = expand(':p:h:h') + let g:vim_ai_complete_default = { \ "engine": "complete", \ "options": { @@ -73,6 +75,9 @@ endif if !exists("g:vim_ai_token_file_path") let g:vim_ai_token_file_path = "~/.config/openai.token" endif +if !exists("g:vim_ai_roles_config_file") + let g:vim_ai_roles_config_file = s:plugin_root . "/roles-example.ini" +endif function! vim_ai_config#ExtendDeep(defaults, override) abort let l:result = a:defaults -- cgit v1.2.3 From 5709ab1ce75f05e9b57bf052f3ad92aa74f61c16 Mon Sep 17 00:00:00 2001 From: Martin Bielik Date: Sat, 9 Mar 2024 22:44:27 +0100 Subject: removed unused log --- autoload/vim_ai.vim | 1 - 1 file changed, 1 deletion(-) (limited to 'autoload') diff --git a/autoload/vim_ai.vim b/autoload/vim_ai.vim index b6f4617..47a9f03 100644 --- a/autoload/vim_ai.vim +++ b/autoload/vim_ai.vim @@ -254,6 +254,5 @@ endfunction function! vim_ai#RoleCompletion(A,L,P) abort execute "py3file " . s:roles_py call map(l:role_list, '"/" . v:val') - echom a:A return filter(l:role_list, 'v:val =~ "^' . a:A . '"') endfunction -- cgit v1.2.3