From 60f123341288df77a466528f2f1875f81c0cc450 Mon Sep 17 00:00:00 2001
From: Martin Bielik <mx.bielik@gmail.com>
Date: Mon, 13 Mar 2023 20:52:45 +0100
Subject: stream complete/edit commands

---
 py/chat.py | 32 ++++++++++++++++++--------------
 1 file changed, 18 insertions(+), 14 deletions(-)

(limited to 'py/chat.py')

diff --git a/py/chat.py b/py/chat.py
index b6db8db..066227b 100644
--- a/py/chat.py
+++ b/py/chat.py
@@ -29,20 +29,24 @@ if not messages:
     file_content = ">>> user\n\n" + file_content
     messages.append({"role": "user", "content": file_content })
 
-vim.command("normal! Go\n<<< assistant\n\n")
-vim.command("redraw")
-
-response = openai.ChatCompletion.create(
-  model="gpt-3.5-turbo",
-  messages=messages,
-  stream=True,
-)
-
-for resp in response:
-    if 'content' in resp['choices'][0]['delta']:
-        text = resp['choices'][0]['delta']['content']
+if messages[-1]["content"].strip():
+
+    vim.command("normal! Go\n<<< assistant\n\n")
+    vim.command("redraw")
+
+    response = openai.ChatCompletion.create(
+      model="gpt-3.5-turbo",
+      messages=messages,
+      stream=True,
+    )
+
+    generating_text = False
+    for resp in response:
+        text = resp['choices'][0]['delta'].get('content', '')
+        if not text.strip() and not generating_text:
+            continue # trim newlines from the beginning
         vim.command("normal! a" + text)
         vim.command("redraw")
 
-vim.command("normal! a\n\n>>> user\n")
-vim.command("redraw")
+    vim.command("normal! a\n\n>>> user\n\n")
+    vim.command("redraw")
-- 
cgit v1.2.3