diff options
| author | Martin Bielik <mx.bielik@gmail.com> | 2023-04-04 22:41:14 +0200 |
|---|---|---|
| committer | Martin Bielik <mx.bielik@gmail.com> | 2023-04-04 22:41:14 +0200 |
| commit | c97fbfdccb65334243f91bb574b7fb00740726dc (patch) | |
| tree | 9b022438a962c1fdde281b5958f9b68ae2e4b470 /py/complete.py | |
| parent | c66a57be70e7c3ba48e74ca6e666d74cf00258d8 (diff) | |
| download | vim-ai-c97fbfdccb65334243f91bb574b7fb00740726dc.tar.gz | |
combine initial prompt with empty chat prompt
Diffstat (limited to 'py/complete.py')
| -rw-r--r-- | py/complete.py | 3 |
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) |