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

lightning_sdk/lightning_cloud/openapi/models/v1_get_model_files_url_response.py

@@ -0,0 +1,149 @@
+# 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 V1GetModelFilesUrlResponse(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 = {
+        'files': 'list[V1ModelFile]',
+        'offset': 'str'
+    }
+
+    attribute_map = {
+        'files': 'files',
+        'offset': 'offset'
+    }
+
+    def __init__(self, files: 'list[V1ModelFile]' =None, offset: 'str' =None):  # noqa: E501
+        """V1GetModelFilesUrlResponse - a model defined in Swagger"""  # noqa: E501
+        self._files = None
+        self._offset = None
+        self.discriminator = None
+        if files is not None:
+            self.files = files
+        if offset is not None:
+            self.offset = offset
+
+    @property
+    def files(self) -> 'list[V1ModelFile]':
+        """Gets the files of this V1GetModelFilesUrlResponse.  # noqa: E501
+
+
+        :return: The files of this V1GetModelFilesUrlResponse.  # noqa: E501
+        :rtype: list[V1ModelFile]
+        """
+        return self._files
+
+    @files.setter
+    def files(self, files: 'list[V1ModelFile]'):
+        """Sets the files of this V1GetModelFilesUrlResponse.
+
+
+        :param files: The files of this V1GetModelFilesUrlResponse.  # noqa: E501
+        :type: list[V1ModelFile]
+        """
+
+        self._files = files
+
+    @property
+    def offset(self) -> 'str':
+        """Gets the offset of this V1GetModelFilesUrlResponse.  # noqa: E501
+
+
+        :return: The offset of this V1GetModelFilesUrlResponse.  # noqa: E501
+        :rtype: str
+        """
+        return self._offset
+
+    @offset.setter
+    def offset(self, offset: 'str'):
+        """Sets the offset of this V1GetModelFilesUrlResponse.
+
+
+        :param offset: The offset of this V1GetModelFilesUrlResponse.  # noqa: E501
+        :type: str
+        """
+
+        self._offset = offset
+
+    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(V1GetModelFilesUrlResponse, 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: 'V1GetModelFilesUrlResponse') -> bool:
+        """Returns true if both objects are equal"""
+        if not isinstance(other, V1GetModelFilesUrlResponse):
+            return False
+
+        return self.__dict__ == other.__dict__
+
+    def __ne__(self, other: 'V1GetModelFilesUrlResponse') -> bool:
+        """Returns true if both objects are not equal"""
+        return not self == other

lightning_sdk/lightning_cloud/openapi/models/v1_get_project_balance_response.py

@@ -43,6 +43,7 @@ class V1GetProjectBalanceResponse(object):
     swagger_types = {
         'account_id': 'str',
         'balance': 'float',
+        'next_free_credits_grant_at': 'datetime',
         'project_id': 'str',
         'transactions': 'list[V1Transaction]'
     }
@@ -50,14 +51,16 @@ class V1GetProjectBalanceResponse(object):
     attribute_map = {
         'account_id': 'accountId',
         'balance': 'balance',
+        'next_free_credits_grant_at': 'nextFreeCreditsGrantAt',
         'project_id': 'projectId',
         'transactions': 'transactions'
     }
 
-    def __init__(self, account_id: 'str' =None, balance: 'float' =None, project_id: 'str' =None, transactions: 'list[V1Transaction]' =None):  # noqa: E501
+    def __init__(self, account_id: 'str' =None, balance: 'float' =None, next_free_credits_grant_at: 'datetime' =None, project_id: 'str' =None, transactions: 'list[V1Transaction]' =None):  # noqa: E501
         """V1GetProjectBalanceResponse - a model defined in Swagger"""  # noqa: E501
         self._account_id = None
         self._balance = None
+        self._next_free_credits_grant_at = None
         self._project_id = None
         self._transactions = None
         self.discriminator = None
@@ -65,6 +68,8 @@ class V1GetProjectBalanceResponse(object):
             self.account_id = account_id
         if balance is not None:
             self.balance = balance
+        if next_free_credits_grant_at is not None:
+            self.next_free_credits_grant_at = next_free_credits_grant_at
         if project_id is not None:
             self.project_id = project_id
         if transactions is not None:
@@ -112,6 +117,27 @@ class V1GetProjectBalanceResponse(object):
 
         self._balance = balance
 
+    @property
+    def next_free_credits_grant_at(self) -> 'datetime':
+        """Gets the next_free_credits_grant_at of this V1GetProjectBalanceResponse.  # noqa: E501
+
+
+        :return: The next_free_credits_grant_at of this V1GetProjectBalanceResponse.  # noqa: E501
+        :rtype: datetime
+        """
+        return self._next_free_credits_grant_at
+
+    @next_free_credits_grant_at.setter
+    def next_free_credits_grant_at(self, next_free_credits_grant_at: 'datetime'):
+        """Sets the next_free_credits_grant_at of this V1GetProjectBalanceResponse.
+
+
+        :param next_free_credits_grant_at: The next_free_credits_grant_at of this V1GetProjectBalanceResponse.  # noqa: E501
+        :type: datetime
+        """
+
+        self._next_free_credits_grant_at = next_free_credits_grant_at
+
     @property
     def project_id(self) -> 'str':
         """Gets the project_id of this V1GetProjectBalanceResponse.  # noqa: E501

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

lightning_sdk/lightning_cloud/openapi/models/v1_job_spec.py

@@ -57,6 +57,7 @@ class V1JobSpec(object):
         'instance_type': 'str',
         'modified_volume': 'bool',
         'readiness_probe': 'V1JobHealthCheckConfig',
+        'regions': 'list[str]',
         'resources': 'V1Resources',
         'restart_policy': 'str',
         'run_id': 'str',
@@ -81,6 +82,7 @@ class V1JobSpec(object):
         'instance_type': 'instanceType',
         'modified_volume': 'modifiedVolume',
         'readiness_probe': 'readinessProbe',
+        'regions': 'regions',
         'resources': 'resources',
         'restart_policy': 'restartPolicy',
         'run_id': 'runId',
@@ -88,7 +90,7 @@ class V1JobSpec(object):
         'volumes': 'volumes'
     }
 
-    def __init__(self, artifacts_destination: 'str' =None, artifacts_source: 'str' =None, cloudspace_id: 'str' =None, cluster_id: 'str' =None, command: 'str' =None, data_paths: 'list[V1DataPath]' =None, entrypoint: 'str' =None, env: 'list[V1EnvVar]' =None, image: 'str' =None, image_cluster_credentials: 'bool' =None, image_secret_ref: 'str' =None, include_credentials: 'bool' =None, instance_name: 'str' =None, instance_type: 'str' =None, modified_volume: 'bool' =None, readiness_probe: 'V1JobHealthCheckConfig' =None, resources: 'V1Resources' =None, restart_policy: 'str' =None, run_id: 'str' =None, spot: 'bool' =None, volumes: 'list[V1Volume]' =None):  # noqa: E501
+    def __init__(self, artifacts_destination: 'str' =None, artifacts_source: 'str' =None, cloudspace_id: 'str' =None, cluster_id: 'str' =None, command: 'str' =None, data_paths: 'list[V1DataPath]' =None, entrypoint: 'str' =None, env: 'list[V1EnvVar]' =None, image: 'str' =None, image_cluster_credentials: 'bool' =None, image_secret_ref: 'str' =None, include_credentials: 'bool' =None, instance_name: 'str' =None, instance_type: 'str' =None, modified_volume: 'bool' =None, readiness_probe: 'V1JobHealthCheckConfig' =None, regions: 'list[str]' =None, resources: 'V1Resources' =None, restart_policy: 'str' =None, run_id: 'str' =None, spot: 'bool' =None, volumes: 'list[V1Volume]' =None):  # noqa: E501
         """V1JobSpec - a model defined in Swagger"""  # noqa: E501
         self._artifacts_destination = None
         self._artifacts_source = None
@@ -106,6 +108,7 @@ class V1JobSpec(object):
         self._instance_type = None
         self._modified_volume = None
         self._readiness_probe = None
+        self._regions = None
         self._resources = None
         self._restart_policy = None
         self._run_id = None
@@ -144,6 +147,8 @@ class V1JobSpec(object):
             self.modified_volume = modified_volume
         if readiness_probe is not None:
             self.readiness_probe = readiness_probe
+        if regions is not None:
+            self.regions = regions
         if resources is not None:
             self.resources = resources
         if restart_policy is not None:
@@ -491,6 +496,27 @@ class V1JobSpec(object):
 
         self._readiness_probe = readiness_probe
 
+    @property
+    def regions(self) -> 'list[str]':
+        """Gets the regions of this V1JobSpec.  # noqa: E501
+
+
+        :return: The regions of this V1JobSpec.  # noqa: E501
+        :rtype: list[str]
+        """
+        return self._regions
+
+    @regions.setter
+    def regions(self, regions: 'list[str]'):
+        """Sets the regions of this V1JobSpec.
+
+
+        :param regions: The regions of this V1JobSpec.  # noqa: E501
+        :type: list[str]
+        """
+
+        self._regions = regions
+
     @property
     def resources(self) -> 'V1Resources':
         """Gets the resources of this V1JobSpec.  # noqa: E501

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

lightning_sdk/lightning_cloud/openapi/models/v1_managed_model.py

@@ -47,7 +47,8 @@ class V1ManagedModel(object):
         'endpoint_id': 'str',
         'id': 'str',
         'name': 'str',
-        'prompt_token_price': 'float'
+        'prompt_token_price': 'float',
+        'status': 'V1AssistantModelStatus'
     }
 
     attribute_map = {
@@ -57,10 +58,11 @@ class V1ManagedModel(object):
         'endpoint_id': 'endpointId',
         'id': 'id',
         'name': 'name',
-        'prompt_token_price': 'promptTokenPrice'
+        'prompt_token_price': 'promptTokenPrice',
+        'status': 'status'
     }
 
-    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
+    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, status: 'V1AssistantModelStatus' =None):  # noqa: E501
         """V1ManagedModel - a model defined in Swagger"""  # noqa: E501
         self._abilities = None
         self._completion_token_price = None
@@ -69,6 +71,7 @@ class V1ManagedModel(object):
         self._id = None
         self._name = None
         self._prompt_token_price = None
+        self._status = None
         self.discriminator = None
         if abilities is not None:
             self.abilities = abilities
@@ -84,6 +87,8 @@ class V1ManagedModel(object):
             self.name = name
         if prompt_token_price is not None:
             self.prompt_token_price = prompt_token_price
+        if status is not None:
+            self.status = status
 
     @property
     def abilities(self) -> 'V1ManagedModelAbilities':
@@ -232,6 +237,27 @@ class V1ManagedModel(object):
 
         self._prompt_token_price = prompt_token_price
 
+    @property
+    def status(self) -> 'V1AssistantModelStatus':
+        """Gets the status of this V1ManagedModel.  # noqa: E501
+
+
+        :return: The status of this V1ManagedModel.  # noqa: E501
+        :rtype: V1AssistantModelStatus
+        """
+        return self._status
+
+    @status.setter
+    def status(self, status: 'V1AssistantModelStatus'):
+        """Sets the status of this V1ManagedModel.
+
+
+        :param status: The status of this V1ManagedModel.  # noqa: E501
+        :type: V1AssistantModelStatus
+        """
+
+        self._status = status
+
     def to_dict(self) -> dict:
         """Returns the model properties as a dict"""
         result = {}