diff --git a/mentat/terminal/prompt_session.py b/mentat/terminal/prompt_session.py index cd6171503..ae6bf14d2 100644 --- a/mentat/terminal/prompt_session.py +++ b/mentat/terminal/prompt_session.py @@ -60,12 +60,16 @@ def prompt_continuation( def _setup_bindings(self): self.bindings = KeyBindings() - @self.bindings.add("s-down") - @self.bindings.add("c-j") + @Condition + def not_searching() -> bool: + return not get_app().layout.is_searching + + @self.bindings.add("s-down", filter=not_searching) + @self.bindings.add("c-j", filter=not_searching) def _(event: KeyPressEvent): event.current_buffer.insert_text("\n") - @self.bindings.add("enter") + @self.bindings.add("enter", filter=not_searching) def _(event: KeyPressEvent): event.current_buffer.validate_and_handle() @@ -86,8 +90,8 @@ def _(event: KeyPressEvent): if suggestion: event.current_buffer.insert_text(suggestion.text) - @self.bindings.add("c-c") - @self.bindings.add("c-d") + @self.bindings.add("c-c", filter=not_searching) + @self.bindings.add("c-d", filter=not_searching) def _(event: KeyPressEvent): if event.current_buffer.text != "": event.current_buffer.reset()