lightning-sdk 0.1.32__py3-none-any.whl → 0.1.33__py3-none-any.whl

lightning_sdk/__init__.py

@@ -27,5 +27,5 @@ __all__ = [
     "AIHub",
 ]
 
-__version__ = "0.1.32"
+__version__ = "0.1.33"
 _check_version_and_prompt_upgrade(__version__)

lightning_sdk/ai_hub.py

@@ -1,6 +1,12 @@
-from typing import Dict, List
+from typing import TYPE_CHECKING, Any, Dict, List, Optional, Union
 
-from lightning_sdk.api.ai_hub_api import AIHubApi
+from lightning_sdk.api import AIHubApi, UserApi
+from lightning_sdk.lightning_cloud import login
+from lightning_sdk.user import User
+from lightning_sdk.utils.resolve import _resolve_org, _resolve_teamspace
+
+if TYPE_CHECKING:
+    from lightning_sdk import Organization, Teamspace
 
 
 class AIHub:
@@ -13,10 +19,17 @@ class AIHub:
 
     def __init__(self) -> None:
         self._api = AIHubApi()
+        self._auth = None
+
+    def list_apis(self, search: Optional[str] = None) -> List[Dict[str, str]]:
+        """Get a list of AI Hub API templates.
 
-    def list_apis(self) -> List[Dict[str, str]]:
-        """Get a list of AI Hub API templates."""
-        api_templates = self._api.list_apis()
+        Example:
+            api_hub = AIHub()
+            api_list = api_hub.list_apis(search="Llama")
+        """
+        search_query = search or ""
+        api_templates = self._api.list_apis(search_query=search_query)
         results = []
         for template in api_templates:
             result = {
@@ -24,6 +37,74 @@ class AIHub:
                 "name": template.name,
                 "description": template.description,
                 "creator_username": template.creator_username,
+                "created_on": template.creation_timestamp.strftime("%Y-%m-%d %H:%M:%S")
+                if template.creation_timestamp
+                else None,
             }
             results.append(result)
         return results
+
+    def _authenticate(
+        self,
+        teamspace: Optional[Union[str, "Teamspace"]] = None,
+        org: Optional[Union[str, "Organization"]] = None,
+        user: Optional[Union[str, "User"]] = None,
+    ) -> "Teamspace":
+        if self._auth is None:
+            self._auth = login.Auth()
+        try:
+            self._auth.authenticate()
+            user = User(name=UserApi()._get_user_by_id(self._auth.user_id).username)
+        except ConnectionError as e:
+            raise e
+
+        org = _resolve_org(org)
+        teamspace = _resolve_teamspace(teamspace=teamspace, org=org, user=user if org is None else None)
+        if teamspace is None:
+            raise ValueError("You need to pass a teamspace or an org for your deployment.")
+        return teamspace
+
+    def deploy(
+        self,
+        api_id: str,
+        cluster: Optional[str] = None,
+        name: Optional[str] = None,
+        teamspace: Optional[Union[str, "Teamspace"]] = None,
+        org: Optional[Union[str, "Organization"]] = None,
+        **kwargs: Dict[str, Any],
+    ) -> Dict[str, Union[str, bool]]:
+        """Deploy an API from the AI Hub.
+
+        Example:
+            from lightning_sdk import AIHub
+            ai_hub = AIHub()
+            deployment = ai_hub.deploy("temp_01jc37n6qpqkdptjpyep0z06hy", batch_size=10)
+
+        Args:
+            api_id: The ID of the API you want to deploy.
+            cluster: The cluster where you want to deploy the API, such as "lightning-public-prod".
+            name: Name for the deployed API. Defaults to None.
+            teamspace: The team or group for deployment. Defaults to None.
+            org: The organization for deployment. Defaults to None.
+            **kwargs: Additional keyword arguments for deployment.
+
+        Returns:
+            A dictionary containing the name of the deployed API,
+            the URL to access it, and whether it is interruptible.
+
+        Raises:
+            ValueError: If a teamspace or organization is not provided.
+            ConnectionError: If there is an issue with logging in.
+        """
+        teamspace = self._authenticate(teamspace, org)
+        teamspace_id = teamspace.id
+
+        deployment = self._api.deploy_api(
+            template_id=api_id, cluster_id=cluster, project_id=teamspace_id, name=name, **kwargs
+        )
+        return {
+            "id": deployment.id,
+            "name": deployment.name,
+            "base_url": deployment.status.urls[0],
+            "interruptible": deployment.spec.spot,
+        }

lightning_sdk/api/ai_hub_api.py

@@ -1,5 +1,11 @@
-from typing import List
+import re
+from typing import List, Optional
 
+from lightning_sdk.lightning_cloud.openapi.models import (
+    CreateDeploymentRequestDefinesASpecForTheJobThatAllowsForAutoscalingJobs,
+    V1Deployment,
+    V1ParameterizationSpec,
+)
 from lightning_sdk.lightning_cloud.openapi.models.v1_deployment_template_gallery_response import (
     V1DeploymentTemplateGalleryResponse,
 )
@@ -8,8 +14,48 @@ from lightning_sdk.lightning_cloud.rest_client import LightningClient
 
 class AIHubApi:
     def __init__(self) -> None:
-        self._client = LightningClient(max_tries=7)
+        self._client = LightningClient(max_tries=3)
 
-    def list_apis(self) -> List[V1DeploymentTemplateGalleryResponse]:
+    def list_apis(self, search_query: str) -> List[V1DeploymentTemplateGalleryResponse]:
         kwargs = {"show_globally_visible": True}
-        return self._client.deployment_templates_service_list_published_deployment_templates(**kwargs).templates
+        return self._client.deployment_templates_service_list_published_deployment_templates(
+            search_query=search_query, **kwargs
+        ).templates
+
+    @staticmethod
+    def _parse_and_update_args(cmd: str, **kwargs: dict) -> list:
+        """Parse the command and update the arguments with the provided kwargs.
+
+        >>> _parse_and_update_args("--arg1 1 --arg2=2", arg1=3)
+        ['--arg1 3']
+        """
+        keys = [key.lstrip("-") for key in re.findall(r"--\w+", cmd)]
+        arguments = {}
+        for key in keys:
+            if key in kwargs:
+                arguments[key] = kwargs[key]
+        return [f"--{k} {v}" for k, v in arguments.items()]
+
+    @staticmethod
+    def _resolve_api_arguments(parameter_spec: "V1ParameterizationSpec", **kwargs: dict) -> str:
+        return " ".join(AIHubApi._parse_and_update_args(parameter_spec.command, **kwargs))
+
+    def deploy_api(
+        self, template_id: str, project_id: str, cluster_id: str, name: Optional[str], **kwargs: dict
+    ) -> V1Deployment:
+        template = self._client.deployment_templates_service_get_deployment_template(template_id)
+        name = name or template.name
+        template.spec_v2.endpoint.id = None
+        command = self._resolve_api_arguments(template.parameter_spec, **kwargs)
+        template.spec_v2.job.command = command
+        return self._client.jobs_service_create_deployment(
+            project_id=project_id,
+            body=CreateDeploymentRequestDefinesASpecForTheJobThatAllowsForAutoscalingJobs(
+                autoscaling=template.spec_v2.autoscaling,
+                cluster_id=cluster_id,
+                endpoint=template.spec_v2.endpoint,
+                name=name,
+                replicas=0,
+                spec=template.spec_v2.job,
+            ),
+        )

lightning_sdk/api/studio_api.py

@@ -5,7 +5,7 @@ import time
 import warnings
 import zipfile
 from threading import Event, Thread
-from typing import Any, Dict, Mapping, Optional, Tuple
+from typing import Any, Dict, Mapping, Optional, Tuple, Union
 
 import backoff
 import requests
@@ -18,6 +18,7 @@ from lightning_sdk.api.utils import (
     _DummyBody,
     _DummyResponse,
     _FileUploader,
+    _machine_to_compute_name,
     _sanitize_studio_remote_path,
 )
 from lightning_sdk.api.utils import (
@@ -146,9 +147,11 @@ class StudioApi:
         """Retries checking the sync_in_progress value of the code status when there's an AttributeError."""
         return self.get_studio_status(studio_id, teamspace_id).in_use.sync_in_progress
 
-    def start_studio(self, studio_id: str, teamspace_id: str, machine: Machine, interruptible: False) -> None:
+    def start_studio(
+        self, studio_id: str, teamspace_id: str, machine: Union[Machine, str], interruptible: False
+    ) -> None:
         """Start an existing Studio."""
-        if machine == Machine.CPU_SMALL:
+        if _machine_to_compute_name(machine) == _machine_to_compute_name(Machine.CPU_SMALL):
             warnings.warn(
                 f"{Machine.CPU_SMALL} is not a valid machine for starting a Studio. "
                 "It is reserved for running jobs only. "
@@ -158,7 +161,7 @@ class StudioApi:
 
         self._client.cloud_space_service_start_cloud_space_instance(
             IdStartBody(
-                compute_config=V1UserRequestedComputeConfig(name=_MACHINE_TO_COMPUTE_NAME[machine], spot=interruptible)
+                compute_config=V1UserRequestedComputeConfig(name=_machine_to_compute_name(machine), spot=interruptible)
             ),
             teamspace_id,
             studio_id,
@@ -204,9 +207,11 @@ class StudioApi:
     def _get_studio_instance_status_from_object(self, studio: V1CloudSpace) -> Optional[str]:
         return getattr(getattr(studio.code_status, "in_use", None), "phase", None)
 
-    def _request_switch(self, studio_id: str, teamspace_id: str, machine: Machine, interruptible: bool) -> None:
+    def _request_switch(
+        self, studio_id: str, teamspace_id: str, machine: Union[Machine, str], interruptible: bool
+    ) -> None:
         """Switches given Studio to a new machine type."""
-        if machine == Machine.CPU_SMALL:
+        if _machine_to_compute_name(machine) == _machine_to_compute_name(Machine.CPU_SMALL):
             warnings.warn(
                 f"{Machine.CPU_SMALL} is not a valid machine for switching a Studio. "
                 "It is reserved for running jobs only. "
@@ -214,7 +219,7 @@ class StudioApi:
             )
             machine = Machine.CPU
 
-        compute_name = _MACHINE_TO_COMPUTE_NAME[machine]
+        compute_name = _machine_to_compute_name(machine)
         # TODO: UI sends disk size here, maybe we need to also?
         body = IdCodeconfigBody(compute_config=V1UserRequestedComputeConfig(name=compute_name, spot=interruptible))
         self._client.cloud_space_service_update_cloud_space_instance_config(
@@ -223,7 +228,9 @@ class StudioApi:
             body=body,
         )
 
-    def switch_studio_machine(self, studio_id: str, teamspace_id: str, machine: Machine, interruptible: bool) -> None:
+    def switch_studio_machine(
+        self, studio_id: str, teamspace_id: str, machine: Union[Machine, str], interruptible: bool
+    ) -> None:
         """Switches given Studio to a new machine type."""
         self._request_switch(
             studio_id=studio_id, teamspace_id=teamspace_id, machine=machine, interruptible=interruptible

lightning_sdk/api/utils.py

@@ -4,7 +4,7 @@ import os
 from concurrent.futures import ThreadPoolExecutor
 from functools import partial
 from pathlib import Path
-from typing import Any, Dict, List, Optional, Tuple
+from typing import Any, Dict, List, Optional, Tuple, Union
 
 import backoff
 import requests
@@ -340,6 +340,13 @@ _MACHINE_TO_COMPUTE_NAME: Dict[Machine, str] = {
     Machine.H200_X_8: "p5e.48xlarge",
 }
 
+
+def _machine_to_compute_name(machine: Union[Machine, str]) -> str:
+    if isinstance(machine, Machine):
+        return _MACHINE_TO_COMPUTE_NAME[machine]
+    return machine
+
+
 _COMPUTE_NAME_TO_MACHINE: Dict[str, Machine] = {v: k for k, v in _MACHINE_TO_COMPUTE_NAME.items()}
 
 _DEFAULT_CLOUD_URL = "https://lightning.ai:443"

lightning_sdk/lightning_cloud/openapi/models/create_deployment_request_defines_a_spec_for_the_job_that_allows_for_autoscaling_jobs.py

@@ -45,6 +45,7 @@ class CreateDeploymentRequestDefinesASpecForTheJobThatAllowsForAutoscalingJobs(o
         'cluster_id': 'str',
         'endpoint': 'V1Endpoint',
         'name': 'str',
+        'parameter_spec': 'V1ParameterizationSpec',
         'replicas': 'int',
         'spec': 'V1JobSpec',
         'strategy': 'V1DeploymentStrategy'
@@ -55,17 +56,19 @@ class CreateDeploymentRequestDefinesASpecForTheJobThatAllowsForAutoscalingJobs(o
         'cluster_id': 'clusterId',
         'endpoint': 'endpoint',
         'name': 'name',
+        'parameter_spec': 'parameterSpec',
         'replicas': 'replicas',
         'spec': 'spec',
         'strategy': 'strategy'
     }
 
-    def __init__(self, autoscaling: 'V1AutoscalingSpec' =None, cluster_id: 'str' =None, endpoint: 'V1Endpoint' =None, name: 'str' =None, replicas: 'int' =None, spec: 'V1JobSpec' =None, strategy: 'V1DeploymentStrategy' =None):  # noqa: E501
+    def __init__(self, autoscaling: 'V1AutoscalingSpec' =None, cluster_id: 'str' =None, endpoint: 'V1Endpoint' =None, name: 'str' =None, parameter_spec: 'V1ParameterizationSpec' =None, replicas: 'int' =None, spec: 'V1JobSpec' =None, strategy: 'V1DeploymentStrategy' =None):  # noqa: E501
         """CreateDeploymentRequestDefinesASpecForTheJobThatAllowsForAutoscalingJobs - a model defined in Swagger"""  # noqa: E501
         self._autoscaling = None
         self._cluster_id = None
         self._endpoint = None
         self._name = None
+        self._parameter_spec = None
         self._replicas = None
         self._spec = None
         self._strategy = None
@@ -78,6 +81,8 @@ class CreateDeploymentRequestDefinesASpecForTheJobThatAllowsForAutoscalingJobs(o
             self.endpoint = endpoint
         if name is not None:
             self.name = name
+        if parameter_spec is not None:
+            self.parameter_spec = parameter_spec
         if replicas is not None:
             self.replicas = replicas
         if spec is not None:
@@ -169,6 +174,27 @@ class CreateDeploymentRequestDefinesASpecForTheJobThatAllowsForAutoscalingJobs(o
 
         self._name = name
 
+    @property
+    def parameter_spec(self) -> 'V1ParameterizationSpec':
+        """Gets the parameter_spec of this CreateDeploymentRequestDefinesASpecForTheJobThatAllowsForAutoscalingJobs.  # noqa: E501
+
+
+        :return: The parameter_spec of this CreateDeploymentRequestDefinesASpecForTheJobThatAllowsForAutoscalingJobs.  # noqa: E501
+        :rtype: V1ParameterizationSpec
+        """
+        return self._parameter_spec
+
+    @parameter_spec.setter
+    def parameter_spec(self, parameter_spec: 'V1ParameterizationSpec'):
+        """Sets the parameter_spec of this CreateDeploymentRequestDefinesASpecForTheJobThatAllowsForAutoscalingJobs.
+
+
+        :param parameter_spec: The parameter_spec of this CreateDeploymentRequestDefinesASpecForTheJobThatAllowsForAutoscalingJobs.  # noqa: E501
+        :type: V1ParameterizationSpec
+        """
+
+        self._parameter_spec = parameter_spec
+
     @property
     def replicas(self) -> 'int':
         """Gets the replicas of this CreateDeploymentRequestDefinesASpecForTheJobThatAllowsForAutoscalingJobs.  # noqa: E501

lightning_sdk/lightning_cloud/openapi/models/deployments_id_body.py

@@ -47,6 +47,7 @@ class DeploymentsIdBody(object):
         'endpoint': 'V1Endpoint',
         'is_published': 'bool',
         'name': 'str',
+        'parameter_spec': 'V1ParameterizationSpec',
         'release_id': 'str',
         'replicas': 'int',
         'spec': 'V1JobSpec',
@@ -63,6 +64,7 @@ class DeploymentsIdBody(object):
         'endpoint': 'endpoint',
         'is_published': 'isPublished',
         'name': 'name',
+        'parameter_spec': 'parameterSpec',
         'release_id': 'releaseId',
         'replicas': 'replicas',
         'spec': 'spec',
@@ -72,7 +74,7 @@ class DeploymentsIdBody(object):
         'user_id': 'userId'
     }
 
-    def __init__(self, autoscaling: 'V1AutoscalingSpec' =None, created_at: 'datetime' =None, desired_state: 'V1DeploymentState' =None, endpoint: 'V1Endpoint' =None, is_published: 'bool' =None, name: 'str' =None, release_id: 'str' =None, replicas: 'int' =None, spec: 'V1JobSpec' =None, status: 'V1DeploymentStatus' =None, strategy: 'V1DeploymentStrategy' =None, updated_at: 'datetime' =None, user_id: 'str' =None):  # noqa: E501
+    def __init__(self, autoscaling: 'V1AutoscalingSpec' =None, created_at: 'datetime' =None, desired_state: 'V1DeploymentState' =None, endpoint: 'V1Endpoint' =None, is_published: 'bool' =None, name: 'str' =None, parameter_spec: 'V1ParameterizationSpec' =None, release_id: 'str' =None, replicas: 'int' =None, spec: 'V1JobSpec' =None, status: 'V1DeploymentStatus' =None, strategy: 'V1DeploymentStrategy' =None, updated_at: 'datetime' =None, user_id: 'str' =None):  # noqa: E501
         """DeploymentsIdBody - a model defined in Swagger"""  # noqa: E501
         self._autoscaling = None
         self._created_at = None
@@ -80,6 +82,7 @@ class DeploymentsIdBody(object):
         self._endpoint = None
         self._is_published = None
         self._name = None
+        self._parameter_spec = None
         self._release_id = None
         self._replicas = None
         self._spec = None
@@ -100,6 +103,8 @@ class DeploymentsIdBody(object):
             self.is_published = is_published
         if name is not None:
             self.name = name
+        if parameter_spec is not None:
+            self.parameter_spec = parameter_spec
         if release_id is not None:
             self.release_id = release_id
         if replicas is not None:
@@ -241,6 +246,27 @@ class DeploymentsIdBody(object):
 
         self._name = name
 
+    @property
+    def parameter_spec(self) -> 'V1ParameterizationSpec':
+        """Gets the parameter_spec of this DeploymentsIdBody.  # noqa: E501
+
+
+        :return: The parameter_spec of this DeploymentsIdBody.  # noqa: E501
+        :rtype: V1ParameterizationSpec
+        """
+        return self._parameter_spec
+
+    @parameter_spec.setter
+    def parameter_spec(self, parameter_spec: 'V1ParameterizationSpec'):
+        """Sets the parameter_spec of this DeploymentsIdBody.
+
+
+        :param parameter_spec: The parameter_spec of this DeploymentsIdBody.  # noqa: E501
+        :type: V1ParameterizationSpec
+        """
+
+        self._parameter_spec = parameter_spec
+
     @property
     def release_id(self) -> 'str':
         """Gets the release_id of this DeploymentsIdBody.  # noqa: E501

lightning_sdk/lightning_cloud/openapi/models/v1_deployment.py

@@ -48,6 +48,7 @@ class V1Deployment(object):
         'id': 'str',
         'is_published': 'bool',
         'name': 'str',
+        'parameter_spec': 'V1ParameterizationSpec',
         'project_id': 'str',
         'release_id': 'str',
         'replicas': 'int',
@@ -66,6 +67,7 @@ class V1Deployment(object):
         'id': 'id',
         'is_published': 'isPublished',
         'name': 'name',
+        'parameter_spec': 'parameterSpec',
         'project_id': 'projectId',
         'release_id': 'releaseId',
         'replicas': 'replicas',
@@ -76,7 +78,7 @@ class V1Deployment(object):
         'user_id': 'userId'
     }
 
-    def __init__(self, autoscaling: 'V1AutoscalingSpec' =None, created_at: 'datetime' =None, desired_state: 'V1DeploymentState' =None, endpoint: 'V1Endpoint' =None, id: 'str' =None, is_published: 'bool' =None, name: 'str' =None, project_id: 'str' =None, release_id: 'str' =None, replicas: 'int' =None, spec: 'V1JobSpec' =None, status: 'V1DeploymentStatus' =None, strategy: 'V1DeploymentStrategy' =None, updated_at: 'datetime' =None, user_id: 'str' =None):  # noqa: E501
+    def __init__(self, autoscaling: 'V1AutoscalingSpec' =None, created_at: 'datetime' =None, desired_state: 'V1DeploymentState' =None, endpoint: 'V1Endpoint' =None, id: 'str' =None, is_published: 'bool' =None, name: 'str' =None, parameter_spec: 'V1ParameterizationSpec' =None, project_id: 'str' =None, release_id: 'str' =None, replicas: 'int' =None, spec: 'V1JobSpec' =None, status: 'V1DeploymentStatus' =None, strategy: 'V1DeploymentStrategy' =None, updated_at: 'datetime' =None, user_id: 'str' =None):  # noqa: E501
         """V1Deployment - a model defined in Swagger"""  # noqa: E501
         self._autoscaling = None
         self._created_at = None
@@ -85,6 +87,7 @@ class V1Deployment(object):
         self._id = None
         self._is_published = None
         self._name = None
+        self._parameter_spec = None
         self._project_id = None
         self._release_id = None
         self._replicas = None
@@ -108,6 +111,8 @@ class V1Deployment(object):
             self.is_published = is_published
         if name is not None:
             self.name = name
+        if parameter_spec is not None:
+            self.parameter_spec = parameter_spec
         if project_id is not None:
             self.project_id = project_id
         if release_id is not None:
@@ -272,6 +277,27 @@ class V1Deployment(object):
 
         self._name = name
 
+    @property
+    def parameter_spec(self) -> 'V1ParameterizationSpec':
+        """Gets the parameter_spec of this V1Deployment.  # noqa: E501
+
+
+        :return: The parameter_spec of this V1Deployment.  # noqa: E501
+        :rtype: V1ParameterizationSpec
+        """
+        return self._parameter_spec
+
+    @parameter_spec.setter
+    def parameter_spec(self, parameter_spec: 'V1ParameterizationSpec'):
+        """Sets the parameter_spec of this V1Deployment.
+
+
+        :param parameter_spec: The parameter_spec of this V1Deployment.  # noqa: E501
+        :type: V1ParameterizationSpec
+        """
+
+        self._parameter_spec = parameter_spec
+
     @property
     def project_id(self) -> 'str':
         """Gets the project_id of this V1Deployment.  # noqa: E501

lightning_sdk/studio.py

@@ -3,6 +3,7 @@ import warnings
 from typing import TYPE_CHECKING, Any, Mapping, Optional, Tuple, Union
 
 from lightning_sdk.api.studio_api import StudioApi
+from lightning_sdk.api.utils import _machine_to_compute_name
 from lightning_sdk.constants import _LIGHTNING_DEBUG
 from lightning_sdk.machine import Machine
 from lightning_sdk.organization import Organization
@@ -145,14 +146,14 @@ class Studio:
         """Returns the cluster the Studio is running on."""
         return self._studio.cluster_id
 
-    def start(self, machine: Machine = Machine.CPU, interruptible: bool = False) -> None:
+    def start(self, machine: Union[Machine, str] = Machine.CPU, interruptible: bool = False) -> None:
         """Starts a Studio on the specified machine type (default: CPU-4)."""
         status = self.status
         if status == Status.Running:
-            curr_machine = self.machine
-            if curr_machine != machine:
+            curr_machine = _machine_to_compute_name(self.machine) if self.machine is not None else None
+            if curr_machine != _machine_to_compute_name(machine):
                 raise RuntimeError(
-                    f"Requested to start studio on {machine}, but studio is already running on {curr_machine}."
+                    f"Requested to start studio on {machine}, but studio is already running on {self.machine}."
                     " Consider switching instead!"
                 )
             _logger.info(f"Studio {self.name} is already running")
@@ -180,7 +181,7 @@ class Studio:
         kwargs = self._studio_api.duplicate_studio(self._studio.id, self._teamspace.id, self._teamspace.id)
         return Studio(**kwargs)
 
-    def switch_machine(self, machine: Machine, interruptible: bool = False) -> None:
+    def switch_machine(self, machine: Union[Machine, str], interruptible: bool = False) -> None:
         """Switches machine to the provided machine type/.
 
         Args:

{lightning_sdk-0.1.32.dist-info → lightning_sdk-0.1.33.dist-info}/METADATA

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

{lightning_sdk-0.1.32.dist-info → lightning_sdk-0.1.33.dist-info}/RECORD

@@ -1,7 +1,7 @@
 docs/source/conf.py,sha256=r8yX20eC-4mHhMTd0SbQb5TlSWHhO6wnJ0VJ_FBFpag,13249
-lightning_sdk/__init__.py,sha256=R_77kIsdEkN0Lxnf5ETmShr62qDXfjG_gi2HIeT9LmA,925
+lightning_sdk/__init__.py,sha256=rB9uPGPKZ4KsCOSgqGSglzmuOvrxBtIp_o_kdXHqlZM,925
 lightning_sdk/agents.py,sha256=ly6Ma1j0ZgGPFyvPvMN28JWiB9dATIstFa5XM8pMi6I,1577
-lightning_sdk/ai_hub.py,sha256=wVgLzqDzv0d3yWyoOEvE8epR97pWu-MoAYqysiO2evs,785
+lightning_sdk/ai_hub.py,sha256=yjCkfmJRKI9zpIn3qUZ7J5C1lccTt6UnKD6iv0xXsX0,3966
 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
@@ -9,19 +9,19 @@ lightning_sdk/organization.py,sha256=WCfzdgjtvY1_A07DnxOpp74V2JR2gQwtXbIEcFDnoVU
 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=ivwRwPSJtve2iLOwjatXKCtNNt-bAVkhEpB-GRqsQz4,16588
+lightning_sdk/studio.py,sha256=wvZrk07ZuG6Bf6c4XK8gMBumyiz_90JUScczdiWq1rU,16763
 lightning_sdk/teamspace.py,sha256=T3nxbZG1HtKu40F8gtpapqn-GVqSL6nBv7d32vmB_DY,10567
 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=DKVc80OY6Ehx1KoOEkHSnDZJxs6lh28C6I-4erJ2zew,572
+lightning_sdk/api/ai_hub_api.py,sha256=Gf4-0FXQROpIXiuQipp-pjLFGNuU-0hft7V_rpxg3Hc,2479
 lightning_sdk/api/deployment_api.py,sha256=9HhOHz7ElmjgX88YkPUtETPgz_jMFkK1nTO18-jUniI,21505
 lightning_sdk/api/job_api.py,sha256=pdQ3R5x04fyC49UqggWCIuPDLCmLYrnq791Ij-3vMr0,8457
 lightning_sdk/api/org_api.py,sha256=Ze3z_ATVrukobujV5YdC42DKj45Vuwl7X52q_Vr-o3U,803
-lightning_sdk/api/studio_api.py,sha256=es_ZpFwSDotYgWvJ4MunY7qEpSqQLcA7amTK9fG3tII,25924
+lightning_sdk/api/studio_api.py,sha256=ypgTAUJhwfdsEOJHfog4JsgZiLp9ZQA0KggSjLiI0ZA,26143
 lightning_sdk/api/teamspace_api.py,sha256=o-GBR3KLo298kDxO0myx-qlcCzSZnbR2OhZ73tt_5B8,9871
 lightning_sdk/api/user_api.py,sha256=sL7RIjjtmZmvCZWx7BBZslhj1BeNh4Idn-RVcdmf7M0,2598
-lightning_sdk/api/utils.py,sha256=Pnze_iy14cScQdOs8oELLXP-9WncpyigVEaohiT7uLM,20380
+lightning_sdk/api/utils.py,sha256=eHxq0dU_e9Pq5XjPlp7sNJ5_N3d44EtHKpbEiPPMkOQ,20562
 lightning_sdk/cli/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 lightning_sdk/cli/download.py,sha256=b6dpUMLhro6p0y6CleCHxggQnZlZMxCq0bjuf8B6_eA,6191
 lightning_sdk/cli/entrypoint.py,sha256=DD9rbmNqH6uE0IKMzfOenVymyrvfTMYi_zDPHr4Pk5E,1258
@@ -110,10 +110,10 @@ lightning_sdk/lightning_cloud/openapi/models/command_argument_command_argument_t
 lightning_sdk/lightning_cloud/openapi/models/conversations_id_body.py,sha256=blW66WBRE-qMCkzLCOrx4EqmM1vI42O3MvP9rIlx_tE,3651
 lightning_sdk/lightning_cloud/openapi/models/create.py,sha256=xrtxd9B0VkqfdszT5GXOmfD5PLmvPrbxTd2Ktjq55eE,14373
 lightning_sdk/lightning_cloud/openapi/models/create_checkout_session_request_wallet_type.py,sha256=6mUpEXvoZxyrnOvfT9PfF34E3OGvRMzMfLFndUe-jKU,3217
-lightning_sdk/lightning_cloud/openapi/models/create_deployment_request_defines_a_spec_for_the_job_that_allows_for_autoscaling_jobs.py,sha256=wGwtlmnWMhWMyJ2opvaV8llc7_tKPprpNjSOQTJeyVU,9970
+lightning_sdk/lightning_cloud/openapi/models/create_deployment_request_defines_a_spec_for_the_job_that_allows_for_autoscaling_jobs.py,sha256=tbgFc6advfcyZCmeTiGzPGdASHbOzV9jxUa3DPgaapA,11124
 lightning_sdk/lightning_cloud/openapi/models/data_connection_mount_data_connection_mount_copy_status.py,sha256=ytC9VwBzQMvpOSZJ78uekfrMmwGRaTOclX5XC3a-dMs,3343
 lightning_sdk/lightning_cloud/openapi/models/datasets_id_body.py,sha256=OcNzTWEbLGF36eY-yImNCmTwY4ZMX2P8yPuHAvc6KSk,5533
-lightning_sdk/lightning_cloud/openapi/models/deployments_id_body.py,sha256=OQJ_v3nlNxgPEs6A6JE4uSMv1qZsZqJfRGtSjo16vRM,12741
+lightning_sdk/lightning_cloud/openapi/models/deployments_id_body.py,sha256=CzAP4ulw4Q7nBFyLU3xHckqDLDKY7TRAB5ZaM64mV8o,13675
 lightning_sdk/lightning_cloud/openapi/models/deploymenttemplates_id_body.py,sha256=2TNCtt0YINExIsa6GYNNps5ton8QN-PPWSo6juG-oRo,14223
 lightning_sdk/lightning_cloud/openapi/models/endpoints_id_body.py,sha256=-Nv94Q8XoAT3tfYVhcjutpihfoNS2b2STbTHJr1gomo,12309
 lightning_sdk/lightning_cloud/openapi/models/experiment_name_variant_name_body.py,sha256=f-pBOnWAd03BreivwAA0D1P1wEBGq2jaC1nK-GtGxVA,3910
@@ -384,7 +384,7 @@ lightning_sdk/lightning_cloud/openapi/models/v1_delete_studio_job_response.py,sh
 lightning_sdk/lightning_cloud/openapi/models/v1_delete_user_slurm_job_response.py,sha256=kmy5RJKy63bRzBcWdMk00-vT3m2lVs05cic7jfU1JyQ,3058
 lightning_sdk/lightning_cloud/openapi/models/v1_dependency_cache_state.py,sha256=VL2MY_XvHWAd8toG-myqZlhuoEwjAIX1cemNSXxqk-k,3210
 lightning_sdk/lightning_cloud/openapi/models/v1_dependency_file_info.py,sha256=Fyh7eRcxFJ8Y4LHA3uJe_D5F9RhOxT4ChllHph_ryfQ,4571
-lightning_sdk/lightning_cloud/openapi/models/v1_deployment.py,sha256=7938YMxE3gyAJ6TyQogPiLu5PCwGKRPPJF74KG6n7lM,13732
+lightning_sdk/lightning_cloud/openapi/models/v1_deployment.py,sha256=gcZLLsjCX21_IanX_pmHC2TkDhFoH8feGKAd9sXhbwo,14646
 lightning_sdk/lightning_cloud/openapi/models/v1_deployment_event.py,sha256=XDpq8DM8sN2UKCIlBWGEk-vzjGQGNoc0KjyRcCzDqZ4,9994
 lightning_sdk/lightning_cloud/openapi/models/v1_deployment_event_type.py,sha256=uuogrGRymVVSz-OcYoJ7eZyFj3POXY22EAQJkXpTVm4,3197
 lightning_sdk/lightning_cloud/openapi/models/v1_deployment_metrics.py,sha256=xHhh0-aPxmyyH3g1Ks2h2oFnEy-CtZjDWH68jjetPEQ,3611
@@ -808,9 +808,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.32.dist-info/LICENSE,sha256=uFIuZwj5z-4TeF2UuacPZ1o17HkvKObT8fY50qN84sg,1064
-lightning_sdk-0.1.32.dist-info/METADATA,sha256=vxqkGNSCH82QlLfXB4ne_BdEdQWjqRpAZDlD02jAb3M,3941
-lightning_sdk-0.1.32.dist-info/WHEEL,sha256=P9jw-gEje8ByB7_hXoICnHtVCrEwMQh-630tKvQWehc,91
-lightning_sdk-0.1.32.dist-info/entry_points.txt,sha256=msB9PJWIJ784dX-OP8by51d4IbKYH3Fj1vCuA9oXjHY,68
-lightning_sdk-0.1.32.dist-info/top_level.txt,sha256=ps8doKILFXmN7F1mHncShmnQoTxKBRPIcchC8TpoBw4,19
-lightning_sdk-0.1.32.dist-info/RECORD,,
+lightning_sdk-0.1.33.dist-info/LICENSE,sha256=uFIuZwj5z-4TeF2UuacPZ1o17HkvKObT8fY50qN84sg,1064
+lightning_sdk-0.1.33.dist-info/METADATA,sha256=Q20ldoup46qvs8rCqTg3mCuO0e4si6qBOiXyWMwrbvs,3941
+lightning_sdk-0.1.33.dist-info/WHEEL,sha256=P9jw-gEje8ByB7_hXoICnHtVCrEwMQh-630tKvQWehc,91
+lightning_sdk-0.1.33.dist-info/entry_points.txt,sha256=msB9PJWIJ784dX-OP8by51d4IbKYH3Fj1vCuA9oXjHY,68
+lightning_sdk-0.1.33.dist-info/top_level.txt,sha256=ps8doKILFXmN7F1mHncShmnQoTxKBRPIcchC8TpoBw4,19
+lightning_sdk-0.1.33.dist-info/RECORD,,