From 9db190c977d1f76e2f574882e4edde062eecbb0d Mon Sep 17 00:00:00 2001
From: Martin Bielik <mx.bielik@gmail.com>
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/vim_ai.vim')

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
-- 
cgit v1.2.3