lightning-sdk 0.1.55__py3-none-any.whl → 0.1.57__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 (72) hide show
  1. lightning_sdk/__init__.py +3 -2
  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/mmt_api.py +1 -4
  7. lightning_sdk/api/studio_api.py +5 -7
  8. lightning_sdk/api/teamspace_api.py +7 -0
  9. lightning_sdk/api/utils.py +1 -27
  10. lightning_sdk/cli/ai_hub.py +61 -10
  11. lightning_sdk/cli/configure.py +137 -0
  12. lightning_sdk/cli/connect.py +47 -0
  13. lightning_sdk/cli/delete.py +83 -32
  14. lightning_sdk/cli/download.py +177 -90
  15. lightning_sdk/cli/entrypoint.py +50 -15
  16. lightning_sdk/cli/generate.py +51 -42
  17. lightning_sdk/cli/inspect.py +45 -3
  18. lightning_sdk/cli/jobs_menu.py +2 -1
  19. lightning_sdk/cli/list.py +139 -55
  20. lightning_sdk/cli/mmts_menu.py +2 -1
  21. lightning_sdk/cli/run.py +3 -9
  22. lightning_sdk/cli/serve.py +1 -2
  23. lightning_sdk/cli/start.py +2 -2
  24. lightning_sdk/cli/stop.py +5 -3
  25. lightning_sdk/cli/studios_menu.py +24 -1
  26. lightning_sdk/cli/switch.py +2 -2
  27. lightning_sdk/cli/teamspace_menu.py +2 -1
  28. lightning_sdk/cli/upload.py +6 -4
  29. lightning_sdk/helpers.py +20 -0
  30. lightning_sdk/job/job.py +1 -1
  31. lightning_sdk/lightning_cloud/openapi/__init__.py +9 -0
  32. lightning_sdk/lightning_cloud/openapi/api/cluster_service_api.py +105 -0
  33. lightning_sdk/lightning_cloud/openapi/api/data_connection_service_api.py +105 -0
  34. lightning_sdk/lightning_cloud/openapi/api/jobs_service_api.py +226 -0
  35. lightning_sdk/lightning_cloud/openapi/api/lit_logger_service_api.py +4 -4
  36. lightning_sdk/lightning_cloud/openapi/api/lit_registry_service_api.py +7 -3
  37. lightning_sdk/lightning_cloud/openapi/api/projects_service_api.py +1 -5
  38. lightning_sdk/lightning_cloud/openapi/models/__init__.py +9 -0
  39. lightning_sdk/lightning_cloud/openapi/models/agents_id_body.py +105 -1
  40. lightning_sdk/lightning_cloud/openapi/models/deployments_id_body.py +29 -3
  41. lightning_sdk/lightning_cloud/openapi/models/id_reportrestarttimings_body.py +123 -0
  42. lightning_sdk/lightning_cloud/openapi/models/id_visibility_body1.py +149 -0
  43. lightning_sdk/lightning_cloud/openapi/models/model_id_visibility_body.py +27 -1
  44. lightning_sdk/lightning_cloud/openapi/models/project_id_litregistry_body.py +2 -0
  45. lightning_sdk/lightning_cloud/openapi/models/setup.py +149 -0
  46. lightning_sdk/lightning_cloud/openapi/models/v1_assistant.py +105 -1
  47. lightning_sdk/lightning_cloud/openapi/models/v1_cluster_accelerator.py +27 -1
  48. lightning_sdk/lightning_cloud/openapi/models/v1_deployment.py +29 -3
  49. lightning_sdk/lightning_cloud/openapi/models/v1_gcp_data_connection_setup.py +123 -0
  50. lightning_sdk/lightning_cloud/openapi/models/v1_get_cluster_accelerator_demand_response.py +123 -0
  51. lightning_sdk/lightning_cloud/openapi/models/v1_job.py +27 -1
  52. lightning_sdk/lightning_cloud/openapi/models/v1_job_spec.py +27 -1
  53. lightning_sdk/lightning_cloud/openapi/models/v1_lit_registry_artifact.py +27 -1
  54. lightning_sdk/lightning_cloud/openapi/models/v1_lit_registry_project.py +8 -0
  55. lightning_sdk/lightning_cloud/openapi/models/v1_lit_repository.py +27 -1
  56. lightning_sdk/lightning_cloud/openapi/models/v1_report_restart_timings_response.py +97 -0
  57. lightning_sdk/lightning_cloud/openapi/models/v1_restart_timing.py +175 -0
  58. lightning_sdk/lightning_cloud/openapi/models/v1_setup_data_connection_response.py +123 -0
  59. lightning_sdk/lightning_cloud/openapi/models/v1_update_deployment_visibility_response.py +97 -0
  60. lightning_sdk/lightning_cloud/openapi/models/v1_update_metrics_stream_visibility_response.py +27 -1
  61. lightning_sdk/lightning_cloud/openapi/models/v1_update_model_visibility_response.py +27 -1
  62. lightning_sdk/lightning_cloud/openapi/models/v1_user_features.py +79 -1
  63. lightning_sdk/lightning_cloud/openapi/models/v1_validate_deployment_image_request.py +27 -1
  64. lightning_sdk/machine.py +59 -27
  65. lightning_sdk/studio.py +5 -1
  66. lightning_sdk/teamspace.py +25 -0
  67. {lightning_sdk-0.1.55.dist-info → lightning_sdk-0.1.57.dist-info}/METADATA +3 -1
  68. {lightning_sdk-0.1.55.dist-info → lightning_sdk-0.1.57.dist-info}/RECORD +72 -61
  69. {lightning_sdk-0.1.55.dist-info → lightning_sdk-0.1.57.dist-info}/LICENSE +0 -0
  70. {lightning_sdk-0.1.55.dist-info → lightning_sdk-0.1.57.dist-info}/WHEEL +0 -0
  71. {lightning_sdk-0.1.55.dist-info → lightning_sdk-0.1.57.dist-info}/entry_points.txt +0 -0
  72. {lightning_sdk-0.1.55.dist-info → lightning_sdk-0.1.57.dist-info}/top_level.txt +0 -0
@@ -124,11 +124,13 @@ from lightning_sdk.lightning_cloud.openapi.models.id_publications_body import Id
124
124
  from lightning_sdk.lightning_cloud.openapi.models.id_publications_body1 import IdPublicationsBody1
125
125
  from lightning_sdk.lightning_cloud.openapi.models.id_release_body import IdReleaseBody
126
126
  from lightning_sdk.lightning_cloud.openapi.models.id_reportlogsactivity_body import IdReportlogsactivityBody
127
+ from lightning_sdk.lightning_cloud.openapi.models.id_reportrestarttimings_body import IdReportrestarttimingsBody
127
128
  from lightning_sdk.lightning_cloud.openapi.models.id_start_body import IdStartBody
128
129
  from lightning_sdk.lightning_cloud.openapi.models.id_storage_body import IdStorageBody
129
130
  from lightning_sdk.lightning_cloud.openapi.models.id_uploads_body import IdUploadsBody
130
131
  from lightning_sdk.lightning_cloud.openapi.models.id_uploads_body1 import IdUploadsBody1
131
132
  from lightning_sdk.lightning_cloud.openapi.models.id_visibility_body import IdVisibilityBody
133
+ from lightning_sdk.lightning_cloud.openapi.models.id_visibility_body1 import IdVisibilityBody1
132
134
  from lightning_sdk.lightning_cloud.openapi.models.jobs_id_body import JobsIdBody
133
135
  from lightning_sdk.lightning_cloud.openapi.models.jobs_id_body1 import JobsIdBody1
134
136
  from lightning_sdk.lightning_cloud.openapi.models.jobs_id_body2 import JobsIdBody2
@@ -181,6 +183,7 @@ from lightning_sdk.lightning_cloud.openapi.models.servers_server_id_body import
181
183
  from lightning_sdk.lightning_cloud.openapi.models.service_artifact_artifact_kind import ServiceArtifactArtifactKind
182
184
  from lightning_sdk.lightning_cloud.openapi.models.service_health_service_status import ServiceHealthServiceStatus
183
185
  from lightning_sdk.lightning_cloud.openapi.models.serviceexecution_id_body import ServiceexecutionIdBody
186
+ from lightning_sdk.lightning_cloud.openapi.models.setup import Setup
184
187
  from lightning_sdk.lightning_cloud.openapi.models.slurm_jobs_body import SlurmJobsBody
185
188
  from lightning_sdk.lightning_cloud.openapi.models.snowflake_export_body import SnowflakeExportBody
186
189
  from lightning_sdk.lightning_cloud.openapi.models.snowflake_query_body import SnowflakeQueryBody
@@ -439,6 +442,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_gpu_system_metrics import V
439
442
  from lightning_sdk.lightning_cloud.openapi.models.v1_gallery_app import V1GalleryApp
440
443
  from lightning_sdk.lightning_cloud.openapi.models.v1_gallery_component import V1GalleryComponent
441
444
  from lightning_sdk.lightning_cloud.openapi.models.v1_gcp_data_connection import V1GcpDataConnection
445
+ from lightning_sdk.lightning_cloud.openapi.models.v1_gcp_data_connection_setup import V1GcpDataConnectionSetup
442
446
  from lightning_sdk.lightning_cloud.openapi.models.v1_generate_ssh_key_pair_request import V1GenerateSSHKeyPairRequest
443
447
  from lightning_sdk.lightning_cloud.openapi.models.v1_get_affiliate_link_response import V1GetAffiliateLinkResponse
444
448
  from lightning_sdk.lightning_cloud.openapi.models.v1_get_agent_job_env_response import V1GetAgentJobEnvResponse
@@ -446,6 +450,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_get_agent_job_logs_metadata
446
450
  from lightning_sdk.lightning_cloud.openapi.models.v1_get_artifacts_page_response import V1GetArtifactsPageResponse
447
451
  from lightning_sdk.lightning_cloud.openapi.models.v1_get_cloud_space_instance_status_response import V1GetCloudSpaceInstanceStatusResponse
448
452
  from lightning_sdk.lightning_cloud.openapi.models.v1_get_cloud_space_size_response import V1GetCloudSpaceSizeResponse
453
+ from lightning_sdk.lightning_cloud.openapi.models.v1_get_cluster_accelerator_demand_response import V1GetClusterAcceleratorDemandResponse
449
454
  from lightning_sdk.lightning_cloud.openapi.models.v1_get_cluster_credentials_response import V1GetClusterCredentialsResponse
450
455
  from lightning_sdk.lightning_cloud.openapi.models.v1_get_deployment_routing_telemetry_aggregated_response import V1GetDeploymentRoutingTelemetryAggregatedResponse
451
456
  from lightning_sdk.lightning_cloud.openapi.models.v1_get_deployment_routing_telemetry_response import V1GetDeploymentRoutingTelemetryResponse
@@ -701,6 +706,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_refresh_response import V1R
701
706
  from lightning_sdk.lightning_cloud.openapi.models.v1_region_state import V1RegionState
702
707
  from lightning_sdk.lightning_cloud.openapi.models.v1_regional_load_balancer import V1RegionalLoadBalancer
703
708
  from lightning_sdk.lightning_cloud.openapi.models.v1_report_logs_activity_response import V1ReportLogsActivityResponse
709
+ from lightning_sdk.lightning_cloud.openapi.models.v1_report_restart_timings_response import V1ReportRestartTimingsResponse
704
710
  from lightning_sdk.lightning_cloud.openapi.models.v1_request_cluster_access_request import V1RequestClusterAccessRequest
705
711
  from lightning_sdk.lightning_cloud.openapi.models.v1_request_cluster_access_response import V1RequestClusterAccessResponse
706
712
  from lightning_sdk.lightning_cloud.openapi.models.v1_request_verification_code_response import V1RequestVerificationCodeResponse
@@ -710,6 +716,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_resources import V1Resource
710
716
  from lightning_sdk.lightning_cloud.openapi.models.v1_response_choice import V1ResponseChoice
711
717
  from lightning_sdk.lightning_cloud.openapi.models.v1_response_choice_delta import V1ResponseChoiceDelta
712
718
  from lightning_sdk.lightning_cloud.openapi.models.v1_restart_cloud_space_instance_response import V1RestartCloudSpaceInstanceResponse
719
+ from lightning_sdk.lightning_cloud.openapi.models.v1_restart_timing import V1RestartTiming
713
720
  from lightning_sdk.lightning_cloud.openapi.models.v1_restore_deployment_release_response import V1RestoreDeploymentReleaseResponse
714
721
  from lightning_sdk.lightning_cloud.openapi.models.v1_role import V1Role
715
722
  from lightning_sdk.lightning_cloud.openapi.models.v1_rolling_update_strategy import V1RollingUpdateStrategy
@@ -733,6 +740,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_service_artifact import V1S
733
740
  from lightning_sdk.lightning_cloud.openapi.models.v1_service_execution import V1ServiceExecution
734
741
  from lightning_sdk.lightning_cloud.openapi.models.v1_service_health import V1ServiceHealth
735
742
  from lightning_sdk.lightning_cloud.openapi.models.v1_setup_confirmed_ssh_public_key_response import V1SetupConfirmedSSHPublicKeyResponse
743
+ from lightning_sdk.lightning_cloud.openapi.models.v1_setup_data_connection_response import V1SetupDataConnectionResponse
736
744
  from lightning_sdk.lightning_cloud.openapi.models.v1_should_start_syncing_response import V1ShouldStartSyncingResponse
737
745
  from lightning_sdk.lightning_cloud.openapi.models.v1_signed_url import V1SignedUrl
738
746
  from lightning_sdk.lightning_cloud.openapi.models.v1_slurm_node import V1SlurmNode
@@ -769,6 +777,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_update_cloud_space_publicat
769
777
  from lightning_sdk.lightning_cloud.openapi.models.v1_update_cluster_accelerators_request import V1UpdateClusterAcceleratorsRequest
770
778
  from lightning_sdk.lightning_cloud.openapi.models.v1_update_cluster_accelerators_response import V1UpdateClusterAcceleratorsResponse
771
779
  from lightning_sdk.lightning_cloud.openapi.models.v1_update_cluster_availability_request import V1UpdateClusterAvailabilityRequest
780
+ from lightning_sdk.lightning_cloud.openapi.models.v1_update_deployment_visibility_response import V1UpdateDeploymentVisibilityResponse
772
781
  from lightning_sdk.lightning_cloud.openapi.models.v1_update_index_response import V1UpdateIndexResponse
773
782
  from lightning_sdk.lightning_cloud.openapi.models.v1_update_lit_page_response import V1UpdateLitPageResponse
774
783
  from lightning_sdk.lightning_cloud.openapi.models.v1_update_lit_repository_response import V1UpdateLitRepositoryResponse
@@ -1529,6 +1529,111 @@ class ClusterServiceApi(object):
1529
1529
  _request_timeout=params.get('_request_timeout'),
1530
1530
  collection_formats=collection_formats)
1531
1531
 
1532
+ def cluster_service_get_cluster_accelerator_demand(self, cluster_id: 'str', id: 'str', **kwargs) -> 'V1GetClusterAcceleratorDemandResponse': # noqa: E501
1533
+ """cluster_service_get_cluster_accelerator_demand # noqa: E501
1534
+
1535
+ This method makes a synchronous HTTP request by default. To make an
1536
+ asynchronous HTTP request, please pass async_req=True
1537
+ >>> thread = api.cluster_service_get_cluster_accelerator_demand(cluster_id, id, async_req=True)
1538
+ >>> result = thread.get()
1539
+
1540
+ :param async_req bool
1541
+ :param str cluster_id: (required)
1542
+ :param str id: (required)
1543
+ :param bool spot:
1544
+ :return: V1GetClusterAcceleratorDemandResponse
1545
+ If the method is called asynchronously,
1546
+ returns the request thread.
1547
+ """
1548
+ kwargs['_return_http_data_only'] = True
1549
+ if kwargs.get('async_req'):
1550
+ return self.cluster_service_get_cluster_accelerator_demand_with_http_info(cluster_id, id, **kwargs) # noqa: E501
1551
+ else:
1552
+ (data) = self.cluster_service_get_cluster_accelerator_demand_with_http_info(cluster_id, id, **kwargs) # noqa: E501
1553
+ return data
1554
+
1555
+ def cluster_service_get_cluster_accelerator_demand_with_http_info(self, cluster_id: 'str', id: 'str', **kwargs) -> 'V1GetClusterAcceleratorDemandResponse': # noqa: E501
1556
+ """cluster_service_get_cluster_accelerator_demand # noqa: E501
1557
+
1558
+ This method makes a synchronous HTTP request by default. To make an
1559
+ asynchronous HTTP request, please pass async_req=True
1560
+ >>> thread = api.cluster_service_get_cluster_accelerator_demand_with_http_info(cluster_id, id, async_req=True)
1561
+ >>> result = thread.get()
1562
+
1563
+ :param async_req bool
1564
+ :param str cluster_id: (required)
1565
+ :param str id: (required)
1566
+ :param bool spot:
1567
+ :return: V1GetClusterAcceleratorDemandResponse
1568
+ If the method is called asynchronously,
1569
+ returns the request thread.
1570
+ """
1571
+
1572
+ all_params = ['cluster_id', 'id', 'spot'] # noqa: E501
1573
+ all_params.append('async_req')
1574
+ all_params.append('_return_http_data_only')
1575
+ all_params.append('_preload_content')
1576
+ all_params.append('_request_timeout')
1577
+
1578
+ params = locals()
1579
+ for key, val in six.iteritems(params['kwargs']):
1580
+ if key not in all_params:
1581
+ raise TypeError(
1582
+ "Got an unexpected keyword argument '%s'"
1583
+ " to method cluster_service_get_cluster_accelerator_demand" % key
1584
+ )
1585
+ params[key] = val
1586
+ del params['kwargs']
1587
+ # verify the required parameter 'cluster_id' is set
1588
+ if ('cluster_id' not in params or
1589
+ params['cluster_id'] is None):
1590
+ raise ValueError("Missing the required parameter `cluster_id` when calling `cluster_service_get_cluster_accelerator_demand`") # noqa: E501
1591
+ # verify the required parameter 'id' is set
1592
+ if ('id' not in params or
1593
+ params['id'] is None):
1594
+ raise ValueError("Missing the required parameter `id` when calling `cluster_service_get_cluster_accelerator_demand`") # noqa: E501
1595
+
1596
+ collection_formats = {}
1597
+
1598
+ path_params = {}
1599
+ if 'cluster_id' in params:
1600
+ path_params['clusterId'] = params['cluster_id'] # noqa: E501
1601
+ if 'id' in params:
1602
+ path_params['id'] = params['id'] # noqa: E501
1603
+
1604
+ query_params = []
1605
+ if 'spot' in params:
1606
+ query_params.append(('spot', params['spot'])) # noqa: E501
1607
+
1608
+ header_params = {}
1609
+
1610
+ form_params = []
1611
+ local_var_files = {}
1612
+
1613
+ body_params = None
1614
+ # HTTP header `Accept`
1615
+ header_params['Accept'] = self.api_client.select_header_accept(
1616
+ ['application/json']) # noqa: E501
1617
+
1618
+ # Authentication setting
1619
+ auth_settings = [] # noqa: E501
1620
+
1621
+ return self.api_client.call_api(
1622
+ '/v1/core/clusters/{clusterId}/accelerator/{id}/demand', 'GET',
1623
+ path_params,
1624
+ query_params,
1625
+ header_params,
1626
+ body=body_params,
1627
+ post_params=form_params,
1628
+ files=local_var_files,
1629
+ response_type='V1GetClusterAcceleratorDemandResponse', # noqa: E501
1630
+ auth_settings=auth_settings,
1631
+ async_req=params.get('async_req'),
1632
+ _return_http_data_only=params.get('_return_http_data_only'),
1633
+ _preload_content=params.get('_preload_content', True),
1634
+ _request_timeout=params.get('_request_timeout'),
1635
+ collection_formats=collection_formats)
1636
+
1532
1637
  def cluster_service_get_cluster_availability(self, **kwargs) -> 'V1ClusterAvailability': # noqa: E501
1533
1638
  """cluster_service_get_cluster_availability # noqa: E501
1534
1639
 
@@ -920,6 +920,111 @@ class DataConnectionServiceApi(object):
920
920
  _request_timeout=params.get('_request_timeout'),
921
921
  collection_formats=collection_formats)
922
922
 
923
+ def data_connection_service_setup_data_connection(self, body: 'Setup', project_id: 'str', **kwargs) -> 'V1SetupDataConnectionResponse': # noqa: E501
924
+ """data_connection_service_setup_data_connection # noqa: E501
925
+
926
+ This method makes a synchronous HTTP request by default. To make an
927
+ asynchronous HTTP request, please pass async_req=True
928
+ >>> thread = api.data_connection_service_setup_data_connection(body, project_id, async_req=True)
929
+ >>> result = thread.get()
930
+
931
+ :param async_req bool
932
+ :param Setup body: (required)
933
+ :param str project_id: (required)
934
+ :return: V1SetupDataConnectionResponse
935
+ If the method is called asynchronously,
936
+ returns the request thread.
937
+ """
938
+ kwargs['_return_http_data_only'] = True
939
+ if kwargs.get('async_req'):
940
+ return self.data_connection_service_setup_data_connection_with_http_info(body, project_id, **kwargs) # noqa: E501
941
+ else:
942
+ (data) = self.data_connection_service_setup_data_connection_with_http_info(body, project_id, **kwargs) # noqa: E501
943
+ return data
944
+
945
+ def data_connection_service_setup_data_connection_with_http_info(self, body: 'Setup', project_id: 'str', **kwargs) -> 'V1SetupDataConnectionResponse': # noqa: E501
946
+ """data_connection_service_setup_data_connection # noqa: E501
947
+
948
+ This method makes a synchronous HTTP request by default. To make an
949
+ asynchronous HTTP request, please pass async_req=True
950
+ >>> thread = api.data_connection_service_setup_data_connection_with_http_info(body, project_id, async_req=True)
951
+ >>> result = thread.get()
952
+
953
+ :param async_req bool
954
+ :param Setup body: (required)
955
+ :param str project_id: (required)
956
+ :return: V1SetupDataConnectionResponse
957
+ If the method is called asynchronously,
958
+ returns the request thread.
959
+ """
960
+
961
+ all_params = ['body', 'project_id'] # noqa: E501
962
+ all_params.append('async_req')
963
+ all_params.append('_return_http_data_only')
964
+ all_params.append('_preload_content')
965
+ all_params.append('_request_timeout')
966
+
967
+ params = locals()
968
+ for key, val in six.iteritems(params['kwargs']):
969
+ if key not in all_params:
970
+ raise TypeError(
971
+ "Got an unexpected keyword argument '%s'"
972
+ " to method data_connection_service_setup_data_connection" % key
973
+ )
974
+ params[key] = val
975
+ del params['kwargs']
976
+ # verify the required parameter 'body' is set
977
+ if ('body' not in params or
978
+ params['body'] is None):
979
+ raise ValueError("Missing the required parameter `body` when calling `data_connection_service_setup_data_connection`") # noqa: E501
980
+ # verify the required parameter 'project_id' is set
981
+ if ('project_id' not in params or
982
+ params['project_id'] is None):
983
+ raise ValueError("Missing the required parameter `project_id` when calling `data_connection_service_setup_data_connection`") # noqa: E501
984
+
985
+ collection_formats = {}
986
+
987
+ path_params = {}
988
+ if 'project_id' in params:
989
+ path_params['projectId'] = params['project_id'] # noqa: E501
990
+
991
+ query_params = []
992
+
993
+ header_params = {}
994
+
995
+ form_params = []
996
+ local_var_files = {}
997
+
998
+ body_params = None
999
+ if 'body' in params:
1000
+ body_params = params['body']
1001
+ # HTTP header `Accept`
1002
+ header_params['Accept'] = self.api_client.select_header_accept(
1003
+ ['application/json']) # noqa: E501
1004
+
1005
+ # HTTP header `Content-Type`
1006
+ header_params['Content-Type'] = self.api_client.select_header_content_type( # noqa: E501
1007
+ ['application/json']) # noqa: E501
1008
+
1009
+ # Authentication setting
1010
+ auth_settings = [] # noqa: E501
1011
+
1012
+ return self.api_client.call_api(
1013
+ '/v1/projects/{projectId}/data-connections/setup', 'POST',
1014
+ path_params,
1015
+ query_params,
1016
+ header_params,
1017
+ body=body_params,
1018
+ post_params=form_params,
1019
+ files=local_var_files,
1020
+ response_type='V1SetupDataConnectionResponse', # noqa: E501
1021
+ auth_settings=auth_settings,
1022
+ async_req=params.get('async_req'),
1023
+ _return_http_data_only=params.get('_return_http_data_only'),
1024
+ _preload_content=params.get('_preload_content', True),
1025
+ _request_timeout=params.get('_request_timeout'),
1026
+ collection_formats=collection_formats)
1027
+
923
1028
  def data_connection_service_update_data_connection(self, body: 'Update', project_id: 'str', id: 'str', **kwargs) -> 'V1DataConnection': # noqa: E501
924
1029
  """data_connection_service_update_data_connection # noqa: E501
925
1030
 
@@ -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
 
@@ -3390,6 +3503,119 @@ class JobsServiceApi(object):
3390
3503
  _request_timeout=params.get('_request_timeout'),
3391
3504
  collection_formats=collection_formats)
3392
3505
 
3506
+ def jobs_service_update_deployment_visibility(self, body: 'IdVisibilityBody', project_id: 'str', id: 'str', **kwargs) -> 'V1UpdateDeploymentVisibilityResponse': # noqa: E501
3507
+ """UpdateDeploymentVisibility updates the deployment visibility, mainly switch between public and private # noqa: E501
3508
+
3509
+ This method makes a synchronous HTTP request by default. To make an
3510
+ asynchronous HTTP request, please pass async_req=True
3511
+ >>> thread = api.jobs_service_update_deployment_visibility(body, project_id, id, async_req=True)
3512
+ >>> result = thread.get()
3513
+
3514
+ :param async_req bool
3515
+ :param IdVisibilityBody body: (required)
3516
+ :param str project_id: (required)
3517
+ :param str id: (required)
3518
+ :return: V1UpdateDeploymentVisibilityResponse
3519
+ If the method is called asynchronously,
3520
+ returns the request thread.
3521
+ """
3522
+ kwargs['_return_http_data_only'] = True
3523
+ if kwargs.get('async_req'):
3524
+ return self.jobs_service_update_deployment_visibility_with_http_info(body, project_id, id, **kwargs) # noqa: E501
3525
+ else:
3526
+ (data) = self.jobs_service_update_deployment_visibility_with_http_info(body, project_id, id, **kwargs) # noqa: E501
3527
+ return data
3528
+
3529
+ def jobs_service_update_deployment_visibility_with_http_info(self, body: 'IdVisibilityBody', project_id: 'str', id: 'str', **kwargs) -> 'V1UpdateDeploymentVisibilityResponse': # noqa: E501
3530
+ """UpdateDeploymentVisibility updates the deployment visibility, mainly switch between public and private # noqa: E501
3531
+
3532
+ This method makes a synchronous HTTP request by default. To make an
3533
+ asynchronous HTTP request, please pass async_req=True
3534
+ >>> thread = api.jobs_service_update_deployment_visibility_with_http_info(body, project_id, id, async_req=True)
3535
+ >>> result = thread.get()
3536
+
3537
+ :param async_req bool
3538
+ :param IdVisibilityBody body: (required)
3539
+ :param str project_id: (required)
3540
+ :param str id: (required)
3541
+ :return: V1UpdateDeploymentVisibilityResponse
3542
+ If the method is called asynchronously,
3543
+ returns the request thread.
3544
+ """
3545
+
3546
+ all_params = ['body', 'project_id', 'id'] # noqa: E501
3547
+ all_params.append('async_req')
3548
+ all_params.append('_return_http_data_only')
3549
+ all_params.append('_preload_content')
3550
+ all_params.append('_request_timeout')
3551
+
3552
+ params = locals()
3553
+ for key, val in six.iteritems(params['kwargs']):
3554
+ if key not in all_params:
3555
+ raise TypeError(
3556
+ "Got an unexpected keyword argument '%s'"
3557
+ " to method jobs_service_update_deployment_visibility" % key
3558
+ )
3559
+ params[key] = val
3560
+ del params['kwargs']
3561
+ # verify the required parameter 'body' is set
3562
+ if ('body' not in params or
3563
+ params['body'] is None):
3564
+ raise ValueError("Missing the required parameter `body` when calling `jobs_service_update_deployment_visibility`") # noqa: E501
3565
+ # verify the required parameter 'project_id' is set
3566
+ if ('project_id' not in params or
3567
+ params['project_id'] is None):
3568
+ raise ValueError("Missing the required parameter `project_id` when calling `jobs_service_update_deployment_visibility`") # noqa: E501
3569
+ # verify the required parameter 'id' is set
3570
+ if ('id' not in params or
3571
+ params['id'] is None):
3572
+ raise ValueError("Missing the required parameter `id` when calling `jobs_service_update_deployment_visibility`") # noqa: E501
3573
+
3574
+ collection_formats = {}
3575
+
3576
+ path_params = {}
3577
+ if 'project_id' in params:
3578
+ path_params['projectId'] = params['project_id'] # noqa: E501
3579
+ if 'id' in params:
3580
+ path_params['id'] = params['id'] # noqa: E501
3581
+
3582
+ query_params = []
3583
+
3584
+ header_params = {}
3585
+
3586
+ form_params = []
3587
+ local_var_files = {}
3588
+
3589
+ body_params = None
3590
+ if 'body' in params:
3591
+ body_params = params['body']
3592
+ # HTTP header `Accept`
3593
+ header_params['Accept'] = self.api_client.select_header_accept(
3594
+ ['application/json']) # noqa: E501
3595
+
3596
+ # HTTP header `Content-Type`
3597
+ header_params['Content-Type'] = self.api_client.select_header_content_type( # noqa: E501
3598
+ ['application/json']) # noqa: E501
3599
+
3600
+ # Authentication setting
3601
+ auth_settings = [] # noqa: E501
3602
+
3603
+ return self.api_client.call_api(
3604
+ '/v1/projects/{projectId}/deployments/{id}/visibility', 'PUT',
3605
+ path_params,
3606
+ query_params,
3607
+ header_params,
3608
+ body=body_params,
3609
+ post_params=form_params,
3610
+ files=local_var_files,
3611
+ response_type='V1UpdateDeploymentVisibilityResponse', # noqa: E501
3612
+ auth_settings=auth_settings,
3613
+ async_req=params.get('async_req'),
3614
+ _return_http_data_only=params.get('_return_http_data_only'),
3615
+ _preload_content=params.get('_preload_content', True),
3616
+ _request_timeout=params.get('_request_timeout'),
3617
+ collection_formats=collection_formats)
3618
+
3393
3619
  def jobs_service_update_job(self, body: 'JobsIdBody1', project_id: 'str', id: 'str', **kwargs) -> 'V1Job': # noqa: E501
3394
3620
  """jobs_service_update_job # noqa: E501
3395
3621
 
@@ -1534,7 +1534,7 @@ class LitLoggerServiceApi(object):
1534
1534
  _request_timeout=params.get('_request_timeout'),
1535
1535
  collection_formats=collection_formats)
1536
1536
 
1537
- def lit_logger_service_update_metrics_stream_visibility(self, body: 'IdVisibilityBody', project_id: 'str', id: 'str', **kwargs) -> 'V1UpdateMetricsStreamVisibilityResponse': # noqa: E501
1537
+ def lit_logger_service_update_metrics_stream_visibility(self, body: 'IdVisibilityBody1', project_id: 'str', id: 'str', **kwargs) -> 'V1UpdateMetricsStreamVisibilityResponse': # noqa: E501
1538
1538
  """lit_logger_service_update_metrics_stream_visibility # noqa: E501
1539
1539
 
1540
1540
  This method makes a synchronous HTTP request by default. To make an
@@ -1543,7 +1543,7 @@ class LitLoggerServiceApi(object):
1543
1543
  >>> result = thread.get()
1544
1544
 
1545
1545
  :param async_req bool
1546
- :param IdVisibilityBody body: (required)
1546
+ :param IdVisibilityBody1 body: (required)
1547
1547
  :param str project_id: (required)
1548
1548
  :param str id: (required)
1549
1549
  :return: V1UpdateMetricsStreamVisibilityResponse
@@ -1557,7 +1557,7 @@ class LitLoggerServiceApi(object):
1557
1557
  (data) = self.lit_logger_service_update_metrics_stream_visibility_with_http_info(body, project_id, id, **kwargs) # noqa: E501
1558
1558
  return data
1559
1559
 
1560
- def lit_logger_service_update_metrics_stream_visibility_with_http_info(self, body: 'IdVisibilityBody', project_id: 'str', id: 'str', **kwargs) -> 'V1UpdateMetricsStreamVisibilityResponse': # noqa: E501
1560
+ def lit_logger_service_update_metrics_stream_visibility_with_http_info(self, body: 'IdVisibilityBody1', project_id: 'str', id: 'str', **kwargs) -> 'V1UpdateMetricsStreamVisibilityResponse': # noqa: E501
1561
1561
  """lit_logger_service_update_metrics_stream_visibility # noqa: E501
1562
1562
 
1563
1563
  This method makes a synchronous HTTP request by default. To make an
@@ -1566,7 +1566,7 @@ class LitLoggerServiceApi(object):
1566
1566
  >>> result = thread.get()
1567
1567
 
1568
1568
  :param async_req bool
1569
- :param IdVisibilityBody body: (required)
1569
+ :param IdVisibilityBody1 body: (required)
1570
1570
  :param str project_id: (required)
1571
1571
  :param str id: (required)
1572
1572
  :return: V1UpdateMetricsStreamVisibilityResponse
@@ -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.
@@ -563,6 +563,7 @@ class LitRegistryServiceApi(object):
563
563
  :param async_req bool
564
564
  :param str project_id: (required)
565
565
  :param str lit_repo_name: (required)
566
+ :param bool latest_only:
566
567
  :return: V1ListLitRegistryRepositoryImageArtifactVersionsResponse
567
568
  If the method is called asynchronously,
568
569
  returns the request thread.
@@ -585,12 +586,13 @@ class LitRegistryServiceApi(object):
585
586
  :param async_req bool
586
587
  :param str project_id: (required)
587
588
  :param str lit_repo_name: (required)
589
+ :param bool latest_only:
588
590
  :return: V1ListLitRegistryRepositoryImageArtifactVersionsResponse
589
591
  If the method is called asynchronously,
590
592
  returns the request thread.
591
593
  """
592
594
 
593
- all_params = ['project_id', 'lit_repo_name'] # noqa: E501
595
+ all_params = ['project_id', 'lit_repo_name', 'latest_only'] # noqa: E501
594
596
  all_params.append('async_req')
595
597
  all_params.append('_return_http_data_only')
596
598
  all_params.append('_preload_content')
@@ -623,6 +625,8 @@ class LitRegistryServiceApi(object):
623
625
  path_params['litRepoName'] = params['lit_repo_name'] # noqa: E501
624
626
 
625
627
  query_params = []
628
+ if 'latest_only' in params:
629
+ query_params.append(('latestOnly', params['latest_only'])) # noqa: E501
626
630
 
627
631
  header_params = {}
628
632
 
@@ -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: