lightning-sdk 0.1.37__py3-none-any.whl → 0.1.39__py3-none-any.whl

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

lightning_sdk/lightning_cloud/openapi/models/project_id_agents_body.py

@@ -50,10 +50,12 @@ class ProjectIdAgentsBody(object):
         'knowledge': 'str',
         'knowledge_configuration': 'V1KnowledgeConfiguration',
         'model': 'str',
+        'model_provider': 'str',
         'name': 'str',
         'org_id': 'str',
         'prompt_suggestions': 'list[V1PromptSuggestion]',
         'prompt_template': 'str',
+        'publish_status': 'str',
         'thumbnail_url': 'str'
     }
 
@@ -67,14 +69,16 @@ class ProjectIdAgentsBody(object):
         'knowledge': 'knowledge',
         'knowledge_configuration': 'knowledgeConfiguration',
         'model': 'model',
+        'model_provider': 'modelProvider',
         'name': 'name',
         'org_id': 'orgId',
         'prompt_suggestions': 'promptSuggestions',
         'prompt_template': 'promptTemplate',
+        'publish_status': 'publishStatus',
         'thumbnail_url': 'thumbnailUrl'
     }
 
-    def __init__(self, cloudspace_id: 'str' =None, cluster_id: 'str' =None, description: 'str' =None, endpoint: 'V1Endpoint' =None, file_uploads_enabled: 'bool' =None, internal_assistant_name: 'str' =None, knowledge: 'str' =None, knowledge_configuration: 'V1KnowledgeConfiguration' =None, model: 'str' =None, name: 'str' =None, org_id: 'str' =None, prompt_suggestions: 'list[V1PromptSuggestion]' =None, prompt_template: 'str' =None, thumbnail_url: 'str' =None):  # noqa: E501
+    def __init__(self, cloudspace_id: 'str' =None, cluster_id: 'str' =None, description: 'str' =None, endpoint: 'V1Endpoint' =None, file_uploads_enabled: 'bool' =None, internal_assistant_name: 'str' =None, knowledge: 'str' =None, knowledge_configuration: 'V1KnowledgeConfiguration' =None, model: 'str' =None, model_provider: 'str' =None, name: 'str' =None, org_id: 'str' =None, prompt_suggestions: 'list[V1PromptSuggestion]' =None, prompt_template: 'str' =None, publish_status: 'str' =None, thumbnail_url: 'str' =None):  # noqa: E501
         """ProjectIdAgentsBody - a model defined in Swagger"""  # noqa: E501
         self._cloudspace_id = None
         self._cluster_id = None
@@ -85,10 +89,12 @@ class ProjectIdAgentsBody(object):
         self._knowledge = None
         self._knowledge_configuration = None
         self._model = None
+        self._model_provider = None
         self._name = None
         self._org_id = None
         self._prompt_suggestions = None
         self._prompt_template = None
+        self._publish_status = None
         self._thumbnail_url = None
         self.discriminator = None
         if cloudspace_id is not None:
@@ -109,6 +115,8 @@ class ProjectIdAgentsBody(object):
             self.knowledge_configuration = knowledge_configuration
         if model is not None:
             self.model = model
+        if model_provider is not None:
+            self.model_provider = model_provider
         if name is not None:
             self.name = name
         if org_id is not None:
@@ -117,6 +125,8 @@ class ProjectIdAgentsBody(object):
             self.prompt_suggestions = prompt_suggestions
         if prompt_template is not None:
             self.prompt_template = prompt_template
+        if publish_status is not None:
+            self.publish_status = publish_status
         if thumbnail_url is not None:
             self.thumbnail_url = thumbnail_url
 
@@ -309,6 +319,27 @@ class ProjectIdAgentsBody(object):
 
         self._model = model
 
+    @property
+    def model_provider(self) -> 'str':
+        """Gets the model_provider of this ProjectIdAgentsBody.  # noqa: E501
+
+
+        :return: The model_provider of this ProjectIdAgentsBody.  # noqa: E501
+        :rtype: str
+        """
+        return self._model_provider
+
+    @model_provider.setter
+    def model_provider(self, model_provider: 'str'):
+        """Sets the model_provider of this ProjectIdAgentsBody.
+
+
+        :param model_provider: The model_provider of this ProjectIdAgentsBody.  # noqa: E501
+        :type: str
+        """
+
+        self._model_provider = model_provider
+
     @property
     def name(self) -> 'str':
         """Gets the name of this ProjectIdAgentsBody.  # noqa: E501
@@ -393,6 +424,27 @@ class ProjectIdAgentsBody(object):
 
         self._prompt_template = prompt_template
 
+    @property
+    def publish_status(self) -> 'str':
+        """Gets the publish_status of this ProjectIdAgentsBody.  # noqa: E501
+
+
+        :return: The publish_status of this ProjectIdAgentsBody.  # noqa: E501
+        :rtype: str
+        """
+        return self._publish_status
+
+    @publish_status.setter
+    def publish_status(self, publish_status: 'str'):
+        """Sets the publish_status of this ProjectIdAgentsBody.
+
+
+        :param publish_status: The publish_status of this ProjectIdAgentsBody.  # noqa: E501
+        :type: str
+        """
+
+        self._publish_status = publish_status
+
     @property
     def thumbnail_url(self) -> 'str':
         """Gets the thumbnail_url of this ProjectIdAgentsBody.  # noqa: E501

lightning_sdk/lightning_cloud/openapi/models/project_id_cloudspaces_body.py

@@ -50,6 +50,8 @@ class ProjectIdCloudspacesBody(object):
         'disk_size': 'str',
         'display_name': 'str',
         'name': 'str',
+        'plugins': 'list[str]',
+        'requested_run_duration_seconds': 'str',
         'seed_files': 'list[V1CloudSpaceSeedFile]',
         'spot': 'bool'
     }
@@ -64,11 +66,13 @@ class ProjectIdCloudspacesBody(object):
         'disk_size': 'diskSize',
         'display_name': 'displayName',
         'name': 'name',
+        'plugins': 'plugins',
+        'requested_run_duration_seconds': 'requestedRunDurationSeconds',
         'seed_files': 'seedFiles',
         'spot': 'spot'
     }
 
-    def __init__(self, can_download_source_code: 'bool' =None, cloud_space_instance_cpu_image_override: 'str' =None, cloud_space_instance_gpu_image_override: 'str' =None, cluster_id: 'str' =None, compute_name: 'str' =None, data_connection_mounts: 'list[V1DataConnectionMount]' =None, disk_size: 'str' =None, display_name: 'str' =None, name: 'str' =None, seed_files: 'list[V1CloudSpaceSeedFile]' =None, spot: 'bool' =None):  # noqa: E501
+    def __init__(self, can_download_source_code: 'bool' =None, cloud_space_instance_cpu_image_override: 'str' =None, cloud_space_instance_gpu_image_override: 'str' =None, cluster_id: 'str' =None, compute_name: 'str' =None, data_connection_mounts: 'list[V1DataConnectionMount]' =None, disk_size: 'str' =None, display_name: 'str' =None, name: 'str' =None, plugins: 'list[str]' =None, requested_run_duration_seconds: 'str' =None, seed_files: 'list[V1CloudSpaceSeedFile]' =None, spot: 'bool' =None):  # noqa: E501
         """ProjectIdCloudspacesBody - a model defined in Swagger"""  # noqa: E501
         self._can_download_source_code = None
         self._cloud_space_instance_cpu_image_override = None
@@ -79,6 +83,8 @@ class ProjectIdCloudspacesBody(object):
         self._disk_size = None
         self._display_name = None
         self._name = None
+        self._plugins = None
+        self._requested_run_duration_seconds = None
         self._seed_files = None
         self._spot = None
         self.discriminator = None
@@ -100,6 +106,10 @@ class ProjectIdCloudspacesBody(object):
             self.display_name = display_name
         if name is not None:
             self.name = name
+        if plugins is not None:
+            self.plugins = plugins
+        if requested_run_duration_seconds is not None:
+            self.requested_run_duration_seconds = requested_run_duration_seconds
         if seed_files is not None:
             self.seed_files = seed_files
         if spot is not None:
@@ -294,6 +304,48 @@ class ProjectIdCloudspacesBody(object):
 
         self._name = name
 
+    @property
+    def plugins(self) -> 'list[str]':
+        """Gets the plugins of this ProjectIdCloudspacesBody.  # noqa: E501
+
+
+        :return: The plugins of this ProjectIdCloudspacesBody.  # noqa: E501
+        :rtype: list[str]
+        """
+        return self._plugins
+
+    @plugins.setter
+    def plugins(self, plugins: 'list[str]'):
+        """Sets the plugins of this ProjectIdCloudspacesBody.
+
+
+        :param plugins: The plugins of this ProjectIdCloudspacesBody.  # noqa: E501
+        :type: list[str]
+        """
+
+        self._plugins = plugins
+
+    @property
+    def requested_run_duration_seconds(self) -> 'str':
+        """Gets the requested_run_duration_seconds of this ProjectIdCloudspacesBody.  # noqa: E501
+
+
+        :return: The requested_run_duration_seconds of this ProjectIdCloudspacesBody.  # noqa: E501
+        :rtype: str
+        """
+        return self._requested_run_duration_seconds
+
+    @requested_run_duration_seconds.setter
+    def requested_run_duration_seconds(self, requested_run_duration_seconds: 'str'):
+        """Sets the requested_run_duration_seconds of this ProjectIdCloudspacesBody.
+
+
+        :param requested_run_duration_seconds: The requested_run_duration_seconds of this ProjectIdCloudspacesBody.  # noqa: E501
+        :type: str
+        """
+
+        self._requested_run_duration_seconds = requested_run_duration_seconds
+
     @property
     def seed_files(self) -> 'list[V1CloudSpaceSeedFile]':
         """Gets the seed_files of this ProjectIdCloudspacesBody.  # noqa: E501

lightning_sdk/lightning_cloud/openapi/models/project_id_multimachinejobs_body.py

@@ -0,0 +1,201 @@
+# 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 ProjectIdMultimachinejobsBody(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 = {
+        'cluster_id': 'str',
+        'machines': 'int',
+        'name': 'str',
+        'spec': 'V1JobSpec'
+    }
+
+    attribute_map = {
+        'cluster_id': 'clusterId',
+        'machines': 'machines',
+        'name': 'name',
+        'spec': 'spec'
+    }
+
+    def __init__(self, cluster_id: 'str' =None, machines: 'int' =None, name: 'str' =None, spec: 'V1JobSpec' =None):  # noqa: E501
+        """ProjectIdMultimachinejobsBody - a model defined in Swagger"""  # noqa: E501
+        self._cluster_id = None
+        self._machines = None
+        self._name = None
+        self._spec = None
+        self.discriminator = None
+        if cluster_id is not None:
+            self.cluster_id = cluster_id
+        if machines is not None:
+            self.machines = machines
+        if name is not None:
+            self.name = name
+        if spec is not None:
+            self.spec = spec
+
+    @property
+    def cluster_id(self) -> 'str':
+        """Gets the cluster_id of this ProjectIdMultimachinejobsBody.  # noqa: E501
+
+
+        :return: The cluster_id of this ProjectIdMultimachinejobsBody.  # noqa: E501
+        :rtype: str
+        """
+        return self._cluster_id
+
+    @cluster_id.setter
+    def cluster_id(self, cluster_id: 'str'):
+        """Sets the cluster_id of this ProjectIdMultimachinejobsBody.
+
+
+        :param cluster_id: The cluster_id of this ProjectIdMultimachinejobsBody.  # noqa: E501
+        :type: str
+        """
+
+        self._cluster_id = cluster_id
+
+    @property
+    def machines(self) -> 'int':
+        """Gets the machines of this ProjectIdMultimachinejobsBody.  # noqa: E501
+
+
+        :return: The machines of this ProjectIdMultimachinejobsBody.  # noqa: E501
+        :rtype: int
+        """
+        return self._machines
+
+    @machines.setter
+    def machines(self, machines: 'int'):
+        """Sets the machines of this ProjectIdMultimachinejobsBody.
+
+
+        :param machines: The machines of this ProjectIdMultimachinejobsBody.  # noqa: E501
+        :type: int
+        """
+
+        self._machines = machines
+
+    @property
+    def name(self) -> 'str':
+        """Gets the name of this ProjectIdMultimachinejobsBody.  # noqa: E501
+
+
+        :return: The name of this ProjectIdMultimachinejobsBody.  # noqa: E501
+        :rtype: str
+        """
+        return self._name
+
+    @name.setter
+    def name(self, name: 'str'):
+        """Sets the name of this ProjectIdMultimachinejobsBody.
+
+
+        :param name: The name of this ProjectIdMultimachinejobsBody.  # noqa: E501
+        :type: str
+        """
+
+        self._name = name
+
+    @property
+    def spec(self) -> 'V1JobSpec':
+        """Gets the spec of this ProjectIdMultimachinejobsBody.  # noqa: E501
+
+
+        :return: The spec of this ProjectIdMultimachinejobsBody.  # noqa: E501
+        :rtype: V1JobSpec
+        """
+        return self._spec
+
+    @spec.setter
+    def spec(self, spec: 'V1JobSpec'):
+        """Sets the spec of this ProjectIdMultimachinejobsBody.
+
+
+        :param spec: The spec of this ProjectIdMultimachinejobsBody.  # noqa: E501
+        :type: V1JobSpec
+        """
+
+        self._spec = spec
+
+    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(ProjectIdMultimachinejobsBody, 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: 'ProjectIdMultimachinejobsBody') -> bool:
+        """Returns true if both objects are equal"""
+        if not isinstance(other, ProjectIdMultimachinejobsBody):
+            return False
+
+        return self.__dict__ == other.__dict__
+
+    def __ne__(self, other: 'ProjectIdMultimachinejobsBody') -> bool:
+        """Returns true if both objects are not equal"""
+        return not self == other

lightning_sdk/lightning_cloud/openapi/models/update.py

@@ -42,8 +42,7 @@ class Update(object):
     """
     swagger_types = {
         'aws': 'V1AwsDataConnection',
-        'efs': 'V1EfsDataConnection',
-        'efs_folder': 'V1EFSFolderDataConnection',
+        'efs': 'V1EfsConfig',
         'gcp': 'V1GcpDataConnection',
         'gcs_folder': 'V1GCSFolderDataConnection',
         'name': 'str',
@@ -56,7 +55,6 @@ class Update(object):
     attribute_map = {
         'aws': 'aws',
         'efs': 'efs',
-        'efs_folder': 'efsFolder',
         'gcp': 'gcp',
         'gcs_folder': 'gcsFolder',
         'name': 'name',
@@ -66,11 +64,10 @@ class Update(object):
         'writable': 'writable'
     }
 
-    def __init__(self, aws: 'V1AwsDataConnection' =None, efs: 'V1EfsDataConnection' =None, efs_folder: 'V1EFSFolderDataConnection' =None, gcp: 'V1GcpDataConnection' =None, gcs_folder: 'V1GCSFolderDataConnection' =None, name: 'str' =None, run_cmds: 'list[str]' =None, s3_folder: 'V1S3FolderDataConnection' =None, snowflake: 'V1SnowflakeDataConnection' =None, writable: 'bool' =None):  # noqa: E501
+    def __init__(self, aws: 'V1AwsDataConnection' =None, efs: 'V1EfsConfig' =None, gcp: 'V1GcpDataConnection' =None, gcs_folder: 'V1GCSFolderDataConnection' =None, name: 'str' =None, run_cmds: 'list[str]' =None, s3_folder: 'V1S3FolderDataConnection' =None, snowflake: 'V1SnowflakeDataConnection' =None, writable: 'bool' =None):  # noqa: E501
         """Update - a model defined in Swagger"""  # noqa: E501
         self._aws = None
         self._efs = None
-        self._efs_folder = None
         self._gcp = None
         self._gcs_folder = None
         self._name = None
@@ -83,8 +80,6 @@ class Update(object):
             self.aws = aws
         if efs is not None:
             self.efs = efs
-        if efs_folder is not None:
-            self.efs_folder = efs_folder
         if gcp is not None:
             self.gcp = gcp
         if gcs_folder is not None:
@@ -122,47 +117,26 @@ class Update(object):
         self._aws = aws
 
     @property
-    def efs(self) -> 'V1EfsDataConnection':
+    def efs(self) -> 'V1EfsConfig':
         """Gets the efs of this Update.  # noqa: E501
 
 
         :return: The efs of this Update.  # noqa: E501
-        :rtype: V1EfsDataConnection
+        :rtype: V1EfsConfig
         """
         return self._efs
 
     @efs.setter
-    def efs(self, efs: 'V1EfsDataConnection'):
+    def efs(self, efs: 'V1EfsConfig'):
         """Sets the efs of this Update.
 
 
         :param efs: The efs of this Update.  # noqa: E501
-        :type: V1EfsDataConnection
+        :type: V1EfsConfig
         """
 
         self._efs = efs
 
-    @property
-    def efs_folder(self) -> 'V1EFSFolderDataConnection':
-        """Gets the efs_folder of this Update.  # noqa: E501
-
-
-        :return: The efs_folder of this Update.  # noqa: E501
-        :rtype: V1EFSFolderDataConnection
-        """
-        return self._efs_folder
-
-    @efs_folder.setter
-    def efs_folder(self, efs_folder: 'V1EFSFolderDataConnection'):
-        """Sets the efs_folder of this Update.
-
-
-        :param efs_folder: The efs_folder of this Update.  # noqa: E501
-        :type: V1EFSFolderDataConnection
-        """
-
-        self._efs_folder = efs_folder
-
     @property
     def gcp(self) -> 'V1GcpDataConnection':
         """Gets the gcp of this Update.  # noqa: E501