skip when unicode decode error (incomplete byte)

CopilotC-Nvim · Feb 5, 2024 · 043f6f7 · 043f6f7
1 parent f8e9121
commit 043f6f7
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/rplugin/python3/handlers/chat_handler.py b/rplugin/python3/handlers/chat_handler.py
@@ -217,6 +217,10 @@ def _add_chat_messages(
             system_prompt, prompt, code, language=cast(str, file_type), model=model
         ):
             buffer += token.encode("utf-8")
+            try:
+                buffer.decode("utf-8")
+            except UnicodeDecodeError:
+                continue
             if "\n" in buffer.decode("utf-8"):
                 token = buffer.decode("utf-8").split("\n")[0]
                 buffer = buffer[len(token) + 1:]