From 924e3a390f043e979f16113f6b0a55f8c54b1f5e Mon Sep 17 00:00:00 2001
From: Martin Bielik <mx.bielik@gmail.com>
Date: Sun, 25 Jun 2023 11:24:23 +0200
Subject: allow string in initial_prompt, closes #35

---
 py/utils.py | 8 ++++++++
 1 file changed, 8 insertions(+)

(limited to 'py/utils.py')

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 {
-- 
cgit v1.2.3