bub 0.3.7__tar.gz → 0.3.9__tar.gz

{bub-0.3.7 → bub-0.3.9}/.gitignore

@@ -142,6 +142,8 @@ cython_debug/
 
 # Reference directory - ignore all reference projects
 reference/
+!website/src/content/docs/**/build
+!website/src/content/docs/**/reference
 
 # Local legacy backups created during framework migrations
 backup/

{bub-0.3.7 → bub-0.3.9}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: bub
-Version: 0.3.7
+Version: 0.3.9
 Summary: A common shape for agents that live alongside people.
 Project-URL: Homepage, https://bub.build
 Project-URL: Repository, https://github.com/bubbuild/bub
@@ -106,24 +106,28 @@ Key source files:
 
 ```python
 from bub import hookimpl
+from bub.envelope import content_of
 
 
 class EchoPlugin:
     @hookimpl
     def build_prompt(self, message, session_id, state):
-        return f"[echo] {message['content']}"
+        return f"[echo] {content_of(message)}"
 
     @hookimpl
     async def run_model(self, prompt, session_id, state):
         return prompt
+
+
+echo_plugin = EchoPlugin()
 ```
 
 ```toml
 [project.entry-points."bub"]
-echo = "my_package.plugin:EchoPlugin"
+echo = "my_package.plugin:echo_plugin"
 ```
 
-See the [Extending docs](https://bub.build/docs/extending/) for hook guides, packaging, and plugin structure.
+See the [Build docs](https://bub.build/docs/build/) for hook guides, packaging, and plugin structure.
 
 ## CLI
 
@@ -150,7 +154,7 @@ Lines starting with `,` enter internal command mode (`,help`, `,skill name=my-sk
 | `BUB_API_FORMAT`            | `completion`                 | `completion`, `responses`, or `messages`             |
 | `BUB_CLIENT_ARGS`           | —                            | JSON object forwarded to the underlying model client |
 | `BUB_MAX_STEPS`             | `50`                         | Max tool-use loop iterations                         |
-| `BUB_MAX_TOKENS`            | `1024`                       | Max tokens per model call                            |
+| `BUB_MAX_TOKENS`            | `16384`                      | Max tokens per model call                            |
 | `BUB_MODEL_TIMEOUT_SECONDS` | —                            | Model call timeout (seconds)                         |
 
 ## Background
@@ -166,11 +170,11 @@ Read more:
 ## Docs
 
 - [Getting Started](https://bub.build/docs/getting-started/) — install Bub and run the first turn
-- [Architecture](https://bub.build/docs/concepts/architecture/) — the mental model behind the runtime
-- [Channels](https://bub.build/docs/guides/channels/) — run Bub in CLI, Telegram, or your own channel
-- [Skills](https://bub.build/docs/guides/skills/) — discover, inspect, and author Agent Skills in Bub
-- [Extending](https://bub.build/docs/extending/) — write plugins, override hooks, ship tools and skills
-- [Deployment](https://bub.build/docs/guides/deployment/) — Docker, environment, upgrades
+- [Concepts](https://bub.build/docs/concepts/) — the mental model behind the runtime
+- [Channels](https://bub.build/docs/operate/channels/) — run Bub in CLI, Telegram, or your own channel
+- [Skills](https://bub.build/docs/build/skills/) — discover, inspect, and author Agent Skills in Bub
+- [Build](https://bub.build/docs/build/) — write plugins, override hooks, ship tools and skills
+- [Deployment](https://bub.build/docs/operate/deploy/) — Docker, environment, upgrades
 
 ## Development
 

{bub-0.3.7 → bub-0.3.9}/README.md

@@ -70,24 +70,28 @@ Key source files:
 
 ```python
 from bub import hookimpl
+from bub.envelope import content_of
 
 
 class EchoPlugin:
     @hookimpl
     def build_prompt(self, message, session_id, state):
-        return f"[echo] {message['content']}"
+        return f"[echo] {content_of(message)}"
 
     @hookimpl
     async def run_model(self, prompt, session_id, state):
         return prompt
+
+
+echo_plugin = EchoPlugin()
 ```
 
 ```toml
 [project.entry-points."bub"]
-echo = "my_package.plugin:EchoPlugin"
+echo = "my_package.plugin:echo_plugin"
 ```
 
-See the [Extending docs](https://bub.build/docs/extending/) for hook guides, packaging, and plugin structure.
+See the [Build docs](https://bub.build/docs/build/) for hook guides, packaging, and plugin structure.
 
 ## CLI
 
@@ -114,7 +118,7 @@ Lines starting with `,` enter internal command mode (`,help`, `,skill name=my-sk
 | `BUB_API_FORMAT`            | `completion`                 | `completion`, `responses`, or `messages`             |
 | `BUB_CLIENT_ARGS`           | —                            | JSON object forwarded to the underlying model client |
 | `BUB_MAX_STEPS`             | `50`                         | Max tool-use loop iterations                         |
-| `BUB_MAX_TOKENS`            | `1024`                       | Max tokens per model call                            |
+| `BUB_MAX_TOKENS`            | `16384`                      | Max tokens per model call                            |
 | `BUB_MODEL_TIMEOUT_SECONDS` | —                            | Model call timeout (seconds)                         |
 
 ## Background
@@ -130,11 +134,11 @@ Read more:
 ## Docs
 
 - [Getting Started](https://bub.build/docs/getting-started/) — install Bub and run the first turn
-- [Architecture](https://bub.build/docs/concepts/architecture/) — the mental model behind the runtime
-- [Channels](https://bub.build/docs/guides/channels/) — run Bub in CLI, Telegram, or your own channel
-- [Skills](https://bub.build/docs/guides/skills/) — discover, inspect, and author Agent Skills in Bub
-- [Extending](https://bub.build/docs/extending/) — write plugins, override hooks, ship tools and skills
-- [Deployment](https://bub.build/docs/guides/deployment/) — Docker, environment, upgrades
+- [Concepts](https://bub.build/docs/concepts/) — the mental model behind the runtime
+- [Channels](https://bub.build/docs/operate/channels/) — run Bub in CLI, Telegram, or your own channel
+- [Skills](https://bub.build/docs/build/skills/) — discover, inspect, and author Agent Skills in Bub
+- [Build](https://bub.build/docs/build/) — write plugins, override hooks, ship tools and skills
+- [Deployment](https://bub.build/docs/operate/deploy/) — Docker, environment, upgrades
 
 ## Development
 

{bub-0.3.7 → bub-0.3.9}/src/bub/__init__.py

@@ -13,8 +13,20 @@ from bub.configure import Settings, config, ensure_config
 from bub.framework import DEFAULT_HOME, BubFramework
 from bub.hookspecs import hookimpl
 from bub.tools import tool
-
-__all__ = ["BubFramework", "Settings", "config", "ensure_config", "home", "hookimpl", "tool"]
+from bub.turn_admission import AdmitDecision, SteeringBuffer, TurnSnapshot
+
+__all__ = [
+    "AdmitDecision",
+    "BubFramework",
+    "Settings",
+    "SteeringBuffer",
+    "TurnSnapshot",
+    "config",
+    "ensure_config",
+    "home",
+    "hookimpl",
+    "tool",
+]
 
 try:
     __version__ = import_module("bub._version").version

{bub-0.3.7 → bub-0.3.9}/src/bub/__main__.py

@@ -2,22 +2,27 @@
 
 from __future__ import annotations
 
+import sys
+
 import typer
 
 from bub.framework import BubFramework
 
 
 def _instrument_bub() -> None:
+    from loguru import logger
+
+    logger.remove()
+    logger.add(sys.stderr, colorize=True)
+
     try:
         import logfire
+        from logfire.integrations.loguru import LogfireHandler
 
         logfire.configure()
+        logger.add(LogfireHandler(), format="{message}")
     except ImportError:
         pass
-    else:
-        from loguru import logger
-
-        logger.configure(handlers=[logfire.loguru_handler()])
 
 
 def create_cli_app() -> typer.Typer:

{bub-0.3.7 → bub-0.3.9}/src/bub/_version.py

@@ -18,7 +18,7 @@ version_tuple: tuple[int | str, ...]
 commit_id: str | None
 __commit_id__: str | None
 
-__version__ = version = '0.3.7'
-__version_tuple__ = version_tuple = (0, 3, 7)
+__version__ = version = '0.3.9'
+__version_tuple__ = version_tuple = (0, 3, 9)
 
 __commit_id__ = commit_id = None

{bub-0.3.7 → bub-0.3.9}/src/bub/builtin/agent.py

@@ -24,6 +24,7 @@ from republic import (
     StreamEvent,
     StreamState,
     TapeContext,
+    Tool,
     ToolAutoResult,
     ToolContext,
 )
@@ -34,14 +35,14 @@ from bub.builtin.store import ForkTapeStore
 from bub.builtin.tape import TapeService
 from bub.framework import BubFramework
 from bub.skills import discover_skills, render_skills_prompt
-from bub.tools import REGISTRY, model_tools, render_tools_prompt
+from bub.tools import REGISTRY, model_tools, render_tools_prompt, resolve_tool_names
 from bub.types import State
 from bub.utils import workspace_from_state
 
-CONTINUE_PROMPT = "Continue the task."
+CONTINUE_PROMPT = "Continue the task until all targets are completed."
 HINT_RE = re.compile(r"\$([A-Za-z0-9_.-]+)")
 _CONTEXT_LENGTH_PATTERNS = re.compile(
-    r"context.{0,20}length|maximum.{0,20}context|token.{0,10}limit|prompt.{0,10}too long|tokens? > \d+ maximum",
+    r"context.{0,20}(?:length|window)|maximum.{0,20}context|token.{0,10}limit|prompt.{0,10}too long|tokens? > \d+ maximum",
     re.IGNORECASE,
 )
 MAX_AUTO_HANDOFF_RETRIES = 1
@@ -530,12 +531,12 @@ class Agent:
     ) -> AsyncStreamEvents | ToolAutoResult:
         prompt_text = prompt if isinstance(prompt, str) else _extract_text_from_parts(prompt)
         if allowed_tools is not None:
-            allowed_tools = {name.casefold() for name in allowed_tools}
+            allowed_tools = resolve_tool_names(allowed_tools)
         if allowed_skills is not None:
             allowed_skills = {name.casefold() for name in allowed_skills}
             tape.context.state["allowed_skills"] = list(allowed_skills)
         if allowed_tools is not None:
-            tools = [tool for tool in REGISTRY.values() if tool.name.casefold() in allowed_tools]
+            tools = [tool for tool in REGISTRY.values() if tool.name in allowed_tools]
         else:
             tools = list(REGISTRY.values())
         async with asyncio.timeout(self.settings.model_timeout_seconds):
@@ -543,7 +544,7 @@ class Agent:
                 return await tape.stream_events_async(
                     prompt=prompt,
                     system_prompt=self._system_prompt(
-                        prompt_text, state=tape.context.state, allowed_skills=allowed_skills
+                        prompt_text, state=tape.context.state, allowed_skills=allowed_skills, tools=tools
                     ),
                     max_tokens=self.settings.max_tokens,
                     tools=model_tools(tools),
@@ -553,18 +554,20 @@ class Agent:
                 return await tape.run_tools_async(
                     prompt=prompt,
                     system_prompt=self._system_prompt(
-                        prompt_text, state=tape.context.state, allowed_skills=allowed_skills
+                        prompt_text, state=tape.context.state, allowed_skills=allowed_skills, tools=tools
                     ),
                     max_tokens=self.settings.max_tokens,
                     tools=model_tools(tools),
                     model=model,
                 )
 
-    def _system_prompt(self, prompt: str, state: State, allowed_skills: set[str] | None = None) -> str:
+    def _system_prompt(
+        self, prompt: str, state: State, allowed_skills: set[str] | None = None, tools: Iterable[Tool] | None = None
+    ) -> str:
         blocks: list[str] = []
         if result := self.framework.get_system_prompt(prompt=prompt, state=state):
             blocks.append(result)
-        tools_prompt = render_tools_prompt(REGISTRY.values())
+        tools_prompt = render_tools_prompt(tools if tools is not None else REGISTRY.values())
         if tools_prompt:
             blocks.append(tools_prompt)
         workspace = workspace_from_state(state)

{bub-0.3.7 → bub-0.3.9}/src/bub/builtin/cli.py

@@ -21,6 +21,7 @@ from bub.builtin.auth import app as login_app  # noqa: F401
 from bub.channels.message import ChannelMessage
 from bub.envelope import field_of
 from bub.framework import BubFramework
+from bub.types import TurnResult
 
 ONBOARD_BANNER = r"""
  ███████████             █████
@@ -53,7 +54,11 @@ def run(
         context={"sender_id": sender_id},
     )
 
-    result = asyncio.run(framework.process_inbound(inbound))
+    async def _run() -> TurnResult:
+        async with framework.running():
+            return await framework.process_inbound(inbound)
+
+    result = asyncio.run(_run())
     for outbound in result.outbounds:
         rendered = str(field_of(outbound, "content", ""))
         target_channel = str(field_of(outbound, "channel", "stdout"))
@@ -128,8 +133,10 @@ def _find_uv() -> str:
     import shutil
     import sysconfig
 
+    default_path = Path.home() / ".local" / "bin"
+
     bin_path = sysconfig.get_path("scripts")
-    uv_path = shutil.which("uv", path=os.pathsep.join([bin_path, os.getenv("PATH", "")]))
+    uv_path = shutil.which("uv", path=os.pathsep.join([bin_path, str(default_path), os.getenv("PATH", "")]))
     if uv_path is None:
         raise FileNotFoundError("uv executable not found in PATH or scripts directory.")
     return uv_path

{bub-0.3.7 → bub-0.3.9}/src/bub/builtin/hook_impl.py

@@ -1,5 +1,5 @@
 import sys
-from datetime import UTC, datetime
+from datetime import datetime
 from pathlib import Path
 from typing import cast
 
@@ -134,7 +134,7 @@ class BuiltinImpl:
             message.kind = "command"
             return content
         context = field_of(message, "context_str")
-        now = datetime.now(UTC).strftime("%Y-%m-%dT%H:%M:%SZ")
+        now = datetime.now().astimezone().isoformat(timespec="seconds")
         context_prefix = f"{context}\n---Date: {now}---\n" if context else ""
         text = f"{context_prefix}{content}"
 

{bub-0.3.7 → bub-0.3.9}/src/bub/builtin/settings.py

@@ -12,7 +12,7 @@ from pydantic_settings import SettingsConfigDict
 from bub import Settings, config, ensure_config
 
 DEFAULT_MODEL = "openrouter:openrouter/free"
-DEFAULT_MAX_TOKENS = 1024
+DEFAULT_MAX_TOKENS = 16384
 
 
 def provider_specific(setting_name: str) -> Callable[[], dict[str, str] | None]:

{bub-0.3.7 → bub-0.3.9}/src/bub/builtin/shell_manager.py

@@ -13,6 +13,7 @@ class ManagedShell:
     shell_id: str
     cmd: str
     cwd: str | None
+    session_id: str | None
     process: asyncio.subprocess.Process
     output_chunks: list[str] = field(default_factory=list)
     read_tasks: list[asyncio.Task[None]] = field(default_factory=list)
@@ -36,7 +37,7 @@ class ShellManager:
     def __init__(self) -> None:
         self._shells: dict[str, ManagedShell] = {}
 
-    async def start(self, *, cmd: str, cwd: str | None) -> ManagedShell:
+    async def start(self, *, cmd: str, cwd: str | None, session_id: str | None = None) -> ManagedShell:
         process = await asyncio.create_subprocess_shell(
             cmd,
             cwd=cwd,
@@ -44,7 +45,13 @@ class ShellManager:
             stderr=asyncio.subprocess.PIPE,
             executable=self.SHELL,
         )
-        shell = ManagedShell(shell_id=f"bash-{uuid.uuid4().hex[:8]}", cmd=cmd, cwd=cwd, process=process)
+        shell = ManagedShell(
+            shell_id=f"bash-{uuid.uuid4().hex[:8]}",
+            cmd=cmd,
+            cwd=cwd,
+            session_id=session_id,
+            process=process,
+        )
         shell.read_tasks.extend([
             asyncio.create_task(self._drain_stream(shell, process.stdout)),
             asyncio.create_task(self._drain_stream(shell, process.stderr)),
@@ -77,6 +84,13 @@ class ShellManager:
         await self._finalize_shell(shell)
         return shell
 
+    async def terminate_session(self, session_id: str) -> int:
+        shell_ids = [shell.shell_id for shell in self._shells.values() if shell.session_id == session_id]
+        for shell_id in shell_ids:
+            with contextlib.suppress(KeyError):
+                await self.terminate(shell_id)
+        return len(shell_ids)
+
     async def wait_closed(self, shell_id: str) -> ManagedShell:
         shell = self.get(shell_id)
         if shell.returncode is None:

{bub-0.3.7 → bub-0.3.9}/src/bub/builtin/store.py

@@ -101,16 +101,22 @@ class ForkTapeStore:
     @contextlib.asynccontextmanager
     async def fork(self, tape: str, merge_back: bool = True) -> AsyncGenerator[None, None]:
         store = InMemoryTapeStore()
-        token = current_store.set(store)
-        tape_token = current_fork_tape.set(tape)
-        reset_token = current_tape_was_reset.set(False)
+        # Save/restore instead of ContextVar.reset(token) to avoid
+        # "Token was created in a different Context" when cleanup
+        # runs in a different asyncio Task (e.g. cancellation, TaskGroup).
+        prev_store = current_store.get(_empty_store)
+        prev_fork_tape = current_fork_tape.get()
+        prev_was_reset = current_tape_was_reset.get()
+        current_store.set(store)
+        current_fork_tape.set(tape)
+        current_tape_was_reset.set(False)
         try:
             yield
         finally:
             was_reset = current_tape_was_reset.get()
-            current_store.reset(token)
-            current_fork_tape.reset(tape_token)
-            current_tape_was_reset.reset(reset_token)
+            current_store.set(prev_store)
+            current_fork_tape.set(prev_fork_tape)
+            current_tape_was_reset.set(prev_was_reset)
             if merge_back:
                 if was_reset:
                     await self._parent.reset(tape)

{bub-0.3.7 → bub-0.3.9}/src/bub/builtin/tools.py

@@ -79,12 +79,17 @@ async def bash(
     """Run a shell command. Use background=true to keep it running and fetch output later via bash_output."""
     workspace = context.state.get("_runtime_workspace")
     target_cwd = cwd or workspace
-    shell = await shell_manager.start(cmd=cmd, cwd=target_cwd)
+    raw_session_id = context.state.get("session_id")
+    session_id = str(raw_session_id) if raw_session_id is not None else None
+    shell = await shell_manager.start(cmd=cmd, cwd=target_cwd, session_id=session_id)
     if background:
         return f"started: {shell.shell_id}"
     try:
         async with asyncio.timeout(timeout_seconds):
             shell = await shell_manager.wait_closed(shell.shell_id)
+    except asyncio.CancelledError:
+        await shell_manager.terminate(shell.shell_id)
+        raise
     except TimeoutError:
         await shell_manager.terminate(shell.shell_id)
         return f"command timed out after {timeout_seconds} seconds and was terminated"
@@ -307,9 +312,10 @@ def show_help() -> str:
 
 @tool(name="quit", context=True)
 async def quit_tool(*, context: ToolContext) -> str:
-    """Quit the tasks of the current session."""
+    """Abort the tasks of the current session. DO NOT use it in a normal workflow."""
     agent = _get_agent(context)
-    session_id = context.state.get("session_id", "temp/unknown")
+    session_id = str(context.state.get("session_id", "temp/unknown"))
+    await shell_manager.terminate_session(session_id)
     await agent.framework.quit_via_router(session_id)
     return "Session tasks stopped."
 

bub-0.3.9/src/bub/channels/__init__.py

@@ -0,0 +1,5 @@
+from .base import Channel, Interface, Lifecycle
+from .manager import ChannelManager
+from .message import ChannelMessage
+
+__all__ = ["Channel", "ChannelManager", "ChannelMessage", "Interface", "Lifecycle"]

{bub-0.3.7 → bub-0.3.9}/src/bub/channels/base.py

@@ -39,3 +39,11 @@ class Channel(ABC):
     def stream_events(self, message: ChannelMessage, stream: AsyncIterable[StreamEvent]) -> AsyncIterable[StreamEvent]:
         """Optionally wrap the output stream for this channel."""
         return stream
+
+
+class Interface(Channel):
+    """User-facing inbound/outbound surface managed by the channel runtime."""
+
+
+class Lifecycle(Channel):
+    """Background runtime managed alongside channels."""

{bub-0.3.7 → bub-0.3.9}/src/bub/channels/cli/__init__.py

@@ -14,12 +14,11 @@ from prompt_toolkit.key_binding import KeyBindings
 from prompt_toolkit.patch_stdout import patch_stdout
 from republic import StreamEvent
 from rich import get_console
-from rich.live import Live
 
 import bub
 from bub.builtin.agent import Agent
 from bub.builtin.tape import TapeInfo
-from bub.channels.base import Channel
+from bub.channels.base import Interface
 from bub.channels.cli.renderer import CliRenderer
 from bub.channels.message import ChannelMessage
 from bub.envelope import field_of
@@ -27,7 +26,7 @@ from bub.tools import REGISTRY
 from bub.types import MessageHandler
 
 
-class CliChannel(Channel):
+class CliChannel(Interface):
     """A simple CLI channel for testing and debugging."""
 
     name = "cli"
@@ -44,14 +43,13 @@ class CliChannel(Channel):
         self._mode = "agent"  # or "shell"
         self._main_task: asyncio.Task | None = None
         self._renderer = CliRenderer(get_console())
-        self._log_handler_id = self._install_log_sink()
         self._last_tape_info: TapeInfo | None = None
         self._workspace = self._agent.framework.workspace
         self._prompt = self._build_prompt(self._workspace)
 
     def _install_log_sink(self) -> int:
         with contextlib.suppress(ValueError):
-            logger.remove(0)
+            logger.remove()
         return logger.add(self._renderer.log, colorize=False, format="{level:<8} | {message}")
 
     async def _refresh_tape_info(self) -> None:
@@ -66,6 +64,7 @@ class CliChannel(Channel):
             self._message_template["chat_id"] = chat_id
 
     async def start(self, stop_event: asyncio.Event) -> None:
+        self._log_handler_id = self._install_log_sink()
         self._stop_event = stop_event
         self._main_task = asyncio.create_task(self._main_loop())
 
@@ -141,23 +140,21 @@ class CliChannel(Channel):
     async def stream_events(
         self, message: ChannelMessage, stream: AsyncIterable[StreamEvent]
     ) -> AsyncIterable[StreamEvent]:
-        live: Live | None = None
-        text = ""
-        try:
-            async for event in stream:
-                if event.kind == "text":
-                    content = str(event.data.get("delta", ""))
-                    if not content.strip() and not text:
-                        continue  # skip leading whitespace-only events
-                    text += content
-                    if live is None:
-                        live = self._renderer.start_stream(message.kind, text)
-                    else:
-                        self._renderer.update_stream(live, kind=message.kind, text=text)
-                yield event
-        finally:
-            if live is not None:
-                self._renderer.finish_stream(live, kind=message.kind, text=text)
+        head_printed = False
+        console = get_console()
+        async for event in stream:
+            if event.kind == "text":
+                content = str(event.data.get("delta", ""))
+                if not content.strip() and not head_printed:
+                    continue  # skip leading whitespace-only events
+                if not head_printed:
+                    self._renderer.print_head(message.kind)
+                    head_printed = True
+
+                console.print(content, end="", highlight=False)
+            elif event.kind == "final":
+                console.print("\n")  # ensure we end with a newline after the final event
+            yield event
 
     def _build_prompt(self, workspace: Path) -> PromptSession[str]:
         kb = KeyBindings()

{bub-0.3.7 → bub-0.3.9}/src/bub/channels/cli/renderer.py

@@ -5,7 +5,6 @@ from __future__ import annotations
 from dataclasses import dataclass
 
 from rich.console import Console
-from rich.live import Live
 from rich.panel import Panel
 from rich.text import Text
 
@@ -33,54 +32,30 @@ class CliRenderer:
             return
         self.console.print(Text(text, style="bright_black"))
 
-    def panel(self, kind: MessageKind, text: str) -> Panel:
-        title, border_style = self._panel_style(kind)
-        return Panel(text, title=title, border_style=border_style)
-
     def command_output(self, text: str) -> None:
         if not text.strip():
             return
-        self.console.print(self.panel("command", text))
+        self.console.print(f"[cyan bold]Command >[/]\n{text}")
 
     def assistant_output(self, text: str) -> None:
         if not text.strip():
             return
-        self.console.print(self.panel("normal", text))
+        self.console.print(f"[blue bold]Assistant >[/]\n{text}")
 
     def error(self, text: str) -> None:
         if not text.strip():
             return
-        self.console.print(self.panel("error", text))
+        self.console.print(f"[red bold]Error >[/]\n{text}")
+
+    def print_head(self, kind: MessageKind) -> None:
+        if kind == "command":
+            self.console.print("[cyan bold]Command >[/]")
+        elif kind == "error":
+            self.console.print("[red bold]Error >[/]")
+        else:
+            self.console.print("[blue bold]Assistant >[/]")
 
     def log(self, message: object) -> None:
         text = str(message).rstrip()
         if text:
             self.console.print(text)
-
-    def start_stream(self, kind: MessageKind, text: str) -> Live:
-        live = Live(
-            self.panel(kind, text),
-            console=self.console,
-            auto_refresh=False,
-            transient=False,
-            vertical_overflow="visible",
-        )
-        live.start(refresh=True)
-        return live
-
-    def update_stream(self, live: Live, *, kind: MessageKind, text: str) -> None:
-        live.update(self.panel(kind, text), refresh=True)
-
-    def finish_stream(self, live: Live, *, kind: MessageKind, text: str) -> None:
-        live.update(self.panel(kind, text), refresh=True)
-        live.stop()
-
-    @staticmethod
-    def _panel_style(kind: MessageKind) -> tuple[str, str]:
-        match kind:
-            case "error":
-                return "Error", "red"
-            case "command":
-                return "Command", "green"
-            case _:
-                return "Assistant", "blue"