Function _handle_slash

openkb/agent/chat.py:742–831

Return ``"exit"`` to end the REPL, ``"new_session"`` to swap sessions, or ``None`` to continue with the current session.

(
    cmd: str,
    kb_dir: Path,
    session: ChatSession,
    style: Style,
)

async def _handle_slash(
    cmd: str,
    kb_dir: Path,
    session: ChatSession,
    style: Style,
) -> str | None:
    """Return ``"exit"`` to end the REPL, ``"new_session"`` to swap sessions,
    or ``None`` to continue with the current session."""
    parts = cmd.split(maxsplit=1)
    head = parts[0].lower()
    arg = parts[1].strip() if len(parts) > 1 else ""
    # Strip surrounding quotes (user may type /add '/path/to file')
    if len(arg) >= 2 and arg[0] == arg[-1] and arg[0] in ("'", '"'):
        arg = arg[1:-1]
    elif arg and arg[0] in ("'", '"'):
        arg = arg[1:]

    if head in ("/exit", "/quit"):
        _fmt(style, ("class:header", "Bye. Thanks for using OpenKB.\n\n"))
        return "exit"

    if head == "/help":
        _fmt(style, ("class:slash.help", _HELP_TEXT + "\n"))
        return None

    if head == "/clear":
        old_id = session.id
        _fmt(
            style,
            ("class:slash.ok", f"Started new session (previous: {old_id})\n"),
        )
        return "new_session"

    if head == "/save":
        if not session.user_turns:
            _fmt(style, ("class:error", "Nothing to save yet.\n"))
            return None
        from openkb.locks import kb_ingest_lock

        with kb_ingest_lock(kb_dir / ".openkb"):
            path = _save_transcript(kb_dir, session, arg or None)
        _fmt(style, ("class:slash.ok", f"Saved to {path}\n"))
        return None

    if head == "/status":
        from openkb.cli import print_status
        from openkb.locks import kb_read_lock

        with kb_read_lock(kb_dir / ".openkb"):
            print_status(kb_dir)
        return None

    if head == "/list":
        from openkb.cli import print_list
        from openkb.locks import kb_read_lock

        with kb_read_lock(kb_dir / ".openkb"):
            print_list(kb_dir)