lightning-sdk 2025.10.14__py3-none-any.whl → 2025.10.23__py3-none-any.whl

lightning_sdk/lightning_cloud/openapi/models/{v1_product_license_check_response.py → v1_validate_license_response.py}

@@ -28,7 +28,7 @@ if TYPE_CHECKING:
     from datetime import datetime
     from lightning_sdk.lightning_cloud.openapi.models import *
 
-class V1ProductLicenseCheckResponse(object):
+class V1ValidateLicenseResponse(object):
     """NOTE: This class is auto generated by the swagger code generator program.
 
     Do not edit the class manually.
@@ -41,40 +41,40 @@ class V1ProductLicenseCheckResponse(object):
                             and the value is json key in definition.
     """
     swagger_types = {
-        'valid': 'bool'
+        'is_valid': 'bool'
     }
 
     attribute_map = {
-        'valid': 'valid'
+        'is_valid': 'isValid'
     }
 
-    def __init__(self, valid: 'bool' =None):  # noqa: E501
-        """V1ProductLicenseCheckResponse - a model defined in Swagger"""  # noqa: E501
-        self._valid = None
+    def __init__(self, is_valid: 'bool' =None):  # noqa: E501
+        """V1ValidateLicenseResponse - a model defined in Swagger"""  # noqa: E501
+        self._is_valid = None
         self.discriminator = None
-        if valid is not None:
-            self.valid = valid
+        if is_valid is not None:
+            self.is_valid = is_valid
 
     @property
-    def valid(self) -> 'bool':
-        """Gets the valid of this V1ProductLicenseCheckResponse.  # noqa: E501
+    def is_valid(self) -> 'bool':
+        """Gets the is_valid of this V1ValidateLicenseResponse.  # noqa: E501
 
 
-        :return: The valid of this V1ProductLicenseCheckResponse.  # noqa: E501
+        :return: The is_valid of this V1ValidateLicenseResponse.  # noqa: E501
         :rtype: bool
         """
-        return self._valid
+        return self._is_valid
 
-    @valid.setter
-    def valid(self, valid: 'bool'):
-        """Sets the valid of this V1ProductLicenseCheckResponse.
+    @is_valid.setter
+    def is_valid(self, is_valid: 'bool'):
+        """Sets the is_valid of this V1ValidateLicenseResponse.
 
 
-        :param valid: The valid of this V1ProductLicenseCheckResponse.  # noqa: E501
+        :param is_valid: The is_valid of this V1ValidateLicenseResponse.  # noqa: E501
         :type: bool
         """
 
-        self._valid = valid
+        self._is_valid = is_valid
 
     def to_dict(self) -> dict:
         """Returns the model properties as a dict"""
@@ -97,7 +97,7 @@ class V1ProductLicenseCheckResponse(object):
                 ))
             else:
                 result[attr] = value
-        if issubclass(V1ProductLicenseCheckResponse, dict):
+        if issubclass(V1ValidateLicenseResponse, dict):
             for key, value in self.items():
                 result[key] = value
 
@@ -111,13 +111,13 @@ class V1ProductLicenseCheckResponse(object):
         """For `print` and `pprint`"""
         return self.to_str()
 
-    def __eq__(self, other: 'V1ProductLicenseCheckResponse') -> bool:
+    def __eq__(self, other: 'V1ValidateLicenseResponse') -> bool:
         """Returns true if both objects are equal"""
-        if not isinstance(other, V1ProductLicenseCheckResponse):
+        if not isinstance(other, V1ValidateLicenseResponse):
             return False
 
         return self.__dict__ == other.__dict__
 
-    def __ne__(self, other: 'V1ProductLicenseCheckResponse') -> bool:
+    def __ne__(self, other: 'V1ValidateLicenseResponse') -> bool:
         """Returns true if both objects are not equal"""
         return not self == other

lightning_sdk/lightning_cloud/rest_client.py

@@ -2,41 +2,43 @@ import functools
 import logging
 import time
 from functools import wraps
-from typing import Callable, Optional, Any
+from typing import Any, Callable, Optional
 
 import urllib3
+
 from lightning_sdk.lightning_cloud import env
 from lightning_sdk.lightning_cloud.login import Auth
 from lightning_sdk.lightning_cloud.openapi import (
     ApiClient,
+    AssistantsServiceApi,
     AuthServiceApi,
+    BillingServiceApi,
+    CloudSpaceEnvironmentTemplateServiceApi,
     CloudSpaceServiceApi,
     ClusterServiceApi,
     Configuration,
     DataConnectionServiceApi,
+    DatasetServiceApi,
+    DeploymentTemplatesServiceApi,
+    EndpointServiceApi,
+    JobsServiceApi,
     LightningappInstanceServiceApi,
     LightningappV2ServiceApi,
     LightningworkServiceApi,
-    ProjectsServiceApi,
-    SecretServiceApi,
-    SSHPublicKeyServiceApi,
-    DatasetServiceApi,
-    OrganizationsServiceApi,
-    UserServiceApi,
-    BillingServiceApi,
-    EndpointServiceApi,
-    SlurmJobsUserServiceApi,
     LitLoggerServiceApi,
-    JobsServiceApi,
-    AssistantsServiceApi,
-    StorageServiceApi,
-    DeploymentTemplatesServiceApi,
-    ModelsStoreApi,
     LitRegistryServiceApi,
+    ModelsStoreApi,
+    OrganizationsServiceApi,
     PipelinesServiceApi,
-    SchedulesServiceApi,
     ProductLicenseServiceApi,
-    CloudSpaceEnvironmentTemplateServiceApi
+    ProjectsServiceApi,
+    SchedulesServiceApi,
+    SDKCommandHistoryServiceApi,
+    SecretServiceApi,
+    SlurmJobsUserServiceApi,
+    SSHPublicKeyServiceApi,
+    StorageServiceApi,
+    UserServiceApi,
 )
 from lightning_sdk.lightning_cloud.openapi.rest import ApiException
 from lightning_sdk.lightning_cloud.source_code.logs_socket_api import LightningLogsSocketAPI
@@ -76,34 +78,35 @@ def create_swagger_client(check_context: bool = True, with_auth: bool = True):
 
 
 class GridRestClient(
-        LightningLogsSocketAPI,
-        LightningappInstanceServiceApi,
-        LightningappV2ServiceApi,
-        AuthServiceApi,
-        CloudSpaceServiceApi,
-        ClusterServiceApi,
-        ProjectsServiceApi,
-        LightningworkServiceApi,
-        SecretServiceApi,
-        SSHPublicKeyServiceApi,
-        DataConnectionServiceApi,
-        DatasetServiceApi,
-        OrganizationsServiceApi,
-        UserServiceApi,
-        BillingServiceApi,
-        EndpointServiceApi,
-        SlurmJobsUserServiceApi,
-        LitLoggerServiceApi,
-        JobsServiceApi,
-        AssistantsServiceApi,
-        StorageServiceApi,
-        DeploymentTemplatesServiceApi,
-        ModelsStoreApi,
-        LitRegistryServiceApi,
-        PipelinesServiceApi,
-        SchedulesServiceApi,
-        ProductLicenseServiceApi,
-        CloudSpaceEnvironmentTemplateServiceApi
+    LightningLogsSocketAPI,
+    LightningappInstanceServiceApi,
+    LightningappV2ServiceApi,
+    AuthServiceApi,
+    CloudSpaceServiceApi,
+    ClusterServiceApi,
+    ProjectsServiceApi,
+    LightningworkServiceApi,
+    SecretServiceApi,
+    SSHPublicKeyServiceApi,
+    DataConnectionServiceApi,
+    DatasetServiceApi,
+    OrganizationsServiceApi,
+    UserServiceApi,
+    BillingServiceApi,
+    EndpointServiceApi,
+    SlurmJobsUserServiceApi,
+    LitLoggerServiceApi,
+    JobsServiceApi,
+    AssistantsServiceApi,
+    StorageServiceApi,
+    DeploymentTemplatesServiceApi,
+    ModelsStoreApi,
+    LitRegistryServiceApi,
+    PipelinesServiceApi,
+    SchedulesServiceApi,
+    ProductLicenseServiceApi,
+    CloudSpaceEnvironmentTemplateServiceApi,
+    SDKCommandHistoryServiceApi,
 ):
 
     def __init__(self, api_client: Optional[ApiClient] = None, with_auth: bool = True):

lightning_sdk/machine.py

@@ -34,6 +34,7 @@ class Machine:
 
     # supported GPU types
     # supported T4 variations
+    T4_SMALL: ClassVar["Machine"]
     T4: ClassVar["Machine"]
     T4_X_2: ClassVar["Machine"]
     T4_X_4: ClassVar["Machine"]
@@ -162,6 +163,7 @@ Machine.DATA_PREP_ULTRA = Machine(
 
 # GPU machines
 # available T4 machines
+Machine.T4_SMALL = Machine(name="T4_SMALL", slug="lit-t4-1-small", family="T4", accelerator_count=1)
 Machine.T4 = Machine(name="T4", slug="lit-t4-1", family="T4", accelerator_count=1)
 Machine.T4_X_2 = Machine(name="T4_X_2", slug="lit-t4-2", family="T4", accelerator_count=2)
 Machine.T4_X_4 = Machine(name="T4_X_4", slug="lit-t4-4", family="T4", accelerator_count=4)
@@ -216,3 +218,6 @@ Machine.H200 = Machine(name="H200", slug="lit-h200x-1", family="H200", accelerat
 Machine.H200_X_8 = Machine(name="H200_X_8", slug="lit-h200x-8", family="H200", accelerator_count=8)
 # available B200 machines
 Machine.B200_X_8 = Machine(name="B200_X_8", slug="lit-b200x-8", family="B200", accelerator_count=8)
+
+
+DEFAULT_MACHINE = Machine.CPU.name

lightning_sdk/pipeline/steps.py

@@ -239,6 +239,7 @@ class JobStep:
             artifacts_remote=None,
             max_runtime=self.max_runtime,
             machine_image_version=machine_image_version,
+            reuse_snapshot=True,
         )
 
         return V1PipelineStep(

lightning_sdk/studio.py

@@ -11,7 +11,7 @@ from lightning_sdk.api.studio_api import StudioApi
 from lightning_sdk.base_studio import BaseStudio
 from lightning_sdk.constants import _LIGHTNING_DEBUG
 from lightning_sdk.lightning_cloud.openapi import V1ClusterType
-from lightning_sdk.machine import CloudProvider, Machine
+from lightning_sdk.machine import DEFAULT_MACHINE, CloudProvider, Machine
 from lightning_sdk.organization import Organization
 from lightning_sdk.owner import Owner
 from lightning_sdk.status import Status
@@ -103,7 +103,16 @@ class Studio:
         self._plugins = {}
         self._studio = None
 
-        cloud_account = _resolve_deprecated_cluster(cloud_account, cluster)
+        # Check to see if we're inside a studio
+        current_studio = None
+        studio_id = os.environ.get("LIGHTNING_CLOUD_SPACE_ID", None)
+        if studio_id is not None:
+            # We're inside a studio, get it by ID
+            current_studio = self._studio_api.get_studio_by_id(studio_id=studio_id, teamspace_id=self._teamspace.id)
+
+        cloud_account = _resolve_deprecated_cluster(
+            cloud_account, cluster, current_studio.cluster_id if current_studio else None
+        )
         cloud_provider = _resolve_deprecated_provider(cloud_provider, provider)
 
         cls_name = self._cls_name
@@ -119,7 +128,7 @@ class Studio:
 
         self._studio_type = None
         if studio_type:
-            self._base_studio = BaseStudio()
+            self._base_studio = BaseStudio(teamspace=self._teamspace)
             self._available_base_studios = self._base_studio.list()
             for bst in self._available_base_studios:
                 if (
@@ -135,14 +144,14 @@ class Studio:
                     f"Available studio types: "
                     f"{[bst.name.lower().replace(' ', '-') for bst in self._available_base_studios]}"
                 )
+        else:
+            if current_studio:
+                self._studio_type = current_studio.environment_template_id
 
         # Resolve studio name if not provided: explicit → env (LIGHTNING_CLOUD_SPACE_ID) → config defaults
         if name is None and not getattr(self._skip_init, "value", False):
-            studio_id = os.environ.get("LIGHTNING_CLOUD_SPACE_ID", None)
-            if studio_id is not None:
-                # We're inside a studio, get it by ID
-                self._studio = self._studio_api.get_studio_by_id(studio_id=studio_id, teamspace_id=self._teamspace.id)
-                name = self._studio.name
+            if current_studio:
+                name = current_studio.name
             else:
                 # Try config defaults
                 from lightning_sdk.utils.config import Config, DefaultConfigKeys
@@ -272,7 +281,7 @@ class Studio:
 
     def start(
         self,
-        machine: Union[Machine, str] = Machine.CPU,
+        machine: Optional[Union[Machine, str]] = None,
         interruptible: Optional[bool] = None,
         max_runtime: Optional[int] = None,
     ) -> None:
@@ -287,6 +296,22 @@ class Studio:
                 Defaults to 3h
 
         """
+        # Check to see if we're inside a studio and if its running
+        current_studio_machine = None
+        studio_id = os.environ.get("LIGHTNING_CLOUD_SPACE_ID", None)
+        if studio_id is not None:
+            # We're inside a studio, get the machine if it is running
+            current_studio = self._studio_api.get_studio_by_id(studio_id=studio_id, teamspace_id=self._teamspace.id)
+            current_status = self._studio_api._get_studio_instance_status_from_object(current_studio)
+
+            if current_status and _internal_status_to_external_status(current_status) == Status.Running:
+                current_studio_machine = self._studio_api.get_machine(
+                    current_studio.id,
+                    self._teamspace.id,
+                    current_studio.cluster_id,
+                    _get_org_id(self._teamspace),
+                )
+
         status = self.status
 
         if interruptible is None:
@@ -296,9 +321,14 @@ class Studio:
             else:
                 interruptible = self.teamspace.start_studios_on_interruptible
 
-        new_machine = machine
-        if not isinstance(machine, Machine):
-            new_machine = Machine.from_str(machine)
+        new_machine = DEFAULT_MACHINE
+        if machine is not None:
+            new_machine = machine
+        elif current_studio_machine is not None:
+            new_machine = current_studio_machine
+
+        if not isinstance(new_machine, Machine):
+            new_machine = Machine.from_str(new_machine)
 
         if status == Status.Running:
             if new_machine != self.machine:
@@ -353,7 +383,10 @@ class Studio:
         self._studio_api.delete_studio(self._studio.id, self._teamspace.id)
 
     def duplicate(
-        self, target_teamspace: Optional[Union["Teamspace", str]] = None, machine: Machine = Machine.CPU
+        self,
+        target_teamspace: Optional[Union["Teamspace", str]] = None,
+        machine: Machine = Machine.CPU,
+        name: Optional[str] = None,
     ) -> "Studio":
         """Duplicates the existing Studio.
 
@@ -386,6 +419,7 @@ class Studio:
             teamspace_id=self._teamspace.id,
             target_teamspace_id=target_teamspace_id,
             machine=machine,
+            new_name=name,
         )
         return Studio(**kwargs)
 
@@ -659,6 +693,14 @@ class Studio:
         self._assistant_id = assistant.id
         _logger.info(assistant_info)
 
+    def rename(self, new_name: str) -> None:
+        """Renames the current Studio to the provided new name."""
+        if new_name == self._studio.name:
+            return
+
+        self._studio_api._update_cloudspace(self._studio, self._teamspace.id, "display_name", new_name)
+        self._update_studio_reference()
+
     @property
     def auto_sleep(self) -> bool:
         """Returns if a Studio has auto-sleep enabled."""

lightning_sdk/utils/config.py

@@ -1,7 +1,6 @@
 import os
-from collections.abc import Mapping
 from dataclasses import dataclass
-from typing import Any, Dict, Optional, Sequence
+from typing import Any, Dict, Mapping, Optional, Sequence
 
 import yaml
 
@@ -26,6 +25,8 @@ class DefaultConfigKeys:
     cloud_account: str = "cloud_account.name"
     cloud_provider: str = "cloud_provider.name"
 
+    license: str = "license"
+
 
 class ConfigProxy:
     def __init__(self, root: "Config", *path: str) -> None:
@@ -101,6 +102,20 @@ class Config:
         Returns:
             The config value if found, None otherwise
         """
+        return self._get_value_type(key_path, str)
+
+    def get_sub_config(self, key_path: str) -> Optional[Mapping[str, Any]]:
+        """Gets a subconfig from the config using dot notation.
+
+        Args:
+            key_path: the dot-separated path to the subconfig (e.g. "license")
+
+        Returns:
+            The subconfig if found, None otherwise
+        """
+        return self._get_value_type(key_path, Mapping)
+
+    def _get_value_type(self, key_path: str, subtype: type) -> Optional[Any]:
         config = self._load_config()
         if not isinstance(config, Mapping):
             return None
@@ -111,7 +126,7 @@ class Config:
             if not isinstance(curr, dict) or k not in curr:
                 return None
             curr = curr[k]
-        return curr if isinstance(curr, str) else None
+        return curr if isinstance(curr, subtype) else None
 
     def __getattr__(self, name: str) -> ConfigProxy:
         """Returns a proxy to the actual values to allow for nested access.

lightning_sdk/utils/license.py

@@ -0,0 +1,13 @@
+from functools import lru_cache
+
+from lightning_sdk.api.license_api import LicenseApi
+
+
+class License:
+    def __init__(self, license_key: str, product_name: str) -> None:
+        self.license_key = license_key
+        self.product_name = product_name
+
+    @lru_cache(maxsize=1)  # noqa: B019
+    def validate(self) -> bool:
+        return LicenseApi(self.license_key).validate_license(self.license_key, self.product_name)

lightning_sdk/utils/resolve.py

@@ -75,7 +75,9 @@ def _resolve_deprecated_provider(
     return cloud_provider
 
 
-def _resolve_deprecated_cluster(cloud_account: Optional[str], cluster: Optional[str]) -> Optional[str]:
+def _resolve_deprecated_cluster(
+    cloud_account: Optional[str], cluster: Optional[str], current_cloud_account: Optional[str] = None
+) -> Optional[str]:
     if cluster is not None:
         if cloud_account is not None:
             raise ValueError(
@@ -96,6 +98,9 @@ def _resolve_deprecated_cluster(cloud_account: Optional[str], cluster: Optional[
         config = Config()
         cloud_account = config.get_value(DefaultConfigKeys.cloud_account)
 
+        if cloud_account is None:
+            cloud_account = current_cloud_account
+
     return cloud_account
 
 

{lightning_sdk-2025.10.14.dist-info → lightning_sdk-2025.10.23.dist-info}/METADATA

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