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

lightning_sdk/lightning_cloud/openapi/models/v1_data_path.py

@@ -43,20 +43,23 @@ class V1DataPath(object):
     swagger_types = {
         'single_file': 'bool',
         'stored_path': 'str',
-        'target_path': 'str'
+        'target_path': 'str',
+        'updated_at': 'datetime'
     }
 
     attribute_map = {
         'single_file': 'singleFile',
         'stored_path': 'storedPath',
-        'target_path': 'targetPath'
+        'target_path': 'targetPath',
+        'updated_at': 'updatedAt'
     }
 
-    def __init__(self, single_file: 'bool' =None, stored_path: 'str' =None, target_path: 'str' =None):  # noqa: E501
+    def __init__(self, single_file: 'bool' =None, stored_path: 'str' =None, target_path: 'str' =None, updated_at: 'datetime' =None):  # noqa: E501
         """V1DataPath - a model defined in Swagger"""  # noqa: E501
         self._single_file = None
         self._stored_path = None
         self._target_path = None
+        self._updated_at = None
         self.discriminator = None
         if single_file is not None:
             self.single_file = single_file
@@ -64,6 +67,8 @@ class V1DataPath(object):
             self.stored_path = stored_path
         if target_path is not None:
             self.target_path = target_path
+        if updated_at is not None:
+            self.updated_at = updated_at
 
     @property
     def single_file(self) -> 'bool':
@@ -128,6 +133,27 @@ class V1DataPath(object):
 
         self._target_path = target_path
 
+    @property
+    def updated_at(self) -> 'datetime':
+        """Gets the updated_at of this V1DataPath.  # noqa: E501
+
+
+        :return: The updated_at of this V1DataPath.  # noqa: E501
+        :rtype: datetime
+        """
+        return self._updated_at
+
+    @updated_at.setter
+    def updated_at(self, updated_at: 'datetime'):
+        """Sets the updated_at of this V1DataPath.
+
+
+        :param updated_at: The updated_at of this V1DataPath.  # noqa: E501
+        :type: datetime
+        """
+
+        self._updated_at = updated_at
+
     def to_dict(self) -> dict:
         """Returns the model properties as a dict"""
         result = {}

lightning_sdk/lightning_cloud/openapi/models/v1_job_spec.py

@@ -41,6 +41,8 @@ class V1JobSpec(object):
                             and the value is json key in definition.
     """
     swagger_types = {
+        'artifacts_destination': 'str',
+        'artifacts_source': 'str',
         'cloudspace_id': 'str',
         'cluster_id': 'str',
         'command': 'str',
@@ -58,13 +60,13 @@ class V1JobSpec(object):
         'resources': 'V1Resources',
         'restart_policy': 'str',
         'run_id': 'str',
-        'skip_data_connections_setup': 'bool',
-        'skip_filesystem_setup': 'bool',
         'spot': 'bool',
         'volumes': 'list[V1Volume]'
     }
 
     attribute_map = {
+        'artifacts_destination': 'artifactsDestination',
+        'artifacts_source': 'artifactsSource',
         'cloudspace_id': 'cloudspaceId',
         'cluster_id': 'clusterId',
         'command': 'command',
@@ -82,14 +84,14 @@ class V1JobSpec(object):
         'resources': 'resources',
         'restart_policy': 'restartPolicy',
         'run_id': 'runId',
-        'skip_data_connections_setup': 'skipDataConnectionsSetup',
-        'skip_filesystem_setup': 'skipFilesystemSetup',
         'spot': 'spot',
         'volumes': 'volumes'
     }
 
-    def __init__(self, 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, readiness_probe: 'V1JobHealthCheckConfig' =None, resources: 'V1Resources' =None, restart_policy: 'str' =None, run_id: 'str' =None, skip_data_connections_setup: 'bool' =None, skip_filesystem_setup: 'bool' =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, readiness_probe: 'V1JobHealthCheckConfig' =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
         self._cloudspace_id = None
         self._cluster_id = None
         self._command = None
@@ -107,11 +109,13 @@ class V1JobSpec(object):
         self._resources = None
         self._restart_policy = None
         self._run_id = None
-        self._skip_data_connections_setup = None
-        self._skip_filesystem_setup = None
         self._spot = None
         self._volumes = None
         self.discriminator = None
+        if artifacts_destination is not None:
+            self.artifacts_destination = artifacts_destination
+        if artifacts_source is not None:
+            self.artifacts_source = artifacts_source
         if cloudspace_id is not None:
             self.cloudspace_id = cloudspace_id
         if cluster_id is not None:
@@ -146,15 +150,53 @@ class V1JobSpec(object):
             self.restart_policy = restart_policy
         if run_id is not None:
             self.run_id = run_id
-        if skip_data_connections_setup is not None:
-            self.skip_data_connections_setup = skip_data_connections_setup
-        if skip_filesystem_setup is not None:
-            self.skip_filesystem_setup = skip_filesystem_setup
         if spot is not None:
             self.spot = spot
         if volumes is not None:
             self.volumes = volumes
 
+    @property
+    def artifacts_destination(self) -> 'str':
+        """Gets the artifacts_destination of this V1JobSpec.  # noqa: E501
+
+
+        :return: The artifacts_destination of this V1JobSpec.  # noqa: E501
+        :rtype: str
+        """
+        return self._artifacts_destination
+
+    @artifacts_destination.setter
+    def artifacts_destination(self, artifacts_destination: 'str'):
+        """Sets the artifacts_destination of this V1JobSpec.
+
+
+        :param artifacts_destination: The artifacts_destination of this V1JobSpec.  # noqa: E501
+        :type: str
+        """
+
+        self._artifacts_destination = artifacts_destination
+
+    @property
+    def artifacts_source(self) -> 'str':
+        """Gets the artifacts_source of this V1JobSpec.  # noqa: E501
+
+
+        :return: The artifacts_source of this V1JobSpec.  # noqa: E501
+        :rtype: str
+        """
+        return self._artifacts_source
+
+    @artifacts_source.setter
+    def artifacts_source(self, artifacts_source: 'str'):
+        """Sets the artifacts_source of this V1JobSpec.
+
+
+        :param artifacts_source: The artifacts_source of this V1JobSpec.  # noqa: E501
+        :type: str
+        """
+
+        self._artifacts_source = artifacts_source
+
     @property
     def cloudspace_id(self) -> 'str':
         """Gets the cloudspace_id of this V1JobSpec.  # noqa: E501
@@ -512,48 +554,6 @@ class V1JobSpec(object):
 
         self._run_id = run_id
 
-    @property
-    def skip_data_connections_setup(self) -> 'bool':
-        """Gets the skip_data_connections_setup of this V1JobSpec.  # noqa: E501
-
-
-        :return: The skip_data_connections_setup of this V1JobSpec.  # noqa: E501
-        :rtype: bool
-        """
-        return self._skip_data_connections_setup
-
-    @skip_data_connections_setup.setter
-    def skip_data_connections_setup(self, skip_data_connections_setup: 'bool'):
-        """Sets the skip_data_connections_setup of this V1JobSpec.
-
-
-        :param skip_data_connections_setup: The skip_data_connections_setup of this V1JobSpec.  # noqa: E501
-        :type: bool
-        """
-
-        self._skip_data_connections_setup = skip_data_connections_setup
-
-    @property
-    def skip_filesystem_setup(self) -> 'bool':
-        """Gets the skip_filesystem_setup of this V1JobSpec.  # noqa: E501
-
-
-        :return: The skip_filesystem_setup of this V1JobSpec.  # noqa: E501
-        :rtype: bool
-        """
-        return self._skip_filesystem_setup
-
-    @skip_filesystem_setup.setter
-    def skip_filesystem_setup(self, skip_filesystem_setup: 'bool'):
-        """Sets the skip_filesystem_setup of this V1JobSpec.
-
-
-        :param skip_filesystem_setup: The skip_filesystem_setup of this V1JobSpec.  # noqa: E501
-        :type: bool
-        """
-
-        self._skip_filesystem_setup = skip_filesystem_setup
-
     @property
     def spot(self) -> 'bool':
         """Gets the spot of this V1JobSpec.  # noqa: E501

lightning_sdk/lightning_cloud/openapi/models/v1_multi_machine_job.py

@@ -46,6 +46,7 @@ class V1MultiMachineJob(object):
         'desired_state': 'V1MultiMachineJobState',
         'id': 'str',
         'machines': 'int',
+        'message': 'str',
         'name': 'str',
         'project_id': 'str',
         'spec': 'V1JobSpec',
@@ -60,6 +61,7 @@ class V1MultiMachineJob(object):
         'desired_state': 'desiredState',
         'id': 'id',
         'machines': 'machines',
+        'message': 'message',
         'name': 'name',
         'project_id': 'projectId',
         'spec': 'spec',
@@ -68,13 +70,14 @@ class V1MultiMachineJob(object):
         'user_id': 'userId'
     }
 
-    def __init__(self, cloudspace_id: 'str' =None, created_at: 'datetime' =None, desired_state: 'V1MultiMachineJobState' =None, id: 'str' =None, machines: 'int' =None, name: 'str' =None, project_id: 'str' =None, spec: 'V1JobSpec' =None, status: 'V1MultiMachineJobStatus' =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, id: 'str' =None, machines: 'int' =None, message: 'str' =None, name: 'str' =None, project_id: 'str' =None, spec: 'V1JobSpec' =None, status: 'V1MultiMachineJobStatus' =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
         self._desired_state = None
         self._id = None
         self._machines = None
+        self._message = None
         self._name = None
         self._project_id = None
         self._spec = None
@@ -92,6 +95,8 @@ class V1MultiMachineJob(object):
             self.id = id
         if machines is not None:
             self.machines = machines
+        if message is not None:
+            self.message = message
         if name is not None:
             self.name = name
         if project_id is not None:
@@ -210,6 +215,27 @@ class V1MultiMachineJob(object):
 
         self._machines = machines
 
+    @property
+    def message(self) -> 'str':
+        """Gets the message of this V1MultiMachineJob.  # noqa: E501
+
+
+        :return: The message of this V1MultiMachineJob.  # noqa: E501
+        :rtype: str
+        """
+        return self._message
+
+    @message.setter
+    def message(self, message: 'str'):
+        """Sets the message of this V1MultiMachineJob.
+
+
+        :param message: The message of this V1MultiMachineJob.  # noqa: E501
+        :type: str
+        """
+
+        self._message = message
+
     @property
     def name(self) -> 'str':
         """Gets the name of this V1MultiMachineJob.  # noqa: E501

lightning_sdk/lightning_cloud/openapi/models/v1_multi_machine_job_state.py

@@ -39,9 +39,7 @@ class V1MultiMachineJobState(object):
     """
     UNSPECIFIED = "MultiMachineJob_STATE_UNSPECIFIED"
     RUNNING = "MultiMachineJob_STATE_RUNNING"
-    STOP = "MultiMachineJob_STATE_STOP"
     STOPPED = "MultiMachineJob_STATE_STOPPED"
-    DELETE = "MultiMachineJob_STATE_DELETE"
     DELETED = "MultiMachineJob_STATE_DELETED"
     FAILED = "MultiMachineJob_STATE_FAILED"
     """

lightning_sdk/models.py

@@ -0,0 +1,132 @@
+import os
+from pathlib import Path
+from typing import Any, Dict, List, Optional, Tuple, Union
+
+from lightning_sdk.api import OrgApi, TeamspaceApi, UserApi
+from lightning_sdk.lightning_cloud.openapi.models import V1Membership, V1OwnerType
+from lightning_sdk.lightning_cloud.openapi.rest import ApiException
+from lightning_sdk.teamspace import Teamspace
+from lightning_sdk.user import User
+from lightning_sdk.utils.resolve import _get_authed_user
+
+
+def _get_teamspace_and_path(
+    ts: V1Membership, org_api: OrgApi, user_api: UserApi, authed_user: User
+) -> Tuple[str, Dict[str, Any]]:
+    if ts.owner_type == V1OwnerType.ORGANIZATION:
+        org = org_api._get_org_by_id(ts.owner_id)
+        return f"{org.name}/{ts.name}", {"name": ts.name, "org": org.name}
+
+    if ts.owner_type == V1OwnerType.USER and ts.owner_id != authed_user.id:
+        user = user_api._get_user_by_id(ts.owner_id)  # todo: check also the name
+        return f"{user.username}/{ts.name}", {"name": ts.name, "user": User(name=user.username)}
+
+    if ts.owner_type == V1OwnerType.USER:
+        return f"{authed_user.name}/{ts.name}", {"name": ts.name, "user": authed_user}
+
+    raise RuntimeError(f"Unknown organization type {ts.owner_type}")
+
+
+def _list_teamspaces() -> List[str]:
+    org_api = OrgApi()
+    user_api = UserApi()
+    authed_user = _get_authed_user()
+
+    return [
+        _get_teamspace_and_path(ts, org_api, user_api, authed_user)[0]
+        for ts in user_api._get_all_teamspace_memberships("")
+    ]
+
+
+def _get_teamspace(name: str, organization: str) -> Teamspace:
+    """Get a Teamspace object from the SDK."""
+    org_api = OrgApi()
+    user_api = UserApi()
+    authed_user = _get_authed_user()
+
+    requested_teamspace = f"{organization}/{name}".lower()
+
+    for ts in user_api._get_all_teamspace_memberships(""):
+        if ts.name != name:
+            continue
+
+        teamspace_path, teamspace = _get_teamspace_and_path(ts, org_api, user_api, authed_user)
+        if requested_teamspace == teamspace_path:
+            return Teamspace(**teamspace)
+
+    options = f"{os.linesep}\t".join(_list_teamspaces())
+    raise RuntimeError(f"Teamspace `{requested_teamspace}` not found. Available teamspaces: {os.linesep}\t{options}")
+
+
+def _parse_model_name_and_version(name: str) -> Tuple[str, str, str, str]:
+    """Parse the name argument into its components."""
+    try:
+        org_name, teamspace_name, model_name = name.split("/")
+        parts = model_name.split(":")
+        if len(parts) == 1:
+            return org_name, teamspace_name, parts[0], "latest"
+        if len(parts) == 2:
+            return org_name, teamspace_name, parts[0], parts[1]
+        # The rest of the validation for name and version happens in the backend
+        raise ValueError(
+            "Model version is expected to be in the format `entity/modelname:version` separated by a"
+            f" single colon, but got: {name}"
+        )
+    except ValueError as err:
+        raise ValueError(
+            f"Model name must be in the format 'organization/teamspace/model' but you provided '{name}'."
+        ) from err
+
+
+def download_model(
+    name: str,
+    download_dir: Union[Path, str] = ".",
+    progress_bar: bool = True,
+) -> List[str]:
+    """Download a Model.
+
+    Args:
+        name: The name of the Model you want to download.
+        This should have the format <ORGANIZATION-NAME>/<TEAMSPACE-NAME>/<MODEL-NAME>.
+        download_dir: The directory where the Model should be downloaded.
+        progress_bar: Whether to show a progress bar when downloading.
+    """
+    teamspace_owner_name, teamspace_name, model_name, version = _parse_model_name_and_version(name)
+
+    download_dir = Path(download_dir)
+
+    api = TeamspaceApi()
+
+    try:
+        return api.download_model_files(
+            name=model_name,
+            version=version,
+            download_dir=download_dir,
+            teamspace_name=teamspace_name,
+            teamspace_owner_name=teamspace_owner_name,
+            progress_bar=progress_bar,
+        )
+    except ApiException as e:
+        # if we get an error, check if the teamspace actually exists (and print the list)
+        # TODO: ideally this would match a specific error about teamspace not being found
+        _ = _get_teamspace(name=teamspace_name, organization=teamspace_owner_name)
+        raise e
+
+
+def upload_model(name: str, path: Union[Path, str] = ".", cloud_account: Optional[str] = None) -> None:
+    """Upload a Model.
+
+    Args:
+        name: The name of the Model you want to upload.
+        This should have the format <ORGANIZATION-NAME>/<TEAMSPACE-NAME>/<MODEL-NAME>.
+        path: The path to the file or directory you want to upload. Defaults to the current directory.
+        cloud_account: The name of the cloud account to store the Model in.
+    """
+    org_name, teamspace_name, model_name, _ = _parse_model_name_and_version(name)
+    teamspace = _get_teamspace(name=teamspace_name, organization=org_name)
+    teamspace.upload_model(
+        path=path,
+        name=model_name,
+        progress_bar=True,
+        cluster_id=cloud_account,
+    )

lightning_sdk/teamspace.py

@@ -153,7 +153,7 @@ class Teamspace:
 
     def upload_model(
         self,
-        path: str,
+        path: Union[str, Path],
         name: str,
         progress_bar: bool = True,
         cluster_id: Optional[str] = None,
@@ -254,7 +254,8 @@ class Teamspace:
             name=name,
             version=version,
             download_dir=download_dir,
-            teamspace_id=self.id,
+            teamspace_name=self.name,
+            teamspace_owner_name=self.owner.name,
             progress_bar=progress_bar,
         )
 

{lightning_sdk-0.1.38.dist-info → lightning_sdk-0.1.39.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: lightning_sdk
-Version: 0.1.38
+Version: 0.1.39
 Summary: SDK to develop using Lightning AI Studios
 Author-email: Lightning-AI <justus@lightning.ai>
 License: MIT License

{lightning_sdk-0.1.38.dist-info → lightning_sdk-0.1.39.dist-info}/RECORD

@@ -1,44 +1,44 @@
 docs/source/conf.py,sha256=r8yX20eC-4mHhMTd0SbQb5TlSWHhO6wnJ0VJ_FBFpag,13249
-lightning_sdk/__init__.py,sha256=E_I8N6ba8nqKLuK5c-LJrL4x25I0kXkj2NYv91VtIC0,925
+lightning_sdk/__init__.py,sha256=PFDI_PWtaWboTMLPxqcwjLODg56mr8HJupB9Rd4GdQI,925
 lightning_sdk/agents.py,sha256=ly6Ma1j0ZgGPFyvPvMN28JWiB9dATIstFa5XM8pMi6I,1577
 lightning_sdk/ai_hub.py,sha256=V651b07qzxZmU6vKOs_GLkYZ_g1FY0iz-GlacPBCgPk,6526
 lightning_sdk/constants.py,sha256=ztl1PTUBULnqTf3DyKUSJaV_O20hNtUYT6XvAYIrmIk,749
 lightning_sdk/helpers.py,sha256=RnQwUquc_YPotjh6YXOoJvZs8krX_QFhd7kGv4U_spQ,1844
 lightning_sdk/machine.py,sha256=VdFXStR6ilYBEYuxgGWzcAw2TtW-nEQVsh6hz-2aaEw,750
+lightning_sdk/models.py,sha256=2BFfzfqcvBVUBHSn0q3MA9e8IsYhxkQRqvq5FXDma8Q,5091
 lightning_sdk/organization.py,sha256=WCfzdgjtvY1_A07DnxOpp74V2JR2gQwtXbIEcFDnoVU,1232
 lightning_sdk/owner.py,sha256=t5svD2it4C9pbSpVuG9WJL46CYi37JXNziwnXxhiU5U,1361
 lightning_sdk/plugin.py,sha256=2OgP8YtcLo2K1T0pypLGj3dkIaytTw6N11CHHQmRupY,13591
 lightning_sdk/status.py,sha256=kLDhN4-zdsGuZM577JMl1BbUIoF61bUOadW89ZAATFA,219
 lightning_sdk/studio.py,sha256=wvZrk07ZuG6Bf6c4XK8gMBumyiz_90JUScczdiWq1rU,16763
-lightning_sdk/teamspace.py,sha256=AiU88YfAZD44XfXraPUcEVA6qaq4EA0AiVYongGj-i4,10882
+lightning_sdk/teamspace.py,sha256=OjBdC9cR5oJ2OQmlTBVjLSHcsQIt_ltFo4B0xGKwwPU,10949
 lightning_sdk/user.py,sha256=vdn8pZqkAZO0-LoRsBdg0TckRKtd_H3QF4gpiZcl4iY,1130
 lightning_sdk/api/__init__.py,sha256=Qn2VVRvir_gO7w4yxGLkZY-R3T7kdiTPKgQ57BhIA9k,413
 lightning_sdk/api/agents_api.py,sha256=G47TbFo9kYqnBMqdw2RW-lfS1VAUBSXDmzs6fpIEMUs,4059
 lightning_sdk/api/ai_hub_api.py,sha256=_7BFcKYS7uYgCi5Q0Yzfz0mcm5LxezeZvgcDVRvsO9M,5571
-lightning_sdk/api/deployment_api.py,sha256=9HhOHz7ElmjgX88YkPUtETPgz_jMFkK1nTO18-jUniI,21505
-lightning_sdk/api/job_api.py,sha256=nWxpIepzuCYGo-u1DjSMfPB0lvyBYYxkMEiv0ozQEMw,8649
+lightning_sdk/api/deployment_api.py,sha256=LJGbpDPM2ZgP71Ja8rCTwsj2m9gruT91jLQ62a1emRc,21427
+lightning_sdk/api/job_api.py,sha256=jDbnsnIjClE2zQpi-VpzzaRihrkd5jFcks6GcLqIfWA,8840
 lightning_sdk/api/org_api.py,sha256=Ze3z_ATVrukobujV5YdC42DKj45Vuwl7X52q_Vr-o3U,803
 lightning_sdk/api/studio_api.py,sha256=ypgTAUJhwfdsEOJHfog4JsgZiLp9ZQA0KggSjLiI0ZA,26143
-lightning_sdk/api/teamspace_api.py,sha256=kMpQf8io-WhJ4Kd3pSOSe6nZTPqJ2hfUg4bimtmNmVk,10042
+lightning_sdk/api/teamspace_api.py,sha256=TgRhLdsCTLX21EWfGKcn5Y3LtaPVNzZ_eY8bUWL7YYk,10138
 lightning_sdk/api/user_api.py,sha256=sL7RIjjtmZmvCZWx7BBZslhj1BeNh4Idn-RVcdmf7M0,2598
-lightning_sdk/api/utils.py,sha256=RdrxnoreB6SnxSKVWDY0pPb6dZ9m_Y5nq32AAQBx01I,21560
+lightning_sdk/api/utils.py,sha256=3NLuTRME76xKqcYC1hQGUfWzDnb8YsnIYdg4kxKQC_Q,21659
 lightning_sdk/cli/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 lightning_sdk/cli/ai_hub.py,sha256=XcvTgAhm0eX8a5daQyF94pJooNwnygN5WzDTo35iSN0,1613
-lightning_sdk/cli/download.py,sha256=f_pXtLyPXH4r6PVzg7-39O7dd479PTU5iLB0PgTKM7k,6185
+lightning_sdk/cli/download.py,sha256=nyQN3q1vZ0fg4_cfit8cKaokQ9VUd46l_TNcAQWkLwU,5996
 lightning_sdk/cli/entrypoint.py,sha256=yCcRvmvZ9Lrkkq5AOWSVXMDtwU7ov2T6O9QYiVQLBwA,1385
 lightning_sdk/cli/exceptions.py,sha256=QUF3OMAMZwBikvlusimSHSBjb6ywvHpfAumJBEaodSw,169
 lightning_sdk/cli/legacy.py,sha256=ocTVNwlsLRS5aMjbMkwFPjT3uEYvS8C40CJ0PeRRv8g,4707
-lightning_sdk/cli/models.py,sha256=Op5v8XnjdpJksYhvRpW8UL8-XphidTrC0_DkOjZa5Wg,2580
-lightning_sdk/cli/run.py,sha256=gPCGz2FkHCPvmw8sQvPqHH2J3WeX2dtoWZUtDHC-pco,3801
+lightning_sdk/cli/run.py,sha256=rAgCdSZaU30pRIL3oNp3nHkl0_8-x8_qQd-WyW_gizw,5026
 lightning_sdk/cli/studios_menu.py,sha256=0kQGqGel8gAbpdJtjOM1a6NEat_TnIqRNprNn8QiK58,3236
-lightning_sdk/cli/upload.py,sha256=OvNgHTeQ92q2zfiGH0LmZAwGZeAGv1spVNbg-gLnJTA,9372
+lightning_sdk/cli/upload.py,sha256=H9OyipYTYAQ9Mzy2e8jtoaa-B34-uXHbTQTzY2Vmhv4,9078
 lightning_sdk/deployment/__init__.py,sha256=BLu7_cVLp97TYxe6qe-J1zKUSZXAVcvCjgcA7plV2k4,497
 lightning_sdk/deployment/deployment.py,sha256=47-aga7g8emJWrEkqr4FzfhMaNnsWhTSoinOl7FAMI0,15035
 lightning_sdk/job/__init__.py,sha256=1MxjQ6rHkyUHCypSW9RuXuVMVH11WiqhIXcU2LCFMwE,64
-lightning_sdk/job/base.py,sha256=cqErQX93yg4iJDdSCgTcRwtrWnhr3ZRqcTRwcIJG7Y8,5236
-lightning_sdk/job/job.py,sha256=dNUBbXlsTOgE1KNQ9RAGbDnWm68R6yFc8KlVko1MRfM,4455
-lightning_sdk/job/v1.py,sha256=Z_gYL0Smiz4B3vzRmXZMw7ojxkFDfc4L0-1SNoznqt8,5778
-lightning_sdk/job/v2.py,sha256=lxfBr9FwLvKAYJNh4fJPlnlDgzU-VmtDYVP71rQ7qCk,4548
+lightning_sdk/job/base.py,sha256=5Cv1OgQzL98Kl1RhZYLCV331zNnsRq7elYSaJR818ic,6438
+lightning_sdk/job/job.py,sha256=cUg8UZiYrzy96FhZYEdYCkZtgG3bim6IOvQtD2E2gG4,4854
+lightning_sdk/job/v1.py,sha256=tPU4fv00WMOR-l8_cTo7bqAPFRqXlTndfBGO1LZscfQ,6045
+lightning_sdk/job/v2.py,sha256=KH6QogvXHV7xnc_hsMF-NmnXX_467lAuWgs4U_LHxJ4,5123
 lightning_sdk/job/work.py,sha256=JNuFp6TaNvO-V-YoO4tDu8e8AFAkHZj3vzt_tqlD6KE,969
 lightning_sdk/lightning_cloud/__init__.py,sha256=o91SMAlwr4Ke5ESe8fHjqXcj31_h7rT-MlFoXA-n2EI,173
 lightning_sdk/lightning_cloud/__version__.py,sha256=lOfmWHtjmiuSG28TbKQqd2B3nwmSGOlKVFwhaj_cRJk,23
@@ -59,7 +59,7 @@ lightning_sdk/lightning_cloud/openapi/api/auth_service_api.py,sha256=NtdfsFZtQAl
 lightning_sdk/lightning_cloud/openapi/api/billing_service_api.py,sha256=SDJ80TezTWKhE3LR67OI6dhHDHABc4BfG7FjcFX7Ba4,68129
 lightning_sdk/lightning_cloud/openapi/api/cloud_space_service_api.py,sha256=DqHIqMzzOUWueLidgWq7JQs63_2iWXsN8c60XRtzyCw,370286
 lightning_sdk/lightning_cloud/openapi/api/cluster_service_api.py,sha256=11ketCae2QoKwniLAtDEqv_GDNgAJbyxV8rUYEQytnY,145754
-lightning_sdk/lightning_cloud/openapi/api/data_connection_service_api.py,sha256=bvoAUiLsdPvyYMEWWvcEDe9oarIfvLzxz_ZMoADDHOM,50181
+lightning_sdk/lightning_cloud/openapi/api/data_connection_service_api.py,sha256=OcnRdDs5PgAbmCLM0bdnUzJpTWb_OadztNua7AfVngQ,50422
 lightning_sdk/lightning_cloud/openapi/api/dataset_service_api.py,sha256=wCrcXOMuVs5rA4PHyrVJqpInO-Vy1hulddjCJUenPJc,23508
 lightning_sdk/lightning_cloud/openapi/api/deployment_templates_service_api.py,sha256=VK4ZuwoE0pvWEfSWr2KZNoutRtjovJFH3-6NqFmQjNg,31887
 lightning_sdk/lightning_cloud/openapi/api/endpoint_service_api.py,sha256=IgERziSMhxLxK5ATpsNpNs5Ybpu577mCH7f6yBCr7uw,113191
@@ -71,7 +71,7 @@ lightning_sdk/lightning_cloud/openapi/api/lightningapp_v2_service_api.py,sha256=
 lightning_sdk/lightning_cloud/openapi/api/lightningwork_service_api.py,sha256=dXKzbr4I7TDW-PFsqgiaaw4RtYrzB1X6NZB_hxgMnts,43747
 lightning_sdk/lightning_cloud/openapi/api/lit_logger_service_api.py,sha256=Fo2TsyFV2if4dGuesvQPJFZXw96Wr2hh_bTX8ocRwIE,67154
 lightning_sdk/lightning_cloud/openapi/api/lit_page_service_api.py,sha256=xf48UmWpzjs3woKVfFeyyqCLN13NA_qDepIwMw2SRUg,20457
-lightning_sdk/lightning_cloud/openapi/api/models_store_api.py,sha256=jH0ZmE4P32pZqHi_w3ybL71MhvKxt18ToCjETm9TDJE,71974
+lightning_sdk/lightning_cloud/openapi/api/models_store_api.py,sha256=REwcm9ilIetduPkThf0D6ZjQZMPtPy3KWV5mOBqfQdk,77357
 lightning_sdk/lightning_cloud/openapi/api/organizations_service_api.py,sha256=8xYTFSC4yDs32CTcEoWR1ycXX8chLQWRCsKAXJlOCls,88781
 lightning_sdk/lightning_cloud/openapi/api/profiler_service_api.py,sha256=hM2vTawBBuOYZrxN1SjZ6mU5ldvB_jUXRcLb0as0F4M,28286
 lightning_sdk/lightning_cloud/openapi/api/projects_service_api.py,sha256=a0x8UcufASV-CZzmIu-37iJiwhTlzalB07uctELD48Q,85065
@@ -148,7 +148,7 @@ lightning_sdk/lightning_cloud/openapi/models/id_publications_body.py,sha256=ya6b
 lightning_sdk/lightning_cloud/openapi/models/id_publications_body1.py,sha256=H-p5V2Ee0ae6meOGmv-Gm7hKeVFf4f5IREyFEumEMjc,3670
 lightning_sdk/lightning_cloud/openapi/models/id_release_body.py,sha256=j7p1BUSf74WKdWfkQv8ABwYx8egJVKf8wxxvC1R2KxY,3690
 lightning_sdk/lightning_cloud/openapi/models/id_reportlogsactivity_body.py,sha256=l4w-s7Q3VX-FKfSkUbDNA0eTmLPeXk_-k3PrzRAhviA,3830
-lightning_sdk/lightning_cloud/openapi/models/id_start_body.py,sha256=oXFItTT6ZcBhqPmXEJZEq1wUMEjlxRNKJO3jvuzkaD4,3900
+lightning_sdk/lightning_cloud/openapi/models/id_start_body.py,sha256=qySSqwn-O3pUA5kGoGGcr5YADSWUhY9P_EoXtgbltGw,4602
 lightning_sdk/lightning_cloud/openapi/models/id_storage_body.py,sha256=am89naxDX9PxhWQk9hLkuFlVb0WZp9UXmi0hSj5OP9I,4920
 lightning_sdk/lightning_cloud/openapi/models/id_uploads_body.py,sha256=4fKv6N6q5me42n6x9CXnVPlYjQ9uUj9N-kOSe71cXQw,3651
 lightning_sdk/lightning_cloud/openapi/models/id_uploads_body1.py,sha256=NqIds1ra2paAdJpjdMG4D4_z7RNx-omRpXdBTx6_ukc,5232
@@ -175,7 +175,7 @@ lightning_sdk/lightning_cloud/openapi/models/profiler_enabled_body.py,sha256=4Yn
 lightning_sdk/lightning_cloud/openapi/models/project_id_agentmanagedendpoints_body.py,sha256=pK1HjgWsXhU7OOKwYdt-SwklzF39N3Ap7Lyrk75T43k,4673
 lightning_sdk/lightning_cloud/openapi/models/project_id_agentmanagedmodels_body.py,sha256=jsMJS8werSAo1LLKdVjYs97wgL2FPIEvQSbbuyDB6mc,4553
 lightning_sdk/lightning_cloud/openapi/models/project_id_agents_body.py,sha256=oE3OGIhX0FfiKqE96YJmHnrRRAXgqrFsIAw840ib64Q,16193
-lightning_sdk/lightning_cloud/openapi/models/project_id_cloudspaces_body.py,sha256=RlOK35alRUHH63pt_ijqMpDhUPsCkZSGSQ7TqY6n-TE,14431
+lightning_sdk/lightning_cloud/openapi/models/project_id_cloudspaces_body.py,sha256=q36orD3j5-RjVaMtFp3q_DFD1ZbrHBnT2VLyciDsIeM,15176
 lightning_sdk/lightning_cloud/openapi/models/project_id_clusters_body.py,sha256=ewyBgf8oHlhgXKmn4YdjuyG84iONOeNXiWtbRQJfj4U,4348
 lightning_sdk/lightning_cloud/openapi/models/project_id_datasets_body.py,sha256=NkuH_enmOZ0MnYJ_hH2iVP1-Y1a9XinHR_c-5DaJxqU,15094
 lightning_sdk/lightning_cloud/openapi/models/project_id_endpoints_body.py,sha256=bnCgn3Jz3KE7blDn7chONuxJuc7Wg1eYM1QG0hxGR9U,7507
@@ -336,7 +336,7 @@ lightning_sdk/lightning_cloud/openapi/models/v1_data_connection.py,sha256=s3IUV6
 lightning_sdk/lightning_cloud/openapi/models/v1_data_connection_artifact.py,sha256=g91LNJXfc3oDV3JJEZ-Fyy-g19crePaZFnbYtj4NfkM,6824
 lightning_sdk/lightning_cloud/openapi/models/v1_data_connection_mount.py,sha256=NxMM-0hO9QjCw_0bOAsLa4DFZMM_ZFqc2XpeSOMu5P0,6814
 lightning_sdk/lightning_cloud/openapi/models/v1_data_connection_state.py,sha256=eSUVNEhmB-RKoFz2Aehur-IjPIBDd8Z2JPFY4-SSyog,3295
-lightning_sdk/lightning_cloud/openapi/models/v1_data_path.py,sha256=uPDS3qp5puZSZfOxAIrZaV4wG8toUCOUdw7uX_Fjs6I,5150
+lightning_sdk/lightning_cloud/openapi/models/v1_data_path.py,sha256=CJq2RJsuVd2wRwe52nYFmR5iZHrh1vQ8yMWWDLsIwv0,5892
 lightning_sdk/lightning_cloud/openapi/models/v1_dataset.py,sha256=wvuxddnly3oTU32gDj-RhvU1YqAFSHsvdCGZhl3vdB4,16987
 lightning_sdk/lightning_cloud/openapi/models/v1_dataset_type.py,sha256=ZMFPL9mcFlkzT_1ukZu4ndUoARLqsgsG6teZFPISKOc,3065
 lightning_sdk/lightning_cloud/openapi/models/v1_delete_affiliate_link_response.py,sha256=b9k7sRsVDMN9bvwxe5kAn9l9QCnV91wokgXEW4V0_b0,3064
@@ -504,7 +504,7 @@ lightning_sdk/lightning_cloud/openapi/models/v1_job_health_check_config.py,sha25
 lightning_sdk/lightning_cloud/openapi/models/v1_job_log_entry.py,sha256=Bhjx6rx3cJy6tarZ1zLXycZwmpZInMROecDo5ODsakg,4971
 lightning_sdk/lightning_cloud/openapi/models/v1_job_logs_page.py,sha256=uPeDC0SKPuojeSAgV-6Ei43A_nXkgxvQG1qSxd8w8To,6309
 lightning_sdk/lightning_cloud/openapi/models/v1_job_logs_response.py,sha256=9Ae7cFQTJHhI1XBsnHqNGQLFXEMeAJEw7T53jY4BMZg,4467
-lightning_sdk/lightning_cloud/openapi/models/v1_job_spec.py,sha256=N4fWyVPRH4ZZHSIIEwTr9a_phQlpnkq2I_qFi8CutA8,19260
+lightning_sdk/lightning_cloud/openapi/models/v1_job_spec.py,sha256=4DgXLTe8YkAxE6fhJRAKbeS7kz_pC6MyFo2VWM8Y21U,19031
 lightning_sdk/lightning_cloud/openapi/models/v1_job_timing.py,sha256=qnnGfuKbAU_IEWYZFPaBGTX2DFAvBvgoXllQk7yhMFk,5462
 lightning_sdk/lightning_cloud/openapi/models/v1_joinable_organization.py,sha256=IkuGZ89NWOMpe8JsxXJvkRKdcDUvV4F3akA_USvl5rs,9586
 lightning_sdk/lightning_cloud/openapi/models/v1_keep_alive_cloud_space_instance_response.py,sha256=NRrP0aqQbpd8ejQUApJ2L1vUoSx1UhNOlSh4JeQ2_es,3112
@@ -639,8 +639,8 @@ lightning_sdk/lightning_cloud/openapi/models/v1_metrics_tracker.py,sha256=b6ihJR
 lightning_sdk/lightning_cloud/openapi/models/v1_model.py,sha256=Vcg7XYYpTyC_Ri2tXBwHk4DSlh14v3D7tJckT-wUs0M,12630
 lightning_sdk/lightning_cloud/openapi/models/v1_model_version_archive.py,sha256=dZ_SM_IKbyRxjRSAmfdBWkLEMIWAGLQxYxBmwIjEUcM,12857
 lightning_sdk/lightning_cloud/openapi/models/v1_mount_target.py,sha256=7hWm0_z3weM2V6j900nUpCMceF3q-hXH1tVI1rama2U,5823
-lightning_sdk/lightning_cloud/openapi/models/v1_multi_machine_job.py,sha256=Jf4wZkYcy6u2AWvs-Poo6KCOevvhVk1xUvwRyABXudo,11005
-lightning_sdk/lightning_cloud/openapi/models/v1_multi_machine_job_state.py,sha256=WLRj0swuWjbdOtpqncMIhNsF9bYOMUB5h_jFn1RBkeE,3383
+lightning_sdk/lightning_cloud/openapi/models/v1_multi_machine_job.py,sha256=fYEWpNPNzAR60XRzn_oaczDG9U-yIFgR1g_u9gW2_Gg,11686
+lightning_sdk/lightning_cloud/openapi/models/v1_multi_machine_job_state.py,sha256=3P-RVvjEsR0gdwEfS7UoL9_j4da4yKzHouK4do7gwU0,3299
 lightning_sdk/lightning_cloud/openapi/models/v1_multi_machine_job_status.py,sha256=x_q3EoAcUK77tENFP0HbjoegkP8u1g2TQe0P62Ed-xY,8835
 lightning_sdk/lightning_cloud/openapi/models/v1_named_get_logger_metrics.py,sha256=VygKKjSPGIbWT9RxbmJfOSlNR-Fq12yKj8wT_SHd2ZQ,3966
 lightning_sdk/lightning_cloud/openapi/models/v1_network_config.py,sha256=bHDZVIsqgiSi7eBf72RunvrLR-d-4b9S7oDMAZFBvCM,5476
@@ -822,9 +822,9 @@ lightning_sdk/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSu
 lightning_sdk/utils/dynamic.py,sha256=glUTO1JC9APtQ6Gr9SO02a3zr56-sPAXM5C3NrTpgyQ,1959
 lightning_sdk/utils/enum.py,sha256=h2JRzqoBcSlUdanFHmkj_j5DleBHAu1esQYUsdNI-hU,4106
 lightning_sdk/utils/resolve.py,sha256=gU3MSko9Y7rE4jcnVwstNBaW83OFnSgvM-N44Ibrc_A,5148
-lightning_sdk-0.1.38.dist-info/LICENSE,sha256=uFIuZwj5z-4TeF2UuacPZ1o17HkvKObT8fY50qN84sg,1064
-lightning_sdk-0.1.38.dist-info/METADATA,sha256=oxfqzVjo2Kz5H6jemi-WJqW7cfpTD0lRk1WY6eienEQ,3920
-lightning_sdk-0.1.38.dist-info/WHEEL,sha256=P9jw-gEje8ByB7_hXoICnHtVCrEwMQh-630tKvQWehc,91
-lightning_sdk-0.1.38.dist-info/entry_points.txt,sha256=msB9PJWIJ784dX-OP8by51d4IbKYH3Fj1vCuA9oXjHY,68
-lightning_sdk-0.1.38.dist-info/top_level.txt,sha256=ps8doKILFXmN7F1mHncShmnQoTxKBRPIcchC8TpoBw4,19
-lightning_sdk-0.1.38.dist-info/RECORD,,
+lightning_sdk-0.1.39.dist-info/LICENSE,sha256=uFIuZwj5z-4TeF2UuacPZ1o17HkvKObT8fY50qN84sg,1064
+lightning_sdk-0.1.39.dist-info/METADATA,sha256=roqOUzNAM6XxDxtxhbzynlLVbbZAv6WKXgnY45BUOhs,3920
+lightning_sdk-0.1.39.dist-info/WHEEL,sha256=P9jw-gEje8ByB7_hXoICnHtVCrEwMQh-630tKvQWehc,91
+lightning_sdk-0.1.39.dist-info/entry_points.txt,sha256=msB9PJWIJ784dX-OP8by51d4IbKYH3Fj1vCuA9oXjHY,68
+lightning_sdk-0.1.39.dist-info/top_level.txt,sha256=ps8doKILFXmN7F1mHncShmnQoTxKBRPIcchC8TpoBw4,19
+lightning_sdk-0.1.39.dist-info/RECORD,,