lightning-sdk 0.1.58__py3-none-any.whl → 0.2.0__py3-none-any.whl

lightning_sdk/lightning_cloud/openapi/models/v1_job_spec.py

@@ -57,7 +57,9 @@ class V1JobSpec(object):
         'instance_type': 'str',
         'modified_volume': 'bool',
         'path_mappings': 'list[V1PathMapping]',
+        'pipeline_id': 'str',
         'quantity': 'int',
+        'rank': 'int',
         'readiness_probe': 'V1JobHealthCheckConfig',
         'regions': 'list[str]',
         'requested_run_duration_seconds': 'str',
@@ -85,7 +87,9 @@ class V1JobSpec(object):
         'instance_type': 'instanceType',
         'modified_volume': 'modifiedVolume',
         'path_mappings': 'pathMappings',
+        'pipeline_id': 'pipelineId',
         'quantity': 'quantity',
+        'rank': 'rank',
         'readiness_probe': 'readinessProbe',
         'regions': 'regions',
         'requested_run_duration_seconds': 'requestedRunDurationSeconds',
@@ -96,7 +100,7 @@ class V1JobSpec(object):
         'volumes': 'volumes'
     }
 
-    def __init__(self, artifacts_destination: 'str' =None, artifacts_source: 'str' =None, cloudspace_id: 'str' =None, cluster_id: 'str' =None, command: 'str' =None, data_paths: 'list[V1DataPath]' =None, entrypoint: 'str' =None, env: 'list[V1EnvVar]' =None, image: 'str' =None, image_cluster_credentials: 'bool' =None, image_secret_ref: 'str' =None, include_credentials: 'bool' =None, instance_name: 'str' =None, instance_type: 'str' =None, modified_volume: 'bool' =None, path_mappings: 'list[V1PathMapping]' =None, quantity: 'int' =None, readiness_probe: 'V1JobHealthCheckConfig' =None, regions: 'list[str]' =None, requested_run_duration_seconds: 'str' =None, resources: 'V1Resources' =None, restart_policy: 'str' =None, run_id: 'str' =None, spot: 'bool' =None, volumes: 'list[V1Volume]' =None):  # noqa: E501
+    def __init__(self, artifacts_destination: 'str' =None, artifacts_source: 'str' =None, cloudspace_id: 'str' =None, cluster_id: 'str' =None, command: 'str' =None, data_paths: 'list[V1DataPath]' =None, entrypoint: 'str' =None, env: 'list[V1EnvVar]' =None, image: 'str' =None, image_cluster_credentials: 'bool' =None, image_secret_ref: 'str' =None, include_credentials: 'bool' =None, instance_name: 'str' =None, instance_type: 'str' =None, modified_volume: 'bool' =None, path_mappings: 'list[V1PathMapping]' =None, pipeline_id: 'str' =None, quantity: 'int' =None, rank: 'int' =None, readiness_probe: 'V1JobHealthCheckConfig' =None, regions: 'list[str]' =None, requested_run_duration_seconds: 'str' =None, resources: 'V1Resources' =None, restart_policy: 'str' =None, run_id: 'str' =None, spot: 'bool' =None, volumes: 'list[V1Volume]' =None):  # noqa: E501
         """V1JobSpec - a model defined in Swagger"""  # noqa: E501
         self._artifacts_destination = None
         self._artifacts_source = None
@@ -114,7 +118,9 @@ class V1JobSpec(object):
         self._instance_type = None
         self._modified_volume = None
         self._path_mappings = None
+        self._pipeline_id = None
         self._quantity = None
+        self._rank = None
         self._readiness_probe = None
         self._regions = None
         self._requested_run_duration_seconds = None
@@ -156,8 +162,12 @@ class V1JobSpec(object):
             self.modified_volume = modified_volume
         if path_mappings is not None:
             self.path_mappings = path_mappings
+        if pipeline_id is not None:
+            self.pipeline_id = pipeline_id
         if quantity is not None:
             self.quantity = quantity
+        if rank is not None:
+            self.rank = rank
         if readiness_probe is not None:
             self.readiness_probe = readiness_probe
         if regions is not None:
@@ -511,6 +521,27 @@ class V1JobSpec(object):
 
         self._path_mappings = path_mappings
 
+    @property
+    def pipeline_id(self) -> 'str':
+        """Gets the pipeline_id of this V1JobSpec.  # noqa: E501
+
+
+        :return: The pipeline_id of this V1JobSpec.  # noqa: E501
+        :rtype: str
+        """
+        return self._pipeline_id
+
+    @pipeline_id.setter
+    def pipeline_id(self, pipeline_id: 'str'):
+        """Sets the pipeline_id of this V1JobSpec.
+
+
+        :param pipeline_id: The pipeline_id of this V1JobSpec.  # noqa: E501
+        :type: str
+        """
+
+        self._pipeline_id = pipeline_id
+
     @property
     def quantity(self) -> 'int':
         """Gets the quantity of this V1JobSpec.  # noqa: E501
@@ -532,6 +563,27 @@ class V1JobSpec(object):
 
         self._quantity = quantity
 
+    @property
+    def rank(self) -> 'int':
+        """Gets the rank of this V1JobSpec.  # noqa: E501
+
+
+        :return: The rank of this V1JobSpec.  # noqa: E501
+        :rtype: int
+        """
+        return self._rank
+
+    @rank.setter
+    def rank(self, rank: 'int'):
+        """Sets the rank of this V1JobSpec.
+
+
+        :param rank: The rank of this V1JobSpec.  # noqa: E501
+        :type: int
+        """
+
+        self._rank = rank
+
     @property
     def readiness_probe(self) -> 'V1JobHealthCheckConfig':
         """Gets the readiness_probe of this V1JobSpec.  # noqa: E501

lightning_sdk/lightning_cloud/openapi/models/v1_job_timing.py

@@ -44,6 +44,7 @@ class V1JobTiming(object):
         'end': 'datetime',
         'name': 'str',
         'payload': 'str',
+        'rank': 'int',
         'start': 'datetime'
     }
 
@@ -51,14 +52,16 @@ class V1JobTiming(object):
         'end': 'end',
         'name': 'name',
         'payload': 'payload',
+        'rank': 'rank',
         'start': 'start'
     }
 
-    def __init__(self, end: 'datetime' =None, name: 'str' =None, payload: 'str' =None, start: 'datetime' =None):  # noqa: E501
+    def __init__(self, end: 'datetime' =None, name: 'str' =None, payload: 'str' =None, rank: 'int' =None, start: 'datetime' =None):  # noqa: E501
         """V1JobTiming - a model defined in Swagger"""  # noqa: E501
         self._end = None
         self._name = None
         self._payload = None
+        self._rank = None
         self._start = None
         self.discriminator = None
         if end is not None:
@@ -67,6 +70,8 @@ class V1JobTiming(object):
             self.name = name
         if payload is not None:
             self.payload = payload
+        if rank is not None:
+            self.rank = rank
         if start is not None:
             self.start = start
 
@@ -133,6 +138,27 @@ class V1JobTiming(object):
 
         self._payload = payload
 
+    @property
+    def rank(self) -> 'int':
+        """Gets the rank of this V1JobTiming.  # noqa: E501
+
+
+        :return: The rank of this V1JobTiming.  # noqa: E501
+        :rtype: int
+        """
+        return self._rank
+
+    @rank.setter
+    def rank(self, rank: 'int'):
+        """Sets the rank of this V1JobTiming.
+
+
+        :param rank: The rank of this V1JobTiming.  # noqa: E501
+        :type: int
+        """
+
+        self._rank = rank
+
     @property
     def start(self) -> 'datetime':
         """Gets the start of this V1JobTiming.  # noqa: E501

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

lightning_sdk/lightning_cloud/openapi/models/v1_lit_registry_artifact.py

@@ -41,6 +41,7 @@ class V1LitRegistryArtifact(object):
                             and the value is json key in definition.
     """
     swagger_types = {
+        'exposed_ports': 'list[str]',
         'full_hash_digest': 'str',
         'location': 'str',
         'pull_time': 'datetime',
@@ -51,6 +52,7 @@ class V1LitRegistryArtifact(object):
     }
 
     attribute_map = {
+        'exposed_ports': 'exposedPorts',
         'full_hash_digest': 'fullHashDigest',
         'location': 'location',
         'pull_time': 'pullTime',
@@ -60,8 +62,9 @@ class V1LitRegistryArtifact(object):
         'tag_name': 'tagName'
     }
 
-    def __init__(self, full_hash_digest: 'str' =None, location: 'str' =None, pull_time: 'datetime' =None, push_time: 'datetime' =None, short_hash_digest: 'str' =None, size: 'str' =None, tag_name: 'str' =None):  # noqa: E501
+    def __init__(self, exposed_ports: 'list[str]' =None, full_hash_digest: 'str' =None, location: 'str' =None, pull_time: 'datetime' =None, push_time: 'datetime' =None, short_hash_digest: 'str' =None, size: 'str' =None, tag_name: 'str' =None):  # noqa: E501
         """V1LitRegistryArtifact - a model defined in Swagger"""  # noqa: E501
+        self._exposed_ports = None
         self._full_hash_digest = None
         self._location = None
         self._pull_time = None
@@ -70,6 +73,8 @@ class V1LitRegistryArtifact(object):
         self._size = None
         self._tag_name = None
         self.discriminator = None
+        if exposed_ports is not None:
+            self.exposed_ports = exposed_ports
         if full_hash_digest is not None:
             self.full_hash_digest = full_hash_digest
         if location is not None:
@@ -85,6 +90,27 @@ class V1LitRegistryArtifact(object):
         if tag_name is not None:
             self.tag_name = tag_name
 
+    @property
+    def exposed_ports(self) -> 'list[str]':
+        """Gets the exposed_ports of this V1LitRegistryArtifact.  # noqa: E501
+
+
+        :return: The exposed_ports of this V1LitRegistryArtifact.  # noqa: E501
+        :rtype: list[str]
+        """
+        return self._exposed_ports
+
+    @exposed_ports.setter
+    def exposed_ports(self, exposed_ports: 'list[str]'):
+        """Sets the exposed_ports of this V1LitRegistryArtifact.
+
+
+        :param exposed_ports: The exposed_ports of this V1LitRegistryArtifact.  # noqa: E501
+        :type: list[str]
+        """
+
+        self._exposed_ports = exposed_ports
+
     @property
     def full_hash_digest(self) -> 'str':
         """Gets the full_hash_digest of this V1LitRegistryArtifact.  # noqa: E501

lightning_sdk/lightning_cloud/openapi/models/v1_lit_repository.py

@@ -42,6 +42,7 @@ class V1LitRepository(object):
     """
     swagger_types = {
         'artifact_count': 'str',
+        'artifacts': 'list[V1LitRegistryArtifact]',
         'creation_time': 'datetime',
         'description': 'str',
         'id': 'str',
@@ -53,6 +54,7 @@ class V1LitRepository(object):
 
     attribute_map = {
         'artifact_count': 'artifactCount',
+        'artifacts': 'artifacts',
         'creation_time': 'creationTime',
         'description': 'description',
         'id': 'id',
@@ -62,9 +64,10 @@ class V1LitRepository(object):
         'update_time': 'updateTime'
     }
 
-    def __init__(self, artifact_count: 'str' =None, creation_time: 'datetime' =None, description: 'str' =None, id: 'str' =None, latest_artifact: 'V1LitRegistryArtifact' =None, name: 'str' =None, pull_count: 'str' =None, update_time: 'datetime' =None):  # noqa: E501
+    def __init__(self, artifact_count: 'str' =None, artifacts: 'list[V1LitRegistryArtifact]' =None, creation_time: 'datetime' =None, description: 'str' =None, id: 'str' =None, latest_artifact: 'V1LitRegistryArtifact' =None, name: 'str' =None, pull_count: 'str' =None, update_time: 'datetime' =None):  # noqa: E501
         """V1LitRepository - a model defined in Swagger"""  # noqa: E501
         self._artifact_count = None
+        self._artifacts = None
         self._creation_time = None
         self._description = None
         self._id = None
@@ -75,6 +78,8 @@ class V1LitRepository(object):
         self.discriminator = None
         if artifact_count is not None:
             self.artifact_count = artifact_count
+        if artifacts is not None:
+            self.artifacts = artifacts
         if creation_time is not None:
             self.creation_time = creation_time
         if description is not None:
@@ -111,6 +116,29 @@ class V1LitRepository(object):
 
         self._artifact_count = artifact_count
 
+    @property
+    def artifacts(self) -> 'list[V1LitRegistryArtifact]':
+        """Gets the artifacts of this V1LitRepository.  # noqa: E501
+
+        All artifacts for this repo, this plus lit registry are coming from behind the scenes listLitRegistryRepositoryImageArtifactVersions. We'll use these for a complete view of all registry container artifacts vs the UI having to make the calls separately.  # noqa: E501
+
+        :return: The artifacts of this V1LitRepository.  # noqa: E501
+        :rtype: list[V1LitRegistryArtifact]
+        """
+        return self._artifacts
+
+    @artifacts.setter
+    def artifacts(self, artifacts: 'list[V1LitRegistryArtifact]'):
+        """Sets the artifacts of this V1LitRepository.
+
+        All artifacts for this repo, this plus lit registry are coming from behind the scenes listLitRegistryRepositoryImageArtifactVersions. We'll use these for a complete view of all registry container artifacts vs the UI having to make the calls separately.  # noqa: E501
+
+        :param artifacts: The artifacts of this V1LitRepository.  # noqa: E501
+        :type: list[V1LitRegistryArtifact]
+        """
+
+        self._artifacts = artifacts
+
     @property
     def creation_time(self) -> 'datetime':
         """Gets the creation_time of this V1LitRepository.  # noqa: E501

lightning_sdk/lightning_cloud/openapi/models/v1_managed_model.py

@@ -43,6 +43,7 @@ class V1ManagedModel(object):
     swagger_types = {
         'abilities': 'V1ManagedModelAbilities',
         'completion_token_price': 'float',
+        'deployment_details': 'V1DeploymentDetails',
         'description': 'str',
         'endpoint_id': 'str',
         'id': 'str',
@@ -54,6 +55,7 @@ class V1ManagedModel(object):
     attribute_map = {
         'abilities': 'abilities',
         'completion_token_price': 'completionTokenPrice',
+        'deployment_details': 'deploymentDetails',
         'description': 'description',
         'endpoint_id': 'endpointId',
         'id': 'id',
@@ -62,10 +64,11 @@ class V1ManagedModel(object):
         '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
+    def __init__(self, abilities: 'V1ManagedModelAbilities' =None, completion_token_price: 'float' =None, deployment_details: 'V1DeploymentDetails' =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._deployment_details = None
         self._description = None
         self._endpoint_id = None
         self._id = None
@@ -77,6 +80,8 @@ class V1ManagedModel(object):
             self.abilities = abilities
         if completion_token_price is not None:
             self.completion_token_price = completion_token_price
+        if deployment_details is not None:
+            self.deployment_details = deployment_details
         if description is not None:
             self.description = description
         if endpoint_id is not None:
@@ -132,6 +137,27 @@ class V1ManagedModel(object):
 
         self._completion_token_price = completion_token_price
 
+    @property
+    def deployment_details(self) -> 'V1DeploymentDetails':
+        """Gets the deployment_details of this V1ManagedModel.  # noqa: E501
+
+
+        :return: The deployment_details of this V1ManagedModel.  # noqa: E501
+        :rtype: V1DeploymentDetails
+        """
+        return self._deployment_details
+
+    @deployment_details.setter
+    def deployment_details(self, deployment_details: 'V1DeploymentDetails'):
+        """Sets the deployment_details of this V1ManagedModel.
+
+
+        :param deployment_details: The deployment_details of this V1ManagedModel.  # noqa: E501
+        :type: V1DeploymentDetails
+        """
+
+        self._deployment_details = deployment_details
+
     @property
     def description(self) -> 'str':
         """Gets the description of this V1ManagedModel.  # noqa: E501

lightning_sdk/lightning_cloud/openapi/models/v1_multi_machine_job.py

@@ -49,6 +49,7 @@ class V1MultiMachineJob(object):
         'machines': 'int',
         'message': 'str',
         'name': 'str',
+        'pipeline_id': 'str',
         'project_id': 'str',
         'spec': 'V1JobSpec',
         'state': 'V1MultiMachineJobState',
@@ -67,6 +68,7 @@ class V1MultiMachineJob(object):
         'machines': 'machines',
         'message': 'message',
         'name': 'name',
+        'pipeline_id': 'pipelineId',
         'project_id': 'projectId',
         'spec': 'spec',
         'state': 'state',
@@ -76,7 +78,7 @@ class V1MultiMachineJob(object):
         'user_id': 'userId'
     }
 
-    def __init__(self, cloudspace_id: 'str' =None, created_at: 'datetime' =None, desired_state: 'V1MultiMachineJobState' =None, fault_tolerance: 'V1MultiMachineJobFaultTolerance' =None, id: 'str' =None, machines: 'int' =None, message: 'str' =None, name: 'str' =None, project_id: 'str' =None, spec: 'V1JobSpec' =None, state: 'V1MultiMachineJobState' =None, status: 'V1MultiMachineJobStatus' =None, total_cost: 'float' =None, updated_at: 'datetime' =None, user_id: 'str' =None):  # noqa: E501
+    def __init__(self, cloudspace_id: 'str' =None, created_at: 'datetime' =None, desired_state: 'V1MultiMachineJobState' =None, fault_tolerance: 'V1MultiMachineJobFaultTolerance' =None, id: 'str' =None, machines: 'int' =None, message: 'str' =None, name: 'str' =None, pipeline_id: 'str' =None, project_id: 'str' =None, spec: 'V1JobSpec' =None, state: 'V1MultiMachineJobState' =None, status: 'V1MultiMachineJobStatus' =None, total_cost: 'float' =None, updated_at: 'datetime' =None, user_id: 'str' =None):  # noqa: E501
         """V1MultiMachineJob - a model defined in Swagger"""  # noqa: E501
         self._cloudspace_id = None
         self._created_at = None
@@ -86,6 +88,7 @@ class V1MultiMachineJob(object):
         self._machines = None
         self._message = None
         self._name = None
+        self._pipeline_id = None
         self._project_id = None
         self._spec = None
         self._state = None
@@ -110,6 +113,8 @@ class V1MultiMachineJob(object):
             self.message = message
         if name is not None:
             self.name = name
+        if pipeline_id is not None:
+            self.pipeline_id = pipeline_id
         if project_id is not None:
             self.project_id = project_id
         if spec is not None:
@@ -293,6 +298,27 @@ class V1MultiMachineJob(object):
 
         self._name = name
 
+    @property
+    def pipeline_id(self) -> 'str':
+        """Gets the pipeline_id of this V1MultiMachineJob.  # noqa: E501
+
+
+        :return: The pipeline_id of this V1MultiMachineJob.  # noqa: E501
+        :rtype: str
+        """
+        return self._pipeline_id
+
+    @pipeline_id.setter
+    def pipeline_id(self, pipeline_id: 'str'):
+        """Sets the pipeline_id of this V1MultiMachineJob.
+
+
+        :param pipeline_id: The pipeline_id of this V1MultiMachineJob.  # noqa: E501
+        :type: str
+        """
+
+        self._pipeline_id = pipeline_id
+
     @property
     def project_id(self) -> 'str':
         """Gets the project_id of this V1MultiMachineJob.  # noqa: E501

lightning_sdk/lightning_cloud/openapi/models/v1_multi_machine_job_state.py

@@ -44,6 +44,8 @@ class V1MultiMachineJobState(object):
     DELETED = "MultiMachineJob_STATE_DELETED"
     FAILED = "MultiMachineJob_STATE_FAILED"
     COMPLETED = "MultiMachineJob_STATE_COMPLETED"
+    STOP = "MultiMachineJob_STATE_STOP"
+    DELETE = "MultiMachineJob_STATE_DELETE"
     """
     Attributes:
       swagger_types (dict): The key is attribute name