drydock-cli 1.7.6__tar.gz → 1.9.0__tar.gz

drydock_cli-1.9.0/PKG-INFO

@@ -0,0 +1,258 @@
+Metadata-Version: 2.4
+Name: drydock-cli
+Version: 1.9.0
+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: beautifulsoup4>=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: pypdf>=4.0.0
+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`.
+
+### Consultant Model
+
+Set a smarter model for the `/consult` command:
+
+```toml
+# In ~/.drydock/config.toml
+consultant_model = "gemini-2.5-pro"
+```
+
+The consultant provides read-only advice — it never calls tools. Use `/consult <question>` to ask it. The response appears in your conversation so the local model can see it.
+
+### 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-1.9.0/drydock/cli/cli.py

@@ -0,0 +1,215 @@
+from __future__ import annotations
+
+import argparse
+from pathlib import Path
+import sys
+
+from rich import print as rprint
+import tomli_w
+
+from drydock import __version__
+from drydock.cli.textual_ui.app import run_textual_ui
+from drydock.core.agent_loop import AgentLoop
+from drydock.core.agents.models import BuiltinAgentName
+from drydock.core.config import (
+    MissingAPIKeyError,
+    MissingPromptFileError,
+    VibeConfig,
+    load_dotenv_values,
+)
+from drydock.core.config.harness_files import get_harness_files_manager
+from drydock.core.logger import logger
+from drydock.core.paths import HISTORY_FILE
+from drydock.core.programmatic import run_programmatic
+from drydock.core.session.session_loader import SessionLoader
+from drydock.core.types import EntrypointMetadata, LLMMessage, OutputFormat, Role
+from drydock.core.utils import ConversationLimitException
+from drydock.setup.onboarding import run_onboarding
+
+
+def get_initial_agent_name(args: argparse.Namespace) -> str:
+    if args.prompt is not None and args.agent == BuiltinAgentName.DEFAULT:
+        return BuiltinAgentName.AUTO_APPROVE
+    return args.agent
+
+
+def get_prompt_from_stdin() -> str | None:
+    if sys.stdin.isatty():
+        return None
+    try:
+        if content := sys.stdin.read().strip():
+            sys.stdin = sys.__stdin__ = open("/dev/tty")
+            return content
+    except KeyboardInterrupt:
+        pass
+    except OSError:
+        return None
+
+    return None
+
+
+def load_config_or_exit() -> VibeConfig:
+    try:
+        return VibeConfig.load()
+    except MissingAPIKeyError:
+        run_onboarding()
+        return VibeConfig.load()
+    except MissingPromptFileError as e:
+        rprint(f"[yellow]Invalid system prompt id: {e}[/]")
+        sys.exit(1)
+    except ValueError as e:
+        rprint(f"[yellow]{e}[/]")
+        sys.exit(1)
+
+
+def bootstrap_config_files() -> None:
+    mgr = get_harness_files_manager()
+    config_file = mgr.user_config_file
+    if not config_file.exists():
+        try:
+            config_file.parent.mkdir(parents=True, exist_ok=True)
+            with config_file.open("wb") as f:
+                tomli_w.dump(VibeConfig.create_default(), f)
+        except Exception as e:
+            rprint(f"[yellow]Could not create default config file: {e}[/]")
+
+    history_file = HISTORY_FILE.path
+    if not history_file.exists():
+        try:
+            history_file.parent.mkdir(parents=True, exist_ok=True)
+            history_file.write_text("Hello Drydock!\n", "utf-8")
+        except Exception as e:
+            rprint(f"[yellow]Could not create history file: {e}[/]")
+
+
+def load_session(
+    args: argparse.Namespace, config: VibeConfig
+) -> tuple[list[LLMMessage], Path] | None:
+    if not args.continue_session and not args.resume:
+        return None
+
+    if not config.session_logging.enabled:
+        rprint(
+            "[red]Session logging is disabled. "
+            "Enable it in config to use --continue or --resume[/]"
+        )
+        sys.exit(1)
+
+    session_to_load = None
+    if args.continue_session:
+        session_to_load = SessionLoader.find_latest_session(config.session_logging)
+        if not session_to_load:
+            rprint(
+                f"[red]No previous sessions found in "
+                f"{config.session_logging.save_dir}[/]"
+            )
+            sys.exit(1)
+    else:
+        session_to_load = SessionLoader.find_session_by_id(
+            args.resume, config.session_logging
+        )
+        if not session_to_load:
+            rprint(
+                f"[red]Session '{args.resume}' not found in "
+                f"{config.session_logging.save_dir}[/]"
+            )
+            sys.exit(1)
+
+    try:
+        loaded_messages, _ = SessionLoader.load_session(session_to_load)
+        return loaded_messages, session_to_load
+    except Exception as e:
+        rprint(f"[red]Failed to load session: {e}[/]")
+        sys.exit(1)
+
+
+def _resume_previous_session(
+    agent_loop: AgentLoop, loaded_messages: list[LLMMessage], session_path: Path
+) -> None:
+    non_system_messages = [msg for msg in loaded_messages if msg.role != Role.system]
+    agent_loop.messages.extend(non_system_messages)
+
+    _, metadata = SessionLoader.load_session(session_path)
+    session_id = metadata.get("session_id", agent_loop.session_id)
+    agent_loop.session_id = session_id
+    agent_loop.session_logger.resume_existing_session(session_id, session_path)
+
+    logger.info(
+        "Resumed session %s with %d messages", session_id, len(non_system_messages)
+    )
+
+
+def run_cli(args: argparse.Namespace) -> None:
+    load_dotenv_values()
+    bootstrap_config_files()
+
+    if args.setup:
+        run_onboarding()
+        sys.exit(0)
+
+    try:
+        initial_agent_name = get_initial_agent_name(args)
+        config = load_config_or_exit()
+
+        if args.enabled_tools:
+            config.enabled_tools = args.enabled_tools
+
+        loaded_session = load_session(args, config)
+
+        stdin_prompt = get_prompt_from_stdin()
+        if args.prompt is not None:
+            config.disabled_tools = [*config.disabled_tools, "ask_user_question"]
+            programmatic_prompt = args.prompt or stdin_prompt
+            if not programmatic_prompt:
+                print(
+                    "Error: No prompt provided for programmatic mode", file=sys.stderr
+                )
+                sys.exit(1)
+            output_format = OutputFormat(
+                args.output if hasattr(args, "output") else "text"
+            )
+
+            try:
+                final_response = run_programmatic(
+                    config=config,
+                    prompt=programmatic_prompt,
+                    max_turns=args.max_turns,
+                    max_price=args.max_price,
+                    output_format=output_format,
+                    previous_messages=loaded_session[0] if loaded_session else None,
+                    agent_name=initial_agent_name,
+                )
+                if final_response:
+                    print(final_response)
+                sys.exit(0)
+            except ConversationLimitException as e:
+                print(e, file=sys.stderr)
+                sys.exit(1)
+            except RuntimeError as e:
+                print(f"Error: {e}", file=sys.stderr)
+                sys.exit(1)
+        else:
+            agent_loop = AgentLoop(
+                config,
+                agent_name=initial_agent_name,
+                enable_streaming=True,
+                entrypoint_metadata=EntrypointMetadata(
+                    agent_entrypoint="cli",
+                    agent_version=__version__,
+                    client_name="drydock_cli",
+                    client_version=__version__,
+                ),
+            )
+
+            if loaded_session:
+                _resume_previous_session(agent_loop, *loaded_session)
+
+            run_textual_ui(
+                agent_loop=agent_loop,
+                initial_prompt=args.initial_prompt or stdin_prompt,
+                teleport_on_start=args.teleport,
+            )
+
+    except (KeyboardInterrupt, EOFError):
+        rprint("\n[dim]Bye![/]")
+        sys.exit(0)

drydock_cli-1.9.0/drydock/cli/entrypoint.py

@@ -0,0 +1,231 @@
+from __future__ import annotations
+
+import argparse
+import os
+from pathlib import Path
+import sys
+
+from rich import print as rprint
+
+from drydock import __version__
+from drydock.core.agents.models import BuiltinAgentName
+from drydock.core.config.harness_files import init_harness_files_manager
+from drydock.core.trusted_folders import has_trustable_content, trusted_folders_manager
+from drydock.setup.trusted_folders.trust_folder_dialog import (
+    TrustDialogQuitException,
+    ask_trust_folder,
+)
+
+
+def parse_arguments() -> argparse.Namespace:
+    parser = argparse.ArgumentParser(description="Run the Drydock interactive CLI")
+    parser.add_argument(
+        "-v", "--version", action="version", version=f"%(prog)s {__version__}"
+    )
+    parser.add_argument(
+        "initial_prompt",
+        nargs="?",
+        metavar="PROMPT",
+        help="Initial prompt to start the interactive session with.",
+    )
+    parser.add_argument(
+        "-p",
+        "--prompt",
+        nargs="?",
+        const="",
+        metavar="TEXT",
+        help="Run in programmatic mode: send prompt, auto-approve all tools, "
+        "output response, and exit.",
+    )
+    parser.add_argument(
+        "--max-turns",
+        type=int,
+        metavar="N",
+        help="Maximum number of assistant turns "
+        "(only applies in programmatic mode with -p).",
+    )
+    parser.add_argument(
+        "--max-price",
+        type=float,
+        metavar="DOLLARS",
+        help="Maximum cost in dollars (only applies in programmatic mode with -p). "
+        "Session will be interrupted if cost exceeds this limit.",
+    )
+    parser.add_argument(
+        "--enabled-tools",
+        action="append",
+        metavar="TOOL",
+        help="Enable specific tools. In programmatic mode (-p), this disables "
+        "all other tools. "
+        "Can use exact names, glob patterns (e.g., 'bash*'), or "
+        "regex with 're:' prefix. Can be specified multiple times.",
+    )
+    parser.add_argument(
+        "--output",
+        type=str,
+        choices=["text", "json", "streaming"],
+        default="text",
+        help="Output format for programmatic mode (-p): 'text' "
+        "for human-readable (default), 'json' for all messages at end, "
+        "'streaming' for newline-delimited JSON per message.",
+    )
+    parser.add_argument(
+        "--json-schema",
+        metavar="FILE",
+        help="Path to JSON Schema file. Constrains the final output to match the schema.",
+    )
+    parser.add_argument(
+        "--agent",
+        metavar="NAME",
+        default=BuiltinAgentName.DEFAULT,
+        help="Agent to use (builtin: default, plan, accept-edits, auto-approve, "
+        "or custom from ~/.drydock/agents/NAME.toml)",
+    )
+    parser.add_argument("--setup", action="store_true", help="Setup API key and exit")
+    parser.add_argument(
+        "--workdir",
+        type=Path,
+        metavar="DIR",
+        help="Change to this directory before running",
+    )
+
+    parser.add_argument(
+        "--dangerously-skip-permissions",
+        action="store_true",
+        help="Skip all tool permission checks. Equivalent to --agent auto-approve. "
+        "Use with caution — tools will execute without confirmation.",
+    )
+    parser.add_argument(
+        "-k", "--insecure",
+        action="store_true",
+        help="Disable SSL certificate verification for web searches and API calls. "
+        "Useful behind corporate proxies with self-signed certificates.",
+    )
+    parser.add_argument(
+        "--consultant",
+        metavar="MODEL",
+        help="Enable consultant mode: when the agent is uncertain or stuck in a loop, "
+        "it can call a more capable model (e.g., 'gemini-2.5-pro') for single-turn advice. "
+        "The consultant is NOT used for tool calls, only reasoning.",
+    )
+
+    parser.add_argument(
+        "--local",
+        metavar="URL",
+        nargs="?",
+        const="http://localhost:8000/v1",
+        help="Use a local LLM server (default: http://localhost:8000/v1). "
+        "Sets up a 'local' provider and model automatically. "
+        "Example: drydock --local http://localhost:11434/v1",
+    )
+
+    # Feature flag for teleport, not exposed to the user yet
+    parser.add_argument("--teleport", action="store_true", help=argparse.SUPPRESS)
+
+    continuation_group = parser.add_mutually_exclusive_group()
+    continuation_group.add_argument(
+        "-c",
+        "--continue",
+        action="store_true",
+        dest="continue_session",
+        help="Continue from the most recent saved session",
+    )
+    continuation_group.add_argument(
+        "--resume",
+        metavar="SESSION_ID",
+        help="Resume a specific session by its ID (supports partial matching)",
+    )
+    return parser.parse_args()
+
+
+def check_and_resolve_trusted_folder() -> None:
+    try:
+        cwd = Path.cwd()
+    except FileNotFoundError:
+        rprint(
+            "[red]Error: Current working directory no longer exists.[/]\n"
+            "[yellow]The directory you started drydock from has been deleted. "
+            "Please change to an existing directory and try again, "
+            "or use --workdir to specify a working directory.[/]"
+        )
+        sys.exit(1)
+
+    if not has_trustable_content(cwd) or cwd.resolve() == Path.home().resolve():
+        return
+
+    is_folder_trusted = trusted_folders_manager.is_trusted(cwd)
+
+    if is_folder_trusted is not None:
+        return
+
+    try:
+        is_folder_trusted = ask_trust_folder(cwd)
+    except (KeyboardInterrupt, EOFError, TrustDialogQuitException):
+        sys.exit(0)
+    except Exception as e:
+        rprint(f"[yellow]Error showing trust dialog: {e}[/]")
+        return
+
+    if is_folder_trusted is True:
+        trusted_folders_manager.add_trusted(cwd)
+    elif is_folder_trusted is False:
+        trusted_folders_manager.add_untrusted(cwd)
+
+
+def main() -> None:
+    args = parse_arguments()
+
+    if args.workdir:
+        workdir = args.workdir.expanduser().resolve()
+        if not workdir.is_dir():
+            rprint(
+                f"[red]Error: --workdir does not exist or is not a directory: {workdir}[/]"
+            )
+            sys.exit(1)
+        os.chdir(workdir)
+
+    # --dangerously-skip-permissions → force auto-approve agent
+    if args.dangerously_skip_permissions:
+        args.agent = BuiltinAgentName.AUTO_APPROVE
+
+    # --insecure → disable SSL verification globally
+    if args.insecure:
+        os.environ["DRYDOCK_INSECURE"] = "1"
+        os.environ["CURL_CA_BUNDLE"] = ""
+        os.environ["REQUESTS_CA_BUNDLE"] = ""
+        import ssl
+        ssl._create_default_https_context = ssl._create_unverified_context
+
+    # --consultant → store for agent loop to use
+    if args.consultant:
+        os.environ["DRYDOCK_CONSULTANT_MODEL"] = args.consultant
+
+    # --local → set up local LLM provider without editing config
+    if getattr(args, "local", None):
+        os.environ["DRYDOCK_LOCAL_URL"] = args.local
+        # Detect model name from the server
+        try:
+            import httpx
+            resp = httpx.get(f"{args.local}/models", timeout=5)
+            if resp.status_code == 200:
+                models = resp.json().get("data", [])
+                if models:
+                    model_name = models[0]["id"]
+                    os.environ["DRYDOCK_LOCAL_MODEL"] = model_name
+                    rprint(f"[green]Using local model: {model_name} at {args.local}[/]")
+        except Exception:
+            os.environ["DRYDOCK_LOCAL_MODEL"] = "local"
+            rprint(f"[yellow]Using local server at {args.local} (couldn't detect model name)[/]")
+
+    is_interactive = args.prompt is None
+    if is_interactive:
+        check_and_resolve_trusted_folder()
+    init_harness_files_manager("user", "project")
+
+    from drydock.cli.cli import run_cli
+
+    run_cli(args)
+
+
+if __name__ == "__main__":
+    main()