Function _make_prompt_session

openkb/agent/chat.py:270–311

(
    session: ChatSession, style: Style, use_color: bool, kb_dir: Path
)

def _make_prompt_session(
    session: ChatSession, style: Style, use_color: bool, kb_dir: Path
) -> PromptSession:
    from prompt_toolkit.filters import has_completions
    from prompt_toolkit.history import FileHistory
    from prompt_toolkit.key_binding import KeyBindings

    kb = KeyBindings()

    @kb.add("tab", filter=has_completions)
    def _accept_completion(event: Any) -> None:
        """Tab accepts the current completion (like zsh), not cycle."""
        buf = event.current_buffer
        state = buf.complete_state
        if not state:
            return
        # Only one candidate or already selected — accept immediately
        if state.current_completion:
            buf.apply_completion(state.current_completion)
        elif len(state.completions) == 1:
            buf.apply_completion(state.completions[0])
        else:
            # Multiple candidates, nothing selected — highlight first
            buf.go_to_completion(0)

    @kb.add("tab", filter=~has_completions)
    def _trigger_completion(event: Any) -> None:
        """Tab triggers completion when menu is not open."""
        buf = event.current_buffer
        buf.start_completion()

    history_path = kb_dir / ".openkb" / "chat_history"
    return PromptSession(
        message=FormattedText([("class:prompt", ">>> ")]),
        style=style,
        completer=_ChatCompleter(),
        complete_style=CompleteStyle.MULTI_COLUMN,
        complete_while_typing=False,
        key_bindings=kb,
        history=FileHistory(str(history_path)),
        bottom_toolbar=(lambda: _bottom_toolbar(session)) if use_color else None,
    )


def _make_rich_console() -> Any: