lightning-sdk 0.1.2__py3-none-any.whl → 0.1.46__py3-none-any.whl

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

lightning_sdk/lightning_cloud/openapi/models/v1_message.py

@@ -41,39 +41,85 @@ class V1Message(object):
                             and the value is json key in definition.
     """
     swagger_types = {
+        'assistant_id': 'str',
         'author': 'V1MessageAuthor',
-        'content': 'V1MessageContent',
+        'completion_tokens': 'str',
+        'content': 'list[V1MessageContent]',
         'conversation_id': 'str',
         'created_at': 'datetime',
-        'id': 'str'
+        'executable': 'bool',
+        'id': 'str',
+        'model': 'str',
+        'prompt_tokens': 'str'
     }
 
     attribute_map = {
+        'assistant_id': 'assistantId',
         'author': 'author',
+        'completion_tokens': 'completionTokens',
         'content': 'content',
         'conversation_id': 'conversationId',
         'created_at': 'createdAt',
-        'id': 'id'
+        'executable': 'executable',
+        'id': 'id',
+        'model': 'model',
+        'prompt_tokens': 'promptTokens'
     }
 
-    def __init__(self, author: 'V1MessageAuthor' =None, content: 'V1MessageContent' =None, conversation_id: 'str' =None, created_at: 'datetime' =None, id: 'str' =None):  # noqa: E501
+    def __init__(self, assistant_id: 'str' =None, author: 'V1MessageAuthor' =None, completion_tokens: 'str' =None, content: 'list[V1MessageContent]' =None, conversation_id: 'str' =None, created_at: 'datetime' =None, executable: 'bool' =None, id: 'str' =None, model: 'str' =None, prompt_tokens: 'str' =None):  # noqa: E501
         """V1Message - a model defined in Swagger"""  # noqa: E501
+        self._assistant_id = None
         self._author = None
+        self._completion_tokens = None
         self._content = None
         self._conversation_id = None
         self._created_at = None
+        self._executable = None
         self._id = None
+        self._model = None
+        self._prompt_tokens = None
         self.discriminator = None
+        if assistant_id is not None:
+            self.assistant_id = assistant_id
         if author is not None:
             self.author = author
+        if completion_tokens is not None:
+            self.completion_tokens = completion_tokens
         if content is not None:
             self.content = content
         if conversation_id is not None:
             self.conversation_id = conversation_id
         if created_at is not None:
             self.created_at = created_at
+        if executable is not None:
+            self.executable = executable
         if id is not None:
             self.id = id
+        if model is not None:
+            self.model = model
+        if prompt_tokens is not None:
+            self.prompt_tokens = prompt_tokens
+
+    @property
+    def assistant_id(self) -> 'str':
+        """Gets the assistant_id of this V1Message.  # noqa: E501
+
+
+        :return: The assistant_id of this V1Message.  # noqa: E501
+        :rtype: str
+        """
+        return self._assistant_id
+
+    @assistant_id.setter
+    def assistant_id(self, assistant_id: 'str'):
+        """Sets the assistant_id of this V1Message.
+
+
+        :param assistant_id: The assistant_id of this V1Message.  # noqa: E501
+        :type: str
+        """
+
+        self._assistant_id = assistant_id
 
     @property
     def author(self) -> 'V1MessageAuthor':
@@ -97,22 +143,43 @@ class V1Message(object):
         self._author = author
 
     @property
-    def content(self) -> 'V1MessageContent':
+    def completion_tokens(self) -> 'str':
+        """Gets the completion_tokens of this V1Message.  # noqa: E501
+
+
+        :return: The completion_tokens of this V1Message.  # noqa: E501
+        :rtype: str
+        """
+        return self._completion_tokens
+
+    @completion_tokens.setter
+    def completion_tokens(self, completion_tokens: 'str'):
+        """Sets the completion_tokens of this V1Message.
+
+
+        :param completion_tokens: The completion_tokens of this V1Message.  # noqa: E501
+        :type: str
+        """
+
+        self._completion_tokens = completion_tokens
+
+    @property
+    def content(self) -> 'list[V1MessageContent]':
         """Gets the content of this V1Message.  # noqa: E501
 
 
         :return: The content of this V1Message.  # noqa: E501
-        :rtype: V1MessageContent
+        :rtype: list[V1MessageContent]
         """
         return self._content
 
     @content.setter
-    def content(self, content: 'V1MessageContent'):
+    def content(self, content: 'list[V1MessageContent]'):
         """Sets the content of this V1Message.
 
 
         :param content: The content of this V1Message.  # noqa: E501
-        :type: V1MessageContent
+        :type: list[V1MessageContent]
         """
 
         self._content = content
@@ -159,6 +226,27 @@ class V1Message(object):
 
         self._created_at = created_at
 
+    @property
+    def executable(self) -> 'bool':
+        """Gets the executable of this V1Message.  # noqa: E501
+
+
+        :return: The executable of this V1Message.  # noqa: E501
+        :rtype: bool
+        """
+        return self._executable
+
+    @executable.setter
+    def executable(self, executable: 'bool'):
+        """Sets the executable of this V1Message.
+
+
+        :param executable: The executable of this V1Message.  # noqa: E501
+        :type: bool
+        """
+
+        self._executable = executable
+
     @property
     def id(self) -> 'str':
         """Gets the id of this V1Message.  # noqa: E501
@@ -180,6 +268,48 @@ class V1Message(object):
 
         self._id = id
 
+    @property
+    def model(self) -> 'str':
+        """Gets the model of this V1Message.  # noqa: E501
+
+
+        :return: The model of this V1Message.  # noqa: E501
+        :rtype: str
+        """
+        return self._model
+
+    @model.setter
+    def model(self, model: 'str'):
+        """Sets the model of this V1Message.
+
+
+        :param model: The model of this V1Message.  # noqa: E501
+        :type: str
+        """
+
+        self._model = model
+
+    @property
+    def prompt_tokens(self) -> 'str':
+        """Gets the prompt_tokens of this V1Message.  # noqa: E501
+
+
+        :return: The prompt_tokens of this V1Message.  # noqa: E501
+        :rtype: str
+        """
+        return self._prompt_tokens
+
+    @prompt_tokens.setter
+    def prompt_tokens(self, prompt_tokens: 'str'):
+        """Sets the prompt_tokens of this V1Message.
+
+
+        :param prompt_tokens: The prompt_tokens of this V1Message.  # noqa: E501
+        :type: str
+        """
+
+        self._prompt_tokens = prompt_tokens
+
     def to_dict(self) -> dict:
         """Returns the model properties as a dict"""
         result = {}

lightning_sdk/lightning_cloud/openapi/models/v1_message_content.py

@@ -41,7 +41,7 @@ class V1MessageContent(object):
                             and the value is json key in definition.
     """
     swagger_types = {
-        'content_type': 'str',
+        'content_type': 'V1MessageContentType',
         'parts': 'list[str]'
     }
 
@@ -50,7 +50,7 @@ class V1MessageContent(object):
         'parts': 'parts'
     }
 
-    def __init__(self, content_type: 'str' =None, parts: 'list[str]' =None):  # noqa: E501
+    def __init__(self, content_type: 'V1MessageContentType' =None, parts: 'list[str]' =None):  # noqa: E501
         """V1MessageContent - a model defined in Swagger"""  # noqa: E501
         self._content_type = None
         self._parts = None
@@ -61,22 +61,22 @@ class V1MessageContent(object):
             self.parts = parts
 
     @property
-    def content_type(self) -> 'str':
+    def content_type(self) -> 'V1MessageContentType':
         """Gets the content_type of this V1MessageContent.  # noqa: E501
 
 
         :return: The content_type of this V1MessageContent.  # noqa: E501
-        :rtype: str
+        :rtype: V1MessageContentType
         """
         return self._content_type
 
     @content_type.setter
-    def content_type(self, content_type: 'str'):
+    def content_type(self, content_type: 'V1MessageContentType'):
         """Sets the content_type of this V1MessageContent.
 
 
         :param content_type: The content_type of this V1MessageContent.  # noqa: E501
-        :type: str
+        :type: V1MessageContentType
         """
 
         self._content_type = content_type