drydock-cli 2.6.129__tar.gz → 2.6.131__tar.gz

{drydock_cli-2.6.129 → drydock_cli-2.6.131}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: drydock-cli
-Version: 2.6.129
+Version: 2.6.131
 Summary: Local-first CLI coding agent — tested with Gemma 4 26B via vLLM
 Project-URL: Homepage, https://github.com/fbobe321/drydock
 Project-URL: Repository, https://github.com/fbobe321/drydock

{drydock_cli-2.6.129 → drydock_cli-2.6.131}/drydock/cli/textual_ui/widgets/messages.py

@@ -163,7 +163,24 @@ class AssistantMessage(StreamingMessageBase):
         self.add_class("assistant-message")
 
     async def append_content(self, content: str) -> None:
-        await super().append_content(_preserve_line_breaks(content))
+        # _preserve_line_breaks doubles \n → \n\n WITHIN a chunk. Streaming
+        # chunks routinely split on the newline boundary ("abc\n" then
+        # "def"), which that function can't upgrade — the single \n stays
+        # and markdown renders it as a space, collapsing lists and
+        # paragraphs. We bridge the gap here by checking the join point:
+        # if the accumulated content ends with a single \n and the new
+        # chunk starts with non-whitespace, inject an extra \n so markdown
+        # sees the paragraph break. Handles issue #5 (v2.6.130).
+        prepared = _preserve_line_breaks(content)
+        if (
+            prepared
+            and self._content.endswith("\n")
+            and not self._content.endswith("\n\n")
+            and not self._content.endswith("\t\n")
+            and prepared[0] not in ("\n", " ", "\t")
+        ):
+            prepared = "\n" + prepared
+        await super().append_content(prepared)
 
     def compose(self) -> ComposeResult:
         if self._content:

{drydock_cli-2.6.129 → drydock_cli-2.6.131}/drydock/core/llm/format.py

@@ -228,6 +228,12 @@ class APIToolFormatHandler:
             # Strip leading "thought"/"(thought)" filler and match the
             # `call:name{...}` shape so the agent loop sees empty content
             # and can fire its recovery nudge.
+            #
+            # IMPORTANT: `stripped` is for PATTERN MATCHING ONLY. We must
+            # NOT replace `content` with `stripped` in the no-match case —
+            # that ate leading/trailing spaces from every streaming chunk
+            # (issue #1: a Gemma chunk like " received" became "received"
+            # so the rendered text had no spaces between words).
             stripped = content.strip()
             # Peel "thought" or "(thought)" prefix (case-insensitive, optional
             # whitespace)
@@ -250,7 +256,12 @@ class APIToolFormatHandler:
                 # this..." — only the bare-word leak with `/` or `\n`.
                 content = None
             else:
-                content = stripped or None
+                # Preserve original whitespace. Only collapse to None when
+                # the chunk is genuinely empty after stripping (so we don't
+                # ship pure-whitespace chunks to downstream consumers).
+                if not stripped:
+                    content = None
+                # else: leave `content` as-is (with leading/trailing spaces)
 
         clean_message = {
             "role": message.role,

{drydock_cli-2.6.129 → drydock_cli-2.6.131}/drydock/core/tools/builtins/bash.py

@@ -370,7 +370,19 @@ class Bash(
             return False
 
         def is_allowlisted(command: str) -> bool:
-            return any(command.startswith(pattern) for pattern in self.config.allowlist)
+            # Direct prefix match first ("find . -name x" vs "find").
+            if any(command.startswith(pattern) for pattern in self.config.allowlist):
+                return True
+            # Also match on basename so "/usr/bin/find ..." or "./grep ..."
+            # auto-approve alongside bare "find" / "grep". Without this,
+            # the model invoking a fully-qualified path would drop to the
+            # approval prompt unnecessarily (GitHub issue #6).
+            parts = command.split(None, 1)
+            if not parts:
+                return False
+            base = os.path.basename(parts[0])
+            rest = f"{base} {parts[1]}" if len(parts) > 1 else base
+            return any(rest.startswith(pattern) for pattern in self.config.allowlist)
 
         for part in command_parts:
             if is_denylisted(part):

{drydock_cli-2.6.129 → drydock_cli-2.6.131}/drydock/core/tools/builtins/todo.py

@@ -4,7 +4,7 @@ from collections.abc import AsyncGenerator
 from enum import StrEnum, auto
 from typing import Any, ClassVar
 
-from pydantic import BaseModel, Field, field_validator
+from pydantic import BaseModel, Field, field_validator, model_validator
 
 from drydock.core.tools.base import (
     BaseTool,
@@ -92,6 +92,19 @@ class TodoArgs(BaseModel):
             return result
         return v
 
+    @model_validator(mode="after")
+    def _infer_write_action(self) -> "TodoArgs":
+        """Gemma 4 often calls todo with a todos list but forgets
+        action='write' (leaving it at the default 'read'). Drydock
+        would then silently do a read, return 0 todos, and the model
+        would loop trying to write and read empty. If todos is
+        populated, treat that as a write — the caller's intent is
+        clear. See GitHub issue #4.
+        """
+        if self.todos and self.action == "read":
+            object.__setattr__(self, "action", "write")
+        return self
+
 
 class TodoResult(BaseModel):
     message: str

{drydock_cli-2.6.129 → drydock_cli-2.6.131}/drydock/setup/onboarding/__init__.py

@@ -6,7 +6,12 @@ from rich import print as rprint
 from textual.app import App
 
 from drydock.core.paths import GLOBAL_ENV_FILE
-from drydock.setup.onboarding.screens import ApiKeyScreen, WelcomeScreen
+from drydock.setup.onboarding.screens import (
+    ApiKeyScreen,
+    ChoiceScreen,
+    LocalModelScreen,
+    WelcomeScreen,
+)
 
 
 class OnboardingApp(App[str | None]):
@@ -16,6 +21,8 @@ class OnboardingApp(App[str | None]):
         self.theme = "textual-ansi"
 
         self.install_screen(WelcomeScreen(), "welcome")
+        self.install_screen(ChoiceScreen(), "choice")
+        self.install_screen(LocalModelScreen(), "local_model")
         self.install_screen(ApiKeyScreen(), "api_key")
         self.push_screen("welcome")
 

drydock_cli-2.6.131/drydock/setup/onboarding/screens/__init__.py

@@ -0,0 +1,13 @@
+from __future__ import annotations
+
+from drydock.setup.onboarding.screens.api_key import ApiKeyScreen
+from drydock.setup.onboarding.screens.choice import ChoiceScreen
+from drydock.setup.onboarding.screens.local_model import LocalModelScreen
+from drydock.setup.onboarding.screens.welcome import WelcomeScreen
+
+__all__ = [
+    "ApiKeyScreen",
+    "ChoiceScreen",
+    "LocalModelScreen",
+    "WelcomeScreen",
+]

{drydock_cli-2.6.129 → drydock_cli-2.6.131}/drydock/setup/onboarding/screens/api_key.py

@@ -19,10 +19,10 @@ from drydock.core.telemetry.send import TelemetryClient
 from drydock.setup.onboarding.base import OnboardingScreen
 
 PROVIDER_HELP = {
-    "mistral": ("https://console.mistral.ai/codestral/cli", "Mistral AI Studio")
+    "mistral": ("https://console.mistral.ai/codestral/cli", "Mistral AI Studio"),
 }
 CONFIG_DOCS_URL = (
-    "https://github.com/mistralai/mistral-vibe?tab=readme-ov-file#configuration"
+    "https://github.com/fbobe321/drydock?tab=readme-ov-file#configuration"
 )
 
 

drydock_cli-2.6.131/drydock/setup/onboarding/screens/choice.py

@@ -0,0 +1,65 @@
+"""Cloud-vs-local choice screen — step 2 of onboarding.
+
+Asks the user which kind of LLM they're connecting to. Answers github
+issue #3: a fresh install used to drop straight into a Mistral API key
+prompt with no obvious way to use a local vLLM instead. Now the user
+sees the choice up front.
+"""
+from __future__ import annotations
+
+from typing import ClassVar
+
+from textual.app import ComposeResult
+from textual.binding import Binding, BindingType
+from textual.containers import Center, Vertical
+from textual.widgets import OptionList
+from textual.widgets.option_list import Option
+
+from drydock.cli.textual_ui.widgets.no_markup_static import NoMarkupStatic
+from drydock.setup.onboarding.base import OnboardingScreen
+
+
+class ChoiceScreen(OnboardingScreen):
+    BINDINGS: ClassVar[list[BindingType]] = [
+        Binding("ctrl+c", "cancel", "Cancel", show=False),
+        Binding("escape", "cancel", "Cancel", show=False),
+    ]
+
+    NEXT_SCREEN = None  # set dynamically based on selection
+
+    def compose(self) -> ComposeResult:
+        with Vertical(id="choice-outer"):
+            yield NoMarkupStatic("", classes="spacer")
+            yield Center(NoMarkupStatic(
+                "How will you connect to an LLM?",
+                id="choice-title",
+            ))
+            with Center():
+                with Vertical(id="choice-content"):
+                    yield OptionList(
+                        Option(
+                            "Local server (vLLM, Ollama, LM Studio, llama.cpp)",
+                            id="local",
+                        ),
+                        Option(
+                            "Cloud API (Mistral)",
+                            id="cloud",
+                        ),
+                        id="choice-options",
+                    )
+                    yield NoMarkupStatic(
+                        "[dim]↑↓ Navigate · Enter Select · Esc Cancel[/]",
+                        id="choice-help",
+                    )
+            yield NoMarkupStatic("", classes="spacer")
+
+    def on_mount(self) -> None:
+        self.query_one(OptionList).focus()
+
+    def on_option_list_option_selected(
+        self, event: OptionList.OptionSelected
+    ) -> None:
+        if event.option.id == "local":
+            self.app.switch_screen("local_model")
+        elif event.option.id == "cloud":
+            self.app.switch_screen("api_key")

drydock_cli-2.6.131/drydock/setup/onboarding/screens/local_model.py

@@ -0,0 +1,138 @@
+"""Local-LLM setup screen — alternative to the Mistral API key path.
+
+Asks the user for an api_base URL and a model name, writes a minimal
+config.toml pointing at it, and exits onboarding marking it complete.
+The user can refine later via `/setup-model` or by editing config.toml
+directly. This is the "I'll run my own vLLM/Ollama/LM Studio" path
+described in github issue #3.
+"""
+from __future__ import annotations
+
+from pathlib import Path
+from typing import ClassVar
+
+import tomli_w
+from textual.app import ComposeResult
+from textual.binding import Binding, BindingType
+from textual.containers import Center, Horizontal, Vertical
+from textual.validation import Length, Regex
+from textual.widgets import Input
+
+from drydock.cli.textual_ui.widgets.no_markup_static import NoMarkupStatic
+from drydock.setup.onboarding.base import OnboardingScreen
+
+
+_DEFAULT_API_BASE = "http://localhost:8000/v1"
+_DEFAULT_MODEL = "local"
+
+
+def _write_local_config(api_base: str, model_name: str) -> Path:
+    """Write a minimal local-model config to ~/.drydock/config.toml.
+
+    Adds a 'local' provider pointing at api_base, a 'local' model alias
+    pointing at the user's model name, and sets it as the active model.
+    Doesn't touch the existing Mistral defaults — the user can still
+    add a Mistral key later if they want.
+    """
+    config_dir = Path.home() / ".drydock"
+    config_dir.mkdir(parents=True, exist_ok=True)
+    cfg_path = config_dir / "config.toml"
+    payload = {
+        "active_model": model_name or _DEFAULT_MODEL,
+        "providers": [
+            {
+                "name": "local",
+                "api_base": api_base,
+                "api_key_env_var": "",
+                "backend": "generic",
+            },
+        ],
+        "models": [
+            {
+                "name": model_name or _DEFAULT_MODEL,
+                "provider": "local",
+                "alias": model_name or _DEFAULT_MODEL,
+                "input_price": 0.0,
+                "output_price": 0.0,
+            },
+        ],
+    }
+    with cfg_path.open("wb") as f:
+        tomli_w.dump(payload, f)
+    return cfg_path
+
+
+class LocalModelScreen(OnboardingScreen):
+    """Step-2 alternative: configure a local OpenAI-compatible endpoint."""
+
+    BINDINGS: ClassVar[list[BindingType]] = [
+        Binding("ctrl+c", "cancel", "Cancel", show=False),
+        Binding("escape", "cancel", "Cancel", show=False),
+    ]
+
+    NEXT_SCREEN = None
+
+    def compose(self) -> ComposeResult:
+        self.api_base_input = Input(
+            value=_DEFAULT_API_BASE,
+            id="api-base",
+            placeholder=_DEFAULT_API_BASE,
+            validators=[
+                Length(minimum=1, failure_description="API base URL required."),
+                Regex(
+                    r"^https?://.+",
+                    failure_description="Must start with http:// or https://",
+                ),
+            ],
+        )
+        self.model_input = Input(
+            value="",
+            id="model-name",
+            placeholder="model name (e.g. gemma4, llama-3-8b)",
+            validators=[Length(minimum=1, failure_description="Model name required.")],
+        )
+        with Vertical(id="local-model-outer"):
+            yield NoMarkupStatic("", classes="spacer")
+            yield Center(NoMarkupStatic(
+                "Local LLM setup",
+                id="local-model-title",
+            ))
+            with Center():
+                with Vertical(id="local-model-content"):
+                    yield NoMarkupStatic(
+                        "Point Drydock at any OpenAI-compatible endpoint "
+                        "(vLLM, Ollama, LM Studio, llama.cpp, etc.):",
+                        id="local-model-hint",
+                    )
+                    yield NoMarkupStatic("API base URL:")
+                    yield Center(Horizontal(self.api_base_input, id="api-base-row"))
+                    yield NoMarkupStatic("Model name (as your server reports it):")
+                    yield Center(Horizontal(self.model_input, id="model-row"))
+                    yield NoMarkupStatic("", id="feedback")
+                    yield NoMarkupStatic(
+                        "[dim]Tab to switch fields · Enter on the model "
+                        "field to save · Esc to skip[/]",
+                        id="local-model-help",
+                    )
+            yield NoMarkupStatic("", classes="spacer")
+
+    def on_mount(self) -> None:
+        self.api_base_input.focus()
+
+    def on_input_submitted(self, event: Input.Submitted) -> None:
+        # Enter on api-base → jump to model field; Enter on model → save.
+        if event.input.id == "api-base":
+            self.model_input.focus()
+            return
+        if event.input.id == "model-name":
+            self._save_and_finish()
+
+    def _save_and_finish(self) -> None:
+        api_base = self.api_base_input.value.strip() or _DEFAULT_API_BASE
+        model_name = self.model_input.value.strip() or _DEFAULT_MODEL
+        try:
+            _write_local_config(api_base, model_name)
+        except OSError as err:
+            self.app.exit(f"save_error:{err}")
+            return
+        self.app.exit("completed")

{drydock_cli-2.6.129 → drydock_cli-2.6.131}/drydock/setup/onboarding/screens/welcome.py

@@ -50,7 +50,7 @@ class WelcomeScreen(OnboardingScreen):
         Binding("escape", "cancel", "Cancel", show=False),
     ]
 
-    NEXT_SCREEN = "api_key"
+    NEXT_SCREEN = "choice"
 
     def __init__(self) -> None:
         super().__init__()

{drydock_cli-2.6.129 → drydock_cli-2.6.131}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "drydock-cli"
-version = "2.6.129"
+version = "2.6.131"
 description = "Local-first CLI coding agent — tested with Gemma 4 26B via vLLM"
 readme = "README.md"
 requires-python = ">=3.12"

drydock_cli-2.6.131/scripts/poll_issues.sh

@@ -0,0 +1,102 @@
+#!/bin/bash
+# Hourly poller: notifies via Telegram about NEW open issues opened by
+# fbobe321 in fbobe321/drydock since the last poll. Runs Mon-Fri 7am-5pm
+# via cron. State pointer at ~/.drydock/last_issue_poll keeps track of
+# which issues we've already alerted on so we don't re-notify.
+#
+# Cron line (added separately to crontab):
+#   0 7-17 * * 1-5  /data3/drydock/scripts/poll_issues.sh >> /data3/drydock/logs/poll_issues.log 2>&1
+set -euo pipefail
+
+PYTHON="/home/bobef/miniconda3/bin/python3"
+TOKEN_FILE="$HOME/.config/drydock/github_token"
+STATE_FILE="$HOME/.drydock/last_issue_poll.json"
+DRYDOCK="/data3/drydock"
+
+if [ ! -f "$TOKEN_FILE" ]; then
+    echo "[$(date)] ERROR: GitHub token missing at $TOKEN_FILE" >&2
+    exit 1
+fi
+
+mkdir -p "$(dirname "$STATE_FILE")"
+
+$PYTHON - <<'PYEOF'
+"""Find new issues opened by fbobe321 since last poll, notify Telegram."""
+import json
+import os
+import sys
+import urllib.request
+from datetime import datetime, timezone
+from pathlib import Path
+
+REPO = "fbobe321/drydock"
+AUTHOR = "fbobe321"
+TOKEN = Path.home().joinpath(".config/drydock/github_token").read_text().strip()
+STATE = Path.home().joinpath(".drydock/last_issue_poll.json")
+NOTIFY_SCRIPT = "/data3/drydock/scripts/notify_release.py"
+
+
+def gh_get(url: str):
+    req = urllib.request.Request(url)
+    req.add_header("Authorization", f"token {TOKEN}")
+    req.add_header("Accept", "application/vnd.github+json")
+    with urllib.request.urlopen(req, timeout=15) as resp:
+        return json.loads(resp.read().decode())
+
+
+def load_state() -> dict:
+    if STATE.is_file():
+        try:
+            return json.loads(STATE.read_text())
+        except json.JSONDecodeError:
+            pass
+    return {"seen_issue_numbers": []}
+
+
+def save_state(state: dict) -> None:
+    STATE.write_text(json.dumps(state, indent=2))
+
+
+def notify_telegram(text: str) -> None:
+    """Use the existing release notifier in 'status' mode (plain text)."""
+    import subprocess
+    subprocess.run(
+        [sys.executable, NOTIFY_SCRIPT, "status", text],
+        check=False, timeout=10,
+    )
+
+
+state = load_state()
+seen = set(state.get("seen_issue_numbers", []))
+
+issues = gh_get(
+    f"https://api.github.com/repos/{REPO}/issues"
+    f"?state=open&creator={AUTHOR}&per_page=50&sort=created&direction=desc"
+)
+
+new_items = []
+for it in issues:
+    if "pull_request" in it:
+        continue
+    n = it["number"]
+    if n in seen:
+        continue
+    new_items.append(it)
+    seen.add(n)
+
+if new_items:
+    lines = [f"⚓ {len(new_items)} new drydock issue(s) from fbobe321:"]
+    for it in new_items[:5]:
+        lines.append(f"  #{it['number']}: {it['title'][:80]}")
+    if len(new_items) > 5:
+        lines.append(f"  ...and {len(new_items) - 5} more")
+    lines.append(f"  https://github.com/{REPO}/issues")
+    notify_telegram("\n".join(lines))
+    print(f"[{datetime.now(timezone.utc).isoformat()}] notified {len(new_items)} new issues")
+else:
+    print(f"[{datetime.now(timezone.utc).isoformat()}] no new issues")
+
+state["seen_issue_numbers"] = sorted(seen)
+state["last_poll"] = datetime.now(timezone.utc).isoformat()
+save_state(state)
+PYEOF

drydock_cli-2.6.131/tests/test_issue_fixes.py

@@ -0,0 +1,118 @@
+"""Regression tests for GitHub issues #4, #5, #6."""
+from __future__ import annotations
+
+import os
+import pytest
+
+
+# Issue #4: todo broken — action defaults to read, silently ignores todos
+
+class TestIssue4TodoInferWrite:
+    def test_todos_without_action_infers_write(self):
+        from drydock.core.tools.builtins.todo import TodoArgs
+        args = TodoArgs(todos=[{"content": "first", "status": "pending"}])
+        assert args.action == "write"
+
+    def test_empty_todos_stays_read(self):
+        from drydock.core.tools.builtins.todo import TodoArgs
+        args = TodoArgs()
+        assert args.action == "read"
+
+    def test_explicit_read_with_todos_still_writes(self):
+        """Intent is clear from the todos list. Model's confused 'read'
+        shouldn't stomp a populated list."""
+        from drydock.core.tools.builtins.todo import TodoArgs
+        args = TodoArgs(action="read", todos=[{"content": "x"}])
+        assert args.action == "write"
+
+    def test_explicit_write_no_todos_still_writes(self):
+        """User explicitly said write but sent nothing — honor it (empties)."""
+        from drydock.core.tools.builtins.todo import TodoArgs
+        args = TodoArgs(action="write", todos=[])
+        assert args.action == "write"
+
+
+# Issue #5: cross-chunk newlines collapse to spaces in streaming output
+
+class TestIssue5StreamingLineBreaks:
+    @pytest.mark.asyncio
+    async def test_streaming_preserves_cross_chunk_paragraph_breaks(self, tmp_path):
+        """AssistantMessage.append_content must upgrade a single \\n at a
+        chunk boundary to \\n\\n so markdown renders it as a paragraph
+        break. Without this, 'hello\\n' + 'world' becomes 'hello\\nworld'
+        which markdown prints as 'hello world'."""
+        from drydock.cli.textual_ui.widgets.messages import AssistantMessage
+
+        # Bypass Textual mount — directly test the content-building path.
+        # AssistantMessage.__init__ calls super().__init__ which sets
+        # self._content; append_content mutates it via super().append_content.
+        msg = object.__new__(AssistantMessage)
+        msg._content = ""
+        msg._content_initialized = False
+        msg._stream = None
+
+        async def fake_super_append(c: str) -> None:
+            msg._content += c
+
+        # Monkey-patch only the super().append_content invocation.
+        from drydock.cli.textual_ui.widgets.messages import (
+            StreamingMessageBase,
+        )
+        original = StreamingMessageBase.append_content
+
+        async def shim(self, content: str) -> None:
+            self._content += content
+
+        StreamingMessageBase.append_content = shim
+        try:
+            await AssistantMessage.append_content(msg, "hello\n")
+            await AssistantMessage.append_content(msg, "world")
+        finally:
+            StreamingMessageBase.append_content = original
+
+        # hello\n + world must yield hello\n\nworld so markdown renders
+        # it as two paragraphs, not "hello world".
+        assert "hello\n\nworld" in msg._content
+
+
+# Issue #6: /find and /grep via bash should auto-approve even with full paths
+
+class TestIssue6BashAllowlistBasename:
+    def _bash_tool(self):
+        from drydock.core.tools.builtins.bash import Bash, BashToolConfig
+        from drydock.core.tools.base import BaseToolState
+        return Bash(config=BashToolConfig(), state=BaseToolState())
+
+    def test_bare_find_allowlisted(self):
+        from drydock.core.tools.builtins.bash import BashArgs
+        from drydock.core.tools.base import ToolPermission
+        tool = self._bash_tool()
+        res = tool.resolve_permission(BashArgs(command="find . -name '*.py'"))
+        assert res == ToolPermission.ALWAYS
+
+    def test_full_path_find_allowlisted(self):
+        """New fix: /usr/bin/find must behave like bare find for
+        auto-approval (basename match)."""
+        from drydock.core.tools.builtins.bash import BashArgs
+        from drydock.core.tools.base import ToolPermission
+        tool = self._bash_tool()
+        res = tool.resolve_permission(BashArgs(command="/usr/bin/find . -name '*.py'"))
+        assert res == ToolPermission.ALWAYS
+
+    def test_relative_path_grep_allowlisted(self):
+        from drydock.core.tools.builtins.bash import BashArgs
+        from drydock.core.tools.base import ToolPermission
+        tool = self._bash_tool()
+        res = tool.resolve_permission(BashArgs(command="./grep -r pattern ."))
+        assert res == ToolPermission.ALWAYS
+
+    def test_unknown_command_still_asks(self):
+        """Safety: basename match doesn't accidentally allow random
+        binaries. `/usr/bin/sudo` has basename `sudo` which isn't
+        allowlisted → no auto-approve."""
+        from drydock.core.tools.builtins.bash import BashArgs
+        tool = self._bash_tool()
+        res = tool.resolve_permission(BashArgs(command="/usr/bin/sudo whoami"))
+        # Returns None (caller prompts for approval) — not ALWAYS.
+        from drydock.core.tools.base import ToolPermission
+        assert res != ToolPermission.ALWAYS

{drydock_cli-2.6.129 → drydock_cli-2.6.131}/tests/test_reasoning_content.py

@@ -287,6 +287,48 @@ class TestAPIToolFormatHandlerReasoningContent:
         assert result.content == "Hello"
         assert result.reasoning_content is None
 
+    def test_process_api_response_message_preserves_leading_space(self):
+        """Streaming chunks like ' received' must keep their leading space —
+        otherwise the rendered text reads 'Whatwouldyoulikemetotest' (issue
+        #1: spaces stripped from streaming output with Gemma vLLM)."""
+        handler = APIToolFormatHandler()
+        mock_message = MagicMock(spec=["role", "content", "tool_calls"])
+        mock_message.role = "assistant"
+        mock_message.content = " received"
+        mock_message.tool_calls = None
+        result = handler.process_api_response_message(mock_message)
+        assert result.content == " received"
+
+    def test_process_api_response_message_preserves_trailing_space(self):
+        handler = APIToolFormatHandler()
+        mock_message = MagicMock(spec=["role", "content", "tool_calls"])
+        mock_message.role = "assistant"
+        mock_message.content = "What "
+        mock_message.tool_calls = None
+        result = handler.process_api_response_message(mock_message)
+        assert result.content == "What "
+
+    def test_process_api_response_message_preserves_internal_whitespace(self):
+        """Multi-word chunks must preserve all internal whitespace too."""
+        handler = APIToolFormatHandler()
+        mock_message = MagicMock(spec=["role", "content", "tool_calls"])
+        mock_message.role = "assistant"
+        mock_message.content = " I am working\n  correctly."
+        mock_message.tool_calls = None
+        result = handler.process_api_response_message(mock_message)
+        assert result.content == " I am working\n  correctly."
+
+    def test_process_api_response_message_pure_whitespace_becomes_none(self):
+        """A chunk that's entirely whitespace should still collapse to None
+        — we don't want to ship '   ' as a content chunk."""
+        handler = APIToolFormatHandler()
+        mock_message = MagicMock(spec=["role", "content", "tool_calls"])
+        mock_message.role = "assistant"
+        mock_message.content = "    "
+        mock_message.tool_calls = None
+        result = handler.process_api_response_message(mock_message)
+        assert result.content is None
+
 
 class TestAgentLoopStreamingReasoningEvents:
     @pytest.mark.asyncio

{drydock_cli-2.6.129 → drydock_cli-2.6.131}/uv.lock

@@ -326,7 +326,7 @@ wheels = [
 
 [[package]]
 name = "drydock-cli"
-version = "2.6.127"
+version = "2.6.130"
 source = { editable = "." }
 dependencies = [
     { name = "agent-client-protocol" },

drydock_cli-2.6.129/.auto_release.lock

@@ -1 +0,0 @@
-1768422