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

lightning_sdk/__init__.py

@@ -32,6 +32,6 @@ __all__ = [
     "User",
 ]
 
-__version__ = "2025.09.04"
+__version__ = "2025.09.11"
 _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/login.py

@@ -17,6 +17,11 @@ from starlette.background import BackgroundTask
 from starlette.responses import RedirectResponse
 
 from lightning_sdk.lightning_cloud import env
+from lightning_sdk.lightning_cloud.openapi import ApiClient, Configuration
+from lightning_sdk.lightning_cloud.openapi.api import \
+    AuthServiceApi
+from lightning_sdk.lightning_cloud.openapi.models.v1_guest_login_request import \
+    V1GuestLoginRequest
 
 logger = logging.getLogger(__name__)
 
@@ -132,6 +137,61 @@ class Auth:
             "We couldn't find any credentials linked to your account. Please try logging in using the CLI command `lightning login`"
         )
 
+    def guest_login(self) -> Optional[str]:
+        """Performs guest user authentication.
+        This method sends a request to the guest login endpoint to get temporary
+        credentials, saves them, and returns the authorization header.
+        Useful to log experiments as a non signed in user, using a guest account
+        in the background.
+        Returns
+        -------
+        Optional[str]
+            The authorization header to use for subsequent requests.
+        Raises
+        ------
+        RuntimeError
+            If the guest login request fails.
+        ValueError
+            If the response from the server is invalid.
+        """
+
+        config = Configuration()
+        config.host = env.LIGHTNING_CLOUD_URL
+        api_client = ApiClient(configuration=config)
+        auth_api = AuthServiceApi(api_client)
+
+        logger.debug(f"Attempting guest login to {config.host}")
+
+        try:
+            # The body is an empty object for a guest login.
+            body = V1GuestLoginRequest()
+            credentials = auth_api.auth_service_guest_login(body)
+
+        except requests.RequestException as e:
+            logger.error(f"Guest login request failed: {e}")
+            raise RuntimeError(
+                "Failed to connect to the guest login endpoint. "
+                "Please check your network connection and the server status."
+            ) from e
+
+        # attributes based on the `V1GuestLoginResponse` model.
+        user = getattr(credentials, "user", None)
+        user_id = getattr(user, "id", None) if user else None
+        api_key = getattr(user, "api_key", None) if user else None
+
+        if not all([user_id, api_key]):
+            logger.error(
+                f"Incomplete credentials received from guest login: {credentials}"
+            )
+            raise ValueError(
+                "The guest login response did not contain the required 'user_id' and 'api_key' fields."
+            )
+
+        self.save(user_id=user_id, api_key=api_key)
+        logger.info("Successfully authenticated as a guest user.")
+
+        return self.auth_header
+
 
 class AuthServer:
 

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
@@ -204,6 +207,7 @@ from lightning_sdk.lightning_cloud.openapi.models.project_id_schedules_body impo
 from lightning_sdk.lightning_cloud.openapi.models.project_id_secrets_body import ProjectIdSecretsBody
 from lightning_sdk.lightning_cloud.openapi.models.project_id_snowflake_body import ProjectIdSnowflakeBody
 from lightning_sdk.lightning_cloud.openapi.models.project_id_storage_body import ProjectIdStorageBody
+from lightning_sdk.lightning_cloud.openapi.models.project_tab_management_messages import ProjectTabManagementMessages
 from lightning_sdk.lightning_cloud.openapi.models.projects_id_body import ProjectsIdBody
 from lightning_sdk.lightning_cloud.openapi.models.projects_project_id_body import ProjectsProjectIdBody
 from lightning_sdk.lightning_cloud.openapi.models.protobuf_any import ProtobufAny
@@ -433,6 +437,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 +536,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 +612,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 +634,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 +714,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
@@ -857,6 +871,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_project_membership_invite i
 from lightning_sdk.lightning_cloud.openapi.models.v1_project_membership_role_binding import V1ProjectMembershipRoleBinding
 from lightning_sdk.lightning_cloud.openapi.models.v1_project_settings import V1ProjectSettings
 from lightning_sdk.lightning_cloud.openapi.models.v1_project_storage import V1ProjectStorage
+from lightning_sdk.lightning_cloud.openapi.models.v1_project_tab import V1ProjectTab
 from lightning_sdk.lightning_cloud.openapi.models.v1_prompt_suggestion import V1PromptSuggestion
 from lightning_sdk.lightning_cloud.openapi.models.v1_publish_cloud_space_response import V1PublishCloudSpaceResponse
 from lightning_sdk.lightning_cloud.openapi.models.v1_published_cloud_space_response import V1PublishedCloudSpaceResponse
@@ -879,6 +894,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
@@ -988,6 +1004,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_update_metrics_stream_visib
 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_project_tab_order_response import V1UpdateProjectTabOrderResponse
 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