bub 0.4.2__tar.gz → 0.4.3__tar.gz

{bub-0.4.2 → bub-0.4.3}/PKG-INFO

@@ -1,11 +1,12 @@
-Metadata-Version: 2.4
+Metadata-Version: 2.5
 Name: bub
-Version: 0.4.2
+Version: 0.4.3
 Summary: A common shape for agents that live alongside people.
 Project-URL: Homepage, https://bub.build
 Project-URL: Repository, https://github.com/bubbuild/bub
 Project-URL: Documentation, https://bub.build
 Author-email: Chojan Shang <psiace@apache.org>, Frost Ming <me@frostming.com>, Yihong <zouzou0208@gmail.com>
+License: Apache-2.0
 License-File: LICENSE
 Classifier: Intended Audience :: Developers
 Classifier: Programming Language :: Python
@@ -154,7 +155,7 @@ Lines starting with `,` enter internal command mode (`,help`, `,skill name=my-sk
 | `BUB_API_BASE`              | —                            | Custom provider endpoint                             |
 | `BUB_CLIENT_ARGS`           | —                            | JSON object forwarded to the underlying model client |
 | `BUB_COMPLETION_ARGS`       | —                            | JSON object forwarded to each completion call         |
-| `BUB_MAX_STEPS`             | `50`                         | Max tool-use loop iterations                         |
+| `BUB_MAX_STEPS`             | unlimited                    | Tool-use loop limit; must be a positive integer      |
 | `BUB_MAX_TOKENS`            | `16384`                      | Max tokens per model call                            |
 | `BUB_MODEL_TIMEOUT_SECONDS` | —                            | Model call timeout (seconds)                         |
 

{bub-0.4.2 → bub-0.4.3}/README.md

@@ -117,7 +117,7 @@ Lines starting with `,` enter internal command mode (`,help`, `,skill name=my-sk
 | `BUB_API_BASE`              | —                            | Custom provider endpoint                             |
 | `BUB_CLIENT_ARGS`           | —                            | JSON object forwarded to the underlying model client |
 | `BUB_COMPLETION_ARGS`       | —                            | JSON object forwarded to each completion call         |
-| `BUB_MAX_STEPS`             | `50`                         | Max tool-use loop iterations                         |
+| `BUB_MAX_STEPS`             | unlimited                    | Tool-use loop limit; must be a positive integer      |
 | `BUB_MAX_TOKENS`            | `16384`                      | Max tokens per model call                            |
 | `BUB_MODEL_TIMEOUT_SECONDS` | —                            | Model call timeout (seconds)                         |
 

{bub-0.4.2 → bub-0.4.3}/pyproject.toml

@@ -9,6 +9,7 @@ authors = [
 ]
 readme = "README.md"
 requires-python = ">=3.12,<4.0"
+license = { text = "Apache-2.0" }
 classifiers = [
     "Intended Audience :: Developers",
     "Programming Language :: Python",

{bub-0.4.2 → bub-0.4.3}/src/bub/_version.py

@@ -18,7 +18,7 @@ version_tuple: tuple[int | str, ...]
 commit_id: str | None
 __commit_id__: str | None
 
-__version__ = version = '0.4.2'
-__version_tuple__ = version_tuple = (0, 4, 2)
+__version__ = version = '0.4.3'
+__version_tuple__ = version_tuple = (0, 4, 3)
 
 __commit_id__ = commit_id = None

{bub-0.4.2 → bub-0.4.3}/src/bub/builtin/agent.py

@@ -22,12 +22,12 @@ from bub.builtin.model_runner import (
     is_context_length_error,
 )
 from bub.builtin.settings import load_settings
-from bub.builtin.tape import Tape
 from bub.envelope import field_of
 from bub.framework import BubFramework
 from bub.skills import discover_skills, render_skills_prompt
+from bub.store import AsyncTapeStoreAdapter, InMemoryTapeStore, is_async_tape_store
 from bub.streaming import AsyncStreamEvents, StreamEvent, StreamState
-from bub.tape import AsyncTapeStoreAdapter, InMemoryTapeStore, is_async_tape_store
+from bub.tape import Tape
 from bub.tools import (
     REGISTRY,
     Tool,
@@ -37,7 +37,6 @@ from bub.tools import (
 from bub.turn import TurnState
 from bub.utils import workspace_from_state
 
-CONTINUE_PROMPT = "Continue the task until all targets are completed."
 HINT_RE = re.compile(r"\$([A-Za-z0-9_.-]+)")
 MAX_AUTO_HANDOFF_RETRIES = 1
 
@@ -291,7 +290,7 @@ class Agent:
                 )
                 return
 
-            next_prompt = self._continue_prompt(tape)
+            next_prompt = await self.framework.continue_prompt(prompt=next_prompt, tape=tape, state=state)
             await tape.append_event(
                 "loop.step",
                 {
@@ -398,11 +397,6 @@ class Agent:
             blocks.append(skills_prompt)
         return "\n\n".join(blocks)
 
-    def _continue_prompt(self, tape: Tape) -> str:
-        if "context" in tape.context.state:
-            return f"{CONTINUE_PROMPT} [context: {tape.context.state['context']}]"
-        return CONTINUE_PROMPT
-
     def _has_steering_messages(self, state: TurnState) -> bool:
         steering_inbox = self.framework.get_steering_inbox()
         return bool(steering_inbox and steering_inbox.message_count(state) > 0)

{bub-0.4.2 → bub-0.4.3}/src/bub/builtin/hook_impl.py

@@ -1,5 +1,6 @@
 import sys
 from datetime import datetime
+from difflib import get_close_matches
 from pathlib import Path
 from typing import cast
 
@@ -18,9 +19,11 @@ from bub.channels.message import ChannelMessage, MediaItem
 from bub.envelope import Envelope, content_of, field_of
 from bub.framework import BubFramework
 from bub.hooks import hookimpl
+from bub.hooks.interception import ToolCall, ToolCallDecision
 from bub.model_selection import ModelChoice, ModelOptions
-from bub.streaming import AsyncStreamEvents
-from bub.tape import TapeContext, TapeStore
+from bub.store import TapeStore
+from bub.streaming import AsyncStreamEvents, StreamState
+from bub.tape import Tape, TapeContext
 from bub.turn import TurnState
 
 AGENTS_FILE_NAME = "AGENTS.md"
@@ -57,6 +60,7 @@ When responding to a channel message, you MUST:
 Excessively long context may cause model call failures. In this case, you MAY use tape.info to retrieve the token usage and you SHOULD use tape.handoff tool to shorten the retrieved history.
 </context_contract>
 """
+DEFAULT_CONTINUE_PROMPT = "Continue the task until all targets are completed."
 
 
 class BuiltinImpl:
@@ -214,6 +218,13 @@ class BuiltinImpl:
             model=state.get("model"),
         )
 
+    @hookimpl
+    def continue_prompt(self, prompt: str | list[dict], tape: Tape, state: StreamState) -> str:
+        del prompt, state
+        if "context" in tape.context.state:
+            return f"{DEFAULT_CONTINUE_PROMPT} [context: {tape.context.state['context']}]"
+        return DEFAULT_CONTINUE_PROMPT
+
     @hookimpl
     def register_cli_commands(self, app: typer.Typer) -> None:
         from bub.builtin import cli
@@ -355,7 +366,7 @@ class BuiltinImpl:
     @hookimpl
     def provide_tape_store(self) -> TapeStore:
         import bub
-        from bub.builtin.store import FileTapeStore
+        from bub.store import FileTapeStore
 
         return FileTapeStore(directory=bub.home / "tapes")
 
@@ -378,3 +389,31 @@ class BuiltinImpl:
         if channel_router is None:
             return None
         return await channel_router.admit_channel_message(session_id=session_id, message=message, turn=turn)
+
+    @hookimpl
+    async def before_tool_call(
+        self,
+        call: ToolCall,
+        state: TurnState,
+    ) -> ToolCallDecision | None:
+        """Recover hallucinated/unknown tool names without interrupting the turn.
+
+        When the model invokes a tool outside the current model-facing tool set,
+        replace it with a guidance ``tool_result`` so the model can re-issue a
+        valid call on the next step.
+        """
+        from bub.tools import REGISTRY, model_tools
+
+        available_tools = tuple(tool_item.name for tool_item in model_tools(REGISTRY.values()))
+        if call.tool in available_tools:
+            return None
+
+        matches = get_close_matches(call.tool, available_tools, n=3, cutoff=0.6)
+        if matches:
+            suggestions = "\n".join(f"- {name}" for name in matches)
+            guidance = f"Tool `{call.tool}` does not exist. Did you mean one of the following?\n{suggestions}"
+        elif "skill" in available_tools:
+            guidance = f"Tool `{call.tool}` does not exist. Invoke the `skill` tool to list available skills."
+        else:
+            guidance = f"Tool `{call.tool}` does not exist. No similar tool is available."
+        return ToolCallDecision.replace(guidance)

{bub-0.4.2 → bub-0.4.3}/src/bub/builtin/model_runner.py

@@ -28,7 +28,6 @@ from pydantic import TypeAdapter, ValidationError
 
 from bub.builtin.codex_provider import OpenaiCodexProvider, should_use_openai_codex_provider
 from bub.builtin.settings import AgentSettings, ModelCandidate
-from bub.builtin.tape import Tape
 from bub.errors import BubError, ErrorKind
 from bub.hooks.interception import (
     AgentHooks,
@@ -37,6 +36,7 @@ from bub.hooks.interception import (
     LlmCallResult,
 )
 from bub.streaming import AsyncStreamEvents, StreamEvent, StreamState
+from bub.tape import Tape
 from bub.tools import Tool, ToolContext, ToolExecutor
 
 CONTEXT_LENGTH_PATTERNS = re.compile(
@@ -461,10 +461,23 @@ def tool_invocation_from_native(
     tool_call: ChatCompletionMessageToolCall,
     tool_map: dict[str, Tool],
 ) -> tuple[Tool, dict[str, Any]]:
+    """Resolve a model tool call to (runtime tool, arguments).
+
+    An unknown tool name is not treated as a fatal error: it is surfaced as a
+    placeholder ``Tool`` so the invocation flows through ``ToolExecutor`` and
+    builtin hooks (e.g. ``before_tool_call``) can recover it into a guidance
+    ``tool_result`` instead of interrupting the turn. If no hook replaces the
+    call, the placeholder raises a clear tool error rather than succeeding with
+    an empty result.
+    """
     tool_name, arguments = parse_native_function_call(tool_call)
     tool_obj = tool_map.get(tool_name)
     if tool_obj is None:
-        raise BubError(ErrorKind.TOOL, f"Unknown tool name: {tool_name}.")
+
+        def raise_unknown_tool(**_: Any) -> None:
+            raise BubError(ErrorKind.TOOL, f"Unknown tool name: {tool_name}.")
+
+        return Tool(name=tool_name, handler=raise_unknown_tool), arguments
     return tool_obj, arguments
 
 

{bub-0.4.2 → bub-0.4.3}/src/bub/builtin/settings.py

@@ -3,6 +3,7 @@ from __future__ import annotations
 import os
 import pathlib
 import re
+import sys
 from dataclasses import dataclass
 from typing import Any
 
@@ -56,7 +57,7 @@ class AgentSettings(Settings):
     fallback_models: list[str] | None = None
     api_key: str | dict[str, str] | None = None
     api_base: str | dict[str, str] | None = None
-    max_steps: int = 50
+    max_steps: int = Field(default=sys.maxsize, gt=0)
     max_tokens: int = DEFAULT_MAX_TOKENS
     model_timeout_seconds: int | None = None
     client_args: dict[str, Any] = Field(default_factory=dict)

{bub-0.4.2 → bub-0.4.3}/src/bub/channels/cli/__init__.py

@@ -25,7 +25,6 @@ from rich.tree import Tree
 
 import bub
 from bub.builtin.agent import Agent
-from bub.builtin.tape import TapeInfo
 from bub.channels.admission import AdmitDecision, TurnSnapshot
 from bub.channels.base import Interface
 from bub.channels.cli.ansi_bridge import render_to_ansi
@@ -39,6 +38,7 @@ from bub.channels.contracts import MessageHandler
 from bub.channels.message import ChannelMessage
 from bub.envelope import Envelope, field_of
 from bub.streaming import StreamEvent
+from bub.tape import TapeInfo
 from bub.tools import REGISTRY, tool_call_reporter
 
 _GENERATION_SPINNER: str = SPINNERS["dots"]["frames"]  # type: ignore[assignment]
@@ -54,10 +54,12 @@ class _StreamPrinter:
         expand_thinking: bool,
         presenter: TerminalPresenter,
         writer: MarkdownWriter | None = None,
+        print_end: Callable[[], None] | None = None,
         invalidate: Callable[[], None] | None = None,
     ) -> None:
         self._console = console
         self._print_head = print_head
+        self._print_end = print_end or (lambda: None)
         self._expand_thinking = expand_thinking
         self._presenter = presenter
         self._reasoning_chars = 0
@@ -82,13 +84,10 @@ class _StreamPrinter:
 
     async def _record_reasoning(self, reasoning: str) -> None:
         if not self._expand_thinking:
-            if self._reasoning_chars == 0:
-                await self._ensure_head()
             self._reasoning_chars += len(reasoning)
             self._invalidate()
             return
 
-        await self._ensure_head()
         if not self._reasoning_streaming:
             await self._print(Text("[-] Thinking", style="dim"))
             self._reasoning_streaming = True
@@ -97,19 +96,21 @@ class _StreamPrinter:
     async def _print_content(self, content: str) -> bool:
         if not (content.strip() or self.head_printed or self._reasoning_chars or self._reasoning_streaming):
             return False
-        await self._ensure_head()
         await self._close_reasoning_stream()
         await self._flush_reasoning()
+        if content.strip() or self.head_printed:
+            await self._ensure_head()
         await self._write_text(content)
         return True
 
     async def finish(self) -> None:
         await self._close_reasoning_stream()
-        if self._reasoning_chars:
-            await self._ensure_head()
         await self._flush_reasoning()
         if self._writer.has_content():
+            await self._ensure_head()
             await self._flush_text()
+        else:
+            await self._ensure_panel_end()
 
     async def _print_stream_boundary(self) -> None:
         await self.finish()
@@ -122,6 +123,12 @@ class _StreamPrinter:
         await self._presenter.write(self._print_head)
         self.head_printed = True
 
+    async def _ensure_panel_end(self) -> None:
+        if not self.head_printed:
+            return
+        await self._presenter.write(self._print_end)
+        self.head_printed = False
+
     async def _close_reasoning_stream(self) -> None:
         if not self._reasoning_streaming:
             return
@@ -163,6 +170,8 @@ class _StreamPrinter:
         def commit() -> None:
             self._console.print(finished)
             self._writer.clear()
+            self._print_end()
+            self.head_printed = False
 
         await self._presenter.write(commit)
         self._ansi_cache = None
@@ -353,6 +362,7 @@ class CliChannel(Interface):
         printer = _StreamPrinter(
             console=console,
             print_head=lambda: self._renderer.print_head(message.kind),
+            print_end=lambda: self._renderer.print_end(message.kind),
             expand_thinking=self._expand_thinking,
             presenter=self._presenter,
             invalidate=self._invalidate_prompt,

{bub-0.4.2 → bub-0.4.3}/src/bub/channels/cli/renderer.py

@@ -10,6 +10,7 @@ from rich.console import Console
 from rich.panel import Panel
 from rich.text import Text
 
+from bub.channels.cli.writers import PanelEnd, PanelHead
 from bub.channels.message import MessageKind
 
 MAX_TOOL_PAYLOAD_CHARS = 4000
@@ -37,20 +38,10 @@ class CliRenderer:
             return
         self.console.print(Text(text, style="bright_black"))
 
-    def command_output(self, text: str) -> None:
-        if not text.strip():
-            return
-        self.console.print(f"[cyan bold]Command >[/]\n{text}")
-
-    def assistant_output(self, text: str) -> None:
-        if not text.strip():
-            return
-        self.console.print(f"[blue bold]Assistant >[/]\n{text}")
-
     def error(self, text: str) -> None:
         if not text.strip():
             return
-        self.console.print(f"[red bold]Error >[/]\n{text}")
+        self.console.print(Panel(text, title="error", border_style="red"))
 
     def input_echo(self, prompt: str, text: str, steering: bool = False) -> None:
         if not text.strip():
@@ -62,8 +53,7 @@ class CliRenderer:
         self.console.print(Text(_format_tool_call(name, args, kwargs), style="magenta"), new_line_start=True)
 
     def tool_call_success(self, *, name: str, result: Any, elapsed_ms: float) -> None:
-        rendered = _format_tool_payload(result)
-        self._tool_result(f"completed in {elapsed_ms:.0f} ms", rendered, style="green")
+        self._tool_status(f"completed in {elapsed_ms:.0f} ms", style="green")
 
     def tool_call_error(self, *, name: str, error: BaseException, elapsed_ms: float) -> None:
         rendered = _format_tool_payload({"type": error.__class__.__name__, "message": str(error)})
@@ -71,11 +61,19 @@ class CliRenderer:
 
     def print_head(self, kind: MessageKind) -> None:
         if kind == "command":
-            self.console.print("[cyan bold]Command >[/]", new_line_start=True)
+            self.console.print(PanelHead("Command", border_style="cyan"))
         elif kind == "error":
-            self.console.print("[red bold]Error >[/]", new_line_start=True)
+            self.console.print(PanelHead("Error", border_style="red"))
         else:
-            self.console.print("[blue bold]Assistant >[/]", new_line_start=True)
+            self.console.print(PanelHead("Assistant", border_style="blue"))
+
+    def print_end(self, kind: MessageKind) -> None:
+        if kind == "command":
+            self.console.print(PanelEnd(border_style="cyan"))
+        elif kind == "error":
+            self.console.print(PanelEnd(border_style="red"))
+        else:
+            self.console.print(PanelEnd(border_style="blue"))
 
     def log(self, message: object) -> None:
         text = str(message).rstrip()
@@ -83,11 +81,13 @@ class CliRenderer:
             self.console.print(text, new_line_start=True)
 
     def _tool_result(self, label: str, rendered: str, *, style: str) -> None:
-        lines = rendered.splitlines() or [""]
-        self.console.print(Text(f"  ⎿ {label}", style=style), highlight=False)
-        for line in lines:
+        self._tool_status(label, style=style)
+        for line in rendered.splitlines() or [""]:
             self.console.print(Text(f"    {line}", style="bright_black"), highlight=False)
 
+    def _tool_status(self, label: str, *, style: str) -> None:
+        self.console.print(Text(f"  ⎿ {label}", style=style), highlight=False)
+
 
 def _format_tool_call(name: str, args: tuple[Any, ...], kwargs: dict[str, Any]) -> str:
     params = _format_tool_params(args, kwargs)

bub-0.4.3/src/bub/channels/cli/writers.py

@@ -0,0 +1,76 @@
+"""Response-scoped Markdown buffering for CLI streaming output."""
+
+from __future__ import annotations
+
+from rich.console import Console, ConsoleOptions, RenderResult
+from rich.markdown import Markdown
+from rich.panel import Panel
+from rich.segment import Segment
+from rich.text import Text
+
+_MARKDOWN_CODE_THEME = "ansi_dark"
+
+
+def _markdown(content: str) -> Markdown:
+    return Markdown(content, code_theme=_MARKDOWN_CODE_THEME)
+
+
+class PanelHead:
+    """The top border of a Panel, printed before its content is streamed."""
+
+    def __init__(self, title: str, *, border_style: str, width: int | None = None) -> None:
+        self._title = title
+        self._border_style = border_style
+        self._width = width
+
+    def __rich_console__(self, console: Console, options: ConsoleOptions) -> RenderResult:
+        width = self._width if self._width is not None else options.max_width
+        panel = Panel("", title=self._title, border_style=self._border_style, width=width)
+        for line in console.render_lines(panel, options.update(width=width), pad=False)[:1]:
+            yield from line
+        yield Segment.line()
+
+
+class PanelEnd:
+    """The bottom border of a Panel, printed after its content is streamed."""
+
+    def __init__(self, *, border_style: str, width: int | None = None) -> None:
+        self._border_style = border_style
+        self._width = width
+
+    def __rich_console__(self, console: Console, options: ConsoleOptions) -> RenderResult:
+        width = self._width if self._width is not None else options.max_width
+        panel = Panel("", border_style=self._border_style, width=width)
+        for line in console.render_lines(panel, options.update(width=width), pad=False)[-1:]:
+            yield from line
+        yield Segment.line()
+
+
+class MarkdownWriter:
+    """Keep one response segment as a single Markdown document.
+
+    Newlines are Markdown structure, not terminal commit boundaries. The
+    buffer is drained only at an explicit model or tool boundary.
+    """
+
+    def __init__(self) -> None:
+        self._buffer = ""
+
+    def append(self, text: str) -> None:
+        self._buffer += text
+
+    def render_live(self) -> Markdown | Text:
+        if not self._buffer.strip():
+            return Text("")
+        return _markdown(self._buffer)
+
+    def render_final(self) -> Markdown | None:
+        if not self._buffer.strip():
+            return None
+        return _markdown(self._buffer.rstrip())
+
+    def clear(self) -> None:
+        self._buffer = ""
+
+    def has_content(self) -> bool:
+        return bool(self._buffer.strip())

{bub-0.4.2 → bub-0.4.3}/src/bub/framework.py

@@ -22,7 +22,9 @@ from bub.hooks.interception import AgentHooks
 from bub.hooks.runtime import _SKIP_VALUE, HookRuntime
 from bub.hooks.specs import BUB_HOOK_NAMESPACE, BubHookSpecs
 from bub.model_selection import ModelOptions
-from bub.tape import AsyncTapeStore, TapeContext, TapeStore
+from bub.store import AsyncTapeStore, TapeStore
+from bub.streaming import StreamState
+from bub.tape import Tape, TapeContext
 from bub.turn import TurnResult, TurnState
 from bub.utils import maybe_context_manager
 
@@ -122,6 +124,13 @@ class BubFramework:
             prompt = content_of(message)
         return cast("str | list[dict[str, Any]]", prompt)
 
+    async def continue_prompt(self, prompt: str | list[dict], tape: Tape, state: StreamState) -> str:
+        """Build the prompt for the next step of an agent loop."""
+        next_prompt = await self._hook_runtime.call_first("continue_prompt", prompt=prompt, tape=tape, state=state)
+        if isinstance(next_prompt, str):
+            return next_prompt
+        raise TypeError("hook.continue_prompt must return str")
+
     async def build_state(self, message: Envelope, session_id: str) -> TurnState:
         state = {"_runtime_workspace": str(self.workspace), "_runtime_steering_inbox": self.get_steering_inbox()}
         for hook_state in reversed(

{bub-0.4.2 → bub-0.4.3}/src/bub/hooks/specs.py

@@ -19,8 +19,9 @@ from bub.hooks.interception import (
     ToolCallResult,
 )
 from bub.model_selection import ModelOptions
-from bub.streaming import AsyncStreamEvents
-from bub.tape import AsyncTapeStore, TapeContext, TapeStore
+from bub.store import AsyncTapeStore, TapeStore
+from bub.streaming import AsyncStreamEvents, StreamState
+from bub.tape import Tape, TapeContext
 from bub.turn import TurnState
 
 if TYPE_CHECKING:
@@ -64,6 +65,15 @@ class BubHookSpecs:
         """
         raise NotImplementedError
 
+    @hookspec(firstresult=True)
+    def continue_prompt(self, prompt: str | list[dict], tape: Tape, state: StreamState) -> str:
+        """Build the prompt used to continue an agent loop.
+
+        Implementations may be synchronous or asynchronous. The first
+        non-``None`` result in hook priority order is used.
+        """
+        raise NotImplementedError
+
     @hookspec
     def load_state(self, message: Envelope, session_id: str) -> TurnState:
         """Load state snapshot for one session."""