lightning-sdk 0.2.22__py3-none-any.whl → 0.2.24rc0__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 (101) hide show
  1. lightning_sdk/__init__.py +1 -1
  2. lightning_sdk/api/base_studio_api.py +9 -2
  3. lightning_sdk/api/deployment_api.py +9 -9
  4. lightning_sdk/api/license_api.py +2 -2
  5. lightning_sdk/api/llm_api.py +7 -11
  6. lightning_sdk/api/pipeline_api.py +31 -10
  7. lightning_sdk/api/studio_api.py +6 -0
  8. lightning_sdk/base_studio.py +22 -6
  9. lightning_sdk/cli/entrypoint.py +15 -13
  10. lightning_sdk/cli/start.py +5 -2
  11. lightning_sdk/deployment/deployment.py +17 -7
  12. lightning_sdk/lightning_cloud/openapi/__init__.py +20 -0
  13. lightning_sdk/lightning_cloud/openapi/api/__init__.py +1 -0
  14. lightning_sdk/lightning_cloud/openapi/api/assistants_service_api.py +114 -1
  15. lightning_sdk/lightning_cloud/openapi/api/cloud_space_service_api.py +206 -0
  16. lightning_sdk/lightning_cloud/openapi/api/cloudy_service_api.py +129 -0
  17. lightning_sdk/lightning_cloud/openapi/api/cluster_service_api.py +97 -0
  18. lightning_sdk/lightning_cloud/openapi/api/organizations_service_api.py +105 -0
  19. lightning_sdk/lightning_cloud/openapi/api/pipelines_service_api.py +118 -1
  20. lightning_sdk/lightning_cloud/openapi/api/user_service_api.py +105 -0
  21. lightning_sdk/lightning_cloud/openapi/configuration.py +1 -1
  22. lightning_sdk/lightning_cloud/openapi/models/__init__.py +19 -0
  23. lightning_sdk/lightning_cloud/openapi/models/agents_id_body.py +27 -1
  24. lightning_sdk/lightning_cloud/openapi/models/assistant_id_conversations_body.py +81 -3
  25. lightning_sdk/lightning_cloud/openapi/models/cloudspace_id_visibility_body.py +123 -0
  26. lightning_sdk/lightning_cloud/openapi/models/create_deployment_request_defines_a_spec_for_the_job_that_allows_for_autoscaling_jobs.py +27 -1
  27. lightning_sdk/lightning_cloud/openapi/models/credits_autoreplenish_body.py +175 -0
  28. lightning_sdk/lightning_cloud/openapi/models/credits_autoreplenish_body1.py +175 -0
  29. lightning_sdk/lightning_cloud/openapi/models/externalv1_user_status.py +79 -1
  30. lightning_sdk/lightning_cloud/openapi/models/metricsstream_create_body.py +27 -1
  31. lightning_sdk/lightning_cloud/openapi/models/orgs_id_body.py +27 -1
  32. lightning_sdk/lightning_cloud/openapi/models/pipelines_id_body.py +27 -1
  33. lightning_sdk/lightning_cloud/openapi/models/pipelines_id_body1.py +123 -0
  34. lightning_sdk/lightning_cloud/openapi/models/project_id_agents_body.py +27 -1
  35. lightning_sdk/lightning_cloud/openapi/models/project_id_cloudspaces_body.py +53 -1
  36. lightning_sdk/lightning_cloud/openapi/models/update.py +29 -3
  37. lightning_sdk/lightning_cloud/openapi/models/v1_assistant.py +27 -1
  38. lightning_sdk/lightning_cloud/openapi/models/v1_check_cluster_name_availability_request.py +123 -0
  39. lightning_sdk/lightning_cloud/openapi/models/v1_check_cluster_name_availability_response.py +123 -0
  40. lightning_sdk/lightning_cloud/openapi/models/v1_cloud_provider.py +2 -0
  41. lightning_sdk/lightning_cloud/openapi/models/v1_cloud_space.py +79 -1
  42. lightning_sdk/lightning_cloud/openapi/models/v1_cloud_space_environment_config.py +29 -3
  43. lightning_sdk/lightning_cloud/openapi/models/v1_cloud_space_environment_template_config.py +29 -3
  44. lightning_sdk/lightning_cloud/openapi/models/v1_cloud_space_environment_type.py +1 -0
  45. lightning_sdk/lightning_cloud/openapi/models/v1_cloud_space_session.py +29 -3
  46. lightning_sdk/lightning_cloud/openapi/models/v1_cloud_space_specialized_view.py +104 -0
  47. lightning_sdk/lightning_cloud/openapi/models/v1_cloudy_expert.py +279 -0
  48. lightning_sdk/lightning_cloud/openapi/models/v1_cluster_accelerator.py +79 -1
  49. lightning_sdk/lightning_cloud/openapi/models/v1_cluster_capacity_reservation.py +27 -1
  50. lightning_sdk/lightning_cloud/openapi/models/v1_cluster_security_options.py +27 -1
  51. lightning_sdk/lightning_cloud/openapi/models/v1_cluster_spec.py +105 -1
  52. lightning_sdk/lightning_cloud/openapi/models/v1_cluster_status.py +27 -1
  53. lightning_sdk/lightning_cloud/openapi/models/v1_conversation_response_chunk.py +29 -3
  54. lightning_sdk/lightning_cloud/openapi/models/v1_create_cloud_space_environment_template_request.py +29 -3
  55. lightning_sdk/lightning_cloud/openapi/models/v1_create_deployment_request.py +27 -1
  56. lightning_sdk/lightning_cloud/openapi/models/v1_create_organization_request.py +79 -1
  57. lightning_sdk/lightning_cloud/openapi/models/v1_deployment_status.py +47 -21
  58. lightning_sdk/lightning_cloud/openapi/models/v1_external_cluster.py +253 -0
  59. lightning_sdk/lightning_cloud/openapi/models/v1_external_cluster_spec.py +853 -0
  60. lightning_sdk/lightning_cloud/openapi/models/v1_get_job_stats_response.py +53 -1
  61. lightning_sdk/lightning_cloud/openapi/models/v1_get_user_response.py +27 -1
  62. lightning_sdk/lightning_cloud/openapi/models/v1_instance_overprovisioning_spec.py +29 -27
  63. lightning_sdk/lightning_cloud/openapi/models/v1_kubernetes_direct_v1.py +123 -0
  64. lightning_sdk/lightning_cloud/openapi/models/v1_kubernetes_direct_v1_status.py +149 -0
  65. lightning_sdk/lightning_cloud/openapi/models/v1_lightning_run.py +53 -1
  66. lightning_sdk/lightning_cloud/openapi/models/v1_list_cloudy_experts_response.py +123 -0
  67. lightning_sdk/lightning_cloud/openapi/models/v1_list_clusters_response.py +6 -6
  68. lightning_sdk/lightning_cloud/openapi/models/v1_list_project_clusters_response.py +6 -6
  69. lightning_sdk/lightning_cloud/openapi/models/v1_lite_published_cloud_space_response.py +513 -0
  70. lightning_sdk/lightning_cloud/openapi/models/v1_login_request.py +27 -1
  71. lightning_sdk/lightning_cloud/openapi/models/v1_magic_link_login_request.py +29 -3
  72. lightning_sdk/lightning_cloud/openapi/models/v1_magic_link_login_response.py +27 -1
  73. lightning_sdk/lightning_cloud/openapi/models/v1_metrics_stream.py +27 -1
  74. lightning_sdk/lightning_cloud/openapi/models/v1_organization.py +27 -1
  75. lightning_sdk/lightning_cloud/openapi/models/v1_pipeline.py +27 -1
  76. lightning_sdk/lightning_cloud/openapi/models/v1_shared_filesystem.py +131 -1
  77. lightning_sdk/lightning_cloud/openapi/models/v1_token_usage.py +175 -0
  78. lightning_sdk/lightning_cloud/openapi/models/v1_update_cloud_space_visibility_response.py +97 -0
  79. lightning_sdk/lightning_cloud/openapi/models/v1_update_organization_credits_auto_replenish_response.py +97 -0
  80. lightning_sdk/lightning_cloud/openapi/models/v1_update_user_credits_auto_replenish_response.py +97 -0
  81. lightning_sdk/lightning_cloud/openapi/models/v1_update_user_request.py +27 -1
  82. lightning_sdk/lightning_cloud/openapi/models/v1_user_features.py +234 -104
  83. lightning_sdk/lightning_cloud/openapi/models/v1_volume.py +78 -104
  84. lightning_sdk/lightning_cloud/openapi/models/v1_volume_state.py +104 -0
  85. lightning_sdk/llm/llm.py +113 -115
  86. lightning_sdk/llm/public_assistants.json +8 -0
  87. lightning_sdk/pipeline/__init__.py +11 -2
  88. lightning_sdk/pipeline/pipeline.py +54 -14
  89. lightning_sdk/pipeline/printer.py +36 -16
  90. lightning_sdk/pipeline/schedule.py +2 -1
  91. lightning_sdk/pipeline/{types.py → steps.py} +77 -56
  92. lightning_sdk/pipeline/utils.py +65 -3
  93. lightning_sdk/sandbox.py +157 -0
  94. lightning_sdk/services/license.py +12 -6
  95. lightning_sdk/studio.py +10 -1
  96. {lightning_sdk-0.2.22.dist-info → lightning_sdk-0.2.24rc0.dist-info}/METADATA +1 -1
  97. {lightning_sdk-0.2.22.dist-info → lightning_sdk-0.2.24rc0.dist-info}/RECORD +101 -79
  98. {lightning_sdk-0.2.22.dist-info → lightning_sdk-0.2.24rc0.dist-info}/LICENSE +0 -0
  99. {lightning_sdk-0.2.22.dist-info → lightning_sdk-0.2.24rc0.dist-info}/WHEEL +0 -0
  100. {lightning_sdk-0.2.22.dist-info → lightning_sdk-0.2.24rc0.dist-info}/entry_points.txt +0 -0
  101. {lightning_sdk-0.2.22.dist-info → lightning_sdk-0.2.24rc0.dist-info}/top_level.txt +0 -0
@@ -1958,6 +1958,111 @@ class OrganizationsServiceApi(object):
1958
1958
  _request_timeout=params.get('_request_timeout'),
1959
1959
  collection_formats=collection_formats)
1960
1960
 
1961
+ def organizations_service_update_organization_credits_auto_replenish(self, body: 'CreditsAutoreplenishBody', org_id: 'str', **kwargs) -> 'V1UpdateOrganizationCreditsAutoReplenishResponse': # noqa: E501
1962
+ """organizations_service_update_organization_credits_auto_replenish # noqa: E501
1963
+
1964
+ This method makes a synchronous HTTP request by default. To make an
1965
+ asynchronous HTTP request, please pass async_req=True
1966
+ >>> thread = api.organizations_service_update_organization_credits_auto_replenish(body, org_id, async_req=True)
1967
+ >>> result = thread.get()
1968
+
1969
+ :param async_req bool
1970
+ :param CreditsAutoreplenishBody body: (required)
1971
+ :param str org_id: (required)
1972
+ :return: V1UpdateOrganizationCreditsAutoReplenishResponse
1973
+ If the method is called asynchronously,
1974
+ returns the request thread.
1975
+ """
1976
+ kwargs['_return_http_data_only'] = True
1977
+ if kwargs.get('async_req'):
1978
+ return self.organizations_service_update_organization_credits_auto_replenish_with_http_info(body, org_id, **kwargs) # noqa: E501
1979
+ else:
1980
+ (data) = self.organizations_service_update_organization_credits_auto_replenish_with_http_info(body, org_id, **kwargs) # noqa: E501
1981
+ return data
1982
+
1983
+ def organizations_service_update_organization_credits_auto_replenish_with_http_info(self, body: 'CreditsAutoreplenishBody', org_id: 'str', **kwargs) -> 'V1UpdateOrganizationCreditsAutoReplenishResponse': # noqa: E501
1984
+ """organizations_service_update_organization_credits_auto_replenish # noqa: E501
1985
+
1986
+ This method makes a synchronous HTTP request by default. To make an
1987
+ asynchronous HTTP request, please pass async_req=True
1988
+ >>> thread = api.organizations_service_update_organization_credits_auto_replenish_with_http_info(body, org_id, async_req=True)
1989
+ >>> result = thread.get()
1990
+
1991
+ :param async_req bool
1992
+ :param CreditsAutoreplenishBody body: (required)
1993
+ :param str org_id: (required)
1994
+ :return: V1UpdateOrganizationCreditsAutoReplenishResponse
1995
+ If the method is called asynchronously,
1996
+ returns the request thread.
1997
+ """
1998
+
1999
+ all_params = ['body', 'org_id'] # noqa: E501
2000
+ all_params.append('async_req')
2001
+ all_params.append('_return_http_data_only')
2002
+ all_params.append('_preload_content')
2003
+ all_params.append('_request_timeout')
2004
+
2005
+ params = locals()
2006
+ for key, val in six.iteritems(params['kwargs']):
2007
+ if key not in all_params:
2008
+ raise TypeError(
2009
+ "Got an unexpected keyword argument '%s'"
2010
+ " to method organizations_service_update_organization_credits_auto_replenish" % key
2011
+ )
2012
+ params[key] = val
2013
+ del params['kwargs']
2014
+ # verify the required parameter 'body' is set
2015
+ if ('body' not in params or
2016
+ params['body'] is None):
2017
+ raise ValueError("Missing the required parameter `body` when calling `organizations_service_update_organization_credits_auto_replenish`") # noqa: E501
2018
+ # verify the required parameter 'org_id' is set
2019
+ if ('org_id' not in params or
2020
+ params['org_id'] is None):
2021
+ raise ValueError("Missing the required parameter `org_id` when calling `organizations_service_update_organization_credits_auto_replenish`") # noqa: E501
2022
+
2023
+ collection_formats = {}
2024
+
2025
+ path_params = {}
2026
+ if 'org_id' in params:
2027
+ path_params['orgId'] = params['org_id'] # noqa: E501
2028
+
2029
+ query_params = []
2030
+
2031
+ header_params = {}
2032
+
2033
+ form_params = []
2034
+ local_var_files = {}
2035
+
2036
+ body_params = None
2037
+ if 'body' in params:
2038
+ body_params = params['body']
2039
+ # HTTP header `Accept`
2040
+ header_params['Accept'] = self.api_client.select_header_accept(
2041
+ ['application/json']) # noqa: E501
2042
+
2043
+ # HTTP header `Content-Type`
2044
+ header_params['Content-Type'] = self.api_client.select_header_content_type( # noqa: E501
2045
+ ['application/json']) # noqa: E501
2046
+
2047
+ # Authentication setting
2048
+ auth_settings = [] # noqa: E501
2049
+
2050
+ return self.api_client.call_api(
2051
+ '/v1/orgs/{orgId}/credits/auto-replenish', 'PUT',
2052
+ path_params,
2053
+ query_params,
2054
+ header_params,
2055
+ body=body_params,
2056
+ post_params=form_params,
2057
+ files=local_var_files,
2058
+ response_type='V1UpdateOrganizationCreditsAutoReplenishResponse', # noqa: E501
2059
+ auth_settings=auth_settings,
2060
+ async_req=params.get('async_req'),
2061
+ _return_http_data_only=params.get('_return_http_data_only'),
2062
+ _preload_content=params.get('_preload_content', True),
2063
+ _request_timeout=params.get('_request_timeout'),
2064
+ collection_formats=collection_formats)
2065
+
1961
2066
  def organizations_service_validate_auto_join_domain(self, body: 'ValidateautojoindomainDomainBody', org_id: 'str', domain: 'str', **kwargs) -> 'V1ValidateAutoJoinDomainResponse': # noqa: E501
1962
2067
  """AutoJoin Domain management # noqa: E501
1963
2068
 
@@ -43,6 +43,119 @@ class PipelinesServiceApi(object):
43
43
  api_client = ApiClient()
44
44
  self.api_client = api_client
45
45
 
46
+ def pipelines_service_create_child_pipeline(self, body: 'PipelinesIdBody1', project_id: 'str', id: 'str', **kwargs) -> 'V1Pipeline': # noqa: E501
47
+ """pipelines_service_create_child_pipeline # noqa: E501
48
+
49
+ This method makes a synchronous HTTP request by default. To make an
50
+ asynchronous HTTP request, please pass async_req=True
51
+ >>> thread = api.pipelines_service_create_child_pipeline(body, project_id, id, async_req=True)
52
+ >>> result = thread.get()
53
+
54
+ :param async_req bool
55
+ :param PipelinesIdBody1 body: (required)
56
+ :param str project_id: (required)
57
+ :param str id: (required)
58
+ :return: V1Pipeline
59
+ If the method is called asynchronously,
60
+ returns the request thread.
61
+ """
62
+ kwargs['_return_http_data_only'] = True
63
+ if kwargs.get('async_req'):
64
+ return self.pipelines_service_create_child_pipeline_with_http_info(body, project_id, id, **kwargs) # noqa: E501
65
+ else:
66
+ (data) = self.pipelines_service_create_child_pipeline_with_http_info(body, project_id, id, **kwargs) # noqa: E501
67
+ return data
68
+
69
+ def pipelines_service_create_child_pipeline_with_http_info(self, body: 'PipelinesIdBody1', project_id: 'str', id: 'str', **kwargs) -> 'V1Pipeline': # noqa: E501
70
+ """pipelines_service_create_child_pipeline # noqa: E501
71
+
72
+ This method makes a synchronous HTTP request by default. To make an
73
+ asynchronous HTTP request, please pass async_req=True
74
+ >>> thread = api.pipelines_service_create_child_pipeline_with_http_info(body, project_id, id, async_req=True)
75
+ >>> result = thread.get()
76
+
77
+ :param async_req bool
78
+ :param PipelinesIdBody1 body: (required)
79
+ :param str project_id: (required)
80
+ :param str id: (required)
81
+ :return: V1Pipeline
82
+ If the method is called asynchronously,
83
+ returns the request thread.
84
+ """
85
+
86
+ all_params = ['body', 'project_id', 'id'] # noqa: E501
87
+ all_params.append('async_req')
88
+ all_params.append('_return_http_data_only')
89
+ all_params.append('_preload_content')
90
+ all_params.append('_request_timeout')
91
+
92
+ params = locals()
93
+ for key, val in six.iteritems(params['kwargs']):
94
+ if key not in all_params:
95
+ raise TypeError(
96
+ "Got an unexpected keyword argument '%s'"
97
+ " to method pipelines_service_create_child_pipeline" % key
98
+ )
99
+ params[key] = val
100
+ del params['kwargs']
101
+ # verify the required parameter 'body' is set
102
+ if ('body' not in params or
103
+ params['body'] is None):
104
+ raise ValueError("Missing the required parameter `body` when calling `pipelines_service_create_child_pipeline`") # noqa: E501
105
+ # verify the required parameter 'project_id' is set
106
+ if ('project_id' not in params or
107
+ params['project_id'] is None):
108
+ raise ValueError("Missing the required parameter `project_id` when calling `pipelines_service_create_child_pipeline`") # noqa: E501
109
+ # verify the required parameter 'id' is set
110
+ if ('id' not in params or
111
+ params['id'] is None):
112
+ raise ValueError("Missing the required parameter `id` when calling `pipelines_service_create_child_pipeline`") # noqa: E501
113
+
114
+ collection_formats = {}
115
+
116
+ path_params = {}
117
+ if 'project_id' in params:
118
+ path_params['projectId'] = params['project_id'] # noqa: E501
119
+ if 'id' in params:
120
+ path_params['id'] = params['id'] # noqa: E501
121
+
122
+ query_params = []
123
+
124
+ header_params = {}
125
+
126
+ form_params = []
127
+ local_var_files = {}
128
+
129
+ body_params = None
130
+ if 'body' in params:
131
+ body_params = params['body']
132
+ # HTTP header `Accept`
133
+ header_params['Accept'] = self.api_client.select_header_accept(
134
+ ['application/json']) # noqa: E501
135
+
136
+ # HTTP header `Content-Type`
137
+ header_params['Content-Type'] = self.api_client.select_header_content_type( # noqa: E501
138
+ ['application/json']) # noqa: E501
139
+
140
+ # Authentication setting
141
+ auth_settings = [] # noqa: E501
142
+
143
+ return self.api_client.call_api(
144
+ '/v1/projects/{projectId}/pipelines/{id}', 'POST',
145
+ path_params,
146
+ query_params,
147
+ header_params,
148
+ body=body_params,
149
+ post_params=form_params,
150
+ files=local_var_files,
151
+ response_type='V1Pipeline', # noqa: E501
152
+ auth_settings=auth_settings,
153
+ async_req=params.get('async_req'),
154
+ _return_http_data_only=params.get('_return_http_data_only'),
155
+ _preload_content=params.get('_preload_content', True),
156
+ _request_timeout=params.get('_request_timeout'),
157
+ collection_formats=collection_formats)
158
+
46
159
  def pipelines_service_create_pipeline(self, body: 'ProjectIdPipelinesBody', project_id: 'str', **kwargs) -> 'V1Pipeline': # noqa: E501
47
160
  """pipelines_service_create_pipeline # noqa: E501
48
161
 
@@ -462,6 +575,7 @@ class PipelinesServiceApi(object):
462
575
  :param async_req bool
463
576
  :param str project_id: (required)
464
577
  :param str parent_pipeline_id:
578
+ :param str original_pipeline_id:
465
579
  :param str page_token:
466
580
  :param int limit:
467
581
  :param str state:
@@ -488,6 +602,7 @@ class PipelinesServiceApi(object):
488
602
  :param async_req bool
489
603
  :param str project_id: (required)
490
604
  :param str parent_pipeline_id:
605
+ :param str original_pipeline_id:
491
606
  :param str page_token:
492
607
  :param int limit:
493
608
  :param str state:
@@ -497,7 +612,7 @@ class PipelinesServiceApi(object):
497
612
  returns the request thread.
498
613
  """
499
614
 
500
- all_params = ['project_id', 'parent_pipeline_id', 'page_token', 'limit', 'state', 'standalone'] # noqa: E501
615
+ all_params = ['project_id', 'parent_pipeline_id', 'original_pipeline_id', 'page_token', 'limit', 'state', 'standalone'] # noqa: E501
501
616
  all_params.append('async_req')
502
617
  all_params.append('_return_http_data_only')
503
618
  all_params.append('_preload_content')
@@ -526,6 +641,8 @@ class PipelinesServiceApi(object):
526
641
  query_params = []
527
642
  if 'parent_pipeline_id' in params:
528
643
  query_params.append(('parentPipelineId', params['parent_pipeline_id'])) # noqa: E501
644
+ if 'original_pipeline_id' in params:
645
+ query_params.append(('originalPipelineId', params['original_pipeline_id'])) # noqa: E501
529
646
  if 'page_token' in params:
530
647
  query_params.append(('pageToken', params['page_token'])) # noqa: E501
531
648
  if 'limit' in params:
@@ -1284,6 +1284,111 @@ class UserServiceApi(object):
1284
1284
  _request_timeout=params.get('_request_timeout'),
1285
1285
  collection_formats=collection_formats)
1286
1286
 
1287
+ def user_service_update_user_credits_auto_replenish(self, body: 'CreditsAutoreplenishBody1', user_id: 'str', **kwargs) -> 'V1UpdateUserCreditsAutoReplenishResponse': # noqa: E501
1288
+ """user_service_update_user_credits_auto_replenish # noqa: E501
1289
+
1290
+ This method makes a synchronous HTTP request by default. To make an
1291
+ asynchronous HTTP request, please pass async_req=True
1292
+ >>> thread = api.user_service_update_user_credits_auto_replenish(body, user_id, async_req=True)
1293
+ >>> result = thread.get()
1294
+
1295
+ :param async_req bool
1296
+ :param CreditsAutoreplenishBody1 body: (required)
1297
+ :param str user_id: (required)
1298
+ :return: V1UpdateUserCreditsAutoReplenishResponse
1299
+ If the method is called asynchronously,
1300
+ returns the request thread.
1301
+ """
1302
+ kwargs['_return_http_data_only'] = True
1303
+ if kwargs.get('async_req'):
1304
+ return self.user_service_update_user_credits_auto_replenish_with_http_info(body, user_id, **kwargs) # noqa: E501
1305
+ else:
1306
+ (data) = self.user_service_update_user_credits_auto_replenish_with_http_info(body, user_id, **kwargs) # noqa: E501
1307
+ return data
1308
+
1309
+ def user_service_update_user_credits_auto_replenish_with_http_info(self, body: 'CreditsAutoreplenishBody1', user_id: 'str', **kwargs) -> 'V1UpdateUserCreditsAutoReplenishResponse': # noqa: E501
1310
+ """user_service_update_user_credits_auto_replenish # noqa: E501
1311
+
1312
+ This method makes a synchronous HTTP request by default. To make an
1313
+ asynchronous HTTP request, please pass async_req=True
1314
+ >>> thread = api.user_service_update_user_credits_auto_replenish_with_http_info(body, user_id, async_req=True)
1315
+ >>> result = thread.get()
1316
+
1317
+ :param async_req bool
1318
+ :param CreditsAutoreplenishBody1 body: (required)
1319
+ :param str user_id: (required)
1320
+ :return: V1UpdateUserCreditsAutoReplenishResponse
1321
+ If the method is called asynchronously,
1322
+ returns the request thread.
1323
+ """
1324
+
1325
+ all_params = ['body', 'user_id'] # noqa: E501
1326
+ all_params.append('async_req')
1327
+ all_params.append('_return_http_data_only')
1328
+ all_params.append('_preload_content')
1329
+ all_params.append('_request_timeout')
1330
+
1331
+ params = locals()
1332
+ for key, val in six.iteritems(params['kwargs']):
1333
+ if key not in all_params:
1334
+ raise TypeError(
1335
+ "Got an unexpected keyword argument '%s'"
1336
+ " to method user_service_update_user_credits_auto_replenish" % key
1337
+ )
1338
+ params[key] = val
1339
+ del params['kwargs']
1340
+ # verify the required parameter 'body' is set
1341
+ if ('body' not in params or
1342
+ params['body'] is None):
1343
+ raise ValueError("Missing the required parameter `body` when calling `user_service_update_user_credits_auto_replenish`") # noqa: E501
1344
+ # verify the required parameter 'user_id' is set
1345
+ if ('user_id' not in params or
1346
+ params['user_id'] is None):
1347
+ raise ValueError("Missing the required parameter `user_id` when calling `user_service_update_user_credits_auto_replenish`") # noqa: E501
1348
+
1349
+ collection_formats = {}
1350
+
1351
+ path_params = {}
1352
+ if 'user_id' in params:
1353
+ path_params['userId'] = params['user_id'] # noqa: E501
1354
+
1355
+ query_params = []
1356
+
1357
+ header_params = {}
1358
+
1359
+ form_params = []
1360
+ local_var_files = {}
1361
+
1362
+ body_params = None
1363
+ if 'body' in params:
1364
+ body_params = params['body']
1365
+ # HTTP header `Accept`
1366
+ header_params['Accept'] = self.api_client.select_header_accept(
1367
+ ['application/json']) # noqa: E501
1368
+
1369
+ # HTTP header `Content-Type`
1370
+ header_params['Content-Type'] = self.api_client.select_header_content_type( # noqa: E501
1371
+ ['application/json']) # noqa: E501
1372
+
1373
+ # Authentication setting
1374
+ auth_settings = [] # noqa: E501
1375
+
1376
+ return self.api_client.call_api(
1377
+ '/v1/users/{userId}/credits/auto-replenish', 'PUT',
1378
+ path_params,
1379
+ query_params,
1380
+ header_params,
1381
+ body=body_params,
1382
+ post_params=form_params,
1383
+ files=local_var_files,
1384
+ response_type='V1UpdateUserCreditsAutoReplenishResponse', # noqa: E501
1385
+ auth_settings=auth_settings,
1386
+ async_req=params.get('async_req'),
1387
+ _return_http_data_only=params.get('_return_http_data_only'),
1388
+ _preload_content=params.get('_preload_content', True),
1389
+ _request_timeout=params.get('_request_timeout'),
1390
+ collection_formats=collection_formats)
1391
+
1287
1392
  def user_service_update_user_viewed_new_features(self, body: 'V1UpdateUserViewedNewFeaturesRequest', **kwargs) -> 'V1UpdateUserViewedNewFeaturesResponse': # noqa: E501
1288
1393
  """we use this to update user viewed new features # noqa: E501
1289
1394
 
@@ -70,7 +70,7 @@ class Configuration(six.with_metaclass(TypeWithDefault, object)):
70
70
  self.password = ""
71
71
  # Logging Settings
72
72
  self.logger = {}
73
- self.logger["package_logger"] = logging.getLogger("lightning_sdk.lightning_cloud.openapi")
73
+ self.logger["package_logger"] = logging.getLogger("lightning_cloud.openapi")
74
74
  self.logger["urllib3_logger"] = logging.getLogger("urllib3")
75
75
  # Log format
76
76
  self.logger_format = '%(asctime)s %(levelname)s %(message)s'
@@ -44,6 +44,7 @@ from lightning_sdk.lightning_cloud.openapi.models.cloud_space_id_versions_body i
44
44
  from lightning_sdk.lightning_cloud.openapi.models.cloudspace_id_metric_body import CloudspaceIdMetricBody
45
45
  from lightning_sdk.lightning_cloud.openapi.models.cloudspace_id_runs_body import CloudspaceIdRunsBody
46
46
  from lightning_sdk.lightning_cloud.openapi.models.cloudspace_id_systemmetrics_body import CloudspaceIdSystemmetricsBody
47
+ from lightning_sdk.lightning_cloud.openapi.models.cloudspace_id_visibility_body import CloudspaceIdVisibilityBody
47
48
  from lightning_sdk.lightning_cloud.openapi.models.cloudspaces_id_body import CloudspacesIdBody
48
49
  from lightning_sdk.lightning_cloud.openapi.models.cluster_id_capacityblock_body import ClusterIdCapacityblockBody
49
50
  from lightning_sdk.lightning_cloud.openapi.models.cluster_id_capacityreservations_body import ClusterIdCapacityreservationsBody
@@ -57,6 +58,8 @@ from lightning_sdk.lightning_cloud.openapi.models.conversations_id_body import C
57
58
  from lightning_sdk.lightning_cloud.openapi.models.create import Create
58
59
  from lightning_sdk.lightning_cloud.openapi.models.create_checkout_session_request_wallet_type import CreateCheckoutSessionRequestWalletType
59
60
  from lightning_sdk.lightning_cloud.openapi.models.create_deployment_request_defines_a_spec_for_the_job_that_allows_for_autoscaling_jobs import CreateDeploymentRequestDefinesASpecForTheJobThatAllowsForAutoscalingJobs
61
+ from lightning_sdk.lightning_cloud.openapi.models.credits_autoreplenish_body import CreditsAutoreplenishBody
62
+ from lightning_sdk.lightning_cloud.openapi.models.credits_autoreplenish_body1 import CreditsAutoreplenishBody1
60
63
  from lightning_sdk.lightning_cloud.openapi.models.data_connection_mount_data_connection_mount_copy_status import DataConnectionMountDataConnectionMountCopyStatus
61
64
  from lightning_sdk.lightning_cloud.openapi.models.dataset_id_versions_body import DatasetIdVersionsBody
62
65
  from lightning_sdk.lightning_cloud.openapi.models.dataset_id_visibility_body import DatasetIdVisibilityBody
@@ -129,6 +132,7 @@ from lightning_sdk.lightning_cloud.openapi.models.org_id_memberships_body import
129
132
  from lightning_sdk.lightning_cloud.openapi.models.org_id_roles_body import OrgIdRolesBody
130
133
  from lightning_sdk.lightning_cloud.openapi.models.orgs_id_body import OrgsIdBody
131
134
  from lightning_sdk.lightning_cloud.openapi.models.pipelines_id_body import PipelinesIdBody
135
+ from lightning_sdk.lightning_cloud.openapi.models.pipelines_id_body1 import PipelinesIdBody1
132
136
  from lightning_sdk.lightning_cloud.openapi.models.pipelinetemplates_id_body import PipelinetemplatesIdBody
133
137
  from lightning_sdk.lightning_cloud.openapi.models.profiler_captures_body import ProfilerCapturesBody
134
138
  from lightning_sdk.lightning_cloud.openapi.models.profiler_enabled_body import ProfilerEnabledBody
@@ -230,6 +234,8 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_cpu_system_metrics import V
230
234
  from lightning_sdk.lightning_cloud.openapi.models.v1_cancel_cloud_space_instance_switch_response import V1CancelCloudSpaceInstanceSwitchResponse
231
235
  from lightning_sdk.lightning_cloud.openapi.models.v1_cancellation_metadata import V1CancellationMetadata
232
236
  from lightning_sdk.lightning_cloud.openapi.models.v1_capacity_block_offering import V1CapacityBlockOffering
237
+ from lightning_sdk.lightning_cloud.openapi.models.v1_check_cluster_name_availability_request import V1CheckClusterNameAvailabilityRequest
238
+ from lightning_sdk.lightning_cloud.openapi.models.v1_check_cluster_name_availability_response import V1CheckClusterNameAvailabilityResponse
233
239
  from lightning_sdk.lightning_cloud.openapi.models.v1_check_external_service_status_response import V1CheckExternalServiceStatusResponse
234
240
  from lightning_sdk.lightning_cloud.openapi.models.v1_check_snowflake_connection_response import V1CheckSnowflakeConnectionResponse
235
241
  from lightning_sdk.lightning_cloud.openapi.models.v1_checkbox import V1Checkbox
@@ -260,10 +266,12 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_cloud_space_publication_typ
260
266
  from lightning_sdk.lightning_cloud.openapi.models.v1_cloud_space_seed_file import V1CloudSpaceSeedFile
261
267
  from lightning_sdk.lightning_cloud.openapi.models.v1_cloud_space_session import V1CloudSpaceSession
262
268
  from lightning_sdk.lightning_cloud.openapi.models.v1_cloud_space_source_type import V1CloudSpaceSourceType
269
+ from lightning_sdk.lightning_cloud.openapi.models.v1_cloud_space_specialized_view import V1CloudSpaceSpecializedView
263
270
  from lightning_sdk.lightning_cloud.openapi.models.v1_cloud_space_state import V1CloudSpaceState
264
271
  from lightning_sdk.lightning_cloud.openapi.models.v1_cloud_space_version import V1CloudSpaceVersion
265
272
  from lightning_sdk.lightning_cloud.openapi.models.v1_cloud_space_version_publication import V1CloudSpaceVersionPublication
266
273
  from lightning_sdk.lightning_cloud.openapi.models.v1_cloudflare_v1 import V1CloudflareV1
274
+ from lightning_sdk.lightning_cloud.openapi.models.v1_cloudy_expert import V1CloudyExpert
267
275
  from lightning_sdk.lightning_cloud.openapi.models.v1_cluster_accelerator import V1ClusterAccelerator
268
276
  from lightning_sdk.lightning_cloud.openapi.models.v1_cluster_availability import V1ClusterAvailability
269
277
  from lightning_sdk.lightning_cloud.openapi.models.v1_cluster_capacity_reservation import V1ClusterCapacityReservation
@@ -454,6 +462,8 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_execute_in_cloud_space_sess
454
462
  from lightning_sdk.lightning_cloud.openapi.models.v1_execute_snowflake_query_response import V1ExecuteSnowflakeQueryResponse
455
463
  from lightning_sdk.lightning_cloud.openapi.models.v1_experiment import V1Experiment
456
464
  from lightning_sdk.lightning_cloud.openapi.models.v1_export_snowflake_query_response import V1ExportSnowflakeQueryResponse
465
+ from lightning_sdk.lightning_cloud.openapi.models.v1_external_cluster import V1ExternalCluster
466
+ from lightning_sdk.lightning_cloud.openapi.models.v1_external_cluster_spec import V1ExternalClusterSpec
457
467
  from lightning_sdk.lightning_cloud.openapi.models.v1_external_search_user import V1ExternalSearchUser
458
468
  from lightning_sdk.lightning_cloud.openapi.models.v1_file_endpoint import V1FileEndpoint
459
469
  from lightning_sdk.lightning_cloud.openapi.models.v1_filestore_data_connection import V1FilestoreDataConnection
@@ -549,6 +559,8 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_job_type import V1JobType
549
559
  from lightning_sdk.lightning_cloud.openapi.models.v1_joinable_organization import V1JoinableOrganization
550
560
  from lightning_sdk.lightning_cloud.openapi.models.v1_keep_alive_cloud_space_instance_response import V1KeepAliveCloudSpaceInstanceResponse
551
561
  from lightning_sdk.lightning_cloud.openapi.models.v1_knowledge_configuration import V1KnowledgeConfiguration
562
+ from lightning_sdk.lightning_cloud.openapi.models.v1_kubernetes_direct_v1 import V1KubernetesDirectV1
563
+ from lightning_sdk.lightning_cloud.openapi.models.v1_kubernetes_direct_v1_status import V1KubernetesDirectV1Status
552
564
  from lightning_sdk.lightning_cloud.openapi.models.v1_lambda_labs_direct_v1 import V1LambdaLabsDirectV1
553
565
  from lightning_sdk.lightning_cloud.openapi.models.v1_lightning_app_user import V1LightningAppUser
554
566
  from lightning_sdk.lightning_cloud.openapi.models.v1_lightning_auth import V1LightningAuth
@@ -584,6 +596,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_list_cloud_space_tags_respo
584
596
  from lightning_sdk.lightning_cloud.openapi.models.v1_list_cloud_space_version_publications_response import V1ListCloudSpaceVersionPublicationsResponse
585
597
  from lightning_sdk.lightning_cloud.openapi.models.v1_list_cloud_space_versions_response import V1ListCloudSpaceVersionsResponse
586
598
  from lightning_sdk.lightning_cloud.openapi.models.v1_list_cloud_spaces_response import V1ListCloudSpacesResponse
599
+ from lightning_sdk.lightning_cloud.openapi.models.v1_list_cloudy_experts_response import V1ListCloudyExpertsResponse
587
600
  from lightning_sdk.lightning_cloud.openapi.models.v1_list_cluster_accelerators_response import V1ListClusterAcceleratorsResponse
588
601
  from lightning_sdk.lightning_cloud.openapi.models.v1_list_cluster_availabilities_response import V1ListClusterAvailabilitiesResponse
589
602
  from lightning_sdk.lightning_cloud.openapi.models.v1_list_cluster_capacity_reservations_response import V1ListClusterCapacityReservationsResponse
@@ -674,6 +687,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_lit_page_type import V1LitP
674
687
  from lightning_sdk.lightning_cloud.openapi.models.v1_lit_registry_artifact import V1LitRegistryArtifact
675
688
  from lightning_sdk.lightning_cloud.openapi.models.v1_lit_registry_project import V1LitRegistryProject
676
689
  from lightning_sdk.lightning_cloud.openapi.models.v1_lit_repository import V1LitRepository
690
+ from lightning_sdk.lightning_cloud.openapi.models.v1_lite_published_cloud_space_response import V1LitePublishedCloudSpaceResponse
677
691
  from lightning_sdk.lightning_cloud.openapi.models.v1_locked_resource import V1LockedResource
678
692
  from lightning_sdk.lightning_cloud.openapi.models.v1_log_page import V1LogPage
679
693
  from lightning_sdk.lightning_cloud.openapi.models.v1_logger_artifact import V1LoggerArtifact
@@ -853,6 +867,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_system_metrics_aggregated i
853
867
  from lightning_sdk.lightning_cloud.openapi.models.v1_system_metrics_list import V1SystemMetricsList
854
868
  from lightning_sdk.lightning_cloud.openapi.models.v1_telemetry import V1Telemetry
855
869
  from lightning_sdk.lightning_cloud.openapi.models.v1_timestamp_code_telemetry import V1TimestampCodeTelemetry
870
+ from lightning_sdk.lightning_cloud.openapi.models.v1_token_usage import V1TokenUsage
856
871
  from lightning_sdk.lightning_cloud.openapi.models.v1_transaction import V1Transaction
857
872
  from lightning_sdk.lightning_cloud.openapi.models.v1_transfer_cloud_space_response import V1TransferCloudSpaceResponse
858
873
  from lightning_sdk.lightning_cloud.openapi.models.v1_transfer_org_balance_response import V1TransferOrgBalanceResponse
@@ -866,6 +881,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_update_agent_status_respons
866
881
  from lightning_sdk.lightning_cloud.openapi.models.v1_update_billing_subscription_request import V1UpdateBillingSubscriptionRequest
867
882
  from lightning_sdk.lightning_cloud.openapi.models.v1_update_cloud_space_collab_response import V1UpdateCloudSpaceCollabResponse
868
883
  from lightning_sdk.lightning_cloud.openapi.models.v1_update_cloud_space_publication_response import V1UpdateCloudSpacePublicationResponse
884
+ from lightning_sdk.lightning_cloud.openapi.models.v1_update_cloud_space_visibility_response import V1UpdateCloudSpaceVisibilityResponse
869
885
  from lightning_sdk.lightning_cloud.openapi.models.v1_update_cluster_accelerators_request import V1UpdateClusterAcceleratorsRequest
870
886
  from lightning_sdk.lightning_cloud.openapi.models.v1_update_cluster_accelerators_response import V1UpdateClusterAcceleratorsResponse
871
887
  from lightning_sdk.lightning_cloud.openapi.models.v1_update_cluster_availability_request import V1UpdateClusterAvailabilityRequest
@@ -877,9 +893,11 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_update_lit_page_response im
877
893
  from lightning_sdk.lightning_cloud.openapi.models.v1_update_lit_repository_response import V1UpdateLitRepositoryResponse
878
894
  from lightning_sdk.lightning_cloud.openapi.models.v1_update_metrics_stream_visibility_response import V1UpdateMetricsStreamVisibilityResponse
879
895
  from lightning_sdk.lightning_cloud.openapi.models.v1_update_model_visibility_response import V1UpdateModelVisibilityResponse
896
+ from lightning_sdk.lightning_cloud.openapi.models.v1_update_organization_credits_auto_replenish_response import V1UpdateOrganizationCreditsAutoReplenishResponse
880
897
  from lightning_sdk.lightning_cloud.openapi.models.v1_update_project_cluster_accelerators_response import V1UpdateProjectClusterAcceleratorsResponse
881
898
  from lightning_sdk.lightning_cloud.openapi.models.v1_update_shared_metrics_stream_response import V1UpdateSharedMetricsStreamResponse
882
899
  from lightning_sdk.lightning_cloud.openapi.models.v1_update_snowflake_query_response import V1UpdateSnowflakeQueryResponse
900
+ from lightning_sdk.lightning_cloud.openapi.models.v1_update_user_credits_auto_replenish_response import V1UpdateUserCreditsAutoReplenishResponse
883
901
  from lightning_sdk.lightning_cloud.openapi.models.v1_update_user_request import V1UpdateUserRequest
884
902
  from lightning_sdk.lightning_cloud.openapi.models.v1_update_user_viewed_new_features_request import V1UpdateUserViewedNewFeaturesRequest
885
903
  from lightning_sdk.lightning_cloud.openapi.models.v1_update_user_viewed_new_features_response import V1UpdateUserViewedNewFeaturesResponse
@@ -914,6 +932,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_validate_managed_model_resp
914
932
  from lightning_sdk.lightning_cloud.openapi.models.v1_verify_verification_response import V1VerifyVerificationResponse
915
933
  from lightning_sdk.lightning_cloud.openapi.models.v1_voltage_park_direct_v1 import V1VoltageParkDirectV1
916
934
  from lightning_sdk.lightning_cloud.openapi.models.v1_volume import V1Volume
935
+ from lightning_sdk.lightning_cloud.openapi.models.v1_volume_state import V1VolumeState
917
936
  from lightning_sdk.lightning_cloud.openapi.models.v1_vultr_direct_v1 import V1VultrDirectV1
918
937
  from lightning_sdk.lightning_cloud.openapi.models.v1_weka_data_connection import V1WekaDataConnection
919
938
  from lightning_sdk.lightning_cloud.openapi.models.v1_work import V1Work
@@ -42,6 +42,7 @@ class AgentsIdBody(object):
42
42
  """
43
43
  swagger_types = {
44
44
  'cloudspace_id': 'str',
45
+ 'cloudy_compatible': 'bool',
45
46
  'cluster_id': 'str',
46
47
  'created_at': 'datetime',
47
48
  'deployment_details': 'V1DeploymentDetails',
@@ -68,6 +69,7 @@ class AgentsIdBody(object):
68
69
 
69
70
  attribute_map = {
70
71
  'cloudspace_id': 'cloudspaceId',
72
+ 'cloudy_compatible': 'cloudyCompatible',
71
73
  'cluster_id': 'clusterId',
72
74
  'created_at': 'createdAt',
73
75
  'deployment_details': 'deploymentDetails',
@@ -92,9 +94,10 @@ class AgentsIdBody(object):
92
94
  'user_id': 'userId'
93
95
  }
94
96
 
95
- def __init__(self, cloudspace_id: 'str' =None, cluster_id: 'str' =None, created_at: 'datetime' =None, deployment_details: 'V1DeploymentDetails' =None, description: 'str' =None, endpoint_id: 'str' =None, expected_cold_start_time: 'str' =None, file_uploads_enabled: 'bool' =None, internal_assistant_name: 'str' =None, knowledge: 'str' =None, knowledge_configuration: 'V1KnowledgeConfiguration' =None, managed_endpoint_id: 'str' =None, model: 'str' =None, model_provider: 'str' =None, name: 'str' =None, org_id: 'str' =None, prompt_suggestions: 'list[V1PromptSuggestion]' =None, prompt_template: 'str' =None, publish_status: 'str' =None, status: 'V1AssistantModelStatus' =None, thumbnail_url: 'str' =None, updated_at: 'datetime' =None, user_id: 'str' =None): # noqa: E501
97
+ def __init__(self, cloudspace_id: 'str' =None, cloudy_compatible: 'bool' =None, cluster_id: 'str' =None, created_at: 'datetime' =None, deployment_details: 'V1DeploymentDetails' =None, description: 'str' =None, endpoint_id: 'str' =None, expected_cold_start_time: 'str' =None, file_uploads_enabled: 'bool' =None, internal_assistant_name: 'str' =None, knowledge: 'str' =None, knowledge_configuration: 'V1KnowledgeConfiguration' =None, managed_endpoint_id: 'str' =None, model: 'str' =None, model_provider: 'str' =None, name: 'str' =None, org_id: 'str' =None, prompt_suggestions: 'list[V1PromptSuggestion]' =None, prompt_template: 'str' =None, publish_status: 'str' =None, status: 'V1AssistantModelStatus' =None, thumbnail_url: 'str' =None, updated_at: 'datetime' =None, user_id: 'str' =None): # noqa: E501
96
98
  """AgentsIdBody - a model defined in Swagger""" # noqa: E501
97
99
  self._cloudspace_id = None
100
+ self._cloudy_compatible = None
98
101
  self._cluster_id = None
99
102
  self._created_at = None
100
103
  self._deployment_details = None
@@ -120,6 +123,8 @@ class AgentsIdBody(object):
120
123
  self.discriminator = None
121
124
  if cloudspace_id is not None:
122
125
  self.cloudspace_id = cloudspace_id
126
+ if cloudy_compatible is not None:
127
+ self.cloudy_compatible = cloudy_compatible
123
128
  if cluster_id is not None:
124
129
  self.cluster_id = cluster_id
125
130
  if created_at is not None:
@@ -186,6 +191,27 @@ class AgentsIdBody(object):
186
191
 
187
192
  self._cloudspace_id = cloudspace_id
188
193
 
194
+ @property
195
+ def cloudy_compatible(self) -> 'bool':
196
+ """Gets the cloudy_compatible of this AgentsIdBody. # noqa: E501
197
+
198
+
199
+ :return: The cloudy_compatible of this AgentsIdBody. # noqa: E501
200
+ :rtype: bool
201
+ """
202
+ return self._cloudy_compatible
203
+
204
+ @cloudy_compatible.setter
205
+ def cloudy_compatible(self, cloudy_compatible: 'bool'):
206
+ """Sets the cloudy_compatible of this AgentsIdBody.
207
+
208
+
209
+ :param cloudy_compatible: The cloudy_compatible of this AgentsIdBody. # noqa: E501
210
+ :type: bool
211
+ """
212
+
213
+ self._cloudy_compatible = cloudy_compatible
214
+
189
215
  @property
190
216
  def cluster_id(self) -> 'str':
191
217
  """Gets the cluster_id of this AgentsIdBody. # noqa: E501