lightning-sdk 0.1.41__py3-none-any.whl → 0.1.43__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 (77) hide show
  1. lightning_sdk/__init__.py +1 -1
  2. lightning_sdk/ai_hub.py +8 -3
  3. lightning_sdk/api/ai_hub_api.py +3 -3
  4. lightning_sdk/api/deployment_api.py +6 -6
  5. lightning_sdk/api/job_api.py +32 -6
  6. lightning_sdk/api/mmt_api.py +59 -19
  7. lightning_sdk/api/studio_api.py +37 -19
  8. lightning_sdk/api/teamspace_api.py +34 -29
  9. lightning_sdk/api/utils.py +46 -34
  10. lightning_sdk/cli/ai_hub.py +3 -3
  11. lightning_sdk/cli/entrypoint.py +3 -1
  12. lightning_sdk/cli/run.py +122 -12
  13. lightning_sdk/cli/serve.py +218 -0
  14. lightning_sdk/deployment/deployment.py +18 -12
  15. lightning_sdk/job/base.py +118 -24
  16. lightning_sdk/job/job.py +98 -9
  17. lightning_sdk/job/v1.py +75 -18
  18. lightning_sdk/job/v2.py +51 -15
  19. lightning_sdk/job/work.py +36 -7
  20. lightning_sdk/lightning_cloud/openapi/__init__.py +12 -0
  21. lightning_sdk/lightning_cloud/openapi/api/jobs_service_api.py +215 -5
  22. lightning_sdk/lightning_cloud/openapi/api/lit_logger_service_api.py +218 -0
  23. lightning_sdk/lightning_cloud/openapi/api/models_store_api.py +226 -0
  24. lightning_sdk/lightning_cloud/openapi/api/snowflake_service_api.py +21 -1
  25. lightning_sdk/lightning_cloud/openapi/models/__init__.py +12 -0
  26. lightning_sdk/lightning_cloud/openapi/models/deploymenttemplates_id_body.py +27 -1
  27. lightning_sdk/lightning_cloud/openapi/models/id_visibility_body.py +123 -0
  28. lightning_sdk/lightning_cloud/openapi/models/model_id_versions_body.py +29 -3
  29. lightning_sdk/lightning_cloud/openapi/models/project_id_multimachinejobs_body.py +27 -1
  30. lightning_sdk/lightning_cloud/openapi/models/project_id_snowflake_body.py +15 -67
  31. lightning_sdk/lightning_cloud/openapi/models/query_query_id_body.py +17 -69
  32. lightning_sdk/lightning_cloud/openapi/models/snowflake_export_body.py +29 -81
  33. lightning_sdk/lightning_cloud/openapi/models/snowflake_query_body.py +17 -69
  34. lightning_sdk/lightning_cloud/openapi/models/v1_get_model_file_url_response.py +27 -1
  35. lightning_sdk/lightning_cloud/openapi/models/v1_get_model_files_response.py +17 -17
  36. lightning_sdk/lightning_cloud/openapi/models/v1_get_model_files_url_response.py +149 -0
  37. lightning_sdk/lightning_cloud/openapi/models/v1_get_project_balance_response.py +27 -1
  38. lightning_sdk/lightning_cloud/openapi/models/v1_list_multi_machine_job_events_response.py +123 -0
  39. lightning_sdk/lightning_cloud/openapi/models/v1_metrics_stream.py +27 -1
  40. lightning_sdk/lightning_cloud/openapi/models/v1_model_file.py +175 -0
  41. lightning_sdk/lightning_cloud/openapi/models/v1_multi_machine_job.py +27 -1
  42. lightning_sdk/lightning_cloud/openapi/models/v1_multi_machine_job_event.py +331 -0
  43. lightning_sdk/lightning_cloud/openapi/models/v1_multi_machine_job_event_type.py +104 -0
  44. lightning_sdk/lightning_cloud/openapi/models/v1_multi_machine_job_fault_tolerance.py +149 -0
  45. lightning_sdk/lightning_cloud/openapi/models/v1_multi_machine_job_fault_tolerance_strategy.py +105 -0
  46. lightning_sdk/lightning_cloud/openapi/models/v1_multi_machine_job_status.py +27 -1
  47. lightning_sdk/lightning_cloud/openapi/models/v1_rule_resource.py +1 -0
  48. lightning_sdk/lightning_cloud/openapi/models/v1_snowflake_data_connection.py +29 -81
  49. lightning_sdk/lightning_cloud/openapi/models/v1_system_metrics.py +29 -3
  50. lightning_sdk/lightning_cloud/openapi/models/v1_trainium_system_metrics.py +175 -0
  51. lightning_sdk/lightning_cloud/openapi/models/v1_update_metrics_stream_visibility_response.py +97 -0
  52. lightning_sdk/lightning_cloud/openapi/models/v1_user_features.py +27 -53
  53. lightning_sdk/lightning_cloud/openapi/models/v1_validate_deployment_image_request.py +149 -0
  54. lightning_sdk/lightning_cloud/openapi/models/v1_validate_deployment_image_response.py +97 -0
  55. lightning_sdk/lightning_cloud/rest_client.py +2 -0
  56. lightning_sdk/mmt/__init__.py +4 -0
  57. lightning_sdk/mmt/base.py +278 -0
  58. lightning_sdk/mmt/mmt.py +267 -0
  59. lightning_sdk/mmt/v1.py +181 -0
  60. lightning_sdk/mmt/v2.py +188 -0
  61. lightning_sdk/plugin.py +43 -16
  62. lightning_sdk/services/file_endpoint.py +11 -5
  63. lightning_sdk/studio.py +16 -9
  64. lightning_sdk/teamspace.py +21 -8
  65. lightning_sdk/utils/resolve.py +18 -0
  66. {lightning_sdk-0.1.41.dist-info → lightning_sdk-0.1.43.dist-info}/METADATA +4 -1
  67. {lightning_sdk-0.1.41.dist-info → lightning_sdk-0.1.43.dist-info}/RECORD +71 -59
  68. {lightning_sdk-0.1.41.dist-info → lightning_sdk-0.1.43.dist-info}/entry_points.txt +0 -1
  69. lightning_sdk/_mmt/__init__.py +0 -3
  70. lightning_sdk/_mmt/base.py +0 -180
  71. lightning_sdk/_mmt/mmt.py +0 -161
  72. lightning_sdk/_mmt/v1.py +0 -69
  73. lightning_sdk/_mmt/v2.py +0 -141
  74. lightning_sdk/cli/mmt.py +0 -137
  75. {lightning_sdk-0.1.41.dist-info → lightning_sdk-0.1.43.dist-info}/LICENSE +0 -0
  76. {lightning_sdk-0.1.41.dist-info → lightning_sdk-0.1.43.dist-info}/WHEEL +0 -0
  77. {lightning_sdk-0.1.41.dist-info → lightning_sdk-0.1.43.dist-info}/top_level.txt +0 -0
@@ -145,7 +145,7 @@ class JobsServiceApi(object):
145
145
  collection_formats=collection_formats)
146
146
 
147
147
  def jobs_service_create_deployment(self, body: 'CreateDeploymentRequestDefinesASpecForTheJobThatAllowsForAutoscalingJobs', project_id: 'str', **kwargs) -> 'V1Deployment': # noqa: E501
148
- """Job deployments # noqa: E501
148
+ """jobs_service_create_deployment # noqa: E501
149
149
 
150
150
  This method makes a synchronous HTTP request by default. To make an
151
151
  asynchronous HTTP request, please pass async_req=True
@@ -167,7 +167,7 @@ class JobsServiceApi(object):
167
167
  return data
168
168
 
169
169
  def jobs_service_create_deployment_with_http_info(self, body: 'CreateDeploymentRequestDefinesASpecForTheJobThatAllowsForAutoscalingJobs', project_id: 'str', **kwargs) -> 'V1Deployment': # noqa: E501
170
- """Job deployments # noqa: E501
170
+ """jobs_service_create_deployment # noqa: E501
171
171
 
172
172
  This method makes a synchronous HTTP request by default. To make an
173
173
  asynchronous HTTP request, please pass async_req=True
@@ -2600,7 +2600,8 @@ class JobsServiceApi(object):
2600
2600
  :param str cloudspace_id:
2601
2601
  :param str user_id:
2602
2602
  :param str deployment_id:
2603
- :param bool standalone: Whether to list standalone jobs, not part of a deployment.
2603
+ :param str multi_machine_job_id:
2604
+ :param bool standalone: Whether to list standalone jobs, not part of a deployment or mmt.
2604
2605
  :param str page_token:
2605
2606
  :param int limit:
2606
2607
  :param str state:
@@ -2628,7 +2629,8 @@ class JobsServiceApi(object):
2628
2629
  :param str cloudspace_id:
2629
2630
  :param str user_id:
2630
2631
  :param str deployment_id:
2631
- :param bool standalone: Whether to list standalone jobs, not part of a deployment.
2632
+ :param str multi_machine_job_id:
2633
+ :param bool standalone: Whether to list standalone jobs, not part of a deployment or mmt.
2632
2634
  :param str page_token:
2633
2635
  :param int limit:
2634
2636
  :param str state:
@@ -2637,7 +2639,7 @@ class JobsServiceApi(object):
2637
2639
  returns the request thread.
2638
2640
  """
2639
2641
 
2640
- all_params = ['project_id', 'cloudspace_id', 'user_id', 'deployment_id', 'standalone', 'page_token', 'limit', 'state'] # noqa: E501
2642
+ all_params = ['project_id', 'cloudspace_id', 'user_id', 'deployment_id', 'multi_machine_job_id', 'standalone', 'page_token', 'limit', 'state'] # noqa: E501
2641
2643
  all_params.append('async_req')
2642
2644
  all_params.append('_return_http_data_only')
2643
2645
  all_params.append('_preload_content')
@@ -2670,6 +2672,8 @@ class JobsServiceApi(object):
2670
2672
  query_params.append(('userId', params['user_id'])) # noqa: E501
2671
2673
  if 'deployment_id' in params:
2672
2674
  query_params.append(('deploymentId', params['deployment_id'])) # noqa: E501
2675
+ if 'multi_machine_job_id' in params:
2676
+ query_params.append(('multiMachineJobId', params['multi_machine_job_id'])) # noqa: E501
2673
2677
  if 'standalone' in params:
2674
2678
  query_params.append(('standalone', params['standalone'])) # noqa: E501
2675
2679
  if 'page_token' in params:
@@ -2708,6 +2712,115 @@ class JobsServiceApi(object):
2708
2712
  _request_timeout=params.get('_request_timeout'),
2709
2713
  collection_formats=collection_formats)
2710
2714
 
2715
+ def jobs_service_list_multi_machine_job_events(self, project_id: 'str', id: 'str', **kwargs) -> 'V1ListMultiMachineJobEventsResponse': # noqa: E501
2716
+ """jobs_service_list_multi_machine_job_events # noqa: E501
2717
+
2718
+ This method makes a synchronous HTTP request by default. To make an
2719
+ asynchronous HTTP request, please pass async_req=True
2720
+ >>> thread = api.jobs_service_list_multi_machine_job_events(project_id, id, async_req=True)
2721
+ >>> result = thread.get()
2722
+
2723
+ :param async_req bool
2724
+ :param str project_id: (required)
2725
+ :param str id: (required)
2726
+ :param str release_id:
2727
+ :param str limit:
2728
+ :return: V1ListMultiMachineJobEventsResponse
2729
+ If the method is called asynchronously,
2730
+ returns the request thread.
2731
+ """
2732
+ kwargs['_return_http_data_only'] = True
2733
+ if kwargs.get('async_req'):
2734
+ return self.jobs_service_list_multi_machine_job_events_with_http_info(project_id, id, **kwargs) # noqa: E501
2735
+ else:
2736
+ (data) = self.jobs_service_list_multi_machine_job_events_with_http_info(project_id, id, **kwargs) # noqa: E501
2737
+ return data
2738
+
2739
+ def jobs_service_list_multi_machine_job_events_with_http_info(self, project_id: 'str', id: 'str', **kwargs) -> 'V1ListMultiMachineJobEventsResponse': # noqa: E501
2740
+ """jobs_service_list_multi_machine_job_events # noqa: E501
2741
+
2742
+ This method makes a synchronous HTTP request by default. To make an
2743
+ asynchronous HTTP request, please pass async_req=True
2744
+ >>> thread = api.jobs_service_list_multi_machine_job_events_with_http_info(project_id, id, async_req=True)
2745
+ >>> result = thread.get()
2746
+
2747
+ :param async_req bool
2748
+ :param str project_id: (required)
2749
+ :param str id: (required)
2750
+ :param str release_id:
2751
+ :param str limit:
2752
+ :return: V1ListMultiMachineJobEventsResponse
2753
+ If the method is called asynchronously,
2754
+ returns the request thread.
2755
+ """
2756
+
2757
+ all_params = ['project_id', 'id', 'release_id', 'limit'] # noqa: E501
2758
+ all_params.append('async_req')
2759
+ all_params.append('_return_http_data_only')
2760
+ all_params.append('_preload_content')
2761
+ all_params.append('_request_timeout')
2762
+
2763
+ params = locals()
2764
+ for key, val in six.iteritems(params['kwargs']):
2765
+ if key not in all_params:
2766
+ raise TypeError(
2767
+ "Got an unexpected keyword argument '%s'"
2768
+ " to method jobs_service_list_multi_machine_job_events" % key
2769
+ )
2770
+ params[key] = val
2771
+ del params['kwargs']
2772
+ # verify the required parameter 'project_id' is set
2773
+ if ('project_id' not in params or
2774
+ params['project_id'] is None):
2775
+ raise ValueError("Missing the required parameter `project_id` when calling `jobs_service_list_multi_machine_job_events`") # noqa: E501
2776
+ # verify the required parameter 'id' is set
2777
+ if ('id' not in params or
2778
+ params['id'] is None):
2779
+ raise ValueError("Missing the required parameter `id` when calling `jobs_service_list_multi_machine_job_events`") # noqa: E501
2780
+
2781
+ collection_formats = {}
2782
+
2783
+ path_params = {}
2784
+ if 'project_id' in params:
2785
+ path_params['projectId'] = params['project_id'] # noqa: E501
2786
+ if 'id' in params:
2787
+ path_params['id'] = params['id'] # noqa: E501
2788
+
2789
+ query_params = []
2790
+ if 'release_id' in params:
2791
+ query_params.append(('releaseId', params['release_id'])) # noqa: E501
2792
+ if 'limit' in params:
2793
+ query_params.append(('limit', params['limit'])) # noqa: E501
2794
+
2795
+ header_params = {}
2796
+
2797
+ form_params = []
2798
+ local_var_files = {}
2799
+
2800
+ body_params = None
2801
+ # HTTP header `Accept`
2802
+ header_params['Accept'] = self.api_client.select_header_accept(
2803
+ ['application/json']) # noqa: E501
2804
+
2805
+ # Authentication setting
2806
+ auth_settings = [] # noqa: E501
2807
+
2808
+ return self.api_client.call_api(
2809
+ '/v1/projects/{projectId}/multi-machine-jobs/{id}/events', 'GET',
2810
+ path_params,
2811
+ query_params,
2812
+ header_params,
2813
+ body=body_params,
2814
+ post_params=form_params,
2815
+ files=local_var_files,
2816
+ response_type='V1ListMultiMachineJobEventsResponse', # noqa: E501
2817
+ auth_settings=auth_settings,
2818
+ async_req=params.get('async_req'),
2819
+ _return_http_data_only=params.get('_return_http_data_only'),
2820
+ _preload_content=params.get('_preload_content', True),
2821
+ _request_timeout=params.get('_request_timeout'),
2822
+ collection_formats=collection_formats)
2823
+
2711
2824
  def jobs_service_list_multi_machine_jobs(self, project_id: 'str', **kwargs) -> 'V1ListMultiMachineJobsResponse': # noqa: E501
2712
2825
  """jobs_service_list_multi_machine_jobs # noqa: E501
2713
2826
 
@@ -3611,3 +3724,100 @@ class JobsServiceApi(object):
3611
3724
  _preload_content=params.get('_preload_content', True),
3612
3725
  _request_timeout=params.get('_request_timeout'),
3613
3726
  collection_formats=collection_formats)
3727
+
3728
+ def jobs_service_validate_deployment_image(self, body: 'V1ValidateDeploymentImageRequest', **kwargs) -> 'V1ValidateDeploymentImageResponse': # noqa: E501
3729
+ """Job deployments # noqa: E501
3730
+
3731
+ This method makes a synchronous HTTP request by default. To make an
3732
+ asynchronous HTTP request, please pass async_req=True
3733
+ >>> thread = api.jobs_service_validate_deployment_image(body, async_req=True)
3734
+ >>> result = thread.get()
3735
+
3736
+ :param async_req bool
3737
+ :param V1ValidateDeploymentImageRequest body: (required)
3738
+ :return: V1ValidateDeploymentImageResponse
3739
+ If the method is called asynchronously,
3740
+ returns the request thread.
3741
+ """
3742
+ kwargs['_return_http_data_only'] = True
3743
+ if kwargs.get('async_req'):
3744
+ return self.jobs_service_validate_deployment_image_with_http_info(body, **kwargs) # noqa: E501
3745
+ else:
3746
+ (data) = self.jobs_service_validate_deployment_image_with_http_info(body, **kwargs) # noqa: E501
3747
+ return data
3748
+
3749
+ def jobs_service_validate_deployment_image_with_http_info(self, body: 'V1ValidateDeploymentImageRequest', **kwargs) -> 'V1ValidateDeploymentImageResponse': # noqa: E501
3750
+ """Job deployments # noqa: E501
3751
+
3752
+ This method makes a synchronous HTTP request by default. To make an
3753
+ asynchronous HTTP request, please pass async_req=True
3754
+ >>> thread = api.jobs_service_validate_deployment_image_with_http_info(body, async_req=True)
3755
+ >>> result = thread.get()
3756
+
3757
+ :param async_req bool
3758
+ :param V1ValidateDeploymentImageRequest body: (required)
3759
+ :return: V1ValidateDeploymentImageResponse
3760
+ If the method is called asynchronously,
3761
+ returns the request thread.
3762
+ """
3763
+
3764
+ all_params = ['body'] # noqa: E501
3765
+ all_params.append('async_req')
3766
+ all_params.append('_return_http_data_only')
3767
+ all_params.append('_preload_content')
3768
+ all_params.append('_request_timeout')
3769
+
3770
+ params = locals()
3771
+ for key, val in six.iteritems(params['kwargs']):
3772
+ if key not in all_params:
3773
+ raise TypeError(
3774
+ "Got an unexpected keyword argument '%s'"
3775
+ " to method jobs_service_validate_deployment_image" % key
3776
+ )
3777
+ params[key] = val
3778
+ del params['kwargs']
3779
+ # verify the required parameter 'body' is set
3780
+ if ('body' not in params or
3781
+ params['body'] is None):
3782
+ raise ValueError("Missing the required parameter `body` when calling `jobs_service_validate_deployment_image`") # noqa: E501
3783
+
3784
+ collection_formats = {}
3785
+
3786
+ path_params = {}
3787
+
3788
+ query_params = []
3789
+
3790
+ header_params = {}
3791
+
3792
+ form_params = []
3793
+ local_var_files = {}
3794
+
3795
+ body_params = None
3796
+ if 'body' in params:
3797
+ body_params = params['body']
3798
+ # HTTP header `Accept`
3799
+ header_params['Accept'] = self.api_client.select_header_accept(
3800
+ ['application/json']) # noqa: E501
3801
+
3802
+ # HTTP header `Content-Type`
3803
+ header_params['Content-Type'] = self.api_client.select_header_content_type( # noqa: E501
3804
+ ['application/json']) # noqa: E501
3805
+
3806
+ # Authentication setting
3807
+ auth_settings = [] # noqa: E501
3808
+
3809
+ return self.api_client.call_api(
3810
+ '/v1/deployments/validate', 'POST',
3811
+ path_params,
3812
+ query_params,
3813
+ header_params,
3814
+ body=body_params,
3815
+ post_params=form_params,
3816
+ files=local_var_files,
3817
+ response_type='V1ValidateDeploymentImageResponse', # noqa: E501
3818
+ auth_settings=auth_settings,
3819
+ async_req=params.get('async_req'),
3820
+ _return_http_data_only=params.get('_return_http_data_only'),
3821
+ _preload_content=params.get('_preload_content', True),
3822
+ _request_timeout=params.get('_request_timeout'),
3823
+ collection_formats=collection_formats)
@@ -1009,6 +1009,111 @@ class LitLoggerServiceApi(object):
1009
1009
  _request_timeout=params.get('_request_timeout'),
1010
1010
  collection_formats=collection_formats)
1011
1011
 
1012
+ def lit_logger_service_get_metrics_stream(self, **kwargs) -> 'V1MetricsStream': # noqa: E501
1013
+ """lit_logger_service_get_metrics_stream # noqa: E501
1014
+
1015
+ This method makes a synchronous HTTP request by default. To make an
1016
+ asynchronous HTTP request, please pass async_req=True
1017
+ >>> thread = api.lit_logger_service_get_metrics_stream(async_req=True)
1018
+ >>> result = thread.get()
1019
+
1020
+ :param async_req bool
1021
+ :param str project_name:
1022
+ :param str project_owner_name:
1023
+ :param str name:
1024
+ :param str id:
1025
+ :param int version_number:
1026
+ :return: V1MetricsStream
1027
+ If the method is called asynchronously,
1028
+ returns the request thread.
1029
+ """
1030
+ kwargs['_return_http_data_only'] = True
1031
+ if kwargs.get('async_req'):
1032
+ return self.lit_logger_service_get_metrics_stream_with_http_info(**kwargs) # noqa: E501
1033
+ else:
1034
+ (data) = self.lit_logger_service_get_metrics_stream_with_http_info(**kwargs) # noqa: E501
1035
+ return data
1036
+
1037
+ def lit_logger_service_get_metrics_stream_with_http_info(self, **kwargs) -> 'V1MetricsStream': # noqa: E501
1038
+ """lit_logger_service_get_metrics_stream # noqa: E501
1039
+
1040
+ This method makes a synchronous HTTP request by default. To make an
1041
+ asynchronous HTTP request, please pass async_req=True
1042
+ >>> thread = api.lit_logger_service_get_metrics_stream_with_http_info(async_req=True)
1043
+ >>> result = thread.get()
1044
+
1045
+ :param async_req bool
1046
+ :param str project_name:
1047
+ :param str project_owner_name:
1048
+ :param str name:
1049
+ :param str id:
1050
+ :param int version_number:
1051
+ :return: V1MetricsStream
1052
+ If the method is called asynchronously,
1053
+ returns the request thread.
1054
+ """
1055
+
1056
+ all_params = ['project_name', 'project_owner_name', 'name', 'id', 'version_number'] # noqa: E501
1057
+ all_params.append('async_req')
1058
+ all_params.append('_return_http_data_only')
1059
+ all_params.append('_preload_content')
1060
+ all_params.append('_request_timeout')
1061
+
1062
+ params = locals()
1063
+ for key, val in six.iteritems(params['kwargs']):
1064
+ if key not in all_params:
1065
+ raise TypeError(
1066
+ "Got an unexpected keyword argument '%s'"
1067
+ " to method lit_logger_service_get_metrics_stream" % key
1068
+ )
1069
+ params[key] = val
1070
+ del params['kwargs']
1071
+
1072
+ collection_formats = {}
1073
+
1074
+ path_params = {}
1075
+
1076
+ query_params = []
1077
+ if 'project_name' in params:
1078
+ query_params.append(('projectName', params['project_name'])) # noqa: E501
1079
+ if 'project_owner_name' in params:
1080
+ query_params.append(('projectOwnerName', params['project_owner_name'])) # noqa: E501
1081
+ if 'name' in params:
1082
+ query_params.append(('name', params['name'])) # noqa: E501
1083
+ if 'id' in params:
1084
+ query_params.append(('id', params['id'])) # noqa: E501
1085
+ if 'version_number' in params:
1086
+ query_params.append(('versionNumber', params['version_number'])) # noqa: E501
1087
+
1088
+ header_params = {}
1089
+
1090
+ form_params = []
1091
+ local_var_files = {}
1092
+
1093
+ body_params = None
1094
+ # HTTP header `Accept`
1095
+ header_params['Accept'] = self.api_client.select_header_accept(
1096
+ ['application/json']) # noqa: E501
1097
+
1098
+ # Authentication setting
1099
+ auth_settings = [] # noqa: E501
1100
+
1101
+ return self.api_client.call_api(
1102
+ '/v1/metrics-streams', 'GET',
1103
+ path_params,
1104
+ query_params,
1105
+ header_params,
1106
+ body=body_params,
1107
+ post_params=form_params,
1108
+ files=local_var_files,
1109
+ response_type='V1MetricsStream', # noqa: E501
1110
+ auth_settings=auth_settings,
1111
+ async_req=params.get('async_req'),
1112
+ _return_http_data_only=params.get('_return_http_data_only'),
1113
+ _preload_content=params.get('_preload_content', True),
1114
+ _request_timeout=params.get('_request_timeout'),
1115
+ collection_formats=collection_formats)
1116
+
1012
1117
  def lit_logger_service_get_shared_metrics_stream(self, id: 'str', **kwargs) -> 'V1MetricsStream': # noqa: E501
1013
1118
  """lit_logger_service_get_shared_metrics_stream # noqa: E501
1014
1119
 
@@ -1429,6 +1534,119 @@ class LitLoggerServiceApi(object):
1429
1534
  _request_timeout=params.get('_request_timeout'),
1430
1535
  collection_formats=collection_formats)
1431
1536
 
1537
+ def lit_logger_service_update_metrics_stream_visibility(self, body: 'IdVisibilityBody', project_id: 'str', id: 'str', **kwargs) -> 'V1UpdateMetricsStreamVisibilityResponse': # noqa: E501
1538
+ """lit_logger_service_update_metrics_stream_visibility # noqa: E501
1539
+
1540
+ This method makes a synchronous HTTP request by default. To make an
1541
+ asynchronous HTTP request, please pass async_req=True
1542
+ >>> thread = api.lit_logger_service_update_metrics_stream_visibility(body, project_id, id, async_req=True)
1543
+ >>> result = thread.get()
1544
+
1545
+ :param async_req bool
1546
+ :param IdVisibilityBody body: (required)
1547
+ :param str project_id: (required)
1548
+ :param str id: (required)
1549
+ :return: V1UpdateMetricsStreamVisibilityResponse
1550
+ If the method is called asynchronously,
1551
+ returns the request thread.
1552
+ """
1553
+ kwargs['_return_http_data_only'] = True
1554
+ if kwargs.get('async_req'):
1555
+ return self.lit_logger_service_update_metrics_stream_visibility_with_http_info(body, project_id, id, **kwargs) # noqa: E501
1556
+ else:
1557
+ (data) = self.lit_logger_service_update_metrics_stream_visibility_with_http_info(body, project_id, id, **kwargs) # noqa: E501
1558
+ return data
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
1561
+ """lit_logger_service_update_metrics_stream_visibility # noqa: E501
1562
+
1563
+ This method makes a synchronous HTTP request by default. To make an
1564
+ asynchronous HTTP request, please pass async_req=True
1565
+ >>> thread = api.lit_logger_service_update_metrics_stream_visibility_with_http_info(body, project_id, id, async_req=True)
1566
+ >>> result = thread.get()
1567
+
1568
+ :param async_req bool
1569
+ :param IdVisibilityBody body: (required)
1570
+ :param str project_id: (required)
1571
+ :param str id: (required)
1572
+ :return: V1UpdateMetricsStreamVisibilityResponse
1573
+ If the method is called asynchronously,
1574
+ returns the request thread.
1575
+ """
1576
+
1577
+ all_params = ['body', 'project_id', 'id'] # noqa: E501
1578
+ all_params.append('async_req')
1579
+ all_params.append('_return_http_data_only')
1580
+ all_params.append('_preload_content')
1581
+ all_params.append('_request_timeout')
1582
+
1583
+ params = locals()
1584
+ for key, val in six.iteritems(params['kwargs']):
1585
+ if key not in all_params:
1586
+ raise TypeError(
1587
+ "Got an unexpected keyword argument '%s'"
1588
+ " to method lit_logger_service_update_metrics_stream_visibility" % key
1589
+ )
1590
+ params[key] = val
1591
+ del params['kwargs']
1592
+ # verify the required parameter 'body' is set
1593
+ if ('body' not in params or
1594
+ params['body'] is None):
1595
+ raise ValueError("Missing the required parameter `body` when calling `lit_logger_service_update_metrics_stream_visibility`") # noqa: E501
1596
+ # verify the required parameter 'project_id' is set
1597
+ if ('project_id' not in params or
1598
+ params['project_id'] is None):
1599
+ raise ValueError("Missing the required parameter `project_id` when calling `lit_logger_service_update_metrics_stream_visibility`") # noqa: E501
1600
+ # verify the required parameter 'id' is set
1601
+ if ('id' not in params or
1602
+ params['id'] is None):
1603
+ raise ValueError("Missing the required parameter `id` when calling `lit_logger_service_update_metrics_stream_visibility`") # noqa: E501
1604
+
1605
+ collection_formats = {}
1606
+
1607
+ path_params = {}
1608
+ if 'project_id' in params:
1609
+ path_params['projectId'] = params['project_id'] # noqa: E501
1610
+ if 'id' in params:
1611
+ path_params['id'] = params['id'] # noqa: E501
1612
+
1613
+ query_params = []
1614
+
1615
+ header_params = {}
1616
+
1617
+ form_params = []
1618
+ local_var_files = {}
1619
+
1620
+ body_params = None
1621
+ if 'body' in params:
1622
+ body_params = params['body']
1623
+ # HTTP header `Accept`
1624
+ header_params['Accept'] = self.api_client.select_header_accept(
1625
+ ['application/json']) # noqa: E501
1626
+
1627
+ # HTTP header `Content-Type`
1628
+ header_params['Content-Type'] = self.api_client.select_header_content_type( # noqa: E501
1629
+ ['application/json']) # noqa: E501
1630
+
1631
+ # Authentication setting
1632
+ auth_settings = [] # noqa: E501
1633
+
1634
+ return self.api_client.call_api(
1635
+ '/v1/projects/{projectId}/metrics-streams/{id}/visibility', 'PUT',
1636
+ path_params,
1637
+ query_params,
1638
+ header_params,
1639
+ body=body_params,
1640
+ post_params=form_params,
1641
+ files=local_var_files,
1642
+ response_type='V1UpdateMetricsStreamVisibilityResponse', # noqa: E501
1643
+ auth_settings=auth_settings,
1644
+ async_req=params.get('async_req'),
1645
+ _return_http_data_only=params.get('_return_http_data_only'),
1646
+ _preload_content=params.get('_preload_content', True),
1647
+ _request_timeout=params.get('_request_timeout'),
1648
+ collection_formats=collection_formats)
1649
+
1432
1650
  def lit_logger_service_update_shared_metrics_stream(self, body: 'LitloggermetricsIdBody', id: 'str', **kwargs) -> 'V1UpdateSharedMetricsStreamResponse': # noqa: E501
1433
1651
  """lit_logger_service_update_shared_metrics_stream # noqa: E501
1434
1652