lightning-sdk 0.1.49__py3-none-any.whl → 2025.11.5__py3-none-any.whl

lightning_sdk/lightning_cloud/openapi/models/version_uploads_body1.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 VersionUploadsBody1(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 = {
+        'filepath': 'str'
+    }
+
+    attribute_map = {
+        'filepath': 'filepath'
+    }
+
+    def __init__(self, filepath: 'str' =None):  # noqa: E501
+        """VersionUploadsBody1 - a model defined in Swagger"""  # noqa: E501
+        self._filepath = None
+        self.discriminator = None
+        if filepath is not None:
+            self.filepath = filepath
+
+    @property
+    def filepath(self) -> 'str':
+        """Gets the filepath of this VersionUploadsBody1.  # noqa: E501
+
+
+        :return: The filepath of this VersionUploadsBody1.  # noqa: E501
+        :rtype: str
+        """
+        return self._filepath
+
+    @filepath.setter
+    def filepath(self, filepath: 'str'):
+        """Sets the filepath of this VersionUploadsBody1.
+
+
+        :param filepath: The filepath of this VersionUploadsBody1.  # noqa: E501
+        :type: str
+        """
+
+        self._filepath = filepath
+
+    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(VersionUploadsBody1, 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: 'VersionUploadsBody1') -> bool:
+        """Returns true if both objects are equal"""
+        if not isinstance(other, VersionUploadsBody1):
+            return False
+
+        return self.__dict__ == other.__dict__
+
+    def __ne__(self, other: 'VersionUploadsBody1') -> bool:
+        """Returns true if both objects are not equal"""
+        return not self == other

lightning_sdk/lightning_cloud/openapi/models/versions_version_body1.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 VersionsVersionBody1(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 = {
+        'name': 'str'
+    }
+
+    attribute_map = {
+        'name': 'name'
+    }
+
+    def __init__(self, name: 'str' =None):  # noqa: E501
+        """VersionsVersionBody1 - a model defined in Swagger"""  # noqa: E501
+        self._name = None
+        self.discriminator = None
+        if name is not None:
+            self.name = name
+
+    @property
+    def name(self) -> 'str':
+        """Gets the name of this VersionsVersionBody1.  # noqa: E501
+
+
+        :return: The name of this VersionsVersionBody1.  # noqa: E501
+        :rtype: str
+        """
+        return self._name
+
+    @name.setter
+    def name(self, name: 'str'):
+        """Sets the name of this VersionsVersionBody1.
+
+
+        :param name: The name of this VersionsVersionBody1.  # noqa: E501
+        :type: str
+        """
+
+        self._name = name
+
+    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(VersionsVersionBody1, 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: 'VersionsVersionBody1') -> bool:
+        """Returns true if both objects are equal"""
+        if not isinstance(other, VersionsVersionBody1):
+            return False
+
+        return self.__dict__ == other.__dict__
+
+    def __ne__(self, other: 'VersionsVersionBody1') -> bool:
+        """Returns true if both objects are not equal"""
+        return not self == other

lightning_sdk/lightning_cloud/openapi/models/volumes_id_body.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 VolumesIdBody(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 = {
+        'volume': 'V1Volume'
+    }
+
+    attribute_map = {
+        'volume': 'volume'
+    }
+
+    def __init__(self, volume: 'V1Volume' =None):  # noqa: E501
+        """VolumesIdBody - a model defined in Swagger"""  # noqa: E501
+        self._volume = None
+        self.discriminator = None
+        if volume is not None:
+            self.volume = volume
+
+    @property
+    def volume(self) -> 'V1Volume':
+        """Gets the volume of this VolumesIdBody.  # noqa: E501
+
+
+        :return: The volume of this VolumesIdBody.  # noqa: E501
+        :rtype: V1Volume
+        """
+        return self._volume
+
+    @volume.setter
+    def volume(self, volume: 'V1Volume'):
+        """Sets the volume of this VolumesIdBody.
+
+
+        :param volume: The volume of this VolumesIdBody.  # noqa: E501
+        :type: V1Volume
+        """
+
+        self._volume = volume
+
+    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(VolumesIdBody, 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: 'VolumesIdBody') -> bool:
+        """Returns true if both objects are equal"""
+        if not isinstance(other, VolumesIdBody):
+            return False
+
+        return self.__dict__ == other.__dict__
+
+    def __ne__(self, other: 'VolumesIdBody') -> bool:
+        """Returns true if both objects are not equal"""
+        return not self == other

lightning_sdk/lightning_cloud/rest_client.py

@@ -2,37 +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,
+    LitLoggerServiceApi,
+    LitRegistryServiceApi,
+    ModelsStoreApi,
+    OrganizationsServiceApi,
+    PipelinesServiceApi,
+    ProductLicenseServiceApi,
     ProjectsServiceApi,
+    SchedulesServiceApi,
+    SDKCommandHistoryServiceApi,
     SecretServiceApi,
-    SSHPublicKeyServiceApi,
-    DatasetServiceApi,
-    OrganizationsServiceApi,
-    UserServiceApi,
-    BillingServiceApi,
-    EndpointServiceApi,
     SlurmJobsUserServiceApi,
-    LitLoggerServiceApi,
-    JobsServiceApi,
-    AssistantsServiceApi,
+    SSHPublicKeyServiceApi,
     StorageServiceApi,
-    DeploymentTemplatesServiceApi,
-    ModelsStoreApi,
-    LitRegistryServiceApi
+    UserServiceApi,
 )
 from lightning_sdk.lightning_cloud.openapi.rest import ApiException
 from lightning_sdk.lightning_cloud.source_code.logs_socket_api import LightningLogsSocketAPI
@@ -40,7 +46,7 @@ from lightning_sdk.lightning_cloud.source_code.logs_socket_api import LightningL
 logger = logging.getLogger(__name__)
 
 
-def create_swagger_client(check_context=True):
+def create_swagger_client(check_context: bool = True, with_auth: bool = True):
     """
     Create the swagger client to use the autogenerated code
 
@@ -49,6 +55,8 @@ def create_swagger_client(check_context=True):
     check_context: bool
         If true, check if the context is set. It's only false for APIs that
         doesn't need the context information i.e login
+    with_auth: bool
+        If true, it uses user's Authorization for the api client
     """
     if check_context and not env.CONTEXT:
         raise RuntimeError(
@@ -63,40 +71,46 @@ def create_swagger_client(check_context=True):
     # if present in /grid-cli/grid/cli/cli/artifacts.py
     configuration.ssl_ca_cert = env.SSL_CA_CERT
     api_client = ApiClient(configuration)
-    api_client.default_headers["Authorization"] = Auth().authenticate()
+    if with_auth:
+        api_client.default_headers["Authorization"] = Auth().authenticate()
     api_client.user_agent = f"Grid-CLI-{env.VERSION}"
     return api_client
 
 
 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
+    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):
-        api_client = api_client if api_client else create_swagger_client()
+    def __init__(self, api_client: Optional[ApiClient] = None, with_auth: bool = True):
+        api_client = api_client if api_client else create_swagger_client(with_auth=with_auth)
         api_client.request = request_auth_warning_wrapper(api_client.request)
         super().__init__(api_client)
 
@@ -171,24 +185,23 @@ class LightningClient(GridRestClient):
     Args:
         retry: Whether API calls should follow a retry mechanism with exponential backoff.
         max_tries: Maximum number of attempts (or -1 to retry forever).
+        with_auth: Whether to use user's Authorization for the api client.
 
     """
 
     def __init__(self,
                  retry: bool = True,
-                 max_tries: Optional[int] = None) -> None:
-        super().__init__(api_client=create_swagger_client())
+                 max_tries: Optional[int] = None,
+                 with_auth: bool = True) -> None:
+        super().__init__(api_client=create_swagger_client(with_auth=with_auth))
         if retry:
             for base_class in GridRestClient.__mro__:
                 for name, attribute in base_class.__dict__.items():
-                    if callable(
-                            attribute) and attribute.__name__ != "__init__":
+                    if callable(attribute) and attribute.__name__ != "__init__":
                         setattr(
                             self,
                             name,
-                            _retry_wrapper(self,
-                                           attribute,
-                                           max_tries=max_tries),
+                            _retry_wrapper(self, attribute, max_tries=max_tries),
                         )
 
 

lightning_sdk/lightning_cloud/source_code/logs_socket_api.py

@@ -11,11 +11,17 @@ class LightningLogsSocketAPI:
         if api_client is None:
             api_client = ApiClient()
         self.api_client = api_client
+        self._auth = None
+        self._auth_service = None
+
+    def _init_auth(self):
         self._auth = Auth()
         self._auth.authenticate()
-        self._auth_service = AuthServiceApi(api_client)
+        self._auth_service = AuthServiceApi(self.api_client)
 
     def _get_api_token(self):
+        if not self._auth_service:
+            self._init_auth()
         token_resp = self._auth_service.auth_service_login(body=V1LoginRequest(
             username=self._auth.user_id,
             api_key=self._auth.api_key,
@@ -83,13 +89,12 @@ class LightningLogsSocketAPI:
         -------
         WebSocketApp of the wanted socket
         """
-        _token = self._get_api_token()
         clean_ws_host = urlparse(self.api_client.configuration.host).netloc
         socket_url = self._socket_url(
             host=clean_ws_host,
             project_id=project_id,
             app_id=app_id,
-            token=_token,
+            token=self._get_api_token(),
             component=component,
         )