devsper 2.1.6__py3-none-any.whl

devsper/cli/run_progress.py

@@ -0,0 +1,103 @@
+"""
+CLI run progress: read event log and return status line + parallel task info for live feedback.
+"""
+
+import os
+
+
+def read_run_status(
+    log_path: str | None,
+    worker_count: int = 0,
+) -> tuple[str, list[str]]:
+    """
+    Read event log and return (status_message, running_task_ids).
+    running_task_ids are task ids that have task_started but not yet task_completed.
+    worker_count: if > 0, show "up to N in parallel" in executor messages.
+    """
+    if not log_path or not os.path.isfile(log_path):
+        return "Starting…", []
+
+    events = []
+    try:
+        from devsper.types.event import Event
+
+        with open(log_path, "r") as f:
+            for line in f:
+                line = line.strip()
+                if not line:
+                    continue
+                try:
+                    events.append(Event.model_validate_json(line))
+                except Exception:
+                    continue
+    except Exception:
+        return "Running…", []
+
+    if not events:
+        return "Starting…", []
+
+    # task_id -> description
+    task_descriptions: dict[str, str] = {}
+    # track which tasks have started and which have completed
+    started: set[str] = set()
+    completed: set[str] = set()
+
+    for e in events:
+        payload = e.payload or {}
+        tid = (payload.get("task_id") or "").strip()
+        ev = getattr(e.type, "value", str(e.type))
+
+        if ev == "task_created" and tid:
+            task_descriptions[tid] = (payload.get("description") or "").strip()
+        elif ev in ("task_started", "agent_started") and tid:
+            started.add(tid)
+        elif ev in ("task_completed", "task_failed", "agent_finished") and tid:
+            completed.add(tid)
+
+    running_ids = sorted(started - completed)
+    total_tasks = len(task_descriptions)
+    completed_count = len(completed)
+
+    def _truncate(s: str, max_len: int = 52) -> str:
+        s = (s or "").strip()
+        if len(s) <= max_len:
+            return s
+        return s[: max_len - 1].rstrip() + "…"
+
+    last = events[-1]
+    ev_type = getattr(last.type, "value", str(last.type))
+    payload = last.payload or {}
+    task_id = (payload.get("task_id") or "").strip()
+    desc = task_descriptions.get(task_id)
+
+    if ev_type == "swarm_started":
+        return "Planning your request…", []
+    if ev_type == "planner_started":
+        return "Planning your request…", []
+    if ev_type == "task_created":
+        return f"Planned {total_tasks} task(s). Starting…", []
+    if ev_type == "planner_finished":
+        n = payload.get("subtask_count", total_tasks) or total_tasks
+        tail = f", up to {worker_count} in parallel" if worker_count > 1 else ""
+        return f"Executing {n} step(s){tail}…", []
+    if ev_type == "executor_started":
+        if total_tasks:
+            tail = f" (up to {worker_count} in parallel)" if worker_count > 1 else ""
+            return f"Executing step 1 of {total_tasks}…{tail}", []
+        return "Executing…", []
+    if ev_type in ("agent_started", "task_started"):
+        step_label = f"Step {completed_count + 1} of {total_tasks}: " if total_tasks else ""
+        part = (step_label + _truncate(desc)) if desc else (step_label or "Working on task…")
+        if len(running_ids) > 1:
+            part += f"  [parallel: {len(running_ids)} tasks]"
+        return part.rstrip(": ") or "Working…", running_ids
+    if ev_type in ("task_completed", "agent_finished"):
+        if total_tasks and completed_count < total_tasks:
+            return f"Finished step {completed_count} of {total_tasks}. Next…", running_ids
+        return f"Finished step {completed_count} of {total_tasks}. Assembling…", running_ids
+    if ev_type == "executor_finished":
+        return "Assembling result…", []
+    if ev_type == "swarm_finished":
+        return "Done.", []
+
+    return "Running…", running_ids

devsper/cli/ui/__init__.py

@@ -0,0 +1,65 @@
+"""
+CLI UI: theme, components, logging, progress, errors.
+All CLI code imports from devsper.cli.ui, never from rich directly.
+"""
+
+from devsper.cli.ui.theme import THEME, console, err_console, reconfigure_console
+from devsper.cli.ui.components import (
+    CostDisplay,
+    ErrorPanel,
+    devsperHeader,
+    RoleTag,
+    SectionHeader,
+    StatusBadge,
+    TaskRow,
+)
+from devsper.cli.ui.errors import (
+    ConfigNotFoundError,
+    devsperError,
+    ModelNotFoundError,
+    NoWorkersError,
+    ProviderConnectionError,
+    RedisConnectionError,
+    print_error,
+    print_unexpected_error,
+)
+from devsper.cli.ui.logging import get_logger, set_log_level, get_log_level, devsperLogger
+from devsper.cli.ui.progress import devsperProgress, progress_spinner_style
+from devsper.cli.ui.run_view import RunViewState, run_live_view, print_run_summary
+
+try:
+    from devsper.cli.ui.onboarding import run_init_wizard
+except ImportError:
+    run_init_wizard = None  # type: ignore[misc, assignment]
+
+__all__ = [
+    "CostDisplay",
+    "ConfigNotFoundError",
+    "ErrorPanel",
+    "devsperError",
+    "devsperHeader",
+    "devsperLogger",
+    "devsperProgress",
+    "ModelNotFoundError",
+    "NoWorkersError",
+    "ProviderConnectionError",
+    "RedisConnectionError",
+    "RoleTag",
+    "SectionHeader",
+    "StatusBadge",
+    "TaskRow",
+    "THEME",
+    "console",
+    "err_console",
+    "get_logger",
+    "get_log_level",
+    "print_error",
+    "print_unexpected_error",
+    "print_run_summary",
+    "progress_spinner_style",
+    "reconfigure_console",
+    "run_live_view",
+    "RunViewState",
+    "run_init_wizard",
+    "set_log_level",
+]

devsper/cli/ui/components.py

@@ -0,0 +1,94 @@
+"""
+Reusable Rich renderables for CLI. All CLI code imports from here, not rich directly.
+"""
+
+from rich.panel import Panel
+from rich.rule import Rule
+from rich.table import Table
+from rich.text import Text
+
+from devsper.cli.ui.theme import console
+
+# Status badge styles
+BADGE_STYLES = {
+    "SUCCESS": "hive.success",
+    "FAILED": "hive.error",
+    "RUNNING": "hive.secondary",
+    "SKIPPED": "hive.muted",
+    "CACHED": "hive.primary",
+    "PENDING": "hive.dim",
+}
+
+
+def devsperHeader(
+    version: str = "",
+    model: str = "",
+    workers: int = 0,
+) -> Text:
+    """Single-line wordmark + version. Example: ⬡ devsper  v2.1.5  ·  claude-sonnet-4  ·  4 workers"""
+    parts = [Text("⬡ devsper", style="hive.primary")]
+    if version:
+        parts.append(Text(f"  v{version}", style="hive.muted"))
+    if model:
+        parts.append(Text(f"  ·  {model}", style="hive.muted"))
+    if workers:
+        parts.append(Text(f"  ·  {workers} workers", style="hive.muted"))
+    out = Text()
+    for p in parts:
+        out.append_text(p)
+    return out
+
+
+def StatusBadge(text: str, style: str | None = None) -> Text:
+    """Inline colored badge: [SUCCESS], [FAILED], [RUNNING], etc."""
+    key = text.upper().replace(" ", "_")
+    s = style or BADGE_STYLES.get(key, "hive.muted")
+    return Text(f"[{text}]", style=s)
+
+
+def TaskRow(
+    short_id: str,
+    description: str,
+    role: str = "",
+    duration: str = "—",
+    status: str = "pending",
+    icon: str | None = None,
+) -> Text:
+    """Single-line task summary. status: pending|running|completed|failed|cached|skipped."""
+    if icon is None:
+        icon = {"pending": "○", "running": "◐", "completed": "✓", "failed": "✗", "cached": "⚡", "skipped": "⊘"}.get(status, "○")
+    style = {"completed": "hive.success", "failed": "hive.error", "cached": "hive.primary", "skipped": "hive.muted", "running": "hive.secondary", "pending": "hive.dim"}.get(status, "hive.dim")
+    desc = (description or "")[:45]
+    if len((description or "")) > 45:
+        desc = desc.rstrip() + "…"
+    role_part = f"  {role}" if role else ""
+    return Text().append(icon + " ", style=style).append(short_id + "  ", style="hive.secondary").append(desc, style="dim").append(role_part, style="hive.muted").append("  ", style="").append(duration, style="hive.muted")
+
+
+def RoleTag(role: str) -> Text:
+    """Colored inline tag per role: research->blue, code->teal, analysis->purple, critic->orange, architect->amber."""
+    styles = {"research": "hive.secondary", "code": "hive.tool", "analysis": "hive.agent", "critic": "hive.planner", "architect": "hive.primary"}
+    s = styles.get((role or "").lower(), "hive.muted")
+    return Text(role or "—", style=s)
+
+
+def CostDisplay(usd: float | None) -> Text:
+    """$0.0034 in pink, '—' in muted if None."""
+    if usd is None:
+        return Text("—", style="hive.muted")
+    return Text(f"${usd:.4f}", style="hive.cost")
+
+
+def SectionHeader(title: str) -> Rule:
+    """Amber rule with title."""
+    return Rule(title, style="hive.primary", characters="─")
+
+
+def ErrorPanel(message: str, hint: str | None = None, docs_url: str | None = None) -> Panel:
+    """Red-bordered panel, title 'Error'. No tracebacks to end users."""
+    body = message
+    if hint:
+        body += "\n\n" + f"Hint: {hint}"
+    if docs_url:
+        body += "\n" + f"Docs: {docs_url}"
+    return Panel(body, title="Error", border_style="hive.error")

devsper/cli/ui/errors.py

@@ -0,0 +1,104 @@
+"""
+Typed errors with actionable hints. No raw tracebacks shown to end users.
+"""
+
+from devsper.cli.ui.theme import err_console
+from devsper.cli.ui.components import ErrorPanel
+
+DOCS_BASE = "https://docs.devsper.com"
+REPO_ISSUES = "https://github.com/devsper-com/runtime/issues/new"
+
+
+class devsperError(Exception):
+    """Base CLI error with message, hint, docs, exit code."""
+
+    def __init__(
+        self,
+        message: str,
+        hint: str | None = None,
+        docs_url: str | None = None,
+        exit_code: int = 1,
+    ) -> None:
+        super().__init__(message)
+        self.message = message
+        self.hint = hint
+        self.docs_url = docs_url
+        self.exit_code = exit_code
+
+
+class ProviderConnectionError(devsperError):
+    def __init__(self, message: str, provider: str = "") -> None:
+        super().__init__(
+            message,
+            hint="Check your API key with: devsper credentials list",
+            docs_url=f"{DOCS_BASE}/providers" if provider else None,
+        )
+
+
+class ConfigNotFoundError(devsperError):
+    def __init__(self, message: str = "No configuration found.") -> None:
+        super().__init__(message, hint="Run devsper init to create a configuration")
+
+
+class RedisConnectionError(devsperError):
+    def __init__(self, message: str = "Cannot connect to Redis.") -> None:
+        super().__init__(
+            message,
+            hint="Start Redis with: docker run -p 6379:6379 redis:7-alpine",
+        )
+
+
+class NoWorkersError(devsperError):
+    def __init__(self, message: str = "No workers available.") -> None:
+        super().__init__(message, hint="Start a worker with: devsper node start --role worker")
+
+
+class ModelNotFoundError(devsperError):
+    def __init__(self, model_name: str) -> None:
+        super().__init__(
+            f"Model not found: {model_name}",
+            hint="Available models: devsper doctor | grep model",
+        )
+
+
+def print_error(e: devsperError, json_output: bool = False) -> None:
+    """Print error to stderr. If json_output, print {\"error\", \"hint\", \"exit_code\"}."""
+    if json_output:
+        import json
+        err_console.print(json.dumps({
+            "error": e.message,
+            "hint": e.hint,
+            "docs_url": getattr(e, "docs_url", None),
+            "exit_code": e.exit_code,
+        }))
+        return
+    panel = ErrorPanel(e.message, hint=e.hint, docs_url=getattr(e, "docs_url", None))
+    err_console.print(panel)
+
+
+def print_unexpected_error(exc: BaseException, json_output: bool = False) -> None:
+    """Condensed traceback (last 3 frames) + bug report hint."""
+    if json_output:
+        import json
+        err_console.print(json.dumps({
+            "error": str(exc),
+            "hint": "Unexpected error; see logs.",
+            "exit_code": 1,
+        }))
+        return
+    import traceback
+    tb_lines = traceback.format_exception(type(exc), exc, exc.__traceback__)
+    # Last 3 frames only
+    condensed = "".join(tb_lines[-3:]) if len(tb_lines) >= 3 else "".join(tb_lines)
+    from devsper.cli.ui.theme import THEME
+    from rich.panel import Panel
+    from rich.console import Console
+    c = Console(stderr=True, theme=THEME)
+    c.print(Panel(condensed.strip(), title="Traceback", border_style="hive.dim"))
+    c.print("\n[dim]This looks like a bug. Please report it:[/]")
+    c.print(f"  [link={REPO_ISSUES}]{REPO_ISSUES}[/]")
+    try:
+        import devsper
+        c.print(f"  devsper {getattr(devsper, '__version__', '?')}  Python {__import__('sys').version.split()[0]}  {__import__('platform').system()}")
+    except Exception:
+        pass

devsper/cli/ui/logging.py

@@ -0,0 +1,120 @@
+"""
+Structured logger matching tracing (Rust) compact format for visual consistency.
+All subsystems use get_logger(target) instead of logging.getLogger().
+"""
+
+import os
+import sys
+from datetime import datetime, timezone
+from typing import Any
+
+# Log level: 0=TRACE, 1=DEBUG, 2=INFO, 3=WARN, 4=ERROR. Controlled by env/global.
+_CLI_LOG_LEVEL = 2  # INFO default
+
+
+def set_log_level(level: str | int) -> None:
+    """level: 'trace'|'debug'|'info'|'warn'|'error' or 0-4."""
+    global _CLI_LOG_LEVEL
+    if isinstance(level, int):
+        _CLI_LOG_LEVEL = level
+        return
+    _CLI_LOG_LEVEL = {"trace": 0, "debug": 1, "info": 2, "warn": 3, "warning": 3, "error": 4}.get((level or "").lower(), 2)
+
+
+def get_log_level() -> int:
+    return _CLI_LOG_LEVEL
+
+
+# Target -> style for the target column
+TARGET_STYLES = {
+    "planner": "hive.planner",
+    "executor": "hive.secondary",
+    "agent": "hive.agent",
+    "tool": "hive.tool",
+    "memory": "hive.agent",
+    "scheduler": "hive.muted",
+    "bus": "hive.dim",
+    "swarm": "hive.primary",
+    "hitl": "hive.warning",
+}
+
+
+def _level_style(level: str) -> str:
+    if level == "WARN":
+        return "hive.warning"
+    if level == "ERROR":
+        return "hive.error"
+    if level == "DEBUG":
+        return "hive.muted"
+    if level == "TRACE":
+        return "hive.dim"
+    return ""
+
+
+def _format_ts() -> str:
+    return datetime.now(timezone.utc).strftime("%H:%M:%S.") + f"{datetime.now().microsecond // 1000:03d}"
+
+
+class devsperLogger:
+    """Logger that emits tracing-compatible compact lines: timestamp  LEVEL  target  message  key=val ..."""
+
+    def __init__(self, target: str, run_id_short: str = "") -> None:
+        self.target = target
+        self.run_id_short = (run_id_short or "")[:8]
+
+    def _emit(self, level: str, message: str, **fields: Any) -> None:
+        lvl_num = {"TRACE": 0, "DEBUG": 1, "INFO": 2, "WARN": 3, "ERROR": 4}.get(level, 2)
+        if lvl_num < _CLI_LOG_LEVEL:
+            return
+        ts = _format_ts()
+        target_style = TARGET_STYLES.get(self.target, "hive.muted")
+        level_style = _level_style(level)
+        parts: list[str] = [ts, "  ", level, "  ", self.target, "  ", message]
+        if fields:
+            pairs = "  ".join(f"{k}={v}" for k, v in fields.items())
+            parts.append("  ")
+            parts.append(pairs)
+        if self.run_id_short:
+            parts.append("  ")
+            parts.append(f"run_id={self.run_id_short}")
+        line = "".join(parts)
+        # Use Rich markup for colors when printing
+        try:
+            from devsper.cli.ui.theme import err_console
+            if level_style:
+                line = f"[{level_style}]{level}[/]  [{target_style}]{self.target}[/]  {message}"
+                if fields:
+                    line += "  " + "  ".join(f"[hive.muted]{k}={v}[/]" for k, v in fields.items())
+                if self.run_id_short:
+                    line += f"  [hive.muted]run_id={self.run_id_short}[/]"
+                line = f"[hive.muted]{ts}[/]  " + line
+            else:
+                line = f"[hive.muted]{ts}[/]  {level}  [{target_style}]{self.target}[/]  {message}"
+                if fields:
+                    line += "  " + "  ".join(f"[hive.muted]{k}={v}[/]" for k, v in fields.items())
+                if self.run_id_short:
+                    line += f"  [hive.muted]run_id={self.run_id_short}[/]"
+            err_console.print(line)
+        except Exception:
+            print(line, file=sys.stderr)
+
+    def trace(self, message: str, **fields: Any) -> None:
+        self._emit("TRACE", message, **fields)
+
+    def debug(self, message: str, **fields: Any) -> None:
+        self._emit("DEBUG", message, **fields)
+
+    def info(self, message: str, **fields: Any) -> None:
+        self._emit("INFO", message, **fields)
+
+    def warning(self, message: str, **fields: Any) -> None:
+        self._emit("WARN", message, **fields)
+
+    def error(self, message: str, **fields: Any) -> None:
+        self._emit("ERROR", message, **fields)
+
+
+def get_logger(target: str, run_id: str = "") -> devsperLogger:
+    """Return logger bound to target name. run_id shortened to 8 chars in output."""
+    short = (run_id or "")[:8]
+    return devsperLogger(target, short)

devsper/cli/ui/onboarding.py

@@ -0,0 +1,102 @@
+"""
+Init wizard: welcome screen, provider setup, model selection, workers, features, write config.
+Uses theme and components; --no-interactive writes minimal config and prints next steps.
+"""
+
+import os
+import sys
+from pathlib import Path
+
+from rich.panel import Panel
+from rich.text import Text
+
+from devsper.cli.ui.theme import console
+
+
+DOCS_URL = "https://docs.devsper.com"
+GITHUB_URL = "https://github.com/devsper-com/runtime"
+
+
+def _wordmark_ascii() -> str:
+    """6-line block letters 'devsper' in ASCII. Fallback if pyfiglet missing."""
+    try:
+        import pyfiglet
+        return pyfiglet.figlet_format("devsper", font="standard").rstrip()
+    except ImportError:
+        return """
+ _   _ _   _ __  __ _____  __  __ ___  _   _ ____
+| | | | | | |  \\/  | ____| |  \\/  | _ \\| \\ | |  _ \\
+| |_| | | | | |\\/| |  _|   | |\\/| | | | |  \\| | | | |
+|  _  | |_| | |  | | |___  | |  | | |_| | |\\  | |_| |
+|_| |_|\\___/|_|  |_|_____| |_|  |_|___/|_| \\_|____/
+""".strip()
+
+
+class InitWizard:
+    """Interactive init: welcome, providers, models, workers, features, write config."""
+
+    def __init__(self, cwd: Path | None = None) -> None:
+        self.cwd = cwd or Path.cwd()
+
+    def run(self, no_interactive: bool = False) -> int:
+        """Run full wizard or minimal write when no_interactive."""
+        if no_interactive:
+            return self._run_minimal()
+        return self._run_interactive()
+
+    def _run_minimal(self) -> int:
+        """Write minimal devsper.toml and print next steps."""
+        from devsper.cli.init import _build_init_toml
+        toml = _build_init_toml(workers=4, planner="auto", worker="auto")
+        path = self.cwd / "devsper.toml"
+        path.write_text(toml, encoding="utf-8")
+        console.print("[hive.success]✓[/] Wrote [cyan]devsper.toml[/]")
+        console.print()
+        console.print("Next steps:")
+        console.print("  1) Set API keys: [cyan]devsper credentials set <provider> api_key[/]")
+        console.print("  2) Run: [cyan]devsper run \"your task\"[/]")
+        console.print(f"  Docs: [link={DOCS_URL}]{DOCS_URL}[/]")
+        return 0
+
+    def _run_interactive(self) -> int:
+        """Step 1: Welcome. Then delegate to existing init flow."""
+        # Step 1 — Welcome
+        console.clear()
+        wordmark = _wordmark_ascii()
+        console.print(Text(wordmark, style="hive.primary"))
+        console.print()
+        console.print("Welcome to devsper — distributed AI swarm runtime", style="hive.muted")
+        try:
+            import devsper
+            ver = getattr(devsper, "__version__", "?")
+            console.print(f"v{ver}  ·  [link={DOCS_URL}]{DOCS_URL}[/]  ·  [link={GITHUB_URL}]{GITHUB_URL}[/]", style="hive.muted")
+        except Exception:
+            console.print(f"[link={DOCS_URL}]{DOCS_URL}[/]  ·  [link={GITHUB_URL}]{GITHUB_URL}[/]", style="hive.muted")
+        console.print()
+        try:
+            from rich.prompt import Prompt
+            Prompt.ask("[dim]Press Enter to continue[/]", default="")
+        except Exception:
+            input("Press Enter to continue...")
+        # Delegate to existing init (writes toml and .env)
+        from devsper.cli.init import _run_init_interactive, _write_env_file
+        toml_content, api_keys = _run_init_interactive(self.cwd)
+        path = self.cwd / "devsper.toml"
+        path.write_text(toml_content, encoding="utf-8")
+        if api_keys:
+            _write_env_file(self.cwd, api_keys)
+        console.print()
+        console.print(Panel(
+            "Try it:  [cyan]devsper run \"research quantum computing breakthroughs\"[/]\n\n"
+            f"Docs:    {DOCS_URL}\n"
+            "Discord: https://discord.gg/devsper",
+            title="You're all set",
+            border_style="hive.success",
+        ))
+        return 0
+
+
+def run_init_wizard(no_interactive: bool = False, cwd: Path | None = None) -> int:
+    """Entry point for devsper init."""
+    wizard = InitWizard(cwd=cwd)
+    return wizard.run(no_interactive=no_interactive)

devsper/cli/ui/progress.py

@@ -0,0 +1,43 @@
+"""
+Styled progress bars for long operations. Amber fill, dim track, consistent columns.
+"""
+
+from rich.progress import (
+    Progress,
+    SpinnerColumn,
+    TextColumn,
+    BarColumn,
+    TaskProgressColumn,
+    TimeElapsedColumn,
+)
+from rich.text import Text
+
+from devsper.cli.ui.theme import console
+
+# Spinner names: dots2, line, arrow (Rich built-in)
+SPINNER_THINKING = "dots2"
+SPINNER_NETWORK = "line"
+SPINNER_BUILDING = "arrow"
+SPINNER_SUCCESS = "dots2"  # we override with static ✓ when done
+
+
+def devsperProgress(
+    transient: bool = False,
+    console_use: object = None,
+) -> Progress:
+    """Progress with hive styling: amber bar, dim track, spinner, task count, elapsed."""
+    c = console_use or console
+    return Progress(
+        SpinnerColumn(style="hive.primary", finished_style="hive.success"),
+        TextColumn("[bold]{task.description}"),
+        BarColumn(bar_width=30, complete_style="hive.primary", finished_style="hive.success", pulse_style="hive.dim"),
+        TaskProgressColumn(style="hive.muted"),
+        TimeElapsedColumn(style="hive.muted"),
+        console=c,
+        transient=transient,
+    )
+
+
+def progress_spinner_style(operation: str) -> str:
+    """Spinner style by operation: thinking, network, building, success."""
+    return "hive.primary"