lightning-sdk 0.2.15__py3-none-any.whl → 0.2.17__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 (80) hide show
  1. lightning_sdk/__init__.py +1 -1
  2. lightning_sdk/api/base_studio_api.py +7 -1
  3. lightning_sdk/api/cluster_api.py +83 -1
  4. lightning_sdk/api/llm_api.py +27 -5
  5. lightning_sdk/api/studio_api.py +64 -0
  6. lightning_sdk/api/teamspace_api.py +127 -1
  7. lightning_sdk/api/utils.py +4 -0
  8. lightning_sdk/base_studio.py +14 -1
  9. lightning_sdk/cli/create.py +21 -1
  10. lightning_sdk/cli/deploy/__init__.py +0 -0
  11. lightning_sdk/cli/deploy/_auth.py +189 -0
  12. lightning_sdk/cli/deploy/devbox.py +157 -0
  13. lightning_sdk/cli/{serve.py → deploy/serve.py} +11 -322
  14. lightning_sdk/cli/download.py +44 -16
  15. lightning_sdk/cli/entrypoint.py +1 -1
  16. lightning_sdk/cli/open.py +21 -2
  17. lightning_sdk/cli/start.py +12 -3
  18. lightning_sdk/cli/upload.py +2 -4
  19. lightning_sdk/lightning_cloud/openapi/__init__.py +19 -0
  20. lightning_sdk/lightning_cloud/openapi/api/assistants_service_api.py +126 -1
  21. lightning_sdk/lightning_cloud/openapi/api/cloud_space_environment_template_service_api.py +97 -0
  22. lightning_sdk/lightning_cloud/openapi/api/cloud_space_service_api.py +105 -0
  23. lightning_sdk/lightning_cloud/openapi/api/cluster_service_api.py +105 -0
  24. lightning_sdk/lightning_cloud/openapi/api/jobs_service_api.py +752 -106
  25. lightning_sdk/lightning_cloud/openapi/api/storage_service_api.py +93 -0
  26. lightning_sdk/lightning_cloud/openapi/models/__init__.py +19 -0
  27. lightning_sdk/lightning_cloud/openapi/models/assistant_id_conversations_body.py +53 -1
  28. lightning_sdk/lightning_cloud/openapi/models/cloudspaces_id_body.py +53 -1
  29. lightning_sdk/lightning_cloud/openapi/models/create_deployment_request_defines_a_spec_for_the_job_that_allows_for_autoscaling_jobs.py +27 -1
  30. lightning_sdk/lightning_cloud/openapi/models/deployment_id_alertingpolicies_body.py +357 -0
  31. lightning_sdk/lightning_cloud/openapi/models/deployment_id_alertingpolicies_body1.py +331 -0
  32. lightning_sdk/lightning_cloud/openapi/models/deployments_id_body.py +79 -1
  33. lightning_sdk/lightning_cloud/openapi/models/models_id_body.py +123 -0
  34. lightning_sdk/lightning_cloud/openapi/models/orgs_id_body.py +105 -1
  35. lightning_sdk/lightning_cloud/openapi/models/project_id_cloudspaces_body.py +27 -1
  36. lightning_sdk/lightning_cloud/openapi/models/projects_id_body.py +29 -3
  37. lightning_sdk/lightning_cloud/openapi/models/v1_cloud_space.py +79 -1
  38. lightning_sdk/lightning_cloud/openapi/models/v1_cloud_space_environment_template.py +27 -1
  39. lightning_sdk/lightning_cloud/openapi/models/v1_cloud_space_source_type.py +103 -0
  40. lightning_sdk/lightning_cloud/openapi/models/v1_cluster_tagging_options.py +27 -1
  41. lightning_sdk/lightning_cloud/openapi/models/v1_create_deployment_request.py +27 -1
  42. lightning_sdk/lightning_cloud/openapi/models/v1_data_connection.py +27 -1
  43. lightning_sdk/lightning_cloud/openapi/models/v1_delete_deployment_alerting_policy_response.py +175 -0
  44. lightning_sdk/lightning_cloud/openapi/models/v1_deployment.py +79 -1
  45. lightning_sdk/lightning_cloud/openapi/models/v1_deployment_alerting_event.py +487 -0
  46. lightning_sdk/lightning_cloud/openapi/models/v1_deployment_alerting_policy.py +409 -0
  47. lightning_sdk/lightning_cloud/openapi/models/v1_deployment_alerting_policy_frequency.py +105 -0
  48. lightning_sdk/lightning_cloud/openapi/models/v1_deployment_alerting_policy_operation.py +105 -0
  49. lightning_sdk/lightning_cloud/openapi/models/v1_deployment_alerting_policy_severity.py +106 -0
  50. lightning_sdk/lightning_cloud/openapi/models/v1_deployment_alerting_policy_type.py +111 -0
  51. lightning_sdk/lightning_cloud/openapi/models/v1_deployment_alerting_recipients.py +175 -0
  52. lightning_sdk/lightning_cloud/openapi/models/v1_ge_list_deployment_routing_telemetry_response.py +27 -1
  53. lightning_sdk/lightning_cloud/openapi/models/v1_get_cloud_space_instance_open_ports_response.py +123 -0
  54. lightning_sdk/lightning_cloud/openapi/models/v1_get_deployment_routing_telemetry_content_response.py +123 -0
  55. lightning_sdk/lightning_cloud/openapi/models/v1_get_organization_storage_metadata_response.py +331 -0
  56. lightning_sdk/lightning_cloud/openapi/models/v1_get_user_response.py +1 -27
  57. lightning_sdk/lightning_cloud/openapi/models/v1_google_cloud_direct_v1.py +27 -1
  58. lightning_sdk/lightning_cloud/openapi/models/v1_list_deployment_alerting_events_response.py +123 -0
  59. lightning_sdk/lightning_cloud/openapi/models/v1_list_deployment_alerting_policies_response.py +175 -0
  60. lightning_sdk/lightning_cloud/openapi/models/v1_membership.py +27 -1
  61. lightning_sdk/lightning_cloud/openapi/models/v1_organization.py +105 -1
  62. lightning_sdk/lightning_cloud/openapi/models/v1_project.py +27 -1
  63. lightning_sdk/lightning_cloud/openapi/models/v1_project_membership.py +27 -1
  64. lightning_sdk/lightning_cloud/openapi/models/v1_project_settings.py +29 -3
  65. lightning_sdk/lightning_cloud/openapi/models/v1_project_storage.py +53 -1
  66. lightning_sdk/lightning_cloud/openapi/models/v1_routing_telemetry.py +253 -0
  67. lightning_sdk/lightning_cloud/openapi/models/v1_server_alert_type.py +1 -0
  68. lightning_sdk/lightning_cloud/openapi/models/v1_sleep_server_response.py +97 -0
  69. lightning_sdk/lightning_cloud/openapi/models/v1_update_user_request.py +1 -27
  70. lightning_sdk/lightning_cloud/openapi/models/v1_user_features.py +105 -53
  71. lightning_sdk/lightning_cloud/openapi/models/v1_user_requested_compute_config.py +27 -1
  72. lightning_sdk/llm/llm.py +54 -8
  73. lightning_sdk/studio.py +40 -1
  74. lightning_sdk/teamspace.py +68 -0
  75. {lightning_sdk-0.2.15.dist-info → lightning_sdk-0.2.17.dist-info}/METADATA +1 -1
  76. {lightning_sdk-0.2.15.dist-info → lightning_sdk-0.2.17.dist-info}/RECORD +80 -58
  77. {lightning_sdk-0.2.15.dist-info → lightning_sdk-0.2.17.dist-info}/LICENSE +0 -0
  78. {lightning_sdk-0.2.15.dist-info → lightning_sdk-0.2.17.dist-info}/WHEEL +0 -0
  79. {lightning_sdk-0.2.15.dist-info → lightning_sdk-0.2.17.dist-info}/entry_points.txt +0 -0
  80. {lightning_sdk-0.2.15.dist-info → lightning_sdk-0.2.17.dist-info}/top_level.txt +0 -0
@@ -346,6 +346,99 @@ class StorageServiceApi(object):
346
346
  _request_timeout=params.get('_request_timeout'),
347
347
  collection_formats=collection_formats)
348
348
 
349
+ def storage_service_get_org_storage_metadata(self, org_id: 'str', **kwargs) -> 'V1GetOrganizationStorageMetadataResponse': # noqa: E501
350
+ """storage_service_get_org_storage_metadata # noqa: E501
351
+
352
+ This method makes a synchronous HTTP request by default. To make an
353
+ asynchronous HTTP request, please pass async_req=True
354
+ >>> thread = api.storage_service_get_org_storage_metadata(org_id, async_req=True)
355
+ >>> result = thread.get()
356
+
357
+ :param async_req bool
358
+ :param str org_id: (required)
359
+ :return: V1GetOrganizationStorageMetadataResponse
360
+ If the method is called asynchronously,
361
+ returns the request thread.
362
+ """
363
+ kwargs['_return_http_data_only'] = True
364
+ if kwargs.get('async_req'):
365
+ return self.storage_service_get_org_storage_metadata_with_http_info(org_id, **kwargs) # noqa: E501
366
+ else:
367
+ (data) = self.storage_service_get_org_storage_metadata_with_http_info(org_id, **kwargs) # noqa: E501
368
+ return data
369
+
370
+ def storage_service_get_org_storage_metadata_with_http_info(self, org_id: 'str', **kwargs) -> 'V1GetOrganizationStorageMetadataResponse': # noqa: E501
371
+ """storage_service_get_org_storage_metadata # noqa: E501
372
+
373
+ This method makes a synchronous HTTP request by default. To make an
374
+ asynchronous HTTP request, please pass async_req=True
375
+ >>> thread = api.storage_service_get_org_storage_metadata_with_http_info(org_id, async_req=True)
376
+ >>> result = thread.get()
377
+
378
+ :param async_req bool
379
+ :param str org_id: (required)
380
+ :return: V1GetOrganizationStorageMetadataResponse
381
+ If the method is called asynchronously,
382
+ returns the request thread.
383
+ """
384
+
385
+ all_params = ['org_id'] # noqa: E501
386
+ all_params.append('async_req')
387
+ all_params.append('_return_http_data_only')
388
+ all_params.append('_preload_content')
389
+ all_params.append('_request_timeout')
390
+
391
+ params = locals()
392
+ for key, val in six.iteritems(params['kwargs']):
393
+ if key not in all_params:
394
+ raise TypeError(
395
+ "Got an unexpected keyword argument '%s'"
396
+ " to method storage_service_get_org_storage_metadata" % key
397
+ )
398
+ params[key] = val
399
+ del params['kwargs']
400
+ # verify the required parameter 'org_id' is set
401
+ if ('org_id' not in params or
402
+ params['org_id'] is None):
403
+ raise ValueError("Missing the required parameter `org_id` when calling `storage_service_get_org_storage_metadata`") # noqa: E501
404
+
405
+ collection_formats = {}
406
+
407
+ path_params = {}
408
+ if 'org_id' in params:
409
+ path_params['orgId'] = params['org_id'] # noqa: E501
410
+
411
+ query_params = []
412
+
413
+ header_params = {}
414
+
415
+ form_params = []
416
+ local_var_files = {}
417
+
418
+ body_params = None
419
+ # HTTP header `Accept`
420
+ header_params['Accept'] = self.api_client.select_header_accept(
421
+ ['application/json']) # noqa: E501
422
+
423
+ # Authentication setting
424
+ auth_settings = [] # noqa: E501
425
+
426
+ return self.api_client.call_api(
427
+ '/v1/orgs/{orgId}/storage/metadata', 'GET',
428
+ path_params,
429
+ query_params,
430
+ header_params,
431
+ body=body_params,
432
+ post_params=form_params,
433
+ files=local_var_files,
434
+ response_type='V1GetOrganizationStorageMetadataResponse', # noqa: E501
435
+ auth_settings=auth_settings,
436
+ async_req=params.get('async_req'),
437
+ _return_http_data_only=params.get('_return_http_data_only'),
438
+ _preload_content=params.get('_preload_content', True),
439
+ _request_timeout=params.get('_request_timeout'),
440
+ collection_formats=collection_formats)
441
+
349
442
  def storage_service_get_project_artifact(self, project_id: 'str', **kwargs) -> 'V1GetProjectArtifactResponse': # noqa: E501
350
443
  """storage_service_get_project_artifact # noqa: E501
351
444
 
@@ -59,6 +59,8 @@ from lightning_sdk.lightning_cloud.openapi.models.data_connection_mount_data_con
59
59
  from lightning_sdk.lightning_cloud.openapi.models.dataset_id_versions_body import DatasetIdVersionsBody
60
60
  from lightning_sdk.lightning_cloud.openapi.models.dataset_id_visibility_body import DatasetIdVisibilityBody
61
61
  from lightning_sdk.lightning_cloud.openapi.models.datasets_id_body import DatasetsIdBody
62
+ from lightning_sdk.lightning_cloud.openapi.models.deployment_id_alertingpolicies_body import DeploymentIdAlertingpoliciesBody
63
+ from lightning_sdk.lightning_cloud.openapi.models.deployment_id_alertingpolicies_body1 import DeploymentIdAlertingpoliciesBody1
62
64
  from lightning_sdk.lightning_cloud.openapi.models.deployments_id_body import DeploymentsIdBody
63
65
  from lightning_sdk.lightning_cloud.openapi.models.deploymenttemplates_id_body import DeploymenttemplatesIdBody
64
66
  from lightning_sdk.lightning_cloud.openapi.models.endpoints_id_body import EndpointsIdBody
@@ -116,6 +118,7 @@ from lightning_sdk.lightning_cloud.openapi.models.metricsstream_delete_body impo
116
118
  from lightning_sdk.lightning_cloud.openapi.models.metricsstream_id_body import MetricsstreamIdBody
117
119
  from lightning_sdk.lightning_cloud.openapi.models.model_id_versions_body import ModelIdVersionsBody
118
120
  from lightning_sdk.lightning_cloud.openapi.models.model_id_visibility_body import ModelIdVisibilityBody
121
+ from lightning_sdk.lightning_cloud.openapi.models.models_id_body import ModelsIdBody
119
122
  from lightning_sdk.lightning_cloud.openapi.models.models_model_id_body import ModelsModelIdBody
120
123
  from lightning_sdk.lightning_cloud.openapi.models.multimachinejobs_id_body import MultimachinejobsIdBody
121
124
  from lightning_sdk.lightning_cloud.openapi.models.multipartuploads_upload_id_body import MultipartuploadsUploadIdBody
@@ -251,6 +254,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_cloud_space_publication imp
251
254
  from lightning_sdk.lightning_cloud.openapi.models.v1_cloud_space_publication_type import V1CloudSpacePublicationType
252
255
  from lightning_sdk.lightning_cloud.openapi.models.v1_cloud_space_seed_file import V1CloudSpaceSeedFile
253
256
  from lightning_sdk.lightning_cloud.openapi.models.v1_cloud_space_session import V1CloudSpaceSession
257
+ from lightning_sdk.lightning_cloud.openapi.models.v1_cloud_space_source_type import V1CloudSpaceSourceType
254
258
  from lightning_sdk.lightning_cloud.openapi.models.v1_cloud_space_state import V1CloudSpaceState
255
259
  from lightning_sdk.lightning_cloud.openapi.models.v1_cloud_space_version import V1CloudSpaceVersion
256
260
  from lightning_sdk.lightning_cloud.openapi.models.v1_cloud_space_version_publication import V1CloudSpaceVersionPublication
@@ -349,6 +353,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_delete_cluster_usage_restri
349
353
  from lightning_sdk.lightning_cloud.openapi.models.v1_delete_conversation_response import V1DeleteConversationResponse
350
354
  from lightning_sdk.lightning_cloud.openapi.models.v1_delete_data_connection_response import V1DeleteDataConnectionResponse
351
355
  from lightning_sdk.lightning_cloud.openapi.models.v1_delete_dataset_response import V1DeleteDatasetResponse
356
+ from lightning_sdk.lightning_cloud.openapi.models.v1_delete_deployment_alerting_policy_response import V1DeleteDeploymentAlertingPolicyResponse
352
357
  from lightning_sdk.lightning_cloud.openapi.models.v1_delete_deployment_release_response import V1DeleteDeploymentReleaseResponse
353
358
  from lightning_sdk.lightning_cloud.openapi.models.v1_delete_deployment_response import V1DeleteDeploymentResponse
354
359
  from lightning_sdk.lightning_cloud.openapi.models.v1_delete_endpoint_response import V1DeleteEndpointResponse
@@ -394,6 +399,13 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_dependency_cache_state impo
394
399
  from lightning_sdk.lightning_cloud.openapi.models.v1_dependency_file_info import V1DependencyFileInfo
395
400
  from lightning_sdk.lightning_cloud.openapi.models.v1_deployment import V1Deployment
396
401
  from lightning_sdk.lightning_cloud.openapi.models.v1_deployment_api import V1DeploymentAPI
402
+ from lightning_sdk.lightning_cloud.openapi.models.v1_deployment_alerting_event import V1DeploymentAlertingEvent
403
+ from lightning_sdk.lightning_cloud.openapi.models.v1_deployment_alerting_policy import V1DeploymentAlertingPolicy
404
+ from lightning_sdk.lightning_cloud.openapi.models.v1_deployment_alerting_policy_frequency import V1DeploymentAlertingPolicyFrequency
405
+ from lightning_sdk.lightning_cloud.openapi.models.v1_deployment_alerting_policy_operation import V1DeploymentAlertingPolicyOperation
406
+ from lightning_sdk.lightning_cloud.openapi.models.v1_deployment_alerting_policy_severity import V1DeploymentAlertingPolicySeverity
407
+ from lightning_sdk.lightning_cloud.openapi.models.v1_deployment_alerting_policy_type import V1DeploymentAlertingPolicyType
408
+ from lightning_sdk.lightning_cloud.openapi.models.v1_deployment_alerting_recipients import V1DeploymentAlertingRecipients
397
409
  from lightning_sdk.lightning_cloud.openapi.models.v1_deployment_details import V1DeploymentDetails
398
410
  from lightning_sdk.lightning_cloud.openapi.models.v1_deployment_event import V1DeploymentEvent
399
411
  from lightning_sdk.lightning_cloud.openapi.models.v1_deployment_event_type import V1DeploymentEventType
@@ -463,12 +475,14 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_get_agent_job_env_response
463
475
  from lightning_sdk.lightning_cloud.openapi.models.v1_get_agent_job_logs_metadata_response import V1GetAgentJobLogsMetadataResponse
464
476
  from lightning_sdk.lightning_cloud.openapi.models.v1_get_artifacts_page_response import V1GetArtifactsPageResponse
465
477
  from lightning_sdk.lightning_cloud.openapi.models.v1_get_cloud_space_cold_start_metrics_stats_response import V1GetCloudSpaceColdStartMetricsStatsResponse
478
+ from lightning_sdk.lightning_cloud.openapi.models.v1_get_cloud_space_instance_open_ports_response import V1GetCloudSpaceInstanceOpenPortsResponse
466
479
  from lightning_sdk.lightning_cloud.openapi.models.v1_get_cloud_space_instance_status_response import V1GetCloudSpaceInstanceStatusResponse
467
480
  from lightning_sdk.lightning_cloud.openapi.models.v1_get_cloud_space_instance_system_metrics_aggregate_response import V1GetCloudSpaceInstanceSystemMetricsAggregateResponse
468
481
  from lightning_sdk.lightning_cloud.openapi.models.v1_get_cloud_space_size_response import V1GetCloudSpaceSizeResponse
469
482
  from lightning_sdk.lightning_cloud.openapi.models.v1_get_cluster_accelerator_demand_response import V1GetClusterAcceleratorDemandResponse
470
483
  from lightning_sdk.lightning_cloud.openapi.models.v1_get_cluster_credentials_response import V1GetClusterCredentialsResponse
471
484
  from lightning_sdk.lightning_cloud.openapi.models.v1_get_deployment_routing_telemetry_aggregated_response import V1GetDeploymentRoutingTelemetryAggregatedResponse
485
+ from lightning_sdk.lightning_cloud.openapi.models.v1_get_deployment_routing_telemetry_content_response import V1GetDeploymentRoutingTelemetryContentResponse
472
486
  from lightning_sdk.lightning_cloud.openapi.models.v1_get_deployment_routing_telemetry_response import V1GetDeploymentRoutingTelemetryResponse
473
487
  from lightning_sdk.lightning_cloud.openapi.models.v1_get_folder_index_response import V1GetFolderIndexResponse
474
488
  from lightning_sdk.lightning_cloud.openapi.models.v1_get_job_stats_response import V1GetJobStatsResponse
@@ -485,6 +499,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_get_model_file_upload_urls_
485
499
  from lightning_sdk.lightning_cloud.openapi.models.v1_get_model_file_url_response import V1GetModelFileUrlResponse
486
500
  from lightning_sdk.lightning_cloud.openapi.models.v1_get_model_files_response import V1GetModelFilesResponse
487
501
  from lightning_sdk.lightning_cloud.openapi.models.v1_get_model_files_url_response import V1GetModelFilesUrlResponse
502
+ from lightning_sdk.lightning_cloud.openapi.models.v1_get_organization_storage_metadata_response import V1GetOrganizationStorageMetadataResponse
488
503
  from lightning_sdk.lightning_cloud.openapi.models.v1_get_project_artifact_response import V1GetProjectArtifactResponse
489
504
  from lightning_sdk.lightning_cloud.openapi.models.v1_get_project_balance_response import V1GetProjectBalanceResponse
490
505
  from lightning_sdk.lightning_cloud.openapi.models.v1_get_project_compute_usage_response import V1GetProjectComputeUsageResponse
@@ -573,6 +588,8 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_list_data_connection_artifa
573
588
  from lightning_sdk.lightning_cloud.openapi.models.v1_list_data_connections_response import V1ListDataConnectionsResponse
574
589
  from lightning_sdk.lightning_cloud.openapi.models.v1_list_datasets_response import V1ListDatasetsResponse
575
590
  from lightning_sdk.lightning_cloud.openapi.models.v1_list_default_cluster_accelerators_response import V1ListDefaultClusterAcceleratorsResponse
591
+ from lightning_sdk.lightning_cloud.openapi.models.v1_list_deployment_alerting_events_response import V1ListDeploymentAlertingEventsResponse
592
+ from lightning_sdk.lightning_cloud.openapi.models.v1_list_deployment_alerting_policies_response import V1ListDeploymentAlertingPoliciesResponse
576
593
  from lightning_sdk.lightning_cloud.openapi.models.v1_list_deployment_events_response import V1ListDeploymentEventsResponse
577
594
  from lightning_sdk.lightning_cloud.openapi.models.v1_list_deployment_releases_response import V1ListDeploymentReleasesResponse
578
595
  from lightning_sdk.lightning_cloud.openapi.models.v1_list_deployment_tags_response import V1ListDeploymentTagsResponse
@@ -775,6 +792,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_restart_timing import V1Res
775
792
  from lightning_sdk.lightning_cloud.openapi.models.v1_restore_deployment_release_response import V1RestoreDeploymentReleaseResponse
776
793
  from lightning_sdk.lightning_cloud.openapi.models.v1_role import V1Role
777
794
  from lightning_sdk.lightning_cloud.openapi.models.v1_rolling_update_strategy import V1RollingUpdateStrategy
795
+ from lightning_sdk.lightning_cloud.openapi.models.v1_routing_telemetry import V1RoutingTelemetry
778
796
  from lightning_sdk.lightning_cloud.openapi.models.v1_rule import V1Rule
779
797
  from lightning_sdk.lightning_cloud.openapi.models.v1_rule_action import V1RuleAction
780
798
  from lightning_sdk.lightning_cloud.openapi.models.v1_rule_condition import V1RuleCondition
@@ -805,6 +823,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_setup_data_connection_respo
805
823
  from lightning_sdk.lightning_cloud.openapi.models.v1_shared_filesystem import V1SharedFilesystem
806
824
  from lightning_sdk.lightning_cloud.openapi.models.v1_should_start_syncing_response import V1ShouldStartSyncingResponse
807
825
  from lightning_sdk.lightning_cloud.openapi.models.v1_signed_url import V1SignedUrl
826
+ from lightning_sdk.lightning_cloud.openapi.models.v1_sleep_server_response import V1SleepServerResponse
808
827
  from lightning_sdk.lightning_cloud.openapi.models.v1_slurm_cluster_user import V1SlurmClusterUser
809
828
  from lightning_sdk.lightning_cloud.openapi.models.v1_slurm_node import V1SlurmNode
810
829
  from lightning_sdk.lightning_cloud.openapi.models.v1_slurm_v1 import V1SlurmV1
@@ -44,11 +44,13 @@ class AssistantIdConversationsBody(object):
44
44
  'auto_name': 'bool',
45
45
  'billing_project_id': 'str',
46
46
  'conversation_id': 'str',
47
+ 'internal_conversation': 'bool',
47
48
  'max_tokens': 'str',
48
49
  'message': 'V1Message',
49
50
  'metadata': 'dict(str, str)',
50
51
  'name': 'str',
51
52
  'parent_message_id': 'str',
53
+ 'store': 'bool',
52
54
  'stream': 'bool'
53
55
  }
54
56
 
@@ -56,24 +58,28 @@ class AssistantIdConversationsBody(object):
56
58
  'auto_name': 'autoName',
57
59
  'billing_project_id': 'billingProjectId',
58
60
  'conversation_id': 'conversationId',
61
+ 'internal_conversation': 'internalConversation',
59
62
  'max_tokens': 'maxTokens',
60
63
  'message': 'message',
61
64
  'metadata': 'metadata',
62
65
  'name': 'name',
63
66
  'parent_message_id': 'parentMessageId',
67
+ 'store': 'store',
64
68
  'stream': 'stream'
65
69
  }
66
70
 
67
- def __init__(self, auto_name: 'bool' =None, billing_project_id: 'str' =None, conversation_id: 'str' =None, max_tokens: 'str' =None, message: 'V1Message' =None, metadata: 'dict(str, str)' =None, name: 'str' =None, parent_message_id: 'str' =None, stream: 'bool' =None): # noqa: E501
71
+ def __init__(self, auto_name: 'bool' =None, billing_project_id: 'str' =None, conversation_id: 'str' =None, internal_conversation: 'bool' =None, max_tokens: 'str' =None, message: 'V1Message' =None, metadata: 'dict(str, str)' =None, name: 'str' =None, parent_message_id: 'str' =None, store: 'bool' =None, stream: 'bool' =None): # noqa: E501
68
72
  """AssistantIdConversationsBody - a model defined in Swagger""" # noqa: E501
69
73
  self._auto_name = None
70
74
  self._billing_project_id = None
71
75
  self._conversation_id = None
76
+ self._internal_conversation = None
72
77
  self._max_tokens = None
73
78
  self._message = None
74
79
  self._metadata = None
75
80
  self._name = None
76
81
  self._parent_message_id = None
82
+ self._store = None
77
83
  self._stream = None
78
84
  self.discriminator = None
79
85
  if auto_name is not None:
@@ -82,6 +88,8 @@ class AssistantIdConversationsBody(object):
82
88
  self.billing_project_id = billing_project_id
83
89
  if conversation_id is not None:
84
90
  self.conversation_id = conversation_id
91
+ if internal_conversation is not None:
92
+ self.internal_conversation = internal_conversation
85
93
  if max_tokens is not None:
86
94
  self.max_tokens = max_tokens
87
95
  if message is not None:
@@ -92,6 +100,8 @@ class AssistantIdConversationsBody(object):
92
100
  self.name = name
93
101
  if parent_message_id is not None:
94
102
  self.parent_message_id = parent_message_id
103
+ if store is not None:
104
+ self.store = store
95
105
  if stream is not None:
96
106
  self.stream = stream
97
107
 
@@ -158,6 +168,27 @@ class AssistantIdConversationsBody(object):
158
168
 
159
169
  self._conversation_id = conversation_id
160
170
 
171
+ @property
172
+ def internal_conversation(self) -> 'bool':
173
+ """Gets the internal_conversation of this AssistantIdConversationsBody. # noqa: E501
174
+
175
+
176
+ :return: The internal_conversation of this AssistantIdConversationsBody. # noqa: E501
177
+ :rtype: bool
178
+ """
179
+ return self._internal_conversation
180
+
181
+ @internal_conversation.setter
182
+ def internal_conversation(self, internal_conversation: 'bool'):
183
+ """Sets the internal_conversation of this AssistantIdConversationsBody.
184
+
185
+
186
+ :param internal_conversation: The internal_conversation of this AssistantIdConversationsBody. # noqa: E501
187
+ :type: bool
188
+ """
189
+
190
+ self._internal_conversation = internal_conversation
191
+
161
192
  @property
162
193
  def max_tokens(self) -> 'str':
163
194
  """Gets the max_tokens of this AssistantIdConversationsBody. # noqa: E501
@@ -263,6 +294,27 @@ class AssistantIdConversationsBody(object):
263
294
 
264
295
  self._parent_message_id = parent_message_id
265
296
 
297
+ @property
298
+ def store(self) -> 'bool':
299
+ """Gets the store of this AssistantIdConversationsBody. # noqa: E501
300
+
301
+
302
+ :return: The store of this AssistantIdConversationsBody. # noqa: E501
303
+ :rtype: bool
304
+ """
305
+ return self._store
306
+
307
+ @store.setter
308
+ def store(self, store: 'bool'):
309
+ """Sets the store of this AssistantIdConversationsBody.
310
+
311
+
312
+ :param store: The store of this AssistantIdConversationsBody. # noqa: E501
313
+ :type: bool
314
+ """
315
+
316
+ self._store = store
317
+
266
318
  @property
267
319
  def stream(self) -> 'bool':
268
320
  """Gets the stream of this AssistantIdConversationsBody. # noqa: E501
@@ -41,6 +41,7 @@ class CloudspacesIdBody(object):
41
41
  and the value is json key in definition.
42
42
  """
43
43
  swagger_types = {
44
+ 'auto_switch_machine': 'bool',
44
45
  'code_url': 'str',
45
46
  'data_connection_mounts': 'list[V1DataConnectionMount]',
46
47
  'description': 'str',
@@ -62,6 +63,7 @@ class CloudspacesIdBody(object):
62
63
  'paper_org_avatar_url': 'str',
63
64
  'paper_url': 'str',
64
65
  'publish_with_compute_name': 'str',
66
+ 'switch_to_default_machine_on_idle': 'bool',
65
67
  'tags': 'list[V1ResourceTag]',
66
68
  'thumbnail': 'str',
67
69
  'thumbnail_file_type': 'str',
@@ -69,6 +71,7 @@ class CloudspacesIdBody(object):
69
71
  }
70
72
 
71
73
  attribute_map = {
74
+ 'auto_switch_machine': 'autoSwitchMachine',
72
75
  'code_url': 'codeUrl',
73
76
  'data_connection_mounts': 'dataConnectionMounts',
74
77
  'description': 'description',
@@ -90,14 +93,16 @@ class CloudspacesIdBody(object):
90
93
  'paper_org_avatar_url': 'paperOrgAvatarUrl',
91
94
  'paper_url': 'paperUrl',
92
95
  'publish_with_compute_name': 'publishWithComputeName',
96
+ 'switch_to_default_machine_on_idle': 'switchToDefaultMachineOnIdle',
93
97
  'tags': 'tags',
94
98
  'thumbnail': 'thumbnail',
95
99
  'thumbnail_file_type': 'thumbnailFileType',
96
100
  'user_metadata': 'userMetadata'
97
101
  }
98
102
 
99
- def __init__(self, code_url: 'str' =None, data_connection_mounts: 'list[V1DataConnectionMount]' =None, description: 'str' =None, display_name: 'str' =None, env: 'list[V1EnvVar]' =None, featured: 'bool' =None, hide_files: 'bool' =None, is_cloudspace_private: 'bool' =None, is_code_private: 'bool' =None, is_favorite: 'bool' =None, is_published: 'bool' =None, license: 'str' =None, license_url: 'str' =None, message: 'str' =None, multi_user_edit: 'bool' =None, operating_cost: 'str' =None, paper_authors: 'str' =None, paper_org: 'str' =None, paper_org_avatar_url: 'str' =None, paper_url: 'str' =None, publish_with_compute_name: 'str' =None, tags: 'list[V1ResourceTag]' =None, thumbnail: 'str' =None, thumbnail_file_type: 'str' =None, user_metadata: 'str' =None): # noqa: E501
103
+ def __init__(self, auto_switch_machine: 'bool' =None, code_url: 'str' =None, data_connection_mounts: 'list[V1DataConnectionMount]' =None, description: 'str' =None, display_name: 'str' =None, env: 'list[V1EnvVar]' =None, featured: 'bool' =None, hide_files: 'bool' =None, is_cloudspace_private: 'bool' =None, is_code_private: 'bool' =None, is_favorite: 'bool' =None, is_published: 'bool' =None, license: 'str' =None, license_url: 'str' =None, message: 'str' =None, multi_user_edit: 'bool' =None, operating_cost: 'str' =None, paper_authors: 'str' =None, paper_org: 'str' =None, paper_org_avatar_url: 'str' =None, paper_url: 'str' =None, publish_with_compute_name: 'str' =None, switch_to_default_machine_on_idle: 'bool' =None, tags: 'list[V1ResourceTag]' =None, thumbnail: 'str' =None, thumbnail_file_type: 'str' =None, user_metadata: 'str' =None): # noqa: E501
100
104
  """CloudspacesIdBody - a model defined in Swagger""" # noqa: E501
105
+ self._auto_switch_machine = None
101
106
  self._code_url = None
102
107
  self._data_connection_mounts = None
103
108
  self._description = None
@@ -119,11 +124,14 @@ class CloudspacesIdBody(object):
119
124
  self._paper_org_avatar_url = None
120
125
  self._paper_url = None
121
126
  self._publish_with_compute_name = None
127
+ self._switch_to_default_machine_on_idle = None
122
128
  self._tags = None
123
129
  self._thumbnail = None
124
130
  self._thumbnail_file_type = None
125
131
  self._user_metadata = None
126
132
  self.discriminator = None
133
+ if auto_switch_machine is not None:
134
+ self.auto_switch_machine = auto_switch_machine
127
135
  if code_url is not None:
128
136
  self.code_url = code_url
129
137
  if data_connection_mounts is not None:
@@ -166,6 +174,8 @@ class CloudspacesIdBody(object):
166
174
  self.paper_url = paper_url
167
175
  if publish_with_compute_name is not None:
168
176
  self.publish_with_compute_name = publish_with_compute_name
177
+ if switch_to_default_machine_on_idle is not None:
178
+ self.switch_to_default_machine_on_idle = switch_to_default_machine_on_idle
169
179
  if tags is not None:
170
180
  self.tags = tags
171
181
  if thumbnail is not None:
@@ -175,6 +185,27 @@ class CloudspacesIdBody(object):
175
185
  if user_metadata is not None:
176
186
  self.user_metadata = user_metadata
177
187
 
188
+ @property
189
+ def auto_switch_machine(self) -> 'bool':
190
+ """Gets the auto_switch_machine of this CloudspacesIdBody. # noqa: E501
191
+
192
+
193
+ :return: The auto_switch_machine of this CloudspacesIdBody. # noqa: E501
194
+ :rtype: bool
195
+ """
196
+ return self._auto_switch_machine
197
+
198
+ @auto_switch_machine.setter
199
+ def auto_switch_machine(self, auto_switch_machine: 'bool'):
200
+ """Sets the auto_switch_machine of this CloudspacesIdBody.
201
+
202
+
203
+ :param auto_switch_machine: The auto_switch_machine of this CloudspacesIdBody. # noqa: E501
204
+ :type: bool
205
+ """
206
+
207
+ self._auto_switch_machine = auto_switch_machine
208
+
178
209
  @property
179
210
  def code_url(self) -> 'str':
180
211
  """Gets the code_url of this CloudspacesIdBody. # noqa: E501
@@ -616,6 +647,27 @@ class CloudspacesIdBody(object):
616
647
 
617
648
  self._publish_with_compute_name = publish_with_compute_name
618
649
 
650
+ @property
651
+ def switch_to_default_machine_on_idle(self) -> 'bool':
652
+ """Gets the switch_to_default_machine_on_idle of this CloudspacesIdBody. # noqa: E501
653
+
654
+
655
+ :return: The switch_to_default_machine_on_idle of this CloudspacesIdBody. # noqa: E501
656
+ :rtype: bool
657
+ """
658
+ return self._switch_to_default_machine_on_idle
659
+
660
+ @switch_to_default_machine_on_idle.setter
661
+ def switch_to_default_machine_on_idle(self, switch_to_default_machine_on_idle: 'bool'):
662
+ """Sets the switch_to_default_machine_on_idle of this CloudspacesIdBody.
663
+
664
+
665
+ :param switch_to_default_machine_on_idle: The switch_to_default_machine_on_idle of this CloudspacesIdBody. # noqa: E501
666
+ :type: bool
667
+ """
668
+
669
+ self._switch_to_default_machine_on_idle = switch_to_default_machine_on_idle
670
+
619
671
  @property
620
672
  def tags(self) -> 'list[V1ResourceTag]':
621
673
  """Gets the tags of this CloudspacesIdBody. # noqa: E501
@@ -52,6 +52,7 @@ class CreateDeploymentRequestDefinesASpecForTheJobThatAllowsForAutoscalingJobs(o
52
52
  'name': 'str',
53
53
  'parameter_spec': 'V1ParameterizationSpec',
54
54
  'parent_template_id': 'str',
55
+ 'recipients': 'V1DeploymentAlertingRecipients',
55
56
  'replicas': 'int',
56
57
  'spec': 'V1JobSpec',
57
58
  'strategy': 'V1DeploymentStrategy'
@@ -69,12 +70,13 @@ class CreateDeploymentRequestDefinesASpecForTheJobThatAllowsForAutoscalingJobs(o
69
70
  'name': 'name',
70
71
  'parameter_spec': 'parameterSpec',
71
72
  'parent_template_id': 'parentTemplateId',
73
+ 'recipients': 'recipients',
72
74
  'replicas': 'replicas',
73
75
  'spec': 'spec',
74
76
  'strategy': 'strategy'
75
77
  }
76
78
 
77
- def __init__(self, api_standard: 'str' =None, apis: 'list[V1DeploymentAPI]' =None, autoscaling: 'V1AutoscalingSpec' =None, cloudspace_id: 'str' =None, cluster_id: 'str' =None, endpoint: 'V1Endpoint' =None, from_litserve: 'bool' =None, from_onboarding: 'bool' =None, name: 'str' =None, parameter_spec: 'V1ParameterizationSpec' =None, parent_template_id: 'str' =None, replicas: 'int' =None, spec: 'V1JobSpec' =None, strategy: 'V1DeploymentStrategy' =None): # noqa: E501
79
+ def __init__(self, api_standard: 'str' =None, apis: 'list[V1DeploymentAPI]' =None, autoscaling: 'V1AutoscalingSpec' =None, cloudspace_id: 'str' =None, cluster_id: 'str' =None, endpoint: 'V1Endpoint' =None, from_litserve: 'bool' =None, from_onboarding: 'bool' =None, name: 'str' =None, parameter_spec: 'V1ParameterizationSpec' =None, parent_template_id: 'str' =None, recipients: 'V1DeploymentAlertingRecipients' =None, replicas: 'int' =None, spec: 'V1JobSpec' =None, strategy: 'V1DeploymentStrategy' =None): # noqa: E501
78
80
  """CreateDeploymentRequestDefinesASpecForTheJobThatAllowsForAutoscalingJobs - a model defined in Swagger""" # noqa: E501
79
81
  self._api_standard = None
80
82
  self._apis = None
@@ -87,6 +89,7 @@ class CreateDeploymentRequestDefinesASpecForTheJobThatAllowsForAutoscalingJobs(o
87
89
  self._name = None
88
90
  self._parameter_spec = None
89
91
  self._parent_template_id = None
92
+ self._recipients = None
90
93
  self._replicas = None
91
94
  self._spec = None
92
95
  self._strategy = None
@@ -113,6 +116,8 @@ class CreateDeploymentRequestDefinesASpecForTheJobThatAllowsForAutoscalingJobs(o
113
116
  self.parameter_spec = parameter_spec
114
117
  if parent_template_id is not None:
115
118
  self.parent_template_id = parent_template_id
119
+ if recipients is not None:
120
+ self.recipients = recipients
116
121
  if replicas is not None:
117
122
  self.replicas = replicas
118
123
  if spec is not None:
@@ -351,6 +356,27 @@ class CreateDeploymentRequestDefinesASpecForTheJobThatAllowsForAutoscalingJobs(o
351
356
 
352
357
  self._parent_template_id = parent_template_id
353
358
 
359
+ @property
360
+ def recipients(self) -> 'V1DeploymentAlertingRecipients':
361
+ """Gets the recipients of this CreateDeploymentRequestDefinesASpecForTheJobThatAllowsForAutoscalingJobs. # noqa: E501
362
+
363
+
364
+ :return: The recipients of this CreateDeploymentRequestDefinesASpecForTheJobThatAllowsForAutoscalingJobs. # noqa: E501
365
+ :rtype: V1DeploymentAlertingRecipients
366
+ """
367
+ return self._recipients
368
+
369
+ @recipients.setter
370
+ def recipients(self, recipients: 'V1DeploymentAlertingRecipients'):
371
+ """Sets the recipients of this CreateDeploymentRequestDefinesASpecForTheJobThatAllowsForAutoscalingJobs.
372
+
373
+
374
+ :param recipients: The recipients of this CreateDeploymentRequestDefinesASpecForTheJobThatAllowsForAutoscalingJobs. # noqa: E501
375
+ :type: V1DeploymentAlertingRecipients
376
+ """
377
+
378
+ self._recipients = recipients
379
+
354
380
  @property
355
381
  def replicas(self) -> 'int':
356
382
  """Gets the replicas of this CreateDeploymentRequestDefinesASpecForTheJobThatAllowsForAutoscalingJobs. # noqa: E501