diff options
Diffstat (limited to '')
| -rw-r--r-- | py/chat.py | 22 |
1 files changed, 1 insertions, 21 deletions
@@ -50,8 +50,6 @@ initialize_chat_window() chat_options = parse_chat_header_options() options = {**config_options, **chat_options} -openai_options = make_openai_options(options) -http_options = make_http_options(options) initial_prompt = '\n'.join(options.get('initial_prompt', [])) initial_messages = parse_chat_messages(initial_prompt) @@ -70,25 +68,7 @@ try: print('Answering...') vim.command("redraw") - request = { - 'messages': messages, - **openai_options - } - printDebug("[chat] request: {}", request) - url = options['endpoint_url'] - response = openai_request(url, request, http_options) - - def map_chunk_no_stream(resp): - printDebug("[chat] response: {}", resp) - return resp['choices'][0]['message'].get('content', '') - - def map_chunk_stream(resp): - printDebug("[chat] response: {}", resp) - return resp['choices'][0]['delta'].get('content', '') - - map_chunk = map_chunk_stream if openai_options['stream'] else map_chunk_no_stream - - text_chunks = map(map_chunk, response) + text_chunks = make_chat_text_chunks(messages, options) render_text_chunks(text_chunks, is_selection) vim.command("normal! a\n\n>>> user\n\n") |