lightning-sdk 0.2.23__py3-none-any.whl → 0.2.24rc1__py3-none-any.whl

lightning_sdk/__init__.py

@@ -31,6 +31,6 @@ __all__ = [
     "User",
 ]
 
-__version__ = "0.2.23"
+__version__ = "0.2.24.rc1"
 _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/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:
@@ -75,7 +69,7 @@ class LLMApi:
         self,
         prompt: str,
         system_prompt: Optional[str],
-        max_completion_tokens: int,
+        max_completion_tokens: Optional[int],
         assistant_id: str,
         images: Optional[List[str]] = None,
         conversation_id: Optional[str] = None,
@@ -83,8 +77,13 @@ class LLMApi:
         name: Optional[str] = None,
         metadata: Optional[Dict[str, str]] = None,
         stream: bool = False,
+        **kwargs: Any,
     ) -> Union[V1ConversationResponseChunk, Generator[V1ConversationResponseChunk, None, None]]:
         is_internal_conversation = os.getenv("LIGHTNING_INTERNAL_CONVERSATION", "false").lower() == "true"
+        ephemeral = os.getenv("LIGHTNING_EPHEMERAL", "false").lower() == "true"
+        if ephemeral:
+            conversation_id = None
+            name = None
         body = {
             "message": {
                 "author": {"role": "user"},
@@ -100,6 +99,9 @@ class LLMApi:
             "metadata": metadata or {},
             "internal_conversation": is_internal_conversation,
             "system_prompt": system_prompt,
+            "ephemeral": ephemeral,
+            "parent_conversation_id": kwargs.get("parent_conversation_id", ""),
+            "parent_message_id": kwargs.get("parent_message_id", ""),
         }
         if images:
             for image in images:
@@ -123,7 +125,7 @@ class LLMApi:
         self,
         prompt: str,
         system_prompt: Optional[str],
-        max_completion_tokens: int,
+        max_completion_tokens: Optional[int],
         assistant_id: str,
         images: Optional[List[str]] = None,
         conversation_id: Optional[str] = None,
@@ -131,8 +133,13 @@ class LLMApi:
         name: Optional[str] = None,
         metadata: Optional[Dict[str, str]] = None,
         stream: bool = False,
+        **kwargs: Any,
     ) -> Union[V1ConversationResponseChunk, AsyncGenerator[V1ConversationResponseChunk, None]]:
         is_internal_conversation = os.getenv("LIGHTNING_INTERNAL_CONVERSATION", "false").lower() == "true"
+        ephemeral = os.getenv("LIGHTNING_EPHEMERAL", "false").lower() == "true"
+        if ephemeral:
+            conversation_id = None
+            name = None
         body = {
             "message": {
                 "author": {"role": "user"},
@@ -148,6 +155,9 @@ class LLMApi:
             "metadata": metadata or {},
             "internal_conversation": is_internal_conversation,
             "system_prompt": system_prompt,
+            "ephemeral": ephemeral,
+            "parent_conversation_id": kwargs.get("parent_conversation_id", ""),
+            "parent_message_id": kwargs.get("parent_message_id", ""),
         }
         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,20 +55,17 @@ class PipelineApi:
         body = ProjectIdPipelinesBody(
             name=name,
             steps=steps,
-            shared_filesystem=V1SharedFilesystem(
-                enabled=shared_filesystem,
-                s3_folder=True,
-            ),
+            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:
@@ -77,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
 
@@ -88,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

@@ -124,6 +124,8 @@ class StudioApi:
         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(
@@ -133,6 +135,8 @@ class StudioApi:
             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/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
@@ -62,6 +63,7 @@ from lightning_sdk.lightning_cloud.openapi.api.snowflake_service_api import Snow
 from lightning_sdk.lightning_cloud.openapi.api.storage_service_api import StorageServiceApi
 from lightning_sdk.lightning_cloud.openapi.api.studio_jobs_service_api import StudioJobsServiceApi
 from lightning_sdk.lightning_cloud.openapi.api.user_service_api import UserServiceApi
+from lightning_sdk.lightning_cloud.openapi.api.volume_service_api import VolumeServiceApi
 # import ApiClient
 from lightning_sdk.lightning_cloud.openapi.api_client import ApiClient
 from lightning_sdk.lightning_cloud.openapi.configuration import Configuration
@@ -103,6 +105,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
@@ -175,6 +179,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
@@ -308,10 +313,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
@@ -388,6 +395,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_data_connection import V1Da
 from lightning_sdk.lightning_cloud.openapi.models.v1_data_connection_artifact import V1DataConnectionArtifact
 from lightning_sdk.lightning_cloud.openapi.models.v1_data_connection_mount import V1DataConnectionMount
 from lightning_sdk.lightning_cloud.openapi.models.v1_data_connection_state import V1DataConnectionState
+from lightning_sdk.lightning_cloud.openapi.models.v1_data_connection_tier import V1DataConnectionTier
 from lightning_sdk.lightning_cloud.openapi.models.v1_data_path import V1DataPath
 from lightning_sdk.lightning_cloud.openapi.models.v1_dataset import V1Dataset
 from lightning_sdk.lightning_cloud.openapi.models.v1_dataset_type import V1DatasetType
@@ -569,6 +577,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_get_user_balance_response i
 from lightning_sdk.lightning_cloud.openapi.models.v1_get_user_notification_preferences_response import V1GetUserNotificationPreferencesResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_get_user_response import V1GetUserResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_get_user_storage_breakdown_response import V1GetUserStorageBreakdownResponse
+from lightning_sdk.lightning_cloud.openapi.models.v1_get_volume_response import V1GetVolumeResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_git_credentials import V1GitCredentials
 from lightning_sdk.lightning_cloud.openapi.models.v1_google_cloud_direct_v1 import V1GoogleCloudDirectV1
 from lightning_sdk.lightning_cloud.openapi.models.v1_google_cloud_direct_v1_status import V1GoogleCloudDirectV1Status
@@ -599,6 +608,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
@@ -634,6 +645,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
@@ -860,6 +872,7 @@ 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_action_type import V1ScheduleActionType
 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
@@ -904,6 +917,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
@@ -929,12 +943,15 @@ 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
+from lightning_sdk.lightning_cloud.openapi.models.v1_update_volume_response import V1UpdateVolumeResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_upload_agent_job_artifact_response import V1UploadAgentJobArtifactResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_upload_agent_job_output_response import V1UploadAgentJobOutputResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_upload_lightningapp_instance_artifact_response import V1UploadLightningappInstanceArtifactResponse
@@ -979,6 +996,7 @@ from lightning_sdk.lightning_cloud.openapi.models.version_uploads_body1 import V
 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.volumes_id_body import VolumesIdBody
 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

@@ -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
@@ -43,3 +44,4 @@ from lightning_sdk.lightning_cloud.openapi.api.snowflake_service_api import Snow
 from lightning_sdk.lightning_cloud.openapi.api.storage_service_api import StorageServiceApi
 from lightning_sdk.lightning_cloud.openapi.api.studio_jobs_service_api import StudioJobsServiceApi
 from lightning_sdk.lightning_cloud.openapi.api.user_service_api import UserServiceApi
+from lightning_sdk.lightning_cloud.openapi.api.volume_service_api import VolumeServiceApi