lightning-sdk 0.2.22__py3-none-any.whl → 0.2.24rc0__py3-none-any.whl

lightning_sdk/__init__.py

@@ -31,6 +31,6 @@ __all__ = [
     "User",
 ]
 
-__version__ = "0.2.22"
+__version__ = "0.2.24.rc0"
 _check_version_and_prompt_upgrade(__version__)
 _set_tqdm_envvars_noninteractive()

lightning_sdk/api/base_studio_api.py

@@ -1,10 +1,13 @@
 from typing import Any, List, Optional
 
+from lightning_sdk.lightning_cloud.openapi import (
+    V1CloudSpaceEnvironmentType,
+    V1ListCloudSpaceEnvironmentTemplatesResponse,
+)
 from lightning_sdk.lightning_cloud.openapi.models.update import Update as BaseStudioUpdateBody
 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_type import V1CloudSpaceEnvironmentType
 from lightning_sdk.lightning_cloud.rest_client import LightningClient
 
 
@@ -21,8 +24,12 @@ class BaseStudioApi:
         except ValueError as e:
             raise ValueError(f"Base studio {base_studio_id} does not exist") from e
 
-    def get_all_base_studios(self, org_id: str) -> List[V1CloudSpaceEnvironmentTemplate]:
+    def get_all_base_studios(self, org_id: str, managed: bool = True) -> V1ListCloudSpaceEnvironmentTemplatesResponse:
         """Retrieve all base studios for a given organization."""
+        if managed:
+            return self._client.cloud_space_environment_template_service_list_managed_cloud_space_environment_templates(
+                org_id=org_id
+            )
         return self._client.cloud_space_environment_template_service_list_cloud_space_environment_templates(
             org_id=org_id
         )

lightning_sdk/api/deployment_api.py

@@ -102,10 +102,10 @@ class ExecHealthCheck(HealthCheck):
     def __init__(
         self,
         command: str,
-        timeout_seconds: int = 60,
-        initial_delay_seconds: int = 5,
-        failure_threshold: int = 12,
-        interval_seconds: int = 5,
+        initial_delay_seconds: int = 0,
+        failure_threshold: int = 600,
+        interval_seconds: int = 1,
+        timeout_seconds: int = 30,
     ) -> None:
         self.command = command
         self.timeout_seconds = timeout_seconds
@@ -131,17 +131,17 @@ class HttpHealthCheck(HealthCheck):
         self,
         path: str,
         port: float,
-        timeout_seconds: int = 60,
-        initial_delay_seconds: int = 5,
-        failure_threshold: int = 12,
-        interval_seconds: int = 5,
+        initial_delay_seconds: int = 0,
+        failure_threshold: int = 600,
+        interval_seconds: int = 1,
+        timeout_seconds: int = 30,
     ) -> None:
         self.path = path
         self.port = port
-        self.timeout_seconds = timeout_seconds
         self.initial_delay_seconds = initial_delay_seconds
         self.failure_threshold = failure_threshold
         self.interval_seconds = interval_seconds
+        self.timeout_seconds = timeout_seconds
 
 
 class AutoScalingMetric:

lightning_sdk/api/license_api.py

@@ -10,8 +10,8 @@ LICENSE_CODE = os.environ.get("LICENSE_CODE", "d9s79g79ss")
 LICENSE_SIGNING_URL = f"{env.LIGHTNING_CLOUD_URL}?settings=licenses"
 
 
-def generate_url_user_settings(redirect_to: str = LICENSE_SIGNING_URL) -> str:
-    params = urlencode({"redirectTo": redirect_to, "okbhrt": LICENSE_CODE})
+def generate_url_user_settings(name: str, redirect_to: str = LICENSE_SIGNING_URL) -> str:
+    params = urlencode({"redirectTo": redirect_to, "okbhrt": LICENSE_CODE, "licenseName": name})
     return f"{env.LIGHTNING_CLOUD_URL}/sign-in?{params}"
 
 

lightning_sdk/api/llm_api.py

@@ -18,17 +18,11 @@ class LLMApi:
     def __init__(self) -> None:
         self._client = LightningClient(retry=False, max_tries=0)
 
-    def get_public_models(self) -> List[str]:
-        result = self._client.assistants_service_list_assistants(published=True)
-        return result.assistants
-
-    def get_org_models(self, org_id: str) -> List[str]:
-        result = self._client.assistants_service_list_assistants(org_id=org_id)
-        return result.assistants
-
-    def get_user_models(self, user_id: str) -> List[str]:
-        result = self._client.assistants_service_list_assistants(user_id=user_id)
-        return result.assistants
+    def get_assistant(self, model_provider: str, model_name: str, user_name: str, org_name: str) -> str:
+        result = self._client.assistants_service_get_managed_model_assistant(
+            model_provider=model_provider, model_name=model_name, user_name=user_name, org_name=org_name
+        )
+        return result.id
 
     def _parse_stream_line(self, decoded_line: str) -> Optional[V1ConversationResponseChunk]:
         try:
@@ -99,6 +93,7 @@ class LLMApi:
             "stream": stream,
             "metadata": metadata or {},
             "internal_conversation": is_internal_conversation,
+            "system_prompt": system_prompt,
         }
         if images:
             for image in images:
@@ -146,6 +141,7 @@ class LLMApi:
             "stream": stream,
             "metadata": metadata or {},
             "internal_conversation": is_internal_conversation,
+            "system_prompt": system_prompt,
         }
         if images:
             for image in images:

lightning_sdk/api/pipeline_api.py

@@ -1,5 +1,6 @@
-from typing import TYPE_CHECKING, List, Optional
+from typing import TYPE_CHECKING, List, Optional, Union
 
+from lightning_sdk.api.cluster_api import ClusterApi
 from lightning_sdk.lightning_cloud.openapi.models import (
     ProjectIdPipelinesBody,
     ProjectIdSchedulesBody,
@@ -11,6 +12,7 @@ from lightning_sdk.lightning_cloud.openapi.models import (
 )
 from lightning_sdk.lightning_cloud.openapi.rest import ApiException
 from lightning_sdk.lightning_cloud.rest_client import LightningClient
+from lightning_sdk.teamspace import Teamspace
 
 if TYPE_CHECKING:
     from lightning_sdk.pipeline.schedule import Schedule
@@ -20,7 +22,8 @@ class PipelineApi:
     """Internal API client for Pipeline requests (mainly http requests)."""
 
     def __init__(self) -> None:
-        self._client = LightningClient(retry=False, max_tries=0)
+        self._client = LightningClient(max_tries=0, retry=False)
+        self._cluster_api = ClusterApi()
 
     def get_pipeline_by_id(self, project_id: str, pipeline_id_or_name: str) -> Optional[V1Pipeline]:
         if pipeline_id_or_name.startswith("pip_"):
@@ -43,7 +46,7 @@ class PipelineApi:
     def create_pipeline(
         self,
         name: str,
-        project_id: str,
+        teamspace: Teamspace,
         steps: List["V1PipelineStep"],
         shared_filesystem: bool,
         schedules: List["Schedule"],
@@ -52,19 +55,17 @@ class PipelineApi:
         body = ProjectIdPipelinesBody(
             name=name,
             steps=steps,
-            shared_filesystem=V1SharedFilesystem(
-                enabled=shared_filesystem,
-            ),
+            shared_filesystem=self._prepare_shared_filesytem(shared_filesystem, steps, teamspace),
             parent_pipeline_id=parent_pipeline_id or "",
         )
 
-        pipeline = self._client.pipelines_service_create_pipeline(body, project_id)
+        pipeline = self._client.pipelines_service_create_pipeline(body, teamspace.id)
 
         # Delete the previous schedules
         if parent_pipeline_id is not None:
-            current_schedules = self._client.schedules_service_list_schedules(project_id).schedules
+            current_schedules = self._client.schedules_service_list_schedules(teamspace.id).schedules
             for schedule in current_schedules:
-                self._client.schedules_service_delete_schedule(project_id, schedule.id)
+                self._client.schedules_service_delete_schedule(teamspace.id, schedule.id)
 
         if len(schedules):
             for schedule in schedules:
@@ -76,7 +77,7 @@ class PipelineApi:
                     resource_type=V1ScheduleResourceType.PIPELINE,
                 )
 
-                self._client.schedules_service_create_schedule(body, project_id)
+                self._client.schedules_service_create_schedule(body, teamspace.id)
 
         return pipeline
 
@@ -87,3 +88,23 @@ class PipelineApi:
 
     def delete(self, project_id: str, pipeline_id: str) -> V1DeletePipelineResponse:
         return self._client.pipelines_service_delete_pipeline(project_id, pipeline_id)
+
+    def _prepare_shared_filesytem(
+        self, shared_filesystem: Union[bool, V1SharedFilesystem], steps: List["V1PipelineStep"], teamspace: Teamspace
+    ) -> V1SharedFilesystem:
+        if not shared_filesystem:
+            return V1SharedFilesystem(enabled=False)
+
+        from lightning_sdk.pipeline.utils import _get_cloud_account
+
+        cluster = self._cluster_api.get_cluster(
+            cluster_id=_get_cloud_account(steps), project_id=teamspace.id, org_id=teamspace.owner.id
+        )
+
+        if cluster.spec.aws_v1:
+            return V1SharedFilesystem(enabled=True, s3_folder=True)
+
+        if cluster.spec.google_cloud_v1:
+            return V1SharedFilesystem(enabled=True, gcs_folder=True)
+
+        raise NotImplementedError("This cluster isn't support yet")

lightning_sdk/api/studio_api.py

@@ -123,6 +123,9 @@ class StudioApi:
         teamspace_id: str,
         cloud_account: Optional[str] = None,
         source: Optional[V1CloudSpaceSourceType] = None,
+        disable_secrets: bool = False,
+        sandbox: bool = False,
+        cloud_space_environment_template_id: Optional[str] = None,
     ) -> V1CloudSpace:
         """Create a Studio with a given name in a given Teamspace on a possibly given cloud_account."""
         body = ProjectIdCloudspacesBody(
@@ -131,6 +134,9 @@ class StudioApi:
             display_name=name,
             seed_files=[V1CloudSpaceSeedFile(path="main.py", contents="print('Hello, Lightning World!')\n")],
             source=source,
+            disable_secrets=disable_secrets,
+            sandbox=sandbox,
+            cloud_space_environment_template_id=cloud_space_environment_template_id,
         )
         studio = self._client.cloud_space_service_create_cloud_space(body, teamspace_id)
 

lightning_sdk/base_studio.py

@@ -1,17 +1,23 @@
+from dataclasses import dataclass
 from typing import List, Optional, Union
 
 from lightning_sdk.api.base_studio_api import BaseStudioApi
 from lightning_sdk.api.user_api import UserApi
 from lightning_sdk.lightning_cloud import login
-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_type import V1CloudSpaceEnvironmentType
 from lightning_sdk.organization import Organization
 from lightning_sdk.user import User
 from lightning_sdk.utils.resolve import _resolve_org, _resolve_user
 
 
+@dataclass
+class BaseStudioInfo:
+    id: str
+    name: str
+    managed_id: str
+    description: str
+
+
 class BaseStudio:
     def __init__(
         self,
@@ -61,7 +67,7 @@ class BaseStudio:
         environment_type: Optional[V1CloudSpaceEnvironmentType] = None,
         machine_image_version: Optional[str] = None,
         setup_script_text: Optional[str] = None,
-    ) -> V1CloudSpaceEnvironmentTemplate:
+    ) -> None:
         self._base_studio = self._base_studio_api.update_base_studio(
             self._base_studio.id,
             self._org.id,
@@ -74,10 +80,20 @@ class BaseStudio:
             disabled=disabled,
         )
 
-    def list(self) -> List[V1CloudSpaceEnvironmentTemplate]:
+    def list(self, managed: bool = True) -> List[BaseStudioInfo]:
         """List all base studios in the organization.
 
         Returns:
             List[V1CloudSpaceEnvironmentTemplate]: A list of base studio templates.
         """
-        return self._base_studio_api.get_all_base_studios(self._org.id)
+        result = []
+        for template in self._base_studio_api.get_all_base_studios(self._org.id, managed).templates:
+            result.append(
+                BaseStudioInfo(
+                    id=template.id,
+                    name=template.name,
+                    managed_id=template.managed_id,
+                    description=template.description,
+                ),
+            )
+        return result

lightning_sdk/cli/entrypoint.py

@@ -4,8 +4,9 @@ from types import TracebackType
 from typing import Type
 
 import click
-from rich.console import Console
+from rich.console import Console, Group
 from rich.panel import Panel
+from rich.syntax import Syntax
 from rich.text import Text
 
 from lightning_sdk import __version__
@@ -34,23 +35,24 @@ from lightning_sdk.lightning_cloud.login import Auth
 
 def _notify_exception(exception_type: Type[BaseException], value: BaseException, tb: TracebackType) -> None:
     """CLI won't show tracebacks, just print the exception message."""
-    # if debug mode, print the traceback using rich
     console = Console()
-    if value.args:
-        message = str(value.args[0]) if value.args[0] else str(value)
-    else:
-        message = str(value) or "An unknown error occurred"
 
-    error_content = Text()
-    error_content.append(f"{exception_type.__name__}: ", style="bold red")
-    error_content.append(message, style="white")
+    message = str(value.args[0]) if value.args else str(value) or "An unknown error occurred"
+
+    error_text = Text()
+    error_text.append(f"{exception_type.__name__}: ", style="bold red")
+    error_text.append(message, style="white")
+
+    renderables = [error_text]
 
     if _LIGHTNING_DEBUG:
-        error_content.append("\n\nFull traceback:\n", style="bold yellow")
-        tb_lines = traceback.format_exception(exception_type, value, tb)
-        error_content.append("".join(tb_lines), style="dim white")
+        tb_text = "".join(traceback.format_exception(exception_type, value, tb))
+        renderables.append(Text("\n\nFull traceback:\n", style="bold yellow"))
+        renderables.append(Syntax(tb_text, "python", theme="monokai", line_numbers=False, word_wrap=True))
+    else:
+        renderables.append(Text("\n\nTo see the full traceback, set the LIGHTNING_DEBUG environment variable to 1."))
 
-    console.print(Panel(error_content, title="⚡ Lightning CLI Error", border_style="red"))
+    console.print(Panel(Group(*renderables), title="⚡ Lightning CLI Error", border_style="red"))
 
 
 @click.group(name="lightning", help="Command line interface (CLI) to interact with/manage Lightning AI Studios.")

lightning_sdk/cli/start.py

@@ -60,8 +60,11 @@ def studio(name: str, teamspace: Optional[str] = None, machine: str = "CPU", pro
 
     try:
         studio = Studio(name=name, teamspace=teamspace, org=owner, user=None, create_ok=False, provider=provider)
-    except (RuntimeError, ValueError, ApiException):
-        studio = Studio(name=name, teamspace=teamspace, org=None, user=owner, create_ok=False, provider=provider)
+    except (RuntimeError, ValueError, ApiException) as first_error:
+        try:
+            studio = Studio(name=name, teamspace=teamspace, org=None, user=owner, create_ok=False, provider=provider)
+        except (RuntimeError, ValueError, ApiException) as second_error:
+            raise first_error from second_error
 
     try:
         resolved_machine = getattr(Machine, machine.upper(), Machine(machine, machine))

lightning_sdk/deployment/deployment.py

@@ -197,11 +197,13 @@ class Deployment:
         if commands is not None:
             command = compose_commands(commands)
 
+        autoscaling_metric_name = ("CPU" if machine.is_cpu() else "GPU") if isinstance(machine, Machine) else "CPU"
+
         if autoscale is None:
             autoscale = AutoScaleConfig(
                 min_replicas=0,
                 max_replicas=1,
-                metric="CPU" if machine.is_cpu() else "GPU",
+                metric=autoscaling_metric_name,
                 threshold=90,
             )
 
@@ -380,27 +382,35 @@ class Deployment:
         return None
 
     @property
-    def pending_replicas(self) -> Optional[List[str]]:
+    def pending_replicas(self) -> Optional[int]:
         """The number of pending replicas."""
         if self._deployment:
             self._deployment = self._deployment_api.get_deployment_by_name(self._name, self._teamspace.id)
-            return self._deployment.status.pending_replicas
+            return int(self._deployment.status.pending_replicas)
+        return None
+
+    @property
+    def running_replicas(self) -> Optional[int]:
+        """The number of failing replicas."""
+        if self._deployment:
+            self._deployment = self._deployment_api.get_deployment_by_name(self._name, self._teamspace.id)
+            return int(self._deployment.status.ready_replicas)
         return None
 
     @property
-    def failing_replicas(self) -> Optional[List[str]]:
+    def failing_replicas(self) -> Optional[int]:
         """The number of failing replicas."""
         if self._deployment:
             self._deployment = self._deployment_api.get_deployment_by_name(self._name, self._teamspace.id)
-            return self._deployment.status.failing_replicas
+            return int(self._deployment.status.failing_replicas)
         return None
 
     @property
-    def deleting_replicas(self) -> Optional[List[str]]:
+    def deleting_replicas(self) -> Optional[int]:
         """The number of deleting replicas."""
         if self._deployment:
             self._deployment = self._deployment_api.get_deployment_by_name(self._name, self._teamspace.id)
-            return self._deployment.status.deleting_replicas
+            return int(self._deployment.status.deleting_replicas)
         return None
 
     @property

lightning_sdk/lightning_cloud/openapi/__init__.py

@@ -30,6 +30,7 @@ from lightning_sdk.lightning_cloud.openapi.api.billing_service_api import Billin
 from lightning_sdk.lightning_cloud.openapi.api.blog_posts_service_api import BlogPostsServiceApi
 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.cloudy_service_api import CloudyServiceApi
 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
 from lightning_sdk.lightning_cloud.openapi.api.dataset_service_api import DatasetServiceApi
@@ -89,6 +90,7 @@ from lightning_sdk.lightning_cloud.openapi.models.cloud_space_id_versions_body i
 from lightning_sdk.lightning_cloud.openapi.models.cloudspace_id_metric_body import CloudspaceIdMetricBody
 from lightning_sdk.lightning_cloud.openapi.models.cloudspace_id_runs_body import CloudspaceIdRunsBody
 from lightning_sdk.lightning_cloud.openapi.models.cloudspace_id_systemmetrics_body import CloudspaceIdSystemmetricsBody
+from lightning_sdk.lightning_cloud.openapi.models.cloudspace_id_visibility_body import CloudspaceIdVisibilityBody
 from lightning_sdk.lightning_cloud.openapi.models.cloudspaces_id_body import CloudspacesIdBody
 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
@@ -102,6 +104,8 @@ from lightning_sdk.lightning_cloud.openapi.models.conversations_id_body import C
 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.credits_autoreplenish_body import CreditsAutoreplenishBody
+from lightning_sdk.lightning_cloud.openapi.models.credits_autoreplenish_body1 import CreditsAutoreplenishBody1
 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
@@ -174,6 +178,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.pipelines_id_body1 import PipelinesIdBody1
 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
@@ -275,6 +280,8 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_cpu_system_metrics import V
 from lightning_sdk.lightning_cloud.openapi.models.v1_cancel_cloud_space_instance_switch_response import V1CancelCloudSpaceInstanceSwitchResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_cancellation_metadata import V1CancellationMetadata
 from lightning_sdk.lightning_cloud.openapi.models.v1_capacity_block_offering import V1CapacityBlockOffering
+from lightning_sdk.lightning_cloud.openapi.models.v1_check_cluster_name_availability_request import V1CheckClusterNameAvailabilityRequest
+from lightning_sdk.lightning_cloud.openapi.models.v1_check_cluster_name_availability_response import V1CheckClusterNameAvailabilityResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_check_external_service_status_response import V1CheckExternalServiceStatusResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_check_snowflake_connection_response import V1CheckSnowflakeConnectionResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_checkbox import V1Checkbox
@@ -305,10 +312,12 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_cloud_space_publication_typ
 from lightning_sdk.lightning_cloud.openapi.models.v1_cloud_space_seed_file import V1CloudSpaceSeedFile
 from lightning_sdk.lightning_cloud.openapi.models.v1_cloud_space_session import V1CloudSpaceSession
 from lightning_sdk.lightning_cloud.openapi.models.v1_cloud_space_source_type import V1CloudSpaceSourceType
+from lightning_sdk.lightning_cloud.openapi.models.v1_cloud_space_specialized_view import V1CloudSpaceSpecializedView
 from lightning_sdk.lightning_cloud.openapi.models.v1_cloud_space_state import V1CloudSpaceState
 from lightning_sdk.lightning_cloud.openapi.models.v1_cloud_space_version import V1CloudSpaceVersion
 from lightning_sdk.lightning_cloud.openapi.models.v1_cloud_space_version_publication import V1CloudSpaceVersionPublication
 from lightning_sdk.lightning_cloud.openapi.models.v1_cloudflare_v1 import V1CloudflareV1
+from lightning_sdk.lightning_cloud.openapi.models.v1_cloudy_expert import V1CloudyExpert
 from lightning_sdk.lightning_cloud.openapi.models.v1_cluster_accelerator import V1ClusterAccelerator
 from lightning_sdk.lightning_cloud.openapi.models.v1_cluster_availability import V1ClusterAvailability
 from lightning_sdk.lightning_cloud.openapi.models.v1_cluster_capacity_reservation import V1ClusterCapacityReservation
@@ -499,6 +508,8 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_execute_in_cloud_space_sess
 from lightning_sdk.lightning_cloud.openapi.models.v1_execute_snowflake_query_response import V1ExecuteSnowflakeQueryResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_experiment import V1Experiment
 from lightning_sdk.lightning_cloud.openapi.models.v1_export_snowflake_query_response import V1ExportSnowflakeQueryResponse
+from lightning_sdk.lightning_cloud.openapi.models.v1_external_cluster import V1ExternalCluster
+from lightning_sdk.lightning_cloud.openapi.models.v1_external_cluster_spec import V1ExternalClusterSpec
 from lightning_sdk.lightning_cloud.openapi.models.v1_external_search_user import V1ExternalSearchUser
 from lightning_sdk.lightning_cloud.openapi.models.v1_file_endpoint import V1FileEndpoint
 from lightning_sdk.lightning_cloud.openapi.models.v1_filestore_data_connection import V1FilestoreDataConnection
@@ -594,6 +605,8 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_job_type import V1JobType
 from lightning_sdk.lightning_cloud.openapi.models.v1_joinable_organization import V1JoinableOrganization
 from lightning_sdk.lightning_cloud.openapi.models.v1_keep_alive_cloud_space_instance_response import V1KeepAliveCloudSpaceInstanceResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_knowledge_configuration import V1KnowledgeConfiguration
+from lightning_sdk.lightning_cloud.openapi.models.v1_kubernetes_direct_v1 import V1KubernetesDirectV1
+from lightning_sdk.lightning_cloud.openapi.models.v1_kubernetes_direct_v1_status import V1KubernetesDirectV1Status
 from lightning_sdk.lightning_cloud.openapi.models.v1_lambda_labs_direct_v1 import V1LambdaLabsDirectV1
 from lightning_sdk.lightning_cloud.openapi.models.v1_lightning_app_user import V1LightningAppUser
 from lightning_sdk.lightning_cloud.openapi.models.v1_lightning_auth import V1LightningAuth
@@ -629,6 +642,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_list_cloud_space_tags_respo
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_cloud_space_version_publications_response import V1ListCloudSpaceVersionPublicationsResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_cloud_space_versions_response import V1ListCloudSpaceVersionsResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_cloud_spaces_response import V1ListCloudSpacesResponse
+from lightning_sdk.lightning_cloud.openapi.models.v1_list_cloudy_experts_response import V1ListCloudyExpertsResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_cluster_accelerators_response import V1ListClusterAcceleratorsResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_cluster_availabilities_response import V1ListClusterAvailabilitiesResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_cluster_capacity_reservations_response import V1ListClusterCapacityReservationsResponse
@@ -719,6 +733,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_lit_page_type import V1LitP
 from lightning_sdk.lightning_cloud.openapi.models.v1_lit_registry_artifact import V1LitRegistryArtifact
 from lightning_sdk.lightning_cloud.openapi.models.v1_lit_registry_project import V1LitRegistryProject
 from lightning_sdk.lightning_cloud.openapi.models.v1_lit_repository import V1LitRepository
+from lightning_sdk.lightning_cloud.openapi.models.v1_lite_published_cloud_space_response import V1LitePublishedCloudSpaceResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_locked_resource import V1LockedResource
 from lightning_sdk.lightning_cloud.openapi.models.v1_log_page import V1LogPage
 from lightning_sdk.lightning_cloud.openapi.models.v1_logger_artifact import V1LoggerArtifact
@@ -898,6 +913,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_system_metrics_aggregated i
 from lightning_sdk.lightning_cloud.openapi.models.v1_system_metrics_list import V1SystemMetricsList
 from lightning_sdk.lightning_cloud.openapi.models.v1_telemetry import V1Telemetry
 from lightning_sdk.lightning_cloud.openapi.models.v1_timestamp_code_telemetry import V1TimestampCodeTelemetry
+from lightning_sdk.lightning_cloud.openapi.models.v1_token_usage import V1TokenUsage
 from lightning_sdk.lightning_cloud.openapi.models.v1_transaction import V1Transaction
 from lightning_sdk.lightning_cloud.openapi.models.v1_transfer_cloud_space_response import V1TransferCloudSpaceResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_transfer_org_balance_response import V1TransferOrgBalanceResponse
@@ -911,6 +927,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_update_agent_status_respons
 from lightning_sdk.lightning_cloud.openapi.models.v1_update_billing_subscription_request import V1UpdateBillingSubscriptionRequest
 from lightning_sdk.lightning_cloud.openapi.models.v1_update_cloud_space_collab_response import V1UpdateCloudSpaceCollabResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_update_cloud_space_publication_response import V1UpdateCloudSpacePublicationResponse
+from lightning_sdk.lightning_cloud.openapi.models.v1_update_cloud_space_visibility_response import V1UpdateCloudSpaceVisibilityResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_update_cluster_accelerators_request import V1UpdateClusterAcceleratorsRequest
 from lightning_sdk.lightning_cloud.openapi.models.v1_update_cluster_accelerators_response import V1UpdateClusterAcceleratorsResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_update_cluster_availability_request import V1UpdateClusterAvailabilityRequest
@@ -922,9 +939,11 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_update_lit_page_response im
 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
 from lightning_sdk.lightning_cloud.openapi.models.v1_update_model_visibility_response import V1UpdateModelVisibilityResponse
+from lightning_sdk.lightning_cloud.openapi.models.v1_update_organization_credits_auto_replenish_response import V1UpdateOrganizationCreditsAutoReplenishResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_update_project_cluster_accelerators_response import V1UpdateProjectClusterAcceleratorsResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_update_shared_metrics_stream_response import V1UpdateSharedMetricsStreamResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_update_snowflake_query_response import V1UpdateSnowflakeQueryResponse
+from lightning_sdk.lightning_cloud.openapi.models.v1_update_user_credits_auto_replenish_response import V1UpdateUserCreditsAutoReplenishResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_update_user_request import V1UpdateUserRequest
 from lightning_sdk.lightning_cloud.openapi.models.v1_update_user_viewed_new_features_request import V1UpdateUserViewedNewFeaturesRequest
 from lightning_sdk.lightning_cloud.openapi.models.v1_update_user_viewed_new_features_response import V1UpdateUserViewedNewFeaturesResponse
@@ -959,6 +978,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_validate_managed_model_resp
 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_volume_state import V1VolumeState
 from lightning_sdk.lightning_cloud.openapi.models.v1_vultr_direct_v1 import V1VultrDirectV1
 from lightning_sdk.lightning_cloud.openapi.models.v1_weka_data_connection import V1WekaDataConnection
 from lightning_sdk.lightning_cloud.openapi.models.v1_work import V1Work

lightning_sdk/lightning_cloud/openapi/api/__init__.py

@@ -11,6 +11,7 @@ from lightning_sdk.lightning_cloud.openapi.api.billing_service_api import Billin
 from lightning_sdk.lightning_cloud.openapi.api.blog_posts_service_api import BlogPostsServiceApi
 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.cloudy_service_api import CloudyServiceApi
 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
 from lightning_sdk.lightning_cloud.openapi.api.dataset_service_api import DatasetServiceApi