fixed stream=0 in chat engine

1 files changed, 1 insertions, 21 deletions

diff --git a/py/chat.py b/py/chat.py
index 4850bae..e91ea81 100644
--- a/py/chat.py
+++ b/py/chat.py
@@ -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")