lightning-sdk 0.2.4__py3-none-any.whl → 0.2.6__py3-none-any.whl

lightning_sdk/__init__.py

@@ -31,6 +31,6 @@ __all__ = [
     "User",
 ]
 
-__version__ = "0.2.4"
+__version__ = "0.2.6"
 _check_version_and_prompt_upgrade(__version__)
 _set_tqdm_envvars_noninteractive()

lightning_sdk/ai_hub.py

@@ -110,6 +110,7 @@ class AIHub:
         org: Optional[Union[str, "Organization"]] = None,
         user: Optional[Union[str, "User"]] = None,
         machine: Optional[Union[str, "Machine"]] = None,
+        quantity: Optional[int] = None,
     ) -> V1Deployment:
         """Deploy an API from the AI Hub.
 
@@ -127,7 +128,7 @@ class AIHub:
             api_arguments: Additional API argument, such as model name, or batch size.
             name: Name for the deployed API. Defaults to None.
             cloud_account: The cloud account where you want to run the template, such as "lightning-public-prod".
-                Defaults to None.
+                Defaults to None, which will use the cloud account of the teamspace.
             teamspace: The team or group for deployment. Defaults to None.
             org: The organization for deployment. Don't pass user with this. Defaults to None.
             user: The user for deployment. Don't pass org with this. Defaults to None.
@@ -158,6 +159,7 @@ class AIHub:
             name=name,
             api_arguments=api_arguments,
             machine=machine,
+            quantity=quantity,
         )
 
         url = (

lightning_sdk/api/ai_hub_api.py

@@ -1,4 +1,5 @@
 import traceback
+import warnings
 from typing import Dict, List, Optional, Tuple, Union
 
 import backoff
@@ -122,6 +123,7 @@ class AIHubApi:
         name: Optional[str],
         api_arguments: Dict[str, str],
         machine: Optional[Union[str, Machine]],
+        quantity: Optional[int],
     ) -> V1Deployment:
         template = self._client.deployment_templates_service_get_deployment_template(template_id)
         name = name or template.name
@@ -141,6 +143,19 @@ class AIHubApi:
             apply_change(template.spec_v2.job, "instance_name", machine)
             apply_change(template.spec_v2.job, "instance_type", machine)
 
+        if quantity != template.spec_v2.job.quantity:
+            # If the quantity is different from the published template quantity, override it with warnging
+            warnings.warn(
+                "Overriding the quantity of the template with the provided quantity. "
+                "This may result in unexpected behavior. "
+                f"Please verify the template (https://lightning.ai/lightning-ai/ai-hub/{template_id}) "
+                "and asscoiated parameters before running."
+            )
+            apply_change(template.spec_v2.job, "quantity", quantity)
+
+        # Override the cluster_id with the cloud_account if it is provided
+        if len(cloud_account) > 0:
+            apply_change(template.spec_v2.job, "cluster_id", cloud_account)
         return self._client.jobs_service_create_deployment(
             project_id=project_id,
             body=CreateDeploymentRequestDefinesASpecForTheJobThatAllowsForAutoscalingJobs(

lightning_sdk/api/studio_api.py

@@ -140,11 +140,6 @@ class StudioApi:
         startup_status = self.get_studio_status(studio_id, teamspace_id).in_use.startup_status
         return startup_status and startup_status.top_up_restore_finished
 
-    @backoff.on_exception(backoff.expo, AttributeError, max_tries=10)
-    def _check_code_status_sync_in_progress(self, studio_id: str, teamspace_id: str) -> bool:
-        """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: Union[Machine, str], interruptible: False
     ) -> None:
@@ -165,9 +160,6 @@ class StudioApi:
             studio_id,
         )
 
-        while self._check_code_status_sync_in_progress(studio_id, teamspace_id):
-            time.sleep(1)
-
         while True:
             if self._check_code_status_top_up_restore_finished(studio_id, teamspace_id):
                 break

lightning_sdk/cli/serve.py

@@ -1,5 +1,6 @@
 import os
 import subprocess
+from datetime import datetime
 from pathlib import Path
 from typing import Optional, Union
 
@@ -7,11 +8,11 @@ import click
 from rich.console import Console
 from rich.progress import Progress, SpinnerColumn, TextColumn, TimeElapsedColumn
 from rich.prompt import Confirm
+from rich.syntax import Syntax
 
 from lightning_sdk import Machine, Teamspace
 from lightning_sdk.api.deployment_api import DeploymentApi
 from lightning_sdk.api.lit_container_api import LitContainerApi
-from lightning_sdk.cli.exceptions import StudioCliError
 from lightning_sdk.cli.teamspace_menu import _TeamspacesMenu
 from lightning_sdk.serve import _LitServeDeployer
 
@@ -109,7 +110,7 @@ def api(
     script_path: str,
     easy: bool,
     cloud: bool,
-    name: str,
+    name: Optional[str],
     non_interactive: bool,
     machine: str,
     interruptible: bool,
@@ -137,6 +138,7 @@ def api(
         user=user,
         cloud_account=cloud_account,
         port=port,
+        replicas=replicas,
         min_replica=min_replica,
         max_replica=max_replica,
         include_credentials=not no_credentials,
@@ -172,6 +174,10 @@ def api_impl(
 
     _LitServeDeployer.generate_client() if easy else None
 
+    if not repository:
+        timestr = datetime.now().strftime("%b-%d-%H_%M")
+        repository = f"litserve-{timestr}".lower()
+
     if cloud:
         repository = repository or "litserve-model"
         machine = Machine.from_str(machine)
@@ -223,12 +229,30 @@ def _handle_cloud(
     replicas: Optional[int] = 1,
     include_credentials: Optional[bool] = True,
 ) -> None:
+    deployment_name = os.path.basename(repository)
+    tag = tag if tag else "latest"
     if teamspace is None:
         menu = _TeamspacesMenu()
         resolved_teamspace = menu._resolve_teamspace(teamspace)
     else:
         resolved_teamspace = Teamspace(name=teamspace, org=org, user=user)
 
+    if DeploymentApi().get_deployment_by_name(deployment_name, resolved_teamspace.id):
+        syntax = Syntax(
+            "from lightning_sdk import Deployment\n\n"
+            f"deployment = Deployment('{deployment_name}', teamspace={resolved_teamspace.name})\n"
+            "deployment.update(...)",
+            "python",
+            line_numbers=True,
+        )
+
+        console.print(
+            f"\nDeployment with name [b]{deployment_name}[/b] already running. "
+            "To update the deployment, use the Deployment API:\n",
+            syntax,
+        )
+        return
+
     port = port or 8000
     ls_deployer = _LitServeDeployer()
     path = ls_deployer.dockerize_api(script_path, port=port, gpu=not machine.is_cpu(), tag=tag, print_success=False)
@@ -242,14 +266,10 @@ def _handle_cloud(
         console.print("Please fix the Dockerfile and try again.", style="red")
         return
 
-    tag = tag if tag else "latest"
-
-    lit_cr = LitContainerApi()
-    deployment_name = os.path.basename(repository)
-
     ls_deployer.authenticate()
-    if DeploymentApi().get_deployment_by_name(deployment_name, resolved_teamspace.id):
-        raise StudioCliError(f"Deployment {deployment_name} already exists. Please choose a different name.") from None
+    # list containers to create the project if it doesn't exist
+    lit_cr = LitContainerApi()
+    lit_cr.list_containers(resolved_teamspace.id)
 
     with Progress(
         SpinnerColumn(),

lightning_sdk/deployment/deployment.py

@@ -32,7 +32,7 @@ from lightning_sdk.organization import Organization
 from lightning_sdk.services.utilities import _get_cluster
 from lightning_sdk.teamspace import Teamspace
 from lightning_sdk.user import User
-from lightning_sdk.utils.resolve import _resolve_deprecated_cluster, _resolve_teamspace, _resolve_user
+from lightning_sdk.utils.resolve import _resolve_deprecated_cluster, _resolve_org, _resolve_teamspace, _resolve_user
 
 
 class Deployment:
@@ -74,6 +74,7 @@ class Deployment:
 
         self._name = name
         self._user = _resolve_user(self._user or user)
+        self._org = _resolve_org(org)
 
         self._teamspace = _resolve_teamspace(
             teamspace=teamspace,
@@ -96,6 +97,8 @@ class Deployment:
             self._name = deployment.name
             self._is_created = True
             self._deployment = deployment
+        else:
+            self._deployment = None
 
     def start(
         self,
@@ -235,6 +238,13 @@ class Deployment:
         """All the deployment replicas will be stopped and all their traffic blocked."""
         self._deployment = self._deployment_api.stop(self._deployment)
 
+    @property
+    def name(self) -> Optional[str]:
+        if self._deployment:
+            self._deployment = self._deployment_api.get_deployment_by_name(self._name, self._teamspace.id)
+            self._name = self._deployment.name
+        return self._name
+
     @property
     def replicas(self) -> Optional[int]:
         """The default number of replicas the release starts with."""
@@ -248,7 +258,7 @@ class Deployment:
         """The minimum number of replicas."""
         if self._deployment:
             self._deployment = self._deployment_api.get_deployment_by_name(self._name, self._teamspace.id)
-            return self._deployment.autoscaling.min_replicas
+            return int(self._deployment.autoscaling.min_replicas)
         return None
 
     @property
@@ -256,7 +266,7 @@ class Deployment:
         """The maximum number of replicas."""
         if self._deployment:
             self._deployment = self._deployment_api.get_deployment_by_name(self._name, self._teamspace.id)
-            return self._deployment.autoscaling.max_replicas
+            return int(self._deployment.autoscaling.max_replicas)
         return None
 
     @property
@@ -371,9 +381,13 @@ class Deployment:
             return self._deployment.spec.include_credentials
         return None
 
+    @property
+    def org(self) -> Optional[Organization]:
+        return self._org
+
     @property
     def user(self) -> Optional[User]:
-        """The teamspace of the deployment."""
+        """The user of the deployment."""
         return self._user
 
     @property
@@ -385,6 +399,20 @@ class Deployment:
     def is_started(self) -> bool:
         return self._is_created
 
+    @property
+    def is_stopped(self) -> Optional[bool]:
+        if self._deployment:
+            self._deployment = self._deployment_api.get_deployment_by_name(self._name, self._teamspace.id)
+            return self._deployment.autoscaling.max_replicas == 0
+        return None
+
+    @property
+    def image(self) -> Optional[str]:
+        if self._deployment:
+            self._deployment = self._deployment_api.get_deployment_by_name(self._name, self._teamspace.id)
+            return self._deployment.spec.image
+        return None
+
     @property
     def _session(self) -> Any:
         if self._request_session is None:

lightning_sdk/lightning_cloud/openapi/__init__.py

@@ -27,6 +27,7 @@ from lightning_sdk.lightning_cloud.openapi.api.analytics_service_api import Anal
 from lightning_sdk.lightning_cloud.openapi.api.assistants_service_api import AssistantsServiceApi
 from lightning_sdk.lightning_cloud.openapi.api.auth_service_api import AuthServiceApi
 from lightning_sdk.lightning_cloud.openapi.api.billing_service_api import BillingServiceApi
+from lightning_sdk.lightning_cloud.openapi.api.cloud_space_environment_template_service_api import CloudSpaceEnvironmentTemplateServiceApi
 from lightning_sdk.lightning_cloud.openapi.api.cloud_space_service_api import CloudSpaceServiceApi
 from lightning_sdk.lightning_cloud.openapi.api.cluster_service_api import ClusterServiceApi
 from lightning_sdk.lightning_cloud.openapi.api.data_connection_service_api import DataConnectionServiceApi
@@ -39,16 +40,19 @@ from lightning_sdk.lightning_cloud.openapi.api.jobs_service_api import JobsServi
 from lightning_sdk.lightning_cloud.openapi.api.lightningapp_instance_service_api import LightningappInstanceServiceApi
 from lightning_sdk.lightning_cloud.openapi.api.lightningapp_v2_service_api import LightningappV2ServiceApi
 from lightning_sdk.lightning_cloud.openapi.api.lightningwork_service_api import LightningworkServiceApi
+from lightning_sdk.lightning_cloud.openapi.api.lit_dataset_service_api import LitDatasetServiceApi
 from lightning_sdk.lightning_cloud.openapi.api.lit_logger_service_api import LitLoggerServiceApi
 from lightning_sdk.lightning_cloud.openapi.api.lit_page_service_api import LitPageServiceApi
 from lightning_sdk.lightning_cloud.openapi.api.lit_registry_service_api import LitRegistryServiceApi
 from lightning_sdk.lightning_cloud.openapi.api.models_store_api import ModelsStoreApi
 from lightning_sdk.lightning_cloud.openapi.api.organizations_service_api import OrganizationsServiceApi
+from lightning_sdk.lightning_cloud.openapi.api.pipeline_templates_service_api import PipelineTemplatesServiceApi
 from lightning_sdk.lightning_cloud.openapi.api.pipelines_service_api import PipelinesServiceApi
 from lightning_sdk.lightning_cloud.openapi.api.profiler_service_api import ProfilerServiceApi
 from lightning_sdk.lightning_cloud.openapi.api.projects_service_api import ProjectsServiceApi
 from lightning_sdk.lightning_cloud.openapi.api.quest_service_api import QuestServiceApi
 from lightning_sdk.lightning_cloud.openapi.api.ssh_public_key_service_api import SSHPublicKeyServiceApi
+from lightning_sdk.lightning_cloud.openapi.api.schedules_service_api import SchedulesServiceApi
 from lightning_sdk.lightning_cloud.openapi.api.secret_service_api import SecretServiceApi
 from lightning_sdk.lightning_cloud.openapi.api.slurm_jobs_user_service_api import SlurmJobsUserServiceApi
 from lightning_sdk.lightning_cloud.openapi.api.snowflake_service_api import SnowflakeServiceApi
@@ -81,6 +85,7 @@ from lightning_sdk.lightning_cloud.openapi.models.cloudspaces_id_body import Clo
 from lightning_sdk.lightning_cloud.openapi.models.cluster_id_capacityblock_body import ClusterIdCapacityblockBody
 from lightning_sdk.lightning_cloud.openapi.models.cluster_id_capacityreservations_body import ClusterIdCapacityreservationsBody
 from lightning_sdk.lightning_cloud.openapi.models.cluster_id_proxies_body import ClusterIdProxiesBody
+from lightning_sdk.lightning_cloud.openapi.models.cluster_id_slurmusers_body import ClusterIdSlurmusersBody
 from lightning_sdk.lightning_cloud.openapi.models.cluster_id_usagerestrictions_body import ClusterIdUsagerestrictionsBody
 from lightning_sdk.lightning_cloud.openapi.models.clusters_id_body import ClustersIdBody
 from lightning_sdk.lightning_cloud.openapi.models.clusters_id_body1 import ClustersIdBody1
@@ -90,10 +95,13 @@ from lightning_sdk.lightning_cloud.openapi.models.create import Create
 from lightning_sdk.lightning_cloud.openapi.models.create_checkout_session_request_wallet_type import CreateCheckoutSessionRequestWalletType
 from lightning_sdk.lightning_cloud.openapi.models.create_deployment_request_defines_a_spec_for_the_job_that_allows_for_autoscaling_jobs import CreateDeploymentRequestDefinesASpecForTheJobThatAllowsForAutoscalingJobs
 from lightning_sdk.lightning_cloud.openapi.models.data_connection_mount_data_connection_mount_copy_status import DataConnectionMountDataConnectionMountCopyStatus
+from lightning_sdk.lightning_cloud.openapi.models.dataset_id_versions_body import DatasetIdVersionsBody
+from lightning_sdk.lightning_cloud.openapi.models.dataset_id_visibility_body import DatasetIdVisibilityBody
 from lightning_sdk.lightning_cloud.openapi.models.datasets_id_body import DatasetsIdBody
 from lightning_sdk.lightning_cloud.openapi.models.deployments_id_body import DeploymentsIdBody
 from lightning_sdk.lightning_cloud.openapi.models.deploymenttemplates_id_body import DeploymenttemplatesIdBody
 from lightning_sdk.lightning_cloud.openapi.models.endpoints_id_body import EndpointsIdBody
+from lightning_sdk.lightning_cloud.openapi.models.environmenttemplates_id_body import EnvironmenttemplatesIdBody
 from lightning_sdk.lightning_cloud.openapi.models.experiment_name_variant_name_body import ExperimentNameVariantNameBody
 from lightning_sdk.lightning_cloud.openapi.models.externalv1_cloud_space_instance_status import Externalv1CloudSpaceInstanceStatus
 from lightning_sdk.lightning_cloud.openapi.models.externalv1_cluster import Externalv1Cluster
@@ -137,6 +145,7 @@ from lightning_sdk.lightning_cloud.openapi.models.jobs_id_body import JobsIdBody
 from lightning_sdk.lightning_cloud.openapi.models.jobs_id_body1 import JobsIdBody1
 from lightning_sdk.lightning_cloud.openapi.models.jobs_id_body2 import JobsIdBody2
 from lightning_sdk.lightning_cloud.openapi.models.jobs_id_body3 import JobsIdBody3
+from lightning_sdk.lightning_cloud.openapi.models.litdatasets_dataset_id_body import LitdatasetsDatasetIdBody
 from lightning_sdk.lightning_cloud.openapi.models.litloggermetrics_id_body import LitloggermetricsIdBody
 from lightning_sdk.lightning_cloud.openapi.models.litpages_id_body import LitpagesIdBody
 from lightning_sdk.lightning_cloud.openapi.models.litregistry_lit_repo_name_body import LitregistryLitRepoNameBody
@@ -154,6 +163,7 @@ from lightning_sdk.lightning_cloud.openapi.models.org_id_memberships_body import
 from lightning_sdk.lightning_cloud.openapi.models.org_id_roles_body import OrgIdRolesBody
 from lightning_sdk.lightning_cloud.openapi.models.orgs_id_body import OrgsIdBody
 from lightning_sdk.lightning_cloud.openapi.models.pipelines_id_body import PipelinesIdBody
+from lightning_sdk.lightning_cloud.openapi.models.pipelinetemplates_id_body import PipelinetemplatesIdBody
 from lightning_sdk.lightning_cloud.openapi.models.profiler_captures_body import ProfilerCapturesBody
 from lightning_sdk.lightning_cloud.openapi.models.profiler_enabled_body import ProfilerEnabledBody
 from lightning_sdk.lightning_cloud.openapi.models.project_id_agentmanagedendpoints_body import ProjectIdAgentmanagedendpointsBody
@@ -166,12 +176,14 @@ from lightning_sdk.lightning_cloud.openapi.models.project_id_fileendpoints_body
 from lightning_sdk.lightning_cloud.openapi.models.project_id_getapp_body import ProjectIdGetappBody
 from lightning_sdk.lightning_cloud.openapi.models.project_id_invite_body import ProjectIdInviteBody
 from lightning_sdk.lightning_cloud.openapi.models.project_id_jobs_body import ProjectIdJobsBody
+from lightning_sdk.lightning_cloud.openapi.models.project_id_litdatasets_body import ProjectIdLitdatasetsBody
 from lightning_sdk.lightning_cloud.openapi.models.project_id_litregistry_body import ProjectIdLitregistryBody
 from lightning_sdk.lightning_cloud.openapi.models.project_id_memberships_body import ProjectIdMembershipsBody
 from lightning_sdk.lightning_cloud.openapi.models.project_id_models_body import ProjectIdModelsBody
 from lightning_sdk.lightning_cloud.openapi.models.project_id_multimachinejobs_body import ProjectIdMultimachinejobsBody
 from lightning_sdk.lightning_cloud.openapi.models.project_id_pipelines_body import ProjectIdPipelinesBody
 from lightning_sdk.lightning_cloud.openapi.models.project_id_projectclustersbindings_body import ProjectIdProjectclustersbindingsBody
+from lightning_sdk.lightning_cloud.openapi.models.project_id_schedules_body import ProjectIdSchedulesBody
 from lightning_sdk.lightning_cloud.openapi.models.project_id_secrets_body import ProjectIdSecretsBody
 from lightning_sdk.lightning_cloud.openapi.models.project_id_serviceexecution_body import ProjectIdServiceexecutionBody
 from lightning_sdk.lightning_cloud.openapi.models.project_id_snowflake_body import ProjectIdSnowflakeBody
@@ -181,8 +193,10 @@ from lightning_sdk.lightning_cloud.openapi.models.projects_project_id_body impor
 from lightning_sdk.lightning_cloud.openapi.models.protobuf_any import ProtobufAny
 from lightning_sdk.lightning_cloud.openapi.models.query_query_id_body import QueryQueryIdBody
 from lightning_sdk.lightning_cloud.openapi.models.rpc_status import RpcStatus
+from lightning_sdk.lightning_cloud.openapi.models.schedules_id_body import SchedulesIdBody
 from lightning_sdk.lightning_cloud.openapi.models.secrets_id_body import SecretsIdBody
 from lightning_sdk.lightning_cloud.openapi.models.secrets_id_body1 import SecretsIdBody1
+from lightning_sdk.lightning_cloud.openapi.models.server_id_alerts_body import ServerIdAlertsBody
 from lightning_sdk.lightning_cloud.openapi.models.servers_server_id_body import ServersServerIdBody
 from lightning_sdk.lightning_cloud.openapi.models.service_artifact_artifact_kind import ServiceArtifactArtifactKind
 from lightning_sdk.lightning_cloud.openapi.models.service_health_service_status import ServiceHealthServiceStatus
@@ -198,7 +212,9 @@ from lightning_sdk.lightning_cloud.openapi.models.stream_result_of_v1_get_long_r
 from lightning_sdk.lightning_cloud.openapi.models.studioapp_jobs_body import StudioappJobsBody
 from lightning_sdk.lightning_cloud.openapi.models.update import Update
 from lightning_sdk.lightning_cloud.openapi.models.upload_id_complete_body import UploadIdCompleteBody
+from lightning_sdk.lightning_cloud.openapi.models.upload_id_complete_body1 import UploadIdCompleteBody1
 from lightning_sdk.lightning_cloud.openapi.models.upload_id_parts_body import UploadIdPartsBody
+from lightning_sdk.lightning_cloud.openapi.models.upload_id_parts_body1 import UploadIdPartsBody1
 from lightning_sdk.lightning_cloud.openapi.models.uploads_upload_id_body import UploadsUploadIdBody
 from lightning_sdk.lightning_cloud.openapi.models.uploads_upload_id_body1 import UploadsUploadIdBody1
 from lightning_sdk.lightning_cloud.openapi.models.usagerestrictions_id_body import UsagerestrictionsIdBody
@@ -257,6 +273,8 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_cloud_space_artifact_event_
 from lightning_sdk.lightning_cloud.openapi.models.v1_cloud_space_code_version import V1CloudSpaceCodeVersion
 from lightning_sdk.lightning_cloud.openapi.models.v1_cloud_space_code_version_status import V1CloudSpaceCodeVersionStatus
 from lightning_sdk.lightning_cloud.openapi.models.v1_cloud_space_engagement_response import V1CloudSpaceEngagementResponse
+from lightning_sdk.lightning_cloud.openapi.models.v1_cloud_space_environment_template import V1CloudSpaceEnvironmentTemplate
+from lightning_sdk.lightning_cloud.openapi.models.v1_cloud_space_environment_template_config import V1CloudSpaceEnvironmentTemplateConfig
 from lightning_sdk.lightning_cloud.openapi.models.v1_cloud_space_instance_collab_status import V1CloudSpaceInstanceCollabStatus
 from lightning_sdk.lightning_cloud.openapi.models.v1_cloud_space_instance_config import V1CloudSpaceInstanceConfig
 from lightning_sdk.lightning_cloud.openapi.models.v1_cloud_space_instance_startup_status import V1CloudSpaceInstanceStartupStatus
@@ -288,6 +306,8 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_collab_action import V1Coll
 from lightning_sdk.lightning_cloud.openapi.models.v1_collab_session import V1CollabSession
 from lightning_sdk.lightning_cloud.openapi.models.v1_command_argument import V1CommandArgument
 from lightning_sdk.lightning_cloud.openapi.models.v1_complete_agent_multipart_upload import V1CompleteAgentMultipartUpload
+from lightning_sdk.lightning_cloud.openapi.models.v1_complete_lit_dataset_multi_part_upload_response import V1CompleteLitDatasetMultiPartUploadResponse
+from lightning_sdk.lightning_cloud.openapi.models.v1_complete_lit_dataset_upload_response import V1CompleteLitDatasetUploadResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_complete_model_upload_response import V1CompleteModelUploadResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_complete_multi_part_upload_response import V1CompleteMultiPartUploadResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_complete_onboarding_response import V1CompleteOnboardingResponse
@@ -308,6 +328,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_create_billing_portal_sessi
 from lightning_sdk.lightning_cloud.openapi.models.v1_create_checkout_session_request import V1CreateCheckoutSessionRequest
 from lightning_sdk.lightning_cloud.openapi.models.v1_create_checkout_session_response import V1CreateCheckoutSessionResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_create_cloud_space_app_instance_response import V1CreateCloudSpaceAppInstanceResponse
+from lightning_sdk.lightning_cloud.openapi.models.v1_create_cloud_space_environment_template_request import V1CreateCloudSpaceEnvironmentTemplateRequest
 from lightning_sdk.lightning_cloud.openapi.models.v1_create_cloud_space_instance_metric_response import V1CreateCloudSpaceInstanceMetricResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_create_cluster_capacity_reservation_response import V1CreateClusterCapacityReservationResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_create_cluster_encryption_keys_request import V1CreateClusterEncryptionKeysRequest
@@ -317,14 +338,17 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_create_cluster_response imp
 from lightning_sdk.lightning_cloud.openapi.models.v1_create_deployment_request import V1CreateDeploymentRequest
 from lightning_sdk.lightning_cloud.openapi.models.v1_create_deployment_template_request import V1CreateDeploymentTemplateRequest
 from lightning_sdk.lightning_cloud.openapi.models.v1_create_job_request import V1CreateJobRequest
+from lightning_sdk.lightning_cloud.openapi.models.v1_create_lit_dataset_multi_part_upload_response import V1CreateLitDatasetMultiPartUploadResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_create_lit_page_request import V1CreateLitPageRequest
 from lightning_sdk.lightning_cloud.openapi.models.v1_create_lit_page_response import V1CreateLitPageResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_create_managed_endpoint_response import V1CreateManagedEndpointResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_create_multi_machine_job_request import V1CreateMultiMachineJobRequest
 from lightning_sdk.lightning_cloud.openapi.models.v1_create_multi_part_upload_response import V1CreateMultiPartUploadResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_create_organization_request import V1CreateOrganizationRequest
+from lightning_sdk.lightning_cloud.openapi.models.v1_create_pipeline_template_request import V1CreatePipelineTemplateRequest
 from lightning_sdk.lightning_cloud.openapi.models.v1_create_project_request import V1CreateProjectRequest
 from lightning_sdk.lightning_cloud.openapi.models.v1_create_ssh_public_key_request import V1CreateSSHPublicKeyRequest
+from lightning_sdk.lightning_cloud.openapi.models.v1_create_server_alert_response import V1CreateServerAlertResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_create_shared_metrics_stream_request import V1CreateSharedMetricsStreamRequest
 from lightning_sdk.lightning_cloud.openapi.models.v1_create_shared_metrics_stream_response import V1CreateSharedMetricsStreamResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_create_snowflake_connection_response import V1CreateSnowflakeConnectionResponse
@@ -342,6 +366,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_dataset_type import V1Datas
 from lightning_sdk.lightning_cloud.openapi.models.v1_delete_affiliate_link_response import V1DeleteAffiliateLinkResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_delete_assistant_response import V1DeleteAssistantResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_delete_cloud_space_app_response import V1DeleteCloudSpaceAppResponse
+from lightning_sdk.lightning_cloud.openapi.models.v1_delete_cloud_space_environment_template_response import V1DeleteCloudSpaceEnvironmentTemplateResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_delete_cloud_space_response import V1DeleteCloudSpaceResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_delete_cloud_space_session_response import V1DeleteCloudSpaceSessionResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_delete_cloud_space_version_publication_response import V1DeleteCloudSpaceVersionPublicationResponse
@@ -364,6 +389,8 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_delete_lightning_run_respon
 from lightning_sdk.lightning_cloud.openapi.models.v1_delete_lightningapp_instance_artifact_response import V1DeleteLightningappInstanceArtifactResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_delete_lightningapp_instance_response import V1DeleteLightningappInstanceResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_delete_lightningwork_response import V1DeleteLightningworkResponse
+from lightning_sdk.lightning_cloud.openapi.models.v1_delete_lit_dataset_response import V1DeleteLitDatasetResponse
+from lightning_sdk.lightning_cloud.openapi.models.v1_delete_lit_dataset_version_response import V1DeleteLitDatasetVersionResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_delete_lit_page_response import V1DeleteLitPageResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_delete_lit_repository_response import V1DeleteLitRepositoryResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_delete_logger_artifact_response import V1DeleteLoggerArtifactResponse
@@ -386,6 +413,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_delete_project_membership_r
 from lightning_sdk.lightning_cloud.openapi.models.v1_delete_project_response import V1DeleteProjectResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_delete_project_role_response import V1DeleteProjectRoleResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_delete_ssh_public_key_response import V1DeleteSSHPublicKeyResponse
+from lightning_sdk.lightning_cloud.openapi.models.v1_delete_schedule_response import V1DeleteScheduleResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_delete_secret_response import V1DeleteSecretResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_delete_service_execution_response import V1DeleteServiceExecutionResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_delete_shared_metrics_stream_response import V1DeleteSharedMetricsStreamResponse
@@ -473,6 +501,8 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_get_job_system_metrics_resp
 from lightning_sdk.lightning_cloud.openapi.models.v1_get_lightning_run_source_code_download_url_response import V1GetLightningRunSourceCodeDownloadUrlResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_get_lightningapp_instance_open_ports_response import V1GetLightningappInstanceOpenPortsResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_get_lightningapp_instance_system_metrics_response import V1GetLightningappInstanceSystemMetricsResponse
+from lightning_sdk.lightning_cloud.openapi.models.v1_get_lit_dataset_file_upload_urls_response import V1GetLitDatasetFileUploadUrlsResponse
+from lightning_sdk.lightning_cloud.openapi.models.v1_get_lit_dataset_files_url_response import V1GetLitDatasetFilesUrlResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_get_lit_page_response import V1GetLitPageResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_get_logger_metrics_response import V1GetLoggerMetricsResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_get_long_running_command_in_cloud_space_response import V1GetLongRunningCommandInCloudSpaceResponse
@@ -544,6 +574,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_list_agent_job_artifacts_re
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_agent_jobs_response import V1ListAgentJobsResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_assistants_response import V1ListAssistantsResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_cloud_space_apps_response import V1ListCloudSpaceAppsResponse
+from lightning_sdk.lightning_cloud.openapi.models.v1_list_cloud_space_environment_templates_response import V1ListCloudSpaceEnvironmentTemplatesResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_cloud_space_instances_response import V1ListCloudSpaceInstancesResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_cloud_space_publications_response import V1ListCloudSpacePublicationsResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_cloud_space_python_versions_response import V1ListCloudSpacePythonVersionsResponse
@@ -589,6 +620,8 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_list_lightningapp_instance_
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_lightningapp_instances_response import V1ListLightningappInstancesResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_lightningwork_events_response import V1ListLightningworkEventsResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_lightningwork_response import V1ListLightningworkResponse
+from lightning_sdk.lightning_cloud.openapi.models.v1_list_lit_dataset_versions_response import V1ListLitDatasetVersionsResponse
+from lightning_sdk.lightning_cloud.openapi.models.v1_list_lit_datasets_response import V1ListLitDatasetsResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_lit_pages_response import V1ListLitPagesResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_lit_registry_repository_image_artifact_versions_response import V1ListLitRegistryRepositoryImageArtifactVersionsResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_logger_artifact_response import V1ListLoggerArtifactResponse
@@ -619,12 +652,17 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_list_project_roles_response
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_published_cloud_spaces_response import V1ListPublishedCloudSpacesResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_published_deployment_templates_response import V1ListPublishedDeploymentTemplatesResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_quests_response import V1ListQuestsResponse
+from lightning_sdk.lightning_cloud.openapi.models.v1_list_slurm_cluster_users_response import V1ListSLURMClusterUsersResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_ssh_public_keys_response import V1ListSSHPublicKeysResponse
+from lightning_sdk.lightning_cloud.openapi.models.v1_list_schedules_response import V1ListSchedulesResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_secrets_response import V1ListSecretsResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_service_execution_lightningapp_instances_response import V1ListServiceExecutionLightningappInstancesResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_service_executions_response import V1ListServiceExecutionsResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_studio_jobs_response import V1ListStudioJobsResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_user_slurm_jobs_response import V1ListUserSLURMJobsResponse
+from lightning_sdk.lightning_cloud.openapi.models.v1_lit_dataset import V1LitDataset
+from lightning_sdk.lightning_cloud.openapi.models.v1_lit_dataset_file import V1LitDatasetFile
+from lightning_sdk.lightning_cloud.openapi.models.v1_lit_dataset_version_archive import V1LitDatasetVersionArchive
 from lightning_sdk.lightning_cloud.openapi.models.v1_lit_page import V1LitPage
 from lightning_sdk.lightning_cloud.openapi.models.v1_lit_page_type import V1LitPageType
 from lightning_sdk.lightning_cloud.openapi.models.v1_lit_registry_artifact import V1LitRegistryArtifact
@@ -686,10 +724,15 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_path_mapping import V1PathM
 from lightning_sdk.lightning_cloud.openapi.models.v1_path_telemetry import V1PathTelemetry
 from lightning_sdk.lightning_cloud.openapi.models.v1_phase_type import V1PhaseType
 from lightning_sdk.lightning_cloud.openapi.models.v1_pipeline import V1Pipeline
-from lightning_sdk.lightning_cloud.openapi.models.v1_pipeline_schedule import V1PipelineSchedule
+from lightning_sdk.lightning_cloud.openapi.models.v1_pipeline_parameter import V1PipelineParameter
+from lightning_sdk.lightning_cloud.openapi.models.v1_pipeline_parameter_placement import V1PipelineParameterPlacement
+from lightning_sdk.lightning_cloud.openapi.models.v1_pipeline_parameter_placement_type import V1PipelineParameterPlacementType
+from lightning_sdk.lightning_cloud.openapi.models.v1_pipeline_parameter_type import V1PipelineParameterType
 from lightning_sdk.lightning_cloud.openapi.models.v1_pipeline_step import V1PipelineStep
 from lightning_sdk.lightning_cloud.openapi.models.v1_pipeline_step_status import V1PipelineStepStatus
 from lightning_sdk.lightning_cloud.openapi.models.v1_pipeline_step_type import V1PipelineStepType
+from lightning_sdk.lightning_cloud.openapi.models.v1_pipeline_template import V1PipelineTemplate
+from lightning_sdk.lightning_cloud.openapi.models.v1_pipeline_template_visibility_type import V1PipelineTemplateVisibilityType
 from lightning_sdk.lightning_cloud.openapi.models.v1_plugin import V1Plugin
 from lightning_sdk.lightning_cloud.openapi.models.v1_plugins_list_response import V1PluginsListResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_post_cloud_space_artifact_events_response import V1PostCloudSpaceArtifactEventsResponse
@@ -730,6 +773,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_report_restart_timings_resp
 from lightning_sdk.lightning_cloud.openapi.models.v1_request_cluster_access_request import V1RequestClusterAccessRequest
 from lightning_sdk.lightning_cloud.openapi.models.v1_request_cluster_access_response import V1RequestClusterAccessResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_request_verification_code_response import V1RequestVerificationCodeResponse
+from lightning_sdk.lightning_cloud.openapi.models.v1_reservation_details import V1ReservationDetails
 from lightning_sdk.lightning_cloud.openapi.models.v1_resource_tag import V1ResourceTag
 from lightning_sdk.lightning_cloud.openapi.models.v1_resource_visibility import V1ResourceVisibility
 from lightning_sdk.lightning_cloud.openapi.models.v1_resources import V1Resources
@@ -749,12 +793,17 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_s3_folder_data_connection i
 from lightning_sdk.lightning_cloud.openapi.models.v1_slurm_job import V1SLURMJob
 from lightning_sdk.lightning_cloud.openapi.models.v1_ssh_key_pair import V1SSHKeyPair
 from lightning_sdk.lightning_cloud.openapi.models.v1_ssh_public_key import V1SSHPublicKey
+from lightning_sdk.lightning_cloud.openapi.models.v1_schedule import V1Schedule
+from lightning_sdk.lightning_cloud.openapi.models.v1_schedule_resource_type import V1ScheduleResourceType
 from lightning_sdk.lightning_cloud.openapi.models.v1_search_job_logs_response import V1SearchJobLogsResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_search_user import V1SearchUser
 from lightning_sdk.lightning_cloud.openapi.models.v1_search_users_response import V1SearchUsersResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_secret import V1Secret
 from lightning_sdk.lightning_cloud.openapi.models.v1_secret_type import V1SecretType
 from lightning_sdk.lightning_cloud.openapi.models.v1_select import V1Select
+from lightning_sdk.lightning_cloud.openapi.models.v1_server_alert import V1ServerAlert
+from lightning_sdk.lightning_cloud.openapi.models.v1_server_alert_severity import V1ServerAlertSeverity
+from lightning_sdk.lightning_cloud.openapi.models.v1_server_alert_type import V1ServerAlertType
 from lightning_sdk.lightning_cloud.openapi.models.v1_server_check_in_response import V1ServerCheckInResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_service_artifact import V1ServiceArtifact
 from lightning_sdk.lightning_cloud.openapi.models.v1_service_execution import V1ServiceExecution
@@ -764,6 +813,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_setup_data_connection_respo
 from lightning_sdk.lightning_cloud.openapi.models.v1_shared_filesystem import V1SharedFilesystem
 from lightning_sdk.lightning_cloud.openapi.models.v1_should_start_syncing_response import V1ShouldStartSyncingResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_signed_url import V1SignedUrl
+from lightning_sdk.lightning_cloud.openapi.models.v1_slurm_cluster_user import V1SlurmClusterUser
 from lightning_sdk.lightning_cloud.openapi.models.v1_slurm_node import V1SlurmNode
 from lightning_sdk.lightning_cloud.openapi.models.v1_slurm_v1 import V1SlurmV1
 from lightning_sdk.lightning_cloud.openapi.models.v1_slurm_v1_job_status import V1SlurmV1JobStatus
@@ -778,6 +828,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_storage_asset_type import V
 from lightning_sdk.lightning_cloud.openapi.models.v1_storage_system_metrics import V1StorageSystemMetrics
 from lightning_sdk.lightning_cloud.openapi.models.v1_studio_job import V1StudioJob
 from lightning_sdk.lightning_cloud.openapi.models.v1_studio_job_app import V1StudioJobApp
+from lightning_sdk.lightning_cloud.openapi.models.v1_subnet_spec import V1SubnetSpec
 from lightning_sdk.lightning_cloud.openapi.models.v1_switch_cloud_space_instance_response import V1SwitchCloudSpaceInstanceResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_system_info import V1SystemInfo
 from lightning_sdk.lightning_cloud.openapi.models.v1_system_metrics import V1SystemMetrics
@@ -801,6 +852,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_update_cluster_availability
 from lightning_sdk.lightning_cloud.openapi.models.v1_update_deployment_visibility_response import V1UpdateDeploymentVisibilityResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_update_index_response import V1UpdateIndexResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_update_job_visibility_response import V1UpdateJobVisibilityResponse
+from lightning_sdk.lightning_cloud.openapi.models.v1_update_lit_dataset_visibility_response import V1UpdateLitDatasetVisibilityResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_update_lit_page_response import V1UpdateLitPageResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_update_lit_repository_response import V1UpdateLitRepositoryResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_update_metrics_stream_visibility_response import V1UpdateMetricsStreamVisibilityResponse
@@ -840,15 +892,19 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_validate_managed_endpoint_r
 from lightning_sdk.lightning_cloud.openapi.models.v1_validate_managed_endpoint_response import V1ValidateManagedEndpointResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_validate_managed_model_response import V1ValidateManagedModelResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_verify_verification_response import V1VerifyVerificationResponse
+from lightning_sdk.lightning_cloud.openapi.models.v1_voltage_park_direct_v1 import V1VoltageParkDirectV1
 from lightning_sdk.lightning_cloud.openapi.models.v1_volume import V1Volume
 from lightning_sdk.lightning_cloud.openapi.models.v1_vultr_direct_v1 import V1VultrDirectV1
 from lightning_sdk.lightning_cloud.openapi.models.v1_work import V1Work
 from lightning_sdk.lightning_cloud.openapi.models.validate import Validate
 from lightning_sdk.lightning_cloud.openapi.models.validateautojoindomain_domain_body import ValidateautojoindomainDomainBody
 from lightning_sdk.lightning_cloud.openapi.models.version_default_body import VersionDefaultBody
+from lightning_sdk.lightning_cloud.openapi.models.version_default_body1 import VersionDefaultBody1
 from lightning_sdk.lightning_cloud.openapi.models.version_uploads_body import VersionUploadsBody
+from lightning_sdk.lightning_cloud.openapi.models.version_uploads_body1 import VersionUploadsBody1
 from lightning_sdk.lightning_cloud.openapi.models.versions_id_body import VersionsIdBody
 from lightning_sdk.lightning_cloud.openapi.models.versions_version_body import VersionsVersionBody
+from lightning_sdk.lightning_cloud.openapi.models.versions_version_body1 import VersionsVersionBody1
 from lightning_sdk.lightning_cloud.openapi.models.works_id_body import WorksIdBody
 from lightning_sdk.lightning_cloud.openapi.models.v1_image_spec import V1ImageSpec as Gridv1ImageSpec
 from lightning_sdk.lightning_cloud.openapi.models.clusters_id_body import ClustersIdBody as Body

lightning_sdk/lightning_cloud/openapi/api/__init__.py

@@ -8,6 +8,7 @@ from lightning_sdk.lightning_cloud.openapi.api.analytics_service_api import Anal
 from lightning_sdk.lightning_cloud.openapi.api.assistants_service_api import AssistantsServiceApi
 from lightning_sdk.lightning_cloud.openapi.api.auth_service_api import AuthServiceApi
 from lightning_sdk.lightning_cloud.openapi.api.billing_service_api import BillingServiceApi
+from lightning_sdk.lightning_cloud.openapi.api.cloud_space_environment_template_service_api import CloudSpaceEnvironmentTemplateServiceApi
 from lightning_sdk.lightning_cloud.openapi.api.cloud_space_service_api import CloudSpaceServiceApi
 from lightning_sdk.lightning_cloud.openapi.api.cluster_service_api import ClusterServiceApi
 from lightning_sdk.lightning_cloud.openapi.api.data_connection_service_api import DataConnectionServiceApi
@@ -20,16 +21,19 @@ from lightning_sdk.lightning_cloud.openapi.api.jobs_service_api import JobsServi
 from lightning_sdk.lightning_cloud.openapi.api.lightningapp_instance_service_api import LightningappInstanceServiceApi
 from lightning_sdk.lightning_cloud.openapi.api.lightningapp_v2_service_api import LightningappV2ServiceApi
 from lightning_sdk.lightning_cloud.openapi.api.lightningwork_service_api import LightningworkServiceApi
+from lightning_sdk.lightning_cloud.openapi.api.lit_dataset_service_api import LitDatasetServiceApi
 from lightning_sdk.lightning_cloud.openapi.api.lit_logger_service_api import LitLoggerServiceApi
 from lightning_sdk.lightning_cloud.openapi.api.lit_page_service_api import LitPageServiceApi
 from lightning_sdk.lightning_cloud.openapi.api.lit_registry_service_api import LitRegistryServiceApi
 from lightning_sdk.lightning_cloud.openapi.api.models_store_api import ModelsStoreApi
 from lightning_sdk.lightning_cloud.openapi.api.organizations_service_api import OrganizationsServiceApi
+from lightning_sdk.lightning_cloud.openapi.api.pipeline_templates_service_api import PipelineTemplatesServiceApi
 from lightning_sdk.lightning_cloud.openapi.api.pipelines_service_api import PipelinesServiceApi
 from lightning_sdk.lightning_cloud.openapi.api.profiler_service_api import ProfilerServiceApi
 from lightning_sdk.lightning_cloud.openapi.api.projects_service_api import ProjectsServiceApi
 from lightning_sdk.lightning_cloud.openapi.api.quest_service_api import QuestServiceApi
 from lightning_sdk.lightning_cloud.openapi.api.ssh_public_key_service_api import SSHPublicKeyServiceApi
+from lightning_sdk.lightning_cloud.openapi.api.schedules_service_api import SchedulesServiceApi
 from lightning_sdk.lightning_cloud.openapi.api.secret_service_api import SecretServiceApi
 from lightning_sdk.lightning_cloud.openapi.api.slurm_jobs_user_service_api import SlurmJobsUserServiceApi
 from lightning_sdk.lightning_cloud.openapi.api.snowflake_service_api import SnowflakeServiceApi