zach
/
RA.Aid
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Improve interruption UX.

This commit is contained in:
user 2024-12-23 14:10:54 -05:00
parent 34fcb76ad8
commit 1d8b44e1a6
1 changed files with 8 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
ra_aid/tools/human.py
View File

@ -1,3 +1,4 @@
import sys
from langchain_core.tools import tool from langchain_core.tools import tool
from prompt_toolkit import PromptSession from prompt_toolkit import PromptSession
from prompt_toolkit.key_binding import KeyBindings from prompt_toolkit.key_binding import KeyBindings
@ -16,6 +17,13 @@ def create_keybindings():
"""Trigger submission when Ctrl+D is pressed.""" """Trigger submission when Ctrl+D is pressed."""
event.current_buffer.validate_and_handle() event.current_buffer.validate_and_handle()
@bindings.add('c-c')
def exit(event):
"""Trigger exit when Ctrl+C is pressed."""
print()
print(" 👋 Bye!")
sys.exit(0)
return bindings return bindings
@tool @tool