fixed complete command roles after refactoring

madox2 · Dec 12, 2024 · 3cdf03b · 3cdf03b
1 parent c9bc248
commit 3cdf03b
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 4 deletions.
diff --git a/py/chat.py b/py/chat.py
@@ -4,7 +4,7 @@
 plugin_root = vim.eval("s:plugin_root")
 vim.command(f"py3file {plugin_root}/py/utils.py")
 
-prompt, config = load_config_and_prompt()
+prompt, config = load_config_and_prompt('chat')
 config_options = config['options']
 config_ui = config['ui']
 

diff --git a/py/complete.py b/py/complete.py
@@ -4,7 +4,7 @@
 plugin_root = vim.eval("s:plugin_root")
 vim.command(f"py3file {plugin_root}/py/utils.py")
 
-prompt, config = load_config_and_prompt()
+prompt, config = load_config_and_prompt('complete')
 config_options = config['options']
 config_ui = config['ui']
 

diff --git a/py/utils.py b/py/utils.py
@@ -44,13 +44,13 @@ def load_api_key(config_token_file_path):
 
     return (api_key, org_id)
 
-def load_config_and_prompt():
+def load_config_and_prompt(command_type):
     prompt, role_options = parse_prompt_and_role(vim.eval("l:prompt"))
     config = vim.eval("l:config")
     config['options'] = {
         **normalize_options(config['options']),
         **normalize_options(role_options['options_default']),
-        **normalize_options(role_options['options_chat']),
+        **normalize_options(role_options['options_' + command_type]),
     }
     return prompt, config