Function _run_enhanced_chat_loop

src/mcp_cli/chat/chat_handler.py:416–608

Run the main chat loop with enhanced streaming support. Args: ui: UI manager with streaming coordination ctx: Chat context convo: Conversation processor with streaming support max_turns: Maximum conversation turns before forcing exit (default: 100)

(
    ui: ChatUIManager,
    ctx: ChatContext,
    convo: ConversationProcessor,
    max_turns: int = 100,
)

async def _run_enhanced_chat_loop(
    ui: ChatUIManager,
    ctx: ChatContext,
    convo: ConversationProcessor,
    max_turns: int = 100,
) -> None:
    """
    Run the main chat loop with enhanced streaming support.

    Args:
        ui: UI manager with streaming coordination
        ctx: Chat context
        convo: Conversation processor with streaming support
        max_turns: Maximum conversation turns before forcing exit (default: 100)
    """
    # Shared queue: terminal reader task and browser WebSocket both put messages here.
    # This lets browser input arrive during the terminal prompt wait.
    # Type is Any because we also put _INTERRUPT sentinel objects on the queue.
    input_queue: asyncio.Queue = asyncio.Queue()

    # Wire dashboard bridge so browser USER_MESSAGE/USER_COMMAND go into the queue.
    if bridge := getattr(ctx, "dashboard_bridge", None):
        bridge.set_input_queue(input_queue)

    # Gate the prompt display: the reader waits for this event before showing
    # the "💬 You:" prompt, preventing streaming / tool output from overwriting it.
    prompt_ready = asyncio.Event()
    prompt_ready.set()  # Ready immediately for the first prompt

    # Background task: reads terminal input and forwards to the queue.
    reader_task = asyncio.create_task(
        _terminal_reader(ui, input_queue, ready=prompt_ready)
    )

    try:
        while True:
            try:
                user_msg = await input_queue.get()

                # Handle interrupt sentinel forwarded from _terminal_reader
                if user_msg is _INTERRUPT:
                    logger.info(
                        "Interrupt forwarded from reader — streaming=%s, tools_running=%s",
                        ui.is_streaming_response,
                        ui.tools_running,
                    )
                    if ui.is_streaming_response:
                        output.warning("\nStreaming interrupted - type 'exit' to quit.")
                        ui.interrupt_streaming()
                    elif ui.tools_running:
                        output.warning(
                            "\nTool execution interrupted - type 'exit' to quit."
                        )
                        ui._interrupt_now()
                    else:
                        output.warning("\nInterrupted - type 'exit' to quit.")
                    prompt_ready.set()
                    continue