Merge pull request #111 from WecoAI/dev

Add run sharing + run credit cost lookup + bump version (0.3.14)

Author: aliroberts

pyproject.toml

@@ -8,7 +8,7 @@ name = "weco"
 authors = [{ name = "Weco AI Team", email = "contact@weco.ai" }]
 description = "Documentation for `weco`, a CLI for using Weco AI's code optimizer."
 readme = "README.md"
-version = "0.3.13"
+version = "0.3.14"
 license = { file = "LICENSE" }
 requires-python = ">=3.8"
 dependencies = [

weco/api.py

@@ -450,3 +450,33 @@ def submit_execution_result(
         return None
     except Exception:
         return None
+
+
+# --- Share API Functions ---
+
+
+def create_share_link(
+    console: Console, run_id: str, auth_headers: dict = {}, timeout: Union[int, Tuple[int, int]] = (5, 30)
+) -> Optional[str]:
+    """Create a public share link for a run.
+
+    Args:
+        console: Rich console for output.
+        run_id: The run ID to share.
+        auth_headers: Authentication headers.
+        timeout: Request timeout.
+
+    Returns:
+        The share ID if successful, or None on failure.
+    """
+    try:
+        response = requests.post(f"{__base_url__}/runs/{run_id}/share", headers=auth_headers, timeout=timeout)
+        response.raise_for_status()
+        result = response.json()
+        return result.get("share_id")
+    except requests.exceptions.HTTPError as e:
+        handle_api_error(e, console)
+        return None
+    except Exception as e:
+        console.print(f"[bold red]Error creating share link: {e}[/]")
+        return None

weco/cli.py

@@ -182,6 +182,12 @@ def _parse_credit_amount(value: str) -> float:
         help="Amount of credits to purchase (minimum 2, defaults to 10)",
     )
 
+    # Credits cost command
+    cost_parser = credits_subparsers.add_parser("cost", help="Check credit spend for a run")
+    cost_parser.add_argument(
+        "run_id", type=str, help="The run ID to check credit spend for (e.g., '0002e071-1b67-411f-a514-36947f0c4b31')"
+    )
+
     # Credits autotopup command
     autotopup_parser = credits_subparsers.add_parser("autotopup", help="Configure automatic top-up")
     autotopup_parser.add_argument("--enable", action="store_true", help="Enable automatic top-up")
@@ -387,6 +393,21 @@ def _main() -> None:
     )
     configure_resume_parser(resume_parser)
 
+    # --- Share Command Parser Setup ---
+    share_parser = subparsers.add_parser(
+        "share", help="Create a public share link for a run", formatter_class=argparse.RawTextHelpFormatter
+    )
+    share_parser.add_argument(
+        "run_id", type=str, help="The UUID of the run to share (e.g., '0002e071-1b67-411f-a514-36947f0c4b31')"
+    )
+    share_parser.add_argument(
+        "--output",
+        type=str,
+        choices=["rich", "plain"],
+        default="rich",
+        help="Output mode: 'rich' for interactive terminal UI (default), 'plain' for machine-readable text output suitable for LLM agents.",
+    )
+
     # --- Setup Command Parser Setup ---
     setup_parser = subparsers.add_parser("setup", help="Set up Weco for use with AI tools")
     configure_setup_parser(setup_parser)
@@ -426,6 +447,11 @@ def _main() -> None:
         sys.exit(0)
     elif args.command == "resume":
         execute_resume_command(args)
+    elif args.command == "share":
+        from .share import handle_share_command
+
+        handle_share_command(run_id=args.run_id, output_mode=args.output, console=console)
+        sys.exit(0)
     elif args.command == "setup":
         from .setup import handle_setup_command
 

weco/credits.py

@@ -26,6 +26,8 @@ def handle_credits_command(args, console: Console) -> None:
 
     if args.credits_command == "balance" or args.credits_command is None:
         check_balance(console, auth_headers)
+    elif args.credits_command == "cost":
+        check_run_cost(console, auth_headers, args.run_id)
     elif args.credits_command == "topup":
         topup_credits(console, auth_headers, args.amount)
     elif args.credits_command == "autotopup":
@@ -65,6 +67,45 @@ def check_balance(console: Console, auth_headers: dict) -> None:
         console.print(f"[bold red]Unexpected error: {e}[/]")
 
 
+def check_run_cost(console: Console, auth_headers: dict, run_id: str) -> None:
+    """Check and display credit spend for a specific run."""
+    try:
+        response = requests.get(f"{__base_url__}/billing/run/{run_id}/cost", headers=auth_headers, timeout=10)
+        response.raise_for_status()
+        data = response.json()
+
+        total = data.get("credits_spent_total", 0)
+        steps = data.get("steps", [])
+
+        table = Table(title=f"Credit Spend for Run {run_id}", show_header=True, header_style="bold cyan")
+        table.add_column("Step", style="dim", justify="right")
+        table.add_column("Node ID", style="dim")
+        table.add_column("Cost", style="green", justify="right")
+
+        for step in steps:
+            step_num = str(step.get("step", "-"))
+            node_id = step.get("node_id", "-") or "-"
+            credits = step.get("credits_spent", 0)
+            table.add_row(step_num, node_id, f"${credits:.2f}")
+
+        table.add_section()
+        table.add_row("", "[bold]Total[/]", f"[bold]${total:.2f}[/]")
+
+        console.print(table)
+
+    except requests.exceptions.HTTPError as e:
+        if e.response.status_code == 401:
+            console.print("[bold red]Authentication failed. Please log in again with 'weco login'.[/]")
+        elif e.response.status_code == 403:
+            console.print("[bold red]Permission denied. You do not own this run.[/]")
+        elif e.response.status_code == 404:
+            console.print(f"[bold red]Run not found: {run_id}[/]")
+        else:
+            console.print(f"[bold red]Error checking run cost: {e}[/]")
+    except Exception as e:
+        console.print(f"[bold red]Unexpected error: {e}[/]")
+
+
 def topup_credits(console: Console, auth_headers: dict, amount: float) -> None:
     """Initiate a credit top-up via Stripe."""
     try:

weco/setup.py

@@ -14,7 +14,7 @@
 from urllib.request import urlopen
 
 from rich.console import Console
-from rich.prompt import Confirm
+from rich.prompt import Confirm, Prompt
 
 from .events import (
     send_event,
@@ -512,56 +512,78 @@ def setup_cursor(console: Console, local_path: pathlib.Path | None = None) -> No
 SETUP_HANDLERS = {"claude-code": setup_claude_code, "cursor": setup_cursor}
 
 
-def handle_setup_command(args, console: Console) -> None:
-    """Handle the setup command with its subcommands."""
-    available_tools = ", ".join(SETUP_HANDLERS)
-    ctx = create_event_context()
+def prompt_tool_selection(console: Console) -> list[str]:
+    """Prompt the user to select which tool(s) to set up.
 
-    if args.tool is None:
-        console.print("[bold red]Error:[/] Please specify a tool to set up.")
-        console.print(f"Available tools: {available_tools}")
-        console.print("\nUsage: weco setup <tool>")
-        sys.exit(1)
+    Returns:
+        List of tool names to set up.
+    """
+    tool_names = list(SETUP_HANDLERS.keys())
+    all_option = len(tool_names) + 1
 
-    handler = SETUP_HANDLERS.get(args.tool)
-    if handler is None:
-        console.print(f"[bold red]Error:[/] Unknown tool: {args.tool}")
-        console.print(f"Available tools: {available_tools}")
-        sys.exit(1)
+    console.print("\n[bold cyan]Available tools to set up:[/]\n")
+    for i, name in enumerate(tool_names, 1):
+        console.print(f"  {i}. {name}")
+    console.print(f"  {all_option}. All of the above")
+
+    valid_choices = [str(i) for i in range(1, all_option + 1)]
+    choice = Prompt.ask("\n[bold]Select an option[/]", choices=valid_choices, show_choices=True)
+
+    idx = int(choice)
+    if idx == all_option:
+        return tool_names
+    return [tool_names[idx - 1]]
 
-    # Extract local path if provided
-    local_path = None
-    if hasattr(args, "local") and args.local:
-        local_path = pathlib.Path(args.local).expanduser().resolve()
 
-    # Determine source type for event reporting
+def _run_setup_for_tool(tool: str, console: Console, local_path: pathlib.Path | None, ctx) -> None:
+    """Run setup for a single tool with event tracking and error handling."""
     source = "local" if local_path else "download"
 
-    # Send skill install started event
-    send_event(SkillInstallStartedEvent(tool=args.tool, source=source), ctx)
+    send_event(SkillInstallStartedEvent(tool=tool, source=source), ctx)
 
     start_time = time.time()
 
     try:
+        handler = SETUP_HANDLERS[tool]
         handler(console, local_path=local_path)
 
-        # Send successful completion event
         duration_ms = int((time.time() - start_time) * 1000)
-        send_event(SkillInstallCompletedEvent(tool=args.tool, source=source, duration_ms=duration_ms), ctx)
+        send_event(SkillInstallCompletedEvent(tool=tool, source=source, duration_ms=duration_ms), ctx)
 
     except DownloadError as e:
-        # Send failure event
-        send_event(SkillInstallFailedEvent(tool=args.tool, source=source, error_type="download_error", stage="download"), ctx)
+        send_event(SkillInstallFailedEvent(tool=tool, source=source, error_type="download_error", stage="download"), ctx)
         console.print(f"[bold red]Error:[/] {e}")
         sys.exit(1)
     except SafetyError as e:
-        # Send failure event
-        send_event(SkillInstallFailedEvent(tool=args.tool, source=source, error_type="safety_error", stage="setup"), ctx)
+        send_event(SkillInstallFailedEvent(tool=tool, source=source, error_type="safety_error", stage="setup"), ctx)
         console.print(f"[bold red]Safety Error:[/] {e}")
         sys.exit(1)
     except (SetupError, FileNotFoundError, OSError, ValueError) as e:
-        # Send failure event
         error_type = type(e).__name__
-        send_event(SkillInstallFailedEvent(tool=args.tool, source=source, error_type=error_type, stage="setup"), ctx)
+        send_event(SkillInstallFailedEvent(tool=tool, source=source, error_type=error_type, stage="setup"), ctx)
         console.print(f"[bold red]Error:[/] {e}")
         sys.exit(1)
+
+
+def handle_setup_command(args, console: Console) -> None:
+    """Handle the setup command with its subcommands."""
+    ctx = create_event_context()
+
+    if args.tool is None:
+        selected_tools = prompt_tool_selection(console)
+    else:
+        handler = SETUP_HANDLERS.get(args.tool)
+        if handler is None:
+            available_tools = ", ".join(SETUP_HANDLERS)
+            console.print(f"[bold red]Error:[/] Unknown tool: {args.tool}")
+            console.print(f"Available tools: {available_tools}")
+            sys.exit(1)
+        selected_tools = [args.tool]
+
+    # Extract local path if provided
+    local_path = None
+    if hasattr(args, "local") and args.local:
+        local_path = pathlib.Path(args.local).expanduser().resolve()
+
+    for tool in selected_tools:
+        _run_setup_for_tool(tool, console, local_path, ctx)

weco/share.py

@@ -0,0 +1,43 @@
+"""Share command handler for the Weco CLI."""
+
+import sys
+from rich.console import Console
+
+from . import __dashboard_url__
+from .api import create_share_link
+from .auth import handle_authentication
+
+
+def handle_share_command(run_id: str, output_mode: str, console: Console) -> None:
+    """Handle the `weco share <run_id>` command.
+
+    Creates a public share link for the given run. Requires CLI sharing
+    to be enabled in the user's account settings on the dashboard.
+
+    Args:
+        run_id: UUID of the run to share.
+        output_mode: 'rich' or 'plain'.
+        console: Rich console instance for output.
+    """
+    # Authenticate
+    _, auth_headers = handle_authentication(console)
+    if not auth_headers:
+        sys.exit(1)
+
+    # Attempt to create the share link
+    share_id = create_share_link(console=console, run_id=run_id, auth_headers=auth_headers)
+
+    if share_id is None:
+        # Error message was already printed by create_share_link / handle_api_error
+        sys.exit(1)
+
+    share_url = f"{__dashboard_url__}/share/{share_id}"
+
+    if output_mode == "plain":
+        console.print(share_url)
+    else:
+        console.print()
+        console.print("[bold green]Share link created![/]")
+        console.print(f"[bold]{share_url}[/]")
+        console.print()
+        console.print("[dim]Anyone with this link can view the run's optimization progress and results.[/]")