lightning-sdk 2025.7.22__py3-none-any.whl → 2025.7.31__py3-none-any.whl

lightning_sdk/__init__.py

@@ -32,6 +32,6 @@ __all__ = [
     "User",
 ]
 
-__version__ = "2025.07.22"
+__version__ = "2025.07.31"
 _check_version_and_prompt_upgrade(__version__)
 _set_tqdm_envvars_noninteractive()

lightning_sdk/api/cloud_account_api.py

@@ -1,3 +1,4 @@
+from functools import lru_cache
 from typing import TYPE_CHECKING, Dict, List, Optional, Union
 
 from lightning_sdk.lightning_cloud.openapi import (
@@ -6,6 +7,7 @@ from lightning_sdk.lightning_cloud.openapi import (
     V1ClusterType,
     V1ExternalCluster,
     V1ListClusterAcceleratorsResponse,
+    V1ListDefaultClusterAcceleratorsResponse,
 )
 from lightning_sdk.lightning_cloud.rest_client import LightningClient
 
@@ -33,6 +35,7 @@ class CloudAccountApi:
             raise ValueError(f"CloudAccount {cloud_account_id} does not exist")
         return res
 
+    @lru_cache(maxsize=None)  # noqa: B019
     def list_cloud_accounts(self, teamspace_id: str) -> List[V1ExternalCluster]:
         """Lists the cloud accounts for a given teamspace.
 
@@ -42,10 +45,22 @@ class CloudAccountApi:
         Returns:
             A list of cloud accounts
         """
-        res = self._client.cluster_service_list_project_clusters(
+        res_project = self._client.cluster_service_list_project_clusters(
             project_id=teamspace_id,
         )
-        return res.clusters
+        res_global = self._client.cluster_service_list_clusters(
+            project_id=teamspace_id,
+        )
+
+        # can't use set here because the cloud_accounts are not hashable
+        cloud_accounts = []
+        cloud_account_ids = []
+        for cloud_account in res_project.clusters + res_global.clusters:
+            if cloud_account.id not in cloud_account_ids:
+                cloud_accounts.append(cloud_account)
+                cloud_account_ids.append(cloud_account.id)
+
+        return cloud_accounts
 
     def get_cloud_account_non_org(self, teamspace_id: str, cloud_account_id: str) -> Optional[V1ExternalCluster]:
         for cluster in self.list_cloud_accounts(teamspace_id=teamspace_id):
@@ -54,21 +69,52 @@ class CloudAccountApi:
 
         return None
 
-    def list_cloud_account_accelerators(self, cloud_account_id: str, org_id: str) -> V1ListClusterAcceleratorsResponse:
+    @lru_cache(maxsize=None)  # noqa: B019
+    def list_cloud_account_accelerators(
+        self,
+        teamspace_id: str,
+        cloud_account_id: str,
+        org_id: str,
+    ) -> Union[V1ListClusterAcceleratorsResponse, V1ListDefaultClusterAcceleratorsResponse]:
         """Lists the accelerators for a given cloud account.
 
         Args:
             cloud_account_id: cluster ID to list accelerators for
-            org_id: The owning org of this project
         """
-        res = self._client.cluster_service_list_cluster_accelerators(
-            id=cloud_account_id,
-            org_id=org_id,
-        )
+        # map cloud_account to provider
+        cloud_provider = None
+        is_default = True
+        for cloud_account in self.list_cloud_accounts(teamspace_id=teamspace_id):
+            if cloud_account.id == cloud_account_id:
+                is_default = cloud_account.spec.cluster_type == V1ClusterType.GLOBAL
+                cloud_provider = self._get_cloud_account_provider(cloud_account)
+                break
+
+        if cloud_provider is None:
+            raise ValueError(
+                f"Cloud Account {cloud_account_id} is not a default cloud account. Are you in the correct teamspace?"
+            )
+
+        if is_default:
+            res = self._list_default_cluster_accelerators(teamspace_id=teamspace_id, cloud_provider=cloud_provider)
+        else:
+            res = self._client.cluster_service_list_cluster_accelerators(
+                id=cloud_account_id,
+                org_id=org_id,
+            )
+
         if not res:
             raise ValueError(f"CloudAccount {cloud_account_id} does not exist")
         return res
 
+    def _list_default_cluster_accelerators(
+        self, teamspace_id: str, cloud_provider: Union[str, "CloudProvider"]
+    ) -> V1ListDefaultClusterAcceleratorsResponse:
+        return self._client.cluster_service_list_default_cluster_accelerators(
+            project_id=teamspace_id, cloud_provider=str(cloud_provider)
+        )
+
+    @lru_cache(maxsize=None)  # noqa: B019
     def list_global_cloud_accounts(self, teamspace_id: str) -> List[V1ExternalCluster]:
         """Lists the accelerators for a given teamspace.
 
@@ -79,6 +125,10 @@ class CloudAccountApi:
         if not cloud_accounts:
             raise ValueError(f"Teamspace {teamspace_id} does not exist")
         filtered_cloud_accounts = filter(lambda x: x.spec.cluster_type == V1ClusterType.GLOBAL, cloud_accounts)
+        # TODO: remove aggregate filter once finished
+        filtered_cloud_accounts = filter(
+            lambda x: x.spec.driver != V1CloudProvider.LIGHTNING_AGGREGATE, filtered_cloud_accounts
+        )
         return list(filtered_cloud_accounts)
 
     def get_cloud_account_provider_mapping(self, teamspace_id: str) -> Dict["CloudProvider", str]:

lightning_sdk/api/job_api.py

@@ -17,7 +17,7 @@ from lightning_sdk.lightning_cloud.openapi import (
     JobsIdBody1,
     ProjectIdJobsBody,
     V1CloudSpace,
-    V1ComputeConfig,
+    V1ClusterAccelerator,
     V1DownloadJobLogsResponse,
     V1DownloadLightningappInstanceLogsResponse,
     V1EnvVar,
@@ -94,15 +94,48 @@ class JobApiV1:
     def get_work(self, job_id: str, teamspace_id: str, work_id: str) -> Externalv1Lightningwork:
         return self._client.lightningwork_service_get_lightningwork(project_id=teamspace_id, app_id=job_id, id=work_id)
 
-    def get_machine_from_work(self, work: Externalv1Lightningwork) -> Machine:
+    def get_machine_from_work(self, work: Externalv1Lightningwork, org_id: str) -> Machine:
         spec: V1LightningworkSpec = work.spec
         # prefer user-requested config if specified
         user_requested_compute_config: V1UserRequestedComputeConfig = spec.user_requested_compute_config
-        if user_requested_compute_config.name:
-            return Machine.from_str(user_requested_compute_config.name)
-        compute_config: V1ComputeConfig = spec.compute_config
+        accelerators = self._get_machines_for_cloud_account(
+            teamspace_id=work.project_id,
+            cloud_account_id=spec.cluster_id,
+            org_id=org_id,
+        )
+
+        identifier = None
+
+        if user_requested_compute_config and user_requested_compute_config.name:
+            identifier = user_requested_compute_config.name
+        else:
+            identifier = spec.compute_config.instance_type
+
+        for accelerator in accelerators:
+            if identifier in (
+                accelerator.slug,
+                accelerator.slug_multi_cloud,
+                accelerator.instance_id,
+            ):
+                return Machine.from_str(accelerator.slug_multi_cloud)
+
+        return Machine.from_str(identifier)
 
-        return Machine.from_str(compute_config.instance_type)
+    def _get_machines_for_cloud_account(
+        self, teamspace_id: str, cloud_account_id: str, org_id: str
+    ) -> List[V1ClusterAccelerator]:
+        from lightning_sdk.api.cloud_account_api import CloudAccountApi
+
+        cloud_account_api = CloudAccountApi()
+        accelerators = cloud_account_api.list_cloud_account_accelerators(
+            teamspace_id=teamspace_id,
+            cloud_account_id=cloud_account_id,
+            org_id=org_id,
+        )
+        if not accelerators:
+            return []
+
+        return list(filter(lambda acc: acc.enabled, accelerators.accelerator))
 
     def get_studio_name(self, job: Externalv1LightningappInstance) -> str:
         cs: V1CloudSpace = self._client.cloud_space_service_get_cloud_space(
@@ -215,7 +248,7 @@ class JobApiV2:
         path_mappings: Optional[Dict[str, str]],
         artifacts_local: Optional[str],  # deprecated in favor of path_mappings
         artifacts_remote: Optional[str],  # deprecated in favor of path_mappings
-        max_runtime: Optional[str] = None,
+        max_runtime: Optional[int] = None,
     ) -> V1Job:
         body = self._create_job_body(
             name=name,
@@ -380,11 +413,41 @@ class JobApiV2:
             return Status.Stopping
         return Status.Pending
 
-    def _get_job_machine_from_spec(self, spec: V1JobSpec) -> "Machine":
-        instance_name = spec.instance_name
-        instance_type = spec.instance_type
+    def _get_job_machine_from_spec(self, spec: V1JobSpec, teamspace_id: str, org_id: str) -> "Machine":
+        accelerators = self._get_machines_for_cloud_account(
+            teamspace_id=teamspace_id,
+            cloud_account_id=spec.cluster_id,
+            org_id=org_id,
+        )
+
+        for accelerator in accelerators:
+            possible_identifiers = (
+                accelerator.slug,
+                accelerator.slug_multi_cloud,
+                accelerator.instance_id,
+            )
+            if (spec.instance_name and spec.instance_name in possible_identifiers) or (
+                spec.instance_type and spec.instance_type in possible_identifiers
+            ):
+                return Machine.from_str(accelerator.slug_multi_cloud)
+
+        return Machine.from_str(spec.instance_name or spec.instance_type)
+
+    def _get_machines_for_cloud_account(
+        self, teamspace_id: str, cloud_account_id: str, org_id: str
+    ) -> List[V1ClusterAccelerator]:
+        from lightning_sdk.api.cloud_account_api import CloudAccountApi
+
+        cloud_account_api = CloudAccountApi()
+        accelerators = cloud_account_api.list_cloud_account_accelerators(
+            teamspace_id=teamspace_id,
+            cloud_account_id=cloud_account_id,
+            org_id=org_id,
+        )
+        if not accelerators:
+            return []
 
-        return Machine.from_str(instance_name, instance_type or instance_name)
+        return list(filter(lambda acc: acc.enabled, accelerators.accelerator))
 
     def get_total_cost(self, job: V1Job) -> float:
         return job.total_cost

lightning_sdk/api/llm_api.py

@@ -98,7 +98,7 @@ class LLMApi:
                     {"contentType": "text", "parts": [prompt]},
                 ],
             },
-            "max_completion_tokens": max_completion_tokens,
+            "max_tokens": max_completion_tokens,
             "conversation_id": conversation_id,
             "billing_project_id": billing_project_id,
             "name": name,

lightning_sdk/api/mmt_api.py

@@ -2,7 +2,7 @@ import json
 import time
 from typing import TYPE_CHECKING, Dict, List, Optional, Union
 
-from lightning_sdk.api.job_api import JobApiV1
+from lightning_sdk.api.job_api import JobApiV1, V1ClusterAccelerator
 from lightning_sdk.api.utils import (
     _create_app,
     _machine_to_compute_name,
@@ -247,11 +247,41 @@ class MMTApiV2:
     def get_command(self, job: V1MultiMachineJob) -> str:
         return job.spec.command
 
-    def _get_job_machine_from_spec(self, spec: V1JobSpec) -> "Machine":
-        instance_name = spec.instance_name
-        instance_type = spec.instance_type
+    def _get_job_machine_from_spec(self, spec: V1JobSpec, teamspace_id: str, org_id: str) -> "Machine":
+        accelerators = self._get_machines_for_cloud_account(
+            teamspace_id=teamspace_id,
+            cloud_account_id=spec.cluster_id,
+            org_id=org_id,
+        )
+
+        for accelerator in accelerators:
+            possible_identifiers = (
+                accelerator.slug,
+                accelerator.slug_multi_cloud,
+                accelerator.instance_id,
+            )
+            if (spec.instance_name and spec.instance_name in possible_identifiers) or (
+                spec.instance_type and spec.instance_type in possible_identifiers
+            ):
+                return Machine.from_str(accelerator.slug_multi_cloud)
+
+        return Machine.from_str(spec.instance_name or spec.instance_type)
+
+    def _get_machines_for_cloud_account(
+        self, teamspace_id: str, cloud_account_id: str, org_id: str
+    ) -> List[V1ClusterAccelerator]:
+        from lightning_sdk.api.cloud_account_api import CloudAccountApi
+
+        cloud_account_api = CloudAccountApi()
+        accelerators = cloud_account_api.list_cloud_account_accelerators(
+            teamspace_id=teamspace_id,
+            cloud_account_id=cloud_account_id,
+            org_id=org_id,
+        )
+        if not accelerators:
+            return []
 
-        return Machine.from_str(instance_name, instance_type or instance_name)
+        return list(filter(lambda acc: acc.enabled, accelerators.accelerator))
 
     def get_total_cost(self, job: V1MultiMachineJob) -> float:
         return job.total_cost

lightning_sdk/api/studio_api.py

@@ -4,7 +4,7 @@ import tempfile
 import time
 import zipfile
 from threading import Event, Thread
-from typing import Any, Dict, Generator, Mapping, Optional, Tuple, Union
+from typing import Any, Dict, Generator, List, Mapping, Optional, Tuple, Union
 
 import backoff
 import requests
@@ -37,6 +37,7 @@ from lightning_sdk.lightning_cloud.openapi import (
     V1CloudSpaceSeedFile,
     V1CloudSpaceSourceType,
     V1CloudSpaceState,
+    V1ClusterAccelerator,
     V1EndpointType,
     V1GetCloudSpaceInstanceStatusResponse,
     V1GetLongRunningCommandInCloudSpaceResponse,
@@ -267,12 +268,24 @@ class StudioApi:
                 break
             time.sleep(1)
 
-    def get_machine(self, studio_id: str, teamspace_id: str) -> Machine:
+    def get_machine(self, studio_id: str, teamspace_id: str, cloud_account_id: str, org_id: str) -> Machine:
         """Get the current machine type the given Studio is running on."""
         response: V1CloudSpaceInstanceConfig = self._client.cloud_space_service_get_cloud_space_instance_config(
             project_id=teamspace_id, id=studio_id
         )
-        return Machine(response.compute_config.name, response.compute_config.name)
+        accelerators = self._get_machines_for_cloud_account(
+            teamspace_id=teamspace_id, cloud_account_id=cloud_account_id, org_id=org_id
+        )
+
+        for accelerator in accelerators:
+            if response.compute_config.name in (
+                accelerator.slug,
+                accelerator.slug_multi_cloud,
+                accelerator.instance_id,
+            ):
+                return Machine.from_str(accelerator.slug_multi_cloud)
+
+        return Machine.from_str(response.compute_config.name)
 
     def get_interruptible(self, studio_id: str, teamspace_id: str) -> bool:
         """Get whether the Studio is running on a interruptible instance."""
@@ -282,6 +295,22 @@ class StudioApi:
 
         return response.compute_config.spot
 
+    def _get_machines_for_cloud_account(
+        self, teamspace_id: str, cloud_account_id: str, org_id: str
+    ) -> List[V1ClusterAccelerator]:
+        from lightning_sdk.api.cloud_account_api import CloudAccountApi
+
+        cloud_account_api = CloudAccountApi()
+        accelerators = cloud_account_api.list_cloud_account_accelerators(
+            teamspace_id=teamspace_id,
+            cloud_account_id=cloud_account_id,
+            org_id=org_id,
+        )
+        if not accelerators:
+            return []
+
+        return list(filter(lambda acc: acc.enabled, accelerators.accelerator))
+
     def _get_detached_command_status(
         self, studio_id: str, teamspace_id: str, session_id: str
     ) -> V1GetLongRunningCommandInCloudSpaceResponse:

lightning_sdk/api/utils.py

@@ -331,7 +331,9 @@ class _DummyResponse:
 
 def _machine_to_compute_name(machine: Union[Machine, str]) -> str:
     if isinstance(machine, Machine):
-        return machine.instance_type
+        if machine.instance_type is not None:
+            return machine.instance_type
+        return machine.slug
     return machine
 
 
@@ -352,7 +354,9 @@ def _get_registry_url() -> str:
 
 
 def _sanitize_studio_remote_path(path: str, studio_id: str) -> str:
-    return f"/cloudspaces/{studio_id}/code/content/{path.replace('/teamspace/studios/this_studio/', '')}"
+    path = path.replace("/teamspace/studios/this_studio/", "")
+    root = f"/cloudspaces/{studio_id}/code/content/"
+    return os.path.join(root, path)
 
 
 def _resolve_teamspace_remote_path(path: str) -> str:

lightning_sdk/cli/create.py

@@ -12,7 +12,9 @@ from lightning_sdk.cli.teamspace_menu import _TeamspacesMenu
 from lightning_sdk.machine import CloudProvider
 from lightning_sdk.utils.resolve import _resolve_deprecated_provider
 
-_MACHINE_VALUES = tuple([machine.name for machine in Machine.__dict__.values() if isinstance(machine, Machine)])
+_MACHINE_VALUES = tuple(
+    [machine.name for machine in Machine.__dict__.values() if isinstance(machine, Machine) and machine._include_in_cli]
+)
 _PROVIDER_VALUES = tuple([provider.value for provider in CloudProvider])
 
 

lightning_sdk/cli/deploy/serve.py

@@ -25,7 +25,9 @@ from lightning_sdk.cli.deploy._auth import (
 from lightning_sdk.cli.deploy.devbox import _handle_devbox
 from lightning_sdk.serve import _LitServeDeployer
 
-_MACHINE_VALUES = tuple([machine.name for machine in Machine.__dict__.values() if isinstance(machine, Machine)])
+_MACHINE_VALUES = tuple(
+    [machine.name for machine in Machine.__dict__.values() if isinstance(machine, Machine) and machine._include_in_cli]
+)
 
 
 class _ServeGroup(click.Group):

lightning_sdk/cli/download.py

@@ -17,6 +17,27 @@ from lightning_sdk.studio import Studio
 from lightning_sdk.utils.resolve import _get_authed_user, skip_studio_init
 
 
+def _expand_remote_path(path: str) -> str:
+    """Expand and normalize remote CLI paths.
+
+    - Strips leading `~/` or `~`
+    - Expands `~` to the user's home but returns relative to it
+    - Returns an empty string if path is empty or `~`
+    """
+    if not path:
+        return ""
+
+    local_home = os.path.expanduser("~")
+
+    # Expand to absolute path and remove the local home prefix if present
+    path = os.path.expanduser(path)
+    if path.startswith(local_home):
+        path = path[len(local_home) :]
+
+    # Remove any leading "/" or "~" remnants
+    return path.lstrip("/~")
+
+
 @click.group(name="download")
 def download() -> None:
     """Download resources from Lightning AI."""
@@ -86,17 +107,20 @@ def folder(
         raise ValueError("Either --studio or --teamspace must be provided, not both")
 
     if studio:
+        path = _expand_remote_path(path)
         resolved_downloader = _resolve_studio(studio)
     elif teamspace:
         menu = _TeamspacesMenu()
         resolved_downloader = menu._resolve_teamspace(teamspace)
+    else:
+        raise ValueError("Either --studio or --teamspace must be provided")
 
     if not path:
         local_path /= resolved_downloader.name
         path = ""
 
     try:
-        if not path:
+        if not path and teamspace:
             raise FileNotFoundError()
         resolved_downloader.download_folder(remote_path=path, target_path=str(local_path))
     except Exception as e:

lightning_sdk/cli/entrypoint.py

@@ -50,7 +50,9 @@ def _notify_exception(exception_type: Type[BaseException], value: BaseException,
         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."))
+        renderables.append(Text("\n\n🐞 To view the full traceback, set: LIGHTNING_DEBUG=1"))
+
+    renderables.append(Text("\n📘 Need help? Run: lightning <command> --help", style="cyan"))
 
     console.print(Panel(Group(*renderables), title="⚡ Lightning CLI Error", border_style="red"))
 

lightning_sdk/cli/list.py

@@ -274,7 +274,11 @@ def machines() -> None:
     table.add_column("Name")
 
     # Get all machine types from the enum
-    machine_types = [name for name in dir(Machine) if isinstance(getattr(Machine, name), Machine)]
+    machine_types = [
+        name
+        for name in dir(Machine)
+        if isinstance(getattr(Machine, name), Machine) and getattr(Machine, name)._include_in_cli
+    ]
 
     # Add rows to table
     for name in sorted(machine_types):

lightning_sdk/cli/run.py

@@ -8,7 +8,9 @@ from lightning_sdk.machine import Machine
 from lightning_sdk.mmt import MMT
 from lightning_sdk.teamspace import Teamspace
 
-_MACHINE_VALUES = tuple([machine.name for machine in Machine.__dict__.values() if isinstance(machine, Machine)])
+_MACHINE_VALUES = tuple(
+    [machine.name for machine in Machine.__dict__.values() if isinstance(machine, Machine) and machine._include_in_cli]
+)
 
 
 @click.group(name="run")

lightning_sdk/cli/start.py

@@ -6,7 +6,9 @@ from lightning_sdk import Machine, Studio
 from lightning_sdk.lightning_cloud.openapi.rest import ApiException
 from lightning_sdk.machine import CloudProvider
 
-_MACHINE_VALUES = tuple([machine.name for machine in Machine.__dict__.values() if isinstance(machine, Machine)])
+_MACHINE_VALUES = tuple(
+    [machine.name for machine in Machine.__dict__.values() if isinstance(machine, Machine) and machine._include_in_cli]
+)
 _PROVIDER_VALUES = tuple([provider.value for provider in CloudProvider])
 
 

lightning_sdk/cli/switch.py

@@ -5,7 +5,9 @@ import click
 from lightning_sdk import Machine, Studio
 from lightning_sdk.lightning_cloud.openapi.rest import ApiException
 
-_MACHINE_VALUES = tuple([machine.name for machine in Machine.__dict__.values() if isinstance(machine, Machine)])
+_MACHINE_VALUES = tuple(
+    [machine.name for machine in Machine.__dict__.values() if isinstance(machine, Machine) and machine._include_in_cli]
+)
 
 
 @click.group("switch")

lightning_sdk/job/v2.py

@@ -4,6 +4,7 @@ from lightning_sdk.api.job_api import JobApiV2
 from lightning_sdk.api.utils import _get_cloud_url
 from lightning_sdk.job.base import _BaseJob
 from lightning_sdk.status import Status
+from lightning_sdk.utils.resolve import _get_org_id
 
 if TYPE_CHECKING:
     from lightning_sdk.machine import CloudProvider, Machine
@@ -165,7 +166,12 @@ class _JobV2(_BaseJob):
     def machine(self) -> Union["Machine", str]:
         """The machine type the job is running on."""
         # only fetch the job it it hasn't been fetched yet as machine cannot change over time
-        return self._job_api._get_job_machine_from_spec(self._guaranteed_job.spec)
+
+        return self._job_api._get_job_machine_from_spec(
+            self._guaranteed_job.spec,
+            self.teamspace.id,
+            _get_org_id(self.teamspace),
+        )
 
     @property
     def artifact_path(self) -> Optional[str]:

lightning_sdk/job/work.py

@@ -1,6 +1,7 @@
 from typing import TYPE_CHECKING, Any, Optional, Protocol, Union
 
 from lightning_sdk.api.job_api import JobApiV1
+from lightning_sdk.utils.resolve import _get_org_id
 
 if TYPE_CHECKING:
     from lightning_sdk.job.base import MachineDict
@@ -52,7 +53,10 @@ class Work:
 
     @property
     def machine(self) -> Union["Machine", str]:
-        return self._job_api.get_machine_from_work(self._guaranteed_work)
+        return self._job_api.get_machine_from_work(
+            self._guaranteed_work,
+            org_id=_get_org_id(self._teamspace),
+        )
 
     @property
     def artifact_path(self) -> Optional[str]:

lightning_sdk/lightning_cloud/openapi/__init__.py

@@ -243,6 +243,7 @@ from lightning_sdk.lightning_cloud.openapi.models.user_id_affiliatelinks_body im
 from lightning_sdk.lightning_cloud.openapi.models.user_id_membershiprolebindings_body import UserIdMembershiprolebindingsBody
 from lightning_sdk.lightning_cloud.openapi.models.user_id_membershiprolebindings_body1 import UserIdMembershiprolebindingsBody1
 from lightning_sdk.lightning_cloud.openapi.models.user_id_upgradetrigger_body import UserIdUpgradetriggerBody
+from lightning_sdk.lightning_cloud.openapi.models.user_user_id_body import UserUserIdBody
 from lightning_sdk.lightning_cloud.openapi.models.v1_aws_cluster_credentials import V1AWSClusterCredentials
 from lightning_sdk.lightning_cloud.openapi.models.v1_aws_direct_v1 import V1AWSDirectV1
 from lightning_sdk.lightning_cloud.openapi.models.v1_aws_direct_v1_status import V1AWSDirectV1Status
@@ -325,6 +326,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_cloud_space_version import
 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_cloudy_settings import V1CloudySettings
 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
@@ -533,6 +535,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_filesystem_work import V1Fi
 from lightning_sdk.lightning_cloud.openapi.models.v1_find_capacity_block_offering_response import V1FindCapacityBlockOfferingResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_flowserver import V1Flowserver
 from lightning_sdk.lightning_cloud.openapi.models.v1_folder_index_status import V1FolderIndexStatus
+from lightning_sdk.lightning_cloud.openapi.models.v1_function_call import V1FunctionCall
 from lightning_sdk.lightning_cloud.openapi.models.v1_function_tool import V1FunctionTool
 from lightning_sdk.lightning_cloud.openapi.models.v1_gcp_direct_vpc import V1GCPDirectVPC
 from lightning_sdk.lightning_cloud.openapi.models.v1_gcs_folder_data_connection import V1GCSFolderDataConnection
@@ -555,6 +558,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_get_cloud_space_instance_sy
 from lightning_sdk.lightning_cloud.openapi.models.v1_get_cloud_space_size_response import V1GetCloudSpaceSizeResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_get_cluster_accelerator_demand_response import V1GetClusterAcceleratorDemandResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_get_cluster_credentials_response import V1GetClusterCredentialsResponse
+from lightning_sdk.lightning_cloud.openapi.models.v1_get_cluster_health_response import V1GetClusterHealthResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_get_deployment_routing_telemetry_aggregated_response import V1GetDeploymentRoutingTelemetryAggregatedResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_get_deployment_routing_telemetry_content_response import V1GetDeploymentRoutingTelemetryContentResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_get_deployment_routing_telemetry_response import V1GetDeploymentRoutingTelemetryResponse
@@ -729,7 +733,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_list_project_memberships_re
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_project_roles_response import V1ListProjectRolesResponse
 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_published_managed_endpoint_models_response import V1ListPublishedManagedEndpointModelsResponse
+from lightning_sdk.lightning_cloud.openapi.models.v1_list_published_managed_endpoints_response import V1ListPublishedManagedEndpointsResponse
 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
@@ -930,6 +934,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_telemetry import V1Telemetr
 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_tool import V1Tool
+from lightning_sdk.lightning_cloud.openapi.models.v1_tool_call import V1ToolCall
 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