lightning-sdk 0.1.54__py3-none-any.whl → 0.1.56__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 (61) hide show
  1. lightning_sdk/__init__.py +1 -1
  2. lightning_sdk/ai_hub.py +22 -0
  3. lightning_sdk/api/ai_hub_api.py +21 -2
  4. lightning_sdk/api/deployment_api.py +4 -3
  5. lightning_sdk/api/job_api.py +5 -10
  6. lightning_sdk/api/lit_container_api.py +27 -7
  7. lightning_sdk/api/mmt_api.py +1 -4
  8. lightning_sdk/api/studio_api.py +5 -7
  9. lightning_sdk/api/teamspace_api.py +7 -0
  10. lightning_sdk/api/utils.py +1 -27
  11. lightning_sdk/cli/configure.py +92 -0
  12. lightning_sdk/cli/connect.py +31 -0
  13. lightning_sdk/cli/delete.py +32 -3
  14. lightning_sdk/cli/download.py +1 -1
  15. lightning_sdk/cli/entrypoint.py +14 -1
  16. lightning_sdk/cli/generate.py +35 -0
  17. lightning_sdk/cli/inspect.py +4 -2
  18. lightning_sdk/cli/jobs_menu.py +2 -1
  19. lightning_sdk/cli/list.py +49 -6
  20. lightning_sdk/cli/mmts_menu.py +2 -1
  21. lightning_sdk/cli/run.py +3 -3
  22. lightning_sdk/cli/serve.py +1 -2
  23. lightning_sdk/cli/start.py +43 -0
  24. lightning_sdk/cli/stop.py +30 -2
  25. lightning_sdk/cli/studios_menu.py +24 -1
  26. lightning_sdk/cli/switch.py +43 -0
  27. lightning_sdk/cli/teamspace_menu.py +2 -1
  28. lightning_sdk/cli/upload.py +6 -4
  29. lightning_sdk/lightning_cloud/openapi/__init__.py +6 -0
  30. lightning_sdk/lightning_cloud/openapi/api/cluster_service_api.py +115 -2
  31. lightning_sdk/lightning_cloud/openapi/api/jobs_service_api.py +113 -0
  32. lightning_sdk/lightning_cloud/openapi/api/lit_registry_service_api.py +216 -2
  33. lightning_sdk/lightning_cloud/openapi/api/projects_service_api.py +1 -5
  34. lightning_sdk/lightning_cloud/openapi/models/__init__.py +6 -0
  35. lightning_sdk/lightning_cloud/openapi/models/cluster_id_usagerestrictions_body.py +27 -1
  36. lightning_sdk/lightning_cloud/openapi/models/id_reportrestarttimings_body.py +123 -0
  37. lightning_sdk/lightning_cloud/openapi/models/project_id_litregistry_body.py +2 -0
  38. lightning_sdk/lightning_cloud/openapi/models/usagerestrictions_id_body.py +27 -1
  39. lightning_sdk/lightning_cloud/openapi/models/v1_cloud_provider.py +3 -0
  40. lightning_sdk/lightning_cloud/openapi/models/v1_cluster_accelerator.py +53 -1
  41. lightning_sdk/lightning_cloud/openapi/models/v1_get_cluster_accelerator_demand_response.py +123 -0
  42. lightning_sdk/lightning_cloud/openapi/models/v1_job.py +53 -1
  43. lightning_sdk/lightning_cloud/openapi/models/v1_job_spec.py +27 -1
  44. lightning_sdk/lightning_cloud/openapi/models/v1_lambda_labs_direct_v1.py +55 -3
  45. lightning_sdk/lightning_cloud/openapi/models/v1_list_lit_registry_repository_image_artifact_versions_response.py +231 -0
  46. lightning_sdk/lightning_cloud/openapi/models/v1_lit_registry_artifact.py +279 -0
  47. lightning_sdk/lightning_cloud/openapi/models/v1_lit_registry_project.py +8 -0
  48. lightning_sdk/lightning_cloud/openapi/models/v1_lit_repository.py +27 -1
  49. lightning_sdk/lightning_cloud/openapi/models/v1_report_restart_timings_response.py +97 -0
  50. lightning_sdk/lightning_cloud/openapi/models/v1_restart_timing.py +175 -0
  51. lightning_sdk/lightning_cloud/openapi/models/v1_user_features.py +27 -53
  52. lightning_sdk/lightning_cloud/openapi/models/v1_validate_deployment_image_request.py +27 -1
  53. lightning_sdk/machine.py +59 -27
  54. lightning_sdk/studio.py +5 -1
  55. lightning_sdk/teamspace.py +25 -0
  56. {lightning_sdk-0.1.54.dist-info → lightning_sdk-0.1.56.dist-info}/METADATA +2 -1
  57. {lightning_sdk-0.1.54.dist-info → lightning_sdk-0.1.56.dist-info}/RECORD +61 -50
  58. {lightning_sdk-0.1.54.dist-info → lightning_sdk-0.1.56.dist-info}/LICENSE +0 -0
  59. {lightning_sdk-0.1.54.dist-info → lightning_sdk-0.1.56.dist-info}/WHEEL +0 -0
  60. {lightning_sdk-0.1.54.dist-info → lightning_sdk-0.1.56.dist-info}/entry_points.txt +0 -0
  61. {lightning_sdk-0.1.54.dist-info → lightning_sdk-0.1.56.dist-info}/top_level.txt +0 -0
@@ -3035,6 +3035,119 @@ class JobsServiceApi(object):
3035
3035
  _request_timeout=params.get('_request_timeout'),
3036
3036
  collection_formats=collection_formats)
3037
3037
 
3038
+ def jobs_service_report_restart_timings(self, body: 'IdReportrestarttimingsBody', project_id: 'str', id: 'str', **kwargs) -> 'V1ReportRestartTimingsResponse': # noqa: E501
3039
+ """The tired proxy collects the time at which the user logs started and inform the CP # noqa: E501
3040
+
3041
+ This method makes a synchronous HTTP request by default. To make an
3042
+ asynchronous HTTP request, please pass async_req=True
3043
+ >>> thread = api.jobs_service_report_restart_timings(body, project_id, id, async_req=True)
3044
+ >>> result = thread.get()
3045
+
3046
+ :param async_req bool
3047
+ :param IdReportrestarttimingsBody body: (required)
3048
+ :param str project_id: (required)
3049
+ :param str id: (required)
3050
+ :return: V1ReportRestartTimingsResponse
3051
+ If the method is called asynchronously,
3052
+ returns the request thread.
3053
+ """
3054
+ kwargs['_return_http_data_only'] = True
3055
+ if kwargs.get('async_req'):
3056
+ return self.jobs_service_report_restart_timings_with_http_info(body, project_id, id, **kwargs) # noqa: E501
3057
+ else:
3058
+ (data) = self.jobs_service_report_restart_timings_with_http_info(body, project_id, id, **kwargs) # noqa: E501
3059
+ return data
3060
+
3061
+ def jobs_service_report_restart_timings_with_http_info(self, body: 'IdReportrestarttimingsBody', project_id: 'str', id: 'str', **kwargs) -> 'V1ReportRestartTimingsResponse': # noqa: E501
3062
+ """The tired proxy collects the time at which the user logs started and inform the CP # noqa: E501
3063
+
3064
+ This method makes a synchronous HTTP request by default. To make an
3065
+ asynchronous HTTP request, please pass async_req=True
3066
+ >>> thread = api.jobs_service_report_restart_timings_with_http_info(body, project_id, id, async_req=True)
3067
+ >>> result = thread.get()
3068
+
3069
+ :param async_req bool
3070
+ :param IdReportrestarttimingsBody body: (required)
3071
+ :param str project_id: (required)
3072
+ :param str id: (required)
3073
+ :return: V1ReportRestartTimingsResponse
3074
+ If the method is called asynchronously,
3075
+ returns the request thread.
3076
+ """
3077
+
3078
+ all_params = ['body', 'project_id', 'id'] # noqa: E501
3079
+ all_params.append('async_req')
3080
+ all_params.append('_return_http_data_only')
3081
+ all_params.append('_preload_content')
3082
+ all_params.append('_request_timeout')
3083
+
3084
+ params = locals()
3085
+ for key, val in six.iteritems(params['kwargs']):
3086
+ if key not in all_params:
3087
+ raise TypeError(
3088
+ "Got an unexpected keyword argument '%s'"
3089
+ " to method jobs_service_report_restart_timings" % key
3090
+ )
3091
+ params[key] = val
3092
+ del params['kwargs']
3093
+ # verify the required parameter 'body' is set
3094
+ if ('body' not in params or
3095
+ params['body'] is None):
3096
+ raise ValueError("Missing the required parameter `body` when calling `jobs_service_report_restart_timings`") # noqa: E501
3097
+ # verify the required parameter 'project_id' is set
3098
+ if ('project_id' not in params or
3099
+ params['project_id'] is None):
3100
+ raise ValueError("Missing the required parameter `project_id` when calling `jobs_service_report_restart_timings`") # noqa: E501
3101
+ # verify the required parameter 'id' is set
3102
+ if ('id' not in params or
3103
+ params['id'] is None):
3104
+ raise ValueError("Missing the required parameter `id` when calling `jobs_service_report_restart_timings`") # noqa: E501
3105
+
3106
+ collection_formats = {}
3107
+
3108
+ path_params = {}
3109
+ if 'project_id' in params:
3110
+ path_params['projectId'] = params['project_id'] # noqa: E501
3111
+ if 'id' in params:
3112
+ path_params['id'] = params['id'] # noqa: E501
3113
+
3114
+ query_params = []
3115
+
3116
+ header_params = {}
3117
+
3118
+ form_params = []
3119
+ local_var_files = {}
3120
+
3121
+ body_params = None
3122
+ if 'body' in params:
3123
+ body_params = params['body']
3124
+ # HTTP header `Accept`
3125
+ header_params['Accept'] = self.api_client.select_header_accept(
3126
+ ['application/json']) # noqa: E501
3127
+
3128
+ # HTTP header `Content-Type`
3129
+ header_params['Content-Type'] = self.api_client.select_header_content_type( # noqa: E501
3130
+ ['application/json']) # noqa: E501
3131
+
3132
+ # Authentication setting
3133
+ auth_settings = [] # noqa: E501
3134
+
3135
+ return self.api_client.call_api(
3136
+ '/v1/projects/{projectId}/jobs/{id}/report-restart-timings', 'PUT',
3137
+ path_params,
3138
+ query_params,
3139
+ header_params,
3140
+ body=body_params,
3141
+ post_params=form_params,
3142
+ files=local_var_files,
3143
+ response_type='V1ReportRestartTimingsResponse', # noqa: E501
3144
+ auth_settings=auth_settings,
3145
+ async_req=params.get('async_req'),
3146
+ _return_http_data_only=params.get('_return_http_data_only'),
3147
+ _preload_content=params.get('_preload_content', True),
3148
+ _request_timeout=params.get('_request_timeout'),
3149
+ collection_formats=collection_formats)
3150
+
3038
3151
  def jobs_service_restore_deployment_release(self, body: 'object', project_id: 'str', deployment_id: 'str', id: 'str', **kwargs) -> 'V1RestoreDeploymentReleaseResponse': # noqa: E501
3039
3152
  """jobs_service_restore_deployment_release # noqa: E501
3040
3153
 
@@ -53,7 +53,7 @@ class LitRegistryServiceApi(object):
53
53
 
54
54
  :param async_req bool
55
55
  :param ProjectIdLitregistryBody body: (required)
56
- :param str project_id: (required)
56
+ :param str project_id: The teamspace project_id (required)
57
57
  :return: V1LitRegistryProject
58
58
  If the method is called asynchronously,
59
59
  returns the request thread.
@@ -75,7 +75,7 @@ class LitRegistryServiceApi(object):
75
75
 
76
76
  :param async_req bool
77
77
  :param ProjectIdLitregistryBody body: (required)
78
- :param str project_id: (required)
78
+ :param str project_id: The teamspace project_id (required)
79
79
  :return: V1LitRegistryProject
80
80
  If the method is called asynchronously,
81
81
  returns the request thread.
@@ -342,6 +342,115 @@ class LitRegistryServiceApi(object):
342
342
  _request_timeout=params.get('_request_timeout'),
343
343
  collection_formats=collection_formats)
344
344
 
345
+ def lit_registry_service_get_lit_registry_repository_image_artifact_version_by_digest(self, project_id: 'str', lit_repo_name: 'str', full_hash_digest: 'str', **kwargs) -> 'V1LitRegistryArtifact': # noqa: E501
346
+ """lit_registry_service_get_lit_registry_repository_image_artifact_version_by_digest # noqa: E501
347
+
348
+ This method makes a synchronous HTTP request by default. To make an
349
+ asynchronous HTTP request, please pass async_req=True
350
+ >>> thread = api.lit_registry_service_get_lit_registry_repository_image_artifact_version_by_digest(project_id, lit_repo_name, full_hash_digest, async_req=True)
351
+ >>> result = thread.get()
352
+
353
+ :param async_req bool
354
+ :param str project_id: (required)
355
+ :param str lit_repo_name: (required)
356
+ :param str full_hash_digest: (required)
357
+ :return: V1LitRegistryArtifact
358
+ If the method is called asynchronously,
359
+ returns the request thread.
360
+ """
361
+ kwargs['_return_http_data_only'] = True
362
+ if kwargs.get('async_req'):
363
+ return self.lit_registry_service_get_lit_registry_repository_image_artifact_version_by_digest_with_http_info(project_id, lit_repo_name, full_hash_digest, **kwargs) # noqa: E501
364
+ else:
365
+ (data) = self.lit_registry_service_get_lit_registry_repository_image_artifact_version_by_digest_with_http_info(project_id, lit_repo_name, full_hash_digest, **kwargs) # noqa: E501
366
+ return data
367
+
368
+ def lit_registry_service_get_lit_registry_repository_image_artifact_version_by_digest_with_http_info(self, project_id: 'str', lit_repo_name: 'str', full_hash_digest: 'str', **kwargs) -> 'V1LitRegistryArtifact': # noqa: E501
369
+ """lit_registry_service_get_lit_registry_repository_image_artifact_version_by_digest # noqa: E501
370
+
371
+ This method makes a synchronous HTTP request by default. To make an
372
+ asynchronous HTTP request, please pass async_req=True
373
+ >>> thread = api.lit_registry_service_get_lit_registry_repository_image_artifact_version_by_digest_with_http_info(project_id, lit_repo_name, full_hash_digest, async_req=True)
374
+ >>> result = thread.get()
375
+
376
+ :param async_req bool
377
+ :param str project_id: (required)
378
+ :param str lit_repo_name: (required)
379
+ :param str full_hash_digest: (required)
380
+ :return: V1LitRegistryArtifact
381
+ If the method is called asynchronously,
382
+ returns the request thread.
383
+ """
384
+
385
+ all_params = ['project_id', 'lit_repo_name', 'full_hash_digest'] # 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 lit_registry_service_get_lit_registry_repository_image_artifact_version_by_digest" % key
397
+ )
398
+ params[key] = val
399
+ del params['kwargs']
400
+ # verify the required parameter 'project_id' is set
401
+ if ('project_id' not in params or
402
+ params['project_id'] is None):
403
+ raise ValueError("Missing the required parameter `project_id` when calling `lit_registry_service_get_lit_registry_repository_image_artifact_version_by_digest`") # noqa: E501
404
+ # verify the required parameter 'lit_repo_name' is set
405
+ if ('lit_repo_name' not in params or
406
+ params['lit_repo_name'] is None):
407
+ raise ValueError("Missing the required parameter `lit_repo_name` when calling `lit_registry_service_get_lit_registry_repository_image_artifact_version_by_digest`") # noqa: E501
408
+ # verify the required parameter 'full_hash_digest' is set
409
+ if ('full_hash_digest' not in params or
410
+ params['full_hash_digest'] is None):
411
+ raise ValueError("Missing the required parameter `full_hash_digest` when calling `lit_registry_service_get_lit_registry_repository_image_artifact_version_by_digest`") # noqa: E501
412
+
413
+ collection_formats = {}
414
+
415
+ path_params = {}
416
+ if 'project_id' in params:
417
+ path_params['projectId'] = params['project_id'] # noqa: E501
418
+ if 'lit_repo_name' in params:
419
+ path_params['litRepoName'] = params['lit_repo_name'] # noqa: E501
420
+ if 'full_hash_digest' in params:
421
+ path_params['fullHashDigest'] = params['full_hash_digest'] # noqa: E501
422
+
423
+ query_params = []
424
+
425
+ header_params = {}
426
+
427
+ form_params = []
428
+ local_var_files = {}
429
+
430
+ body_params = None
431
+ # HTTP header `Accept`
432
+ header_params['Accept'] = self.api_client.select_header_accept(
433
+ ['application/json']) # noqa: E501
434
+
435
+ # Authentication setting
436
+ auth_settings = [] # noqa: E501
437
+
438
+ return self.api_client.call_api(
439
+ '/v1/projects/{projectId}/lit-registry/{litRepoName}/artifacts/{fullHashDigest}', 'GET',
440
+ path_params,
441
+ query_params,
442
+ header_params,
443
+ body=body_params,
444
+ post_params=form_params,
445
+ files=local_var_files,
446
+ response_type='V1LitRegistryArtifact', # noqa: E501
447
+ auth_settings=auth_settings,
448
+ async_req=params.get('async_req'),
449
+ _return_http_data_only=params.get('_return_http_data_only'),
450
+ _preload_content=params.get('_preload_content', True),
451
+ _request_timeout=params.get('_request_timeout'),
452
+ collection_formats=collection_formats)
453
+
345
454
  def lit_registry_service_get_lit_repository(self, project_id: 'str', lit_repo_name: 'str', **kwargs) -> 'V1LitRepository': # noqa: E501
346
455
  """lit_registry_service_get_lit_repository # noqa: E501
347
456
 
@@ -443,6 +552,111 @@ class LitRegistryServiceApi(object):
443
552
  _request_timeout=params.get('_request_timeout'),
444
553
  collection_formats=collection_formats)
445
554
 
555
+ def lit_registry_service_list_lit_registry_repository_image_artifact_versions(self, project_id: 'str', lit_repo_name: 'str', **kwargs) -> 'V1ListLitRegistryRepositoryImageArtifactVersionsResponse': # noqa: E501
556
+ """lit_registry_service_list_lit_registry_repository_image_artifact_versions # noqa: E501
557
+
558
+ This method makes a synchronous HTTP request by default. To make an
559
+ asynchronous HTTP request, please pass async_req=True
560
+ >>> thread = api.lit_registry_service_list_lit_registry_repository_image_artifact_versions(project_id, lit_repo_name, async_req=True)
561
+ >>> result = thread.get()
562
+
563
+ :param async_req bool
564
+ :param str project_id: (required)
565
+ :param str lit_repo_name: (required)
566
+ :param bool latest_only:
567
+ :return: V1ListLitRegistryRepositoryImageArtifactVersionsResponse
568
+ If the method is called asynchronously,
569
+ returns the request thread.
570
+ """
571
+ kwargs['_return_http_data_only'] = True
572
+ if kwargs.get('async_req'):
573
+ return self.lit_registry_service_list_lit_registry_repository_image_artifact_versions_with_http_info(project_id, lit_repo_name, **kwargs) # noqa: E501
574
+ else:
575
+ (data) = self.lit_registry_service_list_lit_registry_repository_image_artifact_versions_with_http_info(project_id, lit_repo_name, **kwargs) # noqa: E501
576
+ return data
577
+
578
+ def lit_registry_service_list_lit_registry_repository_image_artifact_versions_with_http_info(self, project_id: 'str', lit_repo_name: 'str', **kwargs) -> 'V1ListLitRegistryRepositoryImageArtifactVersionsResponse': # noqa: E501
579
+ """lit_registry_service_list_lit_registry_repository_image_artifact_versions # noqa: E501
580
+
581
+ This method makes a synchronous HTTP request by default. To make an
582
+ asynchronous HTTP request, please pass async_req=True
583
+ >>> thread = api.lit_registry_service_list_lit_registry_repository_image_artifact_versions_with_http_info(project_id, lit_repo_name, async_req=True)
584
+ >>> result = thread.get()
585
+
586
+ :param async_req bool
587
+ :param str project_id: (required)
588
+ :param str lit_repo_name: (required)
589
+ :param bool latest_only:
590
+ :return: V1ListLitRegistryRepositoryImageArtifactVersionsResponse
591
+ If the method is called asynchronously,
592
+ returns the request thread.
593
+ """
594
+
595
+ all_params = ['project_id', 'lit_repo_name', 'latest_only'] # noqa: E501
596
+ all_params.append('async_req')
597
+ all_params.append('_return_http_data_only')
598
+ all_params.append('_preload_content')
599
+ all_params.append('_request_timeout')
600
+
601
+ params = locals()
602
+ for key, val in six.iteritems(params['kwargs']):
603
+ if key not in all_params:
604
+ raise TypeError(
605
+ "Got an unexpected keyword argument '%s'"
606
+ " to method lit_registry_service_list_lit_registry_repository_image_artifact_versions" % key
607
+ )
608
+ params[key] = val
609
+ del params['kwargs']
610
+ # verify the required parameter 'project_id' is set
611
+ if ('project_id' not in params or
612
+ params['project_id'] is None):
613
+ raise ValueError("Missing the required parameter `project_id` when calling `lit_registry_service_list_lit_registry_repository_image_artifact_versions`") # noqa: E501
614
+ # verify the required parameter 'lit_repo_name' is set
615
+ if ('lit_repo_name' not in params or
616
+ params['lit_repo_name'] is None):
617
+ raise ValueError("Missing the required parameter `lit_repo_name` when calling `lit_registry_service_list_lit_registry_repository_image_artifact_versions`") # noqa: E501
618
+
619
+ collection_formats = {}
620
+
621
+ path_params = {}
622
+ if 'project_id' in params:
623
+ path_params['projectId'] = params['project_id'] # noqa: E501
624
+ if 'lit_repo_name' in params:
625
+ path_params['litRepoName'] = params['lit_repo_name'] # noqa: E501
626
+
627
+ query_params = []
628
+ if 'latest_only' in params:
629
+ query_params.append(('latestOnly', params['latest_only'])) # noqa: E501
630
+
631
+ header_params = {}
632
+
633
+ form_params = []
634
+ local_var_files = {}
635
+
636
+ body_params = None
637
+ # HTTP header `Accept`
638
+ header_params['Accept'] = self.api_client.select_header_accept(
639
+ ['application/json']) # noqa: E501
640
+
641
+ # Authentication setting
642
+ auth_settings = [] # noqa: E501
643
+
644
+ return self.api_client.call_api(
645
+ '/v1/projects/{projectId}/lit-registry/{litRepoName}/artifacts', 'GET',
646
+ path_params,
647
+ query_params,
648
+ header_params,
649
+ body=body_params,
650
+ post_params=form_params,
651
+ files=local_var_files,
652
+ response_type='V1ListLitRegistryRepositoryImageArtifactVersionsResponse', # noqa: E501
653
+ auth_settings=auth_settings,
654
+ async_req=params.get('async_req'),
655
+ _return_http_data_only=params.get('_return_http_data_only'),
656
+ _preload_content=params.get('_preload_content', True),
657
+ _request_timeout=params.get('_request_timeout'),
658
+ collection_formats=collection_formats)
659
+
446
660
  def lit_registry_service_update_lit_repository_metadata(self, body: 'LitregistryLitRepoNameBody', project_id: 'str', lit_repo_name: 'str', **kwargs) -> 'V1UpdateLitRepositoryResponse': # noqa: E501
447
661
  """lit_registry_service_update_lit_repository_metadata # noqa: E501
448
662
 
@@ -578,7 +578,6 @@ class ProjectsServiceApi(object):
578
578
 
579
579
  :param async_req bool
580
580
  :param str id: (required)
581
- :param bool skip_balance_check:
582
581
  :param str balance_destination_id:
583
582
  :param bool is_org:
584
583
  :return: V1DeleteProjectResponse
@@ -602,7 +601,6 @@ class ProjectsServiceApi(object):
602
601
 
603
602
  :param async_req bool
604
603
  :param str id: (required)
605
- :param bool skip_balance_check:
606
604
  :param str balance_destination_id:
607
605
  :param bool is_org:
608
606
  :return: V1DeleteProjectResponse
@@ -610,7 +608,7 @@ class ProjectsServiceApi(object):
610
608
  returns the request thread.
611
609
  """
612
610
 
613
- all_params = ['id', 'skip_balance_check', 'balance_destination_id', 'is_org'] # noqa: E501
611
+ all_params = ['id', 'balance_destination_id', 'is_org'] # noqa: E501
614
612
  all_params.append('async_req')
615
613
  all_params.append('_return_http_data_only')
616
614
  all_params.append('_preload_content')
@@ -637,8 +635,6 @@ class ProjectsServiceApi(object):
637
635
  path_params['id'] = params['id'] # noqa: E501
638
636
 
639
637
  query_params = []
640
- if 'skip_balance_check' in params:
641
- query_params.append(('skipBalanceCheck', params['skip_balance_check'])) # noqa: E501
642
638
  if 'balance_destination_id' in params:
643
639
  query_params.append(('balanceDestinationId', params['balance_destination_id'])) # noqa: E501
644
640
  if 'is_org' in params:
@@ -87,6 +87,7 @@ from lightning_sdk.lightning_cloud.openapi.models.id_publications_body import Id
87
87
  from lightning_sdk.lightning_cloud.openapi.models.id_publications_body1 import IdPublicationsBody1
88
88
  from lightning_sdk.lightning_cloud.openapi.models.id_release_body import IdReleaseBody
89
89
  from lightning_sdk.lightning_cloud.openapi.models.id_reportlogsactivity_body import IdReportlogsactivityBody
90
+ from lightning_sdk.lightning_cloud.openapi.models.id_reportrestarttimings_body import IdReportrestarttimingsBody
90
91
  from lightning_sdk.lightning_cloud.openapi.models.id_start_body import IdStartBody
91
92
  from lightning_sdk.lightning_cloud.openapi.models.id_storage_body import IdStorageBody
92
93
  from lightning_sdk.lightning_cloud.openapi.models.id_uploads_body import IdUploadsBody
@@ -409,6 +410,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_get_agent_job_logs_metadata
409
410
  from lightning_sdk.lightning_cloud.openapi.models.v1_get_artifacts_page_response import V1GetArtifactsPageResponse
410
411
  from lightning_sdk.lightning_cloud.openapi.models.v1_get_cloud_space_instance_status_response import V1GetCloudSpaceInstanceStatusResponse
411
412
  from lightning_sdk.lightning_cloud.openapi.models.v1_get_cloud_space_size_response import V1GetCloudSpaceSizeResponse
413
+ from lightning_sdk.lightning_cloud.openapi.models.v1_get_cluster_accelerator_demand_response import V1GetClusterAcceleratorDemandResponse
412
414
  from lightning_sdk.lightning_cloud.openapi.models.v1_get_cluster_credentials_response import V1GetClusterCredentialsResponse
413
415
  from lightning_sdk.lightning_cloud.openapi.models.v1_get_deployment_routing_telemetry_aggregated_response import V1GetDeploymentRoutingTelemetryAggregatedResponse
414
416
  from lightning_sdk.lightning_cloud.openapi.models.v1_get_deployment_routing_telemetry_response import V1GetDeploymentRoutingTelemetryResponse
@@ -534,6 +536,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_list_lightningapp_instances
534
536
  from lightning_sdk.lightning_cloud.openapi.models.v1_list_lightningwork_events_response import V1ListLightningworkEventsResponse
535
537
  from lightning_sdk.lightning_cloud.openapi.models.v1_list_lightningwork_response import V1ListLightningworkResponse
536
538
  from lightning_sdk.lightning_cloud.openapi.models.v1_list_lit_pages_response import V1ListLitPagesResponse
539
+ from lightning_sdk.lightning_cloud.openapi.models.v1_list_lit_registry_repository_image_artifact_versions_response import V1ListLitRegistryRepositoryImageArtifactVersionsResponse
537
540
  from lightning_sdk.lightning_cloud.openapi.models.v1_list_logger_artifact_response import V1ListLoggerArtifactResponse
538
541
  from lightning_sdk.lightning_cloud.openapi.models.v1_list_managed_endpoints_response import V1ListManagedEndpointsResponse
539
542
  from lightning_sdk.lightning_cloud.openapi.models.v1_list_memberships_response import V1ListMembershipsResponse
@@ -569,6 +572,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_list_studio_jobs_response i
569
572
  from lightning_sdk.lightning_cloud.openapi.models.v1_list_user_slurm_jobs_response import V1ListUserSLURMJobsResponse
570
573
  from lightning_sdk.lightning_cloud.openapi.models.v1_lit_page import V1LitPage
571
574
  from lightning_sdk.lightning_cloud.openapi.models.v1_lit_page_type import V1LitPageType
575
+ from lightning_sdk.lightning_cloud.openapi.models.v1_lit_registry_artifact import V1LitRegistryArtifact
572
576
  from lightning_sdk.lightning_cloud.openapi.models.v1_lit_registry_project import V1LitRegistryProject
573
577
  from lightning_sdk.lightning_cloud.openapi.models.v1_lit_repository import V1LitRepository
574
578
  from lightning_sdk.lightning_cloud.openapi.models.v1_locked_resource import V1LockedResource
@@ -662,6 +666,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_refresh_response import V1R
662
666
  from lightning_sdk.lightning_cloud.openapi.models.v1_region_state import V1RegionState
663
667
  from lightning_sdk.lightning_cloud.openapi.models.v1_regional_load_balancer import V1RegionalLoadBalancer
664
668
  from lightning_sdk.lightning_cloud.openapi.models.v1_report_logs_activity_response import V1ReportLogsActivityResponse
669
+ from lightning_sdk.lightning_cloud.openapi.models.v1_report_restart_timings_response import V1ReportRestartTimingsResponse
665
670
  from lightning_sdk.lightning_cloud.openapi.models.v1_request_cluster_access_request import V1RequestClusterAccessRequest
666
671
  from lightning_sdk.lightning_cloud.openapi.models.v1_request_cluster_access_response import V1RequestClusterAccessResponse
667
672
  from lightning_sdk.lightning_cloud.openapi.models.v1_request_verification_code_response import V1RequestVerificationCodeResponse
@@ -671,6 +676,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_resources import V1Resource
671
676
  from lightning_sdk.lightning_cloud.openapi.models.v1_response_choice import V1ResponseChoice
672
677
  from lightning_sdk.lightning_cloud.openapi.models.v1_response_choice_delta import V1ResponseChoiceDelta
673
678
  from lightning_sdk.lightning_cloud.openapi.models.v1_restart_cloud_space_instance_response import V1RestartCloudSpaceInstanceResponse
679
+ from lightning_sdk.lightning_cloud.openapi.models.v1_restart_timing import V1RestartTiming
674
680
  from lightning_sdk.lightning_cloud.openapi.models.v1_restore_deployment_release_response import V1RestoreDeploymentReleaseResponse
675
681
  from lightning_sdk.lightning_cloud.openapi.models.v1_role import V1Role
676
682
  from lightning_sdk.lightning_cloud.openapi.models.v1_rolling_update_strategy import V1RollingUpdateStrategy
@@ -43,25 +43,30 @@ class ClusterIdUsagerestrictionsBody(object):
43
43
  swagger_types = {
44
44
  'instance_type': 'str',
45
45
  'max_instances': 'int',
46
+ 'org_id': 'str',
46
47
  'restriction_type': 'str'
47
48
  }
48
49
 
49
50
  attribute_map = {
50
51
  'instance_type': 'instanceType',
51
52
  'max_instances': 'maxInstances',
53
+ 'org_id': 'orgId',
52
54
  'restriction_type': 'restrictionType'
53
55
  }
54
56
 
55
- def __init__(self, instance_type: 'str' =None, max_instances: 'int' =None, restriction_type: 'str' =None): # noqa: E501
57
+ def __init__(self, instance_type: 'str' =None, max_instances: 'int' =None, org_id: 'str' =None, restriction_type: 'str' =None): # noqa: E501
56
58
  """ClusterIdUsagerestrictionsBody - a model defined in Swagger""" # noqa: E501
57
59
  self._instance_type = None
58
60
  self._max_instances = None
61
+ self._org_id = None
59
62
  self._restriction_type = None
60
63
  self.discriminator = None
61
64
  if instance_type is not None:
62
65
  self.instance_type = instance_type
63
66
  if max_instances is not None:
64
67
  self.max_instances = max_instances
68
+ if org_id is not None:
69
+ self.org_id = org_id
65
70
  if restriction_type is not None:
66
71
  self.restriction_type = restriction_type
67
72
 
@@ -107,6 +112,27 @@ class ClusterIdUsagerestrictionsBody(object):
107
112
 
108
113
  self._max_instances = max_instances
109
114
 
115
+ @property
116
+ def org_id(self) -> 'str':
117
+ """Gets the org_id of this ClusterIdUsagerestrictionsBody. # noqa: E501
118
+
119
+
120
+ :return: The org_id of this ClusterIdUsagerestrictionsBody. # noqa: E501
121
+ :rtype: str
122
+ """
123
+ return self._org_id
124
+
125
+ @org_id.setter
126
+ def org_id(self, org_id: 'str'):
127
+ """Sets the org_id of this ClusterIdUsagerestrictionsBody.
128
+
129
+
130
+ :param org_id: The org_id of this ClusterIdUsagerestrictionsBody. # noqa: E501
131
+ :type: str
132
+ """
133
+
134
+ self._org_id = org_id
135
+
110
136
  @property
111
137
  def restriction_type(self) -> 'str':
112
138
  """Gets the restriction_type of this ClusterIdUsagerestrictionsBody. # noqa: E501
@@ -0,0 +1,123 @@
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 IdReportrestarttimingsBody(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
+ 'timings': 'list[V1RestartTiming]'
45
+ }
46
+
47
+ attribute_map = {
48
+ 'timings': 'timings'
49
+ }
50
+
51
+ def __init__(self, timings: 'list[V1RestartTiming]' =None): # noqa: E501
52
+ """IdReportrestarttimingsBody - a model defined in Swagger""" # noqa: E501
53
+ self._timings = None
54
+ self.discriminator = None
55
+ if timings is not None:
56
+ self.timings = timings
57
+
58
+ @property
59
+ def timings(self) -> 'list[V1RestartTiming]':
60
+ """Gets the timings of this IdReportrestarttimingsBody. # noqa: E501
61
+
62
+
63
+ :return: The timings of this IdReportrestarttimingsBody. # noqa: E501
64
+ :rtype: list[V1RestartTiming]
65
+ """
66
+ return self._timings
67
+
68
+ @timings.setter
69
+ def timings(self, timings: 'list[V1RestartTiming]'):
70
+ """Sets the timings of this IdReportrestarttimingsBody.
71
+
72
+
73
+ :param timings: The timings of this IdReportrestarttimingsBody. # noqa: E501
74
+ :type: list[V1RestartTiming]
75
+ """
76
+
77
+ self._timings = timings
78
+
79
+ def to_dict(self) -> dict:
80
+ """Returns the model properties as a dict"""
81
+ result = {}
82
+
83
+ for attr, _ in six.iteritems(self.swagger_types):
84
+ value = getattr(self, attr)
85
+ if isinstance(value, list):
86
+ result[attr] = list(map(
87
+ lambda x: x.to_dict() if hasattr(x, "to_dict") else x,
88
+ value
89
+ ))
90
+ elif hasattr(value, "to_dict"):
91
+ result[attr] = value.to_dict()
92
+ elif isinstance(value, dict):
93
+ result[attr] = dict(map(
94
+ lambda item: (item[0], item[1].to_dict())
95
+ if hasattr(item[1], "to_dict") else item,
96
+ value.items()
97
+ ))
98
+ else:
99
+ result[attr] = value
100
+ if issubclass(IdReportrestarttimingsBody, dict):
101
+ for key, value in self.items():
102
+ result[key] = value
103
+
104
+ return result
105
+
106
+ def to_str(self) -> str:
107
+ """Returns the string representation of the model"""
108
+ return pprint.pformat(self.to_dict())
109
+
110
+ def __repr__(self) -> str:
111
+ """For `print` and `pprint`"""
112
+ return self.to_str()
113
+
114
+ def __eq__(self, other: 'IdReportrestarttimingsBody') -> bool:
115
+ """Returns true if both objects are equal"""
116
+ if not isinstance(other, IdReportrestarttimingsBody):
117
+ return False
118
+
119
+ return self.__dict__ == other.__dict__
120
+
121
+ def __ne__(self, other: 'IdReportrestarttimingsBody') -> bool:
122
+ """Returns true if both objects are not equal"""
123
+ return not self == other
@@ -59,6 +59,7 @@ class ProjectIdLitregistryBody(object):
59
59
  def lit_reg_project_name(self) -> 'str':
60
60
  """Gets the lit_reg_project_name of this ProjectIdLitregistryBody. # noqa: E501
61
61
 
62
+ This will be the name of the project in the registry, 1:1 with org-name|username/teamspace-name name. # noqa: E501
62
63
 
63
64
  :return: The lit_reg_project_name of this ProjectIdLitregistryBody. # noqa: E501
64
65
  :rtype: str
@@ -69,6 +70,7 @@ class ProjectIdLitregistryBody(object):
69
70
  def lit_reg_project_name(self, lit_reg_project_name: 'str'):
70
71
  """Sets the lit_reg_project_name of this ProjectIdLitregistryBody.
71
72
 
73
+ This will be the name of the project in the registry, 1:1 with org-name|username/teamspace-name name. # noqa: E501
72
74
 
73
75
  :param lit_reg_project_name: The lit_reg_project_name of this ProjectIdLitregistryBody. # noqa: E501
74
76
  :type: str