lightning-sdk 2025.9.4__py3-none-any.whl → 2025.9.10__py3-none-any.whl

lightning_sdk/__init__.py

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

lightning_sdk/api/llm_api.py

@@ -16,6 +16,7 @@ from lightning_sdk.lightning_cloud.login import Auth, AuthServer
 from lightning_sdk.lightning_cloud.openapi.models import (
     StreamResultOfV1ConversationResponseChunk,
     V1ConversationResponseChunk,
+    V1ManagedModel,
     V1ResponseChoice,
     V1ResponseChoiceDelta,
 )
@@ -67,13 +68,23 @@ def authenticate(model: str, shall_confirm: bool = True) -> None:
 class LLMApi:
     def __init__(self) -> None:
         self._client = LightningClient(retry=False, max_tries=0)
+        self._assistant = None
+        self._model = None
 
     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
         )
+        self._assistant = result
         return result.id
 
+    def get_model_metadata(self, teamspace_id: str, model_name: str) -> V1ManagedModel:
+        if self._assistant and not self._model:
+            self._model = self._client.assistants_service_get_managed_model_by_name(
+                teamspace_id, self._assistant.managed_endpoint_id, model_name
+            )
+        return self._model
+
     def _parse_stream_line(self, decoded_line: str) -> Optional[V1ConversationResponseChunk]:
         try:
             payload = json.loads(decoded_line)
@@ -132,6 +143,7 @@ class LLMApi:
         metadata: Optional[Dict[str, str]] = None,
         stream: bool = False,
         tools: Optional[List[Dict[str, Any]]] = None,
+        reasoning_effort: Optional[str] = None,
         **kwargs: Any,
     ) -> Union[V1ConversationResponseChunk, Generator[V1ConversationResponseChunk, None, None]]:
         is_internal_conversation = os.getenv("LIGHTNING_INTERNAL_CONVERSATION", "false").lower() == "true"
@@ -161,6 +173,9 @@ class LLMApi:
         if max_completion_tokens is not None:
             body["max_completion_tokens"] = max_completion_tokens
 
+        if reasoning_effort:
+            body["reasoning_effort"] = reasoning_effort
+
         if images:
             for image in images:
                 url = image
@@ -191,6 +206,7 @@ class LLMApi:
         name: Optional[str] = None,
         metadata: Optional[Dict[str, str]] = None,
         stream: bool = False,
+        reasoning_effort: Optional[str] = None,
         **kwargs: Any,
     ) -> Union[V1ConversationResponseChunk, AsyncGenerator[V1ConversationResponseChunk, None]]:
         is_internal_conversation = os.getenv("LIGHTNING_INTERNAL_CONVERSATION", "false").lower() == "true"
@@ -220,6 +236,9 @@ class LLMApi:
         if max_completion_tokens is not None:
             body["max_completion_tokens"] = max_completion_tokens
 
+        if reasoning_effort:
+            body["reasoning_effort"] = reasoning_effort
+
         if images:
             for image in images:
                 url = image

lightning_sdk/api/studio_api.py

@@ -261,12 +261,19 @@ class StudioApi:
         return getattr(getattr(studio.code_status, "in_use", None), "phase", None)
 
     def _request_switch(
-        self, studio_id: str, teamspace_id: str, machine: Union[Machine, str], interruptible: bool
+        self,
+        studio_id: str,
+        teamspace_id: str,
+        machine: Union[Machine, str],
+        interruptible: bool,
+        cloud_account: Optional[str],
     ) -> None:
         """Switches given Studio to a new machine type."""
         compute_name = _machine_to_compute_name(machine)
         # TODO: UI sends disk size here, maybe we need to also?
         body = IdCodeconfigBody(compute_config=V1UserRequestedComputeConfig(name=compute_name, spot=interruptible))
+        if cloud_account:
+            body.compute_config.cluster_override = cloud_account
         self._client.cloud_space_service_update_cloud_space_instance_config(
             id=studio_id,
             project_id=teamspace_id,
@@ -274,11 +281,20 @@ class StudioApi:
         )
 
     def switch_studio_machine(
-        self, studio_id: str, teamspace_id: str, machine: Union[Machine, str], interruptible: bool
+        self,
+        studio_id: str,
+        teamspace_id: str,
+        machine: Union[Machine, str],
+        interruptible: bool,
+        cloud_account: Optional[str],
     ) -> None:
         """Switches given Studio to a new machine type."""
         self._request_switch(
-            studio_id=studio_id, teamspace_id=teamspace_id, machine=machine, interruptible=interruptible
+            studio_id=studio_id,
+            teamspace_id=teamspace_id,
+            machine=machine,
+            interruptible=interruptible,
+            cloud_account=cloud_account,
         )
 
         # Wait until it's time to switch
@@ -324,12 +340,17 @@ class StudioApi:
         machine: Union[Machine, str],
         interruptible: bool,
         progress: Any,  # StudioProgressTracker - avoid circular import
+        cloud_account: Optional[str],
     ) -> None:
         """Switches given Studio to a new machine type with progress tracking."""
         progress.update_progress(10, "Requesting machine switch...")
 
         self._request_switch(
-            studio_id=studio_id, teamspace_id=teamspace_id, machine=machine, interruptible=interruptible
+            studio_id=studio_id,
+            teamspace_id=teamspace_id,
+            machine=machine,
+            interruptible=interruptible,
+            cloud_account=cloud_account,
         )
 
         progress.update_progress(20, "Waiting for machine allocation...")

lightning_sdk/lightning_cloud/openapi/__init__.py

@@ -146,6 +146,7 @@ from lightning_sdk.lightning_cloud.openapi.models.id_output_body import IdOutput
 from lightning_sdk.lightning_cloud.openapi.models.id_publications_body import IdPublicationsBody
 from lightning_sdk.lightning_cloud.openapi.models.id_publications_body1 import IdPublicationsBody1
 from lightning_sdk.lightning_cloud.openapi.models.id_release_body import IdReleaseBody
+from lightning_sdk.lightning_cloud.openapi.models.id_render_body import IdRenderBody
 from lightning_sdk.lightning_cloud.openapi.models.id_reportlogsactivity_body import IdReportlogsactivityBody
 from lightning_sdk.lightning_cloud.openapi.models.id_reportrestarttimings_body import IdReportrestarttimingsBody
 from lightning_sdk.lightning_cloud.openapi.models.id_sleepconfig_body import IdSleepconfigBody
@@ -159,6 +160,7 @@ from lightning_sdk.lightning_cloud.openapi.models.jobs_id_body import JobsIdBody
 from lightning_sdk.lightning_cloud.openapi.models.jobs_id_body1 import JobsIdBody1
 from lightning_sdk.lightning_cloud.openapi.models.jobs_id_body2 import JobsIdBody2
 from lightning_sdk.lightning_cloud.openapi.models.jobs_id_body3 import JobsIdBody3
+from lightning_sdk.lightning_cloud.openapi.models.kubernetestemplates_id_body import KubernetestemplatesIdBody
 from lightning_sdk.lightning_cloud.openapi.models.litdatasets_dataset_id_body import LitdatasetsDatasetIdBody
 from lightning_sdk.lightning_cloud.openapi.models.litloggermetrics_id_body import LitloggermetricsIdBody
 from lightning_sdk.lightning_cloud.openapi.models.litpages_id_body import LitpagesIdBody
@@ -193,6 +195,7 @@ from lightning_sdk.lightning_cloud.openapi.models.project_id_endpoints_body impo
 from lightning_sdk.lightning_cloud.openapi.models.project_id_getapp_body import ProjectIdGetappBody
 from lightning_sdk.lightning_cloud.openapi.models.project_id_invite_body import ProjectIdInviteBody
 from lightning_sdk.lightning_cloud.openapi.models.project_id_jobs_body import ProjectIdJobsBody
+from lightning_sdk.lightning_cloud.openapi.models.project_id_kubernetestemplates_body import ProjectIdKubernetestemplatesBody
 from lightning_sdk.lightning_cloud.openapi.models.project_id_litdatasets_body import ProjectIdLitdatasetsBody
 from lightning_sdk.lightning_cloud.openapi.models.project_id_litregistry_body import ProjectIdLitregistryBody
 from lightning_sdk.lightning_cloud.openapi.models.project_id_memberships_body import ProjectIdMembershipsBody
@@ -433,6 +436,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_delete_endpoint_response im
 from lightning_sdk.lightning_cloud.openapi.models.v1_delete_git_credentials_response import V1DeleteGitCredentialsResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_delete_index_response import V1DeleteIndexResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_delete_job_response import V1DeleteJobResponse
+from lightning_sdk.lightning_cloud.openapi.models.v1_delete_kubernetes_template_response import V1DeleteKubernetesTemplateResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_delete_lightning_run_response import V1DeleteLightningRunResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_delete_lightningapp_instance_artifact_response import V1DeleteLightningappInstanceArtifactResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_delete_lightningapp_instance_response import V1DeleteLightningappInstanceResponse
@@ -531,6 +535,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_filesystem_slurm_job import
 from lightning_sdk.lightning_cloud.openapi.models.v1_filesystem_snowflake_connection import V1FilesystemSnowflakeConnection
 from lightning_sdk.lightning_cloud.openapi.models.v1_filesystem_work import V1FilesystemWork
 from lightning_sdk.lightning_cloud.openapi.models.v1_find_capacity_block_offering_response import V1FindCapacityBlockOfferingResponse
+from lightning_sdk.lightning_cloud.openapi.models.v1_firewall_rule import V1FirewallRule
 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
@@ -606,6 +611,9 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_health_check_http_get impor
 from lightning_sdk.lightning_cloud.openapi.models.v1_ids_logger_metrics import V1IdsLoggerMetrics
 from lightning_sdk.lightning_cloud.openapi.models.v1_image_spec import V1ImageSpec
 from lightning_sdk.lightning_cloud.openapi.models.v1_image_state import V1ImageState
+from lightning_sdk.lightning_cloud.openapi.models.v1_incident_event import V1IncidentEvent
+from lightning_sdk.lightning_cloud.openapi.models.v1_incident_severity import V1IncidentSeverity
+from lightning_sdk.lightning_cloud.openapi.models.v1_incident_type import V1IncidentType
 from lightning_sdk.lightning_cloud.openapi.models.v1_index import V1Index
 from lightning_sdk.lightning_cloud.openapi.models.v1_input import V1Input
 from lightning_sdk.lightning_cloud.openapi.models.v1_instance_overprovisioning_spec import V1InstanceOverprovisioningSpec
@@ -625,10 +633,13 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_job_spec import V1JobSpec
 from lightning_sdk.lightning_cloud.openapi.models.v1_job_timing import V1JobTiming
 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_k8s_incident_indexes import V1K8sIncidentIndexes
 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_kubernetes_template import V1KubernetesTemplate
+from lightning_sdk.lightning_cloud.openapi.models.v1_kubernetes_template_property import V1KubernetesTemplateProperty
 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
@@ -702,10 +713,12 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_list_filesystem_snowflake_r
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_gallery_components_response import V1ListGalleryComponentsResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_gallery_lightningapps_response import V1ListGalleryLightningappsResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_git_credentials_response import V1ListGitCredentialsResponse
+from lightning_sdk.lightning_cloud.openapi.models.v1_list_incident_events_response import V1ListIncidentEventsResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_job_files_response import V1ListJobFilesResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_job_resources_response import V1ListJobResourcesResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_jobs_response import V1ListJobsResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_joinable_organizations_response import V1ListJoinableOrganizationsResponse
+from lightning_sdk.lightning_cloud.openapi.models.v1_list_kubernetes_templates_response import V1ListKubernetesTemplatesResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_lightning_run_response import V1ListLightningRunResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_lightningapp_instance_artifacts_response import V1ListLightningappInstanceArtifactsResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_list_lightningapp_instance_events_response import V1ListLightningappInstanceEventsResponse
@@ -879,6 +892,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_refresh_request import V1Re
 from lightning_sdk.lightning_cloud.openapi.models.v1_refresh_response import V1RefreshResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_region_state import V1RegionState
 from lightning_sdk.lightning_cloud.openapi.models.v1_regional_load_balancer import V1RegionalLoadBalancer
+from lightning_sdk.lightning_cloud.openapi.models.v1_render_kubernetes_template_response import V1RenderKubernetesTemplateResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_report_cloud_space_instance_stop_at_response import V1ReportCloudSpaceInstanceStopAtResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_report_cloud_space_instance_system_metrics_response import V1ReportCloudSpaceInstanceSystemMetricsResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_report_k8s_cluster_metrics_response import V1ReportK8sClusterMetricsResponse