diff options
| author | Martin Bielik <mx.bielik@gmail.com> | 2023-04-16 13:20:08 +0200 |
|---|---|---|
| committer | Martin Bielik <mx.bielik@gmail.com> | 2023-04-16 13:20:08 +0200 |
| commit | f2b83895297444a9c4220b776f3243951d446aba (patch) | |
| tree | ec1df4af144543203feffe3ad10014b6390e82f4 /py/utils.py | |
| parent | c7e88f343518ea46e30c1a156af242a4b3b1fc2c (diff) | |
| download | vim-ai-f2b83895297444a9c4220b776f3243951d446aba.tar.gz | |
nvim keyboard interrupt handling
Diffstat (limited to 'py/utils.py')
| -rw-r--r-- | py/utils.py | 11 |
1 files changed, 5 insertions, 6 deletions
diff --git a/py/utils.py b/py/utils.py index 4243c4c..2da0302 100644 --- a/py/utils.py +++ b/py/utils.py @@ -134,12 +134,11 @@ def print_info_message(msg): vim.command("echohl None") def handle_completion_error(error): - if isinstance(error, KeyboardInterrupt): + # nvim throws - pynvim.api.common.NvimError: Keyboard interrupt + is_nvim_keyboard_interrupt = "Keyboard interrupt" in str(error) + if isinstance(error, KeyboardInterrupt) or is_nvim_keyboard_interrupt: print_info_message("Completion cancelled...") - elif isinstance(error, URLError): - if isinstance(error.reason, socket.timeout): - print_info_message("Request timeout...") - else: - raise error + elif isinstance(error, URLError) and isinstance(error.reason, socket.timeout): + print_info_message("Request timeout...") else: raise error |