lightning-sdk 2025.8.6rc2__py3-none-any.whl → 2025.11.5__py3-none-any.whl

lightning_sdk/lightning_cloud/openapi/models/v1_validate_license_response.py

@@ -0,0 +1,123 @@
+# coding: utf-8
+
+"""
+    external/v1/auth_service.proto
+
+    No description provided (generated by Swagger Codegen https://github.com/swagger-api/swagger-codegen)  # noqa: E501
+
+    OpenAPI spec version: version not set
+    
+    Generated by: https://github.com/swagger-api/swagger-codegen.git
+
+    NOTE
+    ----
+    standard swagger-codegen-cli for this python client has been modified
+    by custom templates. The purpose of these templates is to include
+    typing information in the API and Model code. Please refer to the
+    main grid repository for more info
+"""
+
+import pprint
+import re  # noqa: F401
+
+from typing import TYPE_CHECKING
+
+import six
+
+if TYPE_CHECKING:
+    from datetime import datetime
+    from lightning_sdk.lightning_cloud.openapi.models import *
+
+class V1ValidateLicenseResponse(object):
+    """NOTE: This class is auto generated by the swagger code generator program.
+
+    Do not edit the class manually.
+    """
+    """
+    Attributes:
+      swagger_types (dict): The key is attribute name
+                            and the value is attribute type.
+      attribute_map (dict): The key is attribute name
+                            and the value is json key in definition.
+    """
+    swagger_types = {
+        'is_valid': 'bool'
+    }
+
+    attribute_map = {
+        'is_valid': 'isValid'
+    }
+
+    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 is_valid is not None:
+            self.is_valid = is_valid
+
+    @property
+    def is_valid(self) -> 'bool':
+        """Gets the is_valid of this V1ValidateLicenseResponse.  # noqa: E501
+
+
+        :return: The is_valid of this V1ValidateLicenseResponse.  # noqa: E501
+        :rtype: bool
+        """
+        return self._is_valid
+
+    @is_valid.setter
+    def is_valid(self, is_valid: 'bool'):
+        """Sets the is_valid of this V1ValidateLicenseResponse.
+
+
+        :param is_valid: The is_valid of this V1ValidateLicenseResponse.  # noqa: E501
+        :type: bool
+        """
+
+        self._is_valid = is_valid
+
+    def to_dict(self) -> dict:
+        """Returns the model properties as a dict"""
+        result = {}
+
+        for attr, _ in six.iteritems(self.swagger_types):
+            value = getattr(self, attr)
+            if isinstance(value, list):
+                result[attr] = list(map(
+                    lambda x: x.to_dict() if hasattr(x, "to_dict") else x,
+                    value
+                ))
+            elif hasattr(value, "to_dict"):
+                result[attr] = value.to_dict()
+            elif isinstance(value, dict):
+                result[attr] = dict(map(
+                    lambda item: (item[0], item[1].to_dict())
+                    if hasattr(item[1], "to_dict") else item,
+                    value.items()
+                ))
+            else:
+                result[attr] = value
+        if issubclass(V1ValidateLicenseResponse, dict):
+            for key, value in self.items():
+                result[key] = value
+
+        return result
+
+    def to_str(self) -> str:
+        """Returns the string representation of the model"""
+        return pprint.pformat(self.to_dict())
+
+    def __repr__(self) -> str:
+        """For `print` and `pprint`"""
+        return self.to_str()
+
+    def __eq__(self, other: 'V1ValidateLicenseResponse') -> bool:
+        """Returns true if both objects are equal"""
+        if not isinstance(other, V1ValidateLicenseResponse):
+            return False
+
+        return self.__dict__ == other.__dict__
+
+    def __ne__(self, other: 'V1ValidateLicenseResponse') -> bool:
+        """Returns true if both objects are not equal"""
+        return not self == other

lightning_sdk/lightning_cloud/openapi/models/v1_validate_storage_transfer_response.py

@@ -0,0 +1,123 @@
+# coding: utf-8
+
+"""
+    external/v1/auth_service.proto
+
+    No description provided (generated by Swagger Codegen https://github.com/swagger-api/swagger-codegen)  # noqa: E501
+
+    OpenAPI spec version: version not set
+    
+    Generated by: https://github.com/swagger-api/swagger-codegen.git
+
+    NOTE
+    ----
+    standard swagger-codegen-cli for this python client has been modified
+    by custom templates. The purpose of these templates is to include
+    typing information in the API and Model code. Please refer to the
+    main grid repository for more info
+"""
+
+import pprint
+import re  # noqa: F401
+
+from typing import TYPE_CHECKING
+
+import six
+
+if TYPE_CHECKING:
+    from datetime import datetime
+    from lightning_sdk.lightning_cloud.openapi.models import *
+
+class V1ValidateStorageTransferResponse(object):
+    """NOTE: This class is auto generated by the swagger code generator program.
+
+    Do not edit the class manually.
+    """
+    """
+    Attributes:
+      swagger_types (dict): The key is attribute name
+                            and the value is attribute type.
+      attribute_map (dict): The key is attribute name
+                            and the value is json key in definition.
+    """
+    swagger_types = {
+        'valid_perms': 'bool'
+    }
+
+    attribute_map = {
+        'valid_perms': 'validPerms'
+    }
+
+    def __init__(self, valid_perms: 'bool' =None):  # noqa: E501
+        """V1ValidateStorageTransferResponse - a model defined in Swagger"""  # noqa: E501
+        self._valid_perms = None
+        self.discriminator = None
+        if valid_perms is not None:
+            self.valid_perms = valid_perms
+
+    @property
+    def valid_perms(self) -> 'bool':
+        """Gets the valid_perms of this V1ValidateStorageTransferResponse.  # noqa: E501
+
+
+        :return: The valid_perms of this V1ValidateStorageTransferResponse.  # noqa: E501
+        :rtype: bool
+        """
+        return self._valid_perms
+
+    @valid_perms.setter
+    def valid_perms(self, valid_perms: 'bool'):
+        """Sets the valid_perms of this V1ValidateStorageTransferResponse.
+
+
+        :param valid_perms: The valid_perms of this V1ValidateStorageTransferResponse.  # noqa: E501
+        :type: bool
+        """
+
+        self._valid_perms = valid_perms
+
+    def to_dict(self) -> dict:
+        """Returns the model properties as a dict"""
+        result = {}
+
+        for attr, _ in six.iteritems(self.swagger_types):
+            value = getattr(self, attr)
+            if isinstance(value, list):
+                result[attr] = list(map(
+                    lambda x: x.to_dict() if hasattr(x, "to_dict") else x,
+                    value
+                ))
+            elif hasattr(value, "to_dict"):
+                result[attr] = value.to_dict()
+            elif isinstance(value, dict):
+                result[attr] = dict(map(
+                    lambda item: (item[0], item[1].to_dict())
+                    if hasattr(item[1], "to_dict") else item,
+                    value.items()
+                ))
+            else:
+                result[attr] = value
+        if issubclass(V1ValidateStorageTransferResponse, dict):
+            for key, value in self.items():
+                result[key] = value
+
+        return result
+
+    def to_str(self) -> str:
+        """Returns the string representation of the model"""
+        return pprint.pformat(self.to_dict())
+
+    def __repr__(self) -> str:
+        """For `print` and `pprint`"""
+        return self.to_str()
+
+    def __eq__(self, other: 'V1ValidateStorageTransferResponse') -> bool:
+        """Returns true if both objects are equal"""
+        if not isinstance(other, V1ValidateStorageTransferResponse):
+            return False
+
+        return self.__dict__ == other.__dict__
+
+    def __ne__(self, other: 'V1ValidateStorageTransferResponse') -> bool:
+        """Returns true if both objects are not equal"""
+        return not self == other

lightning_sdk/lightning_cloud/openapi/models/v1_voltage_park_direct_v1.py

@@ -44,22 +44,25 @@ class V1VoltageParkDirectV1(object):
         'credentials_secret_id': 'str',
         'primary_region': 'str',
         'regions': 'list[str]',
-        'reserved_instance_ids': 'list[str]'
+        'reserved_instance_ids': 'list[str]',
+        'subnet_cidr_ranges': 'list[str]'
     }
 
     attribute_map = {
         'credentials_secret_id': 'credentialsSecretId',
         'primary_region': 'primaryRegion',
         'regions': 'regions',
-        'reserved_instance_ids': 'reservedInstanceIds'
+        'reserved_instance_ids': 'reservedInstanceIds',
+        'subnet_cidr_ranges': 'subnetCidrRanges'
     }
 
-    def __init__(self, credentials_secret_id: 'str' =None, primary_region: 'str' =None, regions: 'list[str]' =None, reserved_instance_ids: 'list[str]' =None):  # noqa: E501
+    def __init__(self, credentials_secret_id: 'str' =None, primary_region: 'str' =None, regions: 'list[str]' =None, reserved_instance_ids: 'list[str]' =None, subnet_cidr_ranges: 'list[str]' =None):  # noqa: E501
         """V1VoltageParkDirectV1 - a model defined in Swagger"""  # noqa: E501
         self._credentials_secret_id = None
         self._primary_region = None
         self._regions = None
         self._reserved_instance_ids = None
+        self._subnet_cidr_ranges = None
         self.discriminator = None
         if credentials_secret_id is not None:
             self.credentials_secret_id = credentials_secret_id
@@ -69,6 +72,8 @@ class V1VoltageParkDirectV1(object):
             self.regions = regions
         if reserved_instance_ids is not None:
             self.reserved_instance_ids = reserved_instance_ids
+        if subnet_cidr_ranges is not None:
+            self.subnet_cidr_ranges = subnet_cidr_ranges
 
     @property
     def credentials_secret_id(self) -> 'str':
@@ -156,6 +161,27 @@ class V1VoltageParkDirectV1(object):
 
         self._reserved_instance_ids = reserved_instance_ids
 
+    @property
+    def subnet_cidr_ranges(self) -> 'list[str]':
+        """Gets the subnet_cidr_ranges of this V1VoltageParkDirectV1.  # noqa: E501
+
+
+        :return: The subnet_cidr_ranges of this V1VoltageParkDirectV1.  # noqa: E501
+        :rtype: list[str]
+        """
+        return self._subnet_cidr_ranges
+
+    @subnet_cidr_ranges.setter
+    def subnet_cidr_ranges(self, subnet_cidr_ranges: 'list[str]'):
+        """Sets the subnet_cidr_ranges of this V1VoltageParkDirectV1.
+
+
+        :param subnet_cidr_ranges: The subnet_cidr_ranges of this V1VoltageParkDirectV1.  # noqa: E501
+        :type: list[str]
+        """
+
+        self._subnet_cidr_ranges = subnet_cidr_ranges
+
     def to_dict(self) -> dict:
         """Returns the model properties as a dict"""
         result = {}

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/lightning_cloud/utils/data_connection.py

@@ -1,7 +1,7 @@
 import os
 from time import sleep, time
 from lightning_sdk.lightning_cloud import rest_client
-from lightning_sdk.lightning_cloud.openapi import Create, V1AwsDataConnection, V1S3FolderDataConnection, V1EfsConfig, V1GcpDataConnection, V1FilestoreDataConnection, V1DataConnectionTier
+from lightning_sdk.lightning_cloud.openapi import Create, V1AwsDataConnection, V1S3FolderDataConnection, V1EfsConfig, V1GcpDataConnection, V1FilestoreDataConnection, V1DataConnectionTier, V1R2DataConnection
 from lightning_sdk.lightning_cloud.openapi.rest import ApiException
 import urllib3
 
@@ -327,6 +327,56 @@ def create_filestore_folder(folder_name: str, region: str, capacity_gb: int = 10
         sleep(1)
 
 
+def create_cloud_agnostic_folder(folder_name: str, create_timeout: int = 30) -> None:
+    """
+    Utility function to create a Cloud-Agnostic (R2) folder.
+
+    Args:
+        folder_name: The name of the folder to create.
+        create_timeout (int): The timeout for the folder creation.
+    """
+    client = rest_client.LightningClient(retry=False)
+
+    project_id = os.getenv("LIGHTNING_CLOUD_PROJECT_ID")
+    cluster_id = os.getenv("LIGHTNING_CLUSTER_ID")
+
+    # Get existing data connections
+    data_connections = client.data_connection_service_list_data_connections(project_id).data_connections
+
+    for connection in data_connections:
+        existing_folder_name = getattr(connection, 'name', None)
+        isCloudAgnostic = getattr(connection, 'r2_folder', None) is not None
+
+        if existing_folder_name == folder_name and isCloudAgnostic:
+            return
+
+    # If we get here, no matching folder was found, proceed with creation
+    body = Create(
+        name=folder_name,
+        create_resources=True,
+        r2=V1R2DataConnection(name=folder_name),
+    )
+    try:
+        connection = client.data_connection_service_create_data_connection(body, project_id)
+    except ApiException as e:
+        # Note: This function can be called in a distributed way.
+        # There is a race condition where one machine might create the entry before another machine
+        # and this request would fail with duplicated key
+        # In this case, it is fine not to raise
+        if'duplicate key value violates unique constraint' in str(e.body):
+            return
+        raise e from None
+
+    except urllib3.exceptions.HTTPError as e:
+        raise e from None
+
+    # Wait for the filesystem picks up the newly added Cloud-Agnostic folder
+    start = time()
+
+    while not os.path.isdir(f"/teamspace/lightning_storage/{folder_name}") and (time() - start) < create_timeout:
+        sleep(1)
+
+
 def delete_data_connection(name: str):
     """Utility to delete a data connection