Function _run_turn

openkb/agent/chat.py:326–426

Run one agent turn with streaming output and persist the new history.

(
    agent: Any,
    session: ChatSession,
    user_input: str,
    style: Style,
    *,
    use_color: bool = True,
    raw: bool = False,
)

async def _run_turn(
    agent: Any,
    session: ChatSession,
    user_input: str,
    style: Style,
    *,
    use_color: bool = True,
    raw: bool = False,
) -> None:
    """Run one agent turn with streaming output and persist the new history."""
    from agents import (
        RawResponsesStreamEvent,
        RunItemStreamEvent,
        Runner,
    )
    from openai.types.responses import ResponseTextDeltaEvent

    new_input = session.history + [{"role": "user", "content": user_input}]

    result = Runner.run_streamed(agent, new_input, max_turns=MAX_TURNS)

    print()
    collected: list[str] = []
    segment: list[str] = []
    last_was_text = False
    need_blank_before_text = False

    if use_color and not raw:
        from rich.live import Live

        console = _make_rich_console()
    else:
        console = None  # type: ignore[assignment]

    def _start_live() -> Any:
        if console is None:
            return None
        lv = Live(console=console, vertical_overflow="visible")
        lv.start()
        return lv

    live = _start_live()

    try:
        async for event in result.stream_events():
            if isinstance(event, RawResponsesStreamEvent):
                if isinstance(event.data, ResponseTextDeltaEvent):
                    text = event.data.delta
                    if text:
                        if need_blank_before_text:
                            if console is not None:
                                print()
                                segment = []
                                live = _start_live()
                            else:
                                sys.stdout.write("\n")
                            need_blank_before_text = False
                        collected.append(text)