summaryrefslogtreecommitdiff
path: root/py/complete.py
diff options
context:
space:
mode:
Diffstat (limited to 'py/complete.py')
-rw-r--r--py/complete.py3
1 files changed, 2 insertions, 1 deletions
diff --git a/py/complete.py b/py/complete.py
index 96ae032..2f151d3 100644
--- a/py/complete.py
+++ b/py/complete.py
@@ -18,7 +18,8 @@ def complete_engine():
return text_chunks
def chat_engine():
- initial_prompt = options.get('initial_prompt', '')
+ initial_prompt = options.get('initial_prompt', [])
+ initial_prompt = '\n'.join(initial_prompt)
chat_content = f"{initial_prompt}\n\n>>> user\n\n{prompt}".strip()
messages = parse_chat_messages(chat_content)
response = openai.ChatCompletion.create(messages=messages, stream=True, **request_options)