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

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

lightning_sdk/lightning_cloud/openapi/models/v1_message_content_type.py

@@ -0,0 +1,103 @@
+# 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 V1MessageContentType(object):
+    """NOTE: This class is auto generated by the swagger code generator program.
+
+    Do not edit the class manually.
+    """
+
+    """
+    allowed enum values
+    """
+    TEXT = "text"
+    IMAGE = "image"
+    """
+    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 = {
+    }
+
+    attribute_map = {
+    }
+
+    def __init__(self):  # noqa: E501
+        """V1MessageContentType - a model defined in Swagger"""  # noqa: E501
+        self.discriminator = None
+
+    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(V1MessageContentType, 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: 'V1MessageContentType') -> bool:
+        """Returns true if both objects are equal"""
+        if not isinstance(other, V1MessageContentType):
+            return False
+
+        return self.__dict__ == other.__dict__
+
+    def __ne__(self, other: 'V1MessageContentType') -> bool:
+        """Returns true if both objects are not equal"""
+        return not self == other

lightning_sdk/lightning_cloud/openapi/models/v1_metadata.py

@@ -42,14 +42,19 @@ class V1Metadata(object):
     """
     swagger_types = {
         'annotations': 'dict(str, str)',
+        'cluster_id': 'str',
         'creation_timestamp': 'datetime',
         'deletion_timestamp': 'datetime',
+        'deployment_id': 'str',
         'display_name': 'str',
         'finalizers': 'list[str]',
         'id': 'str',
         'labels': 'dict(str, str)',
         'last_updated_by_user_at': 'datetime',
+        'linked_project_id': 'str',
+        'linked_user_id': 'str',
         'name': 'str',
+        'org_id': 'str',
         'project_id': 'str',
         'resource_version': 'str',
         'updated_at': 'datetime'
@@ -57,40 +62,54 @@ class V1Metadata(object):
 
     attribute_map = {
         'annotations': 'annotations',
+        'cluster_id': 'clusterId',
         'creation_timestamp': 'creationTimestamp',
         'deletion_timestamp': 'deletionTimestamp',
+        'deployment_id': 'deploymentId',
         'display_name': 'displayName',
         'finalizers': 'finalizers',
         'id': 'id',
         'labels': 'labels',
         'last_updated_by_user_at': 'lastUpdatedByUserAt',
+        'linked_project_id': 'linkedProjectId',
+        'linked_user_id': 'linkedUserId',
         'name': 'name',
+        'org_id': 'orgId',
         'project_id': 'projectId',
         'resource_version': 'resourceVersion',
         'updated_at': 'updatedAt'
     }
 
-    def __init__(self, annotations: 'dict(str, str)' =None, creation_timestamp: 'datetime' =None, deletion_timestamp: 'datetime' =None, display_name: 'str' =None, finalizers: 'list[str]' =None, id: 'str' =None, labels: 'dict(str, str)' =None, last_updated_by_user_at: 'datetime' =None, name: 'str' =None, project_id: 'str' =None, resource_version: 'str' =None, updated_at: 'datetime' =None):  # noqa: E501
+    def __init__(self, annotations: 'dict(str, str)' =None, cluster_id: 'str' =None, creation_timestamp: 'datetime' =None, deletion_timestamp: 'datetime' =None, deployment_id: 'str' =None, display_name: 'str' =None, finalizers: 'list[str]' =None, id: 'str' =None, labels: 'dict(str, str)' =None, last_updated_by_user_at: 'datetime' =None, linked_project_id: 'str' =None, linked_user_id: 'str' =None, name: 'str' =None, org_id: 'str' =None, project_id: 'str' =None, resource_version: 'str' =None, updated_at: 'datetime' =None):  # noqa: E501
         """V1Metadata - a model defined in Swagger"""  # noqa: E501
         self._annotations = None
+        self._cluster_id = None
         self._creation_timestamp = None
         self._deletion_timestamp = None
+        self._deployment_id = None
         self._display_name = None
         self._finalizers = None
         self._id = None
         self._labels = None
         self._last_updated_by_user_at = None
+        self._linked_project_id = None
+        self._linked_user_id = None
         self._name = None
+        self._org_id = None
         self._project_id = None
         self._resource_version = None
         self._updated_at = None
         self.discriminator = None
         if annotations is not None:
             self.annotations = annotations
+        if cluster_id is not None:
+            self.cluster_id = cluster_id
         if creation_timestamp is not None:
             self.creation_timestamp = creation_timestamp
         if deletion_timestamp is not None:
             self.deletion_timestamp = deletion_timestamp
+        if deployment_id is not None:
+            self.deployment_id = deployment_id
         if display_name is not None:
             self.display_name = display_name
         if finalizers is not None:
@@ -101,8 +120,14 @@ class V1Metadata(object):
             self.labels = labels
         if last_updated_by_user_at is not None:
             self.last_updated_by_user_at = last_updated_by_user_at
+        if linked_project_id is not None:
+            self.linked_project_id = linked_project_id
+        if linked_user_id is not None:
+            self.linked_user_id = linked_user_id
         if name is not None:
             self.name = name
+        if org_id is not None:
+            self.org_id = org_id
         if project_id is not None:
             self.project_id = project_id
         if resource_version is not None:
@@ -131,6 +156,27 @@ class V1Metadata(object):
 
         self._annotations = annotations
 
+    @property
+    def cluster_id(self) -> 'str':
+        """Gets the cluster_id of this V1Metadata.  # noqa: E501
+
+
+        :return: The cluster_id of this V1Metadata.  # noqa: E501
+        :rtype: str
+        """
+        return self._cluster_id
+
+    @cluster_id.setter
+    def cluster_id(self, cluster_id: 'str'):
+        """Sets the cluster_id of this V1Metadata.
+
+
+        :param cluster_id: The cluster_id of this V1Metadata.  # noqa: E501
+        :type: str
+        """
+
+        self._cluster_id = cluster_id
+
     @property
     def creation_timestamp(self) -> 'datetime':
         """Gets the creation_timestamp of this V1Metadata.  # noqa: E501
@@ -173,6 +219,27 @@ class V1Metadata(object):
 
         self._deletion_timestamp = deletion_timestamp
 
+    @property
+    def deployment_id(self) -> 'str':
+        """Gets the deployment_id of this V1Metadata.  # noqa: E501
+
+
+        :return: The deployment_id of this V1Metadata.  # noqa: E501
+        :rtype: str
+        """
+        return self._deployment_id
+
+    @deployment_id.setter
+    def deployment_id(self, deployment_id: 'str'):
+        """Sets the deployment_id of this V1Metadata.
+
+
+        :param deployment_id: The deployment_id of this V1Metadata.  # noqa: E501
+        :type: str
+        """
+
+        self._deployment_id = deployment_id
+
     @property
     def display_name(self) -> 'str':
         """Gets the display_name of this V1Metadata.  # noqa: E501
@@ -280,6 +347,48 @@ class V1Metadata(object):
 
         self._last_updated_by_user_at = last_updated_by_user_at
 
+    @property
+    def linked_project_id(self) -> 'str':
+        """Gets the linked_project_id of this V1Metadata.  # noqa: E501
+
+
+        :return: The linked_project_id of this V1Metadata.  # noqa: E501
+        :rtype: str
+        """
+        return self._linked_project_id
+
+    @linked_project_id.setter
+    def linked_project_id(self, linked_project_id: 'str'):
+        """Sets the linked_project_id of this V1Metadata.
+
+
+        :param linked_project_id: The linked_project_id of this V1Metadata.  # noqa: E501
+        :type: str
+        """
+
+        self._linked_project_id = linked_project_id
+
+    @property
+    def linked_user_id(self) -> 'str':
+        """Gets the linked_user_id of this V1Metadata.  # noqa: E501
+
+
+        :return: The linked_user_id of this V1Metadata.  # noqa: E501
+        :rtype: str
+        """
+        return self._linked_user_id
+
+    @linked_user_id.setter
+    def linked_user_id(self, linked_user_id: 'str'):
+        """Sets the linked_user_id of this V1Metadata.
+
+
+        :param linked_user_id: The linked_user_id of this V1Metadata.  # noqa: E501
+        :type: str
+        """
+
+        self._linked_user_id = linked_user_id
+
     @property
     def name(self) -> 'str':
         """Gets the name of this V1Metadata.  # noqa: E501
@@ -301,6 +410,27 @@ class V1Metadata(object):
 
         self._name = name
 
+    @property
+    def org_id(self) -> 'str':
+        """Gets the org_id of this V1Metadata.  # noqa: E501
+
+
+        :return: The org_id of this V1Metadata.  # noqa: E501
+        :rtype: str
+        """
+        return self._org_id
+
+    @org_id.setter
+    def org_id(self, org_id: 'str'):
+        """Sets the org_id of this V1Metadata.
+
+
+        :param org_id: The org_id of this V1Metadata.  # noqa: E501
+        :type: str
+        """
+
+        self._org_id = org_id
+
     @property
     def project_id(self) -> 'str':
         """Gets the project_id of this V1Metadata.  # noqa: E501