blob: bb5356eeb5ae607c107ad3f1a6bd02f594b3678a (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
|
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])
role_names = [name for name in role_names if name != DEFAULT_ROLE_NAME]
return list(role_names)
|
