diff options
Diffstat (limited to 'autoload/vim_ai.vim')
| -rw-r--r-- | autoload/vim_ai.vim | 9 |
1 files changed, 9 insertions, 0 deletions
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('<sfile>: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 |