stream complete/edit commands

1 files changed, 18 insertions, 14 deletions

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")