From 8fde389664ca59773c38dc0ec1a434a98bc2428b Mon Sep 17 00:00:00 2001 From: Martin Bielik Date: Mon, 16 Dec 2024 23:46:17 +0100 Subject: command-type only roles --- autoload/vim_ai.vim | 16 ++++++++++++++-- 1 file changed, 14 insertions(+), 2 deletions(-) (limited to 'autoload/vim_ai.vim') diff --git a/autoload/vim_ai.vim b/autoload/vim_ai.vim index 03ac978..f6cbed5 100644 --- a/autoload/vim_ai.vim +++ b/autoload/vim_ai.vim @@ -299,9 +299,21 @@ function! vim_ai#AIRedoRun() abort endif endfunction -function! vim_ai#RoleCompletion(A,L,P) abort +function! s:RoleCompletion(A, command_type) abort call s:ImportPythonModules() - let l:role_list = py3eval("load_ai_role_names()") + let l:role_list = py3eval("load_ai_role_names(unwrap('a:command_type'))") call map(l:role_list, '"/" . v:val') return filter(l:role_list, 'v:val =~ "^' . a:A . '"') endfunction + +function! vim_ai#RoleCompletionComplete(A,L,P) abort + return s:RoleCompletion(a:A, 'complete') +endfunction + +function! vim_ai#RoleCompletionEdit(A,L,P) abort + return s:RoleCompletion(a:A, 'edit') +endfunction + +function! vim_ai#RoleCompletionChat(A,L,P) abort + return s:RoleCompletion(a:A, 'chat') +endfunction -- cgit v1.2.3