lightning-sdk 0.1.3__py3-none-any.whl → 0.1.47__py3-none-any.whl

lightning_sdk/lightning_cloud/openapi/models/v1_managed_model.py

@@ -0,0 +1,305 @@
+# 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 V1ManagedModel(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 = {
+        'abilities': 'V1ManagedModelAbilities',
+        'completion_token_price': 'float',
+        'description': 'str',
+        'endpoint_id': 'str',
+        'id': 'str',
+        'name': 'str',
+        'prompt_token_price': 'float',
+        'status': 'V1AssistantModelStatus'
+    }
+
+    attribute_map = {
+        'abilities': 'abilities',
+        'completion_token_price': 'completionTokenPrice',
+        'description': 'description',
+        'endpoint_id': 'endpointId',
+        'id': 'id',
+        'name': 'name',
+        '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, status: 'V1AssistantModelStatus' =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
+        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
+        if completion_token_price is not None:
+            self.completion_token_price = completion_token_price
+        if description is not None:
+            self.description = description
+        if endpoint_id is not None:
+            self.endpoint_id = endpoint_id
+        if id is not None:
+            self.id = id
+        if name is not None:
+            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':
+        """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
+
+
+        :return: The completion_token_price of this V1ManagedModel.  # noqa: E501
+        :rtype: float
+        """
+        return self._completion_token_price
+
+    @completion_token_price.setter
+    def completion_token_price(self, completion_token_price: 'float'):
+        """Sets the completion_token_price of this V1ManagedModel.
+
+
+        :param completion_token_price: The completion_token_price of this V1ManagedModel.  # noqa: E501
+        :type: float
+        """
+
+        self._completion_token_price = completion_token_price
+
+    @property
+    def description(self) -> 'str':
+        """Gets the description of this V1ManagedModel.  # noqa: E501
+
+
+        :return: The description of this V1ManagedModel.  # noqa: E501
+        :rtype: str
+        """
+        return self._description
+
+    @description.setter
+    def description(self, description: 'str'):
+        """Sets the description of this V1ManagedModel.
+
+
+        :param description: The description of this V1ManagedModel.  # noqa: E501
+        :type: str
+        """
+
+        self._description = description
+
+    @property
+    def endpoint_id(self) -> 'str':
+        """Gets the endpoint_id of this V1ManagedModel.  # noqa: E501
+
+
+        :return: The endpoint_id of this V1ManagedModel.  # noqa: E501
+        :rtype: str
+        """
+        return self._endpoint_id
+
+    @endpoint_id.setter
+    def endpoint_id(self, endpoint_id: 'str'):
+        """Sets the endpoint_id of this V1ManagedModel.
+
+
+        :param endpoint_id: The endpoint_id of this V1ManagedModel.  # noqa: E501
+        :type: str
+        """
+
+        self._endpoint_id = endpoint_id
+
+    @property
+    def id(self) -> 'str':
+        """Gets the id of this V1ManagedModel.  # noqa: E501
+
+
+        :return: The id of this V1ManagedModel.  # noqa: E501
+        :rtype: str
+        """
+        return self._id
+
+    @id.setter
+    def id(self, id: 'str'):
+        """Sets the id of this V1ManagedModel.
+
+
+        :param id: The id of this V1ManagedModel.  # noqa: E501
+        :type: str
+        """
+
+        self._id = id
+
+    @property
+    def name(self) -> 'str':
+        """Gets the name of this V1ManagedModel.  # noqa: E501
+
+
+        :return: The name of this V1ManagedModel.  # noqa: E501
+        :rtype: str
+        """
+        return self._name
+
+    @name.setter
+    def name(self, name: 'str'):
+        """Sets the name of this V1ManagedModel.
+
+
+        :param name: The name of this V1ManagedModel.  # noqa: E501
+        :type: str
+        """
+
+        self._name = name
+
+    @property
+    def prompt_token_price(self) -> 'float':
+        """Gets the prompt_token_price of this V1ManagedModel.  # noqa: E501
+
+
+        :return: The prompt_token_price of this V1ManagedModel.  # noqa: E501
+        :rtype: float
+        """
+        return self._prompt_token_price
+
+    @prompt_token_price.setter
+    def prompt_token_price(self, prompt_token_price: 'float'):
+        """Sets the prompt_token_price of this V1ManagedModel.
+
+
+        :param prompt_token_price: The prompt_token_price of this V1ManagedModel.  # noqa: E501
+        :type: float
+        """
+
+        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 = {}
+
+        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(V1ManagedModel, 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: 'V1ManagedModel') -> bool:
+        """Returns true if both objects are equal"""
+        if not isinstance(other, V1ManagedModel):
+            return False
+
+        return self.__dict__ == other.__dict__
+
+    def __ne__(self, other: 'V1ManagedModel') -> bool:
+        """Returns true if both objects are not equal"""
+        return not self == other

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_membership.py

@@ -43,13 +43,16 @@ class V1Membership(object):
     swagger_types = {
         'balance': 'float',
         'created_at': 'datetime',
+        'creator_id': 'str',
         'datastore_count': 'str',
         'description': 'str',
         'display_name': 'str',
         'free_credits_enabled': 'bool',
+        'is_default': 'bool',
         'job_count': 'str',
         'membership_count': 'str',
         'name': 'str',
+        'next_free_credits_grant_at': 'datetime',
         'owner_id': 'str',
         'owner_type': 'V1OwnerType',
         'project_id': 'str',
@@ -62,13 +65,16 @@ class V1Membership(object):
     attribute_map = {
         'balance': 'balance',
         'created_at': 'createdAt',
+        'creator_id': 'creatorId',
         'datastore_count': 'datastoreCount',
         'description': 'description',
         'display_name': 'displayName',
         'free_credits_enabled': 'freeCreditsEnabled',
+        'is_default': 'isDefault',
         'job_count': 'jobCount',
         'membership_count': 'membershipCount',
         'name': 'name',
+        'next_free_credits_grant_at': 'nextFreeCreditsGrantAt',
         'owner_id': 'ownerId',
         'owner_type': 'ownerType',
         'project_id': 'projectId',
@@ -78,17 +84,20 @@ class V1Membership(object):
         'user_id': 'userId'
     }
 
-    def __init__(self, balance: 'float' =None, created_at: 'datetime' =None, datastore_count: 'str' =None, description: 'str' =None, display_name: 'str' =None, free_credits_enabled: 'bool' =None, job_count: 'str' =None, membership_count: 'str' =None, name: 'str' =None, owner_id: 'str' =None, owner_type: 'V1OwnerType' =None, project_id: 'str' =None, quotas: 'V1Quotas' =None, roles: 'list[V1Role]' =None, updated_at: 'datetime' =None, user_id: 'str' =None):  # noqa: E501
+    def __init__(self, balance: 'float' =None, created_at: 'datetime' =None, creator_id: 'str' =None, datastore_count: 'str' =None, description: 'str' =None, display_name: 'str' =None, free_credits_enabled: 'bool' =None, is_default: 'bool' =None, job_count: 'str' =None, membership_count: 'str' =None, name: 'str' =None, next_free_credits_grant_at: 'datetime' =None, owner_id: 'str' =None, owner_type: 'V1OwnerType' =None, project_id: 'str' =None, quotas: 'V1Quotas' =None, roles: 'list[V1Role]' =None, updated_at: 'datetime' =None, user_id: 'str' =None):  # noqa: E501
         """V1Membership - a model defined in Swagger"""  # noqa: E501
         self._balance = None
         self._created_at = None
+        self._creator_id = None
         self._datastore_count = None
         self._description = None
         self._display_name = None
         self._free_credits_enabled = None
+        self._is_default = None
         self._job_count = None
         self._membership_count = None
         self._name = None
+        self._next_free_credits_grant_at = None
         self._owner_id = None
         self._owner_type = None
         self._project_id = None
@@ -101,6 +110,8 @@ class V1Membership(object):
             self.balance = balance
         if created_at is not None:
             self.created_at = created_at
+        if creator_id is not None:
+            self.creator_id = creator_id
         if datastore_count is not None:
             self.datastore_count = datastore_count
         if description is not None:
@@ -109,12 +120,16 @@ class V1Membership(object):
             self.display_name = display_name
         if free_credits_enabled is not None:
             self.free_credits_enabled = free_credits_enabled
+        if is_default is not None:
+            self.is_default = is_default
         if job_count is not None:
             self.job_count = job_count
         if membership_count is not None:
             self.membership_count = membership_count
         if name is not None:
             self.name = name
+        if next_free_credits_grant_at is not None:
+            self.next_free_credits_grant_at = next_free_credits_grant_at
         if owner_id is not None:
             self.owner_id = owner_id
         if owner_type is not None:
@@ -172,6 +187,27 @@ class V1Membership(object):
 
         self._created_at = created_at
 
+    @property
+    def creator_id(self) -> 'str':
+        """Gets the creator_id of this V1Membership.  # noqa: E501
+
+
+        :return: The creator_id of this V1Membership.  # noqa: E501
+        :rtype: str
+        """
+        return self._creator_id
+
+    @creator_id.setter
+    def creator_id(self, creator_id: 'str'):
+        """Sets the creator_id of this V1Membership.
+
+
+        :param creator_id: The creator_id of this V1Membership.  # noqa: E501
+        :type: str
+        """
+
+        self._creator_id = creator_id
+
     @property
     def datastore_count(self) -> 'str':
         """Gets the datastore_count of this V1Membership.  # noqa: E501
@@ -256,6 +292,27 @@ class V1Membership(object):
 
         self._free_credits_enabled = free_credits_enabled
 
+    @property
+    def is_default(self) -> 'bool':
+        """Gets the is_default of this V1Membership.  # noqa: E501
+
+
+        :return: The is_default of this V1Membership.  # noqa: E501
+        :rtype: bool
+        """
+        return self._is_default
+
+    @is_default.setter
+    def is_default(self, is_default: 'bool'):
+        """Sets the is_default of this V1Membership.
+
+
+        :param is_default: The is_default of this V1Membership.  # noqa: E501
+        :type: bool
+        """
+
+        self._is_default = is_default
+
     @property
     def job_count(self) -> 'str':
         """Gets the job_count of this V1Membership.  # noqa: E501
@@ -319,6 +376,27 @@ class V1Membership(object):
 
         self._name = name
 
+    @property
+    def next_free_credits_grant_at(self) -> 'datetime':
+        """Gets the next_free_credits_grant_at of this V1Membership.  # noqa: E501
+
+
+        :return: The next_free_credits_grant_at of this V1Membership.  # 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 V1Membership.
+
+
+        :param next_free_credits_grant_at: The next_free_credits_grant_at of this V1Membership.  # noqa: E501
+        :type: datetime
+        """
+
+        self._next_free_credits_grant_at = next_free_credits_grant_at
+
     @property
     def owner_id(self) -> 'str':
         """Gets the owner_id of this V1Membership.  # noqa: E501