drydock-cli 0.5.0__tar.gz → 0.5.2__tar.gz

drydock_cli-0.5.2/PKG-INFO

@@ -0,0 +1,245 @@
+Metadata-Version: 2.4
+Name: drydock-cli
+Version: 0.5.2
+Summary: Drydock — nautical CLI coding agent
+Project-URL: Homepage, https://github.com/fbobe321/drydock
+Project-URL: Repository, https://github.com/fbobe321/drydock
+Project-URL: Issues, https://github.com/fbobe321/drydock/issues
+Project-URL: Documentation, https://github.com/fbobe321/drydock#readme
+Author: Drydock contributors, Mistral AI (original)
+License: Apache-2.0
+License-File: LICENSE
+License-File: NOTICE
+Keywords: ai,cli,coding-assistant,developer-tools,drydock,llm,nautical
+Classifier: Development Status :: 5 - Production/Stable
+Classifier: Intended Audience :: Developers
+Classifier: License :: OSI Approved :: Apache Software License
+Classifier: Operating System :: OS Independent
+Classifier: Programming Language :: Python
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.12
+Classifier: Topic :: Software Development
+Classifier: Topic :: Software Development :: Libraries :: Python Modules
+Classifier: Topic :: Utilities
+Requires-Python: >=3.12
+Requires-Dist: agent-client-protocol==0.8.1
+Requires-Dist: anyio>=4.12.0
+Requires-Dist: cachetools>=5.5.0
+Requires-Dist: cryptography<=46.0.3,>=44.0.0
+Requires-Dist: gitpython>=3.1.46
+Requires-Dist: giturlparse>=0.14.0
+Requires-Dist: google-auth>=2.0.0
+Requires-Dist: httpx>=0.28.1
+Requires-Dist: keyring>=25.6.0
+Requires-Dist: markdownify>=1.2.2
+Requires-Dist: mcp>=1.14.0
+Requires-Dist: mistralai==1.12.4
+Requires-Dist: packaging>=24.1
+Requires-Dist: pexpect>=4.9.0
+Requires-Dist: pydantic-settings>=2.12.0
+Requires-Dist: pydantic>=2.12.4
+Requires-Dist: pyperclip>=1.11.0
+Requires-Dist: python-dotenv>=1.0.0
+Requires-Dist: pyyaml>=6.0.0
+Requires-Dist: requests>=2.20.0
+Requires-Dist: rich>=14.0.0
+Requires-Dist: textual-speedups>=0.2.1
+Requires-Dist: textual>=7.4.0
+Requires-Dist: tomli-w>=1.2.0
+Requires-Dist: tree-sitter-bash>=0.25.1
+Requires-Dist: tree-sitter>=0.25.2
+Requires-Dist: watchfiles>=1.1.1
+Requires-Dist: zstandard>=0.25.0
+Description-Content-Type: text/markdown
+
+# DryDock
+
+[![Python Version](https://img.shields.io/badge/python-3.12%2B-blue)](https://www.python.org/downloads/release/python-3120/)
+[![License](https://img.shields.io/github/license/fbobe321/drydock)](https://github.com/fbobe321/drydock/blob/main/LICENSE)
+
+```
+         |    |    |
+        )_)  )_)  )_)
+       )___))___))___)\
+      )____)____)_____)\\
+    _____|____|____|____\\\__
+---\                   /----
+    \_________________/
+  ~~~~ ~~~ ~~~~ ~~~ ~~~~
+   ~~~ ~~~~ ~~~ ~~~ ~~~
+```
+
+**Nautical CLI coding agent. Chart your course. Execute with precision.**
+
+DryDock is a command-line coding assistant that works with any LLM provider. It provides a conversational interface to your codebase, allowing you to use natural language to explore, modify, and interact with your projects through a powerful set of tools.
+
+> [!WARNING]
+> DryDock works on Windows, but we officially support and target UNIX environments.
+
+### Install
+
+```bash
+pip install drydock-cli
+```
+
+Or with uv:
+
+```bash
+uv tool install drydock-cli
+```
+
+## Features
+
+- **Interactive Chat**: A conversational AI agent that understands your requests and breaks down complex tasks.
+- **Powerful Toolset**: Read, write, and patch files. Execute shell commands. Search code with `grep`. Manage todos. Delegate to subagents.
+- **Project-Aware Context**: DryDock automatically scans your project's file structure and Git status.
+- **Conda/Pip Support**: Auto-approves `pip install`, `conda install`, `pytest`, and other dev commands.
+- **Bundled Skills**: Ships with skills like `create-presentation` for PowerPoint generation.
+- **MCP Support**: Connect Model Context Protocol servers for extended capabilities.
+- **Safety First**: Tool execution approval with `--dangerously-skip-permissions` for full auto-approve.
+
+### Built-in Agents
+
+- **`default`**: Standard agent that requires approval for tool executions.
+- **`plan`**: Read-only agent for exploration and planning.
+- **`accept-edits`**: Auto-approves file edits only.
+- **`auto-approve`**: Auto-approves all tool executions.
+
+```bash
+drydock --agent plan
+```
+
+## Quick Start
+
+1. Navigate to your project directory and run:
+
+   ```bash
+   drydock
+   ```
+
+2. First run creates a config at `~/.drydock/config.toml` and prompts for your API key.
+
+3. Start chatting:
+
+   ```
+   > Can you find all TODO comments in this project?
+   ```
+
+## Usage
+
+### Interactive Mode
+
+```bash
+drydock                        # Start interactive session
+drydock "Fix the login bug"    # Start with a prompt
+drydock --continue             # Resume last session
+drydock --resume abc123        # Resume specific session
+```
+
+**Keyboard shortcuts:**
+- `Ctrl+C` — Cancel current operation (double-tap to quit)
+- `Shift+Tab` — Toggle auto-approve mode
+- `Ctrl+O` — Toggle tool output
+- `Ctrl+G` — Open external editor
+- `@` — File path autocompletion
+- `!command` — Run shell command directly
+
+### Programmatic Mode
+
+```bash
+drydock --prompt "Analyze the codebase" --max-turns 5 --output json
+drydock --dangerously-skip-permissions -p "Fix all lint errors"
+```
+
+### Trust Folder System
+
+DryDock includes a trust folder system. When you run DryDock in a directory with a `.drydock` folder, it asks you to confirm trust. Managed via `~/.drydock/trusted_folders.toml`.
+
+## Configuration
+
+DryDock is configured via `config.toml`. It looks first in `./.drydock/config.toml`, then `~/.drydock/config.toml`.
+
+### API Key
+
+```bash
+drydock --setup                              # Interactive setup
+export MISTRAL_API_KEY="your_key"            # Or set env var
+```
+
+Keys are saved to `~/.drydock/.env`.
+
+### Custom Agents
+
+Create agent configs in `~/.drydock/agents/`:
+
+```toml
+# ~/.drydock/agents/redteam.toml
+active_model = "devstral-2"
+system_prompt_id = "redteam"
+disabled_tools = ["search_replace", "write_file"]
+```
+
+```bash
+drydock --agent redteam
+```
+
+### Custom Prompts
+
+Create markdown files in `~/.drydock/prompts/`:
+
+```toml
+system_prompt_id = "my_custom_prompt"
+```
+
+### Skills
+
+DryDock discovers skills from:
+1. Custom paths in `config.toml` via `skill_paths`
+2. Project `.drydock/skills/` or `.agents/skills/`
+3. Global `~/.drydock/skills/`
+4. Bundled skills (shipped with the package)
+
+### MCP Servers
+
+```toml
+[[mcp_servers]]
+name = "fetch_server"
+transport = "stdio"
+command = "uvx"
+args = ["mcp-server-fetch"]
+```
+
+### Custom DryDock Home
+
+```bash
+export DRYDOCK_HOME="/path/to/custom/home"
+```
+
+This affects where DryDock looks for `config.toml`, `.env`, `agents/`, `prompts/`, `skills/`, and `logs/`.
+
+## Slash Commands
+
+Type `/help` in the input for available commands. Create custom slash commands via the skills system.
+
+## Session Management
+
+```bash
+drydock --continue              # Continue last session
+drydock --resume abc123         # Resume specific session
+drydock --workdir /path/to/dir  # Set working directory
+```
+
+## Resources
+
+- [CHANGELOG](CHANGELOG.md)
+- [CONTRIBUTING](CONTRIBUTING.md)
+- [ACP Setup](docs/acp-setup.md) — Editor/IDE integration
+
+## License
+
+Copyright 2025 Mistral AI (original work)
+Copyright 2026 DryDock contributors (modifications)
+
+Licensed under the Apache License, Version 2.0. See [LICENSE](LICENSE) for details.
+
+DryDock is a fork of [mistralai/mistral-vibe](https://github.com/mistralai/mistral-vibe) (Apache 2.0). See [NOTICE](NOTICE) for attribution.

drydock_cli-0.5.2/README.md

@@ -0,0 +1,191 @@
+# DryDock
+
+[![Python Version](https://img.shields.io/badge/python-3.12%2B-blue)](https://www.python.org/downloads/release/python-3120/)
+[![License](https://img.shields.io/github/license/fbobe321/drydock)](https://github.com/fbobe321/drydock/blob/main/LICENSE)
+
+```
+         |    |    |
+        )_)  )_)  )_)
+       )___))___))___)\
+      )____)____)_____)\\
+    _____|____|____|____\\\__
+---\                   /----
+    \_________________/
+  ~~~~ ~~~ ~~~~ ~~~ ~~~~
+   ~~~ ~~~~ ~~~ ~~~ ~~~
+```
+
+**Nautical CLI coding agent. Chart your course. Execute with precision.**
+
+DryDock is a command-line coding assistant that works with any LLM provider. It provides a conversational interface to your codebase, allowing you to use natural language to explore, modify, and interact with your projects through a powerful set of tools.
+
+> [!WARNING]
+> DryDock works on Windows, but we officially support and target UNIX environments.
+
+### Install
+
+```bash
+pip install drydock-cli
+```
+
+Or with uv:
+
+```bash
+uv tool install drydock-cli
+```
+
+## Features
+
+- **Interactive Chat**: A conversational AI agent that understands your requests and breaks down complex tasks.
+- **Powerful Toolset**: Read, write, and patch files. Execute shell commands. Search code with `grep`. Manage todos. Delegate to subagents.
+- **Project-Aware Context**: DryDock automatically scans your project's file structure and Git status.
+- **Conda/Pip Support**: Auto-approves `pip install`, `conda install`, `pytest`, and other dev commands.
+- **Bundled Skills**: Ships with skills like `create-presentation` for PowerPoint generation.
+- **MCP Support**: Connect Model Context Protocol servers for extended capabilities.
+- **Safety First**: Tool execution approval with `--dangerously-skip-permissions` for full auto-approve.
+
+### Built-in Agents
+
+- **`default`**: Standard agent that requires approval for tool executions.
+- **`plan`**: Read-only agent for exploration and planning.
+- **`accept-edits`**: Auto-approves file edits only.
+- **`auto-approve`**: Auto-approves all tool executions.
+
+```bash
+drydock --agent plan
+```
+
+## Quick Start
+
+1. Navigate to your project directory and run:
+
+   ```bash
+   drydock
+   ```
+
+2. First run creates a config at `~/.drydock/config.toml` and prompts for your API key.
+
+3. Start chatting:
+
+   ```
+   > Can you find all TODO comments in this project?
+   ```
+
+## Usage
+
+### Interactive Mode
+
+```bash
+drydock                        # Start interactive session
+drydock "Fix the login bug"    # Start with a prompt
+drydock --continue             # Resume last session
+drydock --resume abc123        # Resume specific session
+```
+
+**Keyboard shortcuts:**
+- `Ctrl+C` — Cancel current operation (double-tap to quit)
+- `Shift+Tab` — Toggle auto-approve mode
+- `Ctrl+O` — Toggle tool output
+- `Ctrl+G` — Open external editor
+- `@` — File path autocompletion
+- `!command` — Run shell command directly
+
+### Programmatic Mode
+
+```bash
+drydock --prompt "Analyze the codebase" --max-turns 5 --output json
+drydock --dangerously-skip-permissions -p "Fix all lint errors"
+```
+
+### Trust Folder System
+
+DryDock includes a trust folder system. When you run DryDock in a directory with a `.drydock` folder, it asks you to confirm trust. Managed via `~/.drydock/trusted_folders.toml`.
+
+## Configuration
+
+DryDock is configured via `config.toml`. It looks first in `./.drydock/config.toml`, then `~/.drydock/config.toml`.
+
+### API Key
+
+```bash
+drydock --setup                              # Interactive setup
+export MISTRAL_API_KEY="your_key"            # Or set env var
+```
+
+Keys are saved to `~/.drydock/.env`.
+
+### Custom Agents
+
+Create agent configs in `~/.drydock/agents/`:
+
+```toml
+# ~/.drydock/agents/redteam.toml
+active_model = "devstral-2"
+system_prompt_id = "redteam"
+disabled_tools = ["search_replace", "write_file"]
+```
+
+```bash
+drydock --agent redteam
+```
+
+### Custom Prompts
+
+Create markdown files in `~/.drydock/prompts/`:
+
+```toml
+system_prompt_id = "my_custom_prompt"
+```
+
+### Skills
+
+DryDock discovers skills from:
+1. Custom paths in `config.toml` via `skill_paths`
+2. Project `.drydock/skills/` or `.agents/skills/`
+3. Global `~/.drydock/skills/`
+4. Bundled skills (shipped with the package)
+
+### MCP Servers
+
+```toml
+[[mcp_servers]]
+name = "fetch_server"
+transport = "stdio"
+command = "uvx"
+args = ["mcp-server-fetch"]
+```
+
+### Custom DryDock Home
+
+```bash
+export DRYDOCK_HOME="/path/to/custom/home"
+```
+
+This affects where DryDock looks for `config.toml`, `.env`, `agents/`, `prompts/`, `skills/`, and `logs/`.
+
+## Slash Commands
+
+Type `/help` in the input for available commands. Create custom slash commands via the skills system.
+
+## Session Management
+
+```bash
+drydock --continue              # Continue last session
+drydock --resume abc123         # Resume specific session
+drydock --workdir /path/to/dir  # Set working directory
+```
+
+## Resources
+
+- [CHANGELOG](CHANGELOG.md)
+- [CONTRIBUTING](CONTRIBUTING.md)
+- [ACP Setup](docs/acp-setup.md) — Editor/IDE integration
+
+## License
+
+Copyright 2025 Mistral AI (original work)
+Copyright 2026 DryDock contributors (modifications)
+
+Licensed under the Apache License, Version 2.0. See [LICENSE](LICENSE) for details.
+
+DryDock is a fork of [mistralai/mistral-vibe](https://github.com/mistralai/mistral-vibe) (Apache 2.0). See [NOTICE](NOTICE) for attribution.

{drydock_cli-0.5.0 → drydock_cli-0.5.2}/drydock/core/agent_loop.py

@@ -288,6 +288,18 @@ class AgentLoop:
 
     async def act(self, msg: str) -> AsyncGenerator[BaseEvent]:
         self._clean_message_history()
+
+        # Load project state for cross-session context
+        try:
+            from drydock.core.session.state_file import load_state
+            state_content = load_state()
+            if state_content:
+                self._inject_system_note(
+                    f"Previous session state:\n{state_content[:500]}"
+                )
+        except Exception:
+            pass  # Non-critical
+
         async for event in self._conversation_loop(msg):
             yield event
 

{drydock_cli-0.5.0 → drydock_cli-0.5.2}/drydock/core/middleware.py

@@ -98,35 +98,63 @@ class AutoCompactMiddleware:
 
 
 class ContextWarningMiddleware:
+    """Tiered context warnings inspired by GSD's context monitoring.
+
+    Warns at multiple thresholds as context fills up:
+    - 50% used: soft warning ("you're halfway through")
+    - 65% used: moderate warning ("wrap up current task")
+    - 75% used: critical warning ("finish NOW or compact")
+    - 85% used: emergency ("stop exploring, make final edit")
+
+    Debounced: only warns every 5 tool calls to avoid spamming.
+    """
+
+    _TIERS = [
+        (0.50, "soft", "You've used {pct:.0f}% of context ({used:,}/{max:,} tokens). Start wrapping up your current task."),
+        (0.65, "moderate", "Context at {pct:.0f}%. Stop exploring — finish your current edit and verify it works."),
+        (0.75, "critical", "Context at {pct:.0f}% — CRITICAL. Make your final edit NOW. No more grep/read_file. Use /compact if you need to continue."),
+        (0.85, "emergency", "Context nearly full ({pct:.0f}%). You MUST stop after this turn. Apply your fix with search_replace immediately or the session will degrade."),
+    ]
+
     def __init__(
         self, threshold_percent: float = 0.5, max_context: int | None = None
     ) -> None:
         self.threshold_percent = threshold_percent
         self.max_context = max_context
-        self.has_warned = False
+        self._tier_warned: set[str] = set()
+        self._calls_since_last_warn = 0
+        self._debounce_interval = 5
 
     async def before_turn(self, context: ConversationContext) -> MiddlewareResult:
-        if self.has_warned:
+        max_context = self.max_context
+        if max_context is None or max_context == 0:
             return MiddlewareResult()
 
-        max_context = self.max_context
-        if max_context is None:
+        self._calls_since_last_warn += 1
+
+        # Debounce: don't warn on every single turn
+        if self._calls_since_last_warn < self._debounce_interval:
             return MiddlewareResult()
 
-        if context.stats.context_tokens >= max_context * self.threshold_percent:
-            self.has_warned = True
+        pct_used = context.stats.context_tokens / max_context
 
-            percentage_used = (context.stats.context_tokens / max_context) * 100
-            warning_msg = f"<{VIBE_WARNING_TAG}>You have used {percentage_used:.0f}% of your total context ({context.stats.context_tokens:,}/{max_context:,} tokens)</{VIBE_WARNING_TAG}>"
+        # Find the highest tier we've crossed but haven't warned about
+        for threshold, tier_name, template in reversed(self._TIERS):
+            if pct_used >= threshold and tier_name not in self._tier_warned:
+                self._tier_warned.add(tier_name)
+                self._calls_since_last_warn = 0
 
-            return MiddlewareResult(
-                action=MiddlewareAction.INJECT_MESSAGE, message=warning_msg
-            )
+                warning_msg = f"<{VIBE_WARNING_TAG}>{template.format(pct=pct_used * 100, used=context.stats.context_tokens, max=max_context)}</{VIBE_WARNING_TAG}>"
+                return MiddlewareResult(
+                    action=MiddlewareAction.INJECT_MESSAGE, message=warning_msg
+                )
+                break
 
         return MiddlewareResult()
 
     def reset(self, reset_reason: ResetReason = ResetReason.STOP) -> None:
-        self.has_warned = False
+        self._tier_warned.clear()
+        self._calls_since_last_warn = 0
 
 
 def make_plan_agent_reminder(plan_file_path: str) -> str:

{drydock_cli-0.5.0 → drydock_cli-0.5.2}/drydock/core/prompts/cli.md

@@ -59,6 +59,16 @@ Flip-flopping (add X → remove X → add X) is a critical failure. Commit to a
 Ambiguous Prompts
 If the user's message is very short or ambiguous (e.g., "test", "check", "fix"), ask what they want before exploring the filesystem. Do NOT start scanning directories or running `find` on a vague prompt. Ask: "What would you like me to test/check/fix?"
 
+Deviation Handling
+When your fix attempt hits unexpected issues, follow these rules:
+1. **Bug in your fix** → Auto-fix immediately. Re-read the error, adjust your edit, retry.
+2. **Missing dependency/import** → Auto-resolve. Add the import or install the package.
+3. **Blocking issue** (wrong file, missing context) → Auto-resolve by grepping for the right location.
+4. **Architectural decision** (should we refactor? change the API? add a new module?) → STOP and ask the user. Do NOT make architectural decisions unilaterally.
+5. **Scope change** (the fix requires changes to 5+ files, or touches unrelated code) → STOP and ask the user before proceeding.
+
+Rules 1-3: fix silently. Rules 4-5: always ask.
+
 Response Format
 No Noise
 No greetings, outros, hedging, puffery, or tool narration.

drydock_cli-0.5.2/drydock/core/session/state_file.py

@@ -0,0 +1,143 @@
+"""Structured state file for cross-session memory.
+
+Maintains a `.drydock/state.md` file in the project directory that tracks:
+- Current task and status
+- Files modified in this session
+- Key decisions made
+- Blockers and next steps
+
+Max 100 lines. Loaded on session start, updated on session end.
+Enables "resume where I left off" workflows.
+
+Inspired by GSD's STATE.md concept.
+"""
+
+from __future__ import annotations
+
+import logging
+from datetime import datetime
+from pathlib import Path
+
+logger = logging.getLogger(__name__)
+
+_MAX_LINES = 100
+_STATE_FILENAME = "state.md"
+
+
+def _get_state_path(project_dir: Path | None = None) -> Path:
+    """Get the path to the state file."""
+    base = project_dir or Path.cwd()
+    return base / ".drydock" / _STATE_FILENAME
+
+
+def load_state(project_dir: Path | None = None) -> str:
+    """Load the state file content, or empty string if it doesn't exist."""
+    path = _get_state_path(project_dir)
+    try:
+        if path.exists():
+            content = path.read_text(encoding="utf-8")
+            logger.info("Loaded session state from %s (%d lines)", path, content.count("\n"))
+            return content
+    except (OSError, PermissionError) as e:
+        logger.warning("Could not read state file %s: %s", path, e)
+    return ""
+
+
+def save_state(
+    project_dir: Path | None = None,
+    *,
+    current_task: str = "",
+    files_modified: list[str] | None = None,
+    decisions: list[str] | None = None,
+    blockers: list[str] | None = None,
+    next_steps: list[str] | None = None,
+    custom_notes: str = "",
+) -> None:
+    """Save the state file with structured session information."""
+    path = _get_state_path(project_dir)
+
+    lines: list[str] = [
+        f"# DryDock Session State",
+        f"_Updated: {datetime.now().strftime('%Y-%m-%d %H:%M')}_",
+        "",
+    ]
+
+    if current_task:
+        lines.extend([f"## Current Task", current_task, ""])
+
+    if files_modified:
+        lines.append("## Files Modified")
+        for f in files_modified[-20:]:  # Keep last 20
+            lines.append(f"- `{f}`")
+        lines.append("")
+
+    if decisions:
+        lines.append("## Decisions")
+        for d in decisions[-10:]:  # Keep last 10
+            lines.append(f"- {d}")
+        lines.append("")
+
+    if blockers:
+        lines.append("## Blockers")
+        for b in blockers:
+            lines.append(f"- {b}")
+        lines.append("")
+
+    if next_steps:
+        lines.append("## Next Steps")
+        for n in next_steps:
+            lines.append(f"- {n}")
+        lines.append("")
+
+    if custom_notes:
+        lines.extend(["## Notes", custom_notes, ""])
+
+    # Enforce max lines
+    content = "\n".join(lines[:_MAX_LINES])
+
+    try:
+        path.parent.mkdir(parents=True, exist_ok=True)
+        path.write_text(content, encoding="utf-8")
+        logger.info("Saved session state to %s", path)
+    except (OSError, PermissionError) as e:
+        logger.warning("Could not write state file %s: %s", path, e)
+
+
+def update_state_files_modified(
+    project_dir: Path | None = None,
+    files: list[str] | None = None,
+) -> None:
+    """Quick update: append modified files to the state without rewriting everything."""
+    if not files:
+        return
+
+    path = _get_state_path(project_dir)
+    try:
+        existing = path.read_text(encoding="utf-8") if path.exists() else ""
+
+        if "## Files Modified" not in existing:
+            # Append a new section
+            existing += "\n## Files Modified\n"
+
+        for f in files:
+            entry = f"- `{f}`"
+            if entry not in existing:
+                existing += f"{entry}\n"
+
+        # Enforce max lines
+        lines = existing.split("\n")[:_MAX_LINES]
+        path.parent.mkdir(parents=True, exist_ok=True)
+        path.write_text("\n".join(lines), encoding="utf-8")
+    except (OSError, PermissionError):
+        pass  # Non-critical
+
+
+def clear_state(project_dir: Path | None = None) -> None:
+    """Remove the state file."""
+    path = _get_state_path(project_dir)
+    try:
+        if path.exists():
+            path.unlink()
+            logger.info("Cleared session state at %s", path)
+    except (OSError, PermissionError):
+        pass