summaryrefslogtreecommitdiff
path: root/py/utils.py
diff options
context:
space:
mode:
Diffstat (limited to 'py/utils.py')
-rw-r--r--py/utils.py9
1 files changed, 9 insertions, 0 deletions
diff --git a/py/utils.py b/py/utils.py
index bd69401..2e0fc2d 100644
--- a/py/utils.py
+++ b/py/utils.py
@@ -125,3 +125,12 @@ def openai_request(url, data, options):
else:
openai_obj = json.loads(line_data)
yield openai_obj
+
+def handle_completion_error(error):
+ if isinstance(error, KeyboardInterrupt):
+ vim.command("normal! a Ctrl-C...")
+ if isinstance(error, URLError):
+ if isinstance(error.reason, socket.timeout):
+ vim.command("normal! aRequest timeout...")
+ else:
+ raise error