alloy-runtime-cli 0.1.0__py3-none-any.whl

cli/commands/agents/update/presenter.py

@@ -0,0 +1,68 @@
+"""Presenter for agent update output."""
+
+from cli.commands.agents.update.fields import (
+    format_generation_params,
+    format_models,
+    format_tags,
+    format_tools,
+)
+from cli.infrastructure.formatting.fields import (
+    format_datetime,
+    format_optional,
+    format_uuid,
+)
+from cli.infrastructure.output import OutputService
+from cli.infrastructure.renderers.entity_renderer import FieldConfig
+from alloy_runtime_types.dtos.agents import UpdateAgentResponse
+
+
+def present_agent_update_success(response: UpdateAgentResponse) -> None:
+    """Present successful agent update.
+
+    Args:
+        response: Agent update response from server
+    """
+    output = OutputService.get()
+    output.success("Agent updated successfully")
+
+    fields = [
+        FieldConfig("id", "ID", lambda v: format_uuid(v, short=False)),
+        FieldConfig("name", "Name"),
+        FieldConfig("description", "Description", format_optional),
+        FieldConfig("models", "Models", lambda _: format_models(response)),
+        FieldConfig("tools", "Tools", lambda _: format_tools(response)),
+        FieldConfig("tags", "Tags", lambda _: format_tags(response)),
+        FieldConfig(
+            "generation_params",
+            "Generation Params",
+            lambda _: format_generation_params(response),
+        ),
+        FieldConfig(
+            "system_instruction_version_id",
+            "System Prompt ID",
+            lambda v: format_optional(v, lambda x: format_uuid(x, short=False)),
+        ),
+        FieldConfig(
+            "input_schema_id",
+            "Input Schema ID",
+            lambda v: format_optional(v, lambda x: format_uuid(x, short=False)),
+        ),
+        FieldConfig(
+            "output_schema_id",
+            "Output Schema ID",
+            lambda v: format_optional(v, lambda x: format_uuid(x, short=False)),
+        ),
+        FieldConfig(
+            "organization_id",
+            "Organization ID",
+            lambda v: format_optional(v, lambda x: format_uuid(x, short=True)),
+        ),
+        FieldConfig(
+            "user_id",
+            "User ID",
+            lambda v: format_optional(v, lambda x: format_uuid(x, short=True)),
+        ),
+        FieldConfig("updated_at", "Updated At", format_datetime),
+    ]
+
+    output.entity(response, f"Agent: {response.name}", fields)

cli/commands/audio/transcribe/command.py

@@ -0,0 +1,144 @@
+"""Audio transcribe command implementation."""
+
+import mimetypes
+import sys
+from pathlib import Path
+
+import typer
+
+from cli.commands.audio.transcribe.presenter import present_transcription
+from cli.infrastructure.command import async_command, authenticated_client
+
+MAX_FILE_SIZE_BYTES = 25 * 1024 * 1024  # 25 MB
+
+
+def resolve_audio_input(file: str) -> tuple[bytes, str, str]:
+    """Validate and read audio input from a file path or stdin.
+
+    Args:
+        file: File path to an audio file, or '-' for stdin.
+
+    Returns:
+        Tuple of (file_bytes, filename, content_type).
+
+    Raises:
+        typer.BadParameter: If the file is not found, not readable,
+            or exceeds the 25MB size limit.
+    """
+    if file == "-":
+        return _read_stdin()
+
+    return _read_file(file)
+
+
+def _read_stdin() -> tuple[bytes, str, str]:
+    """Read audio bytes from stdin.
+
+    Returns:
+        Tuple of (file_bytes, filename, content_type).
+
+    Raises:
+        typer.BadParameter: If stdin data exceeds size limit.
+    """
+    file_bytes = sys.stdin.buffer.read()
+    if len(file_bytes) > MAX_FILE_SIZE_BYTES:
+        size_mb = len(file_bytes) / (1024 * 1024)
+        raise typer.BadParameter(
+            f"Stdin data ({size_mb:.1f} MB) exceeds the 25 MB upload limit."
+        )
+    return file_bytes, "audio.wav", "audio/wav"
+
+
+def _read_file(file: str) -> tuple[bytes, str, str]:
+    """Read and validate an audio file from disk.
+
+    Args:
+        file: Path to the audio file.
+
+    Returns:
+        Tuple of (file_bytes, filename, content_type).
+
+    Raises:
+        typer.BadParameter: If the file is not found, not a regular file,
+            or exceeds the 25MB size limit.
+    """
+    path = Path(file).expanduser()
+
+    if not path.exists():
+        raise typer.BadParameter(f"File not found: {file}")
+
+    if not path.is_file():
+        raise typer.BadParameter(f"Not a file: {file}")
+
+    file_size = path.stat().st_size
+    if file_size > MAX_FILE_SIZE_BYTES:
+        size_mb = file_size / (1024 * 1024)
+        raise typer.BadParameter(
+            f"File ({size_mb:.1f} MB) exceeds the 25 MB upload limit."
+        )
+
+    file_bytes = path.read_bytes()
+    filename = path.name
+    content_type = _detect_content_type(path)
+
+    return file_bytes, filename, content_type
+
+
+def _detect_content_type(path: Path) -> str:
+    """Detect the MIME content type from the file extension.
+
+    Args:
+        path: Path to the file.
+
+    Returns:
+        MIME type string. Defaults to 'audio/wav' if detection fails.
+    """
+    guessed, _ = mimetypes.guess_type(str(path))
+    if guessed and guessed.startswith("audio/"):
+        return guessed
+    return "audio/wav"
+
+
+def audio_transcribe_command(
+    file: str = typer.Argument(
+        ...,
+        help="Path to audio file, or '-' for stdin.",
+    ),
+) -> None:
+    """Transcribe audio to text.
+
+    Accepts a local audio file or piped stdin. The file must be under 25 MB.
+    Content type is auto-detected from the file extension.
+
+    Examples:
+        ai audio transcribe recording.wav
+        ai audio transcribe ~/voice-memo.mp3
+        cat audio.wav | ai audio transcribe -
+    """
+    file_bytes, filename, content_type = resolve_audio_input(file)
+    _execute_transcribe(
+        file_bytes=file_bytes, filename=filename, content_type=content_type
+    )
+
+
+@async_command
+async def _execute_transcribe(
+    file_bytes: bytes,
+    filename: str,
+    content_type: str,
+) -> None:
+    """Execute the transcribe operation.
+
+    Args:
+        file_bytes: Raw audio bytes to upload.
+        filename: Name of the audio file.
+        content_type: MIME content type of the audio.
+    """
+    async with authenticated_client() as (_config, client):
+        response = await client.transcribe_audio(
+            file_bytes,
+            filename=filename,
+            content_type=content_type,
+        )
+
+    present_transcription(response)

cli/commands/audio/transcribe/presenter.py

@@ -0,0 +1,15 @@
+"""Presenter for audio transcribe command output."""
+
+from alloy_runtime_types.dtos.audio import AudioTranscriptResponse
+
+
+def present_transcription(response: AudioTranscriptResponse) -> None:
+    """Print transcribed text to stdout.
+
+    Output is piping-friendly: only the transcribed text goes to stdout
+    with no decoration or metadata.
+
+    Args:
+        response: Transcription response from the server.
+    """
+    print(response.text)

cli/commands/auth/login/command.py

@@ -0,0 +1,80 @@
+"""Login command implementation."""
+
+import typer
+
+from cli.infrastructure.auth_storage import save_credentials
+from cli.infrastructure.command import async_command
+from cli.infrastructure.config import get_api_url
+from cli.infrastructure.console import Confirm, Prompt, get_console
+from cli.infrastructure.output import print_info, print_success
+from cli.infrastructure.provider_setup import prompt_provider_credentials
+from alloy_runtime_sdk.api_client.client import ApiClient
+from alloy_runtime_types.dtos.auth import LoginRequest
+
+
+def login_command(
+    email: str | None = typer.Option(None, "-e", "--email", help="Email address"),
+    password: str | None = typer.Option(
+        None, "--password", help="Password (avoid: exposes in shell history)"
+    ),
+) -> None:
+    """Login and save API key.
+
+    First-time users will be prompted to set up provider credentials.
+
+    Examples:
+
+        alloy login
+        alloy login -e user@example.com
+    """
+    _execute_login(email, password)
+
+
+@async_command
+async def _execute_login(
+    email: str | None,
+    password: str | None,
+) -> None:
+    """Execute the login operation.
+
+    Args:
+        email: Optional email (will prompt if not provided)
+        password: Optional password (will prompt if not provided)
+    """
+    console = get_console()
+    api_url = get_api_url()
+
+    # Prompt for credentials if not provided
+    if not email:
+        email = Prompt.ask("[cyan]Email[/cyan]")
+
+    if not password:
+        password = Prompt.ask("[cyan]Password[/cyan]", password=True)
+
+    console.print("\n[blue]Authenticating...[/blue]")
+
+    # Create unauthenticated client
+    async with ApiClient(base_url=api_url, api_key="") as client:
+        response = await client.login(LoginRequest(email=email, password=password))
+
+    # Save credentials
+    save_credentials(api_url, response.api_key)
+
+    print_success(f"Logged in as {response.name} ({response.email})")
+    print_info("API key saved to ~/.alloy-runtime/config")
+    print_info(f"Organization: {response.organization_id}")
+
+    # Prompt for provider credentials if new user
+    if response.is_new_user:
+        console.print("\n[yellow]Welcome! This is your first login.[/yellow]")
+        if Confirm.ask(
+            "Would you like to set up AI provider credentials now?", default=True
+        ):
+            await prompt_provider_credentials(api_url, response.api_key)
+        else:
+            console.print(
+                "\n[dim]You can set up provider credentials later with:[/dim]"
+            )
+            console.print(
+                "[dim]  alloy credentials update <credential-id> --value <api-key>[/dim]"
+            )

cli/commands/auth/signup/command.py

@@ -0,0 +1,115 @@
+"""Signup command implementation."""
+
+import typer
+
+from cli.infrastructure.auth_storage import save_credentials
+from cli.infrastructure.command import async_command
+from cli.infrastructure.config import get_api_url
+from cli.infrastructure.console import Confirm, Prompt, get_console
+from cli.infrastructure.output import print_info, print_success
+from cli.infrastructure.provider_setup import prompt_provider_credentials
+from alloy_runtime_sdk.api_client.client import ApiClient
+from alloy_runtime_types.dtos.auth import SignupRequest
+
+
+def signup_command(
+    email: str | None = typer.Option(None, "-e", "--email", help="Email address"),
+    password: str | None = typer.Option(
+        None, "--password", help="Password (avoid: exposes in shell history)"
+    ),
+    name: str | None = typer.Option(None, "-n", "--name", help="Display name"),
+    org_name: str | None = typer.Option(
+        None,
+        "--org",
+        help="Organization name",
+    ),
+) -> None:
+    """Create a new Alloy Runtime account.
+
+    You'll be prompted to set up provider credentials after signup.
+
+    Examples:
+
+        ai signup
+        ai signup -e user@example.com -n "John Doe" --org "My Company"
+    """
+    _execute_signup(email, password, name, org_name)
+
+
+@async_command
+async def _execute_signup(
+    email: str | None,
+    password: str | None,
+    name: str | None,
+    org_name: str | None,
+) -> None:
+    """Execute the signup operation.
+
+    Args:
+        email: Optional email (will prompt if not provided)
+        password: Optional password (will prompt if not provided)
+        name: Optional name (will prompt if not provided)
+        org_name: Optional organization name
+    """
+    console = get_console()
+    api_url = get_api_url()
+
+    # Prompt for required fields if not provided
+    if not email:
+        email = Prompt.ask("[cyan]Email address[/cyan]")
+
+    if not name:
+        name = Prompt.ask("[cyan]Your name[/cyan]")
+
+    if not password:
+        while True:
+            password = Prompt.ask("[cyan]Password[/cyan] (min 8 chars)", password=True)
+            confirm = Prompt.ask("[cyan]Confirm password[/cyan]", password=True)
+
+            if password != confirm:
+                console.print("[red]Passwords do not match. Please try again.[/red]\n")
+                continue
+
+            if password and len(password) < 8:
+                console.print("[red]Password must be at least 8 characters.[/red]\n")
+                continue
+
+            break
+
+    console.print("\n[blue]Creating your account...[/blue]")
+
+    # Create unauthenticated client
+    async with ApiClient(base_url=api_url, api_key="") as client:
+        response = await client.signup(
+            SignupRequest(
+                email=email,
+                password=password,
+                name=name,
+                organization_name=org_name,
+            )
+        )
+
+    # Save credentials
+    save_credentials(api_url, response.api_key)
+
+    console.print()
+    print_success(f"Account created for {response.name}")
+    print_info(f"Email: {response.email}")
+    print_info(f"Organization ID: {response.organization_id}")
+    print_info("API key saved to ~/.alloy-runtime/config")
+
+    # Prompt for provider setup
+    console.print(
+        "\n[bold yellow]Let's set up your AI provider credentials[/bold yellow]"
+    )
+    console.print(
+        "[dim]You'll need API keys from providers like OpenAI, Anthropic, or Google.[/dim]\n"
+    )
+
+    if Confirm.ask("Set up provider credentials now?", default=True):
+        await prompt_provider_credentials(api_url, response.api_key)
+    else:
+        console.print("\n[dim]You can add provider credentials later with:[/dim]")
+        console.print(
+            "[dim]  ai credentials update <credential-id> --value <api-key>[/dim]"
+        )

cli/commands/billing/__init__.py

@@ -0,0 +1 @@
+# Empty per project convention - import directly from submodules

cli/commands/billing/costs/__init__.py

@@ -0,0 +1 @@
+# Empty per project convention - import directly from submodules

cli/commands/billing/costs/by_agent/__init__.py

@@ -0,0 +1 @@
+# Empty per project convention - import directly from submodules

cli/commands/billing/costs/by_agent/command.py

@@ -0,0 +1,57 @@
+"""Billing project costs by agent command implementation."""
+
+from typing import Optional
+from uuid import UUID
+
+import typer
+
+from cli.commands.billing.costs.by_agent.presenter import present_billing_costs_by_agent
+from cli.commands.flag_utils import validate_uuid
+from cli.infrastructure.command import async_command, authenticated_client
+
+
+def billing_costs_by_agent_command(
+    project_id: str = typer.Argument(
+        ...,
+        help="Billing project UUID",
+    ),
+    start_date: Optional[str] = typer.Option(
+        None,
+        "--from",
+        help="Start date (ISO 8601, defaults to 30 days ago)",
+    ),
+    end_date: Optional[str] = typer.Option(
+        None,
+        "--to",
+        help="End date (ISO 8601, defaults to now)",
+    ),
+) -> None:
+    """Get cost breakdown by agent for a billing project.
+
+    Examples:
+        ai billing costs by-agent 123e4567-89ab...
+        ai billing costs by-agent 123e4567-89ab... --from 2026-01-01 --to 2026-01-31
+    """
+    project_uuid = validate_uuid(project_id, "billing project")
+    _execute_costs_by_agent(
+        project_id=project_uuid,
+        start_date=start_date,
+        end_date=end_date,
+    )
+
+
+@async_command
+async def _execute_costs_by_agent(
+    project_id: UUID,
+    start_date: Optional[str],
+    end_date: Optional[str],
+) -> None:
+    """Execute get billing project costs by agent operation."""
+    async with authenticated_client() as (_config, client):
+        response = await client.get_billing_project_costs_by_agent(
+            project_id,
+            start_date=start_date,
+            end_date=end_date,
+        )
+
+    present_billing_costs_by_agent(response)

cli/commands/billing/costs/by_agent/presenter.py

@@ -0,0 +1,81 @@
+"""Presenter for billing project costs by agent command output."""
+
+from decimal import Decimal
+
+from alloy_runtime_types.dtos.billing import ProjectCostByAgent
+
+from cli.infrastructure.output import OutputService
+from cli.infrastructure.renderers.list_renderer import ColumnConfig
+
+
+def _format_cost(cost: Decimal | None) -> str:
+    """Format cost for display."""
+    if cost is None:
+        return "-"
+    return f"${cost:.4f}"
+
+
+def present_billing_costs_by_agent(entries: list[ProjectCostByAgent]) -> None:
+    """Present billing project cost breakdown by agent."""
+    output = OutputService.get()
+
+    if not entries:
+        output.info("No cost data by agent found")
+        return
+
+    columns = [
+        ColumnConfig(
+            source_field="agent_name",
+            header_label="Agent",
+            compact_line=1,
+            compact_style="green",
+        ),
+        ColumnConfig(
+            source_field="agent_id",
+            header_label="Agent ID",
+            formatter=lambda x: str(x) if x else "(transient)",
+            compact_line=2,
+            compact_style="dim",
+        ),
+        ColumnConfig(
+            source_field="execution_count",
+            header_label="Executions",
+            formatter=str,
+            compact_line=1,
+            compact_style="white",
+        ),
+        ColumnConfig(
+            source_field="total_tokens",
+            header_label="Total Tokens",
+            formatter=lambda x: f"{x:,}",
+            compact_line=2,
+            compact_style="dim",
+        ),
+        ColumnConfig(
+            source_field="llm_cost_usd",
+            header_label="LLM Cost",
+            formatter=_format_cost,
+            compact_line=2,
+            compact_style="dim",
+        ),
+        ColumnConfig(
+            source_field="rag_cost_usd",
+            header_label="RAG Cost",
+            formatter=_format_cost,
+            compact_line=2,
+            compact_style="dim",
+        ),
+        ColumnConfig(
+            source_field="total_cost_usd",
+            header_label="Total Cost",
+            formatter=_format_cost,
+            compact_line=1,
+            compact_style="green bold",
+        ),
+    ]
+
+    output.table(
+        items=entries,  # type: ignore[arg-type]
+        title=f"Costs by Agent ({len(entries)} agents)",
+        columns=columns,
+    )

cli/commands/billing/costs/by_model/__init__.py

@@ -0,0 +1 @@
+# Empty per project convention - import directly from submodules

cli/commands/billing/costs/by_model/command.py

@@ -0,0 +1,57 @@
+"""Billing project costs by model command implementation."""
+
+from typing import Optional
+from uuid import UUID
+
+import typer
+
+from cli.commands.billing.costs.by_model.presenter import present_billing_costs_by_model
+from cli.commands.flag_utils import validate_uuid
+from cli.infrastructure.command import async_command, authenticated_client
+
+
+def billing_costs_by_model_command(
+    project_id: str = typer.Argument(
+        ...,
+        help="Billing project UUID",
+    ),
+    start_date: Optional[str] = typer.Option(
+        None,
+        "--from",
+        help="Start date (ISO 8601, defaults to 30 days ago)",
+    ),
+    end_date: Optional[str] = typer.Option(
+        None,
+        "--to",
+        help="End date (ISO 8601, defaults to now)",
+    ),
+) -> None:
+    """Get cost breakdown by model for a billing project.
+
+    Examples:
+        ai billing costs by-model 123e4567-89ab...
+        ai billing costs by-model 123e4567-89ab... --from 2026-01-01 --to 2026-01-31
+    """
+    project_uuid = validate_uuid(project_id, "billing project")
+    _execute_costs_by_model(
+        project_id=project_uuid,
+        start_date=start_date,
+        end_date=end_date,
+    )
+
+
+@async_command
+async def _execute_costs_by_model(
+    project_id: UUID,
+    start_date: Optional[str],
+    end_date: Optional[str],
+) -> None:
+    """Execute get billing project costs by model operation."""
+    async with authenticated_client() as (_config, client):
+        response = await client.get_billing_project_costs_by_model(
+            project_id,
+            start_date=start_date,
+            end_date=end_date,
+        )
+
+    present_billing_costs_by_model(response)