diff options
| author | Martin Bielik <martin.bielik@instea.sk> | 2024-12-22 14:55:15 +0100 |
|---|---|---|
| committer | Martin Bielik <martin.bielik@instea.sk> | 2024-12-22 14:55:15 +0100 |
| commit | fbc2bfb445c71985e9fc399d3fac2def2fc6854e (patch) | |
| tree | e93db9bf177d8790c608e3871d9f55d259510855 /py/context.py | |
| parent | 4f1c95386437c8fa6e189c136681950a91fb25c9 (diff) | |
| download | vim-ai-fbc2bfb445c71985e9fc399d3fac2def2fc6854e.tar.gz | |
added image generation
Diffstat (limited to '')
| -rw-r--r-- | py/context.py | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/py/context.py b/py/context.py index 581f8ad..343a050 100644 --- a/py/context.py +++ b/py/context.py @@ -79,7 +79,7 @@ def load_role_config(role): enhance_roles_with_custom_function(roles) - postfixes = ["", ".complete", ".edit", ".chat"] + postfixes = ["", ".complete", ".edit", ".chat", ".image"] if not any([f"{role}{postfix}" in roles for postfix in postfixes]): raise Exception(f"Role `{role}` not found") @@ -91,6 +91,7 @@ def load_role_config(role): 'role_complete': parse_role_section(roles.get(f"{role}.complete", {})), 'role_edit': parse_role_section(roles.get(f"{role}.edit", {})), 'role_chat': parse_role_section(roles.get(f"{role}.chat", {})), + 'role_image': parse_role_section(roles.get(f"{role}.image", {})), } def parse_role_names(prompt): @@ -147,7 +148,7 @@ def make_ai_context(params): user_prompt, role_config = parse_prompt_and_role_config(user_instruction, command_type) final_config = merge_deep([config_default, config_extension, role_config]) - selection_boundary = final_config['options']['selection_boundary'] + selection_boundary = final_config['options'].get('selection_boundary', '') config_prompt = final_config.get('prompt', '') prompt = make_prompt(config_prompt, user_prompt, user_selection, selection_boundary) |