lightning-sdk 0.1.55__py3-none-any.whl → 0.1.57__py3-none-any.whl

lightning_sdk/cli/delete.py

@@ -1,5 +1,8 @@
 from typing import Optional
 
+import click
+from rich.console import Console
+
 from lightning_sdk.cli.exceptions import StudioCliError
 from lightning_sdk.cli.job_and_mmt_action import _JobAndMMTAction
 from lightning_sdk.cli.teamspace_menu import _TeamspacesMenu
@@ -18,15 +21,7 @@ class _Delete(_JobAndMMTAction, _TeamspacesMenu):
             teamspace: The teamspace to delete the container from. Should be specified as {owner}/{name}
                 If not provided, can be selected in an interactive menu.
         """
-        api = LitContainer()
-        resolved_teamspace = self._resolve_teamspace(teamspace=teamspace)
-        try:
-            api.delete_container(container, resolved_teamspace.name, resolved_teamspace.owner.name)
-            print(f"Container {container} deleted successfully.")
-        except Exception as e:
-            raise StudioCliError(
-                f"Could not delete container {container} from project {resolved_teamspace.name}: {e}"
-            ) from None
+        delete_container(container=container, teamspace=teamspace)
 
     def job(self, name: Optional[str] = None, teamspace: Optional[str] = None) -> None:
         """Delete a job.
@@ -38,10 +33,7 @@ class _Delete(_JobAndMMTAction, _TeamspacesMenu):
                 If not specified can be selected interactively.
 
         """
-        job = super().job(name=name, teamspace=teamspace)
-
-        job.delete()
-        print(f"Successfully deleted {job.name}!")
+        job(name=name, teamspace=teamspace)
 
     def mmt(self, name: Optional[str] = None, teamspace: Optional[str] = None) -> None:
         """Delete a multi-machine job.
@@ -53,10 +45,7 @@ class _Delete(_JobAndMMTAction, _TeamspacesMenu):
                 If not specified can be selected interactively.
 
         """
-        mmt = super().mmt(name=name, teamspace=teamspace)
-
-        mmt.delete()
-        print(f"Successfully deleted {mmt.name}!")
+        mmt(name=name, teamspace=teamspace)
 
     def studio(self, name: Optional[str] = None, teamspace: Optional[str] = None) -> None:
         """Delete an existing studio.
@@ -68,18 +57,80 @@ class _Delete(_JobAndMMTAction, _TeamspacesMenu):
             teamspace: The teamspace the studio is part of. Should be of format <OWNER>/<TEAMSPACE_NAME>.
                 If not specified, tries to infer from the environment (e.g. when run from within a Studio.)
         """
-        if teamspace is not None:
-            ts_splits = teamspace.split("/")
-            if len(ts_splits) != 2:
-                raise ValueError(f"Teamspace should be of format <OWNER>/<TEAMSPACE_NAME> but got {teamspace}")
-            owner, teamspace = ts_splits
-        else:
-            owner, teamspace = None, None
-
-        try:
-            studio = Studio(name=name, teamspace=teamspace, org=owner, user=None, create_ok=False)
-        except (RuntimeError, ValueError):
-            studio = Studio(name=name, teamspace=teamspace, org=None, user=owner, create_ok=False)
-
-        studio.delete()
-        print("Studio successfully deleted")
+        studio(name=name, teamspace=teamspace)
+
+
+@click.group()
+def delete() -> None:
+    """Delete resources on the Lightning AI platform."""
+
+
+# @delete.command(name="container")
+# @click.option("--container", help="The name of the container to delete.")
+# @click.option("--teamspace", default=None, help=("The teamspace to delete the container from. "
+#                                                  "Should be specified as {owner}/{name} "
+#                                                  "If not provided, can be selected in an interactive menu."),)
+def delete_container(container: str, teamspace: Optional[str] = None) -> None:
+    """Delete the docker container CONTAINER."""
+    api = LitContainer()
+    menu = _TeamspacesMenu()
+    resolved_teamspace = menu._resolve_teamspace(teamspace=teamspace)
+    try:
+        api.delete_container(container, resolved_teamspace.name, resolved_teamspace.owner.name)
+        Console().print(f"Container {container} deleted successfully.")
+    except Exception as e:
+        raise StudioCliError(
+            f"Could not delete container {container} from project {resolved_teamspace.name}: {e}"
+        ) from None
+
+
+# @delete.command(name="job")
+# @click.option("--name", help="The name of the job to delete.")
+# @click.option("--teamspace", default=None, help=("The teamspace to delete the job from. "
+#                                                  "Should be specified as {owner}/{name} "
+#                                                  "If not provided, can be selected in an interactive menu."),)
+def job(name: Optional[str] = None, teamspace: Optional[str] = None) -> None:
+    """Delete a job."""
+    menu = _JobAndMMTAction()
+    job = menu.job(name=name, teamspace=teamspace)
+
+    job.delete()
+    Console().print(f"Successfully deleted {job.name}!")
+
+
+# @delete.command(name="mmt")
+# @click.option("--name", help="The name of the multi-machine job to delete.")
+# @click.option("--teamspace", default=None, help=("The teamspace to delete the job from. "
+#                                                  "Should be specified as {owner}/{name} "
+#                                                  "If not provided, can be selected in an interactive menu."),)
+def mmt(name: Optional[str] = None, teamspace: Optional[str] = None) -> None:
+    """Delete a multi-machine job."""
+    menu = _JobAndMMTAction()
+    mmt = menu.mmt(name=name, teamspace=teamspace)
+
+    mmt.delete()
+    Console().print(f"Successfully deleted {mmt.name}!")
+
+
+# @delete.command(name="studio")
+# @click.option("--name", help="The name of the studio to delete.")
+# @click.option("--teamspace", default=None, help=("The teamspace to delete the studio from. "
+#                                                  "Should be specified as {owner}/{name} "
+#                                                  "If not provided, can be selected in an interactive menu."),)
+def studio(name: Optional[str] = None, teamspace: Optional[str] = None) -> None:
+    """Delete an existing studio."""
+    if teamspace is not None:
+        ts_splits = teamspace.split("/")
+        if len(ts_splits) != 2:
+            raise ValueError(f"Teamspace should be of format <OWNER>/<TEAMSPACE_NAME> but got {teamspace}")
+        owner, teamspace = ts_splits
+    else:
+        owner, teamspace = None, None
+
+    try:
+        studio = Studio(name=name, teamspace=teamspace, org=owner, user=None, create_ok=False)
+    except (RuntimeError, ValueError):
+        studio = Studio(name=name, teamspace=teamspace, org=None, user=owner, create_ok=False)
+
+    studio.delete()
+    Console().print("Studio successfully deleted")

lightning_sdk/cli/download.py

@@ -3,6 +3,7 @@ import re
 from pathlib import Path
 from typing import Optional
 
+import click
 from rich.console import Console
 
 from lightning_sdk.api.lit_container_api import LitContainerApi
@@ -25,51 +26,7 @@ class _Downloads(_StudiosMenu, _TeamspacesMenu):
             This should have the format <ORGANIZATION-NAME>/<TEAMSPACE-NAME>/<MODEL-NAME>.
           download_dir: The directory where the Model should be downloaded.
         """
-        download_model(
-            name=name,
-            download_dir=download_dir,
-            progress_bar=True,
-        )
-
-    def _resolve_studio(self, studio: Optional[str]) -> Studio:
-        user = _get_authed_user()
-        # if no studio specify suggest/filter only user's studios
-        possible_studios = self._get_possible_studios(user, is_owner=studio is None)
-
-        try:
-            if studio:
-                team_name, studio_name = studio.split("/")
-                options = [st for st in possible_studios if st["teamspace"] == team_name and st["name"] == studio_name]
-                if len(options) == 1:
-                    selected_studio = self._get_studio_from_name(studio, possible_studios)
-                # user can also use the partial studio name as secondary interactive selection
-                else:
-                    # filter matching simple reg expressions or start with the team and studio name
-                    possible_studios = filter(
-                        lambda st: (re.match(team_name, st["teamspace"]) or team_name in st["teamspace"])
-                        and (re.match(studio_name, st["name"]) or studio_name in st["name"]),
-                        possible_studios,
-                    )
-                    if not possible_studios:
-                        raise ValueError(
-                            f"Could not find Studio like '{studio}', please consider update your filtering pattern."
-                        )
-                    selected_studio = self._get_studio_from_interactive_menu(list(possible_studios))
-            else:
-                selected_studio = self._get_studio_from_interactive_menu(possible_studios)
-
-        except KeyboardInterrupt:
-            raise KeyboardInterrupt from None
-
-        # give user friendlier error message
-        except Exception as e:
-            raise StudioCliError(
-                f"Could not find the given Studio {studio} to upload files to. "
-                "Please contact Lightning AI directly to resolve this issue."
-            ) from e
-
-        with skip_studio_init():
-            return Studio(**selected_studio)
+        model(name=name, download_dir=download_dir)
 
     def folder(self, path: str = "", studio: Optional[str] = None, local_path: str = ".") -> None:
         """Download a folder from a Studio.
@@ -84,27 +41,9 @@ class _Downloads(_StudiosMenu, _TeamspacesMenu):
             with filtered studios will be shown for final selection.
           local_path: The path to the directory you want to download the folder to.
         """
-        local_path = Path(local_path)
-        if not local_path.is_dir():
-            raise NotADirectoryError(f"'{local_path}' is not a directory")
-
-        resolved_studio = self._resolve_studio(studio)
+        folder(path=path, studio=studio, local_path=local_path)
 
-        if not path:
-            local_path /= resolved_studio.name
-            path = ""
-
-        try:
-            if not path:
-                raise FileNotFoundError()
-            resolved_studio.download_folder(remote_path=path, target_path=str(local_path))
-        except Exception as e:
-            raise StudioCliError(
-                f"Could not download the folder from the given Studio {studio}. "
-                "Please contact Lightning AI directly to resolve this issue."
-            ) from e
-
-    def file(self, path: str = "", studio: Optional[str] = None, local_path: str = ".") -> None:
+    def file(self, path: str, studio: Optional[str] = None, local_path: str = ".") -> None:
         """Download a file from a Studio.
 
         Args:
@@ -115,25 +54,7 @@ class _Downloads(_StudiosMenu, _TeamspacesMenu):
             with filtered studios will be shown for final selection.
           local_path: The path to the directory you want to download the file to.
         """
-        local_path = Path(local_path)
-        if not local_path.is_dir():
-            raise NotADirectoryError(f"'{local_path}' is not a directory")
-
-        resolved_studio = self._resolve_studio(studio)
-
-        if not path:
-            local_path /= resolved_studio.name
-            path = ""
-
-        try:
-            if not path:
-                raise FileNotFoundError()
-            resolved_studio.download_file(remote_path=path, file_path=str(local_path / os.path.basename(path)))
-        except Exception as e:
-            raise StudioCliError(
-                f"Could not download the file from the given Studio {studio}. "
-                "Please contact Lightning AI directly to resolve this issue."
-            ) from e
+        file(path=path, studio=studio, local_path=local_path)
 
     def container(self, container: str, teamspace: Optional[str] = None, tag: str = "latest") -> None:
         """Download a docker container from a teamspace.
@@ -143,9 +64,175 @@ class _Downloads(_StudiosMenu, _TeamspacesMenu):
           teamspace: The name of the teamspace to download the container from.
           tag: The tag of the container to download.
         """
-        resolved_teamspace = self._resolve_teamspace(teamspace)
-        console = Console()
-        with console.status("Downloading container..."):
-            api = LitContainerApi()
-            api.download_container(container, resolved_teamspace, tag)
-            console.print("Container downloaded successfully", style="green")
+        download_container(container=container, teamspace=teamspace, tag=tag)
+
+
+@click.group(name="download")
+def download() -> None:
+    """Download resources from Lightning AI."""
+
+
+# @download.command(name="model")
+# @click.option(
+#     "--name",
+#     help=(
+#         "The name of the Model you want to download. "
+#         "This should have the format <ORGANIZATION-NAME>/<TEAMSPACE-NAME>/<MODEL-NAME>."
+#     ),
+# )
+# @click.option("--download-dir", default=".", help="The directory where the Model should be downloaded.")
+def model(name: str, download_dir: str = ".") -> None:
+    """Download a Model."""
+    download_model(
+        name=name,
+        download_dir=download_dir,
+        progress_bar=True,
+    )
+
+
+# @download.command(name="folder")
+# @click.option(
+#     "--path",
+#     default="",
+#     help=(
+#         "The relative path within the Studio you want to download. "
+#         "If you leave it empty it will download whole studio and locally creates a "
+#         "new folder with the same name as the selected studio."
+#     ),
+# )
+# @click.option(
+#     "--studio",
+#     default=None,
+#     help=(
+#         "The name of the studio to upload to. "
+#         "Will show a menu with user's owned studios for selection if not specified. "
+#         "If provided, should be in the form of <TEAMSPACE-NAME>/<STUDIO-NAME> where the names are case-sensitive. "
+#         "The teamspace and studio names can be regular expressions to match, "
+#         "a menu filtered studios will be shown for final selection."
+#     ),
+# )
+# @click.option("--local-path", default=".", help="The path to the directory you want to download the folder to.")
+def folder(path: str = "", studio: Optional[str] = None, local_path: str = ".") -> None:
+    """Download a folder from a Studio."""
+    local_path = Path(local_path)
+    if not local_path.is_dir():
+        raise NotADirectoryError(f"'{local_path}' is not a directory")
+
+    menu = _StudiosMenu()
+    resolved_studio = menu._resolve_studio(studio)
+
+    if not path:
+        local_path /= resolved_studio.name
+        path = ""
+
+    try:
+        if not path:
+            raise FileNotFoundError()
+        resolved_studio.download_folder(remote_path=path, target_path=str(local_path))
+    except Exception as e:
+        raise StudioCliError(
+            f"Could not download the folder from the given Studio {studio}. "
+            "Please contact Lightning AI directly to resolve this issue."
+        ) from e
+
+
+# @download.command(name="file")
+# @click.option(
+#     "--path",
+#     default="",
+#     help=(
+#         "The relative path within the Studio you want to download. "
+#         "If you leave it empty it will download whole studio and locally creates a new folder "
+#         "with the same name as the selected studio."
+#     ),
+# )
+# @click.option(
+#     "--studio",
+#     default=None,
+#     help=(
+#         "The name of the studio to upload to. "
+#         "Will show a menu with user's owned studios for selection if not specified. "
+#         "If provided, should be in the form of <TEAMSPACE-NAME>/<STUDIO-NAME> where the names are case-sensitive. "
+#         "The teamspace and studio names can be regular expressions to match, "
+#         "a menu filtered studios will be shown for final selection."
+#     ),
+# )
+# @click.option("--local-path", default=".", help="The path to the directory you want to download the folder to.")
+def file(path: str = "", studio: Optional[str] = None, local_path: str = ".") -> None:
+    """Download a file from a Studio."""
+    local_path = Path(local_path)
+    if not local_path.is_dir():
+        raise NotADirectoryError(f"'{local_path}' is not a directory")
+
+    resolved_studio = _resolve_studio(studio)
+
+    if not path:
+        local_path /= resolved_studio.name
+        path = ""
+
+    try:
+        if not path:
+            raise FileNotFoundError()
+        resolved_studio.download_file(remote_path=path, file_path=str(local_path / os.path.basename(path)))
+    except Exception as e:
+        raise StudioCliError(
+            f"Could not download the file from the given Studio {studio}. "
+            "Please contact Lightning AI directly to resolve this issue."
+        ) from e
+
+
+# @download.command(name="container")
+# @click.argument("container")
+# @click.argument("teamspace", default=None, help="The name of the teamspace to download the container from")
+# @click.argument("tag", default="latest", show_default=True, help="The tag of the container to download.")
+def download_container(container: str, teamspace: Optional[str] = None, tag: str = "latest") -> None:
+    """Download the docker container CONTAINER from a teamspace."""
+    console = Console()
+    menu = _TeamspacesMenu()
+    resolved_teamspace = menu._resolve_teamspace(teamspace)
+    with console.status("Downloading container..."):
+        api = LitContainerApi()
+        api.download_container(container, resolved_teamspace, tag)
+        console.print("Container downloaded successfully", style="green")
+
+
+def _resolve_studio(studio: Optional[str]) -> Studio:
+    user = _get_authed_user()
+    # if no studio specify suggest/filter only user's studios
+    menu = _StudiosMenu()
+    possible_studios = menu._get_possible_studios(user, is_owner=studio is None)
+
+    try:
+        if studio:
+            team_name, studio_name = studio.split("/")
+            options = [st for st in possible_studios if st["teamspace"] == team_name and st["name"] == studio_name]
+            if len(options) == 1:
+                selected_studio = menu._get_studio_from_name(studio, possible_studios)
+            # user can also use the partial studio name as secondary interactive selection
+            else:
+                # filter matching simple reg expressions or start with the team and studio name
+                possible_studios = filter(
+                    lambda st: (re.match(team_name, st["teamspace"]) or team_name in st["teamspace"])
+                    and (re.match(studio_name, st["name"]) or studio_name in st["name"]),
+                    possible_studios,
+                )
+                if not possible_studios:
+                    raise ValueError(
+                        f"Could not find Studio like '{studio}', please consider update your filtering pattern."
+                    )
+                selected_studio = menu._get_studio_from_interactive_menu(list(possible_studios))
+        else:
+            selected_studio = menu._get_studio_from_interactive_menu(possible_studios)
+
+    except KeyboardInterrupt:
+        raise KeyboardInterrupt from None
+
+    # give user friendlier error message
+    except Exception as e:
+        raise StudioCliError(
+            f"Could not find the given Studio {studio} to upload files to. "
+            "Please contact Lightning AI directly to resolve this issue."
+        ) from e
+
+    with skip_studio_init():
+        return Studio(**selected_studio)

lightning_sdk/cli/entrypoint.py

@@ -2,15 +2,20 @@ import sys
 from types import TracebackType
 from typing import Type
 
+import click
 from fire import Fire
 from lightning_utilities.core.imports import RequirementCache
+from rich.console import Console
+from rich.panel import Panel
 
 from lightning_sdk.api.studio_api import _cloud_url
-from lightning_sdk.cli.ai_hub import _AIHub
-from lightning_sdk.cli.delete import _Delete
-from lightning_sdk.cli.download import _Downloads
-from lightning_sdk.cli.generate import _Generate
-from lightning_sdk.cli.inspect import _Inspect
+from lightning_sdk.cli.ai_hub import _AIHub, aihub
+from lightning_sdk.cli.configure import _Configure, configure
+from lightning_sdk.cli.connect import _Connect, connect
+from lightning_sdk.cli.delete import _Delete, delete
+from lightning_sdk.cli.download import _Downloads, download
+from lightning_sdk.cli.generate import _Generate, generate
+from lightning_sdk.cli.inspect import _Inspect, inspect
 from lightning_sdk.cli.legacy import _LegacyLightningCLI
 from lightning_sdk.cli.list import _List
 from lightning_sdk.cli.run import _Run
@@ -41,28 +46,24 @@ class StudioCLI:
         self.start = _Start()
         self.switch = _Switch()
         self.generate = _Generate()
+        self.connect = _Connect()
+        self.configure = _Configure()
 
         sys.excepthook = _notify_exception
 
     def login(self) -> 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
+        return login()
 
     def logout(self) -> None:
         """Logout from Lightning AI Studios."""
-        auth = Auth()
-        auth.clear()
+        return logout()
 
 
 def _notify_exception(exception_type: Type[BaseException], value: BaseException, tb: TracebackType) -> None:  # No
     """CLI won't show tracebacks, just print the exception message."""
-    print(value)
+    console = Console()
+    console.print(Panel(value))
 
 
 def main_cli() -> None:
@@ -70,5 +71,39 @@ def main_cli() -> None:
     Fire(StudioCLI(), name="lightning")
 
 
+@click.group(name="lightning", help="Command line interface (CLI) to interact with/manage Lightning AI Studios.")
+def main_cli_click() -> None:
+    pass
+
+
+# @main_cli_click.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_click.command
+def logout() -> None:
+    """Logout from Lightning AI Studios."""
+    auth = Auth()
+    auth.clear()
+
+
+# TODO: handle exception hook registration
+main_cli_click.add_command(aihub)
+main_cli_click.add_command(configure)
+main_cli_click.add_command(connect)
+main_cli_click.add_command(delete)
+main_cli_click.add_command(download)
+main_cli_click.add_command(generate)
+main_cli_click.add_command(inspect)
+
+
 if __name__ == "__main__":
     main_cli()

lightning_sdk/cli/generate.py

@@ -1,58 +1,67 @@
 from typing import Optional
 
+import click
 from rich.console import Console
 
-from lightning_sdk import Studio
+from lightning_sdk.cli.studios_menu import _StudiosMenu
 
 
 class _Generate:
     """Generate configs (such as ssh for studio) and print them to commandline."""
 
-    console = Console()
-
-    def _generate_ssh_config(self, name: str, studio_id: str) -> str:
-        """Generate SSH config entry for the studio.
-
-        Args:
-            name: Studio name
-            studio_id: Studio space ID
-
-        Returns:
-            str: SSH config entry
-        """
-        return f"""# ssh s_{studio_id}@ssh.lightning.ai
-
-Host {name}
-  User s_{studio_id}
-  Hostname ssh.lightning.ai
-  IdentityFile ~/.ssh/lightning_rsa
-  IdentitiesOnly yes
-  ServerAliveInterval 15
-  ServerAliveCountMax 4
-  StrictHostKeyChecking no
-  UserKnownHostsFile=/dev/null"""
-
     def ssh(self, name: Optional[str] = None, teamspace: Optional[str] = None) -> None:
-        """Get SSH config entry for a studio. Will start the studio if needed.
+        """Get SSH config entry for a studio.
 
         Args:
-            name: The name of the studio to stop.
+            name: The name of the studio to obtain SSH config.
                 If not specified, tries to infer from the environment (e.g. when run from within a Studio.)
             teamspace: The teamspace the studio is part of. Should be of format <OWNER>/<TEAMSPACE_NAME>.
                 If not specified, tries to infer from the environment (e.g. when run from within a Studio.)
         """
-        if teamspace:
-            ts_splits = teamspace.split("/")
-            if len(ts_splits) != 2:
-                raise ValueError(f"Teamspace should be of format <OWNER>/<TEAMSPACE_NAME> but got {teamspace}")
-            owner, teamspace = ts_splits
-        else:
-            owner, teamspace = None, None
-
-        try:
-            studio = Studio(name=name, teamspace=teamspace, org=owner, user=None, create_ok=False)
-        except (RuntimeError, ValueError):
-            studio = Studio(name=name, teamspace=teamspace, org=None, user=owner, create_ok=False)
-
-        # Print the SSH config
-        self.console.print(self._generate_ssh_config(name, studio._studio.id))
+        ssh(name=name, teamspace=teamspace)
+
+
+def _generate_ssh_config(key_path: str, host: str, user: str) -> str:
+    return f"""Host {host}
+      User {user}
+      Hostname ssh.lightning.ai
+      IdentityFile {key_path}
+      IdentitiesOnly yes
+      ServerAliveInterval 15
+      ServerAliveCountMax 4
+      StrictHostKeyChecking no
+      UserKnownHostsFile=/dev/null
+    """
+
+
+@click.group(name="generate")
+def generate() -> None:
+    """Generate configs (such as ssh for studio) and print them to commandline."""
+
+
+# @generate.command(name="ssh")
+# @click.option(
+#     "--name",
+#     default=None,
+#     help=(
+#         "The name of the studio to obtain SSH config. "
+#         "If not specified, tries to infer from the environment (e.g. when run from within a Studio.)"
+#     ),
+# )
+# @click.option(
+#     "--teamspace",
+#     default=None,
+#     help=(
+#         "The teamspace the studio is part of. "
+#         "Should be of format <OWNER>/<TEAMSPACE_NAME>. "
+#         "If not specified, tries to infer from the environment (e.g. when run from within a Studio.)"
+#     ),
+# )
+def ssh(name: Optional[str] = None, teamspace: Optional[str] = None) -> None:
+    """Get SSH config entry for a studio."""
+    menu = _StudiosMenu()
+    studio = menu._get_studio(name=name, teamspace=teamspace)
+
+    conf = _generate_ssh_config(key_path="~/.ssh/lightning_rsa", user=f"s_{studio._studio.id}", host=studio.name)
+    # Print the SSH config
+    Console().print(f"# ssh s_{studio._studio.id}@ssh.lightning.ai\n\n" + conf)