lightning-sdk 2025.8.6rc2__py3-none-any.whl → 2025.11.5__py3-none-any.whl

lightning_sdk/cli/legacy/entrypoint.py

@@ -0,0 +1,110 @@
+import sys
+import traceback
+from types import TracebackType
+from typing import Type
+
+import click
+from rich.console import Console, Group
+from rich.panel import Panel
+from rich.syntax import Syntax
+from rich.text import Text
+
+from lightning_sdk import __version__
+from lightning_sdk.api.studio_api import _cloud_url
+from lightning_sdk.cli.legacy.ai_hub import aihub
+from lightning_sdk.cli.legacy.coloring import CustomHelpFormatter
+from lightning_sdk.cli.legacy.configure import configure
+from lightning_sdk.cli.legacy.connect import connect
+from lightning_sdk.cli.legacy.create import create
+from lightning_sdk.cli.legacy.delete import delete
+from lightning_sdk.cli.legacy.deploy.serve import deploy
+from lightning_sdk.cli.legacy.docker_cli import dockerize
+from lightning_sdk.cli.legacy.download import download
+from lightning_sdk.cli.legacy.generate import generate
+from lightning_sdk.cli.legacy.inspection import inspect
+from lightning_sdk.cli.legacy.list import list_cli
+from lightning_sdk.cli.legacy.open import open
+from lightning_sdk.cli.legacy.run import run
+from lightning_sdk.cli.legacy.start import start
+from lightning_sdk.cli.legacy.stop import stop
+from lightning_sdk.cli.legacy.switch import switch
+from lightning_sdk.cli.legacy.upload import upload
+from lightning_sdk.constants import _LIGHTNING_DEBUG
+from lightning_sdk.lightning_cloud.login import Auth
+
+
+def _notify_exception(exception_type: Type[BaseException], value: BaseException, tb: TracebackType) -> None:
+    """CLI won't show tracebacks, just print the exception message."""
+    console = Console()
+
+    message = str(value.args[0]) if value.args else str(value) or "An unknown error occurred"
+
+    error_text = Text()
+    error_text.append(f"{exception_type.__name__}: ", style="bold red")
+    error_text.append(message, style="white")
+
+    renderables = [error_text]
+
+    if _LIGHTNING_DEBUG:
+        tb_text = "".join(traceback.format_exception(exception_type, value, tb))
+        renderables.append(Text("\n\nFull traceback:\n", style="bold yellow"))
+        renderables.append(Syntax(tb_text, "python", theme="monokai light", line_numbers=False, word_wrap=True))
+    else:
+        renderables.append(Text("\n\n🐞 To view the full traceback, set: LIGHTNING_DEBUG=1"))
+
+    renderables.append(Text("\n📘 Need help? Run: lightning <command> --help", style="cyan"))
+
+    console.print(Panel(Group(*renderables), title="⚡ Lightning CLI Error", border_style="red"))
+
+
+@click.group(name="lightning", help="Command line interface (CLI) to interact with/manage Lightning AI Studios.")
+@click.version_option(__version__, message="Lightning CLI version %(version)s")
+def main_cli() -> None:
+    sys.excepthook = _notify_exception
+
+
+# colorful help messages
+main_cli.context_class.formatter_class = CustomHelpFormatter
+
+
+@main_cli.command
+def login() -> None:
+    """Login to Lightning AI Studios."""
+    auth = Auth()
+    auth.clear()
+
+    try:
+        auth.authenticate()
+    except ConnectionError:
+        raise RuntimeError(f"Unable to connect to {_cloud_url()}. Please check your internet connection.") from None
+
+
+@main_cli.command
+def logout() -> None:
+    """Logout from Lightning AI Studios."""
+    auth = Auth()
+    auth.clear()
+
+
+# additional commands
+main_cli.add_command(aihub)
+main_cli.add_command(configure)
+main_cli.add_command(connect)
+main_cli.add_command(create)
+main_cli.add_command(delete)
+main_cli.add_command(dockerize)
+main_cli.add_command(download)
+main_cli.add_command(generate)
+main_cli.add_command(inspect)
+main_cli.add_command(list_cli)
+main_cli.add_command(run)
+main_cli.add_command(deploy)
+main_cli.add_command(start)
+main_cli.add_command(stop)
+main_cli.add_command(switch)
+main_cli.add_command(upload)
+main_cli.add_command(open)
+
+
+if __name__ == "__main__":
+    main_cli()

lightning_sdk/cli/{generate.py → legacy/generate.py}

@@ -3,7 +3,7 @@ from typing import Optional
 import click
 from rich.console import Console
 
-from lightning_sdk.cli.studios_menu import _StudiosMenu
+from lightning_sdk.cli.legacy.studios_menu import _StudiosMenu
 
 
 @click.group(name="generate")

lightning_sdk/cli/{inspection.py → legacy/inspection.py}

@@ -3,7 +3,7 @@ from typing import Optional
 import click
 from rich.console import Console
 
-from lightning_sdk.cli.job_and_mmt_action import _JobAndMMTAction
+from lightning_sdk.cli.legacy.job_and_mmt_action import _JobAndMMTAction
 
 
 @click.group(name="inspect")

lightning_sdk/cli/{job_and_mmt_action.py → legacy/job_and_mmt_action.py}

@@ -1,13 +1,13 @@
 from typing import Optional
 
-from lightning_sdk.cli.jobs_menu import _JobsMenu
-from lightning_sdk.cli.mmts_menu import _MMTsMenu
-from lightning_sdk.cli.teamspace_menu import _TeamspacesMenu
+from lightning_sdk.cli.legacy.jobs_menu import _JobsMenu
+from lightning_sdk.cli.legacy.mmts_menu import _MMTsMenu
+from lightning_sdk.cli.utils.teamspace_selection import TeamspacesMenu
 from lightning_sdk.job import Job
 from lightning_sdk.mmt import MMT
 
 
-class _JobAndMMTAction(_TeamspacesMenu, _JobsMenu, _MMTsMenu):
+class _JobAndMMTAction(TeamspacesMenu, _JobsMenu, _MMTsMenu):
     """Inspect resources of the Lightning AI platform to get additional details as JSON."""
 
     def job(self, name: Optional[str] = None, teamspace: Optional[str] = None) -> Job:
@@ -20,7 +20,7 @@ class _JobAndMMTAction(_TeamspacesMenu, _JobsMenu, _MMTsMenu):
                 If not specified can be selected interactively.
 
         """
-        resolved_teamspace = self._resolve_teamspace(teamspace)
+        resolved_teamspace = self(teamspace)
         return self._resolve_job(name, teamspace=resolved_teamspace)
 
     def mmt(self, name: Optional[str] = None, teamspace: Optional[str] = None) -> MMT:
@@ -33,5 +33,5 @@ class _JobAndMMTAction(_TeamspacesMenu, _JobsMenu, _MMTsMenu):
                 If not specified can be selected interactively.
 
         """
-        resolved_teamspace = self._resolve_teamspace(teamspace)
+        resolved_teamspace = self(teamspace)
         return self._resolve_mmt(name, teamspace=resolved_teamspace)

lightning_sdk/cli/{jobs_menu.py → legacy/jobs_menu.py}

@@ -3,7 +3,7 @@ from typing import Dict, List, Optional
 from rich.console import Console
 from simple_term_menu import TerminalMenu
 
-from lightning_sdk.cli.exceptions import StudioCliError
+from lightning_sdk.cli.legacy.exceptions import StudioCliError
 from lightning_sdk.job import Job
 from lightning_sdk.teamspace import Teamspace
 

lightning_sdk/cli/{list.py → legacy/list.py}

@@ -7,9 +7,9 @@ from rich.table import Table
 from typing_extensions import Literal
 
 from lightning_sdk import Job, Machine, Studio, Teamspace
-from lightning_sdk.cli.clusters_menu import _ClustersMenu
-from lightning_sdk.cli.teamspace_menu import _TeamspacesMenu
-from lightning_sdk.lightning_cloud.openapi import V1ClusterType, V1MultiMachineJob
+from lightning_sdk.cli.legacy.clusters_menu import _ClustersMenu
+from lightning_sdk.cli.utils.teamspace_selection import TeamspacesMenu
+from lightning_sdk.lightning_cloud.openapi import V1MultiMachineJob
 from lightning_sdk.lit_container import LitContainer
 from lightning_sdk.utils.resolve import _get_authed_user
 
@@ -48,7 +48,7 @@ def studios(
 ) -> None:
     """List studios for a given teamspace."""
     studios = []
-    menu = _TeamspacesMenu()
+    menu = TeamspacesMenu()
     if all and not teamspace:
         user = _get_authed_user()
         possible_teamspaces = menu._get_possible_teamspaces(user)
@@ -56,7 +56,7 @@ def studios(
             teamspace = Teamspace(**ts)
             studios.extend(teamspace.studios)
     else:
-        resolved_teamspace = menu._resolve_teamspace(teamspace=teamspace)
+        resolved_teamspace = menu(teamspace=teamspace)
         studios = resolved_teamspace.studios
 
     table = Table(
@@ -111,7 +111,7 @@ def jobs(
 ) -> None:
     """List jobs for a given teamspace."""
     jobs = []
-    menu = _TeamspacesMenu()
+    menu = TeamspacesMenu()
     if all and not teamspace:
         user = _get_authed_user()
         possible_teamspaces = menu._get_possible_teamspaces(user)
@@ -119,7 +119,7 @@ def jobs(
             teamspace = Teamspace(**ts)
             jobs.extend(teamspace.jobs)
     else:
-        resolved_teamspace = menu._resolve_teamspace(teamspace=teamspace)
+        resolved_teamspace = menu(teamspace=teamspace)
         jobs = resolved_teamspace.jobs
 
     table = Table(pad_edge=True)
@@ -182,7 +182,7 @@ def mmts(
 ) -> None:
     """List multi-machine jobs for a given teamspace."""
     jobs = []
-    menu = _TeamspacesMenu()
+    menu = TeamspacesMenu()
     if all and not teamspace:
         user = _get_authed_user()
         possible_teamspaces = menu._get_possible_teamspaces(user)
@@ -190,7 +190,7 @@ def mmts(
             teamspace = Teamspace(**ts)
             jobs.extend(teamspace.multi_machine_jobs)
     else:
-        resolved_teamspace = menu._resolve_teamspace(teamspace=teamspace)
+        resolved_teamspace = menu(teamspace=teamspace)
         jobs = resolved_teamspace.multi_machine_jobs
 
     table = Table(pad_edge=True)
@@ -242,13 +242,12 @@ def mmts(
 def containers(teamspace: Optional[str] = None, cloud_account: Optional[str] = None) -> None:
     """Display the list of available containers."""
     api = LitContainer()
-    menu = _TeamspacesMenu()
+    menu = TeamspacesMenu()
     clusters_menu = _ClustersMenu()
-    resolved_teamspace = menu._resolve_teamspace(teamspace=teamspace)
+    resolved_teamspace = menu(teamspace=teamspace)
 
     if not cloud_account:
-        cloud_account_obj = clusters_menu._resolve_cluster(resolved_teamspace)
-        cloud_account = "" if cloud_account_obj.spec.cluster_type == V1ClusterType.GLOBAL else cloud_account_obj.id
+        cloud_account = clusters_menu._resolve_cluster(resolved_teamspace)
 
     result = api.list_containers(
         teamspace=resolved_teamspace.name, org=resolved_teamspace.owner.name, cloud_account=cloud_account

lightning_sdk/cli/{mmts_menu.py → legacy/mmts_menu.py}

@@ -3,7 +3,7 @@ from typing import Dict, List, Optional
 from rich.console import Console
 from simple_term_menu import TerminalMenu
 
-from lightning_sdk.cli.exceptions import StudioCliError
+from lightning_sdk.cli.legacy.exceptions import StudioCliError
 from lightning_sdk.mmt import MMT
 from lightning_sdk.teamspace import Teamspace
 

lightning_sdk/cli/{open.py → legacy/open.py}

@@ -6,8 +6,8 @@ from typing import Optional
 import click
 from rich.console import Console
 
-from lightning_sdk.cli.teamspace_menu import _TeamspacesMenu
-from lightning_sdk.cli.upload import _upload_folder
+from lightning_sdk.cli.legacy.upload import _upload_folder
+from lightning_sdk.cli.utils.teamspace_selection import TeamspacesMenu
 from lightning_sdk.studio import Studio
 from lightning_sdk.teamspace import Teamspace
 from lightning_sdk.utils.resolve import _get_studio_url
@@ -49,8 +49,8 @@ def open(path: str = ".", teamspace: Optional[str] = None, cloud_account: Option
     try:
         resolved_teamspace = Teamspace()
     except ValueError:
-        menu = _TeamspacesMenu()
-        resolved_teamspace = menu._resolve_teamspace(teamspace=teamspace)
+        menu = TeamspacesMenu()
+        resolved_teamspace = menu(teamspace=teamspace)
 
     # default cloud account to current studios cloud account if run from studio
     # else it will fall back to teamspace default in the backend

lightning_sdk/cli/{start.py → legacy/start.py}

@@ -103,4 +103,5 @@ def studio(
     except KeyError:
         resolved_machine = machine
 
+    Studio.show_progress = True
     studio.start(resolved_machine)

lightning_sdk/cli/{stop.py → legacy/stop.py}

@@ -3,7 +3,7 @@ from typing import Optional
 import click
 from rich.console import Console
 
-from lightning_sdk.cli.job_and_mmt_action import _JobAndMMTAction
+from lightning_sdk.cli.legacy.job_and_mmt_action import _JobAndMMTAction
 from lightning_sdk.lightning_cloud.openapi.rest import ApiException
 from lightning_sdk.studio import Studio
 

lightning_sdk/cli/{switch.py → legacy/switch.py}

@@ -59,4 +59,5 @@ def studio(name: str, teamspace: Optional[str] = None, machine: str = "CPU") ->
     except KeyError:
         resolved_machine = machine
 
+    Studio.show_progress = True
     studio.switch_machine(resolved_machine)

lightning_sdk/cli/{teamspace_menu.py → legacy/teamspace_menu.py}

@@ -4,7 +4,7 @@ from rich.console import Console
 from simple_term_menu import TerminalMenu
 
 from lightning_sdk.api import OrgApi
-from lightning_sdk.cli.exceptions import StudioCliError
+from lightning_sdk.cli.legacy.exceptions import StudioCliError
 from lightning_sdk.teamspace import Teamspace
 from lightning_sdk.user import User
 from lightning_sdk.utils.resolve import _get_authed_user

lightning_sdk/cli/{upload.py → legacy/upload.py}

@@ -14,13 +14,13 @@ from tqdm import tqdm
 
 from lightning_sdk.api.lit_container_api import DockerNotRunningError, LCRAuthFailedError, LitContainerApi
 from lightning_sdk.api.utils import _get_cloud_url
-from lightning_sdk.cli.exceptions import StudioCliError
-from lightning_sdk.cli.studios_menu import _StudiosMenu
-from lightning_sdk.cli.teamspace_menu import _TeamspacesMenu
+from lightning_sdk.cli.legacy.exceptions import StudioCliError
+from lightning_sdk.cli.legacy.studios_menu import _StudiosMenu
+from lightning_sdk.cli.utils.teamspace_selection import TeamspacesMenu
 from lightning_sdk.constants import _LIGHTNING_DEBUG
 from lightning_sdk.models import upload_model as _upload_model
 from lightning_sdk.studio import Studio
-from lightning_sdk.utils.resolve import _get_authed_user, skip_studio_init
+from lightning_sdk.utils.resolve import _get_authed_user
 
 _STUDIO_UPLOAD_STATUS_PATH = "~/.lightning/studios/uploads"
 
@@ -137,8 +137,8 @@ def upload_container(
     platform: Optional[str] = "linux/amd64",
 ) -> None:
     """Upload a container to Lightning AI's container registry."""
-    menu = _TeamspacesMenu()
-    teamspace = menu._resolve_teamspace(teamspace)
+    menu = TeamspacesMenu()
+    teamspace = menu(teamspace)
     console = Console()
     with Progress(
         SpinnerColumn(),
@@ -285,8 +285,7 @@ def _resolve_studio(studio: Optional[str]) -> Studio:
             "Please contact Lightning AI directly to resolve this issue."
         ) from e
 
-    with skip_studio_init():
-        return Studio(**selected_studio)
+    return Studio(**selected_studio)
 
 
 def _print_docker_push(lines: Generator, console: Console, progress: Progress, push_task: rich.progress.TaskID) -> None:

lightning_sdk/cli/license/__init__.py

@@ -0,0 +1,14 @@
+"""Base Studio CLI commands."""
+
+import click
+
+
+def register_commands(group: click.Group) -> None:
+    """Register base studio commands with the given group."""
+    from lightning_sdk.cli.license.get import get_license
+    from lightning_sdk.cli.license.list import list_licenses
+    from lightning_sdk.cli.license.set import set_license
+
+    group.add_command(list_licenses)
+    group.add_command(get_license)
+    group.add_command(set_license)

lightning_sdk/cli/license/get.py

@@ -0,0 +1,15 @@
+import click
+
+from lightning_sdk.utils.config import _DEFAULT_CONFIG_FILE_PATH, Config, DefaultConfigKeys
+
+
+@click.command("get")
+@click.argument("product_name")
+@click.option("--config-file", help="Path to the config file")
+def get_license(product_name: str, config_file: str = _DEFAULT_CONFIG_FILE_PATH) -> None:
+    """Get a license key for a given product."""
+    cfg = Config(config_file)
+    license_key = cfg.get(f"{DefaultConfigKeys.license}.{product_name}")
+    if license_key:
+        # echo the license key without any additional output to make parsing simpler
+        click.echo(license_key)

lightning_sdk/cli/license/list.py

@@ -0,0 +1,45 @@
+"""License list command."""
+
+from typing import Mapping
+
+import click
+from rich.table import Table
+
+from lightning_sdk.cli.utils.richt_print import rich_to_str
+from lightning_sdk.utils.config import _DEFAULT_CONFIG_FILE_PATH, Config, DefaultConfigKeys
+
+
+@click.command("list")
+@click.option("--include-key", help="Print the key as well", is_flag=True)
+@click.option("--config-file", help="Path to the config file")
+def list_licenses(include_key: bool, config_file: str = _DEFAULT_CONFIG_FILE_PATH) -> None:
+    """List configured licenses.
+
+    Example:
+        lightning license list --include-key
+
+    """
+    return list_impl(include_key=include_key, config_path=config_file)
+
+
+def list_impl(include_key: bool, config_path: str) -> None:
+    cfg = Config(config_file=config_path)
+
+    license_cfg = cfg.get_sub_config(DefaultConfigKeys.license)
+
+    if isinstance(license_cfg, Mapping):
+        table = Table(
+            pad_edge=True,
+        )
+
+        table.add_column("Product")
+        table.add_column("License Key")
+
+        # sort by product_name
+        for product_name, license_key in sorted(license_cfg.items(), key=lambda x: x[0]):
+            table.add_row(product_name, license_key if include_key else "********")
+
+        click.echo(rich_to_str(table), color=True)
+
+    else:
+        click.echo("No licenses configured!")

lightning_sdk/cli/license/set.py

@@ -0,0 +1,13 @@
+import click
+
+from lightning_sdk.utils.config import _DEFAULT_CONFIG_FILE_PATH, Config, DefaultConfigKeys
+
+
+@click.command("set")
+@click.argument("product_name")
+@click.argument("license_key")
+@click.option("--config-file", help="Path to the config file")
+def set_license(product_name: str, license_key: str, config_file: str = _DEFAULT_CONFIG_FILE_PATH) -> None:
+    """Set a license key for a given product."""
+    cfg = Config(config_file)
+    cfg.set(f"{DefaultConfigKeys.license}.{product_name}", license_key)

lightning_sdk/cli/mmt/__init__.py

@@ -0,0 +1,7 @@
+"""MMT CLI commands."""
+
+import click
+
+
+def register_commands(group: click.Group) -> None:
+    """Register MMT commands with the given group."""

lightning_sdk/cli/studio/__init__.py

@@ -0,0 +1,24 @@
+"""Studio CLI commands."""
+
+import click
+
+
+def register_commands(group: click.Group) -> None:
+    """Register studio commands with the given group."""
+    from lightning_sdk.cli.studio.connect import connect_studio
+    from lightning_sdk.cli.studio.create import create_studio
+    from lightning_sdk.cli.studio.delete import delete_studio
+    from lightning_sdk.cli.studio.list import list_studios
+    from lightning_sdk.cli.studio.ssh import ssh_studio
+    from lightning_sdk.cli.studio.start import start_studio
+    from lightning_sdk.cli.studio.stop import stop_studio
+    from lightning_sdk.cli.studio.switch import switch_studio
+
+    group.add_command(delete_studio)
+    group.add_command(create_studio)
+    group.add_command(list_studios)
+    group.add_command(ssh_studio)
+    group.add_command(start_studio)
+    group.add_command(stop_studio)
+    group.add_command(switch_studio)
+    group.add_command(connect_studio)

lightning_sdk/cli/studio/connect.py

@@ -0,0 +1,139 @@
+"""Studio connect command."""
+
+import subprocess
+import sys
+from contextlib import suppress
+from typing import Optional
+
+import click
+
+from lightning_sdk.cli.utils.get_base_studio import get_base_studio_id
+from lightning_sdk.cli.utils.handle_machine_and_gpus_args import handle_machine_and_gpus_args
+from lightning_sdk.cli.utils.richt_print import studio_name_link
+from lightning_sdk.cli.utils.save_to_config import save_studio_to_config, save_teamspace_to_config
+from lightning_sdk.cli.utils.ssh_connection import configure_ssh_internal
+from lightning_sdk.cli.utils.teamspace_selection import TeamspacesMenu
+from lightning_sdk.lightning_cloud.openapi.rest import ApiException
+from lightning_sdk.machine import CloudProvider, Machine
+from lightning_sdk.studio import Studio
+from lightning_sdk.utils.names import random_unique_name
+
+
+def _parse_args_or_get_from_current_studio(
+    teamspace: Optional[str],
+    cloud_account: Optional[str],
+    studio_type: Optional[str],
+    machine: Optional[str],
+    gpus: Optional[str],
+    cloud_provider: Optional[str],
+    name: Optional[str],
+) -> tuple[Optional[str], Optional[str], Optional[str], Optional[str]]:
+    # Parse args provided by user
+    menu = TeamspacesMenu()
+    resolved_teamspace = menu(teamspace)
+    save_teamspace_to_config(resolved_teamspace, overwrite=False)
+
+    template_id = get_base_studio_id(studio_type)
+
+    if cloud_provider is not None:
+        cloud_provider = CloudProvider(cloud_provider)
+
+    name = name or random_unique_name()
+
+    with suppress(ValueError):
+        # Gets current studio context to use its parameters as defaults
+        s = Studio()
+        if not teamspace:
+            resolved_teamspace = s.teamspace
+            save_teamspace_to_config(resolved_teamspace, overwrite=False)
+        if not cloud_account:
+            cloud_account = s.cloud_account
+        if not template_id:
+            template_id = s._studio.environment_template_id
+        if not machine and not gpus:
+            machine = s.machine
+
+    return resolved_teamspace, cloud_account, template_id, machine, cloud_provider, name
+
+
+@click.command("connect")
+@click.argument("name", required=False)
+@click.option("--teamspace", help="Override default teamspace (format: owner/teamspace)")
+@click.option(
+    "--cloud-provider",
+    help="The cloud provider to start the studio on. Defaults to teamspace default.",
+    type=click.Choice(m.name for m in list(CloudProvider)),
+)
+@click.option(
+    "--cloud-account",
+    help="The cloud account to create the studio on. Defaults to teamspace default.",
+    type=click.STRING,
+)
+@click.option(
+    "--machine",
+    help="The machine type to start the studio on. Defaults to CPU-4",
+    type=click.Choice(m.name for m in Machine.__dict__.values() if isinstance(m, Machine) and m._include_in_cli),
+)
+@click.option(
+    "--gpus",
+    help="The number and type of GPUs to start the studio on (format: TYPE:COUNT, e.g. L4:4)",
+    type=click.STRING,
+)
+@click.option(
+    "--studio-type",
+    help="The base studio template name to use for creating the studio. "
+    "Must be lowercase and hyphenated (use '-' instead of spaces). "
+    "Run 'lightning base-studio list' to see all available templates. "
+    "Defaults to the first available template.",
+    type=click.STRING,
+)
+@click.option("--interruptible", is_flag=True, help="Start the studio on an interruptible instance.")
+def connect_studio(
+    name: Optional[str] = None,
+    teamspace: Optional[str] = None,
+    cloud_provider: Optional[str] = None,
+    cloud_account: Optional[str] = None,
+    machine: Optional[str] = None,
+    gpus: Optional[str] = None,
+    studio_type: Optional[str] = None,
+    interruptible: bool = False,
+) -> None:
+    """Connect to a Studio.
+
+    Example:
+        lightning studio connect
+    """
+    teamspace, cloud_account, template_id, machine, cloud_provider, name = _parse_args_or_get_from_current_studio(
+        teamspace, cloud_account, studio_type, machine, gpus, cloud_provider, name
+    )
+
+    try:
+        studio = Studio(
+            name=name,
+            teamspace=teamspace,
+            create_ok=True,
+            cloud_provider=cloud_provider,
+            cloud_account=cloud_account,
+            studio_type=template_id,
+        )
+    except (RuntimeError, ValueError, ApiException):
+        raise ValueError(f"Could not create Studio: '{name}'") from None
+
+    click.echo(f"Connecting to Studio '{studio_name_link(studio)}' ...")
+
+    Studio.show_progress = True
+
+    machine = handle_machine_and_gpus_args(machine, gpus)
+
+    save_studio_to_config(studio)
+    studio.start(machine=machine, interruptible=interruptible)
+
+    ssh_private_key_path = configure_ssh_internal()
+
+    ssh_option = "-o UserKnownHostsFile=/dev/null -o StrictHostKeyChecking=no -o LogLevel=ERROR"
+    try:
+        ssh_command = f"ssh -i {ssh_private_key_path} {ssh_option} s_{studio._studio.id}@ssh.lightning.ai"
+        subprocess.run(ssh_command.split())
+    except Exception as ex:
+        print(f"Failed to establish SSH connection: {ex}")
+        sys.exit(1)