allow string in initial_prompt, closes #35

4 files changed, 11 insertions, 3 deletions

diff --git a/README.md b/README.md
index 8ab679d..1ff7973 100644
--- a/README.md
+++ b/README.md
@@ -252,7 +252,7 @@ END
 
 " :AIChat
 " - options: openai config (see https://platform.openai.com/docs/api-reference/chat)
-" - options.initial_prompt: prompt prepended to every chat request
+" - options.initial_prompt: prompt prepended to every chat request (list of lines or string)
 " - options.request_timeout: request timeout in seconds
 " - options.selection_boundary: seleciton prompt wrapper (eliminates empty responses, see #20)
 " - ui.populate_options: put [chat-options] to the chat header
diff --git a/py/chat.py b/py/chat.py
index e2ef6b1..93d31bf 100644
--- a/py/chat.py
+++ b/py/chat.py
@@ -2,7 +2,7 @@
 plugin_root = vim.eval("s:plugin_root")
 vim.command(f"py3file {plugin_root}/py/utils.py")
 
-config = vim.eval("l:config")
+config = normalize_config(vim.eval("l:config"))
 config_options = config['options']
 config_ui = config['ui']
 prompt = vim.eval("l:prompt").strip()
diff --git a/py/complete.py b/py/complete.py
index fed66c4..c8d45fe 100644
--- a/py/complete.py
+++ b/py/complete.py
@@ -2,7 +2,7 @@
 plugin_root = vim.eval("s:plugin_root")
 vim.command(f"py3file {plugin_root}/py/utils.py")
 
-config = vim.eval("l:config")
+config = normalize_config(vim.eval("l:config"))
 engine = config['engine']
 config_options = config['options']
 openai_options = make_openai_options(config_options)
diff --git a/py/utils.py b/py/utils.py
index 8539347..76ae1e4 100644
--- a/py/utils.py
+++ b/py/utils.py
@@ -26,6 +26,14 @@ def load_api_key():
         raise Exception("Missing OpenAI API key")
     return api_key.strip()
 
+def normalize_config(config):
+    normalized = { **config }
+    # initial prompt can be both a string and a list of strings, normalize it to list
+    if 'initial_prompt' in config['options'] and isinstance(config['options']['initial_prompt'], str):
+        normalized['options']['initial_prompt'] = normalized['options']['initial_prompt'].split('\n')
+    return normalized
+
+
 def make_openai_options(options):
     max_tokens = int(options['max_tokens'])
     return {