summaryrefslogtreecommitdiff
path: root/py/complete.py
diff options
context:
space:
mode:
authorMartin Bielik <mx.bielik@gmail.com>2024-03-09 18:06:14 +0100
committerMartin Bielik <mx.bielik@gmail.com>2024-03-09 18:10:16 +0100
commitb0f7e3b8d92289d1a8086e3eed206f8b5757328b (patch)
treede9d18ee5e5471467fe293e3059f2f138457331c /py/complete.py
parenteb333e39e17d4d5240dbbc186a65bc6e9ab6b44e (diff)
downloadvim-ai-b0f7e3b8d92289d1a8086e3eed206f8b5757328b.tar.gz
read role prompt from config
Diffstat (limited to '')
-rw-r--r--py/complete.py8
1 files changed, 6 insertions, 2 deletions
diff --git a/py/complete.py b/py/complete.py
index debe275..453ef46 100644
--- a/py/complete.py
+++ b/py/complete.py
@@ -6,11 +6,15 @@ vim.command(f"py3file {plugin_root}/py/utils.py")
config = normalize_config(vim.eval("l:config"))
engine = config['engine']
-config_options = config['options']
+
+prompt, role_options = parse_prompt_and_role(vim.eval("l:prompt"))
+config_options = {
+ **config['options'],
+ **role_options,
+}
openai_options = make_openai_options(config_options)
http_options = make_http_options(config_options)
-prompt = vim.eval("l:prompt").strip()
is_selection = vim.eval("l:is_selection")
def complete_engine(prompt):