lightning-sdk 2025.8.14.post0__py3-none-any.whl → 2025.8.18.post0__py3-none-any.whl

lightning_sdk/__init__.py

@@ -32,6 +32,6 @@ __all__ = [
     "User",
 ]
 
-__version__ = "2025.08.14.post0"
+__version__ = "2025.08.18.post0"
 _check_version_and_prompt_upgrade(__version__)
 _set_tqdm_envvars_noninteractive()

lightning_sdk/api/cloud_account_api.py

@@ -181,6 +181,11 @@ class CloudAccountApi:
         cloud_provider: Optional[Union["CloudProvider", str]],
         default_cloud_account: Optional[str],
     ) -> Optional[str]:
+        from lightning_sdk.machine import CloudProvider
+
+        if cloud_provider and not isinstance(cloud_provider, CloudProvider):
+            cloud_provider = CloudProvider(cloud_provider)
+
         if cloud_account:
             if cloud_provider:
                 cloud_account_resp = self.get_cloud_account_non_org(teamspace_id, cloud_account)

lightning_sdk/cli/__init__.py

@@ -0,0 +1 @@
+"""Lightning SDK CLI module."""

lightning_sdk/cli/config/__init__.py

@@ -0,0 +1,14 @@
+"""Config CLI commands."""
+
+import click
+
+
+def register_commands(group: click.Group) -> None:
+    """Register config commands with the given group."""
+    from lightning_sdk.cli.config.get import get
+    from lightning_sdk.cli.config.set import set_value
+    from lightning_sdk.cli.config.show import show
+
+    group.add_command(get)
+    group.add_command(set_value)
+    group.add_command(show)

lightning_sdk/cli/config/get.py

@@ -0,0 +1,57 @@
+import click
+
+from lightning_sdk.utils.config import Config, DefaultConfigKeys
+
+
+@click.group("get")
+def get() -> None:
+    """Get configuration values."""
+
+
+@get.command("user")
+def get_user() -> None:
+    """Get the default user name from the config."""
+    config = Config()
+    user = config.get_value(DefaultConfigKeys.user)
+    click.echo(user)
+
+
+@get.command("org")
+def get_org() -> None:
+    """Get the default organization name from the config."""
+    config = Config()
+    org = config.get_value(DefaultConfigKeys.organization)
+    click.echo(org)
+
+
+@get.command("teamspace")
+def get_teamspace() -> None:
+    """Get the default teamspace name from the config."""
+    config = Config()
+    teamspace_name = config.get_value(DefaultConfigKeys.teamspace_name)
+    teamspace_owner = config.get_value(DefaultConfigKeys.teamspace_owner)
+    click.echo(f"{teamspace_owner}/{teamspace_name}")
+
+
+@get.command("studio")
+def get_studio() -> None:
+    """Get the default sutdio name from the config."""
+    config = Config()
+    studio = config.get_value(DefaultConfigKeys.studio)
+    click.echo(studio)
+
+
+@get.command("cloud-account")
+def get_cloud_account() -> None:
+    """Get the default cloud account name from the config."""
+    config = Config()
+    cloud_account = config.get_value(DefaultConfigKeys.cloud_account)
+    click.echo(cloud_account)
+
+
+@get.command("cloud-provider")
+def get_cloud_provider() -> None:
+    """Get the default cloud provider name from the config."""
+    config = Config()
+    cloud_provider = config.get_value(DefaultConfigKeys.cloud_provider)
+    click.echo(cloud_provider)

lightning_sdk/cli/config/set.py

@@ -0,0 +1,104 @@
+import click
+
+from lightning_sdk.cli.utils.resolve import resolve_teamspace_owner_name_format
+from lightning_sdk.machine import CloudProvider
+from lightning_sdk.organization import Organization
+from lightning_sdk.studio import Studio
+from lightning_sdk.utils.config import Config, DefaultConfigKeys
+from lightning_sdk.utils.resolve import _resolve_org, _resolve_user
+
+
+@click.group("set")
+def set_value() -> None:
+    """Set configuration values."""
+
+
+@set_value.command("user")
+@click.argument("user_name")
+def set_user(user_name: str) -> None:
+    """Set the default user name in the config."""
+    try:
+        _resolve_user(user_name)
+    except Exception:
+        # TODO: make this a generic CLI error
+        raise ValueError(f"Could not resolve user: '{user_name}'. Does the user exist?") from None
+
+    config = Config()
+    setattr(config, DefaultConfigKeys.user, user_name)
+
+
+@set_value.command("org")
+@click.argument("org_name")
+def set_org(org_name: str) -> None:
+    """Set the default organization name in the config."""
+    try:
+        _resolve_org(org_name)
+    except Exception:
+        # TODO: make this a generic CLI error
+        raise ValueError(f"Could not resolve organization: '{org_name}'. Does the organization exist?") from None
+
+    config = Config()
+    setattr(config, DefaultConfigKeys.organization, org_name)
+
+
+@set_value.command("studio")
+@click.argument("studio_name")
+def set_studio(studio_name: str) -> None:
+    """Set the default studio name in the config."""
+    try:
+        studio = Studio(studio_name)
+    except Exception:
+        # TODO: make this a generic CLI error
+        raise ValueError(f"Could not resolve studio: '{studio_name}'. Does the studio exist?") from None
+
+    config = Config()
+    setattr(config, DefaultConfigKeys.studio, studio.name)
+
+
+@set_value.command("teamspace")
+@click.argument("teamspace_name")
+def set_teamspace(teamspace_name: str) -> None:
+    """Set the default teamspace name in the config."""
+    config = Config()
+
+    teamspace_resolved = resolve_teamspace_owner_name_format(teamspace_name)
+
+    if teamspace_resolved is None:
+        # TODO: make this a generic CLI error
+        raise ValueError(
+            f"Could not resolve teamspace: '{teamspace_name}'. "
+            "Teamspace should be specified as 'owner/name'. Does the teamspace exist?"
+        )
+
+    setattr(config, DefaultConfigKeys.teamspace_name, teamspace_resolved.name)
+    setattr(config, DefaultConfigKeys.teamspace_owner, teamspace_resolved.owner.name)
+    if isinstance(teamspace_resolved.owner, Organization):
+        setattr(config, DefaultConfigKeys.teamspace_owner_type, "organization")
+    else:
+        setattr(config, DefaultConfigKeys.teamspace_owner_type, "user")
+
+
+@set_value.command("cloud-account")
+@click.argument("cloud_account_name")
+def set_cloud_account(cloud_account_name: str) -> None:
+    """Set the default cloud account name in the config."""
+    config = Config()
+    setattr(config, DefaultConfigKeys.cloud_account, cloud_account_name)
+
+
+@set_value.command("cloud-provider")
+@click.argument("cloud_provider_name")
+def set_cloud_provider(cloud_provider_name: str) -> None:
+    """Set the default cloud provider name in the config."""
+    config = Config()
+
+    try:
+        cloud_provider = CloudProvider(cloud_provider_name)
+    except ValueError:
+        # TODO: make this a generic CLI error
+        raise ValueError(
+            f"Could not resolve cloud provider: '{cloud_provider_name}'. "
+            f"Supported values are: {', '.join(m.name for m in list(CloudProvider))}"
+        ) from None
+
+    setattr(config, DefaultConfigKeys.cloud_provider, cloud_provider.name)

lightning_sdk/cli/config/show.py

@@ -0,0 +1,9 @@
+import click
+
+from lightning_sdk.utils.config import Config
+
+
+@click.command("show")
+def show() -> None:
+    """Show configuration values."""
+    click.echo(Config())

lightning_sdk/cli/entrypoint.py

@@ -1,42 +1,34 @@
+"""New Lightning CLI entrypoint with organized command groups."""
+
+import os
 import sys
 import traceback
 from types import TracebackType
 from typing import Type
 
 import click
-from rich.console import Console, Group
+from rich.console import 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.ai_hub import aihub
-from lightning_sdk.cli.coloring import CustomHelpFormatter
-from lightning_sdk.cli.configure import configure
-from lightning_sdk.cli.connect import connect
-from lightning_sdk.cli.create import create
-from lightning_sdk.cli.delete import delete
-from lightning_sdk.cli.deploy.serve import deploy
-from lightning_sdk.cli.docker_cli import dockerize
-from lightning_sdk.cli.download import download
-from lightning_sdk.cli.generate import generate
-from lightning_sdk.cli.inspection import inspect
-from lightning_sdk.cli.list import list_cli
-from lightning_sdk.cli.open import open
-from lightning_sdk.cli.run import run
-from lightning_sdk.cli.start import start
-from lightning_sdk.cli.stop import stop
-from lightning_sdk.cli.switch import switch
-from lightning_sdk.cli.upload import upload
+
+# Import legacy groups directly from groups.py
+from lightning_sdk.cli.groups import (
+    config,
+    # job,
+    # mmt,
+    studio,
+)
+from lightning_sdk.cli.utils import CustomHelpFormatter, rich_to_str
 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()
@@ -54,7 +46,8 @@ def _notify_exception(exception_type: Type[BaseException], value: BaseException,
 
     renderables.append(Text("\n📘 Need help? Run: lightning <command> --help", style="cyan"))
 
-    console.print(Panel(Group(*renderables), title="⚡ Lightning CLI Error", border_style="red"))
+    text = rich_to_str(Panel(Group(*renderables), title="⚡ Lightning CLI Error", border_style="red"))
+    click.echo(text, color=True)
 
 
 @click.group(name="lightning", help="Command line interface (CLI) to interact with/manage Lightning AI Studios.")
@@ -63,7 +56,6 @@ def main_cli() -> None:
     sys.excepthook = _notify_exception
 
 
-# colorful help messages
 main_cli.context_class.formatter_class = CustomHelpFormatter
 
 
@@ -86,24 +78,51 @@ def logout() -> None:
     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)
+# Add new command groups
+main_cli.add_command(config)
+# main_cli.add_command(job)
+# main_cli.add_command(mmt)
+main_cli.add_command(studio)
+if os.environ.get("LIGHTNING_EXPERIMENTAL_CLI_ONLY", "0") != "1":
+    #### LEGACY COMMANDS ####
+    # these commands are currently supported for backwards compatibility, but will potentially be removed in the future.
+    # they've grown pretty wild and provide a very inconsistent UX.
+    from lightning_sdk.cli.legacy.ai_hub import aihub
+    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 as open_cmd
+    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
+
+    # Add old command groups
+    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(deploy)
+    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(open_cmd)
+    main_cli.add_command(run)
+    main_cli.add_command(start)
+    main_cli.add_command(stop)
+    main_cli.add_command(switch)
+    main_cli.add_command(upload)
 
 
 if __name__ == "__main__":

lightning_sdk/cli/groups.py

@@ -0,0 +1,35 @@
+"""CLI groups for organizing Lightning SDK commands."""
+
+import click
+
+from lightning_sdk.cli.config import register_commands as register_config_commands
+from lightning_sdk.cli.job import register_commands as register_job_commands
+from lightning_sdk.cli.mmt import register_commands as register_mmt_commands
+from lightning_sdk.cli.studio import register_commands as register_studio_commands
+
+
+@click.group(name="studio")
+def studio() -> None:
+    """Manage Lightning AI Studios."""
+
+
+@click.group(name="job")
+def job() -> None:
+    """Manage Lightning AI Jobs."""
+
+
+@click.group(name="mmt")
+def mmt() -> None:
+    """Manage Lightning AI Multi-Machine Training (MMT)."""
+
+
+@click.group(name="config")
+def config() -> None:
+    """Manage Lightning SDK and CLIconfiguration."""
+
+
+# Register config commands with the main config group
+register_job_commands(job)
+register_mmt_commands(mmt)
+register_studio_commands(studio)
+register_config_commands(config)

lightning_sdk/cli/job/__init__.py

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

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

@@ -7,8 +7,8 @@ from typing import Optional, Union
 import click
 from rich.console import Console
 
-from lightning_sdk.cli.generate import _generate_ssh_config
-from lightning_sdk.cli.studios_menu import _StudiosMenu
+from lightning_sdk.cli.legacy.generate import _generate_ssh_config
+from lightning_sdk.cli.legacy.studios_menu import _StudiosMenu
 from lightning_sdk.lightning_cloud.login import Auth
 
 

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

@@ -4,8 +4,8 @@ from typing import Optional
 
 import click
 
-from lightning_sdk.cli.configure import _configure_ssh_internal
-from lightning_sdk.cli.studios_menu import _StudiosMenu
+from lightning_sdk.cli.legacy.configure import _configure_ssh_internal
+from lightning_sdk.cli.legacy.studios_menu import _StudiosMenu
 
 
 @click.group(name="connect")

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

@@ -8,7 +8,7 @@ from rich.console import Console
 
 from lightning_sdk import Machine, Studio
 from lightning_sdk.api.cloud_account_api import CloudAccountApi
-from lightning_sdk.cli.teamspace_menu import _TeamspacesMenu
+from lightning_sdk.cli.legacy.teamspace_menu import _TeamspacesMenu
 from lightning_sdk.machine import CloudProvider
 from lightning_sdk.utils.resolve import _resolve_deprecated_provider
 

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

@@ -3,9 +3,9 @@ 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
+from lightning_sdk.cli.legacy.exceptions import StudioCliError
+from lightning_sdk.cli.legacy.job_and_mmt_action import _JobAndMMTAction
+from lightning_sdk.cli.legacy.teamspace_menu import _TeamspacesMenu
 from lightning_sdk.lightning_cloud.openapi.rest import ApiException
 from lightning_sdk.lit_container import LitContainer
 from lightning_sdk.studio import Studio

lightning_sdk/cli/{deploy → legacy/deploy}/_auth.py

@@ -10,7 +10,7 @@ from rich.prompt import Confirm
 
 from lightning_sdk import Teamspace
 from lightning_sdk.api import UserApi
-from lightning_sdk.cli.teamspace_menu import _TeamspacesMenu
+from lightning_sdk.cli.legacy.teamspace_menu import _TeamspacesMenu
 from lightning_sdk.lightning_cloud import env
 from lightning_sdk.lightning_cloud.login import Auth, AuthServer
 from lightning_sdk.lightning_cloud.openapi import V1CloudSpace

lightning_sdk/cli/{deploy → legacy/deploy}/devbox.py

@@ -12,8 +12,14 @@ from rich.prompt import Confirm
 from rich.syntax import Syntax
 
 from lightning_sdk import Machine
-from lightning_sdk.cli.deploy._auth import _AuthMode, _Onboarding, authenticate, poll_verified_status, select_teamspace
-from lightning_sdk.cli.upload import (
+from lightning_sdk.cli.legacy.deploy._auth import (
+    _AuthMode,
+    _Onboarding,
+    authenticate,
+    poll_verified_status,
+    select_teamspace,
+)
+from lightning_sdk.cli.legacy.upload import (
     _dump_current_upload_state,
     _resolve_previous_upload_state,
     _single_file_upload,

lightning_sdk/cli/{deploy → legacy/deploy}/serve.py

@@ -15,15 +15,15 @@ from rich.prompt import Confirm
 from lightning_sdk import Machine, Teamspace
 from lightning_sdk.api.lit_container_api import LitContainerApi
 from lightning_sdk.api.utils import _get_registry_url
-from lightning_sdk.cli.clusters_menu import _ClustersMenu
-from lightning_sdk.cli.deploy._auth import (
+from lightning_sdk.cli.legacy.clusters_menu import _ClustersMenu
+from lightning_sdk.cli.legacy.deploy._auth import (
     _AuthMode,
     _Onboarding,
     authenticate,
     poll_verified_status,
     select_teamspace,
 )
-from lightning_sdk.cli.deploy.devbox import _handle_devbox
+from lightning_sdk.cli.legacy.deploy.devbox import _handle_devbox
 from lightning_sdk.serve import _LitServeDeployer
 
 _MACHINE_VALUES = tuple(

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

@@ -9,9 +9,9 @@ from rich.console import Console
 
 from lightning_sdk.api.license_api import LicenseApi
 from lightning_sdk.api.lit_container_api import LitContainerApi
-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.legacy.teamspace_menu import _TeamspacesMenu
 from lightning_sdk.models import download_model
 from lightning_sdk.studio import Studio
 from lightning_sdk.utils.resolve import _get_authed_user, skip_studio_init

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", 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,8 +1,8 @@
 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.legacy.teamspace_menu import _TeamspacesMenu
 from lightning_sdk.job import Job
 from lightning_sdk.mmt import MMT
 

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