bub 0.3.4__tar.gz → 0.3.6__tar.gz

{bub-0.3.4 → bub-0.3.6}/.gitignore

@@ -145,3 +145,4 @@ reference/
 
 # Local legacy backups created during framework migrations
 backup/
+_version.py

{bub-0.3.4 → bub-0.3.6}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: bub
-Version: 0.3.4
+Version: 0.3.6
 Summary: A common shape for agents that live alongside people.
 Project-URL: Homepage, https://bub.build
 Project-URL: Repository, https://github.com/bubbuild/bub
@@ -130,11 +130,14 @@ See the [Extension Guide](https://bub.build/extension-guide/) for hook semantics
 | `bub chat`         | Interactive REPL                  |
 | `bub run MESSAGE`  | One-shot turn                     |
 | `bub gateway`      | Channel listener (Telegram, etc.) |
+| `bub install`      | Install or sync Bub plugin deps   |
+| `bub update`       | Upgrade Bub plugin deps           |
 | `bub login openai` | OpenAI Codex OAuth                |
-| `bub hooks`        | Print hook-to-plugin bindings     |
 
 Lines starting with `,` enter internal command mode (`,help`, `,skill name=my-skill`, `,fs.read path=README.md`).
 
+`bub hooks` still exists for diagnostics, but it is hidden from top-level help. `bub install` and `bub update` manage a separate uv project for Bub plugins, defaulting to `~/.bub/bub-project` or `BUB_PROJECT`.
+
 ## Configuration
 
 | Variable | Default | Description |

{bub-0.3.4 → bub-0.3.6}/README.md

@@ -96,11 +96,14 @@ See the [Extension Guide](https://bub.build/extension-guide/) for hook semantics
 | `bub chat`         | Interactive REPL                  |
 | `bub run MESSAGE`  | One-shot turn                     |
 | `bub gateway`      | Channel listener (Telegram, etc.) |
+| `bub install`      | Install or sync Bub plugin deps   |
+| `bub update`       | Upgrade Bub plugin deps           |
 | `bub login openai` | OpenAI Codex OAuth                |
-| `bub hooks`        | Print hook-to-plugin bindings     |
 
 Lines starting with `,` enter internal command mode (`,help`, `,skill name=my-skill`, `,fs.read path=README.md`).
 
+`bub hooks` still exists for diagnostics, but it is hidden from top-level help. `bub install` and `bub update` manage a separate uv project for Bub plugins, defaulting to `~/.bub/bub-project` or `BUB_PROJECT`.
+
 ## Configuration
 
 | Variable | Default | Description |

{bub-0.3.4 → bub-0.3.6}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "bub"
-version = "0.3.4"
+dynamic = ["version"]
 description = "A common shape for agents that live alongside people."
 authors = [
     { name = "Chojan Shang", email = "psiace@apache.org" },
@@ -64,9 +64,16 @@ dev = [
 ]
 
 [build-system]
-requires = ["hatchling"]
+requires = ["hatchling", "hatch-vcs"]
 build-backend = "hatchling.build"
 
+[tool.hatch.version]
+source = "vcs"
+fallback-version = "0.3.0"
+
+[tool.hatch.build.hooks.vcs]
+version-file = "src/bub/_version.py"
+
 [tool.hatch.build.targets.sdist]
 only-include = ["src/bub", "src/skills", "tests"]
 

bub-0.3.6/src/bub/__init__.py

@@ -0,0 +1,19 @@
+"""Bub framework package."""
+
+from importlib import import_module
+from importlib.metadata import PackageNotFoundError
+from importlib.metadata import version as metadata_version
+
+from bub.framework import BubFramework
+from bub.hookspecs import hookimpl
+from bub.tools import tool
+
+__all__ = ["BubFramework", "hookimpl", "tool"]
+
+try:
+    __version__ = import_module("bub._version").version
+except ModuleNotFoundError:
+    try:
+        __version__ = metadata_version("bub")
+    except PackageNotFoundError:
+        __version__ = "0.0.0"

bub-0.3.6/src/bub/_version.py

@@ -0,0 +1,24 @@
+# file generated by vcs-versioning
+# don't change, don't track in version control
+from __future__ import annotations
+
+__all__ = [
+    "__version__",
+    "__version_tuple__",
+    "version",
+    "version_tuple",
+    "__commit_id__",
+    "commit_id",
+]
+
+version: str
+__version__: str
+__version_tuple__: tuple[int | str, ...]
+version_tuple: tuple[int | str, ...]
+commit_id: str | None
+__commit_id__: str | None
+
+__version__ = version = '0.3.6'
+__version_tuple__ = version_tuple = (0, 3, 6)
+
+__commit_id__ = commit_id = None

{bub-0.3.4 → bub-0.3.6}/src/bub/builtin/agent.py

@@ -13,16 +13,18 @@ from dataclasses import dataclass, replace
 from datetime import UTC, datetime
 from functools import cached_property
 from pathlib import Path
-from typing import Any
+from typing import Any, Literal, overload
 
 from loguru import logger
 from republic import (
     LLM,
     AsyncStreamEvents,
     AsyncTapeStore,
+    RepublicError,
     StreamEvent,
     StreamState,
     TapeContext,
+    ToolAutoResult,
     ToolContext,
 )
 from republic.tape import InMemoryTapeStore, Tape
@@ -89,6 +91,29 @@ class Agent:
         model: str | None = None,
         allowed_skills: Collection[str] | None = None,
         allowed_tools: Collection[str] | None = None,
+    ) -> str:
+        if not prompt:
+            return "error: empty prompt"
+        tape = self.tapes.session_tape(session_id, workspace_from_state(state))
+        tape.context = replace(tape.context, state=state)
+        merge_back = not session_id.startswith("temp/")
+        async with self.tapes.fork_tape(tape.name, merge_back=merge_back):
+            await self.tapes.ensure_bootstrap_anchor(tape.name)
+            if isinstance(prompt, str) and prompt.strip().startswith(","):
+                return await self._run_command(tape=tape, line=prompt.strip())
+            return await self._agent_loop(
+                tape=tape, prompt=prompt, model=model, allowed_skills=allowed_skills, allowed_tools=allowed_tools
+            )
+
+    async def run_stream(
+        self,
+        *,
+        session_id: str,
+        prompt: str | list[dict],
+        state: State,
+        model: str | None = None,
+        allowed_skills: Collection[str] | None = None,
+        allowed_tools: Collection[str] | None = None,
     ) -> AsyncStreamEvents:
         if not prompt:
             events = [
@@ -113,7 +138,12 @@ class Agent:
             ])
         else:
             events = await self._agent_loop(
-                tape=tape, prompt=prompt, model=model, allowed_skills=allowed_skills, allowed_tools=allowed_tools
+                tape=tape,
+                prompt=prompt,
+                model=model,
+                allowed_skills=allowed_skills,
+                allowed_tools=allowed_tools,
+                stream_output=True,
             )
         return self._events_with_callback(events, callback=stack.aclose)
 
@@ -157,6 +187,30 @@ class Agent:
             }
             await self.tapes.append_event(tape.name, "command", event_payload)
 
+    @overload
+    async def _agent_loop(
+        self,
+        *,
+        tape: Tape,
+        prompt: str | list[dict],
+        model: str | None = ...,
+        allowed_skills: Collection[str] | None = ...,
+        allowed_tools: Collection[str] | None = ...,
+        stream_output: Literal[False] = ...,
+    ) -> str: ...
+
+    @overload
+    async def _agent_loop(
+        self,
+        *,
+        tape: Tape,
+        prompt: str | list[dict],
+        model: str | None = ...,
+        allowed_skills: Collection[str] | None = ...,
+        allowed_tools: Collection[str] | None = ...,
+        stream_output: Literal[True] = ...,
+    ) -> AsyncStreamEvents: ...
+
     async def _agent_loop(
         self,
         *,
@@ -165,7 +219,8 @@ class Agent:
         model: str | None = None,
         allowed_skills: Collection[str] | None = None,
         allowed_tools: Collection[str] | None = None,
-    ) -> AsyncStreamEvents:
+        stream_output: bool = False,
+    ) -> AsyncStreamEvents | str:
         next_prompt: str | list[dict] = prompt
         display_model = model or self.settings.model
         await self.tapes.append_event(
@@ -178,16 +233,137 @@ class Agent:
                 "allowed_tools": list(allowed_tools) if allowed_tools else None,
             },
         )
-        state = StreamState()
-        iterator = self._stream_events_with_auto_handoff(
-            tape=tape,
-            prompt=next_prompt,
-            state=state,
-            model=model,
-            allowed_skills=allowed_skills,
-            allowed_tools=allowed_tools,
-        )
-        return AsyncStreamEvents(iterator, state=state)
+        if stream_output:
+            state = StreamState()
+            iterator = self._stream_events_with_auto_handoff(
+                tape=tape,
+                prompt=next_prompt,
+                state=state,
+                model=model,
+                allowed_skills=allowed_skills,
+                allowed_tools=allowed_tools,
+            )
+            return AsyncStreamEvents(iterator, state=state)
+        else:
+            return await self._run_tools_with_auto_handoff(
+                tape=tape,
+                prompt=next_prompt,
+                model=model,
+                allowed_skills=allowed_skills,
+                allowed_tools=allowed_tools,
+            )
+
+    async def _run_tools_with_auto_handoff(
+        self,
+        tape: Tape,
+        prompt: str | list[dict],
+        model: str | None = None,
+        allowed_skills: Collection[str] | None = None,
+        allowed_tools: Collection[str] | None = None,
+    ) -> str:
+        auto_handoff_remaining = MAX_AUTO_HANDOFF_RETRIES
+        display_model = model or self.settings.model
+        next_prompt = prompt
+        for step in range(1, self.settings.max_steps + 1):
+            start = time.monotonic()
+            logger.info("loop.step step={} tape={} model={}", step, tape.name, display_model)
+            await self.tapes.append_event(tape.name, "loop.step.start", {"step": step, "prompt": next_prompt})
+            try:
+                output = await self._run_once(
+                    tape=tape,
+                    prompt=next_prompt,
+                    model=model,
+                    allowed_skills=allowed_skills,
+                    allowed_tools=allowed_tools,
+                )
+            except Exception as exc:
+                elapsed_ms = int((time.monotonic() - start) * 1000)
+                await self.tapes.append_event(
+                    tape.name,
+                    "loop.step",
+                    {
+                        "step": step,
+                        "elapsed_ms": elapsed_ms,
+                        "status": "error",
+                        "error": f"{exc!s}",
+                        "date": datetime.now(UTC).isoformat(),
+                    },
+                )
+                raise
+
+            outcome = _resolve_tool_auto_result(output)
+            elapsed_ms = int((time.monotonic() - start) * 1000)
+            if outcome.kind == "text":
+                await self.tapes.append_event(
+                    tape.name,
+                    "loop.step",
+                    {
+                        "step": step,
+                        "elapsed_ms": elapsed_ms,
+                        "status": "ok",
+                        "date": datetime.now(UTC).isoformat(),
+                    },
+                )
+                return outcome.text
+            if outcome.kind == "continue":
+                if "context" in tape.context.state:
+                    next_prompt = f"{CONTINUE_PROMPT} [context: {tape.context.state['context']}]"
+                else:
+                    next_prompt = CONTINUE_PROMPT
+                await self.tapes.append_event(
+                    tape.name,
+                    "loop.step",
+                    {
+                        "step": step,
+                        "elapsed_ms": elapsed_ms,
+                        "status": "continue",
+                        "date": datetime.now(UTC).isoformat(),
+                    },
+                )
+                continue
+
+            # Check if this is a context-length error that can be recovered via auto-handoff
+            if auto_handoff_remaining > 0 and _is_context_length_error(outcome.error):
+                auto_handoff_remaining -= 1
+                logger.warning(
+                    "auto_handoff: context length exceeded, performing automatic handoff. tape={} step={}",
+                    tape.name,
+                    step,
+                )
+                await self.tapes.handoff(
+                    tape.name,
+                    name="auto_handoff/context_overflow",
+                    state={"reason": "context_length_exceeded", "error": outcome.error},
+                )
+                await self.tapes.append_event(
+                    tape.name,
+                    "loop.step",
+                    {
+                        "step": step,
+                        "elapsed_ms": elapsed_ms,
+                        "status": "auto_handoff",
+                        "error": outcome.error,
+                        "date": datetime.now(UTC).isoformat(),
+                    },
+                )
+                # Retry with original prompt — the handoff anchor will truncate history
+                next_prompt = prompt
+                continue
+
+            await self.tapes.append_event(
+                tape.name,
+                "loop.step",
+                {
+                    "step": step,
+                    "elapsed_ms": elapsed_ms,
+                    "status": "error",
+                    "error": outcome.error,
+                    "date": datetime.now(UTC).isoformat(),
+                },
+            )
+            raise RuntimeError(outcome.error)
+
+        raise RuntimeError(f"max_steps_reached={self.settings.max_steps}")
 
     async def _stream_events_with_auto_handoff(
         self,
@@ -212,6 +388,7 @@ class Agent:
                 model=model,
                 allowed_skills=allowed_skills,
                 allowed_tools=allowed_tools,
+                stream_output=True,
             )
             async for event in output:
                 yield event
@@ -229,7 +406,7 @@ class Agent:
                         },
                     )
                 elif event.kind == "final":
-                    outcome = _resolve_tool_auto_result(event.data)
+                    outcome = _resolve_final_data(event.data, output.error)
 
             state.error = output.error
             state.usage = output.usage
@@ -315,6 +492,30 @@ class Agent:
         expanded_skills = set(HINT_RE.findall(prompt)) & set(skill_index.keys())
         return render_skills_prompt(list(skill_index.values()), expanded_skills=expanded_skills)
 
+    @overload
+    async def _run_once(
+        self,
+        *,
+        tape: Tape,
+        prompt: str | list[dict],
+        model: str | None = ...,
+        allowed_skills: Collection[str] | None = ...,
+        allowed_tools: Collection[str] | None = ...,
+        stream_output: Literal[False] = ...,
+    ) -> ToolAutoResult: ...
+
+    @overload
+    async def _run_once(
+        self,
+        *,
+        tape: Tape,
+        prompt: str | list[dict],
+        model: str | None = ...,
+        allowed_skills: Collection[str] | None = ...,
+        allowed_tools: Collection[str] | None = ...,
+        stream_output: Literal[True] = ...,
+    ) -> AsyncStreamEvents: ...
+
     async def _run_once(
         self,
         *,
@@ -323,7 +524,8 @@ class Agent:
         model: str | None = None,
         allowed_tools: Collection[str] | None = None,
         allowed_skills: Collection[str] | None = None,
-    ) -> AsyncStreamEvents:
+        stream_output: bool = False,
+    ) -> 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}
@@ -335,13 +537,26 @@ class Agent:
         else:
             tools = list(REGISTRY.values())
         async with asyncio.timeout(self.settings.model_timeout_seconds):
-            return await tape.stream_events_async(
-                prompt=prompt,
-                system_prompt=self._system_prompt(prompt_text, state=tape.context.state, allowed_skills=allowed_skills),
-                max_tokens=self.settings.max_tokens,
-                tools=model_tools(tools),
-                model=model,
-            )
+            if stream_output:
+                return await tape.stream_events_async(
+                    prompt=prompt,
+                    system_prompt=self._system_prompt(
+                        prompt_text, state=tape.context.state, allowed_skills=allowed_skills
+                    ),
+                    max_tokens=self.settings.max_tokens,
+                    tools=model_tools(tools),
+                    model=model,
+                )
+            else:
+                return await tape.run_tools_async(
+                    prompt=prompt,
+                    system_prompt=self._system_prompt(
+                        prompt_text, state=tape.context.state, allowed_skills=allowed_skills
+                    ),
+                    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:
         blocks: list[str] = []
@@ -363,12 +578,24 @@ class _ToolAutoOutcome:
     error: str = ""
 
 
-def _resolve_tool_auto_result(final_data: dict[str, Any]) -> _ToolAutoOutcome:
+def _resolve_final_data(final_data: dict[str, Any], error: RepublicError | None) -> _ToolAutoOutcome:
+    if final_data.get("tool_calls") or final_data.get("tool_results"):
+        return _ToolAutoOutcome(kind="continue")
     if (text := final_data.get("text")) is not None:
         return _ToolAutoOutcome(kind="text", text=text)
-    if final_data.get("tool_calls") or final_data.get("tool_results"):
+    error_message = error.message if error else ""
+    return _ToolAutoOutcome(kind="error", error=error_message or "unknown error")
+
+
+def _resolve_tool_auto_result(output: ToolAutoResult) -> _ToolAutoOutcome:
+    if output.kind == "text":
+        return _ToolAutoOutcome(kind="text", text=output.text or "")
+    if output.kind == "tools" or output.tool_calls or output.tool_results:
         return _ToolAutoOutcome(kind="continue")
-    return _ToolAutoOutcome(kind="error", error="unknown error")
+    if output.error is None:
+        return _ToolAutoOutcome(kind="error", error="tool_auto_error: unknown")
+    error_kind = getattr(output.error.kind, "value", str(output.error.kind))
+    return _ToolAutoOutcome(kind="error", error=f"{error_kind}: {output.error.message}")
 
 
 def _build_llm(settings: AgentSettings, tape_store: AsyncTapeStore, tape_context: TapeContext) -> LLM:

{bub-0.3.4 → bub-0.3.6}/src/bub/builtin/cli.py

@@ -4,11 +4,15 @@
 from __future__ import annotations
 
 import asyncio
+import json
 import os
 import subprocess
 import sys
 from functools import lru_cache
+from importlib import metadata
 from pathlib import Path
+from urllib.parse import unquote, urlsplit
+from urllib.request import url2pathname
 
 import typer
 
@@ -79,7 +83,7 @@ def chat(
 
     framework = ctx.ensure_object(BubFramework)
 
-    manager = ChannelManager(framework, enabled_channels=["cli"])
+    manager = ChannelManager(framework, enabled_channels=["cli"], stream_output=True)
     channel = manager.get_channel("cli")
     if channel is None:
         typer.echo("CLI channel not found. Please check your hook implementations.")
@@ -148,15 +152,67 @@ def _build_requirement(spec: str) -> str:
         return f"git+https://github.com/{repo}.git{ref}"
     else:
         # Assume it's a package name in bub-contrib
-        name, *rest = spec.partition("@")
-        ref = "".join(rest)
-        return f"git+{BUB_CONTRIB_REPO}{ref}#subdirectory=packages/{name}"
+        name, has_ref, ref = spec.partition("@")
+        if has_ref:
+            ref = f"@{ref}"
+            return f"git+{BUB_CONTRIB_REPO}{ref}#subdirectory=packages/{name}"
+        else:  # PyPI package name
+            return name
+
+
+def _build_local_requirement_path(url: str, subdirectory: str | None = None) -> str | None:
+    parsed = urlsplit(url)
+    if parsed.scheme != "file":
+        return None
+
+    path = parsed.path
+    if parsed.netloc and parsed.netloc != "localhost":
+        path = f"//{parsed.netloc}{path}"
+    local_path = Path(url2pathname(unquote(path)))
+    if subdirectory:
+        local_path /= subdirectory
+    return os.fspath(local_path)
+
+
+def _build_bub_requirement() -> list[str]:
+    dist = metadata.distribution("bub")
+    dist_name = dist.name
+    direct_url_text = dist.read_text("direct_url.json")
+    if not direct_url_text:
+        return [dist_name]
+
+    direct_url = json.loads(direct_url_text)
+    requirement_url = str(direct_url["url"])
+    subdirectory = direct_url.get("subdirectory")
+    normalized_subdirectory = subdirectory if isinstance(subdirectory, str) and subdirectory else None
+
+    local_path = _build_local_requirement_path(requirement_url, normalized_subdirectory)
+    if local_path is not None:
+        dir_info = direct_url.get("dir_info")
+        editable = isinstance(dir_info, dict) and bool(dir_info.get("editable"))
+        return ["--editable", local_path] if editable else [local_path]
+
+    vcs_info = direct_url.get("vcs_info")
+    if isinstance(vcs_info, dict):
+        vcs = vcs_info.get("vcs")
+        requested_revision = vcs_info.get("requested_revision")
+        if isinstance(vcs, str) and vcs:
+            requirement_url = f"{vcs}+{requirement_url}"
+        if isinstance(requested_revision, str) and requested_revision:
+            requirement_url = f"{requirement_url}@{requested_revision}"
+
+    if normalized_subdirectory:
+        requirement_url = f"{requirement_url}#subdirectory={normalized_subdirectory}"
+
+    return [requirement_url]
 
 
 def _ensure_project(project: Path) -> None:
     if (project / "pyproject.toml").is_file():
         return
     _uv("init", "--bare", "--name", "bub-project", "--app", cwd=project)
+    bub_requirement = _build_bub_requirement()
+    _uv("add", "--active", "--no-sync", *bub_requirement, cwd=project)
 
 
 def install(
@@ -180,8 +236,7 @@ def uninstall(
 ) -> None:
     """Uninstall a plugin from Bub's environment."""
     _ensure_project(project)
-    _uv("remove", "--active", "--no-sync", *packages, cwd=project)
-    _uv("sync", "--active", "--frozen", "--inexact", cwd=project)
+    _uv("remove", "--active", *packages, cwd=project)
 
 
 def update(

{bub-0.3.4 → bub-0.3.6}/src/bub/builtin/hook_impl.py

@@ -106,9 +106,13 @@ class BuiltinImpl:
         return text
 
     @hookimpl
-    async def run_model_stream(self, prompt: str | list[dict], session_id: str, state: State) -> AsyncStreamEvents:
+    async def run_model(self, prompt: str | list[dict], session_id: str, state: State) -> str:
         return await self.agent.run(session_id=session_id, prompt=prompt, state=state)
 
+    @hookimpl
+    async def run_model_stream(self, prompt: str | list[dict], session_id: str, state: State) -> AsyncStreamEvents:
+        return await self.agent.run_stream(session_id=session_id, prompt=prompt, state=state)
+
     @hookimpl
     def register_cli_commands(self, app: typer.Typer) -> None:
         from bub.builtin import cli
@@ -119,8 +123,7 @@ class BuiltinImpl:
         app.command("hooks", hidden=True)(cli.list_hooks)
         app.command("gateway")(cli.gateway)
         app.command("install")(cli.install)
-        # TODO: uninstall command can't work properly
-        # app.command("uninstall")(cli.uninstall)
+        app.command("uninstall")(cli.uninstall)
         app.command("update")(cli.update)
 
     def _read_agents_file(self, state: State) -> str:

{bub-0.3.4 → bub-0.3.6}/src/bub/builtin/shell_manager.py

@@ -3,6 +3,7 @@ from __future__ import annotations
 import asyncio
 import contextlib
 import os
+import shutil
 import uuid
 from dataclasses import dataclass, field
 
@@ -30,6 +31,8 @@ class ManagedShell:
 
 
 class ShellManager:
+    SHELL = shutil.which("bash") or shutil.which("sh") if os.name != "nt" else None
+
     def __init__(self) -> None:
         self._shells: dict[str, ManagedShell] = {}
 
@@ -39,7 +42,7 @@ class ShellManager:
             cwd=cwd,
             stdout=asyncio.subprocess.PIPE,
             stderr=asyncio.subprocess.PIPE,
-            executable="/bin/bash" if os.name != "nt" else None,
+            executable=self.SHELL,
         )
         shell = ManagedShell(shell_id=f"bash-{uuid.uuid4().hex[:8]}", cmd=cmd, cwd=cwd, process=process)
         shell.read_tasks.extend([

{bub-0.3.4 → bub-0.3.6}/src/bub/builtin/tools.py

@@ -42,9 +42,9 @@ class SearchInput(BaseModel):
     limit: int = Field(20, description="Maximum number of search results to return.")
     start: str | None = Field(None, description="Optional start date to filter entries (ISO format).")
     end: str | None = Field(None, description="Optional end date to filter entries (ISO format).")
-    kinds: list[EntryKind] = Field(
+    kinds: list[str] = Field(
         default=["message", "tool_result"],
-        description="Optional list of entry kinds to filter search results.",
+        description="Optional list of entry kinds to filter search results. Can include 'event', 'anchor', 'system', 'message', 'tool_call', 'tool_result'.",
     )
 
 
@@ -267,7 +267,7 @@ async def run_subagent(param: SubAgentInput, *, context: ToolContext) -> str:
     state = {**context.state, "session_id": subagent_session}
     allowed_tools = resolve_tool_names(param.allowed_tools or None, exclude={"subagent"})
     output = ""
-    async for event in await agent.run(
+    async for event in await agent.run_stream(
         session_id=subagent_session,
         prompt=param.prompt,
         state=state,