blob: 94bcdb466c9141b2bc4a32e87520d35991879712 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
|
import vim
import os
if "PYTEST_VERSION" in os.environ:
from utils import *
roles_py_imported = True
def load_ai_role_names(command_type):
roles = read_role_files()
enhance_roles_with_custom_function(roles)
role_names = set()
for name in roles.sections():
parts = name.split('.')
if len(parts) == 1 or parts[-1] == command_type:
role_names.add(parts[0])
return list(role_names)
|
