llamactl 0.2.7a1__py3-none-any.whl → 0.3.0__py3-none-any.whl

llama_deploy/cli/commands/deployment.py

@@ -0,0 +1,300 @@
+"""CLI commands for managing LlamaDeploy deployments.
+
+This command group lets you list, create, edit, refresh, and delete deployments.
+A deployment points the control plane at your Git repository and deployment file
+(e.g., `llama_deploy.yaml`). The control plane pulls your code at the selected
+git ref, reads the config, and runs your app.
+"""
+
+import asyncio
+
+import click
+import questionary
+from llama_deploy.cli.commands.auth import validate_authenticated_profile
+from llama_deploy.cli.styles import HEADER_COLOR, MUTED_COL, PRIMARY_COL, WARNING
+from llama_deploy.core.schema.deployments import DeploymentUpdate
+from rich import print as rprint
+from rich.table import Table
+from rich.text import Text
+
+from ..app import app, console
+from ..client import get_project_client
+from ..interactive_prompts.session_utils import (
+    is_interactive_session,
+)
+from ..interactive_prompts.utils import (
+    confirm_action,
+)
+from ..options import global_options, interactive_option
+from ..textual.deployment_form import create_deployment_form, edit_deployment_form
+from ..textual.deployment_monitor import monitor_deployment_screen
+
+
+@app.group(
+    help="Deploy your app to the cloud.",
+    no_args_is_help=True,
+)
+@global_options
+def deployments() -> None:
+    """Manage deployments"""
+    pass
+
+
+# Deployments commands
+@deployments.command("list")
+@global_options
+@interactive_option
+def list_deployments(interactive: bool) -> None:
+    """List deployments for the configured project."""
+    validate_authenticated_profile(interactive)
+    try:
+        client = get_project_client()
+        deployments = asyncio.run(client.list_deployments())
+
+        if not deployments:
+            rprint(
+                f"[{WARNING}]No deployments found for project {client.project_id}[/]"
+            )
+            return
+
+        table = Table(show_edge=False, box=None, header_style=f"bold {HEADER_COLOR}")
+        table.add_column("Name", style=PRIMARY_COL)
+        table.add_column("Status", style=MUTED_COL)
+        table.add_column("URL", style=MUTED_COL)
+        table.add_column("Repository", style=MUTED_COL)
+
+        for deployment in deployments:
+            name = deployment.id
+            status = deployment.status
+            repo_url = deployment.repo_url
+            gh = "https://github.com/"
+            if repo_url.startswith(gh):
+                repo_url = "gh:" + repo_url.removeprefix(gh)
+
+            table.add_row(
+                name,
+                status,
+                str(deployment.apiserver_url or ""),
+                repo_url,
+            )
+
+        console.print(table)
+
+    except Exception as e:
+        rprint(f"[red]Error: {e}[/red]")
+        raise click.Abort()
+
+
+@deployments.command("get")
+@global_options
+@click.argument("deployment_id", required=False)
+@interactive_option
+def get_deployment(deployment_id: str | None, interactive: bool) -> None:
+    """Get details of a specific deployment"""
+    validate_authenticated_profile(interactive)
+    try:
+        client = get_project_client()
+
+        deployment_id = select_deployment(deployment_id)
+        if not deployment_id:
+            rprint(f"[{WARNING}]No deployment selected[/]")
+            return
+        if interactive:
+            monitor_deployment_screen(deployment_id)
+            return
+
+        deployment = asyncio.run(client.get_deployment(deployment_id))
+
+        table = Table(show_edge=False, box=None, header_style=f"bold {HEADER_COLOR}")
+        table.add_column("Property", style=MUTED_COL, justify="right")
+        table.add_column("Value", style=PRIMARY_COL)
+
+        table.add_row("ID", Text(deployment.id))
+        table.add_row("Project ID", Text(deployment.project_id))
+        table.add_row("Status", Text(deployment.status))
+        table.add_row("Repository", Text(deployment.repo_url))
+        table.add_row("Deployment File", Text(deployment.deployment_file_path))
+        table.add_row("Git Ref", Text(deployment.git_ref or "-"))
+
+        apiserver_url = deployment.apiserver_url
+        table.add_row(
+            "API Server URL",
+            Text(str(apiserver_url) if apiserver_url else "-"),
+        )
+
+        secret_names = deployment.secret_names or []
+        table.add_row("Secrets", Text("\n".join(secret_names), style="italic"))
+
+        console.print(table)
+
+    except Exception as e:
+        rprint(f"[red]Error: {e}[/red]")
+        raise click.Abort()
+
+
+@deployments.command("create")
+@global_options
+@interactive_option
+def create_deployment(
+    interactive: bool,
+) -> None:
+    """Interactively create a new deployment"""
+
+    if not interactive:
+        raise click.ClickException(
+            "This command requires an interactive session. Run in a terminal or provide required arguments explicitly."
+        )
+    validate_authenticated_profile(interactive)
+
+    # Use interactive creation
+    deployment_form = create_deployment_form()
+    if deployment_form is None:
+        rprint(f"[{WARNING}]Cancelled[/]")
+        return
+
+    rprint(
+        f"[green]Created deployment: {deployment_form.name} (id: {deployment_form.id})[/green]"
+    )
+
+
+@deployments.command("delete")
+@global_options
+@click.argument("deployment_id", required=False)
+@interactive_option
+def delete_deployment(deployment_id: str | None, interactive: bool) -> None:
+    """Delete a deployment"""
+    validate_authenticated_profile(interactive)
+    try:
+        client = get_project_client()
+
+        deployment_id = select_deployment(deployment_id, interactive=interactive)
+        if not deployment_id:
+            rprint(f"[{WARNING}]No deployment selected[/]")
+            return
+
+        if interactive:
+            if not confirm_action(f"Delete deployment '{deployment_id}'?"):
+                rprint(f"[{WARNING}]Cancelled[/]")
+                return
+
+        asyncio.run(client.delete_deployment(deployment_id))
+        rprint(f"[green]Deleted deployment: {deployment_id}[/green]")
+
+    except Exception as e:
+        rprint(f"[red]Error: {e}[/red]")
+        raise click.Abort()
+
+
+@deployments.command("edit")
+@global_options
+@click.argument("deployment_id", required=False)
+@interactive_option
+def edit_deployment(deployment_id: str | None, interactive: bool) -> None:
+    """Interactively edit a deployment"""
+    validate_authenticated_profile(interactive)
+    try:
+        client = get_project_client()
+
+        deployment_id = select_deployment(deployment_id, interactive=interactive)
+        if not deployment_id:
+            rprint(f"[{WARNING}]No deployment selected[/]")
+            return
+
+        # Get current deployment details
+        current_deployment = asyncio.run(client.get_deployment(deployment_id))
+
+        # Use the interactive edit form
+        updated_deployment = edit_deployment_form(current_deployment)
+        if updated_deployment is None:
+            rprint(f"[{WARNING}]Cancelled[/]")
+            return
+
+        rprint(
+            f"[green]Successfully updated deployment: {updated_deployment.name}[/green]"
+        )
+
+    except Exception as e:
+        rprint(f"[red]Error: {e}[/red]")
+        raise click.Abort()
+
+
+@deployments.command("update")
+@global_options
+@click.argument("deployment_id", required=False)
+@interactive_option
+def refresh_deployment(deployment_id: str | None, interactive: bool) -> None:
+    """Update the deployment, pulling the latest code from it's branch"""
+    validate_authenticated_profile(interactive)
+    try:
+        deployment_id = select_deployment(deployment_id)
+        if not deployment_id:
+            rprint(f"[{WARNING}]No deployment selected[/]")
+            return
+
+        # Get current deployment details to show what we're refreshing
+        current_deployment = asyncio.run(
+            get_project_client().get_deployment(deployment_id)
+        )
+        deployment_name = current_deployment.name
+        old_git_sha = current_deployment.git_sha or ""
+
+        # Create an empty update to force git SHA refresh with spinner
+        with console.status(f"Refreshing {deployment_name}..."):
+            deployment_update = DeploymentUpdate()
+            updated_deployment = asyncio.run(
+                get_project_client().update_deployment(
+                    deployment_id,
+                    deployment_update,
+                )
+            )
+
+        # Show the git SHA change with short SHAs
+        new_git_sha = updated_deployment.git_sha or ""
+        old_short = old_git_sha[:7] if old_git_sha else "none"
+        new_short = new_git_sha[:7] if new_git_sha else "none"
+
+        if old_git_sha == new_git_sha:
+            rprint(f"No changes: already at {new_short}")
+        else:
+            rprint(f"Updated: {old_short} → {new_short}")
+
+    except Exception as e:
+        rprint(f"[red]Error: {e}[/red]")
+        raise click.Abort()
+
+
+def select_deployment(
+    deployment_id: str | None = None, interactive: bool = is_interactive_session()
+) -> str | None:
+    """
+    Select a deployment interactively if ID not provided.
+    Returns the selected deployment ID or None if cancelled.
+
+    In non-interactive sessions, returns None if deployment_id is not provided.
+    """
+    if deployment_id:
+        return deployment_id
+
+    # Don't attempt interactive selection in non-interactive sessions
+    if not interactive:
+        return None
+
+    client = get_project_client()
+    deployments = asyncio.run(client.list_deployments())
+
+    if not deployments:
+        rprint(f"[{WARNING}]No deployments found for project {client.project_id}[/]")
+        return None
+
+    choices = []
+    for deployment in deployments:
+        name = deployment.name
+        deployment_id = deployment.id
+        status = deployment.status
+        choices.append(
+            questionary.Choice(
+                title=f"{name} ({deployment_id}) - {status}", value=deployment_id
+            )
+        )
+
+    return questionary.select("Select deployment:", choices=choices).ask()

llama_deploy/cli/commands/env.py

@@ -0,0 +1,211 @@
+from importlib import metadata as importlib_metadata
+
+import click
+import questionary
+from llama_deploy.cli.config.schema import Environment
+from llama_deploy.cli.styles import (
+    ACTIVE_INDICATOR,
+    HEADER_COLOR,
+    MUTED_COL,
+    PRIMARY_COL,
+    WARNING,
+)
+from packaging import version as packaging_version
+from rich import print as rprint
+from rich.table import Table
+from rich.text import Text
+
+from ..app import console
+from ..config.env_service import service
+from ..options import global_options, interactive_option
+from .auth import auth
+
+
+@auth.group(
+    name="env",
+    help="Manage environments (control plane API URLs)",
+    no_args_is_help=True,
+)
+@global_options
+def env_group() -> None:
+    pass
+
+
+@env_group.command("list")
+@global_options
+def list_environments_cmd() -> None:
+    try:
+        envs = service.list_environments()
+        current_env = service.get_current_environment()
+
+        if not envs:
+            rprint(f"[{WARNING}]No environments found[/]")
+            return
+
+        table = Table(show_edge=False, box=None, header_style=f"bold {HEADER_COLOR}")
+        table.add_column("  API URL", style=PRIMARY_COL)
+        table.add_column("Requires Auth", style=MUTED_COL)
+
+        for env in envs:
+            text = Text()
+            if env == current_env:
+                text.append("* ", style=ACTIVE_INDICATOR)
+            else:
+                text.append("  ")
+            text.append(env.api_url)
+            table.add_row(text, Text("true" if env.requires_auth else "false"))
+
+        console.print(table)
+    except Exception as e:
+        rprint(f"[red]Error: {e}[/red]")
+        raise click.Abort()
+
+
+@env_group.command("add")
+@click.argument("api_url", required=False)
+@interactive_option
+@global_options
+def add_environment_cmd(api_url: str | None, interactive: bool) -> None:
+    try:
+        if not api_url:
+            if not interactive:
+                raise click.ClickException("API URL is required when not interactive")
+            current_env = service.get_current_environment()
+            entered = questionary.text(
+                "Enter control plane API URL", default=current_env.api_url
+            ).ask()
+            if not entered:
+                rprint(f"[{WARNING}]No environment entered[/]")
+                return
+            api_url = entered.strip()
+
+        api_url = api_url.rstrip("/")
+        env = service.probe_environment(api_url)
+        service.create_or_update_environment(env)
+        rprint(
+            f"[green]Added environment[/green] {env.api_url} (requires_auth={env.requires_auth}, min_llamactl_version={env.min_llamactl_version or '-'})."
+        )
+        _maybe_warn_min_version(env.min_llamactl_version)
+    except Exception as e:
+        rprint(f"[red]Failed to add environment: {e}[/red]")
+        raise click.Abort()
+
+
+@env_group.command("delete")
+@click.argument("api_url", required=False)
+@interactive_option
+@global_options
+def delete_environment_cmd(api_url: str | None, interactive: bool) -> None:
+    try:
+        if not api_url:
+            if not interactive:
+                raise click.ClickException("API URL is required when not interactive")
+            result = _select_environment(
+                service.list_environments(),
+                service.get_current_environment(),
+                "Select environment to delete",
+            )
+
+            if not result:
+                rprint(f"[{WARNING}]No environment selected[/]")
+                return
+            api_url = result.api_url
+
+        api_url = api_url.rstrip("/")
+        deleted = service.delete_environment(api_url)
+        if not deleted:
+            raise click.ClickException(f"Environment '{api_url}' not found")
+        rprint(
+            f"[green]Deleted environment[/green] {api_url} and all associated profiles"
+        )
+    except Exception as e:
+        rprint(f"[red]Failed to delete environment: {e}[/red]")
+        raise click.Abort()
+
+
+@env_group.command("switch")
+@click.argument("api_url", required=False)
+@interactive_option
+@global_options
+def switch_environment_cmd(api_url: str | None, interactive: bool) -> None:
+    try:
+        selected_url = api_url
+
+        if not selected_url and interactive:
+            result = _select_environment(
+                service.list_environments(),
+                service.get_current_environment(),
+                "Select environment",
+            )
+            if not result:
+                rprint(f"[{WARNING}]No environment selected[/]")
+                return
+            selected_url = result.api_url
+
+        if not selected_url:
+            if interactive:
+                rprint(f"[{WARNING}]No environment selected[/]")
+                return
+            raise click.ClickException("API URL is required when not interactive")
+
+        selected_url = selected_url.rstrip("/")
+
+        # Ensure environment exists and switch
+        env = service.switch_environment(selected_url)
+        try:
+            env = service.auto_update_env(env)
+        except Exception as e:
+            rprint(f"[{WARNING}]Failed to resolve environment: {e}[/]")
+            return
+        service.current_auth_service().select_any_profile()
+        rprint(f"[green]Switched to environment[/green] {env.api_url}")
+        _maybe_warn_min_version(env.min_llamactl_version)
+    except Exception as e:
+        rprint(f"[red]Failed to switch environment: {e}[/red]")
+        raise click.Abort()
+
+
+def _get_cli_version() -> str | None:
+    try:
+        return importlib_metadata.version("llamactl")
+    except Exception:
+        return None
+
+
+def _maybe_warn_min_version(min_required: str | None) -> None:
+    if not min_required:
+        return
+    current = _get_cli_version()
+    if not current:
+        return
+    try:
+        if packaging_version.parse(current) < packaging_version.parse(min_required):
+            rprint(
+                f"[{WARNING}]Warning:[/] This environment requires llamactl >= [bold]{min_required}[/bold], you have [bold]{current}[/bold]."
+            )
+    except Exception:
+        # If packaging is not available or parsing fails, skip strict comparison
+        pass
+
+
+def _select_environment(
+    envs: list[Environment],
+    current_env: Environment,
+    message: str = "Select environment",
+) -> Environment | None:
+    envs = service.list_environments()
+    current_env = service.get_current_environment()
+    if not envs:
+        raise click.ClickException(
+            "No environments found. This is a bug and shouldn't happen."
+        )
+    return questionary.select(
+        message,
+        choices=[
+            questionary.Choice(
+                title=f"{env.api_url} {'(current)' if env.api_url == current_env.api_url else ''}",
+                value=env,
+            )
+            for env in envs
+        ],
+    ).ask()