From 2d644e05545be6cd699eb5e834d6ba4468f12b41 Mon Sep 17 00:00:00 2001
From: Martin Bielik <mx.bielik@gmail.com>
Date: Tue, 11 Apr 2023 17:30:16 +0200
Subject: added debug logging

---
 py/chat.py | 13 +++++++++++--
 1 file changed, 11 insertions(+), 2 deletions(-)

(limited to 'py/chat.py')

diff --git a/py/chat.py b/py/chat.py
index 0e3245f..6e94665 100644
--- a/py/chat.py
+++ b/py/chat.py
@@ -74,8 +74,17 @@ try:
         print('Answering...')
         vim.command("redraw")
 
-        response = openai.ChatCompletion.create(messages=messages, stream=True, **request_options)
-        text_chunks = map(lambda resp: resp['choices'][0]['delta'].get('content', ''), response)
+        request = {
+            'stream': True,
+            'messages': messages,
+            **request_options
+        }
+        printDebug("[chat] request: {}", request)
+        response = openai.ChatCompletion.create(**request)
+        def map_chunk(resp):
+            printDebug("[chat] response: {}", resp)
+            return resp['choices'][0]['delta'].get('content', '')
+        text_chunks = map(map_chunk, response)
         render_text_chunks(text_chunks)
 
         vim.command("normal! a\n\n>>> user\n\n")
-- 
cgit v1.2.3