lightning-sdk 0.1.36__py3-none-any.whl → 0.1.38__py3-none-any.whl

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

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

lightning_sdk/lightning_cloud/openapi/models/v1_message.py

@@ -44,7 +44,7 @@ class V1Message(object):
         'assistant_id': 'str',
         'author': 'V1MessageAuthor',
         'completion_tokens': 'str',
-        'content': 'V1MessageContent',
+        'content': 'list[V1MessageContent]',
         'conversation_id': 'str',
         'created_at': 'datetime',
         'executable': 'bool',
@@ -66,7 +66,7 @@ class V1Message(object):
         'prompt_tokens': 'promptTokens'
     }
 
-    def __init__(self, assistant_id: 'str' =None, author: 'V1MessageAuthor' =None, completion_tokens: 'str' =None, content: '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
+    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
@@ -164,22 +164,22 @@ class V1Message(object):
         self._completion_tokens = completion_tokens
 
     @property
-    def content(self) -> 'V1MessageContent':
+    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

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

@@ -61,6 +61,8 @@ class V1MetricsStream(object):
         'system_info': 'V1SystemInfo',
         'tags': 'list[V1MetricsTags]',
         'trackers': 'dict(str, V1MetricsTracker)',
+        'updated_at': 'datetime',
+        'updates_counter': 'int',
         'user_id': 'str',
         'version': 'datetime',
         'version_number': 'int',
@@ -88,13 +90,15 @@ class V1MetricsStream(object):
         'system_info': 'systemInfo',
         'tags': 'tags',
         'trackers': 'trackers',
+        'updated_at': 'updatedAt',
+        'updates_counter': 'updatesCounter',
         'user_id': 'userId',
         'version': 'version',
         'version_number': 'versionNumber',
         'work_id': 'workId'
     }
 
-    def __init__(self, app_id: 'str' =None, cloudspace_id: 'str' =None, cluster_id: 'str' =None, created_at: 'datetime' =None, dark_color: 'str' =None, id: 'str' =None, job_name: 'str' =None, light_color: 'str' =None, name: 'str' =None, persisted: 'bool' =None, phase: 'V1PhaseType' =None, plugin_id: 'str' =None, project_id: 'str' =None, source_id: 'str' =None, stopped_at: 'datetime' =None, store_created_at: 'bool' =None, store_step: 'bool' =None, system_info: 'V1SystemInfo' =None, tags: 'list[V1MetricsTags]' =None, trackers: 'dict(str, V1MetricsTracker)' =None, user_id: 'str' =None, version: 'datetime' =None, version_number: 'int' =None, work_id: 'str' =None):  # noqa: E501
+    def __init__(self, app_id: 'str' =None, cloudspace_id: 'str' =None, cluster_id: 'str' =None, created_at: 'datetime' =None, dark_color: 'str' =None, id: 'str' =None, job_name: 'str' =None, light_color: 'str' =None, name: 'str' =None, persisted: 'bool' =None, phase: 'V1PhaseType' =None, plugin_id: 'str' =None, project_id: 'str' =None, source_id: 'str' =None, stopped_at: 'datetime' =None, store_created_at: 'bool' =None, store_step: 'bool' =None, system_info: 'V1SystemInfo' =None, tags: 'list[V1MetricsTags]' =None, trackers: 'dict(str, V1MetricsTracker)' =None, updated_at: 'datetime' =None, updates_counter: 'int' =None, user_id: 'str' =None, version: 'datetime' =None, version_number: 'int' =None, work_id: 'str' =None):  # noqa: E501
         """V1MetricsStream - a model defined in Swagger"""  # noqa: E501
         self._app_id = None
         self._cloudspace_id = None
@@ -116,6 +120,8 @@ class V1MetricsStream(object):
         self._system_info = None
         self._tags = None
         self._trackers = None
+        self._updated_at = None
+        self._updates_counter = None
         self._user_id = None
         self._version = None
         self._version_number = None
@@ -161,6 +167,10 @@ class V1MetricsStream(object):
             self.tags = tags
         if trackers is not None:
             self.trackers = trackers
+        if updated_at is not None:
+            self.updated_at = updated_at
+        if updates_counter is not None:
+            self.updates_counter = updates_counter
         if user_id is not None:
             self.user_id = user_id
         if version is not None:
@@ -590,6 +600,48 @@ class V1MetricsStream(object):
 
         self._trackers = trackers
 
+    @property
+    def updated_at(self) -> 'datetime':
+        """Gets the updated_at of this V1MetricsStream.  # noqa: E501
+
+
+        :return: The updated_at of this V1MetricsStream.  # noqa: E501
+        :rtype: datetime
+        """
+        return self._updated_at
+
+    @updated_at.setter
+    def updated_at(self, updated_at: 'datetime'):
+        """Sets the updated_at of this V1MetricsStream.
+
+
+        :param updated_at: The updated_at of this V1MetricsStream.  # noqa: E501
+        :type: datetime
+        """
+
+        self._updated_at = updated_at
+
+    @property
+    def updates_counter(self) -> 'int':
+        """Gets the updates_counter of this V1MetricsStream.  # noqa: E501
+
+
+        :return: The updates_counter of this V1MetricsStream.  # noqa: E501
+        :rtype: int
+        """
+        return self._updates_counter
+
+    @updates_counter.setter
+    def updates_counter(self, updates_counter: 'int'):
+        """Sets the updates_counter of this V1MetricsStream.
+
+
+        :param updates_counter: The updates_counter of this V1MetricsStream.  # noqa: E501
+        :type: int
+        """
+
+        self._updates_counter = updates_counter
+
     @property
     def user_id(self) -> 'str':
         """Gets the user_id of this V1MetricsStream.  # noqa: E501