drydock-cli 3.0.99__tar.gz → 3.0.101__tar.gz

{drydock_cli-3.0.99/drydock_cli.egg-info → drydock_cli-3.0.101}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: drydock-cli
-Version: 3.0.99
+Version: 3.0.101
 Summary: Drydock — a local, provider-agnostic terminal coding agent for local LLMs
 Author: Frank Bobe III
 License-Expression: Apache-2.0

{drydock_cli-3.0.99 → drydock_cli-3.0.101}/drydock/tools/__init__.py

@@ -192,13 +192,16 @@ SCHEMAS = [
     },
     {
         "name": "Edit",
-        "description": "Replace exact text in a file. old_string must match exactly.",
+        "description": "Replace exact text in a file. old_string must match exactly. "
+                       "By default old_string must be unique; pass replace_all:true "
+                       "to replace every occurrence.",
         "input_schema": {
             "type": "object",
             "properties": {
                 "file_path": {"type": "string"},
                 "old_string": {"type": "string", "description": "Exact text to find"},
                 "new_string": {"type": "string", "description": "Replacement text"},
+                "replace_all": {"type": "boolean", "description": "Replace ALL occurrences (default false)"},
             },
             "required": ["file_path", "old_string", "new_string"],
         },
@@ -875,9 +878,15 @@ def tool_edit(params: dict, config: dict) -> str:
                     f"to copy the exact text (including indentation)."
                 )
         count = content.count(old)
-        if count > 1:
-            return f"Error: old_string found {count} times in {fp}. Add more context to make it unique."
-        updated = content.replace(old, new, 1)
+        # replace_all (models expect it, like the standard Edit tool): replace
+        # EVERY occurrence. Without it, multiple matches are ambiguous → error.
+        replace_all = bool(params.get("replace_all", False))
+        if count > 1 and not replace_all:
+            return (
+                f"Error: old_string found {count} times in {fp}. Add more context "
+                f"to make it unique, or pass replace_all: true to replace all."
+            )
+        updated = content.replace(old, new) if replace_all else content.replace(old, new, 1)
         if updated == content:
             # No-op edit (old_string == new_string, or the replacement changes
             # nothing). Reporting "Edited" here is a false success that invites
@@ -893,6 +902,8 @@ def tool_edit(params: dict, config: dict) -> str:
         Path(fp).write_text(updated, encoding="utf-8")
         _record_undo(config, fp, content)
         result = f"Edited {fp}: replaced {len(old)} chars with {len(new)} chars"
+        if replace_all and count > 1:
+            result += f" ({count} occurrences)"
         # Advisory post-edit syntax check.
         warn = python_syntax_warning(fp, updated)
         if warn:
@@ -1051,6 +1062,14 @@ def tool_bash(params: dict, config: dict) -> str:
             if cancel is not None and cancel.is_set():
                 kill_process_group(proc)
                 proc.wait()
+                partial = _sanitize_bash_output(
+                    _collapse_repeated_lines("".join(list(chunks)))
+                ).rstrip()
+                if partial:
+                    tail = partial[-_PARTIAL_TAIL:]
+                    if len(partial) > _PARTIAL_TAIL:
+                        tail = "[... earlier output truncated ...]\n" + tail
+                    return f"[stopped by user]\n\n--- output before stop ---\n{tail}"
                 return "[stopped by user]"
             # The SHELL has exited but the pipe is still open → the command
             # backgrounded a child (`cmd &`, a server the task wants to keep

{drydock_cli-3.0.99 → drydock_cli-3.0.101/drydock_cli.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: drydock-cli
-Version: 3.0.99
+Version: 3.0.101
 Summary: Drydock — a local, provider-agnostic terminal coding agent for local LLMs
 Author: Frank Bobe III
 License-Expression: Apache-2.0

{drydock_cli-3.0.99 → drydock_cli-3.0.101}/drydock_cli.egg-info/SOURCES.txt

@@ -57,6 +57,7 @@ tests/test_bash_safety.py
 tests/test_bash_sanitize.py
 tests/test_bash_shell.py
 tests/test_bash_stdin.py
+tests/test_bash_stop_partial.py
 tests/test_bash_timeout_network.py
 tests/test_bash_timeout_param.py
 tests/test_cci.py
@@ -69,6 +70,7 @@ tests/test_context_limit_config.py
 tests/test_detect.py
 tests/test_dispatch.py
 tests/test_e2e_connected.py
+tests/test_edit_replace_all.py
 tests/test_empty_response.py
 tests/test_extract.py
 tests/test_failure_loop.py

{drydock_cli-3.0.99 → drydock_cli-3.0.101}/pyproject.toml

@@ -7,7 +7,7 @@ build-backend = "setuptools.build_meta"
 # PyPI distribution name is drydock-cli (the established install name, continued
 # from the retired v2 fork); the import package + CLI command stay `drydock`.
 name = "drydock-cli"
-version = "3.0.99"
+version = "3.0.101"
 description = "Drydock — a local, provider-agnostic terminal coding agent for local LLMs"
 readme = "README.md"
 requires-python = ">=3.11"

drydock_cli-3.0.101/tests/test_bash_stop_partial.py

@@ -0,0 +1,50 @@
+"""tool_bash must preserve partial output when the user presses STOP.
+Without this fix, cancel.is_set() triggered a bare "[stopped by user]" return
+that discarded all chunks accumulated before the stop — losing useful diagnostics
+(test-run results, build output, etc.) that the model needs to continue.
+
+Mirrors the timeout-partial-output behaviour already in test_bash_background.py.
+"""
+from __future__ import annotations
+
+import threading
+import time
+
+from drydock.tools import tool_bash
+
+
+def test_stop_preserves_partial_output():
+    """Partial output printed before the cancel Event fires must appear in the
+    tool result alongside the '[stopped by user]' marker."""
+    cancel = threading.Event()
+
+    # Fire cancel after a short delay so the command has time to emit output.
+    def _fire():
+        time.sleep(0.5)
+        cancel.set()
+
+    threading.Thread(target=_fire, daemon=True).start()
+    out = tool_bash(
+        {"command": "echo PARTIAL_LINE; echo second_line; sleep 30", "timeout": 20},
+        {"cwd": "/tmp", "_cancel": cancel},
+    )
+    assert "stopped by user" in out
+    assert "PARTIAL_LINE" in out, f"partial output missing from stop result: {out!r}"
+    assert "output before stop" in out
+
+
+def test_stop_with_no_output_returns_plain_marker():
+    """If the command produces no output before the stop, the result is just
+    '[stopped by user]' without a spurious empty block."""
+    cancel = threading.Event()
+
+    def _fire():
+        time.sleep(0.05)
+        cancel.set()
+
+    threading.Thread(target=_fire, daemon=True).start()
+    out = tool_bash(
+        {"command": "sleep 30", "timeout": 20},
+        {"cwd": "/tmp", "_cancel": cancel},
+    )
+    assert out == "[stopped by user]"

drydock_cli-3.0.101/tests/test_edit_replace_all.py

@@ -0,0 +1,37 @@
+"""The Edit tool honors replace_all (models expect it, like the standard Edit
+tool). Without it, multiple matches are an error; with it, every occurrence is
+replaced. Found by probing Edit edge cases."""
+from __future__ import annotations
+
+from drydock.tools import tool_edit
+from drydock import tool_registry
+
+
+def _edit(tmp_path, setup, **kw):
+    f = tmp_path / "f.txt"; f.write_text(setup)
+    out = tool_edit({"file_path": str(f), **kw}, {"cwd": str(tmp_path)})
+    return out, f.read_text()
+
+
+def test_replace_all_replaces_every_occurrence(tmp_path):
+    out, text = _edit(tmp_path, "a a a\n", old_string="a", new_string="b", replace_all=True)
+    assert "Error" not in out and text == "b b b\n" and "3 occurrences" in out
+
+
+def test_default_still_errors_on_multiple(tmp_path):
+    out, text = _edit(tmp_path, "a a a\n", old_string="a", new_string="b")
+    assert "found 3 times" in out and text == "a a a\n"  # unchanged
+
+
+def test_replace_all_single_occurrence(tmp_path):
+    out, text = _edit(tmp_path, "just one\n", old_string="one", new_string="two", replace_all=True)
+    assert text == "just two\n"
+
+
+def test_replace_all_no_match_errors(tmp_path):
+    out, text = _edit(tmp_path, "hello\n", old_string="zzz", new_string="b", replace_all=True)
+    assert "not found" in out and text == "hello\n"
+
+
+def test_replace_all_in_schema():
+    assert "replace_all" in tool_registry.get("Edit").schema["input_schema"]["properties"]