lightning-sdk 2025.8.18__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.18"
+__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/config/get.py

@@ -39,3 +39,19 @@ def get_studio() -> None:
     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

@@ -1,6 +1,7 @@
 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
@@ -75,3 +76,29 @@ def set_teamspace(teamspace_name: str) -> None:
         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/studio/create.py

@@ -18,10 +18,16 @@ from lightning_sdk.studio import Studio
     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,
+)
 def create_studio(
     studio_name: Optional[str] = None,
     teamspace: Optional[str] = None,
     cloud_provider: Optional[str] = None,
+    cloud_account: Optional[str] = None,
 ) -> None:
     """Create a new Studio.
 
@@ -42,8 +48,17 @@ def create_studio(
     else:
         resolved_teamspace = None
 
+    if cloud_provider is not None:
+        cloud_provider = CloudProvider(cloud_provider)
+
     try:
-        studio = Studio(studio_name, teamspace=resolved_teamspace, create_ok=True, cloud_provider=cloud_provider)
+        studio = Studio(
+            studio_name,
+            teamspace=resolved_teamspace,
+            create_ok=True,
+            cloud_provider=cloud_provider,
+            cloud_account=cloud_account,
+        )
     except (RuntimeError, ValueError, ApiException) as e:
         print(e)
         if studio_name:

lightning_sdk/cli/studio/start.py

@@ -22,9 +22,17 @@ from lightning_sdk.studio import Studio
 @click.option("--interruptible", is_flag=True, help="Start the studio on an interruptible instance.")
 @click.option(
     "--cloud-provider",
-    help="The cloud provider to start the studio on. Defaults to teamspace default.",
+    help=(
+        "The cloud provider to start the studio on. Defaults to teamspace default. "
+        "Only used if --create is specified."
+    ),
     type=click.Choice(m.name for m in list(CloudProvider)),
 )
+@click.option(
+    "--cloud-account",
+    help="The cloud account to start the studio on. Defaults to teamspace default. Only used if --create is specified.",
+    type=click.STRING,
+)
 def start_studio(
     studio_name: Optional[str] = None,
     teamspace: Optional[str] = None,
@@ -32,6 +40,7 @@ def start_studio(
     machine: Optional[str] = None,
     interruptible: bool = False,
     cloud_provider: Optional[str] = None,
+    cloud_account: Optional[str] = None,
 ) -> None:
     """Start a Studio.
 
@@ -52,8 +61,17 @@ def start_studio(
     else:
         resolved_teamspace = None
 
+    if cloud_provider is not None:
+        cloud_provider = CloudProvider(cloud_provider)
+
     try:
-        studio = Studio(studio_name, teamspace=resolved_teamspace, create_ok=create, cloud_provider=cloud_provider)
+        studio = Studio(
+            studio_name,
+            teamspace=resolved_teamspace,
+            create_ok=create,
+            cloud_provider=cloud_provider,
+            cloud_account=cloud_account,
+        )
     except (RuntimeError, ValueError, ApiException):
         if studio_name:
             raise ValueError(f"Could not start Studio: '{studio_name}'. Does the Studio exist?") from None

lightning_sdk/studio.py

@@ -81,7 +81,6 @@ class Studio:
             raise ValueError("Couldn't resolve teamspace from the provided name, org, or user")
 
         self._teamspace = _teamspace
-        self._cloud_account = _resolve_deprecated_cluster(cloud_account, cluster)
 
         self._setup_done = self._skip_setup
         self._disable_secrets = disable_secrets

lightning_sdk/utils/config.py

@@ -23,6 +23,9 @@ class DefaultConfigKeys:
 
     studio: str = "studio.name"
 
+    cloud_account: str = "cloud_account.name"
+    cloud_provider: str = "cloud_provider.name"
+
 
 class ConfigProxy:
     def __init__(self, root: "Config", *path: str) -> None:

lightning_sdk/utils/resolve.py

@@ -66,6 +66,12 @@ def _resolve_deprecated_provider(
         )
         return provider
 
+    if cloud_provider is None:
+        from lightning_sdk.utils.config import Config, DefaultConfigKeys
+
+        config = Config()
+        cloud_provider = config.get_value(DefaultConfigKeys.cloud_provider)
+
     return cloud_provider
 
 
@@ -84,6 +90,12 @@ def _resolve_deprecated_cluster(cloud_account: Optional[str], cluster: Optional[
         )
         return cluster
 
+    if cloud_account is None:
+        from lightning_sdk.utils.config import Config, DefaultConfigKeys
+
+        config = Config()
+        cloud_account = config.get_value(DefaultConfigKeys.cloud_account)
+
     return cloud_account
 
 

{lightning_sdk-2025.8.18.dist-info → lightning_sdk-2025.8.18.post0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: lightning_sdk
-Version: 2025.8.18
+Version: 2025.8.18.post0
 Summary: SDK to develop using Lightning AI Studios
 Author-email: Lightning-AI <justus@lightning.ai>
 License: MIT License

{lightning_sdk-2025.8.18.dist-info → lightning_sdk-2025.8.18.post0.dist-info}/RECORD

@@ -1,5 +1,5 @@
 docs/source/conf.py,sha256=r8yX20eC-4mHhMTd0SbQb5TlSWHhO6wnJ0VJ_FBFpag,13249
-lightning_sdk/__init__.py,sha256=Ha5K1ZUBTvAFCsOrwLRJYb2kwwLj-jNYvRgFlxsn4w4,1145
+lightning_sdk/__init__.py,sha256=xjqhb1xsoDuv_SHJPA5eoPZ2K78SgqGKcJJYrF7BuMw,1151
 lightning_sdk/agents.py,sha256=ly6Ma1j0ZgGPFyvPvMN28JWiB9dATIstFa5XM8pMi6I,1577
 lightning_sdk/ai_hub.py,sha256=iI1vNhgcz_Ff1c3rN1ogN7dK-r-HXRj6NMtS2cA14UA,6925
 lightning_sdk/base_studio.py,sha256=_Pwwl37R9GRd7t-f2kO5aQXiLNrP4sUtUNht2ZkP8LE,3678
@@ -14,14 +14,14 @@ lightning_sdk/plugin.py,sha256=f3P5-xZY6x-MX0Fs2z_Q2erSxPSiHZARO0BVkCezHw4,15192
 lightning_sdk/sandbox.py,sha256=_NvnWotEXW2rBiVFZZ4krKXxVjuAqfNh04qELSM0-Pg,5786
 lightning_sdk/serve.py,sha256=uW7zLhQ3X90ifetpxzTb8FNxifv5vIs7qZlgfEjVKzk,11794
 lightning_sdk/status.py,sha256=lLGAuSvXBoXQFEEsEYwdCi0RcSNatUn5OPjJVjDtoM0,386
-lightning_sdk/studio.py,sha256=jqtQ0EVq5T8ShRBFEyEgQQL1sK_r4Q0SvFy39XqDpnY,27220
+lightning_sdk/studio.py,sha256=l3l-TYExmnJnAQIPA6_8R2u1ynT-ewzW-q8hV_nY7nE,27138
 lightning_sdk/teamspace.py,sha256=Ikg3LwGhskD_wAW3GfUOuMgiyEXOV8khleBeIMtN8wA,20751
 lightning_sdk/user.py,sha256=vdn8pZqkAZO0-LoRsBdg0TckRKtd_H3QF4gpiZcl4iY,1130
 lightning_sdk/api/__init__.py,sha256=Qn2VVRvir_gO7w4yxGLkZY-R3T7kdiTPKgQ57BhIA9k,413
 lightning_sdk/api/agents_api.py,sha256=G47TbFo9kYqnBMqdw2RW-lfS1VAUBSXDmzs6fpIEMUs,4059
 lightning_sdk/api/ai_hub_api.py,sha256=azqDZ-PzasVAcoQHno7k7OO_xFOHQ4NDozxF8jEh83Y,7864
 lightning_sdk/api/base_studio_api.py,sha256=3R8tucZX2e9yKHBcY2rWFRP4dxqLrC6H75vdBDkH0ck,3617
-lightning_sdk/api/cloud_account_api.py,sha256=Rxet7QuXjzkeAok-_azvB0r_koO7K-NioAldmFVsibs,8223
+lightning_sdk/api/cloud_account_api.py,sha256=b_Egg5yhDh5G39g1cqA3ZxBDmU9iuDs2PlBMmW0HVnA,8417
 lightning_sdk/api/deployment_api.py,sha256=v2AfoTDkQ-1CBh75FOjFkRpf6yc3U_edDy43uYSn19I,24852
 lightning_sdk/api/job_api.py,sha256=7spFPyotlSQ0Krx6WymslnpSQaaeOmFkydvhtg6p0Ic,16410
 lightning_sdk/api/license_api.py,sha256=XV3RhefyPQDYjwY9AaBZe4rByZTEAnsvLDxcdm9q0Wo,2438
@@ -38,8 +38,8 @@ lightning_sdk/cli/__init__.py,sha256=lksw08t_ZIuHOH47LCIqSVHeZ8cUXI2aJWHYhyujYHM
 lightning_sdk/cli/entrypoint.py,sha256=85icG0fmUqXuFErkD2v_kOlqnr18veD8oxe7gPBWHE8,4556
 lightning_sdk/cli/groups.py,sha256=sgKG2z4xJG7lJ1LphSez-cSnntJReSJxJgBp3Z8htUc,951
 lightning_sdk/cli/config/__init__.py,sha256=_aJ7uZZOdjomZ-ABmz8Cu3fLmnSujnQK4hKAEKM4aF4,395
-lightning_sdk/cli/config/get.py,sha256=ui85ubvxe6LMl4-FRV4IXPKZNPoZRjJBqzmgKNNdJ6E,1102
-lightning_sdk/cli/config/set.py,sha256=m6X70fKkZdxqSSWXxVFrNO3kRKUxLD6iIjm5Ls9AS2s,2681
+lightning_sdk/cli/config/get.py,sha256=JXGRNVGxELS6Es2RESgStzTKnjg5rr-M6yOmx2ty9A8,1606
+lightning_sdk/cli/config/set.py,sha256=B2EXzdpXN8knVLLV-nzSg6I7i3hP1IdYlMssxTr7eMk,3657
 lightning_sdk/cli/config/show.py,sha256=rcarLUSAmbqyEBMFQ9PPHquURFbUCaxoMeBTG1DOpGs,167
 lightning_sdk/cli/job/__init__.py,sha256=HxkswumuGqBnrSHjBElO2CXn9kvH_zVfQEQqd0nwkNU,145
 lightning_sdk/cli/legacy/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -73,11 +73,11 @@ lightning_sdk/cli/legacy/deploy/devbox.py,sha256=SdOtYdGefzldn-WCa6sz31zysyf2nP5
 lightning_sdk/cli/legacy/deploy/serve.py,sha256=bxN04mHlArjHtxR_QkCM0S_RPmL3WIPsgnPgFadSXwI,15478
 lightning_sdk/cli/mmt/__init__.py,sha256=CLgr-ZHHLS6Db_JGqpxbn4G2pYrKi4Qn-uhi8e0kFNc,145
 lightning_sdk/cli/studio/__init__.py,sha256=Reb5a0iHJfNMxmnP30DAmYqpJ-YDQoJcgyVrD2cYDNo,823
-lightning_sdk/cli/studio/create.py,sha256=JQroKr9wfWutrN9zlzIuTVWhAFNkoZuDZnPg7GTZ3FE,1962
+lightning_sdk/cli/studio/create.py,sha256=o1NkEn5H-SNdpE4yC7KyikTkQM14AVGYMRZlT3iEAfA,2343
 lightning_sdk/cli/studio/delete.py,sha256=33IrIsQC4IwsZYRrys4ZxqNbVC2CwWvy6D6ilYXrLuU,1670
 lightning_sdk/cli/studio/list.py,sha256=LxY5R1aQ4OJbo4TWepK7ugrJIju9iq3Z7m2oq3U9Zcc,2387
 lightning_sdk/cli/studio/ssh.py,sha256=0p1DqKPBkrPRNN2AFlhRK9hnuEX3-33ujiBKh24jB_c,3812
-lightning_sdk/cli/studio/start.py,sha256=2qg063z1zw6VenwALjC-HeLVauVJiHpnQvNoPlMu5sk,2496
+lightning_sdk/cli/studio/start.py,sha256=h3AQE5RCgLIcT4XoPJI6-kNVr3J1D6KjY2cUAAPjMdk,2975
 lightning_sdk/cli/studio/stop.py,sha256=pYkS1G-XYLWTxTFEIgEg7xcihzviB_gHNqgBLaW3OwQ,1635
 lightning_sdk/cli/studio/switch.py,sha256=uKuAv4J7qozS6YzPqz-45_ccfVWNkyWF5Pij_xDcG0M,1994
 lightning_sdk/cli/utils/__init__.py,sha256=0gHdWY3bqrIyiFiEh_uSBuxWpOykCjqUc8fPEV0z3no,186
@@ -1131,13 +1131,13 @@ lightning_sdk/services/finetune_llm.py,sha256=5ewjo_bsk5hwvhGoLRdHBv1nP6LeoXDm5_
 lightning_sdk/services/license.py,sha256=Nav9I6nLIwV-iRZRhZR0iSpyzWPyK-AumIl32hX8nmE,17065
 lightning_sdk/services/utilities.py,sha256=WlzKDGqUrQFcT85lnLgHvexf7AB1ppxS4_RBkS9Dbg8,4686
 lightning_sdk/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-lightning_sdk/utils/config.py,sha256=UTvxdd_OiLwIl5q1_9X1SxFPcnsp39m_ONWmixoTRho,5274
+lightning_sdk/utils/config.py,sha256=DkH5e4ljBLjOVWwEBSHsF6EMBf-akSwFjHAZKHpqVbE,5369
 lightning_sdk/utils/dynamic.py,sha256=glUTO1JC9APtQ6Gr9SO02a3zr56-sPAXM5C3NrTpgyQ,1959
 lightning_sdk/utils/enum.py,sha256=h2JRzqoBcSlUdanFHmkj_j5DleBHAu1esQYUsdNI-hU,4106
-lightning_sdk/utils/resolve.py,sha256=LJQwU5L2-SsPUcxtLBo8cWLuyrrE5CpdI1NL372j5SE,9647
-lightning_sdk-2025.8.18.dist-info/LICENSE,sha256=uFIuZwj5z-4TeF2UuacPZ1o17HkvKObT8fY50qN84sg,1064
-lightning_sdk-2025.8.18.dist-info/METADATA,sha256=cV0buIXnH1OUeWalmE-ED3B7yn46HDHm5gVMbFzZGow,4130
-lightning_sdk-2025.8.18.dist-info/WHEEL,sha256=iAkIy5fosb7FzIOwONchHf19Qu7_1wCWyFNR5gu9nU0,91
-lightning_sdk-2025.8.18.dist-info/entry_points.txt,sha256=msB9PJWIJ784dX-OP8by51d4IbKYH3Fj1vCuA9oXjHY,68
-lightning_sdk-2025.8.18.dist-info/top_level.txt,sha256=ps8doKILFXmN7F1mHncShmnQoTxKBRPIcchC8TpoBw4,19
-lightning_sdk-2025.8.18.dist-info/RECORD,,
+lightning_sdk/utils/resolve.py,sha256=r7sh5FvVAqb8Ax_TXED0JR3QJawom6w1wgOjAQFJ-MA,10060
+lightning_sdk-2025.8.18.post0.dist-info/LICENSE,sha256=uFIuZwj5z-4TeF2UuacPZ1o17HkvKObT8fY50qN84sg,1064
+lightning_sdk-2025.8.18.post0.dist-info/METADATA,sha256=S52xMvuWPtO5fS2BIflCw3x9nfkpvqPs28M6wp-GMW0,4136
+lightning_sdk-2025.8.18.post0.dist-info/WHEEL,sha256=iAkIy5fosb7FzIOwONchHf19Qu7_1wCWyFNR5gu9nU0,91
+lightning_sdk-2025.8.18.post0.dist-info/entry_points.txt,sha256=msB9PJWIJ784dX-OP8by51d4IbKYH3Fj1vCuA9oXjHY,68
+lightning_sdk-2025.8.18.post0.dist-info/top_level.txt,sha256=ps8doKILFXmN7F1mHncShmnQoTxKBRPIcchC8TpoBw4,19
+lightning_sdk-2025.8.18.post0.dist-info/RECORD,,