lightning-sdk 2025.9.16__py3-none-any.whl → 2025.9.29__py3-none-any.whl

lightning_sdk/__init__.py

@@ -10,14 +10,16 @@ from lightning_sdk.organization import Organization
 from lightning_sdk.plugin import JobsPlugin, MultiMachineTrainingPlugin, Plugin, SlurmJobsPlugin
 from lightning_sdk.status import Status
 from lightning_sdk.studio import Studio
-from lightning_sdk.teamspace import FolderLocation, Teamspace
+from lightning_sdk.teamspace import ConnectionType, FolderLocation, Teamspace
 from lightning_sdk.user import User
 
 __all__ = [
     "AIHub",
     "Agent",
     "CloudProvider",
+    "ConnectionType",
     "Deployment",
+    "FolderLocation",
     "Job",
     "JobsPlugin",
     "Machine",
@@ -29,10 +31,9 @@ __all__ = [
     "Status",
     "Studio",
     "Teamspace",
-    "FolderLocation",
     "User",
 ]
 
-__version__ = "2025.09.16"
+__version__ = "2025.09.29"
 _check_version_and_prompt_upgrade(__version__)
 _set_tqdm_envvars_noninteractive()

lightning_sdk/api/cloud_account_api.py

@@ -13,6 +13,7 @@ from lightning_sdk.lightning_cloud.rest_client import LightningClient
 
 if TYPE_CHECKING:
     from lightning_sdk.machine import CloudProvider
+    from lightning_sdk.teamspace import ConnectionType
 
 
 class CloudAccountApi:
@@ -133,7 +134,7 @@ class CloudAccountApi:
 
     def get_cloud_account_provider_mapping(self, teamspace_id: str) -> Dict["CloudProvider", V1ExternalCluster]:
         """Gets the cloud account <-> provider mapping."""
-        res = self.list_global_cloud_accounts(teamspace_id=teamspace_id)
+        res = self.list_cloud_accounts(teamspace_id=teamspace_id)
         cloud_accounts = {cloud_account.id: cloud_account for cloud_account in res}
         providers = {cloud_account.id: self._get_cloud_account_provider(cloud_account) for cloud_account in res}
 
@@ -214,3 +215,13 @@ class CloudAccountApi:
             return default_cloud_account
 
         return None
+
+    @staticmethod
+    def get_cloud_provider_for_connection_type(connection_type: "ConnectionType") -> "CloudProvider":
+        from lightning_sdk.machine import CloudProvider
+        from lightning_sdk.teamspace import ConnectionType
+
+        if connection_type == ConnectionType.EFS:
+            return CloudProvider.AWS
+
+        raise ValueError(f"ConnectionType {ConnectionType} currently not supported!")

lightning_sdk/api/job_api.py

@@ -104,22 +104,23 @@ class JobApiV1:
             org_id=org_id,
         )
 
-        identifier = None
+        identifiers = []
 
         if user_requested_compute_config and user_requested_compute_config.name:
-            identifier = user_requested_compute_config.name
+            identifiers.append(user_requested_compute_config.name)
         else:
-            identifier = spec.compute_config.instance_type
+            identifiers.append(spec.compute_config.instance_type)
 
         for accelerator in accelerators:
-            if identifier in (
-                accelerator.slug,
-                accelerator.slug_multi_cloud,
-                accelerator.instance_id,
-            ):
-                return Machine.from_str(accelerator.slug_multi_cloud)
+            for ident in identifiers:
+                if ident in (
+                    accelerator.slug,
+                    accelerator.slug_multi_cloud,
+                    accelerator.instance_id,
+                ):
+                    return Machine._from_accelerator(accelerator)
 
-        return Machine.from_str(identifier)
+        return Machine.from_str(identifiers[0])
 
     def _get_machines_for_cloud_account(
         self, teamspace_id: str, cloud_account_id: str, org_id: str
@@ -431,7 +432,7 @@ class JobApiV2:
             if (spec.instance_name and spec.instance_name in possible_identifiers) or (
                 spec.instance_type and spec.instance_type in possible_identifiers
             ):
-                return Machine.from_str(accelerator.slug_multi_cloud)
+                return Machine._from_accelerator(accelerator)
 
         return Machine.from_str(spec.instance_name or spec.instance_type)
 

lightning_sdk/api/mmt_api.py

@@ -265,7 +265,7 @@ class MMTApiV2:
             if (spec.instance_name and spec.instance_name in possible_identifiers) or (
                 spec.instance_type and spec.instance_type in possible_identifiers
             ):
-                return Machine.from_str(accelerator.slug_multi_cloud)
+                return Machine._from_accelerator(accelerator)
 
         return Machine.from_str(spec.instance_name or spec.instance_type)
 

lightning_sdk/api/studio_api.py

@@ -426,7 +426,7 @@ class StudioApi:
                 accelerator.slug_multi_cloud,
                 accelerator.instance_id,
             ):
-                return Machine.from_str(accelerator.slug_multi_cloud)
+                return Machine._from_accelerator(accelerator)
 
         return Machine.from_str(response.compute_config.name)
 

lightning_sdk/api/teamspace_api.py

@@ -28,6 +28,7 @@ from lightning_sdk.lightning_cloud.openapi import (
     V1Assistant,
     V1CloudSpace,
     V1ClusterAccelerator,
+    V1EfsConfig,
     V1Endpoint,
     V1ExternalCluster,
     V1GCSFolderDataConnection,
@@ -515,3 +516,20 @@ class TeamspaceApi:
                 create_request.gcs_folder = V1GCSFolderDataConnection()
 
         self._client.data_connection_service_create_data_connection(create_request, teamspace_id)
+
+    def new_connection(
+        self, teamspace_id: str, name: str, source: str, cluster: V1ExternalCluster, writable: bool, region: str
+    ) -> None:
+        create_request = Create(
+            name=name,
+            create_resources=False,
+            force=True,
+            writable=writable,
+            cluster_id=cluster.id,
+            access_cluster_ids=[cluster.id],
+        )
+
+        # TODO: Add support for other connection types
+        create_request.efs = V1EfsConfig(file_system_id=source, region=region)
+
+        self._client.data_connection_service_create_data_connection(create_request, teamspace_id)

lightning_sdk/api/user_api.py

@@ -1,5 +1,5 @@
 import re
-from typing import Dict, List, Union
+from typing import Dict, List, Optional, Union
 
 from lightning_sdk.lightning_cloud.login import Auth
 from lightning_sdk.lightning_cloud.openapi import (
@@ -61,8 +61,14 @@ class UserApi:
     def _get_all_teamspace_memberships(
         self,
         user_id: str,  # todo: this is unused, but still required
+        org_id: Optional[str] = None,
     ) -> List[V1Membership]:
-        return self._client.projects_service_list_memberships(filter_by_user_id=True).memberships
+        kwargs: Dict[str, Union[bool, str]] = {"filter_by_user_id": True}
+
+        if org_id is not None:
+            kwargs["organization_id"] = org_id
+
+        return self._client.projects_service_list_memberships(**kwargs).memberships
 
     def _get_authed_user_name(self) -> str:
         """Gets the currently logged-in user."""

lightning_sdk/cli/entrypoint.py

@@ -21,6 +21,7 @@ from lightning_sdk.cli.groups import (
     # job,
     # mmt,
     studio,
+    vm,
 )
 from lightning_sdk.cli.utils import CustomHelpFormatter, rich_to_str
 from lightning_sdk.constants import _LIGHTNING_DEBUG
@@ -40,7 +41,7 @@ def _notify_exception(exception_type: Type[BaseException], value: BaseException,
     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))
+        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"))
 
@@ -83,6 +84,7 @@ main_cli.add_command(config)
 # main_cli.add_command(job)
 # main_cli.add_command(mmt)
 main_cli.add_command(studio)
+main_cli.add_command(vm)
 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.

lightning_sdk/cli/groups.py

@@ -6,6 +6,7 @@ from lightning_sdk.cli.config import register_commands as register_config_comman
 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
+from lightning_sdk.cli.vm import register_commands as register_vm_commands
 
 
 @click.group(name="studio")
@@ -25,7 +26,12 @@ def mmt() -> None:
 
 @click.group(name="config")
 def config() -> None:
-    """Manage Lightning SDK and CLIconfiguration."""
+    """Manage Lightning SDK and CLI configuration."""
+
+
+@click.group(name="vm")
+def vm() -> None:
+    """Manage Lightning AI VMs."""
 
 
 # Register config commands with the main config group
@@ -33,3 +39,4 @@ register_job_commands(job)
 register_mmt_commands(mmt)
 register_studio_commands(studio)
 register_config_commands(config)
+register_vm_commands(vm)

lightning_sdk/cli/legacy/entrypoint.py

@@ -48,7 +48,7 @@ def _notify_exception(exception_type: Type[BaseException], value: BaseException,
     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))
+        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"))
 

lightning_sdk/cli/studio/create.py

@@ -9,7 +9,7 @@ from lightning_sdk.cli.utils.save_to_config import save_teamspace_to_config
 from lightning_sdk.cli.utils.teamspace_selection import TeamspacesMenu
 from lightning_sdk.lightning_cloud.openapi.rest import ApiException
 from lightning_sdk.machine import CloudProvider
-from lightning_sdk.studio import Studio
+from lightning_sdk.studio import VM, Studio
 
 
 @click.command("create")
@@ -36,6 +36,16 @@ def create_studio(
     Example:
         lightning studio create
     """
+    create_impl(name=name, teamspace=teamspace, cloud_provider=cloud_provider, cloud_account=cloud_account, vm=False)
+
+
+def create_impl(
+    name: Optional[str],
+    teamspace: Optional[str],
+    cloud_provider: Optional[str],
+    cloud_account: Optional[str],
+    vm: bool,
+) -> None:
     menu = TeamspacesMenu()
 
     resolved_teamspace = menu(teamspace)
@@ -44,8 +54,12 @@ def create_studio(
     if cloud_provider is not None:
         cloud_provider = CloudProvider(cloud_provider)
 
+    create_cls = VM if vm else Studio
+    cls_name = create_cls.__qualname__
+
     try:
-        studio = Studio(
+        create_cls = VM if vm else Studio
+        studio = create_cls(
             name=name,
             teamspace=resolved_teamspace,
             create_ok=True,
@@ -54,7 +68,7 @@ def create_studio(
         )
     except (RuntimeError, ValueError, ApiException):
         if name:
-            raise ValueError(f"Could not create Studio: '{name}'. Does the Studio exist?") from None
-        raise ValueError(f"Could not create Studio: '{name}'. Please provide a Studio name") from None
+            raise ValueError(f"Could not create {cls_name}: '{name}'. Does the {cls_name} exist?") from None
+        raise ValueError(f"Could not create {cls_name}: '{name}'. Please provide a {cls_name} name") from None
 
-    click.echo(f"Studio {studio_name_link(studio)} created successfully")
+    click.echo(f"{cls_name} {studio_name_link(studio)} created successfully")

lightning_sdk/cli/studio/delete.py

@@ -12,7 +12,7 @@ from lightning_sdk.cli.utils.teamspace_selection import TeamspacesMenu
 @click.option(
     "--name",
     help=(
-        "The name of the studio to start. "
+        "The name of the studio to delete. "
         "If not provided, will try to infer from environment, "
         "use the default value from the config or prompt for interactive selection."
     ),
@@ -25,21 +25,25 @@ def delete_studio(name: Optional[str] = None, teamspace: Optional[str] = None) -
       lightning studio delete --name my-studio
 
     """
+    return delete_impl(name=name, teamspace=teamspace, vm=False)
+
+
+def delete_impl(name: Optional[str], teamspace: Optional[str], vm: bool) -> None:
     menu = TeamspacesMenu()
     resolved_teamspace = menu(teamspace=teamspace)
 
-    menu = StudiosMenu(resolved_teamspace)
+    menu = StudiosMenu(resolved_teamspace, vm=vm)
     studio = menu(studio=name)
 
     studio_name = f"{studio.teamspace.owner.name}/{studio.teamspace.name}/{studio.name}"
     confirmed = click.confirm(
-        f"Are you sure you want to delete studio '{studio_name}'?",
+        f"Are you sure you want to delete {studio._cls_name} '{studio_name}'?",
         abort=True,
     )
     if not confirmed:
-        click.echo("Studio deletion cancelled")
+        click.echo(f"{studio._cls_name} deletion cancelled")
         return
 
     studio.delete()
 
-    click.echo(f"Studio '{studio.name}' deleted successfully")
+    click.echo(f"{studio._cls_name} '{studio.name}' deleted successfully")

lightning_sdk/cli/studio/list.py

@@ -35,13 +35,17 @@ def list_studios(teamspace: Optional[str] = None, all: bool = False, sort_by: Op
         lightning studio list --teamspace owner/teamspace
 
     """
+    return list_impl(teamspace=teamspace, all=all, sort_by=sort_by, vm=False)
+
+
+def list_impl(teamspace: Optional[str], all: bool, sort_by: Optional[str], vm: bool) -> None:  # noqa: A002
     menu = TeamspacesMenu()
     teamspace_resolved = menu(teamspace=teamspace)
     save_teamspace_to_config(teamspace_resolved, overwrite=False)
 
     user = _get_authed_user()
 
-    studios = teamspace_resolved.studios
+    studios = teamspace_resolved.vms if vm else teamspace_resolved.studios
 
     table = Table(
         pad_edge=True,

lightning_sdk/cli/studio/ssh.py

@@ -20,7 +20,7 @@ from lightning_sdk.utils.config import _DEFAULT_CONFIG_FILE_PATH
 @click.option(
     "--name",
     help=(
-        "The name of the studio to start. "
+        "The name of the studio to ssh into. "
         "If not provided, will try to infer from environment, "
         "use the default value from the config or prompt for interactive selection."
     ),
@@ -39,6 +39,10 @@ def ssh_studio(name: Optional[str] = None, teamspace: Optional[str] = None, opti
     Example:
         lightning studio ssh --name my-studio
     """
+    return ssh_impl(name=name, teamspace=teamspace, option=option, vm=False)
+
+
+def ssh_impl(name: Optional[str], teamspace: Optional[str], option: Optional[List[str]], vm: bool) -> None:
     auth = Auth()
     auth.authenticate()
     ssh_private_key_path = _download_ssh_keys(auth.api_key, force_download=False)
@@ -46,8 +50,10 @@ def ssh_studio(name: Optional[str] = None, teamspace: Optional[str] = None, opti
     menu = TeamspacesMenu()
     resolved_teamspace = menu(teamspace=teamspace)
 
-    menu = StudiosMenu(resolved_teamspace)
-    studio = menu(studio=name)
+    menu = StudiosMenu(resolved_teamspace, vm=vm)
+    studio = menu(
+        studio=name,
+    )
     save_studio_to_config(studio)
 
     ssh_options = " -o " + " -o ".join(option) if option else ""

lightning_sdk/cli/studio/start.py

@@ -9,7 +9,7 @@ from lightning_sdk.cli.utils.save_to_config import save_studio_to_config
 from lightning_sdk.cli.utils.studio_selection import StudiosMenu
 from lightning_sdk.cli.utils.teamspace_selection import TeamspacesMenu
 from lightning_sdk.machine import CloudProvider, Machine
-from lightning_sdk.studio import Studio
+from lightning_sdk.studio import VM, Studio
 
 
 @click.command("start")
@@ -57,6 +57,28 @@ def start_studio(
         lightning studio start --name my-studio
 
     """
+    return start_impl(
+        name=name,
+        teamspace=teamspace,
+        create=create,
+        machine=machine,
+        interruptible=interruptible,
+        cloud_provider=cloud_provider,
+        cloud_account=cloud_account,
+        vm=False,
+    )
+
+
+def start_impl(
+    name: Optional[str],
+    teamspace: Optional[str],
+    create: bool,
+    machine: str,
+    interruptible: bool,
+    cloud_provider: Optional[str],
+    cloud_account: Optional[str],
+    vm: bool,
+) -> None:
     menu = TeamspacesMenu()
     resolved_teamspace = menu(teamspace=teamspace)
 
@@ -64,10 +86,11 @@ def start_studio(
         cloud_provider = CloudProvider(cloud_provider)
 
     if not create:
-        menu = StudiosMenu(resolved_teamspace)
+        menu = StudiosMenu(resolved_teamspace, vm=vm)
         studio = menu(studio=name)
     else:
-        studio = Studio(
+        create_cls = VM if vm else Studio
+        studio = create_cls(
             name=name,
             teamspace=resolved_teamspace,
             create_ok=create,

lightning_sdk/cli/studio/stop.py

@@ -14,7 +14,7 @@ from lightning_sdk.cli.utils.teamspace_selection import TeamspacesMenu
 @click.option(
     "--name",
     help=(
-        "The name of the studio to start. "
+        "The name of the studio to stop. "
         "If not provided, will try to infer from environment, "
         "use the default value from the config or prompt for interactive selection."
     ),
@@ -27,15 +27,19 @@ def stop_studio(name: Optional[str] = None, teamspace: Optional[str] = None) ->
         lightning studio stop --name my-studio
 
     """
+    return stop_impl(name=name, teamspace=teamspace, vm=False)
+
+
+def stop_impl(name: Optional[str], teamspace: Optional[str], vm: bool) -> None:
     # missing studio_name and teamspace are handled by the studio class
     menu = TeamspacesMenu()
     resolved_teamspace = menu(teamspace=teamspace)
 
-    menu = StudiosMenu(resolved_teamspace)
+    menu = StudiosMenu(resolved_teamspace, vm=vm)
     studio = menu(studio=name)
 
     studio.stop()
 
     save_studio_to_config(studio)
 
-    click.echo(f"Studio {studio_name_link(studio)} stopped successfully")
+    click.echo(f"{studio._cls_name} {studio_name_link(studio)} stopped successfully")

lightning_sdk/cli/studio/switch.py

@@ -9,14 +9,13 @@ from lightning_sdk.cli.utils.save_to_config import save_studio_to_config
 from lightning_sdk.cli.utils.studio_selection import StudiosMenu
 from lightning_sdk.cli.utils.teamspace_selection import TeamspacesMenu
 from lightning_sdk.machine import Machine
-from lightning_sdk.studio import Studio
 
 
 @click.command("switch")
 @click.option(
     "--name",
     help=(
-        "The name of the studio to start. "
+        "The name of the studio to switch to a different machine. "
         "If not provided, will try to infer from environment, "
         "use the default value from the config or prompt for interactive selection."
     ),
@@ -35,16 +34,33 @@ def switch_studio(
     interruptible: bool = False,
 ) -> None:
     """Switch a Studio to a different machine type."""
+    return switch_impl(
+        name=name,
+        teamspace=teamspace,
+        machine=machine,
+        interruptible=interruptible,
+        vm=False,
+    )
+
+
+def switch_impl(
+    name: Optional[str],
+    teamspace: Optional[str],
+    machine: Optional[str],
+    interruptible: bool,
+    vm: bool,
+) -> None:
     menu = TeamspacesMenu()
     resolved_teamspace = menu(teamspace=teamspace)
 
-    menu = StudiosMenu(resolved_teamspace)
+    menu = StudiosMenu(resolved_teamspace, vm=vm)
     studio = menu(studio=name)
 
     resolved_machine = Machine.from_str(machine)
-    Studio.show_progress = True
+
+    studio.__class__.show_progress = True
     studio.switch_machine(resolved_machine, interruptible=interruptible)
 
     save_studio_to_config(studio)
 
-    click.echo(f"Studio {studio_name_link(studio)} switched to machine '{resolved_machine}' successfully")
+    click.echo(f"{studio._cls_name} {studio_name_link(studio)} switched to machine '{resolved_machine}' successfully")

lightning_sdk/cli/utils/owner_selection.py

@@ -0,0 +1,110 @@
+import os
+from contextlib import suppress
+from typing import Dict, List, Optional, TypedDict
+
+import click
+from simple_term_menu import TerminalMenu
+
+from lightning_sdk.cli.legacy.exceptions import StudioCliError
+from lightning_sdk.organization import Organization
+from lightning_sdk.owner import Owner
+from lightning_sdk.user import User
+from lightning_sdk.utils.resolve import ApiException, _get_authed_user, _resolve_org, _resolve_user
+
+
+class _OwnerMenuType(TypedDict):
+    name: str
+    is_org: bool
+
+
+class OwnerMenu:
+    """This class is used to select a teamspace owner (org/user) from a list of possible owners.
+
+    It can be used to select an owner from a list of possible owners, or to resolve an owner from a name.
+    """
+
+    def _get_owner_from_interactive_menu(self, possible_owners: Dict[str, _OwnerMenuType]) -> _OwnerMenuType:
+        owner_ids = sorted(possible_owners.keys())
+        terminal_menu = self._prepare_terminal_menu_owners([possible_owners[k] for k in owner_ids])
+        terminal_menu.show()
+
+        selected_id = owner_ids[terminal_menu.chosen_menu_index]
+        return possible_owners[selected_id]
+
+    def _get_owner_from_name(self, owner: str, possible_owners: Dict[str, _OwnerMenuType]) -> _OwnerMenuType:
+        for _, ts in possible_owners.items():
+            if ts["name"]:
+                return ts
+
+        click.echo(f"Could not find Owner {owner}, please select it from the list:")
+        return self._get_owner_from_interactive_menu(possible_owners)
+
+    @staticmethod
+    def _prepare_terminal_menu_owners(
+        possible_owners: List[_OwnerMenuType], title: Optional[str] = None
+    ) -> TerminalMenu:
+        if title is None:
+            title = "Please select a Teamspace-Owner out of the following:"
+
+        return TerminalMenu(
+            [f"{to['name']} ({'Organization' if to['is_org'] else 'User'})" for to in possible_owners],
+            title=title,
+            clear_menu_on_exit=True,
+        )
+
+    @staticmethod
+    def _get_possible_owners(user: User) -> Dict[str, _OwnerMenuType]:
+        user_api = user._user_api
+
+        orgs = user_api._get_organizations_for_authed_user()
+        owners: Dict[str, _OwnerMenuType] = {user.id: {"name": user.name, "is_org": False}}
+
+        for org in orgs:
+            owners[org.id] = {"name": org.name, "is_org": True}
+
+        return owners
+
+    def __call__(self, owner: Optional[str] = None) -> Owner:
+        try:
+            # try to resolve the teamspace from the name, environment or config
+            resolved_owner = None
+            with suppress(ApiException, ValueError, RuntimeError):
+                resolved_owner = _resolve_org(owner)
+
+            if resolved_owner is not None:
+                return resolved_owner
+
+            with suppress(ApiException, ValueError, RuntimeError):
+                resolved_owner = _resolve_user(owner)
+
+            if resolved_owner is not None:
+                return resolved_owner
+
+            if os.environ.get("LIGHTNING_NON_INTERACTIVE", "0") == "1":
+                raise ValueError(
+                    "Owner selection is not supported in non-interactive mode. Please provide an owner name."
+                )
+
+            # if the owner is not resolved, try to get the owner from the interactive menu
+            # this could mean that either no owner was provided or the provided owner is not valid
+            user = _get_authed_user()
+
+            possible_owners = self._get_possible_owners(user)
+            if owner is None:
+                owner_dict = self._get_owner_from_interactive_menu(possible_owners=possible_owners)
+
+            else:
+                owner_dict = self._get_owner_from_name(owner=owner, possible_owners=possible_owners)
+
+            if owner_dict.get("is_org", False):
+                return Organization(owner_dict.get("name", None))
+
+            return User(owner_dict.get("name", None))
+        except KeyboardInterrupt:
+            raise KeyboardInterrupt from None
+
+        except Exception as e:
+            raise StudioCliError(
+                f"Could not find the given Teamspace-Owner {owner}. "
+                "Please contact Lightning AI directly to resolve this issue."
+            ) from e