lightning-sdk 0.1.50__py3-none-any.whl → 0.1.53__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 (69) hide show
  1. lightning_sdk/__init__.py +1 -1
  2. lightning_sdk/ai_hub.py +16 -27
  3. lightning_sdk/api/ai_hub_api.py +7 -1
  4. lightning_sdk/api/job_api.py +12 -7
  5. lightning_sdk/api/lit_container_api.py +24 -7
  6. lightning_sdk/api/mmt_api.py +12 -7
  7. lightning_sdk/api/utils.py +52 -0
  8. lightning_sdk/cli/run.py +65 -18
  9. lightning_sdk/cli/serve.py +1 -5
  10. lightning_sdk/cli/upload.py +33 -15
  11. lightning_sdk/helpers.py +1 -1
  12. lightning_sdk/job/base.py +28 -1
  13. lightning_sdk/job/job.py +27 -25
  14. lightning_sdk/job/v1.py +6 -2
  15. lightning_sdk/job/v2.py +12 -12
  16. lightning_sdk/lightning_cloud/login.py +4 -1
  17. lightning_sdk/lightning_cloud/openapi/__init__.py +17 -0
  18. lightning_sdk/lightning_cloud/openapi/api/assistants_service_api.py +105 -0
  19. lightning_sdk/lightning_cloud/openapi/api/cluster_service_api.py +417 -1
  20. lightning_sdk/lightning_cloud/openapi/api/file_system_service_api.py +105 -0
  21. lightning_sdk/lightning_cloud/openapi/api/jobs_service_api.py +5 -1
  22. lightning_sdk/lightning_cloud/openapi/api/lit_registry_service_api.py +113 -0
  23. lightning_sdk/lightning_cloud/openapi/api/storage_service_api.py +101 -0
  24. lightning_sdk/lightning_cloud/openapi/api/user_service_api.py +5 -1
  25. lightning_sdk/lightning_cloud/openapi/models/__init__.py +17 -0
  26. lightning_sdk/lightning_cloud/openapi/models/cluster_id_usagerestrictions_body.py +175 -0
  27. lightning_sdk/lightning_cloud/openapi/models/deployments_id_body.py +27 -1
  28. lightning_sdk/lightning_cloud/openapi/models/id_contactowner_body.py +149 -0
  29. lightning_sdk/lightning_cloud/openapi/models/litregistry_lit_repo_name_body.py +123 -0
  30. lightning_sdk/lightning_cloud/openapi/models/metricsstream_create_body.py +27 -1
  31. lightning_sdk/lightning_cloud/openapi/models/usagerestrictions_id_body.py +175 -0
  32. lightning_sdk/lightning_cloud/openapi/models/v1_assistant_model_status.py +4 -0
  33. lightning_sdk/lightning_cloud/openapi/models/v1_cloud_provider.py +104 -0
  34. lightning_sdk/lightning_cloud/openapi/models/v1_cloud_space_artifact_event.py +149 -0
  35. lightning_sdk/lightning_cloud/openapi/models/v1_cloud_space_artifact_event_type.py +103 -0
  36. lightning_sdk/lightning_cloud/openapi/models/v1_cluster_accelerator.py +81 -3
  37. lightning_sdk/lightning_cloud/openapi/models/v1_cluster_spec.py +27 -1
  38. lightning_sdk/lightning_cloud/openapi/models/v1_cluster_tagging_options.py +29 -3
  39. lightning_sdk/lightning_cloud/openapi/models/v1_cluster_usage_restriction.py +227 -0
  40. lightning_sdk/lightning_cloud/openapi/models/v1_contact_assistant_owner_reason.py +102 -0
  41. lightning_sdk/lightning_cloud/openapi/models/v1_contact_assistant_owner_response.py +97 -0
  42. lightning_sdk/lightning_cloud/openapi/models/v1_delete_cluster_usage_restriction_response.py +97 -0
  43. lightning_sdk/lightning_cloud/openapi/models/v1_deployment.py +27 -1
  44. lightning_sdk/lightning_cloud/openapi/models/v1_deployment_api.py +53 -1
  45. lightning_sdk/lightning_cloud/openapi/models/v1_filesystem_mmt.py +175 -0
  46. lightning_sdk/lightning_cloud/openapi/models/v1_job_spec.py +27 -1
  47. lightning_sdk/lightning_cloud/openapi/models/v1_list_cluster_usage_restrictions_response.py +123 -0
  48. lightning_sdk/lightning_cloud/openapi/models/v1_list_filesystem_mm_ts_response.py +123 -0
  49. lightning_sdk/lightning_cloud/openapi/models/v1_metrics_stream.py +27 -1
  50. lightning_sdk/lightning_cloud/openapi/models/v1_model.py +27 -1
  51. lightning_sdk/lightning_cloud/openapi/models/v1_path_mapping.py +175 -0
  52. lightning_sdk/lightning_cloud/openapi/models/v1_post_cloud_space_artifact_events_response.py +97 -0
  53. lightning_sdk/lightning_cloud/openapi/models/v1_resource_visibility.py +27 -1
  54. lightning_sdk/lightning_cloud/openapi/models/v1_update_lit_repository_response.py +97 -0
  55. lightning_sdk/lightning_cloud/openapi/models/v1_user_features.py +128 -76
  56. lightning_sdk/lightning_cloud/utils/data_connection.py +75 -7
  57. lightning_sdk/mmt/base.py +36 -26
  58. lightning_sdk/mmt/mmt.py +28 -26
  59. lightning_sdk/mmt/v1.py +4 -1
  60. lightning_sdk/mmt/v2.py +14 -13
  61. lightning_sdk/models.py +5 -4
  62. lightning_sdk/studio.py +68 -1
  63. lightning_sdk/utils/resolve.py +7 -0
  64. {lightning_sdk-0.1.50.dist-info → lightning_sdk-0.1.53.dist-info}/METADATA +2 -2
  65. {lightning_sdk-0.1.50.dist-info → lightning_sdk-0.1.53.dist-info}/RECORD +69 -52
  66. {lightning_sdk-0.1.50.dist-info → lightning_sdk-0.1.53.dist-info}/LICENSE +0 -0
  67. {lightning_sdk-0.1.50.dist-info → lightning_sdk-0.1.53.dist-info}/WHEEL +0 -0
  68. {lightning_sdk-0.1.50.dist-info → lightning_sdk-0.1.53.dist-info}/entry_points.txt +0 -0
  69. {lightning_sdk-0.1.50.dist-info → lightning_sdk-0.1.53.dist-info}/top_level.txt +0 -0
@@ -442,3 +442,116 @@ class LitRegistryServiceApi(object):
442
442
  _preload_content=params.get('_preload_content', True),
443
443
  _request_timeout=params.get('_request_timeout'),
444
444
  collection_formats=collection_formats)
445
+
446
+ def lit_registry_service_update_lit_repository_metadata(self, body: 'LitregistryLitRepoNameBody', project_id: 'str', lit_repo_name: 'str', **kwargs) -> 'V1UpdateLitRepositoryResponse': # noqa: E501
447
+ """lit_registry_service_update_lit_repository_metadata # noqa: E501
448
+
449
+ This method makes a synchronous HTTP request by default. To make an
450
+ asynchronous HTTP request, please pass async_req=True
451
+ >>> thread = api.lit_registry_service_update_lit_repository_metadata(body, project_id, lit_repo_name, async_req=True)
452
+ >>> result = thread.get()
453
+
454
+ :param async_req bool
455
+ :param LitregistryLitRepoNameBody body: (required)
456
+ :param str project_id: (required)
457
+ :param str lit_repo_name: (required)
458
+ :return: V1UpdateLitRepositoryResponse
459
+ If the method is called asynchronously,
460
+ returns the request thread.
461
+ """
462
+ kwargs['_return_http_data_only'] = True
463
+ if kwargs.get('async_req'):
464
+ return self.lit_registry_service_update_lit_repository_metadata_with_http_info(body, project_id, lit_repo_name, **kwargs) # noqa: E501
465
+ else:
466
+ (data) = self.lit_registry_service_update_lit_repository_metadata_with_http_info(body, project_id, lit_repo_name, **kwargs) # noqa: E501
467
+ return data
468
+
469
+ def lit_registry_service_update_lit_repository_metadata_with_http_info(self, body: 'LitregistryLitRepoNameBody', project_id: 'str', lit_repo_name: 'str', **kwargs) -> 'V1UpdateLitRepositoryResponse': # noqa: E501
470
+ """lit_registry_service_update_lit_repository_metadata # noqa: E501
471
+
472
+ This method makes a synchronous HTTP request by default. To make an
473
+ asynchronous HTTP request, please pass async_req=True
474
+ >>> thread = api.lit_registry_service_update_lit_repository_metadata_with_http_info(body, project_id, lit_repo_name, async_req=True)
475
+ >>> result = thread.get()
476
+
477
+ :param async_req bool
478
+ :param LitregistryLitRepoNameBody body: (required)
479
+ :param str project_id: (required)
480
+ :param str lit_repo_name: (required)
481
+ :return: V1UpdateLitRepositoryResponse
482
+ If the method is called asynchronously,
483
+ returns the request thread.
484
+ """
485
+
486
+ all_params = ['body', 'project_id', 'lit_repo_name'] # noqa: E501
487
+ all_params.append('async_req')
488
+ all_params.append('_return_http_data_only')
489
+ all_params.append('_preload_content')
490
+ all_params.append('_request_timeout')
491
+
492
+ params = locals()
493
+ for key, val in six.iteritems(params['kwargs']):
494
+ if key not in all_params:
495
+ raise TypeError(
496
+ "Got an unexpected keyword argument '%s'"
497
+ " to method lit_registry_service_update_lit_repository_metadata" % key
498
+ )
499
+ params[key] = val
500
+ del params['kwargs']
501
+ # verify the required parameter 'body' is set
502
+ if ('body' not in params or
503
+ params['body'] is None):
504
+ raise ValueError("Missing the required parameter `body` when calling `lit_registry_service_update_lit_repository_metadata`") # noqa: E501
505
+ # verify the required parameter 'project_id' is set
506
+ if ('project_id' not in params or
507
+ params['project_id'] is None):
508
+ raise ValueError("Missing the required parameter `project_id` when calling `lit_registry_service_update_lit_repository_metadata`") # noqa: E501
509
+ # verify the required parameter 'lit_repo_name' is set
510
+ if ('lit_repo_name' not in params or
511
+ params['lit_repo_name'] is None):
512
+ raise ValueError("Missing the required parameter `lit_repo_name` when calling `lit_registry_service_update_lit_repository_metadata`") # noqa: E501
513
+
514
+ collection_formats = {}
515
+
516
+ path_params = {}
517
+ if 'project_id' in params:
518
+ path_params['projectId'] = params['project_id'] # noqa: E501
519
+ if 'lit_repo_name' in params:
520
+ path_params['litRepoName'] = params['lit_repo_name'] # noqa: E501
521
+
522
+ query_params = []
523
+
524
+ header_params = {}
525
+
526
+ form_params = []
527
+ local_var_files = {}
528
+
529
+ body_params = None
530
+ if 'body' in params:
531
+ body_params = params['body']
532
+ # HTTP header `Accept`
533
+ header_params['Accept'] = self.api_client.select_header_accept(
534
+ ['application/json']) # noqa: E501
535
+
536
+ # HTTP header `Content-Type`
537
+ header_params['Content-Type'] = self.api_client.select_header_content_type( # noqa: E501
538
+ ['application/json']) # noqa: E501
539
+
540
+ # Authentication setting
541
+ auth_settings = [] # noqa: E501
542
+
543
+ return self.api_client.call_api(
544
+ '/v1/projects/{projectId}/lit-registry/{litRepoName}', 'PUT',
545
+ path_params,
546
+ query_params,
547
+ header_params,
548
+ body=body_params,
549
+ post_params=form_params,
550
+ files=local_var_files,
551
+ response_type='V1UpdateLitRepositoryResponse', # noqa: E501
552
+ auth_settings=auth_settings,
553
+ async_req=params.get('async_req'),
554
+ _return_http_data_only=params.get('_return_http_data_only'),
555
+ _preload_content=params.get('_preload_content', True),
556
+ _request_timeout=params.get('_request_timeout'),
557
+ collection_formats=collection_formats)
@@ -875,6 +875,107 @@ class StorageServiceApi(object):
875
875
  _request_timeout=params.get('_request_timeout'),
876
876
  collection_formats=collection_formats)
877
877
 
878
+ def storage_service_post_cloud_space_artifact_events(self, project_id: 'str', cloud_space_id: 'str', **kwargs) -> 'V1PostCloudSpaceArtifactEventsResponse': # noqa: E501
879
+ """storage_service_post_cloud_space_artifact_events # noqa: E501
880
+
881
+ This method makes a synchronous HTTP request by default. To make an
882
+ asynchronous HTTP request, please pass async_req=True
883
+ >>> thread = api.storage_service_post_cloud_space_artifact_events(project_id, cloud_space_id, async_req=True)
884
+ >>> result = thread.get()
885
+
886
+ :param async_req bool
887
+ :param str project_id: (required)
888
+ :param str cloud_space_id: (required)
889
+ :return: V1PostCloudSpaceArtifactEventsResponse
890
+ If the method is called asynchronously,
891
+ returns the request thread.
892
+ """
893
+ kwargs['_return_http_data_only'] = True
894
+ if kwargs.get('async_req'):
895
+ return self.storage_service_post_cloud_space_artifact_events_with_http_info(project_id, cloud_space_id, **kwargs) # noqa: E501
896
+ else:
897
+ (data) = self.storage_service_post_cloud_space_artifact_events_with_http_info(project_id, cloud_space_id, **kwargs) # noqa: E501
898
+ return data
899
+
900
+ def storage_service_post_cloud_space_artifact_events_with_http_info(self, project_id: 'str', cloud_space_id: 'str', **kwargs) -> 'V1PostCloudSpaceArtifactEventsResponse': # noqa: E501
901
+ """storage_service_post_cloud_space_artifact_events # noqa: E501
902
+
903
+ This method makes a synchronous HTTP request by default. To make an
904
+ asynchronous HTTP request, please pass async_req=True
905
+ >>> thread = api.storage_service_post_cloud_space_artifact_events_with_http_info(project_id, cloud_space_id, async_req=True)
906
+ >>> result = thread.get()
907
+
908
+ :param async_req bool
909
+ :param str project_id: (required)
910
+ :param str cloud_space_id: (required)
911
+ :return: V1PostCloudSpaceArtifactEventsResponse
912
+ If the method is called asynchronously,
913
+ returns the request thread.
914
+ """
915
+
916
+ all_params = ['project_id', 'cloud_space_id'] # noqa: E501
917
+ all_params.append('async_req')
918
+ all_params.append('_return_http_data_only')
919
+ all_params.append('_preload_content')
920
+ all_params.append('_request_timeout')
921
+
922
+ params = locals()
923
+ for key, val in six.iteritems(params['kwargs']):
924
+ if key not in all_params:
925
+ raise TypeError(
926
+ "Got an unexpected keyword argument '%s'"
927
+ " to method storage_service_post_cloud_space_artifact_events" % key
928
+ )
929
+ params[key] = val
930
+ del params['kwargs']
931
+ # verify the required parameter 'project_id' is set
932
+ if ('project_id' not in params or
933
+ params['project_id'] is None):
934
+ raise ValueError("Missing the required parameter `project_id` when calling `storage_service_post_cloud_space_artifact_events`") # noqa: E501
935
+ # verify the required parameter 'cloud_space_id' is set
936
+ if ('cloud_space_id' not in params or
937
+ params['cloud_space_id'] is None):
938
+ raise ValueError("Missing the required parameter `cloud_space_id` when calling `storage_service_post_cloud_space_artifact_events`") # noqa: E501
939
+
940
+ collection_formats = {}
941
+
942
+ path_params = {}
943
+ if 'project_id' in params:
944
+ path_params['projectId'] = params['project_id'] # noqa: E501
945
+ if 'cloud_space_id' in params:
946
+ path_params['cloudSpaceId'] = params['cloud_space_id'] # noqa: E501
947
+
948
+ query_params = []
949
+
950
+ header_params = {}
951
+
952
+ form_params = []
953
+ local_var_files = {}
954
+
955
+ body_params = None
956
+ # HTTP header `Accept`
957
+ header_params['Accept'] = self.api_client.select_header_accept(
958
+ ['application/json']) # noqa: E501
959
+
960
+ # Authentication setting
961
+ auth_settings = [] # noqa: E501
962
+
963
+ return self.api_client.call_api(
964
+ '/v1/projects/{projectId}/cloudspaces/{cloudSpaceId}/artifacts/events', 'GET',
965
+ path_params,
966
+ query_params,
967
+ header_params,
968
+ body=body_params,
969
+ post_params=form_params,
970
+ files=local_var_files,
971
+ response_type='V1PostCloudSpaceArtifactEventsResponse', # noqa: E501
972
+ auth_settings=auth_settings,
973
+ async_req=params.get('async_req'),
974
+ _return_http_data_only=params.get('_return_http_data_only'),
975
+ _preload_content=params.get('_preload_content', True),
976
+ _request_timeout=params.get('_request_timeout'),
977
+ collection_formats=collection_formats)
978
+
878
979
  def storage_service_upload_project_artifact(self, body: 'ProjectIdStorageBody', project_id: 'str', **kwargs) -> 'V1UploadProjectArtifactResponse': # noqa: E501
879
980
  """UploadProjectArtifact - starts the multipart upload process and returns the upload ID. You will then need to use the upload ID with UploadProjectArtifactParts to get the upload URLs # noqa: E501
880
981
 
@@ -1167,6 +1167,7 @@ class UserServiceApi(object):
1167
1167
  :param str page_token:
1168
1168
  :param str page_size:
1169
1169
  :param str query:
1170
+ :param str organization_id:
1170
1171
  :return: V1SearchUsersResponse
1171
1172
  If the method is called asynchronously,
1172
1173
  returns the request thread.
@@ -1190,12 +1191,13 @@ class UserServiceApi(object):
1190
1191
  :param str page_token:
1191
1192
  :param str page_size:
1192
1193
  :param str query:
1194
+ :param str organization_id:
1193
1195
  :return: V1SearchUsersResponse
1194
1196
  If the method is called asynchronously,
1195
1197
  returns the request thread.
1196
1198
  """
1197
1199
 
1198
- all_params = ['page_token', 'page_size', 'query'] # noqa: E501
1200
+ all_params = ['page_token', 'page_size', 'query', 'organization_id'] # noqa: E501
1199
1201
  all_params.append('async_req')
1200
1202
  all_params.append('_return_http_data_only')
1201
1203
  all_params.append('_preload_content')
@@ -1222,6 +1224,8 @@ class UserServiceApi(object):
1222
1224
  query_params.append(('pageSize', params['page_size'])) # noqa: E501
1223
1225
  if 'query' in params:
1224
1226
  query_params.append(('query', params['query'])) # noqa: E501
1227
+ if 'organization_id' in params:
1228
+ query_params.append(('organizationId', params['organization_id'])) # noqa: E501
1225
1229
 
1226
1230
  header_params = {}
1227
1231
 
@@ -43,6 +43,7 @@ from lightning_sdk.lightning_cloud.openapi.models.cloudspaces_id_body import Clo
43
43
  from lightning_sdk.lightning_cloud.openapi.models.cluster_id_capacityblock_body import ClusterIdCapacityblockBody
44
44
  from lightning_sdk.lightning_cloud.openapi.models.cluster_id_capacityreservations_body import ClusterIdCapacityreservationsBody
45
45
  from lightning_sdk.lightning_cloud.openapi.models.cluster_id_proxies_body import ClusterIdProxiesBody
46
+ from lightning_sdk.lightning_cloud.openapi.models.cluster_id_usagerestrictions_body import ClusterIdUsagerestrictionsBody
46
47
  from lightning_sdk.lightning_cloud.openapi.models.clusters_id_body import ClustersIdBody
47
48
  from lightning_sdk.lightning_cloud.openapi.models.clusters_id_body1 import ClustersIdBody1
48
49
  from lightning_sdk.lightning_cloud.openapi.models.command_argument_command_argument_type import CommandArgumentCommandArgumentType
@@ -68,6 +69,7 @@ from lightning_sdk.lightning_cloud.openapi.models.id_artifacts_body import IdArt
68
69
  from lightning_sdk.lightning_cloud.openapi.models.id_codeconfig_body import IdCodeconfigBody
69
70
  from lightning_sdk.lightning_cloud.openapi.models.id_collaborate_body import IdCollaborateBody
70
71
  from lightning_sdk.lightning_cloud.openapi.models.id_complete_body import IdCompleteBody
72
+ from lightning_sdk.lightning_cloud.openapi.models.id_contactowner_body import IdContactownerBody
71
73
  from lightning_sdk.lightning_cloud.openapi.models.id_engage_body import IdEngageBody
72
74
  from lightning_sdk.lightning_cloud.openapi.models.id_engage_body1 import IdEngageBody1
73
75
  from lightning_sdk.lightning_cloud.openapi.models.id_execute_body import IdExecuteBody
@@ -96,6 +98,7 @@ from lightning_sdk.lightning_cloud.openapi.models.jobs_id_body2 import JobsIdBod
96
98
  from lightning_sdk.lightning_cloud.openapi.models.jobs_id_body3 import JobsIdBody3
97
99
  from lightning_sdk.lightning_cloud.openapi.models.litloggermetrics_id_body import LitloggermetricsIdBody
98
100
  from lightning_sdk.lightning_cloud.openapi.models.litpages_id_body import LitpagesIdBody
101
+ from lightning_sdk.lightning_cloud.openapi.models.litregistry_lit_repo_name_body import LitregistryLitRepoNameBody
99
102
  from lightning_sdk.lightning_cloud.openapi.models.loggermetrics_id_body import LoggermetricsIdBody
100
103
  from lightning_sdk.lightning_cloud.openapi.models.metrics_stream_id_loggerartifacts_body import MetricsStreamIdLoggerartifactsBody
101
104
  from lightning_sdk.lightning_cloud.openapi.models.metricsstream_create_body import MetricsstreamCreateBody
@@ -153,6 +156,7 @@ from lightning_sdk.lightning_cloud.openapi.models.update import Update
153
156
  from lightning_sdk.lightning_cloud.openapi.models.upload_id_complete_body import UploadIdCompleteBody
154
157
  from lightning_sdk.lightning_cloud.openapi.models.upload_id_parts_body import UploadIdPartsBody
155
158
  from lightning_sdk.lightning_cloud.openapi.models.uploads_upload_id_body import UploadsUploadIdBody
159
+ from lightning_sdk.lightning_cloud.openapi.models.usagerestrictions_id_body import UsagerestrictionsIdBody
156
160
  from lightning_sdk.lightning_cloud.openapi.models.user_id_affiliatelinks_body import UserIdAffiliatelinksBody
157
161
  from lightning_sdk.lightning_cloud.openapi.models.user_id_membershiprolebindings_body import UserIdMembershiprolebindingsBody
158
162
  from lightning_sdk.lightning_cloud.openapi.models.user_id_membershiprolebindings_body1 import UserIdMembershiprolebindingsBody1
@@ -197,11 +201,14 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_capacity_block_offering imp
197
201
  from lightning_sdk.lightning_cloud.openapi.models.v1_check_external_service_status_response import V1CheckExternalServiceStatusResponse
198
202
  from lightning_sdk.lightning_cloud.openapi.models.v1_check_snowflake_connection_response import V1CheckSnowflakeConnectionResponse
199
203
  from lightning_sdk.lightning_cloud.openapi.models.v1_checkbox import V1Checkbox
204
+ from lightning_sdk.lightning_cloud.openapi.models.v1_cloud_provider import V1CloudProvider
200
205
  from lightning_sdk.lightning_cloud.openapi.models.v1_cloud_space import V1CloudSpace
201
206
  from lightning_sdk.lightning_cloud.openapi.models.v1_cloud_space_app import V1CloudSpaceApp
202
207
  from lightning_sdk.lightning_cloud.openapi.models.v1_cloud_space_app_action import V1CloudSpaceAppAction
203
208
  from lightning_sdk.lightning_cloud.openapi.models.v1_cloud_space_app_action_type import V1CloudSpaceAppActionType
204
209
  from lightning_sdk.lightning_cloud.openapi.models.v1_cloud_space_app_owner_type import V1CloudSpaceAppOwnerType
210
+ from lightning_sdk.lightning_cloud.openapi.models.v1_cloud_space_artifact_event import V1CloudSpaceArtifactEvent
211
+ from lightning_sdk.lightning_cloud.openapi.models.v1_cloud_space_artifact_event_type import V1CloudSpaceArtifactEventType
205
212
  from lightning_sdk.lightning_cloud.openapi.models.v1_cloud_space_code_version import V1CloudSpaceCodeVersion
206
213
  from lightning_sdk.lightning_cloud.openapi.models.v1_cloud_space_code_version_status import V1CloudSpaceCodeVersionStatus
207
214
  from lightning_sdk.lightning_cloud.openapi.models.v1_cloud_space_engagement_response import V1CloudSpaceEngagementResponse
@@ -231,6 +238,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_cluster_state import V1Clus
231
238
  from lightning_sdk.lightning_cloud.openapi.models.v1_cluster_status import V1ClusterStatus
232
239
  from lightning_sdk.lightning_cloud.openapi.models.v1_cluster_tagging_options import V1ClusterTaggingOptions
233
240
  from lightning_sdk.lightning_cloud.openapi.models.v1_cluster_type import V1ClusterType
241
+ from lightning_sdk.lightning_cloud.openapi.models.v1_cluster_usage_restriction import V1ClusterUsageRestriction
234
242
  from lightning_sdk.lightning_cloud.openapi.models.v1_collab_action import V1CollabAction
235
243
  from lightning_sdk.lightning_cloud.openapi.models.v1_collab_session import V1CollabSession
236
244
  from lightning_sdk.lightning_cloud.openapi.models.v1_command_argument import V1CommandArgument
@@ -243,6 +251,8 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_complete_upload_project_art
243
251
  from lightning_sdk.lightning_cloud.openapi.models.v1_complete_upload_service_execution_artifact_response import V1CompleteUploadServiceExecutionArtifactResponse
244
252
  from lightning_sdk.lightning_cloud.openapi.models.v1_completed_part import V1CompletedPart
245
253
  from lightning_sdk.lightning_cloud.openapi.models.v1_compute_config import V1ComputeConfig
254
+ from lightning_sdk.lightning_cloud.openapi.models.v1_contact_assistant_owner_reason import V1ContactAssistantOwnerReason
255
+ from lightning_sdk.lightning_cloud.openapi.models.v1_contact_assistant_owner_response import V1ContactAssistantOwnerResponse
246
256
  from lightning_sdk.lightning_cloud.openapi.models.v1_conversation import V1Conversation
247
257
  from lightning_sdk.lightning_cloud.openapi.models.v1_conversation_response_chunk import V1ConversationResponseChunk
248
258
  from lightning_sdk.lightning_cloud.openapi.models.v1_count_metrics_streams_response import V1CountMetricsStreamsResponse
@@ -290,6 +300,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_delete_cluster_capacity_res
290
300
  from lightning_sdk.lightning_cloud.openapi.models.v1_delete_cluster_encryption_key_response import V1DeleteClusterEncryptionKeyResponse
291
301
  from lightning_sdk.lightning_cloud.openapi.models.v1_delete_cluster_proxy_response import V1DeleteClusterProxyResponse
292
302
  from lightning_sdk.lightning_cloud.openapi.models.v1_delete_cluster_response import V1DeleteClusterResponse
303
+ from lightning_sdk.lightning_cloud.openapi.models.v1_delete_cluster_usage_restriction_response import V1DeleteClusterUsageRestrictionResponse
293
304
  from lightning_sdk.lightning_cloud.openapi.models.v1_delete_conversation_response import V1DeleteConversationResponse
294
305
  from lightning_sdk.lightning_cloud.openapi.models.v1_delete_data_connection_response import V1DeleteDataConnectionResponse
295
306
  from lightning_sdk.lightning_cloud.openapi.models.v1_delete_dataset_response import V1DeleteDatasetResponse
@@ -379,6 +390,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_filesystem_app import V1Fil
379
390
  from lightning_sdk.lightning_cloud.openapi.models.v1_filesystem_cloud_space import V1FilesystemCloudSpace
380
391
  from lightning_sdk.lightning_cloud.openapi.models.v1_filesystem_dataset import V1FilesystemDataset
381
392
  from lightning_sdk.lightning_cloud.openapi.models.v1_filesystem_job import V1FilesystemJob
393
+ from lightning_sdk.lightning_cloud.openapi.models.v1_filesystem_mmt import V1FilesystemMMT
382
394
  from lightning_sdk.lightning_cloud.openapi.models.v1_filesystem_slurm_job import V1FilesystemSlurmJob
383
395
  from lightning_sdk.lightning_cloud.openapi.models.v1_filesystem_snowflake_connection import V1FilesystemSnowflakeConnection
384
396
  from lightning_sdk.lightning_cloud.openapi.models.v1_filesystem_work import V1FilesystemWork
@@ -488,6 +500,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_list_cluster_accelerators_r
488
500
  from lightning_sdk.lightning_cloud.openapi.models.v1_list_cluster_availabilities_response import V1ListClusterAvailabilitiesResponse
489
501
  from lightning_sdk.lightning_cloud.openapi.models.v1_list_cluster_capacity_reservations_response import V1ListClusterCapacityReservationsResponse
490
502
  from lightning_sdk.lightning_cloud.openapi.models.v1_list_cluster_proxies_response import V1ListClusterProxiesResponse
503
+ from lightning_sdk.lightning_cloud.openapi.models.v1_list_cluster_usage_restrictions_response import V1ListClusterUsageRestrictionsResponse
491
504
  from lightning_sdk.lightning_cloud.openapi.models.v1_list_clusters_response import V1ListClustersResponse
492
505
  from lightning_sdk.lightning_cloud.openapi.models.v1_list_conversations_response import V1ListConversationsResponse
493
506
  from lightning_sdk.lightning_cloud.openapi.models.v1_list_data_connection_artifacts_response import V1ListDataConnectionArtifactsResponse
@@ -506,6 +519,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_list_filesystem_apps_respon
506
519
  from lightning_sdk.lightning_cloud.openapi.models.v1_list_filesystem_cloud_spaces_response import V1ListFilesystemCloudSpacesResponse
507
520
  from lightning_sdk.lightning_cloud.openapi.models.v1_list_filesystem_datasets_response import V1ListFilesystemDatasetsResponse
508
521
  from lightning_sdk.lightning_cloud.openapi.models.v1_list_filesystem_jobs_response import V1ListFilesystemJobsResponse
522
+ from lightning_sdk.lightning_cloud.openapi.models.v1_list_filesystem_mm_ts_response import V1ListFilesystemMMTsResponse
509
523
  from lightning_sdk.lightning_cloud.openapi.models.v1_list_filesystem_slurm_jobs_response import V1ListFilesystemSlurmJobsResponse
510
524
  from lightning_sdk.lightning_cloud.openapi.models.v1_list_filesystem_snowflake_response import V1ListFilesystemSnowflakeResponse
511
525
  from lightning_sdk.lightning_cloud.openapi.models.v1_list_gallery_components_response import V1ListGalleryComponentsResponse
@@ -609,10 +623,12 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_organization import V1Organ
609
623
  from lightning_sdk.lightning_cloud.openapi.models.v1_owner_type import V1OwnerType
610
624
  from lightning_sdk.lightning_cloud.openapi.models.v1_package_manager import V1PackageManager
611
625
  from lightning_sdk.lightning_cloud.openapi.models.v1_parameterization_spec import V1ParameterizationSpec
626
+ from lightning_sdk.lightning_cloud.openapi.models.v1_path_mapping import V1PathMapping
612
627
  from lightning_sdk.lightning_cloud.openapi.models.v1_path_telemetry import V1PathTelemetry
613
628
  from lightning_sdk.lightning_cloud.openapi.models.v1_phase_type import V1PhaseType
614
629
  from lightning_sdk.lightning_cloud.openapi.models.v1_plugin import V1Plugin
615
630
  from lightning_sdk.lightning_cloud.openapi.models.v1_plugins_list_response import V1PluginsListResponse
631
+ from lightning_sdk.lightning_cloud.openapi.models.v1_post_cloud_space_artifact_events_response import V1PostCloudSpaceArtifactEventsResponse
616
632
  from lightning_sdk.lightning_cloud.openapi.models.v1_presigned_url import V1PresignedUrl
617
633
  from lightning_sdk.lightning_cloud.openapi.models.v1_profiler_capture import V1ProfilerCapture
618
634
  from lightning_sdk.lightning_cloud.openapi.models.v1_profiler_enabled_response import V1ProfilerEnabledResponse
@@ -716,6 +732,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_update_cluster_accelerators
716
732
  from lightning_sdk.lightning_cloud.openapi.models.v1_update_cluster_availability_request import V1UpdateClusterAvailabilityRequest
717
733
  from lightning_sdk.lightning_cloud.openapi.models.v1_update_index_response import V1UpdateIndexResponse
718
734
  from lightning_sdk.lightning_cloud.openapi.models.v1_update_lit_page_response import V1UpdateLitPageResponse
735
+ from lightning_sdk.lightning_cloud.openapi.models.v1_update_lit_repository_response import V1UpdateLitRepositoryResponse
719
736
  from lightning_sdk.lightning_cloud.openapi.models.v1_update_metrics_stream_visibility_response import V1UpdateMetricsStreamVisibilityResponse
720
737
  from lightning_sdk.lightning_cloud.openapi.models.v1_update_model_visibility_response import V1UpdateModelVisibilityResponse
721
738
  from lightning_sdk.lightning_cloud.openapi.models.v1_update_project_cluster_accelerators_response import V1UpdateProjectClusterAcceleratorsResponse
@@ -0,0 +1,175 @@
1
+ # coding: utf-8
2
+
3
+ """
4
+ external/v1/auth_service.proto
5
+
6
+ No description provided (generated by Swagger Codegen https://github.com/swagger-api/swagger-codegen) # noqa: E501
7
+
8
+ OpenAPI spec version: version not set
9
+
10
+ Generated by: https://github.com/swagger-api/swagger-codegen.git
11
+
12
+ NOTE
13
+ ----
14
+ standard swagger-codegen-cli for this python client has been modified
15
+ by custom templates. The purpose of these templates is to include
16
+ typing information in the API and Model code. Please refer to the
17
+ main grid repository for more info
18
+ """
19
+
20
+ import pprint
21
+ import re # noqa: F401
22
+
23
+ from typing import TYPE_CHECKING
24
+
25
+ import six
26
+
27
+ if TYPE_CHECKING:
28
+ from datetime import datetime
29
+ from lightning_sdk.lightning_cloud.openapi.models import *
30
+
31
+ class ClusterIdUsagerestrictionsBody(object):
32
+ """NOTE: This class is auto generated by the swagger code generator program.
33
+
34
+ Do not edit the class manually.
35
+ """
36
+ """
37
+ Attributes:
38
+ swagger_types (dict): The key is attribute name
39
+ and the value is attribute type.
40
+ attribute_map (dict): The key is attribute name
41
+ and the value is json key in definition.
42
+ """
43
+ swagger_types = {
44
+ 'instance_type': 'str',
45
+ 'max_instances': 'int',
46
+ 'restriction_type': 'str'
47
+ }
48
+
49
+ attribute_map = {
50
+ 'instance_type': 'instanceType',
51
+ 'max_instances': 'maxInstances',
52
+ 'restriction_type': 'restrictionType'
53
+ }
54
+
55
+ def __init__(self, instance_type: 'str' =None, max_instances: 'int' =None, restriction_type: 'str' =None): # noqa: E501
56
+ """ClusterIdUsagerestrictionsBody - a model defined in Swagger""" # noqa: E501
57
+ self._instance_type = None
58
+ self._max_instances = None
59
+ self._restriction_type = None
60
+ self.discriminator = None
61
+ if instance_type is not None:
62
+ self.instance_type = instance_type
63
+ if max_instances is not None:
64
+ self.max_instances = max_instances
65
+ if restriction_type is not None:
66
+ self.restriction_type = restriction_type
67
+
68
+ @property
69
+ def instance_type(self) -> 'str':
70
+ """Gets the instance_type of this ClusterIdUsagerestrictionsBody. # noqa: E501
71
+
72
+
73
+ :return: The instance_type of this ClusterIdUsagerestrictionsBody. # noqa: E501
74
+ :rtype: str
75
+ """
76
+ return self._instance_type
77
+
78
+ @instance_type.setter
79
+ def instance_type(self, instance_type: 'str'):
80
+ """Sets the instance_type of this ClusterIdUsagerestrictionsBody.
81
+
82
+
83
+ :param instance_type: The instance_type of this ClusterIdUsagerestrictionsBody. # noqa: E501
84
+ :type: str
85
+ """
86
+
87
+ self._instance_type = instance_type
88
+
89
+ @property
90
+ def max_instances(self) -> 'int':
91
+ """Gets the max_instances of this ClusterIdUsagerestrictionsBody. # noqa: E501
92
+
93
+
94
+ :return: The max_instances of this ClusterIdUsagerestrictionsBody. # noqa: E501
95
+ :rtype: int
96
+ """
97
+ return self._max_instances
98
+
99
+ @max_instances.setter
100
+ def max_instances(self, max_instances: 'int'):
101
+ """Sets the max_instances of this ClusterIdUsagerestrictionsBody.
102
+
103
+
104
+ :param max_instances: The max_instances of this ClusterIdUsagerestrictionsBody. # noqa: E501
105
+ :type: int
106
+ """
107
+
108
+ self._max_instances = max_instances
109
+
110
+ @property
111
+ def restriction_type(self) -> 'str':
112
+ """Gets the restriction_type of this ClusterIdUsagerestrictionsBody. # noqa: E501
113
+
114
+
115
+ :return: The restriction_type of this ClusterIdUsagerestrictionsBody. # noqa: E501
116
+ :rtype: str
117
+ """
118
+ return self._restriction_type
119
+
120
+ @restriction_type.setter
121
+ def restriction_type(self, restriction_type: 'str'):
122
+ """Sets the restriction_type of this ClusterIdUsagerestrictionsBody.
123
+
124
+
125
+ :param restriction_type: The restriction_type of this ClusterIdUsagerestrictionsBody. # noqa: E501
126
+ :type: str
127
+ """
128
+
129
+ self._restriction_type = restriction_type
130
+
131
+ def to_dict(self) -> dict:
132
+ """Returns the model properties as a dict"""
133
+ result = {}
134
+
135
+ for attr, _ in six.iteritems(self.swagger_types):
136
+ value = getattr(self, attr)
137
+ if isinstance(value, list):
138
+ result[attr] = list(map(
139
+ lambda x: x.to_dict() if hasattr(x, "to_dict") else x,
140
+ value
141
+ ))
142
+ elif hasattr(value, "to_dict"):
143
+ result[attr] = value.to_dict()
144
+ elif isinstance(value, dict):
145
+ result[attr] = dict(map(
146
+ lambda item: (item[0], item[1].to_dict())
147
+ if hasattr(item[1], "to_dict") else item,
148
+ value.items()
149
+ ))
150
+ else:
151
+ result[attr] = value
152
+ if issubclass(ClusterIdUsagerestrictionsBody, dict):
153
+ for key, value in self.items():
154
+ result[key] = value
155
+
156
+ return result
157
+
158
+ def to_str(self) -> str:
159
+ """Returns the string representation of the model"""
160
+ return pprint.pformat(self.to_dict())
161
+
162
+ def __repr__(self) -> str:
163
+ """For `print` and `pprint`"""
164
+ return self.to_str()
165
+
166
+ def __eq__(self, other: 'ClusterIdUsagerestrictionsBody') -> bool:
167
+ """Returns true if both objects are equal"""
168
+ if not isinstance(other, ClusterIdUsagerestrictionsBody):
169
+ return False
170
+
171
+ return self.__dict__ == other.__dict__
172
+
173
+ def __ne__(self, other: 'ClusterIdUsagerestrictionsBody') -> bool:
174
+ """Returns true if both objects are not equal"""
175
+ return not self == other
@@ -45,6 +45,7 @@ class DeploymentsIdBody(object):
45
45
  'autoscaling': 'V1AutoscalingSpec',
46
46
  'cloudspace_id': 'str',
47
47
  'created_at': 'datetime',
48
+ 'debug': 'bool',
48
49
  'desired_state': 'V1DeploymentState',
49
50
  'endpoint': 'V1Endpoint',
50
51
  'is_published': 'bool',
@@ -66,6 +67,7 @@ class DeploymentsIdBody(object):
66
67
  'autoscaling': 'autoscaling',
67
68
  'cloudspace_id': 'cloudspaceId',
68
69
  'created_at': 'createdAt',
70
+ 'debug': 'debug',
69
71
  'desired_state': 'desiredState',
70
72
  'endpoint': 'endpoint',
71
73
  'is_published': 'isPublished',
@@ -82,12 +84,13 @@ class DeploymentsIdBody(object):
82
84
  'user_id': 'userId'
83
85
  }
84
86
 
85
- def __init__(self, apis: 'list[V1DeploymentAPI]' =None, autoscaling: 'V1AutoscalingSpec' =None, cloudspace_id: 'str' =None, created_at: 'datetime' =None, desired_state: 'V1DeploymentState' =None, endpoint: 'V1Endpoint' =None, is_published: 'bool' =None, name: 'str' =None, parameter_spec: 'V1ParameterizationSpec' =None, release_id: 'str' =None, replicas: 'int' =None, spec: 'V1JobSpec' =None, status: 'V1DeploymentStatus' =None, strategy: 'V1DeploymentStrategy' =None, template_id: 'str' =None, total_cost: 'float' =None, updated_at: 'datetime' =None, user_id: 'str' =None): # noqa: E501
87
+ def __init__(self, apis: 'list[V1DeploymentAPI]' =None, autoscaling: 'V1AutoscalingSpec' =None, cloudspace_id: 'str' =None, created_at: 'datetime' =None, debug: 'bool' =None, desired_state: 'V1DeploymentState' =None, endpoint: 'V1Endpoint' =None, is_published: 'bool' =None, name: 'str' =None, parameter_spec: 'V1ParameterizationSpec' =None, release_id: 'str' =None, replicas: 'int' =None, spec: 'V1JobSpec' =None, status: 'V1DeploymentStatus' =None, strategy: 'V1DeploymentStrategy' =None, template_id: 'str' =None, total_cost: 'float' =None, updated_at: 'datetime' =None, user_id: 'str' =None): # noqa: E501
86
88
  """DeploymentsIdBody - a model defined in Swagger""" # noqa: E501
87
89
  self._apis = None
88
90
  self._autoscaling = None
89
91
  self._cloudspace_id = None
90
92
  self._created_at = None
93
+ self._debug = None
91
94
  self._desired_state = None
92
95
  self._endpoint = None
93
96
  self._is_published = None
@@ -111,6 +114,8 @@ class DeploymentsIdBody(object):
111
114
  self.cloudspace_id = cloudspace_id
112
115
  if created_at is not None:
113
116
  self.created_at = created_at
117
+ if debug is not None:
118
+ self.debug = debug
114
119
  if desired_state is not None:
115
120
  self.desired_state = desired_state
116
121
  if endpoint is not None:
@@ -224,6 +229,27 @@ class DeploymentsIdBody(object):
224
229
 
225
230
  self._created_at = created_at
226
231
 
232
+ @property
233
+ def debug(self) -> 'bool':
234
+ """Gets the debug of this DeploymentsIdBody. # noqa: E501
235
+
236
+
237
+ :return: The debug of this DeploymentsIdBody. # noqa: E501
238
+ :rtype: bool
239
+ """
240
+ return self._debug
241
+
242
+ @debug.setter
243
+ def debug(self, debug: 'bool'):
244
+ """Sets the debug of this DeploymentsIdBody.
245
+
246
+
247
+ :param debug: The debug of this DeploymentsIdBody. # noqa: E501
248
+ :type: bool
249
+ """
250
+
251
+ self._debug = debug
252
+
227
253
  @property
228
254
  def desired_state(self) -> 'V1DeploymentState':
229
255
  """Gets the desired_state of this DeploymentsIdBody. # noqa: E501