pyspiral 0.4.4__cp310-abi3-macosx_11_0_arm64.whl → 0.6.0__cp310-abi3-macosx_11_0_arm64.whl

spiral/cli/iceberg.py

@@ -0,0 +1,97 @@
+import sys
+from typing import Annotated
+
+import rich
+import typer
+from typer import Argument
+
+from spiral.cli import CONSOLE, ERR_CONSOLE, AsyncTyper, state
+from spiral.cli.types import ProjectArg
+
+app = AsyncTyper(short_help="Apache Iceberg Catalog")
+
+
+@app.command(help="List namespaces.")
+def namespaces(
+    project: ProjectArg,
+    namespace: Annotated[str | None, Argument(help="List only namespaces under this namespace.")] = None,
+):
+    """List Iceberg namespaces."""
+    import pyiceberg.exceptions
+
+    catalog = state.spiral.iceberg.catalog()
+
+    if namespace is None:
+        try:
+            namespaces = catalog.list_namespaces(project)
+        except pyiceberg.exceptions.ForbiddenError:
+            ERR_CONSOLE.print(
+                f"The project, {repr(project)}, does not exist or you lack the "
+                f"`iceberg:view` permission to list namespaces in it.",
+            )
+            raise typer.Exit(code=1)
+    else:
+        try:
+            namespaces = catalog.list_namespaces((project, namespace))
+        except pyiceberg.exceptions.ForbiddenError:
+            ERR_CONSOLE.print(
+                f"The namespace, {repr(project)}.{repr(namespace)}, does not exist or you lack the "
+                f"`iceberg:view` permission to list namespaces in it.",
+            )
+            raise typer.Exit(code=1)
+
+    table = CONSOLE.table.Table("Namespace ID", title="Iceberg namespaces")
+    for ns in namespaces:
+        table.add_row(".".join(ns))
+    CONSOLE.print(table)
+
+
+@app.command(help="List tables.")
+def tables(
+    project: ProjectArg,
+    namespace: Annotated[str | None, Argument(help="Show only tables in the given namespace.")] = None,
+):
+    import pyiceberg.exceptions
+
+    catalog = state.spiral.iceberg.catalog()
+
+    try:
+        if namespace is None:
+            tables = catalog.list_tables(project)
+        else:
+            tables = catalog.list_tables((project, namespace))
+    except pyiceberg.exceptions.ForbiddenError:
+        ERR_CONSOLE.print(
+            f"The namespace, {repr(project)}.{repr(namespace)}, does not exist or you lack the "
+            f"`iceberg:view` permission to list tables in it.",
+        )
+        raise typer.Exit(code=1)
+
+    rich_table = rich.table.Table("table id", title="Iceberg tables")
+    for table in tables:
+        rich_table.add_row(".".join(table))
+    CONSOLE.print(rich_table)
+
+
+@app.command(help="Show the table schema.")
+def schema(
+    project: ProjectArg,
+    namespace: Annotated[str, Argument(help="Table namespace.")],
+    table: Annotated[str, Argument(help="Table name.")],
+):
+    import pyiceberg.exceptions
+
+    catalog = state.spiral.iceberg.catalog()
+
+    try:
+        tbl = catalog.load_table((project, namespace, table))
+    except pyiceberg.exceptions.NoSuchTableError:
+        ERR_CONSOLE.print(f"No table {repr(table)} found in {repr(project)}.{repr(namespace)}", file=sys.stderr)
+        raise typer.Exit(code=1)
+
+    rich_table = rich.table.Table(
+        "Field ID", "Field name", "Type", "Required", "Doc", title=f"{project}.{namespace}.{table}"
+    )
+    for col in tbl.schema().columns:
+        rich_table.add_row(str(col.field_id), col.name, str(col.field_type), str(col.required), col.doc)
+    CONSOLE.print(rich_table)

spiral/cli/key_spaces.py

@@ -0,0 +1,68 @@
+from typing import Annotated
+
+import questionary
+import rich
+import typer
+from questionary import Choice
+from typer import Option
+
+from spiral.api.key_space_indexes import SyncIndexRequest
+from spiral.api.projects import KeySpaceIndexResource
+from spiral.api.types import IndexId
+from spiral.api.workers import ResourceClass
+from spiral.cli import CONSOLE, AsyncTyper, state
+from spiral.cli.types import ProjectArg
+
+app = AsyncTyper(short_help="Key Space Indexes.")
+
+
+def ask_index(project_id, title="Select an index"):
+    indexes: list[KeySpaceIndexResource] = list(state.spiral.project(project_id).list_key_space_indexes())
+
+    if not indexes:
+        CONSOLE.print("[red]No indexes found[/red]")
+        raise typer.Exit(1)
+
+    return questionary.select(
+        title,
+        choices=[Choice(title=index.name, value=index.id) for index in sorted(indexes, key=lambda t: (t.name, t.id))],
+    ).ask()
+
+
+def get_index_id(
+    project: ProjectArg,
+    name: Annotated[str | None, Option(help="Index name.")] = None,
+) -> IndexId:
+    if name is None:
+        return ask_index(project)
+
+    indexes: list[KeySpaceIndexResource] = list(state.spiral.project(project).list_key_space_indexes())
+    for index in indexes:
+        if index.name == name:
+            return index.id
+    raise ValueError(f"Index not found: {name}")
+
+
+@app.command(help="List indexes.")
+def ls(
+    project: ProjectArg,
+):
+    """List indexes."""
+    indexes = state.spiral.project(project).list_key_space_indexes()
+
+    rich_table = rich.table.Table("id", "name", title="Key Space Indexes")
+    for index in indexes:
+        rich_table.add_row(index.id, index.name)
+    CONSOLE.print(rich_table)
+
+
+@app.command(help="Trigger a sync job for an index.")
+def sync(
+    project: ProjectArg,
+    name: Annotated[str | None, Option(help="Index name.")] = None,
+    resources: Annotated[ResourceClass, Option(help="Resources to use for the sync job.")] = ResourceClass.SMALL,
+):
+    """Trigger a sync job."""
+    index_id = get_index_id(project, name)
+    response = state.spiral.api.key_space_indexes.sync_index(index_id, SyncIndexRequest(resources=resources))
+    CONSOLE.print(f"Triggered sync job {response.worker_id} for index {index_id}.")

spiral/cli/login.py

@@ -1,22 +1,21 @@
 import jwt
-from rich import print
 
-from spiral.cli import state
+from spiral.cli import CONSOLE, state
 
 
 def command(org_id: str | None = None, force: bool = False):
     token = state.settings.device_code_auth.authenticate(force=force, org_id=org_id)
-    print("Successfully logged in.")
-    print(token.expose_secret())
+    CONSOLE.print("Successfully logged in.")
+    CONSOLE.print(token.expose_secret(), soft_wrap=True)
 
 
 def whoami():
     """Display the current user's information."""
     payload = jwt.decode(state.settings.authn.token().expose_secret(), options={"verify_signature": False})
-    print(f"{payload['org_id']}")
-    print(f"{payload['sub']}")
+    CONSOLE.print(f"{payload['org_id']}")
+    CONSOLE.print(f"{payload['sub']}")
 
 
 def logout():
     state.settings.device_code_auth.logout()
-    print("Logged out.")
+    CONSOLE.print("Logged out.")

spiral/cli/orgs.py

@@ -2,13 +2,12 @@ import webbrowser
 from typing import Annotated
 
 import jwt
-import rich
 import typer
 from rich.table import Table
 from typer import Option
 
 from spiral.api.organizations import CreateOrgRequest, InviteUserRequest, OrgRole, PortalLinkIntent, PortalLinkRequest
-from spiral.cli import AsyncTyper, state
+from spiral.cli import CONSOLE, ERR_CONSOLE, AsyncTyper, state
 from spiral.cli.types import OrganizationArg
 
 app = AsyncTyper(short_help="Org admin.")
@@ -17,7 +16,7 @@ app = AsyncTyper(short_help="Org admin.")
 @app.command(help="Switch the active organization.")
 def switch(org_id: OrganizationArg):
     state.settings.device_code_auth.authenticate(org_id=org_id)
-    rich.print(f"Switched to organization: {org_id}")
+    CONSOLE.print(f"Switched to organization: {org_id}")
 
 
 @app.command(help="Create a new organization.")
@@ -29,7 +28,7 @@ def create(
     # Authenticate to the new organization
     state.settings.device_code_auth.authenticate(org_id=res.org.id)
 
-    rich.print(f"{res.org.name} [dim]{res.org.id}[/dim]")
+    CONSOLE.print(f"{res.org.name} [dim]{res.org.id}[/dim]")
 
 
 @app.command(help="List organizations.")
@@ -40,7 +39,7 @@ def ls():
     for m in state.settings.api.organization.list_memberships():
         table.add_row("👉" if m.org.id == org_id else "", m.org.id, m.org.name, m.role)
 
-    rich.print(table)
+    CONSOLE.print(table)
 
 
 @app.command(help="Invite a user to the organization.")
@@ -48,7 +47,7 @@ def invite(email: str, role: OrgRole = OrgRole.MEMBER, expires_in_days: int = 7)
     state.settings.api.organization.invite_user(
         InviteUserRequest(email=email, role=role, expires_in_days=expires_in_days)
     )
-    rich.print(f"Invited {email} as a {role.value}.")
+    CONSOLE.print(f"Invited {email} as a {role.value}.")
 
 
 @app.command(help="Configure single sign-on for your organization.")
@@ -78,7 +77,7 @@ def domains():
 
 def _do_action(intent: PortalLinkIntent):
     res = state.settings.api.organization.portal_link(PortalLinkRequest(intent=intent))
-    rich.print(f"Opening the configuration portal:\n{res.url}")
+    CONSOLE.print(f"Opening the configuration portal:\n{res.url}")
     webbrowser.open(res.url)
 
 
@@ -86,5 +85,5 @@ def current_org_id():
     if token := state.settings.authn.token():
         if org_id := jwt.decode(token.expose_secret(), options={"verify_signature": False}).get("org_id"):
             return org_id
-    rich.print("[red]You are not logged in to an organization.[/red]")
+    ERR_CONSOLE.print("You are not logged in to an organization.")
     raise typer.Exit(1)

spiral/cli/printer.py

@@ -2,14 +2,14 @@ import json
 from collections.abc import Iterable
 from typing import TypeVar
 
-import betterproto
+import betterproto2
 from pydantic import BaseModel
 from rich.console import ConsoleRenderable, Group
 from rich.padding import Padding
 from rich.pretty import Pretty
 from rich.table import Table
 
-T = TypeVar("T", bound=betterproto.Message)
+T = TypeVar("T", bound=betterproto2.Message)
 M = TypeVar("M", bound=BaseModel)
 
 
@@ -34,7 +34,7 @@ def table_of_protos(cls: type[T], messages: Iterable[T], fields: list[str] = Non
 
     TODO(ngates): add a CLI switch to emit JSON results instead of tables.
     """
-    cols = fields or cls()._betterproto.sorted_field_names
+    cols = fields or cls()._betterproto2.sorted_field_names
     table = Table(*cols, title=title or f"{cls.__name__}s")
     for msg in messages:
         table.add_row(*[getattr(msg, col, "") for col in cols])

spiral/cli/projects.py

@@ -1,6 +1,5 @@
 from typing import Annotated
 
-import rich
 import typer
 from typer import Option
 
@@ -19,7 +18,7 @@ from spiral.api.projects import (
     Project,
     WorkloadPrincipalConditions,
 )
-from spiral.cli import AsyncTyper, printer, state
+from spiral.cli import CONSOLE, AsyncTyper, printer, state
 from spiral.cli.types import ProjectArg
 
 app = AsyncTyper(short_help="Projects and grants.")
@@ -28,7 +27,7 @@ app = AsyncTyper(short_help="Projects and grants.")
 @app.command(help="List projects.")
 def ls():
     projects = list(state.settings.api.project.list())
-    rich.print(printer.table_of_models(Project, projects))
+    CONSOLE.print(printer.table_of_models(Project, projects))
 
 
 @app.command(help="Create a new project.")
@@ -39,7 +38,7 @@ def create(
     name: Annotated[str | None, Option(help="Friendly name for the project.")] = None,
 ):
     res = state.settings.api.project.create(CreateProjectRequest(id_prefix=id_prefix, name=name))
-    rich.print(f"Created project {res.project.id}")
+    CONSOLE.print(f"Created project {res.project.id}")
 
 
 @app.command(help="Grant a role on a project.")
@@ -127,10 +126,10 @@ def grant(
         ),
     )
 
-    rich.print(f"Granted role {role} on project {project}")
+    CONSOLE.print(f"Granted role {role} on project {project}")
 
 
 @app.command(help="List project grants.")
 def grants(project: ProjectArg):
     project_grants = list(state.settings.api.project.list_grants(project))
-    rich.print(printer.table_of_models(Grant, project_grants, title="Project Grants"))
+    CONSOLE.print(printer.table_of_models(Grant, project_grants, title="Project Grants"))

spiral/cli/tables.py

@@ -0,0 +1,131 @@
+from typing import Annotated
+
+import questionary
+import rich
+import rich.table
+import typer
+from questionary import Choice
+from typer import Argument, Option
+
+from spiral import Spiral
+from spiral.api.projects import TableResource
+from spiral.cli import CONSOLE, ERR_CONSOLE, AsyncTyper, state
+from spiral.cli.types import ProjectArg
+from spiral.debug.manifests import display_manifests
+from spiral.table import Table
+
+app = AsyncTyper(short_help="Spiral Tables.")
+
+
+def ask_table(project_id: str, title: str = "Select a table") -> str:
+    tables: list[TableResource] = list(state.spiral.project(project_id).list_tables())
+
+    if not tables:
+        ERR_CONSOLE.print("No tables found")
+        raise typer.Exit(1)
+
+    return questionary.select(  # pyright: ignore[reportAny]
+        title,
+        choices=[
+            Choice(title=f"{table.dataset}.{table.table}", value=f"{table.dataset}.{table.table}")
+            for table in sorted(tables, key=lambda t: (t.dataset, t.table))
+        ],
+    ).ask()
+
+
+def get_table(
+    project: ProjectArg,
+    table: Annotated[str | None, Option(help="Table name.")] = None,
+    dataset: Annotated[str | None, Option(help="Dataset name.")] = None,
+) -> tuple[str, Table]:
+    if table is None:
+        identifier = ask_table(project)
+    else:
+        identifier = table
+        if dataset is not None:
+            identifier = f"{dataset}.{table}"
+    return identifier, state.spiral.project(project).table(identifier)
+
+
+@app.command(help="List tables.")
+def ls(
+    project: ProjectArg,
+):
+    tables = Spiral().project(project).list_tables()
+
+    rich_table = rich.table.Table("id", "dataset", "name", title="Spiral tables")
+    for table in tables:
+        rich_table.add_row(table.id, table.dataset, table.table)
+    CONSOLE.print(rich_table)
+
+
+@app.command(help="Show the table key schema.")
+def key_schema(
+    project: ProjectArg,
+    table: Annotated[str | None, Option(help="Table name.")] = None,
+    dataset: Annotated[str | None, Option(help="Dataset name.")] = None,
+):
+    _, t = get_table(project, table, dataset)
+    CONSOLE.print(t.key_schema)
+
+
+@app.command(help="Compute the full table schema.")
+def schema(
+    project: ProjectArg,
+    table: Annotated[str | None, Option(help="Table name.")] = None,
+    dataset: Annotated[str | None, Option(help="Dataset name.")] = None,
+):
+    _, t = get_table(project, table, dataset)
+    CONSOLE.print(t.schema())
+
+
+@app.command(help="Flush Write-Ahead-Log.")
+def flush(
+    project: ProjectArg,
+    table: Annotated[str | None, Option(help="Table name.")] = None,
+    dataset: Annotated[str | None, Option(help="Dataset name.")] = None,
+):
+    identifier, t = get_table(project, table, dataset)
+    state.spiral._ops().flush_wal(t.core)  # pyright: ignore[reportPrivateUsage]
+    CONSOLE.print(f"Flushed WAL for table {identifier} in project {project}.")
+
+
+@app.command(help="Display scan.")
+def debug(
+    project: ProjectArg,
+    table: Annotated[str | None, Option(help="Table name.")] = None,
+    dataset: Annotated[str | None, Option(help="Dataset name.")] = None,
+    column_group: Annotated[str, Argument(help="Dot-separated column group path.")] = ".",
+):
+    _, t = get_table(project, table, dataset)
+    scan = state.spiral.scan(t[column_group] if column_group != "." else t)
+    scan._debug()  # pyright: ignore[reportPrivateUsage]
+
+
+@app.command(help="Display all manifests.")
+def manifests(
+    project: ProjectArg,
+    table: Annotated[str | None, Option(help="Table name.")] = None,
+    dataset: Annotated[str | None, Option(help="Dataset name.")] = None,
+):
+    _, t = get_table(project, table, dataset)
+    s = t.snapshot()
+
+    key_space_state = state.spiral._ops().key_space_state(t.core, asof=s.asof)  # pyright: ignore[reportPrivateUsage]
+    key_space_manifest = key_space_state.manifest
+
+    column_groups_states = state.spiral._ops().column_groups_states(t.core, key_space_state, asof=s.asof)  # pyright: ignore[reportPrivateUsage]
+
+    display_manifests(key_space_manifest, ((x.column_group, x.manifest) for x in column_groups_states))
+
+
+@app.command(help="Display the manifests which would be read by a scan of the given column group.")
+def scan_manifests(
+    project: ProjectArg,
+    table: Annotated[str | None, Option(help="Table name.")] = None,
+    dataset: Annotated[str | None, Option(help="Dataset name.")] = None,
+    column_group: Annotated[str, Argument(help="Dot-separated column group path.")] = ".",
+):
+    _, t = get_table(project, table, dataset)
+    scan = state.spiral.scan(t[column_group] if column_group != "." else t)
+    scan._dump_manifests()  # pyright: ignore[reportPrivateUsage]

spiral/cli/telemetry.py

@@ -1,8 +1,7 @@
 import pyperclip
-import rich
 
 from spiral.api.telemetry import IssueExportTokenResponse
-from spiral.cli import AsyncTyper, state
+from spiral.cli import CONSOLE, AsyncTyper, state
 
 app = AsyncTyper(short_help="Client-side telemetry.")
 
@@ -14,5 +13,5 @@ def export():
     command = f"export SPIRAL_OTEL_TOKEN={res.token}"
     pyperclip.copy(command)
 
-    rich.print("Export command copied to clipboard! Paste and run to set [green]SPIRAL_OTEL_TOKEN[/green].")
-    rich.print("[dim]Token is valid for 1h.[/dim]")
+    CONSOLE.print("Export command copied to clipboard! Paste and run to set [green]SPIRAL_OTEL_TOKEN[/green].")
+    CONSOLE.print("[dim]Token is valid for 1h.[/dim]")

spiral/cli/text.py

@@ -0,0 +1,115 @@
+from typing import Annotated
+
+import questionary
+import rich
+import typer
+from questionary import Choice
+from typer import Option
+
+from spiral.api.projects import TextIndexResource
+from spiral.api.text_indexes import CreateWorkerRequest, SyncIndexRequest
+from spiral.api.types import IndexId
+from spiral.api.workers import CPU, GcpRegion, Memory, ResourceClass
+from spiral.cli import CONSOLE, ERR_CONSOLE, AsyncTyper, state
+from spiral.cli.types import ProjectArg
+
+app = AsyncTyper(short_help="Text Indexes.")
+
+
+def ask_index(project_id, title="Select an index"):
+    indexes: list[TextIndexResource] = list(state.spiral.project(project_id).list_text_indexes())
+
+    if not indexes:
+        ERR_CONSOLE.print("No indexes found")
+        raise typer.Exit(1)
+
+    return questionary.select(
+        title,
+        choices=[Choice(title=index.name, value=index.id) for index in sorted(indexes, key=lambda t: (t.name, t.id))],
+    ).ask()
+
+
+def get_index_id(
+    project: ProjectArg,
+    name: Annotated[str | None, Option(help="Index name.")] = None,
+) -> IndexId:
+    if name is None:
+        return ask_index(project)
+
+    indexes: list[TextIndexResource] = list(state.spiral.project(project).list_text_indexes())
+    for index in indexes:
+        if index.name == name:
+            return index.id
+    raise ValueError(f"Index not found: {name}")
+
+
+@app.command(help="List indexes.")
+def ls(
+    project: ProjectArg,
+):
+    """List indexes."""
+    indexes = state.spiral.project(project).list_text_indexes()
+
+    rich_table = rich.table.Table("id", "name", title="Text Indexes")
+    for index in indexes:
+        rich_table.add_row(index.id, index.name)
+    CONSOLE.print(rich_table)
+
+
+@app.command(help="Trigger a sync job for an index.")
+def sync(
+    project: ProjectArg,
+    name: Annotated[str | None, Option(help="Index name.")] = None,
+    resources: Annotated[ResourceClass, Option(help="Resources to use for the sync job.")] = ResourceClass.SMALL,
+):
+    """Trigger a sync job."""
+    index_id = get_index_id(project, name)
+    response = state.spiral.api.text_indexes.sync_index(index_id, SyncIndexRequest(resources=resources))
+    CONSOLE.print(f"Triggered sync job {response.worker_id} for index {index_id}.")
+
+
+@app.command(name="serve", help="Spin up a worker to serve an index.")
+def serve(
+    project: ProjectArg,
+    index: Annotated[str | None, Option(help="Index name.")] = None,
+    region: Annotated[GcpRegion, Option(help="GCP region for the worker.")] = GcpRegion.US_EAST4,
+    cpu: Annotated[CPU, Option(help="CPU resources for the worker.")] = CPU.ONE,
+    memory: Annotated[Memory, Option(help="Memory resources for the worker in MB.")] = Memory.MB_512,
+):
+    """Spin up a worker."""
+    index_id = get_index_id(project, index)
+    request = CreateWorkerRequest(cpu=cpu, memory=memory, region=region)
+    response = state.spiral.api.text_indexes.create_worker(index_id, request)
+    CONSOLE.print(f"Created worker {response.worker_id} for {index_id}.")
+
+
+@app.command(name="workers", help="List search workers.")
+def workers(
+    project: ProjectArg,
+    index: Annotated[str | None, Option(help="Index name.")] = None,
+):
+    """List text search workers."""
+    index_id = get_index_id(project, index)
+    worker_ids = state.spiral.api.text_indexes.list_workers(index_id)
+
+    rich_table = rich.table.Table("Worker ID", "URL", title=f"Text Search Workers for {index_id}")
+    for worker_id in worker_ids:
+        try:
+            worker = state.spiral.api.text_indexes.get_worker(worker_id)
+            rich_table.add_row(
+                worker_id,
+                worker.url,
+            )
+        except Exception:
+            rich_table.add_row(
+                worker_id,
+                "Unavailable",
+            )
+    CONSOLE.print(rich_table)
+
+
+@app.command(name="shutdown", help="Shutdown a search worker.")
+def shutdown(worker_id: str):
+    """Shutdown a worker."""
+    state.spiral.api.text_indexes.shutdown_worker(worker_id)
+    CONSOLE.print(f"Requested worker {worker_id} to shutdown.")

spiral/cli/types.py

@@ -1,20 +1,19 @@
 from typing import Annotated
 
 import questionary
-import rich
 import typer
 from questionary import Choice
 from typer import Argument
 
 from spiral.api.types import OrgId, ProjectId
-from spiral.cli import state
+from spiral.cli import ERR_CONSOLE, state
 
 
 def ask_project(title="Select a project"):
     projects = list(state.settings.api.project.list())
 
     if not projects:
-        rich.print("[red]No projects found[/red]")
+        ERR_CONSOLE.print("No projects found")
         raise typer.Exit(1)
 
     return questionary.select(
@@ -33,7 +32,7 @@ def _org_default():
     memberships = list(state.settings.api.organization.list_memberships())
 
     if not memberships:
-        rich.print("[red]No organizations found[/red]")
+        ERR_CONSOLE.print("No organizations found")
         raise typer.Exit(1)
 
     return questionary.select(

spiral/cli/workloads.py

@@ -2,12 +2,11 @@ from typing import Annotated
 
 import pyperclip
 import questionary
-import rich
 from questionary import Choice
 from typer import Argument, Option
 
 from spiral.api.workloads import CreateWorkloadRequest, IssueWorkloadCredentialsResponse, Workload
-from spiral.cli import AsyncTyper, printer, state
+from spiral.cli import CONSOLE, ERR_CONSOLE, AsyncTyper, printer, state
 from spiral.cli.types import ProjectArg
 
 app = AsyncTyper()
@@ -19,7 +18,7 @@ def create(
     name: Annotated[str | None, Option(help="Friendly name for the workload.")] = None,
 ):
     res = state.settings.api.workload.create(project, CreateWorkloadRequest(name=name))
-    rich.print(f"Created workload {res.workload.id}")
+    CONSOLE.print(f"Created workload {res.workload.id}")
 
 
 @app.command(help="List workloads.")
@@ -27,7 +26,7 @@ def ls(
     project: ProjectArg,
 ):
     workloads = list(state.settings.api.workload.list(project))
-    rich.print(printer.table_of_models(Workload, workloads, fields=["id", "project_id", "name"]))
+    CONSOLE.print(printer.table_of_models(Workload, workloads, fields=["id", "project_id", "name"]))
 
 
 @app.command(help="Issue new workflow credentials.")
@@ -46,14 +45,14 @@ def issue_credentials(workload_id: Annotated[str, Argument(help="Workload ID.")]
 
         if choice == 1:
             pyperclip.copy(res.client_secret)
-            rich.print("[green]Secret copied to clipboard![/green]")
+            CONSOLE.print("[green]Secret copied to clipboard![/green]")
             break
         elif choice == 2:
-            rich.print(f"[green]Token Secret:[/green] {res.client_secret}")
+            CONSOLE.print(f"[green]Token Secret:[/green] {res.client_secret}")
             break
         elif choice == 3:
             break
         else:
-            rich.print("[red]Invalid choice. Please try again.[/red]")
+            ERR_CONSOLE.print("Invalid choice. Please try again.")
 
-    rich.print(f"[green]Token ID:[/green] {res.client_id}")
+    CONSOLE.print(f"[green]Token ID:[/green] {res.client_id}")