fix: do not error out when pressing ctrl+c

Realiserad · Jun 16, 2024 · 07f0222 · 07f0222
1 parent fd13102
commit 07f0222
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/src/fish_ai/switch_context.py b/src/fish_ai/switch_context.py
@@ -18,6 +18,8 @@ def switch_context():
     terminal_menu = TerminalMenu(options)
     terminal_menu.title = 'Select context'
     index = terminal_menu.show()
+    if index is None:
+        return
     active_section = options[index].split(' ')[0]
     config.set(section='fish-ai', option='configuration', value=active_section)
     config.write(open(path.expanduser('~/.config/fish-ai.ini'), 'w'))