lightning-sdk 0.1.39__py3-none-any.whl → 0.1.40__py3-none-any.whl

lightning_sdk/lightning_cloud/openapi/models/v1_deployment_api.py

@@ -0,0 +1,201 @@
+# 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 V1DeploymentAPI(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 = {
+        'body': 'V1Body',
+        'method': 'str',
+        'path': 'str',
+        '_query_params': 'list[V1QueryParam]'
+    }
+
+    attribute_map = {
+        'body': 'body',
+        'method': 'method',
+        'path': 'path',
+        '_query_params': 'queryParams'
+    }
+
+    def __init__(self, body: 'V1Body' =None, method: 'str' =None, path: 'str' =None, _query_params: 'list[V1QueryParam]' =None):  # noqa: E501
+        """V1DeploymentAPI - a model defined in Swagger"""  # noqa: E501
+        self._body = None
+        self._method = None
+        self._path = None
+        self.__query_params = None
+        self.discriminator = None
+        if body is not None:
+            self.body = body
+        if method is not None:
+            self.method = method
+        if path is not None:
+            self.path = path
+        if _query_params is not None:
+            self._query_params = _query_params
+
+    @property
+    def body(self) -> 'V1Body':
+        """Gets the body of this V1DeploymentAPI.  # noqa: E501
+
+
+        :return: The body of this V1DeploymentAPI.  # noqa: E501
+        :rtype: V1Body
+        """
+        return self._body
+
+    @body.setter
+    def body(self, body: 'V1Body'):
+        """Sets the body of this V1DeploymentAPI.
+
+
+        :param body: The body of this V1DeploymentAPI.  # noqa: E501
+        :type: V1Body
+        """
+
+        self._body = body
+
+    @property
+    def method(self) -> 'str':
+        """Gets the method of this V1DeploymentAPI.  # noqa: E501
+
+
+        :return: The method of this V1DeploymentAPI.  # noqa: E501
+        :rtype: str
+        """
+        return self._method
+
+    @method.setter
+    def method(self, method: 'str'):
+        """Sets the method of this V1DeploymentAPI.
+
+
+        :param method: The method of this V1DeploymentAPI.  # noqa: E501
+        :type: str
+        """
+
+        self._method = method
+
+    @property
+    def path(self) -> 'str':
+        """Gets the path of this V1DeploymentAPI.  # noqa: E501
+
+
+        :return: The path of this V1DeploymentAPI.  # noqa: E501
+        :rtype: str
+        """
+        return self._path
+
+    @path.setter
+    def path(self, path: 'str'):
+        """Sets the path of this V1DeploymentAPI.
+
+
+        :param path: The path of this V1DeploymentAPI.  # noqa: E501
+        :type: str
+        """
+
+        self._path = path
+
+    @property
+    def _query_params(self) -> 'list[V1QueryParam]':
+        """Gets the _query_params of this V1DeploymentAPI.  # noqa: E501
+
+
+        :return: The _query_params of this V1DeploymentAPI.  # noqa: E501
+        :rtype: list[V1QueryParam]
+        """
+        return self.__query_params
+
+    @_query_params.setter
+    def _query_params(self, _query_params: 'list[V1QueryParam]'):
+        """Sets the _query_params of this V1DeploymentAPI.
+
+
+        :param _query_params: The _query_params of this V1DeploymentAPI.  # noqa: E501
+        :type: list[V1QueryParam]
+        """
+
+        self.__query_params = _query_params
+
+    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(V1DeploymentAPI, 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: 'V1DeploymentAPI') -> bool:
+        """Returns true if both objects are equal"""
+        if not isinstance(other, V1DeploymentAPI):
+            return False
+
+        return self.__dict__ == other.__dict__
+
+    def __ne__(self, other: 'V1DeploymentAPI') -> bool:
+        """Returns true if both objects are not equal"""
+        return not self == other

lightning_sdk/lightning_cloud/openapi/models/v1_get_model_files_response.py

@@ -42,6 +42,7 @@ class V1GetModelFilesResponse(object):
     """
     swagger_types = {
         'filepaths': 'list[str]',
+        'filesizes': 'list[str]',
         'model_id': 'str',
         'project_id': 'str',
         'size_bytes': 'str',
@@ -51,6 +52,7 @@ class V1GetModelFilesResponse(object):
 
     attribute_map = {
         'filepaths': 'filepaths',
+        'filesizes': 'filesizes',
         'model_id': 'modelId',
         'project_id': 'projectId',
         'size_bytes': 'sizeBytes',
@@ -58,9 +60,10 @@ class V1GetModelFilesResponse(object):
         'version': 'version'
     }
 
-    def __init__(self, filepaths: 'list[str]' =None, model_id: 'str' =None, project_id: 'str' =None, size_bytes: 'str' =None, upload_complete: 'bool' =None, version: 'str' =None):  # noqa: E501
+    def __init__(self, filepaths: 'list[str]' =None, filesizes: 'list[str]' =None, model_id: 'str' =None, project_id: 'str' =None, size_bytes: 'str' =None, upload_complete: 'bool' =None, version: 'str' =None):  # noqa: E501
         """V1GetModelFilesResponse - a model defined in Swagger"""  # noqa: E501
         self._filepaths = None
+        self._filesizes = None
         self._model_id = None
         self._project_id = None
         self._size_bytes = None
@@ -69,6 +72,8 @@ class V1GetModelFilesResponse(object):
         self.discriminator = None
         if filepaths is not None:
             self.filepaths = filepaths
+        if filesizes is not None:
+            self.filesizes = filesizes
         if model_id is not None:
             self.model_id = model_id
         if project_id is not None:
@@ -101,6 +106,27 @@ class V1GetModelFilesResponse(object):
 
         self._filepaths = filepaths
 
+    @property
+    def filesizes(self) -> 'list[str]':
+        """Gets the filesizes of this V1GetModelFilesResponse.  # noqa: E501
+
+
+        :return: The filesizes of this V1GetModelFilesResponse.  # noqa: E501
+        :rtype: list[str]
+        """
+        return self._filesizes
+
+    @filesizes.setter
+    def filesizes(self, filesizes: 'list[str]'):
+        """Sets the filesizes of this V1GetModelFilesResponse.
+
+
+        :param filesizes: The filesizes of this V1GetModelFilesResponse.  # noqa: E501
+        :type: list[str]
+        """
+
+        self._filesizes = filesizes
+
     @property
     def model_id(self) -> 'str':
         """Gets the model_id of this V1GetModelFilesResponse.  # noqa: E501

lightning_sdk/lightning_cloud/openapi/models/v1_managed_model.py

@@ -41,6 +41,7 @@ class V1ManagedModel(object):
                             and the value is json key in definition.
     """
     swagger_types = {
+        'abilities': 'V1ManagedModelAbilities',
         'completion_token_price': 'float',
         'description': 'str',
         'endpoint_id': 'str',
@@ -50,6 +51,7 @@ class V1ManagedModel(object):
     }
 
     attribute_map = {
+        'abilities': 'abilities',
         'completion_token_price': 'completionTokenPrice',
         'description': 'description',
         'endpoint_id': 'endpointId',
@@ -58,8 +60,9 @@ class V1ManagedModel(object):
         'prompt_token_price': 'promptTokenPrice'
     }
 
-    def __init__(self, completion_token_price: 'float' =None, description: 'str' =None, endpoint_id: 'str' =None, id: 'str' =None, name: 'str' =None, prompt_token_price: 'float' =None):  # noqa: E501
+    def __init__(self, abilities: 'V1ManagedModelAbilities' =None, completion_token_price: 'float' =None, description: 'str' =None, endpoint_id: 'str' =None, id: 'str' =None, name: 'str' =None, prompt_token_price: 'float' =None):  # noqa: E501
         """V1ManagedModel - a model defined in Swagger"""  # noqa: E501
+        self._abilities = None
         self._completion_token_price = None
         self._description = None
         self._endpoint_id = None
@@ -67,6 +70,8 @@ class V1ManagedModel(object):
         self._name = None
         self._prompt_token_price = None
         self.discriminator = None
+        if abilities is not None:
+            self.abilities = abilities
         if completion_token_price is not None:
             self.completion_token_price = completion_token_price
         if description is not None:
@@ -80,6 +85,27 @@ class V1ManagedModel(object):
         if prompt_token_price is not None:
             self.prompt_token_price = prompt_token_price
 
+    @property
+    def abilities(self) -> 'V1ManagedModelAbilities':
+        """Gets the abilities of this V1ManagedModel.  # noqa: E501
+
+
+        :return: The abilities of this V1ManagedModel.  # noqa: E501
+        :rtype: V1ManagedModelAbilities
+        """
+        return self._abilities
+
+    @abilities.setter
+    def abilities(self, abilities: 'V1ManagedModelAbilities'):
+        """Sets the abilities of this V1ManagedModel.
+
+
+        :param abilities: The abilities of this V1ManagedModel.  # noqa: E501
+        :type: V1ManagedModelAbilities
+        """
+
+        self._abilities = abilities
+
     @property
     def completion_token_price(self) -> 'float':
         """Gets the completion_token_price of this V1ManagedModel.  # noqa: E501

lightning_sdk/lightning_cloud/openapi/models/v1_managed_model_abilities.py

@@ -0,0 +1,175 @@
+# 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 V1ManagedModelAbilities(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 = {
+        'can_call_hub_deployment': 'bool',
+        'can_receive_files': 'bool',
+        'can_receive_images': 'bool'
+    }
+
+    attribute_map = {
+        'can_call_hub_deployment': 'canCallHubDeployment',
+        'can_receive_files': 'canReceiveFiles',
+        'can_receive_images': 'canReceiveImages'
+    }
+
+    def __init__(self, can_call_hub_deployment: 'bool' =None, can_receive_files: 'bool' =None, can_receive_images: 'bool' =None):  # noqa: E501
+        """V1ManagedModelAbilities - a model defined in Swagger"""  # noqa: E501
+        self._can_call_hub_deployment = None
+        self._can_receive_files = None
+        self._can_receive_images = None
+        self.discriminator = None
+        if can_call_hub_deployment is not None:
+            self.can_call_hub_deployment = can_call_hub_deployment
+        if can_receive_files is not None:
+            self.can_receive_files = can_receive_files
+        if can_receive_images is not None:
+            self.can_receive_images = can_receive_images
+
+    @property
+    def can_call_hub_deployment(self) -> 'bool':
+        """Gets the can_call_hub_deployment of this V1ManagedModelAbilities.  # noqa: E501
+
+
+        :return: The can_call_hub_deployment of this V1ManagedModelAbilities.  # noqa: E501
+        :rtype: bool
+        """
+        return self._can_call_hub_deployment
+
+    @can_call_hub_deployment.setter
+    def can_call_hub_deployment(self, can_call_hub_deployment: 'bool'):
+        """Sets the can_call_hub_deployment of this V1ManagedModelAbilities.
+
+
+        :param can_call_hub_deployment: The can_call_hub_deployment of this V1ManagedModelAbilities.  # noqa: E501
+        :type: bool
+        """
+
+        self._can_call_hub_deployment = can_call_hub_deployment
+
+    @property
+    def can_receive_files(self) -> 'bool':
+        """Gets the can_receive_files of this V1ManagedModelAbilities.  # noqa: E501
+
+
+        :return: The can_receive_files of this V1ManagedModelAbilities.  # noqa: E501
+        :rtype: bool
+        """
+        return self._can_receive_files
+
+    @can_receive_files.setter
+    def can_receive_files(self, can_receive_files: 'bool'):
+        """Sets the can_receive_files of this V1ManagedModelAbilities.
+
+
+        :param can_receive_files: The can_receive_files of this V1ManagedModelAbilities.  # noqa: E501
+        :type: bool
+        """
+
+        self._can_receive_files = can_receive_files
+
+    @property
+    def can_receive_images(self) -> 'bool':
+        """Gets the can_receive_images of this V1ManagedModelAbilities.  # noqa: E501
+
+
+        :return: The can_receive_images of this V1ManagedModelAbilities.  # noqa: E501
+        :rtype: bool
+        """
+        return self._can_receive_images
+
+    @can_receive_images.setter
+    def can_receive_images(self, can_receive_images: 'bool'):
+        """Sets the can_receive_images of this V1ManagedModelAbilities.
+
+
+        :param can_receive_images: The can_receive_images of this V1ManagedModelAbilities.  # noqa: E501
+        :type: bool
+        """
+
+        self._can_receive_images = can_receive_images
+
+    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(V1ManagedModelAbilities, 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: 'V1ManagedModelAbilities') -> bool:
+        """Returns true if both objects are equal"""
+        if not isinstance(other, V1ManagedModelAbilities):
+            return False
+
+        return self.__dict__ == other.__dict__
+
+    def __ne__(self, other: 'V1ManagedModelAbilities') -> bool:
+        """Returns true if both objects are not equal"""
+        return not self == other

lightning_sdk/lightning_cloud/openapi/models/v1_model.py

@@ -54,7 +54,8 @@ class V1Model(object):
         'project_id': 'str',
         'tags': 'list[str]',
         'updated_at': 'datetime',
-        'user_id': 'str'
+        'user_id': 'str',
+        'visibility': 'V1ResourceVisibility'
     }
 
     attribute_map = {
@@ -71,10 +72,11 @@ class V1Model(object):
         'project_id': 'projectId',
         'tags': 'tags',
         'updated_at': 'updatedAt',
-        'user_id': 'userId'
+        'user_id': 'userId',
+        'visibility': 'visibility'
     }
 
-    def __init__(self, categories: 'list[str]' =None, created_at: 'datetime' =None, description: 'str' =None, downloads: 'str' =None, id: 'str' =None, latest_version: 'V1ModelVersionArchive' =None, license: 'str' =None, model_metadata: 'dict(str, str)' =None, name: 'str' =None, private: 'bool' =None, project_id: 'str' =None, tags: 'list[str]' =None, updated_at: 'datetime' =None, user_id: 'str' =None):  # noqa: E501
+    def __init__(self, categories: 'list[str]' =None, created_at: 'datetime' =None, description: 'str' =None, downloads: 'str' =None, id: 'str' =None, latest_version: 'V1ModelVersionArchive' =None, license: 'str' =None, model_metadata: 'dict(str, str)' =None, name: 'str' =None, private: 'bool' =None, project_id: 'str' =None, tags: 'list[str]' =None, updated_at: 'datetime' =None, user_id: 'str' =None, visibility: 'V1ResourceVisibility' =None):  # noqa: E501
         """V1Model - a model defined in Swagger"""  # noqa: E501
         self._categories = None
         self._created_at = None
@@ -90,6 +92,7 @@ class V1Model(object):
         self._tags = None
         self._updated_at = None
         self._user_id = None
+        self._visibility = None
         self.discriminator = None
         if categories is not None:
             self.categories = categories
@@ -119,6 +122,8 @@ class V1Model(object):
             self.updated_at = updated_at
         if user_id is not None:
             self.user_id = user_id
+        if visibility is not None:
+            self.visibility = visibility
 
     @property
     def categories(self) -> 'list[str]':
@@ -414,6 +419,27 @@ class V1Model(object):
 
         self._user_id = user_id
 
+    @property
+    def visibility(self) -> 'V1ResourceVisibility':
+        """Gets the visibility of this V1Model.  # noqa: E501
+
+
+        :return: The visibility of this V1Model.  # noqa: E501
+        :rtype: V1ResourceVisibility
+        """
+        return self._visibility
+
+    @visibility.setter
+    def visibility(self, visibility: 'V1ResourceVisibility'):
+        """Sets the visibility of this V1Model.
+
+
+        :param visibility: The visibility of this V1Model.  # noqa: E501
+        :type: V1ResourceVisibility
+        """
+
+        self._visibility = visibility
+
     def to_dict(self) -> dict:
         """Returns the model properties as a dict"""
         result = {}

lightning_sdk/lightning_cloud/openapi/models/v1_multi_machine_job.py

@@ -50,6 +50,7 @@ class V1MultiMachineJob(object):
         'name': 'str',
         'project_id': 'str',
         'spec': 'V1JobSpec',
+        'state': 'V1MultiMachineJobState',
         'status': 'V1MultiMachineJobStatus',
         'updated_at': 'datetime',
         'user_id': 'str'
@@ -65,12 +66,13 @@ class V1MultiMachineJob(object):
         'name': 'name',
         'project_id': 'projectId',
         'spec': 'spec',
+        'state': 'state',
         'status': 'status',
         'updated_at': 'updatedAt',
         'user_id': 'userId'
     }
 
-    def __init__(self, cloudspace_id: 'str' =None, created_at: 'datetime' =None, desired_state: 'V1MultiMachineJobState' =None, id: 'str' =None, machines: 'int' =None, message: 'str' =None, name: 'str' =None, project_id: 'str' =None, spec: 'V1JobSpec' =None, status: 'V1MultiMachineJobStatus' =None, updated_at: 'datetime' =None, user_id: 'str' =None):  # noqa: E501
+    def __init__(self, cloudspace_id: 'str' =None, created_at: 'datetime' =None, desired_state: 'V1MultiMachineJobState' =None, id: 'str' =None, machines: 'int' =None, message: 'str' =None, name: 'str' =None, project_id: 'str' =None, spec: 'V1JobSpec' =None, state: 'V1MultiMachineJobState' =None, status: 'V1MultiMachineJobStatus' =None, updated_at: 'datetime' =None, user_id: 'str' =None):  # noqa: E501
         """V1MultiMachineJob - a model defined in Swagger"""  # noqa: E501
         self._cloudspace_id = None
         self._created_at = None
@@ -81,6 +83,7 @@ class V1MultiMachineJob(object):
         self._name = None
         self._project_id = None
         self._spec = None
+        self._state = None
         self._status = None
         self._updated_at = None
         self._user_id = None
@@ -103,6 +106,8 @@ class V1MultiMachineJob(object):
             self.project_id = project_id
         if spec is not None:
             self.spec = spec
+        if state is not None:
+            self.state = state
         if status is not None:
             self.status = status
         if updated_at is not None:
@@ -299,6 +304,27 @@ class V1MultiMachineJob(object):
 
         self._spec = spec
 
+    @property
+    def state(self) -> 'V1MultiMachineJobState':
+        """Gets the state of this V1MultiMachineJob.  # noqa: E501
+
+
+        :return: The state of this V1MultiMachineJob.  # noqa: E501
+        :rtype: V1MultiMachineJobState
+        """
+        return self._state
+
+    @state.setter
+    def state(self, state: 'V1MultiMachineJobState'):
+        """Sets the state of this V1MultiMachineJob.
+
+
+        :param state: The state of this V1MultiMachineJob.  # noqa: E501
+        :type: V1MultiMachineJobState
+        """
+
+        self._state = state
+
     @property
     def status(self) -> 'V1MultiMachineJobStatus':
         """Gets the status of this V1MultiMachineJob.  # noqa: E501

lightning_sdk/lightning_cloud/openapi/models/v1_multi_machine_job_state.py

@@ -42,6 +42,7 @@ class V1MultiMachineJobState(object):
     STOPPED = "MultiMachineJob_STATE_STOPPED"
     DELETED = "MultiMachineJob_STATE_DELETED"
     FAILED = "MultiMachineJob_STATE_FAILED"
+    COMPLETED = "MultiMachineJob_STATE_COMPLETED"
     """
     Attributes:
       swagger_types (dict): The key is attribute name