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

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

@@ -6,7 +6,6 @@ from typing import Any, ClassVar, Optional, Tuple
 class CloudProvider(Enum):
     AWS = "AWS"
     GCP = "GCP"
-    VULTR = "VULTR"
     LAMBDA_LABS = "LAMBDA_LABS"
     DGX = "DGX"
     VOLTAGE_PARK = "VOLTAGE_PARK"
@@ -35,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"]
@@ -163,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 (
@@ -242,6 +242,14 @@ class Studio:
             _get_org_id(self._teamspace),
         )
 
+    @property
+    def public_ip(self) -> Optional[str]:
+        """Returns the public IP address of the machine the Studio is running on."""
+        return self._studio_api.get_public_ip(
+            self._studio.id,
+            self._teamspace.id,
+        )
+
     @property
     def interruptible(self) -> bool:
         """Returns whether the Studio is running on a interruptible instance."""
@@ -345,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.
 
@@ -378,6 +389,7 @@ class Studio:
             teamspace_id=self._teamspace.id,
             target_teamspace_id=target_teamspace_id,
             machine=machine,
+            new_name=name,
         )
         return Studio(**kwargs)
 
@@ -651,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.8.dist-info → lightning_sdk-2025.10.22.dist-info}/METADATA

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