lightning-sdk 2025.8.1__py3-none-any.whl → 2025.8.6rc1__py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (65) hide show
  1. lightning_sdk/__init__.py +1 -1
  2. lightning_sdk/api/deployment_api.py +6 -4
  3. lightning_sdk/api/job_api.py +2 -0
  4. lightning_sdk/api/mmt_api.py +2 -0
  5. lightning_sdk/deployment/deployment.py +5 -0
  6. lightning_sdk/lightning_cloud/login.py +1 -7
  7. lightning_sdk/lightning_cloud/openapi/__init__.py +15 -23
  8. lightning_sdk/lightning_cloud/openapi/api/__init__.py +1 -0
  9. lightning_sdk/lightning_cloud/openapi/api/assistants_service_api.py +5 -1
  10. lightning_sdk/lightning_cloud/openapi/api/cloud_space_service_api.py +228 -1
  11. lightning_sdk/lightning_cloud/openapi/api/endpoint_service_api.py +126 -2129
  12. lightning_sdk/lightning_cloud/openapi/api/k8_s_cluster_service_api.py +625 -0
  13. lightning_sdk/lightning_cloud/openapi/api/storage_service_api.py +5 -1
  14. lightning_sdk/lightning_cloud/openapi/models/__init__.py +14 -23
  15. lightning_sdk/lightning_cloud/openapi/models/cluster_id_metrics_body.py +175 -0
  16. lightning_sdk/lightning_cloud/openapi/models/orgs_id_body.py +53 -1
  17. lightning_sdk/lightning_cloud/openapi/models/projects_id_body.py +53 -1
  18. lightning_sdk/lightning_cloud/openapi/models/v1_cloud_space.py +27 -1
  19. lightning_sdk/lightning_cloud/openapi/models/v1_cloud_space_transfer_metadata.py +201 -0
  20. lightning_sdk/lightning_cloud/openapi/models/v1_cloudflare_v1.py +29 -3
  21. lightning_sdk/lightning_cloud/openapi/models/{v1_complete_upload_service_execution_artifact_response.py → v1_complete_running_cloud_space_instance_transfer_response.py} +6 -6
  22. lightning_sdk/lightning_cloud/openapi/models/v1_container_metrics.py +461 -0
  23. lightning_sdk/lightning_cloud/openapi/models/v1_filesystem_metrics.py +227 -0
  24. lightning_sdk/lightning_cloud/openapi/models/{v1_list_file_endpoints_response.py → v1_list_container_metrics_response.py} +23 -23
  25. lightning_sdk/lightning_cloud/openapi/models/{service_artifact_artifact_kind.py → v1_list_node_file_system_metrics_response.py} +6 -13
  26. lightning_sdk/lightning_cloud/openapi/models/{v1_upload_service_execution_artifact_parts_response.py → v1_list_node_metrics_response.py} +23 -23
  27. lightning_sdk/lightning_cloud/openapi/models/v1_list_pod_metrics_response.py +123 -0
  28. lightning_sdk/lightning_cloud/openapi/models/v1_managed_model.py +79 -1
  29. lightning_sdk/lightning_cloud/openapi/models/v1_node_metrics.py +357 -0
  30. lightning_sdk/lightning_cloud/openapi/models/v1_organization.py +53 -1
  31. lightning_sdk/lightning_cloud/openapi/models/v1_pod_metrics.py +435 -0
  32. lightning_sdk/lightning_cloud/openapi/models/v1_project_settings.py +53 -1
  33. lightning_sdk/lightning_cloud/openapi/models/{v1_delete_service_execution_response.py → v1_report_k8s_cluster_metrics_response.py} +6 -6
  34. lightning_sdk/lightning_cloud/openapi/models/{command_argument_command_argument_type.py → v1_request_cloud_space_access_response.py} +6 -12
  35. lightning_sdk/lightning_cloud/openapi/models/v1_update_cloud_space_instance_config_request.py +331 -0
  36. lightning_sdk/lightning_cloud/openapi/models/v1_user_features.py +157 -53
  37. lightning_sdk/lightning_cloud/utils/data_connection.py +56 -1
  38. lightning_sdk/llm/llm.py +2 -13
  39. lightning_sdk/llm/public_assistants.py +17 -0
  40. lightning_sdk/pipeline/steps.py +30 -1
  41. lightning_sdk/services/file_endpoint.py +3 -4
  42. {lightning_sdk-2025.8.1.dist-info → lightning_sdk-2025.8.6rc1.dist-info}/METADATA +1 -1
  43. {lightning_sdk-2025.8.1.dist-info → lightning_sdk-2025.8.6rc1.dist-info}/RECORD +47 -55
  44. lightning_sdk/lightning_cloud/openapi/models/fileendpoints_id_body.py +0 -409
  45. lightning_sdk/lightning_cloud/openapi/models/id_complete_body.py +0 -175
  46. lightning_sdk/lightning_cloud/openapi/models/id_storage_body.py +0 -175
  47. lightning_sdk/lightning_cloud/openapi/models/id_uploads_body1.py +0 -175
  48. lightning_sdk/lightning_cloud/openapi/models/project_id_fileendpoints_body.py +0 -357
  49. lightning_sdk/lightning_cloud/openapi/models/project_id_serviceexecution_body.py +0 -175
  50. lightning_sdk/lightning_cloud/openapi/models/serviceexecution_id_body.py +0 -331
  51. lightning_sdk/lightning_cloud/openapi/models/v1_command_argument.py +0 -305
  52. lightning_sdk/lightning_cloud/openapi/models/v1_delete_file_endpoint_response.py +0 -149
  53. lightning_sdk/lightning_cloud/openapi/models/v1_download_service_execution_artifact_response.py +0 -175
  54. lightning_sdk/lightning_cloud/openapi/models/v1_file_endpoint.py +0 -461
  55. lightning_sdk/lightning_cloud/openapi/models/v1_get_service_execution_status_response.py +0 -175
  56. lightning_sdk/lightning_cloud/openapi/models/v1_list_service_execution_lightningapp_instances_response.py +0 -175
  57. lightning_sdk/lightning_cloud/openapi/models/v1_list_service_executions_response.py +0 -123
  58. lightning_sdk/lightning_cloud/openapi/models/v1_service_artifact.py +0 -201
  59. lightning_sdk/lightning_cloud/openapi/models/v1_service_execution.py +0 -383
  60. lightning_sdk/lightning_cloud/openapi/models/v1_upload_service_execution_artifact_response.py +0 -149
  61. lightning_sdk/llm/public_assistants.json +0 -34
  62. {lightning_sdk-2025.8.1.dist-info → lightning_sdk-2025.8.6rc1.dist-info}/LICENSE +0 -0
  63. {lightning_sdk-2025.8.1.dist-info → lightning_sdk-2025.8.6rc1.dist-info}/WHEEL +0 -0
  64. {lightning_sdk-2025.8.1.dist-info → lightning_sdk-2025.8.6rc1.dist-info}/entry_points.txt +0 -0
  65. {lightning_sdk-2025.8.1.dist-info → lightning_sdk-2025.8.6rc1.dist-info}/top_level.txt +0 -0
lightning_sdk/__init__.py CHANGED
@@ -32,6 +32,6 @@ __all__ = [
32
32
  "User",
33
33
  ]
34
34
 
35
- __version__ = "2025.08.01"
35
+ __version__ = "2025.08.06rc1"
36
36
  _check_version_and_prompt_upgrade(__version__)
37
37
  _set_tqdm_envvars_noninteractive()
@@ -103,7 +103,7 @@ class ExecHealthCheck(HealthCheck):
103
103
  self,
104
104
  command: str,
105
105
  initial_delay_seconds: int = 0,
106
- failure_threshold: int = 600,
106
+ failure_threshold: int = 3600,
107
107
  interval_seconds: int = 1,
108
108
  timeout_seconds: int = 30,
109
109
  ) -> None:
@@ -132,7 +132,7 @@ class HttpHealthCheck(HealthCheck):
132
132
  path: str,
133
133
  port: float,
134
134
  initial_delay_seconds: int = 0,
135
- failure_threshold: int = 600,
135
+ failure_threshold: int = 3600,
136
136
  interval_seconds: int = 1,
137
137
  timeout_seconds: int = 30,
138
138
  ) -> None:
@@ -538,10 +538,10 @@ def to_health_check(
538
538
  # Use Default health check if none is provided
539
539
  if not health_check:
540
540
  return V1JobHealthCheckConfig(
541
- failure_threshold=600,
541
+ failure_threshold=3600,
542
542
  initial_delay_seconds=0,
543
543
  interval_seconds=1,
544
- timeout_seconds=600,
544
+ timeout_seconds=60,
545
545
  )
546
546
 
547
547
  health_check_config = V1JobHealthCheckConfig(
@@ -574,6 +574,7 @@ def to_spec(
574
574
  include_credentials: Optional[bool] = None,
575
575
  cloudspace_id: Optional[None] = None,
576
576
  max_runtime: Optional[int] = None,
577
+ machine_image_version: Optional[str] = None,
577
578
  ) -> V1JobSpec:
578
579
  if cloud_account is None:
579
580
  raise ValueError("The cloud account should be defined.")
@@ -607,6 +608,7 @@ def to_spec(
607
608
  quantity=quantity,
608
609
  include_credentials=include_credentials,
609
610
  cloudspace_id=cloudspace_id,
611
+ machine_image_version=machine_image_version,
610
612
  **optional_spec_kwargs,
611
613
  )
612
614
 
@@ -288,6 +288,7 @@ class JobApiV2:
288
288
  artifacts_local: Optional[str], # deprecated in favor of path_mappings
289
289
  artifacts_remote: Optional[str], # deprecated in favor of path_mappings)
290
290
  max_runtime: Optional[int] = None,
291
+ machine_image_version: Optional[str] = None,
291
292
  ) -> ProjectIdJobsBody:
292
293
  env_vars = []
293
294
  if env is not None:
@@ -322,6 +323,7 @@ class JobApiV2:
322
323
  image_cluster_credentials=cloud_account_auth,
323
324
  image_secret_ref=image_credentials or "",
324
325
  path_mappings=path_mappings_list,
326
+ machine_image_version=machine_image_version,
325
327
  **optional_spec_kwargs,
326
328
  )
327
329
  return ProjectIdJobsBody(name=name, spec=spec)
@@ -129,6 +129,7 @@ class MMTApiV2:
129
129
  artifacts_local: Optional[str], # deprecated in favor of path_mappings
130
130
  artifacts_remote: Optional[str], # deprecated in favor of path_mappings
131
131
  max_runtime: Optional[int] = None,
132
+ machine_image_version: Optional[str] = None,
132
133
  ) -> ProjectIdMultimachinejobsBody:
133
134
  env_vars = []
134
135
  if env is not None:
@@ -163,6 +164,7 @@ class MMTApiV2:
163
164
  image_cluster_credentials=cloud_account_auth,
164
165
  image_secret_ref=image_credentials or "",
165
166
  path_mappings=path_mappings_list,
167
+ machine_image_version=machine_image_version,
166
168
  **optional_spec_kwargs,
167
169
  )
168
170
  return ProjectIdMultimachinejobsBody(
@@ -171,14 +171,18 @@ class Deployment:
171
171
  if self._is_created:
172
172
  raise RuntimeError("This deployment has already been started.")
173
173
 
174
+ machine_image_version = None
175
+
174
176
  if isinstance(studio, Studio):
175
177
  cloudspace_id = studio._studio.id
176
178
  cloud_account = studio._studio.cluster_id
179
+ machine_image_version = studio._studio.machine_image_version
177
180
 
178
181
  if isinstance(studio, str):
179
182
  studio = Studio(studio)
180
183
  cloudspace_id = studio._studio.id
181
184
  cloud_account = studio._studio.cluster_id
185
+ machine_image_version = studio._studio.machine_image_version
182
186
 
183
187
  if cloud_account is None:
184
188
  cloud_account = _resolve_deprecated_cluster(cloud_account, cluster)
@@ -233,6 +237,7 @@ class Deployment:
233
237
  include_credentials=include_credentials if include_credentials is not None else True,
234
238
  cloudspace_id=cloudspace_id,
235
239
  max_runtime=max_runtime,
240
+ machine_image_version=machine_image_version,
236
241
  ),
237
242
  strategy=to_strategy(release_strategy),
238
243
  ),
@@ -44,13 +44,7 @@ class Auth:
44
44
  setattr(self, key.suffix, os.environ.get(key.value, None))
45
45
 
46
46
  # used by authenticate method
47
- self._with_env_var = bool(self.user_id and self.api_key)
48
- if self.api_key and not self.user_id:
49
- raise ValueError(
50
- f"{Keys.USER_ID.value} is missing from env variables. "
51
- "To use env vars for authentication both "
52
- f"{Keys.USER_ID.value} and {Keys.API_KEY.value} should be set."
53
- )
47
+ self._with_env_var = bool(self.api_key)
54
48
 
55
49
  def load(self) -> bool:
56
50
  """Load credentials from disk and update properties with credentials.
@@ -40,6 +40,7 @@ from lightning_sdk.lightning_cloud.openapi.api.experiments_service_api import Ex
40
40
  from lightning_sdk.lightning_cloud.openapi.api.file_system_service_api import FileSystemServiceApi
41
41
  from lightning_sdk.lightning_cloud.openapi.api.git_credentials_service_api import GitCredentialsServiceApi
42
42
  from lightning_sdk.lightning_cloud.openapi.api.jobs_service_api import JobsServiceApi
43
+ from lightning_sdk.lightning_cloud.openapi.api.k8_s_cluster_service_api import K8SClusterServiceApi
43
44
  from lightning_sdk.lightning_cloud.openapi.api.lightningapp_instance_service_api import LightningappInstanceServiceApi
44
45
  from lightning_sdk.lightning_cloud.openapi.api.lightningapp_v2_service_api import LightningappV2ServiceApi
45
46
  from lightning_sdk.lightning_cloud.openapi.api.lightningwork_service_api import LightningworkServiceApi
@@ -95,12 +96,12 @@ from lightning_sdk.lightning_cloud.openapi.models.cloudspace_id_visibility_body
95
96
  from lightning_sdk.lightning_cloud.openapi.models.cloudspaces_id_body import CloudspacesIdBody
96
97
  from lightning_sdk.lightning_cloud.openapi.models.cluster_id_capacityblock_body import ClusterIdCapacityblockBody
97
98
  from lightning_sdk.lightning_cloud.openapi.models.cluster_id_capacityreservations_body import ClusterIdCapacityreservationsBody
99
+ from lightning_sdk.lightning_cloud.openapi.models.cluster_id_metrics_body import ClusterIdMetricsBody
98
100
  from lightning_sdk.lightning_cloud.openapi.models.cluster_id_proxies_body import ClusterIdProxiesBody
99
101
  from lightning_sdk.lightning_cloud.openapi.models.cluster_id_slurmusers_body import ClusterIdSlurmusersBody
100
102
  from lightning_sdk.lightning_cloud.openapi.models.cluster_id_usagerestrictions_body import ClusterIdUsagerestrictionsBody
101
103
  from lightning_sdk.lightning_cloud.openapi.models.clusters_id_body import ClustersIdBody
102
104
  from lightning_sdk.lightning_cloud.openapi.models.clusters_id_body1 import ClustersIdBody1
103
- from lightning_sdk.lightning_cloud.openapi.models.command_argument_command_argument_type import CommandArgumentCommandArgumentType
104
105
  from lightning_sdk.lightning_cloud.openapi.models.conversations_id_body import ConversationsIdBody
105
106
  from lightning_sdk.lightning_cloud.openapi.models.conversations_id_body1 import ConversationsIdBody1
106
107
  from lightning_sdk.lightning_cloud.openapi.models.create import Create
@@ -123,13 +124,11 @@ from lightning_sdk.lightning_cloud.openapi.models.externalv1_cluster import Exte
123
124
  from lightning_sdk.lightning_cloud.openapi.models.externalv1_lightningapp_instance import Externalv1LightningappInstance
124
125
  from lightning_sdk.lightning_cloud.openapi.models.externalv1_lightningwork import Externalv1Lightningwork
125
126
  from lightning_sdk.lightning_cloud.openapi.models.externalv1_user_status import Externalv1UserStatus
126
- from lightning_sdk.lightning_cloud.openapi.models.fileendpoints_id_body import FileendpointsIdBody
127
127
  from lightning_sdk.lightning_cloud.openapi.models.get_endpoint_request_endpoint_ref_type import GetEndpointRequestEndpointRefType
128
128
  from lightning_sdk.lightning_cloud.openapi.models.id_action_body import IdActionBody
129
129
  from lightning_sdk.lightning_cloud.openapi.models.id_artifacts_body import IdArtifactsBody
130
130
  from lightning_sdk.lightning_cloud.openapi.models.id_codeconfig_body import IdCodeconfigBody
131
131
  from lightning_sdk.lightning_cloud.openapi.models.id_collaborate_body import IdCollaborateBody
132
- from lightning_sdk.lightning_cloud.openapi.models.id_complete_body import IdCompleteBody
133
132
  from lightning_sdk.lightning_cloud.openapi.models.id_contactowner_body import IdContactownerBody
134
133
  from lightning_sdk.lightning_cloud.openapi.models.id_engage_body import IdEngageBody
135
134
  from lightning_sdk.lightning_cloud.openapi.models.id_engage_body1 import IdEngageBody1
@@ -150,10 +149,8 @@ from lightning_sdk.lightning_cloud.openapi.models.id_release_body import IdRelea
150
149
  from lightning_sdk.lightning_cloud.openapi.models.id_reportlogsactivity_body import IdReportlogsactivityBody
151
150
  from lightning_sdk.lightning_cloud.openapi.models.id_reportrestarttimings_body import IdReportrestarttimingsBody
152
151
  from lightning_sdk.lightning_cloud.openapi.models.id_start_body import IdStartBody
153
- from lightning_sdk.lightning_cloud.openapi.models.id_storage_body import IdStorageBody
154
152
  from lightning_sdk.lightning_cloud.openapi.models.id_transfer_body import IdTransferBody
155
153
  from lightning_sdk.lightning_cloud.openapi.models.id_uploads_body import IdUploadsBody
156
- from lightning_sdk.lightning_cloud.openapi.models.id_uploads_body1 import IdUploadsBody1
157
154
  from lightning_sdk.lightning_cloud.openapi.models.id_visibility_body import IdVisibilityBody
158
155
  from lightning_sdk.lightning_cloud.openapi.models.id_visibility_body1 import IdVisibilityBody1
159
156
  from lightning_sdk.lightning_cloud.openapi.models.id_visibility_body2 import IdVisibilityBody2
@@ -191,7 +188,6 @@ from lightning_sdk.lightning_cloud.openapi.models.project_id_cloudspaces_body im
191
188
  from lightning_sdk.lightning_cloud.openapi.models.project_id_clusters_body import ProjectIdClustersBody
192
189
  from lightning_sdk.lightning_cloud.openapi.models.project_id_datasets_body import ProjectIdDatasetsBody
193
190
  from lightning_sdk.lightning_cloud.openapi.models.project_id_endpoints_body import ProjectIdEndpointsBody
194
- from lightning_sdk.lightning_cloud.openapi.models.project_id_fileendpoints_body import ProjectIdFileendpointsBody
195
191
  from lightning_sdk.lightning_cloud.openapi.models.project_id_getapp_body import ProjectIdGetappBody
196
192
  from lightning_sdk.lightning_cloud.openapi.models.project_id_invite_body import ProjectIdInviteBody
197
193
  from lightning_sdk.lightning_cloud.openapi.models.project_id_jobs_body import ProjectIdJobsBody
@@ -204,7 +200,6 @@ from lightning_sdk.lightning_cloud.openapi.models.project_id_pipelines_body impo
204
200
  from lightning_sdk.lightning_cloud.openapi.models.project_id_projectclustersbindings_body import ProjectIdProjectclustersbindingsBody
205
201
  from lightning_sdk.lightning_cloud.openapi.models.project_id_schedules_body import ProjectIdSchedulesBody
206
202
  from lightning_sdk.lightning_cloud.openapi.models.project_id_secrets_body import ProjectIdSecretsBody
207
- from lightning_sdk.lightning_cloud.openapi.models.project_id_serviceexecution_body import ProjectIdServiceexecutionBody
208
203
  from lightning_sdk.lightning_cloud.openapi.models.project_id_snowflake_body import ProjectIdSnowflakeBody
209
204
  from lightning_sdk.lightning_cloud.openapi.models.project_id_storage_body import ProjectIdStorageBody
210
205
  from lightning_sdk.lightning_cloud.openapi.models.projects_id_body import ProjectsIdBody
@@ -218,9 +213,7 @@ from lightning_sdk.lightning_cloud.openapi.models.secrets_id_body import Secrets
218
213
  from lightning_sdk.lightning_cloud.openapi.models.secrets_id_body1 import SecretsIdBody1
219
214
  from lightning_sdk.lightning_cloud.openapi.models.server_id_alerts_body import ServerIdAlertsBody
220
215
  from lightning_sdk.lightning_cloud.openapi.models.servers_server_id_body import ServersServerIdBody
221
- from lightning_sdk.lightning_cloud.openapi.models.service_artifact_artifact_kind import ServiceArtifactArtifactKind
222
216
  from lightning_sdk.lightning_cloud.openapi.models.service_health_service_status import ServiceHealthServiceStatus
223
- from lightning_sdk.lightning_cloud.openapi.models.serviceexecution_id_body import ServiceexecutionIdBody
224
217
  from lightning_sdk.lightning_cloud.openapi.models.setup import Setup
225
218
  from lightning_sdk.lightning_cloud.openapi.models.slurm_jobs_body import SlurmJobsBody
226
219
  from lightning_sdk.lightning_cloud.openapi.models.snowflake_export_body import SnowflakeExportBody
@@ -322,6 +315,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_cloud_space_session import
322
315
  from lightning_sdk.lightning_cloud.openapi.models.v1_cloud_space_source_type import V1CloudSpaceSourceType
323
316
  from lightning_sdk.lightning_cloud.openapi.models.v1_cloud_space_specialized_view import V1CloudSpaceSpecializedView
324
317
  from lightning_sdk.lightning_cloud.openapi.models.v1_cloud_space_state import V1CloudSpaceState
318
+ from lightning_sdk.lightning_cloud.openapi.models.v1_cloud_space_transfer_metadata import V1CloudSpaceTransferMetadata
325
319
  from lightning_sdk.lightning_cloud.openapi.models.v1_cloud_space_version import V1CloudSpaceVersion
326
320
  from lightning_sdk.lightning_cloud.openapi.models.v1_cloud_space_version_publication import V1CloudSpaceVersionPublication
327
321
  from lightning_sdk.lightning_cloud.openapi.models.v1_cloudflare_v1 import V1CloudflareV1
@@ -345,21 +339,21 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_cluster_upload import V1Clu
345
339
  from lightning_sdk.lightning_cloud.openapi.models.v1_cluster_usage_restriction import V1ClusterUsageRestriction
346
340
  from lightning_sdk.lightning_cloud.openapi.models.v1_collab_action import V1CollabAction
347
341
  from lightning_sdk.lightning_cloud.openapi.models.v1_collab_session import V1CollabSession
348
- from lightning_sdk.lightning_cloud.openapi.models.v1_command_argument import V1CommandArgument
349
342
  from lightning_sdk.lightning_cloud.openapi.models.v1_complete_agent_multipart_upload import V1CompleteAgentMultipartUpload
350
343
  from lightning_sdk.lightning_cloud.openapi.models.v1_complete_lit_dataset_multi_part_upload_response import V1CompleteLitDatasetMultiPartUploadResponse
351
344
  from lightning_sdk.lightning_cloud.openapi.models.v1_complete_lit_dataset_upload_response import V1CompleteLitDatasetUploadResponse
352
345
  from lightning_sdk.lightning_cloud.openapi.models.v1_complete_model_upload_response import V1CompleteModelUploadResponse
353
346
  from lightning_sdk.lightning_cloud.openapi.models.v1_complete_multi_part_upload_response import V1CompleteMultiPartUploadResponse
354
347
  from lightning_sdk.lightning_cloud.openapi.models.v1_complete_onboarding_response import V1CompleteOnboardingResponse
348
+ from lightning_sdk.lightning_cloud.openapi.models.v1_complete_running_cloud_space_instance_transfer_response import V1CompleteRunningCloudSpaceInstanceTransferResponse
355
349
  from lightning_sdk.lightning_cloud.openapi.models.v1_complete_upload import V1CompleteUpload
356
350
  from lightning_sdk.lightning_cloud.openapi.models.v1_complete_upload_project_artifact_response import V1CompleteUploadProjectArtifactResponse
357
- from lightning_sdk.lightning_cloud.openapi.models.v1_complete_upload_service_execution_artifact_response import V1CompleteUploadServiceExecutionArtifactResponse
358
351
  from lightning_sdk.lightning_cloud.openapi.models.v1_complete_upload_temporary_artifact_request import V1CompleteUploadTemporaryArtifactRequest
359
352
  from lightning_sdk.lightning_cloud.openapi.models.v1_completed_part import V1CompletedPart
360
353
  from lightning_sdk.lightning_cloud.openapi.models.v1_compute_config import V1ComputeConfig
361
354
  from lightning_sdk.lightning_cloud.openapi.models.v1_contact_assistant_owner_reason import V1ContactAssistantOwnerReason
362
355
  from lightning_sdk.lightning_cloud.openapi.models.v1_contact_assistant_owner_response import V1ContactAssistantOwnerResponse
356
+ from lightning_sdk.lightning_cloud.openapi.models.v1_container_metrics import V1ContainerMetrics
363
357
  from lightning_sdk.lightning_cloud.openapi.models.v1_conversation import V1Conversation
364
358
  from lightning_sdk.lightning_cloud.openapi.models.v1_conversation_response_chunk import V1ConversationResponseChunk
365
359
  from lightning_sdk.lightning_cloud.openapi.models.v1_count_metrics_streams_response import V1CountMetricsStreamsResponse
@@ -429,7 +423,6 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_delete_deployment_alerting_
429
423
  from lightning_sdk.lightning_cloud.openapi.models.v1_delete_deployment_release_response import V1DeleteDeploymentReleaseResponse
430
424
  from lightning_sdk.lightning_cloud.openapi.models.v1_delete_deployment_response import V1DeleteDeploymentResponse
431
425
  from lightning_sdk.lightning_cloud.openapi.models.v1_delete_endpoint_response import V1DeleteEndpointResponse
432
- from lightning_sdk.lightning_cloud.openapi.models.v1_delete_file_endpoint_response import V1DeleteFileEndpointResponse
433
426
  from lightning_sdk.lightning_cloud.openapi.models.v1_delete_git_credentials_response import V1DeleteGitCredentialsResponse
434
427
  from lightning_sdk.lightning_cloud.openapi.models.v1_delete_index_response import V1DeleteIndexResponse
435
428
  from lightning_sdk.lightning_cloud.openapi.models.v1_delete_job_response import V1DeleteJobResponse
@@ -464,7 +457,6 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_delete_project_role_respons
464
457
  from lightning_sdk.lightning_cloud.openapi.models.v1_delete_ssh_public_key_response import V1DeleteSSHPublicKeyResponse
465
458
  from lightning_sdk.lightning_cloud.openapi.models.v1_delete_schedule_response import V1DeleteScheduleResponse
466
459
  from lightning_sdk.lightning_cloud.openapi.models.v1_delete_secret_response import V1DeleteSecretResponse
467
- from lightning_sdk.lightning_cloud.openapi.models.v1_delete_service_execution_response import V1DeleteServiceExecutionResponse
468
460
  from lightning_sdk.lightning_cloud.openapi.models.v1_delete_shared_metrics_stream_response import V1DeleteSharedMetricsStreamResponse
469
461
  from lightning_sdk.lightning_cloud.openapi.models.v1_delete_studio_job_response import V1DeleteStudioJobResponse
470
462
  from lightning_sdk.lightning_cloud.openapi.models.v1_delete_user_slurm_job_response import V1DeleteUserSLURMJobResponse
@@ -499,7 +491,6 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_deployment_template_summary
499
491
  from lightning_sdk.lightning_cloud.openapi.models.v1_deployment_template_type import V1DeploymentTemplateType
500
492
  from lightning_sdk.lightning_cloud.openapi.models.v1_download_job_logs_response import V1DownloadJobLogsResponse
501
493
  from lightning_sdk.lightning_cloud.openapi.models.v1_download_lightningapp_instance_logs_response import V1DownloadLightningappInstanceLogsResponse
502
- from lightning_sdk.lightning_cloud.openapi.models.v1_download_service_execution_artifact_response import V1DownloadServiceExecutionArtifactResponse
503
494
  from lightning_sdk.lightning_cloud.openapi.models.v1_drive import V1Drive
504
495
  from lightning_sdk.lightning_cloud.openapi.models.v1_drive_spec import V1DriveSpec
505
496
  from lightning_sdk.lightning_cloud.openapi.models.v1_drive_state import V1DriveState
@@ -522,13 +513,13 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_export_snowflake_query_resp
522
513
  from lightning_sdk.lightning_cloud.openapi.models.v1_external_cluster import V1ExternalCluster
523
514
  from lightning_sdk.lightning_cloud.openapi.models.v1_external_cluster_spec import V1ExternalClusterSpec
524
515
  from lightning_sdk.lightning_cloud.openapi.models.v1_external_search_user import V1ExternalSearchUser
525
- from lightning_sdk.lightning_cloud.openapi.models.v1_file_endpoint import V1FileEndpoint
526
516
  from lightning_sdk.lightning_cloud.openapi.models.v1_filestore_data_connection import V1FilestoreDataConnection
527
517
  from lightning_sdk.lightning_cloud.openapi.models.v1_filesystem_app import V1FilesystemApp
528
518
  from lightning_sdk.lightning_cloud.openapi.models.v1_filesystem_cloud_space import V1FilesystemCloudSpace
529
519
  from lightning_sdk.lightning_cloud.openapi.models.v1_filesystem_dataset import V1FilesystemDataset
530
520
  from lightning_sdk.lightning_cloud.openapi.models.v1_filesystem_job import V1FilesystemJob
531
521
  from lightning_sdk.lightning_cloud.openapi.models.v1_filesystem_mmt import V1FilesystemMMT
522
+ from lightning_sdk.lightning_cloud.openapi.models.v1_filesystem_metrics import V1FilesystemMetrics
532
523
  from lightning_sdk.lightning_cloud.openapi.models.v1_filesystem_slurm_job import V1FilesystemSlurmJob
533
524
  from lightning_sdk.lightning_cloud.openapi.models.v1_filesystem_snowflake_connection import V1FilesystemSnowflakeConnection
534
525
  from lightning_sdk.lightning_cloud.openapi.models.v1_filesystem_work import V1FilesystemWork
@@ -582,7 +573,6 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_get_project_artifact_respon
582
573
  from lightning_sdk.lightning_cloud.openapi.models.v1_get_project_balance_response import V1GetProjectBalanceResponse
583
574
  from lightning_sdk.lightning_cloud.openapi.models.v1_get_project_compute_usage_response import V1GetProjectComputeUsageResponse
584
575
  from lightning_sdk.lightning_cloud.openapi.models.v1_get_project_storage_metadata_response import V1GetProjectStorageMetadataResponse
585
- from lightning_sdk.lightning_cloud.openapi.models.v1_get_service_execution_status_response import V1GetServiceExecutionStatusResponse
586
576
  from lightning_sdk.lightning_cloud.openapi.models.v1_get_settings_response import V1GetSettingsResponse
587
577
  from lightning_sdk.lightning_cloud.openapi.models.v1_get_snowflake_query_response import V1GetSnowflakeQueryResponse
588
578
  from lightning_sdk.lightning_cloud.openapi.models.v1_get_usage_details_response import V1GetUsageDetailsResponse
@@ -666,6 +656,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_list_cluster_capacity_reser
666
656
  from lightning_sdk.lightning_cloud.openapi.models.v1_list_cluster_proxies_response import V1ListClusterProxiesResponse
667
657
  from lightning_sdk.lightning_cloud.openapi.models.v1_list_cluster_usage_restrictions_response import V1ListClusterUsageRestrictionsResponse
668
658
  from lightning_sdk.lightning_cloud.openapi.models.v1_list_clusters_response import V1ListClustersResponse
659
+ from lightning_sdk.lightning_cloud.openapi.models.v1_list_container_metrics_response import V1ListContainerMetricsResponse
669
660
  from lightning_sdk.lightning_cloud.openapi.models.v1_list_conversations_response import V1ListConversationsResponse
670
661
  from lightning_sdk.lightning_cloud.openapi.models.v1_list_data_connection_artifacts_response import V1ListDataConnectionArtifactsResponse
671
662
  from lightning_sdk.lightning_cloud.openapi.models.v1_list_data_connections_response import V1ListDataConnectionsResponse
@@ -680,7 +671,6 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_list_deployment_templates_r
680
671
  from lightning_sdk.lightning_cloud.openapi.models.v1_list_deployments_response import V1ListDeploymentsResponse
681
672
  from lightning_sdk.lightning_cloud.openapi.models.v1_list_endpoints_response import V1ListEndpointsResponse
682
673
  from lightning_sdk.lightning_cloud.openapi.models.v1_list_experiments_response import V1ListExperimentsResponse
683
- from lightning_sdk.lightning_cloud.openapi.models.v1_list_file_endpoints_response import V1ListFileEndpointsResponse
684
674
  from lightning_sdk.lightning_cloud.openapi.models.v1_list_filesystem_apps_response import V1ListFilesystemAppsResponse
685
675
  from lightning_sdk.lightning_cloud.openapi.models.v1_list_filesystem_cloud_spaces_response import V1ListFilesystemCloudSpacesResponse
686
676
  from lightning_sdk.lightning_cloud.openapi.models.v1_list_filesystem_datasets_response import V1ListFilesystemDatasetsResponse
@@ -713,6 +703,8 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_list_model_versions_respons
713
703
  from lightning_sdk.lightning_cloud.openapi.models.v1_list_models_response import V1ListModelsResponse
714
704
  from lightning_sdk.lightning_cloud.openapi.models.v1_list_multi_machine_job_events_response import V1ListMultiMachineJobEventsResponse
715
705
  from lightning_sdk.lightning_cloud.openapi.models.v1_list_multi_machine_jobs_response import V1ListMultiMachineJobsResponse
706
+ from lightning_sdk.lightning_cloud.openapi.models.v1_list_node_file_system_metrics_response import V1ListNodeFileSystemMetricsResponse
707
+ from lightning_sdk.lightning_cloud.openapi.models.v1_list_node_metrics_response import V1ListNodeMetricsResponse
716
708
  from lightning_sdk.lightning_cloud.openapi.models.v1_list_notification_dialogs_response import V1ListNotificationDialogsResponse
717
709
  from lightning_sdk.lightning_cloud.openapi.models.v1_list_org_members_response import V1ListOrgMembersResponse
718
710
  from lightning_sdk.lightning_cloud.openapi.models.v1_list_org_membership_role_binding_response import V1ListOrgMembershipRoleBindingResponse
@@ -721,6 +713,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_list_org_roles_response imp
721
713
  from lightning_sdk.lightning_cloud.openapi.models.v1_list_organization_cluster_encryption_keys_response import V1ListOrganizationClusterEncryptionKeysResponse
722
714
  from lightning_sdk.lightning_cloud.openapi.models.v1_list_organizations_response import V1ListOrganizationsResponse
723
715
  from lightning_sdk.lightning_cloud.openapi.models.v1_list_pipelines_response import V1ListPipelinesResponse
716
+ from lightning_sdk.lightning_cloud.openapi.models.v1_list_pod_metrics_response import V1ListPodMetricsResponse
724
717
  from lightning_sdk.lightning_cloud.openapi.models.v1_list_product_licenses_response import V1ListProductLicensesResponse
725
718
  from lightning_sdk.lightning_cloud.openapi.models.v1_list_profiler_captures_response import V1ListProfilerCapturesResponse
726
719
  from lightning_sdk.lightning_cloud.openapi.models.v1_list_project_artifacts_response import V1ListProjectArtifactsResponse
@@ -739,8 +732,6 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_list_slurm_cluster_users_re
739
732
  from lightning_sdk.lightning_cloud.openapi.models.v1_list_ssh_public_keys_response import V1ListSSHPublicKeysResponse
740
733
  from lightning_sdk.lightning_cloud.openapi.models.v1_list_schedules_response import V1ListSchedulesResponse
741
734
  from lightning_sdk.lightning_cloud.openapi.models.v1_list_secrets_response import V1ListSecretsResponse
742
- from lightning_sdk.lightning_cloud.openapi.models.v1_list_service_execution_lightningapp_instances_response import V1ListServiceExecutionLightningappInstancesResponse
743
- from lightning_sdk.lightning_cloud.openapi.models.v1_list_service_executions_response import V1ListServiceExecutionsResponse
744
735
  from lightning_sdk.lightning_cloud.openapi.models.v1_list_studio_jobs_response import V1ListStudioJobsResponse
745
736
  from lightning_sdk.lightning_cloud.openapi.models.v1_list_user_slurm_jobs_response import V1ListUserSLURMJobsResponse
746
737
  from lightning_sdk.lightning_cloud.openapi.models.v1_lit_dataset import V1LitDataset
@@ -793,6 +784,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_named_get_logger_metrics im
793
784
  from lightning_sdk.lightning_cloud.openapi.models.v1_nebius_direct_v1 import V1NebiusDirectV1
794
785
  from lightning_sdk.lightning_cloud.openapi.models.v1_network_config import V1NetworkConfig
795
786
  from lightning_sdk.lightning_cloud.openapi.models.v1_new_feature import V1NewFeature
787
+ from lightning_sdk.lightning_cloud.openapi.models.v1_node_metrics import V1NodeMetrics
796
788
  from lightning_sdk.lightning_cloud.openapi.models.v1_notification_preference import V1NotificationPreference
797
789
  from lightning_sdk.lightning_cloud.openapi.models.v1_notification_preferences_request import V1NotificationPreferencesRequest
798
790
  from lightning_sdk.lightning_cloud.openapi.models.v1_notification_type import V1NotificationType
@@ -822,6 +814,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_pipeline_template import V1
822
814
  from lightning_sdk.lightning_cloud.openapi.models.v1_pipeline_template_visibility_type import V1PipelineTemplateVisibilityType
823
815
  from lightning_sdk.lightning_cloud.openapi.models.v1_plugin import V1Plugin
824
816
  from lightning_sdk.lightning_cloud.openapi.models.v1_plugins_list_response import V1PluginsListResponse
817
+ from lightning_sdk.lightning_cloud.openapi.models.v1_pod_metrics import V1PodMetrics
825
818
  from lightning_sdk.lightning_cloud.openapi.models.v1_post_cloud_space_artifact_events_response import V1PostCloudSpaceArtifactEventsResponse
826
819
  from lightning_sdk.lightning_cloud.openapi.models.v1_presigned_url import V1PresignedUrl
827
820
  from lightning_sdk.lightning_cloud.openapi.models.v1_product_license import V1ProductLicense
@@ -860,8 +853,10 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_region_state import V1Regio
860
853
  from lightning_sdk.lightning_cloud.openapi.models.v1_regional_load_balancer import V1RegionalLoadBalancer
861
854
  from lightning_sdk.lightning_cloud.openapi.models.v1_report_cloud_space_instance_stop_at_response import V1ReportCloudSpaceInstanceStopAtResponse
862
855
  from lightning_sdk.lightning_cloud.openapi.models.v1_report_cloud_space_instance_system_metrics_response import V1ReportCloudSpaceInstanceSystemMetricsResponse
856
+ from lightning_sdk.lightning_cloud.openapi.models.v1_report_k8s_cluster_metrics_response import V1ReportK8sClusterMetricsResponse
863
857
  from lightning_sdk.lightning_cloud.openapi.models.v1_report_logs_activity_response import V1ReportLogsActivityResponse
864
858
  from lightning_sdk.lightning_cloud.openapi.models.v1_report_restart_timings_response import V1ReportRestartTimingsResponse
859
+ from lightning_sdk.lightning_cloud.openapi.models.v1_request_cloud_space_access_response import V1RequestCloudSpaceAccessResponse
865
860
  from lightning_sdk.lightning_cloud.openapi.models.v1_request_cluster_access_request import V1RequestClusterAccessRequest
866
861
  from lightning_sdk.lightning_cloud.openapi.models.v1_request_cluster_access_response import V1RequestClusterAccessResponse
867
862
  from lightning_sdk.lightning_cloud.openapi.models.v1_request_verification_code_response import V1RequestVerificationCodeResponse
@@ -900,8 +895,6 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_server_alert_phase import V
900
895
  from lightning_sdk.lightning_cloud.openapi.models.v1_server_alert_severity import V1ServerAlertSeverity
901
896
  from lightning_sdk.lightning_cloud.openapi.models.v1_server_alert_type import V1ServerAlertType
902
897
  from lightning_sdk.lightning_cloud.openapi.models.v1_server_check_in_response import V1ServerCheckInResponse
903
- from lightning_sdk.lightning_cloud.openapi.models.v1_service_artifact import V1ServiceArtifact
904
- from lightning_sdk.lightning_cloud.openapi.models.v1_service_execution import V1ServiceExecution
905
898
  from lightning_sdk.lightning_cloud.openapi.models.v1_service_health import V1ServiceHealth
906
899
  from lightning_sdk.lightning_cloud.openapi.models.v1_setup_confirmed_ssh_public_key_response import V1SetupConfirmedSSHPublicKeyResponse
907
900
  from lightning_sdk.lightning_cloud.openapi.models.v1_setup_data_connection_response import V1SetupDataConnectionResponse
@@ -947,6 +940,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_update_agent_status_request
947
940
  from lightning_sdk.lightning_cloud.openapi.models.v1_update_agent_status_response import V1UpdateAgentStatusResponse
948
941
  from lightning_sdk.lightning_cloud.openapi.models.v1_update_billing_subscription_request import V1UpdateBillingSubscriptionRequest
949
942
  from lightning_sdk.lightning_cloud.openapi.models.v1_update_cloud_space_collab_response import V1UpdateCloudSpaceCollabResponse
943
+ from lightning_sdk.lightning_cloud.openapi.models.v1_update_cloud_space_instance_config_request import V1UpdateCloudSpaceInstanceConfigRequest
950
944
  from lightning_sdk.lightning_cloud.openapi.models.v1_update_cloud_space_publication_response import V1UpdateCloudSpacePublicationResponse
951
945
  from lightning_sdk.lightning_cloud.openapi.models.v1_update_cloud_space_visibility_response import V1UpdateCloudSpaceVisibilityResponse
952
946
  from lightning_sdk.lightning_cloud.openapi.models.v1_update_cluster_accelerators_request import V1UpdateClusterAcceleratorsRequest
@@ -976,8 +970,6 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_upload_agent_job_output_res
976
970
  from lightning_sdk.lightning_cloud.openapi.models.v1_upload_lightningapp_instance_artifact_response import V1UploadLightningappInstanceArtifactResponse
977
971
  from lightning_sdk.lightning_cloud.openapi.models.v1_upload_project_artifact_parts_response import V1UploadProjectArtifactPartsResponse
978
972
  from lightning_sdk.lightning_cloud.openapi.models.v1_upload_project_artifact_response import V1UploadProjectArtifactResponse
979
- from lightning_sdk.lightning_cloud.openapi.models.v1_upload_service_execution_artifact_parts_response import V1UploadServiceExecutionArtifactPartsResponse
980
- from lightning_sdk.lightning_cloud.openapi.models.v1_upload_service_execution_artifact_response import V1UploadServiceExecutionArtifactResponse
981
973
  from lightning_sdk.lightning_cloud.openapi.models.v1_upload_settings_response import V1UploadSettingsResponse
982
974
  from lightning_sdk.lightning_cloud.openapi.models.v1_upload_temporary_artifact_request import V1UploadTemporaryArtifactRequest
983
975
  from lightning_sdk.lightning_cloud.openapi.models.v1_upstream_cloud_space import V1UpstreamCloudSpace
@@ -21,6 +21,7 @@ from lightning_sdk.lightning_cloud.openapi.api.experiments_service_api import Ex
21
21
  from lightning_sdk.lightning_cloud.openapi.api.file_system_service_api import FileSystemServiceApi
22
22
  from lightning_sdk.lightning_cloud.openapi.api.git_credentials_service_api import GitCredentialsServiceApi
23
23
  from lightning_sdk.lightning_cloud.openapi.api.jobs_service_api import JobsServiceApi
24
+ from lightning_sdk.lightning_cloud.openapi.api.k8_s_cluster_service_api import K8SClusterServiceApi
24
25
  from lightning_sdk.lightning_cloud.openapi.api.lightningapp_instance_service_api import LightningappInstanceServiceApi
25
26
  from lightning_sdk.lightning_cloud.openapi.api.lightningapp_v2_service_api import LightningappV2ServiceApi
26
27
  from lightning_sdk.lightning_cloud.openapi.api.lightningwork_service_api import LightningworkServiceApi
@@ -1579,6 +1579,7 @@ class AssistantsServiceApi(object):
1579
1579
  >>> result = thread.get()
1580
1580
 
1581
1581
  :param async_req bool
1582
+ :param str org_id:
1582
1583
  :return: V1ListPublishedManagedEndpointsResponse
1583
1584
  If the method is called asynchronously,
1584
1585
  returns the request thread.
@@ -1599,12 +1600,13 @@ class AssistantsServiceApi(object):
1599
1600
  >>> result = thread.get()
1600
1601
 
1601
1602
  :param async_req bool
1603
+ :param str org_id:
1602
1604
  :return: V1ListPublishedManagedEndpointsResponse
1603
1605
  If the method is called asynchronously,
1604
1606
  returns the request thread.
1605
1607
  """
1606
1608
 
1607
- all_params = [] # noqa: E501
1609
+ all_params = ['org_id'] # noqa: E501
1608
1610
  all_params.append('async_req')
1609
1611
  all_params.append('_return_http_data_only')
1610
1612
  all_params.append('_preload_content')
@@ -1625,6 +1627,8 @@ class AssistantsServiceApi(object):
1625
1627
  path_params = {}
1626
1628
 
1627
1629
  query_params = []
1630
+ if 'org_id' in params:
1631
+ query_params.append(('orgId', params['org_id'])) # noqa: E501
1628
1632
 
1629
1633
  header_params = {}
1630
1634