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

lightning_sdk/lightning_cloud/openapi/models/v1_user_features.py

@@ -45,7 +45,6 @@ class V1UserFeatures(object):
         'agents_v2': 'bool',
         'ai_hub_monetization': 'bool',
         'auto_fast_load': 'bool',
-        'auto_join_orgs': 'bool',
         'b2c_experience': 'bool',
         'byo_machine_type': 'bool',
         'cap_add': 'list[str]',
@@ -87,10 +86,11 @@ class V1UserFeatures(object):
         'f262': 'bool',
         'f263': 'bool',
         'f264': 'bool',
+        'f265': 'bool',
+        'f266': 'bool',
         'fair_share': 'bool',
         'featured_studios_admin': 'bool',
         'gcs_connections_optimized': 'bool',
-        'instant_capacity_reservation': 'bool',
         'job_artifacts_v2': 'bool',
         'kubernetes_cluster_ui': 'bool',
         'kubernetes_clusters': 'bool',
@@ -127,7 +127,6 @@ class V1UserFeatures(object):
         'studio_config': 'bool',
         'studio_sharing_v2': 'bool',
         'studio_version_visibility': 'bool',
-        'trainium2': 'bool',
         'vultr': 'bool',
         'weka': 'bool',
         'writable_s3_connections': 'bool'
@@ -138,7 +137,6 @@ class V1UserFeatures(object):
         'agents_v2': 'agentsV2',
         'ai_hub_monetization': 'aiHubMonetization',
         'auto_fast_load': 'autoFastLoad',
-        'auto_join_orgs': 'autoJoinOrgs',
         'b2c_experience': 'b2cExperience',
         'byo_machine_type': 'byoMachineType',
         'cap_add': 'capAdd',
@@ -180,10 +178,11 @@ class V1UserFeatures(object):
         'f262': 'f262',
         'f263': 'f263',
         'f264': 'f264',
+        'f265': 'f265',
+        'f266': 'f266',
         'fair_share': 'fairShare',
         'featured_studios_admin': 'featuredStudiosAdmin',
         'gcs_connections_optimized': 'gcsConnectionsOptimized',
-        'instant_capacity_reservation': 'instantCapacityReservation',
         'job_artifacts_v2': 'jobArtifactsV2',
         'kubernetes_cluster_ui': 'kubernetesClusterUi',
         'kubernetes_clusters': 'kubernetesClusters',
@@ -220,19 +219,17 @@ class V1UserFeatures(object):
         'studio_config': 'studioConfig',
         'studio_sharing_v2': 'studioSharingV2',
         'studio_version_visibility': 'studioVersionVisibility',
-        'trainium2': 'trainium2',
         'vultr': 'vultr',
         'weka': 'weka',
         'writable_s3_connections': 'writableS3Connections'
     }
 
-    def __init__(self, affiliate_links: 'bool' =None, agents_v2: 'bool' =None, ai_hub_monetization: 'bool' =None, auto_fast_load: 'bool' =None, auto_join_orgs: 'bool' =None, b2c_experience: 'bool' =None, byo_machine_type: 'bool' =None, cap_add: 'list[str]' =None, cap_drop: 'list[str]' =None, capacity_reservation_byoc: 'bool' =None, capacity_reservation_dry_run: 'bool' =None, chat_models: 'bool' =None, cloudspace_schedules: 'bool' =None, code_tab: 'bool' =None, collab_screen_sharing: 'bool' =None, control_center_monitoring: 'bool' =None, cost_attribution_settings: 'bool' =None, custom_app_domain: 'bool' =None, datasets: 'bool' =None, default_one_cluster: 'bool' =None, deployment_persistent_disk: 'bool' =None, drive_v2: 'bool' =None, enterprise_compute_admin: 'bool' =None, f227: 'bool' =None, f234: 'bool' =None, f236: 'bool' =None, f239: 'bool' =None, f240: 'bool' =None, f241: 'bool' =None, f243: 'bool' =None, f245: 'bool' =None, f247: 'bool' =None, f248: 'bool' =None, f250: 'bool' =None, f252: 'bool' =None, f253: 'bool' =None, f254: 'bool' =None, f255: 'bool' =None, f257: 'bool' =None, f258: 'bool' =None, f259: 'bool' =None, f260: 'bool' =None, f261: 'bool' =None, f262: 'bool' =None, f263: 'bool' =None, f264: 'bool' =None, fair_share: 'bool' =None, featured_studios_admin: 'bool' =None, gcs_connections_optimized: 'bool' =None, instant_capacity_reservation: 'bool' =None, job_artifacts_v2: 'bool' =None, kubernetes_cluster_ui: 'bool' =None, kubernetes_clusters: 'bool' =None, landing_studios: 'bool' =None, lit_logger: 'bool' =None, marketplace: 'bool' =None, mmt_fault_tolerance: 'bool' =None, mmt_strategy_selector: 'bool' =None, model_api_dashboard: 'bool' =None, multiple_studio_versions: 'bool' =None, nerf_fs_nonpaying: 'bool' =None, org_level_member_permissions: 'bool' =None, org_usage_limits: 'bool' =None, persistent_disk: 'bool' =None, plugin_distributed: 'bool' =None, plugin_inference: 'bool' =None, plugin_label_studio: 'bool' =None, plugin_langflow: 'bool' =None, plugin_python_profiler: 'bool' =None, plugin_sweeps: 'bool' =None, pricing_updates: 'bool' =None, product_generator: 'bool' =None, product_license: 'bool' =None, project_selector: 'bool' =None, publish_pipelines: 'bool' =None, reserved_machines_tab: 'bool' =None, restartable_jobs: 'bool' =None, runnable_public_studio_page: 'bool' =None, security_docs: 'bool' =None, show_dev_admin: 'bool' =None, slurm: 'bool' =None, specialised_studios: 'bool' =None, storage_overuse_deletion: 'bool' =None, studio_config: 'bool' =None, studio_sharing_v2: 'bool' =None, studio_version_visibility: 'bool' =None, trainium2: 'bool' =None, vultr: 'bool' =None, weka: 'bool' =None, writable_s3_connections: 'bool' =None):  # noqa: E501
+    def __init__(self, affiliate_links: 'bool' =None, agents_v2: 'bool' =None, ai_hub_monetization: 'bool' =None, auto_fast_load: 'bool' =None, b2c_experience: 'bool' =None, byo_machine_type: 'bool' =None, cap_add: 'list[str]' =None, cap_drop: 'list[str]' =None, capacity_reservation_byoc: 'bool' =None, capacity_reservation_dry_run: 'bool' =None, chat_models: 'bool' =None, cloudspace_schedules: 'bool' =None, code_tab: 'bool' =None, collab_screen_sharing: 'bool' =None, control_center_monitoring: 'bool' =None, cost_attribution_settings: 'bool' =None, custom_app_domain: 'bool' =None, datasets: 'bool' =None, default_one_cluster: 'bool' =None, deployment_persistent_disk: 'bool' =None, drive_v2: 'bool' =None, enterprise_compute_admin: 'bool' =None, f227: 'bool' =None, f234: 'bool' =None, f236: 'bool' =None, f239: 'bool' =None, f240: 'bool' =None, f241: 'bool' =None, f243: 'bool' =None, f245: 'bool' =None, f247: 'bool' =None, f248: 'bool' =None, f250: 'bool' =None, f252: 'bool' =None, f253: 'bool' =None, f254: 'bool' =None, f255: 'bool' =None, f257: 'bool' =None, f258: 'bool' =None, f259: 'bool' =None, f260: 'bool' =None, f261: 'bool' =None, f262: 'bool' =None, f263: 'bool' =None, f264: 'bool' =None, f265: 'bool' =None, f266: 'bool' =None, fair_share: 'bool' =None, featured_studios_admin: 'bool' =None, gcs_connections_optimized: 'bool' =None, job_artifacts_v2: 'bool' =None, kubernetes_cluster_ui: 'bool' =None, kubernetes_clusters: 'bool' =None, landing_studios: 'bool' =None, lit_logger: 'bool' =None, marketplace: 'bool' =None, mmt_fault_tolerance: 'bool' =None, mmt_strategy_selector: 'bool' =None, model_api_dashboard: 'bool' =None, multiple_studio_versions: 'bool' =None, nerf_fs_nonpaying: 'bool' =None, org_level_member_permissions: 'bool' =None, org_usage_limits: 'bool' =None, persistent_disk: 'bool' =None, plugin_distributed: 'bool' =None, plugin_inference: 'bool' =None, plugin_label_studio: 'bool' =None, plugin_langflow: 'bool' =None, plugin_python_profiler: 'bool' =None, plugin_sweeps: 'bool' =None, pricing_updates: 'bool' =None, product_generator: 'bool' =None, product_license: 'bool' =None, project_selector: 'bool' =None, publish_pipelines: 'bool' =None, reserved_machines_tab: 'bool' =None, restartable_jobs: 'bool' =None, runnable_public_studio_page: 'bool' =None, security_docs: 'bool' =None, show_dev_admin: 'bool' =None, slurm: 'bool' =None, specialised_studios: 'bool' =None, storage_overuse_deletion: 'bool' =None, studio_config: 'bool' =None, studio_sharing_v2: 'bool' =None, studio_version_visibility: 'bool' =None, vultr: 'bool' =None, weka: 'bool' =None, writable_s3_connections: 'bool' =None):  # noqa: E501
         """V1UserFeatures - a model defined in Swagger"""  # noqa: E501
         self._affiliate_links = None
         self._agents_v2 = None
         self._ai_hub_monetization = None
         self._auto_fast_load = None
-        self._auto_join_orgs = None
         self._b2c_experience = None
         self._byo_machine_type = None
         self._cap_add = None
@@ -274,10 +271,11 @@ class V1UserFeatures(object):
         self._f262 = None
         self._f263 = None
         self._f264 = None
+        self._f265 = None
+        self._f266 = None
         self._fair_share = None
         self._featured_studios_admin = None
         self._gcs_connections_optimized = None
-        self._instant_capacity_reservation = None
         self._job_artifacts_v2 = None
         self._kubernetes_cluster_ui = None
         self._kubernetes_clusters = None
@@ -314,7 +312,6 @@ class V1UserFeatures(object):
         self._studio_config = None
         self._studio_sharing_v2 = None
         self._studio_version_visibility = None
-        self._trainium2 = None
         self._vultr = None
         self._weka = None
         self._writable_s3_connections = None
@@ -327,8 +324,6 @@ class V1UserFeatures(object):
             self.ai_hub_monetization = ai_hub_monetization
         if auto_fast_load is not None:
             self.auto_fast_load = auto_fast_load
-        if auto_join_orgs is not None:
-            self.auto_join_orgs = auto_join_orgs
         if b2c_experience is not None:
             self.b2c_experience = b2c_experience
         if byo_machine_type is not None:
@@ -411,14 +406,16 @@ class V1UserFeatures(object):
             self.f263 = f263
         if f264 is not None:
             self.f264 = f264
+        if f265 is not None:
+            self.f265 = f265
+        if f266 is not None:
+            self.f266 = f266
         if fair_share is not None:
             self.fair_share = fair_share
         if featured_studios_admin is not None:
             self.featured_studios_admin = featured_studios_admin
         if gcs_connections_optimized is not None:
             self.gcs_connections_optimized = gcs_connections_optimized
-        if instant_capacity_reservation is not None:
-            self.instant_capacity_reservation = instant_capacity_reservation
         if job_artifacts_v2 is not None:
             self.job_artifacts_v2 = job_artifacts_v2
         if kubernetes_cluster_ui is not None:
@@ -491,8 +488,6 @@ class V1UserFeatures(object):
             self.studio_sharing_v2 = studio_sharing_v2
         if studio_version_visibility is not None:
             self.studio_version_visibility = studio_version_visibility
-        if trainium2 is not None:
-            self.trainium2 = trainium2
         if vultr is not None:
             self.vultr = vultr
         if weka is not None:
@@ -584,27 +579,6 @@ class V1UserFeatures(object):
 
         self._auto_fast_load = auto_fast_load
 
-    @property
-    def auto_join_orgs(self) -> 'bool':
-        """Gets the auto_join_orgs of this V1UserFeatures.  # noqa: E501
-
-
-        :return: The auto_join_orgs of this V1UserFeatures.  # noqa: E501
-        :rtype: bool
-        """
-        return self._auto_join_orgs
-
-    @auto_join_orgs.setter
-    def auto_join_orgs(self, auto_join_orgs: 'bool'):
-        """Sets the auto_join_orgs of this V1UserFeatures.
-
-
-        :param auto_join_orgs: The auto_join_orgs of this V1UserFeatures.  # noqa: E501
-        :type: bool
-        """
-
-        self._auto_join_orgs = auto_join_orgs
-
     @property
     def b2c_experience(self) -> 'bool':
         """Gets the b2c_experience of this V1UserFeatures.  # noqa: E501
@@ -1466,6 +1440,48 @@ class V1UserFeatures(object):
 
         self._f264 = f264
 
+    @property
+    def f265(self) -> 'bool':
+        """Gets the f265 of this V1UserFeatures.  # noqa: E501
+
+
+        :return: The f265 of this V1UserFeatures.  # noqa: E501
+        :rtype: bool
+        """
+        return self._f265
+
+    @f265.setter
+    def f265(self, f265: 'bool'):
+        """Sets the f265 of this V1UserFeatures.
+
+
+        :param f265: The f265 of this V1UserFeatures.  # noqa: E501
+        :type: bool
+        """
+
+        self._f265 = f265
+
+    @property
+    def f266(self) -> 'bool':
+        """Gets the f266 of this V1UserFeatures.  # noqa: E501
+
+
+        :return: The f266 of this V1UserFeatures.  # noqa: E501
+        :rtype: bool
+        """
+        return self._f266
+
+    @f266.setter
+    def f266(self, f266: 'bool'):
+        """Sets the f266 of this V1UserFeatures.
+
+
+        :param f266: The f266 of this V1UserFeatures.  # noqa: E501
+        :type: bool
+        """
+
+        self._f266 = f266
+
     @property
     def fair_share(self) -> 'bool':
         """Gets the fair_share of this V1UserFeatures.  # noqa: E501
@@ -1529,27 +1545,6 @@ class V1UserFeatures(object):
 
         self._gcs_connections_optimized = gcs_connections_optimized
 
-    @property
-    def instant_capacity_reservation(self) -> 'bool':
-        """Gets the instant_capacity_reservation of this V1UserFeatures.  # noqa: E501
-
-
-        :return: The instant_capacity_reservation of this V1UserFeatures.  # noqa: E501
-        :rtype: bool
-        """
-        return self._instant_capacity_reservation
-
-    @instant_capacity_reservation.setter
-    def instant_capacity_reservation(self, instant_capacity_reservation: 'bool'):
-        """Sets the instant_capacity_reservation of this V1UserFeatures.
-
-
-        :param instant_capacity_reservation: The instant_capacity_reservation of this V1UserFeatures.  # noqa: E501
-        :type: bool
-        """
-
-        self._instant_capacity_reservation = instant_capacity_reservation
-
     @property
     def job_artifacts_v2(self) -> 'bool':
         """Gets the job_artifacts_v2 of this V1UserFeatures.  # noqa: E501
@@ -2306,27 +2301,6 @@ class V1UserFeatures(object):
 
         self._studio_version_visibility = studio_version_visibility
 
-    @property
-    def trainium2(self) -> 'bool':
-        """Gets the trainium2 of this V1UserFeatures.  # noqa: E501
-
-
-        :return: The trainium2 of this V1UserFeatures.  # noqa: E501
-        :rtype: bool
-        """
-        return self._trainium2
-
-    @trainium2.setter
-    def trainium2(self, trainium2: 'bool'):
-        """Sets the trainium2 of this V1UserFeatures.
-
-
-        :param trainium2: The trainium2 of this V1UserFeatures.  # noqa: E501
-        :type: bool
-        """
-
-        self._trainium2 = trainium2
-
     @property
     def vultr(self) -> 'bool':
         """Gets the vultr of this V1UserFeatures.  # noqa: E501

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)

lightning_sdk/studio.py

@@ -119,7 +119,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 (
@@ -353,7 +353,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 +389,7 @@ class Studio:
             teamspace_id=self._teamspace.id,
             target_teamspace_id=target_teamspace_id,
             machine=machine,
+            new_name=name,
         )
         return Studio(**kwargs)
 
@@ -659,6 +663,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/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-2025.10.14.dist-info → lightning_sdk-2025.10.22.dist-info}/METADATA

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