From 8402371e6af33dcf6962ffbc8cd73be0e36f6812 Mon Sep 17 00:00:00 2001 From: Martin Bielik Date: Tue, 17 Dec 2024 18:43:51 +0100 Subject: introduced pre-defined default roles --- py/utils.py | 11 +++++++++++ 1 file changed, 11 insertions(+) (limited to 'py/utils.py') diff --git a/py/utils.py b/py/utils.py index ce95ed6..89bef3e 100644 --- a/py/utils.py +++ b/py/utils.py @@ -317,3 +317,14 @@ def make_chat_text_chunks(messages, config_options): map_chunk = map_chunk_stream if openai_options['stream'] else map_chunk_no_stream return map(map_chunk, response) + +def read_role_files(): + plugin_root = vim.eval("s:plugin_root") + default_roles_config_path = str(os.path.join(plugin_root, "roles-default.ini")) + roles_config_path = os.path.expanduser(vim.eval("g:vim_ai_roles_config_file")) + if not os.path.exists(roles_config_path): + raise Exception(f"Role config file does not exist: {roles_config_path}") + + roles = configparser.ConfigParser() + roles.read([default_roles_config_path, roles_config_path]) + return roles -- cgit v1.2.3