pisces-cli 0.1.0__py3-none-any.whl

pisces/config_cmd.py

@@ -0,0 +1,234 @@
+"""
+pisces.config_cmd
+~~~~~~~~~~~~~~~~~
+Implements `pisces config` — interactive configuration wizard.
+
+What it does:
+    - Shows current configuration (with --show)
+    - Interactively updates provider, model, repair, validation settings
+    - Validates that the chosen provider has a key available in the environment
+    - Never reads or writes API keys to pisces.toml
+"""
+
+from __future__ import annotations
+
+from pathlib import Path
+
+from rich import box
+from rich.prompt import Confirm, Prompt
+from rich.table import Table
+
+from pisces.config_manager import (
+    PROVIDER_DEFAULTS,
+    SUPPORTED_PROVIDERS,
+    PiscesConfig,
+    check_api_key_available,
+    load_config,
+    write_config,
+)
+from pisces.tui import (
+    console,
+    print_error,
+    print_section,
+    print_step_ok,
+    print_success,
+    print_warning,
+)
+
+# ─────────────────────────────────────────────────────────────
+# Entry point
+# ─────────────────────────────────────────────────────────────
+
+
+def run_config(project_root: Path, *, show_only: bool = False) -> int:
+    """
+    Entry point for `pisces config`.
+
+    Returns 0 on success, 1 on error.
+    """
+    print_section("pisces config")
+
+    # Load existing config (requires pisces init to have been run)
+    try:
+        cfg = load_config(project_root)
+    except RuntimeError as e:
+        print_error(str(e), hint="Run 'pisces init' first.")
+        return 1
+    except ValueError as e:
+        print_error(f"Invalid configuration: {e}")
+        return 1
+
+    if show_only:
+        _print_config_table(cfg, project_root)
+        return 0
+
+    # Interactive wizard
+    _print_config_table(cfg, project_root)
+    console.print()
+
+    try:
+        cfg = _run_wizard(cfg)
+    except KeyboardInterrupt:
+        console.print("\n[muted]Configuration cancelled.[/]")
+        return 0
+
+    # Save
+    try:
+        write_config(project_root, cfg)
+    except OSError as e:
+        print_error(f"Could not save configuration: {e}")
+        return 1
+
+    print_success("✓  Configuration saved to pisces.toml")
+
+    # Key check
+    _check_provider_key(cfg.llm.provider)
+
+    return 0
+
+
+# ─────────────────────────────────────────────────────────────
+# Current config display table
+# ─────────────────────────────────────────────────────────────
+
+
+def _print_config_table(cfg: PiscesConfig, project_root: Path) -> None:
+    """Print a rich table showing the current configuration."""
+    table = Table(
+        title="[bold]Current Configuration[/]",
+        title_justify="left",
+        box=box.ROUNDED,
+        border_style="blue",
+        header_style="bold cyan",
+        show_lines=False,
+        padding=(0, 2),
+    )
+    table.add_column("Setting", style="dim", min_width=22)
+    table.add_column("Value", style="white", min_width=30)
+    table.add_column("Source", style="muted", min_width=16)
+
+    # API key status (env var check)
+    has_key = check_api_key_available(cfg.llm.provider)
+    key_status = "[success]✓ Found in env[/]" if has_key else "[high]✗ Not set[/]"
+
+    table.add_row("llm.provider", f"[magenta]{cfg.llm.provider}[/]", "pisces.toml")
+    table.add_row("llm.model", f"[cyan]{cfg.llm.model}[/]", "pisces.toml")
+    table.add_row("API key", key_status, "env var")
+    table.add_row("repair.max_attempts", str(cfg.repair.max_attempts), "pisces.toml")
+    table.add_row(
+        "repair.require_approval",
+        "[success]true[/]" if cfg.repair.require_approval else "[high]false[/]",
+        "pisces.toml",
+    )
+    table.add_row(
+        "validation.test_command", f"[code]{cfg.validation.test_command}[/]", "pisces.toml"
+    )
+    table.add_row("Config file", str(project_root / "pisces.toml"), "—")
+
+    console.print()
+    console.print(table)
+    console.print()
+
+
+# ─────────────────────────────────────────────────────────────
+# Interactive wizard
+# ─────────────────────────────────────────────────────────────
+
+
+def _run_wizard(cfg: PiscesConfig) -> PiscesConfig:
+    """Run an interactive prompt-based config wizard. Returns updated cfg."""
+    console.print("[bold]Configure pisces[/] [dim](<press Enter to keep current value>)[/]")
+    console.print()
+
+    # ── LLM Provider ────────────────────────────────────────
+    console.print(f"[dim]Supported providers:[/] {', '.join(SUPPORTED_PROVIDERS)}")
+    provider = (
+        Prompt.ask(
+            "  [bold magenta]LLM provider[/]",
+            default=cfg.llm.provider,
+            console=console,
+        )
+        .lower()
+        .strip()
+    )
+
+    if provider not in SUPPORTED_PROVIDERS:
+        print_warning(f"Unknown provider '{provider}'. Keeping current: {cfg.llm.provider}")
+        provider = cfg.llm.provider
+
+    # ── Model ────────────────────────────────────────────────
+    default_model = PROVIDER_DEFAULTS.get(provider, cfg.llm.model)
+    if provider != cfg.llm.provider:
+        # Provider changed → suggest new provider's default model
+        console.print(f"  [dim]Default model for {provider}:[/] [cyan]{default_model}[/]")
+
+    model = Prompt.ask(
+        "  [bold cyan]Model name[/]",
+        default=cfg.llm.model if provider == cfg.llm.provider else default_model,
+        console=console,
+    ).strip()
+
+    # ── Max repair attempts ──────────────────────────────────
+    attempts_str = Prompt.ask(
+        "  [bold]Max self-healing attempts[/]",
+        default=str(cfg.repair.max_attempts),
+        console=console,
+    ).strip()
+    try:
+        max_attempts = int(attempts_str)
+        if max_attempts < 1:
+            raise ValueError
+    except ValueError:
+        print_warning(f"Invalid value '{attempts_str}'. Keeping: {cfg.repair.max_attempts}")
+        max_attempts = cfg.repair.max_attempts
+
+    # ── Require HITL approval ────────────────────────────────
+    console.print()
+    console.print("  [dim]Human approval (HITL) means pisces will ask before modifying source.[/]")
+    require_approval = Confirm.ask(
+        "  [bold]Require human approval before patching?[/]",
+        default=cfg.repair.require_approval,
+        console=console,
+    )
+
+    # ── Test command ─────────────────────────────────────────
+    console.print()
+    test_command = Prompt.ask(
+        "  [bold]Test command[/] [dim](<used to validate patches>)[/]",
+        default=cfg.validation.test_command,
+        console=console,
+    ).strip()
+
+    # ── Apply ────────────────────────────────────────────────
+    cfg.llm.provider = provider
+    cfg.llm.model = model
+    cfg.repair.max_attempts = max_attempts
+    cfg.repair.require_approval = require_approval
+    cfg.validation.test_command = test_command
+
+    return cfg
+
+
+# ─────────────────────────────────────────────────────────────
+# API key availability check (advisory only)
+# ─────────────────────────────────────────────────────────────
+
+
+def _check_provider_key(provider: str) -> None:
+    """Print a reminder if the provider's API key is not set."""
+    has_key = check_api_key_available(provider)
+    if has_key:
+        print_step_ok(f"API key for '{provider}' found in environment ✓")
+    else:
+        # Local providers don't need keys
+        if provider in ("ollama", "lmstudio"):
+            print_step_ok(f"'{provider}' is a local provider — no API key needed ✓")
+        else:
+            from pisces.config_manager import _PROVIDER_ENV_VARS
+
+            env_var = _PROVIDER_ENV_VARS.get(provider, ["?"])[0]
+            print_warning(
+                f"API key for '{provider}' not found.\n"
+                f"  Set it before running 'pisces fix':\n"
+                f"    export {env_var}=your_key_here"
+            )

pisces/config_manager.py

@@ -0,0 +1,208 @@
+"""loads, merges, and validates pisces configuration
+# pisces.toml should not and never contain sensitive config(eg: API key or maybe cerdentials)
+# it may contain things like say provider name, model, settings or models
+"""
+
+from __future__ import annotations
+
+import os
+import sys
+from dataclasses import dataclass, field
+from pathlib import Path
+from typing import Any
+
+if sys.version_info >= (3, 11):
+    import tomllib
+else:
+    try:
+        import tomli as tomllib
+    except ImportError as e:
+        raise ImportError("Install 'tomli' for Python < 3.11: pip install tomli") from e
+
+    """Environment Variable names for each of the provider"""
+_PROVIDER_ENV_VARS: dict[str, list[str]] = {
+    "openai": ["OPENAI_API_KEY"],
+    "anthropic": ["ANTHROPIC_API_KEY"],
+    "gemini": ["GOOGLE_API_KEY", "GEMINI_API_KEY"],
+    "groq": ["GROQ_API_KEY"],
+    "ollama": [],  # key not needed cos local
+    "lmstudio": [],  # key not needed cos local
+}
+
+# Supported provider and their default models
+PROVIDER_DEFAULTS: dict[str, str] = {
+    "openai": "gpt-4o-mini",
+    "anthropic": "claude-3-5-haiku-20241022",
+    "gemini": "gemini-2.0-flash",
+    "groq": "llama-3.3-70b-versatile",
+    "ollama": "llama3.2",
+    "lmstudio": "local-model",
+}
+
+SUPPORTED_PROVIDERS = list(PROVIDER_DEFAULTS.keys())
+
+"""Config Dataclasses"""
+
+
+@dataclass
+class LLMConfig:
+    provider: str = "groq"
+    model: str = "llama-3.3-70b-versatile"
+
+
+@dataclass
+class RepairConfig:
+    max_attempts: int = 3
+    require_approval: bool = True  # HITL always onn by default
+
+
+@dataclass
+class ValidationConfig:
+    test_command: str = "pytest"
+
+
+@dataclass
+class ScanConfig:
+    exclude: list[str] = field(
+        default_factory=lambda: [
+            ".git",
+            ".venv",
+            "venv",
+            "__pycache__",
+            "node_modules",
+            ".pisces",
+            "dist",
+            "build",
+        ]
+    )
+
+
+@dataclass
+class PiscesConfig:
+    llm: LLMConfig = field(default_factory=LLMConfig)
+    repair: RepairConfig = field(default_factory=RepairConfig)
+    validation: ValidationConfig = field(default_factory=ValidationConfig)
+    scan: ScanConfig = field(default_factory=ScanConfig)
+
+    _project_root: Path | None = field(default=None, repr=False, compare=False)
+
+    """Loader"""
+
+
+def load_config(project_root: Path) -> PiscesConfig:
+    """
+    Load configuration from *project_root*/pisces.toml.
+    Falls back to defaults for any missing keys.
+    Raises RuntimeError if pisces has not been initialized.
+    """
+    toml_path = project_root / "pisces.toml"
+
+    if not toml_path.exists():
+        raise RuntimeError(f"pisces.toml not found in {project_root}\nRun 'pisces init' first.")
+    with toml_path.open("rb") as f:
+        raw: dict[str, Any] = tomllib.load(f)
+
+    cfg = PiscesConfig()
+    cfg._project_root = project_root
+
+    # [llm]
+    llm = raw.get("llm", {})
+    if "provider" in llm:
+        cfg.llm.provider = llm["provider"].lower().strip()
+    if "model" in llm:
+        cfg.llm.model = llm["model"].strip()
+    # [repair]
+    repair = raw.get("repair", {})
+    if "max_attempts" in repair:
+        cfg.repair.max_attempts = int(repair["max_attempts"])
+    if "require_approval" in repair:
+        cfg.repair.require_approval = bool(repair["require_approval"])
+    # [validation]
+    validation = raw.get("validation", {})
+    if "test_command" in validation:
+        cfg.validation.test_command = validation["test_command"].strip()
+    # [scan]
+    scan = raw.get("scan", {})
+    if "exclude" in scan:
+        cfg.scan.exclude = [str(e) for e in scan["exclude"]]
+    _validate_config(cfg)
+    return cfg
+
+
+def _validate_config(cfg: PiscesConfig) -> None:
+    """Raise ValueError for clearly invalid config values."""
+    if cfg.llm.provider not in SUPPORTED_PROVIDERS:
+        raise ValueError(
+            f"Unsupported provider '{cfg.llm.provider}'. "
+            f"Choose from: {', '.join(SUPPORTED_PROVIDERS)}"
+        )
+    if cfg.repair.max_attempts < 1:
+        raise ValueError("repair.max_attempts must be >= 1")
+
+
+# ─────────────────────────────────────────────────────────────
+# Credential lookup — env vars ONLY
+# ─────────────────────────────────────────────────────────────
+def get_api_key(provider: str) -> str | None:
+    """
+    Return the API key for *provider* from environment variables.
+    Returns None for local providers (ollama, lmstudio) that need no key.
+    Raises EnvironmentError if the key is required but missing.
+    """
+    provider = provider.lower()
+    env_vars = _PROVIDER_ENV_VARS.get(provider, [])
+    if not env_vars:
+        return None  # local provider — no key needed
+    for var in env_vars:
+        value = os.environ.get(var, "").strip()
+        if value:
+            return value
+    raise OSError(
+        f"No API key found for provider '{provider}'.\n"
+        f"Set one of these environment variables: {', '.join(env_vars)}\n\n"
+        f"Example (Linux/macOS/WSL):\n"
+        f"  export {env_vars[0]}=your_key_here\n\n"
+        f"Example (Windows CMD):\n"
+        f"  set {env_vars[0]}=your_key_here\n\n"
+        f"Example (Windows PowerShell):\n"
+        f"  $env:{env_vars[0]}='your_key_here'"
+    )
+
+
+def check_api_key_available(provider: str) -> bool:
+    """Return True if an API key is available (or not required)."""
+    try:
+        get_api_key(provider)
+        return True
+    except OSError:
+        return False
+
+
+# ─────────────────────────────────────────────────────────────
+# Config writer (non-sensitive fields only)
+# ─────────────────────────────────────────────────────────────
+def write_config(project_root: Path, cfg: PiscesConfig) -> None:
+    """
+    Write *cfg* to pisces.toml.
+    NEVER writes credentials. Only writes provider name, model, and settings.
+    """
+    toml_path = project_root / "pisces.toml"
+    # Build exclude list as TOML inline array
+    excludes = ", ".join(f'"{e}"' for e in cfg.scan.exclude)
+    content = f"""\
+# pisces project configuration
+# API keys MUST come from environment variables — never put them here.
+[llm]
+provider = "{cfg.llm.provider}"
+model    = "{cfg.llm.model}"
+[repair]
+max_attempts     = {cfg.repair.max_attempts}
+require_approval = {"true" if cfg.repair.require_approval else "false"}
+[validation]
+test_command = "{cfg.validation.test_command}"
+[scan]
+exclude = [
+    {excludes}
+]
+"""
+    toml_path.write_text(content, encoding="utf-8")