lightning-sdk 0.2.23__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 (70) 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/llm_api.py +5 -11
  5. lightning_sdk/api/pipeline_api.py +31 -11
  6. lightning_sdk/api/studio_api.py +4 -0
  7. lightning_sdk/base_studio.py +22 -6
  8. lightning_sdk/deployment/deployment.py +17 -7
  9. lightning_sdk/lightning_cloud/openapi/__init__.py +12 -0
  10. lightning_sdk/lightning_cloud/openapi/api/__init__.py +1 -0
  11. lightning_sdk/lightning_cloud/openapi/api/assistants_service_api.py +114 -1
  12. lightning_sdk/lightning_cloud/openapi/api/cloudy_service_api.py +129 -0
  13. lightning_sdk/lightning_cloud/openapi/api/organizations_service_api.py +105 -0
  14. lightning_sdk/lightning_cloud/openapi/api/pipelines_service_api.py +4 -4
  15. lightning_sdk/lightning_cloud/openapi/api/user_service_api.py +105 -0
  16. lightning_sdk/lightning_cloud/openapi/models/__init__.py +11 -0
  17. lightning_sdk/lightning_cloud/openapi/models/agents_id_body.py +27 -1
  18. lightning_sdk/lightning_cloud/openapi/models/assistant_id_conversations_body.py +53 -1
  19. lightning_sdk/lightning_cloud/openapi/models/credits_autoreplenish_body.py +175 -0
  20. lightning_sdk/lightning_cloud/openapi/models/credits_autoreplenish_body1.py +175 -0
  21. lightning_sdk/lightning_cloud/openapi/models/externalv1_user_status.py +79 -1
  22. lightning_sdk/lightning_cloud/openapi/models/orgs_id_body.py +27 -1
  23. lightning_sdk/lightning_cloud/openapi/models/pipelines_id_body1.py +123 -0
  24. lightning_sdk/lightning_cloud/openapi/models/project_id_agents_body.py +27 -1
  25. lightning_sdk/lightning_cloud/openapi/models/project_id_cloudspaces_body.py +27 -1
  26. lightning_sdk/lightning_cloud/openapi/models/update.py +29 -3
  27. lightning_sdk/lightning_cloud/openapi/models/v1_assistant.py +27 -1
  28. lightning_sdk/lightning_cloud/openapi/models/v1_cloud_provider.py +1 -0
  29. lightning_sdk/lightning_cloud/openapi/models/v1_cloud_space.py +27 -1
  30. lightning_sdk/lightning_cloud/openapi/models/v1_cloud_space_environment_config.py +29 -3
  31. lightning_sdk/lightning_cloud/openapi/models/v1_cloud_space_environment_template_config.py +29 -3
  32. lightning_sdk/lightning_cloud/openapi/models/v1_cloud_space_environment_type.py +1 -0
  33. lightning_sdk/lightning_cloud/openapi/models/v1_cloud_space_specialized_view.py +104 -0
  34. lightning_sdk/lightning_cloud/openapi/models/v1_cloudy_expert.py +279 -0
  35. lightning_sdk/lightning_cloud/openapi/models/v1_cluster_accelerator.py +79 -1
  36. lightning_sdk/lightning_cloud/openapi/models/v1_cluster_capacity_reservation.py +27 -1
  37. lightning_sdk/lightning_cloud/openapi/models/v1_cluster_spec.py +27 -1
  38. lightning_sdk/lightning_cloud/openapi/models/v1_cluster_status.py +27 -1
  39. lightning_sdk/lightning_cloud/openapi/models/v1_conversation_response_chunk.py +29 -3
  40. lightning_sdk/lightning_cloud/openapi/models/v1_create_cloud_space_environment_template_request.py +29 -3
  41. lightning_sdk/lightning_cloud/openapi/models/v1_create_organization_request.py +79 -1
  42. lightning_sdk/lightning_cloud/openapi/models/v1_deployment_status.py +47 -21
  43. lightning_sdk/lightning_cloud/openapi/models/v1_external_cluster_spec.py +27 -1
  44. lightning_sdk/lightning_cloud/openapi/models/v1_get_job_stats_response.py +53 -1
  45. lightning_sdk/lightning_cloud/openapi/models/v1_instance_overprovisioning_spec.py +1 -27
  46. lightning_sdk/lightning_cloud/openapi/models/v1_kubernetes_direct_v1.py +123 -0
  47. lightning_sdk/lightning_cloud/openapi/models/v1_kubernetes_direct_v1_status.py +149 -0
  48. lightning_sdk/lightning_cloud/openapi/models/v1_list_cloudy_experts_response.py +123 -0
  49. lightning_sdk/lightning_cloud/openapi/models/v1_login_request.py +27 -1
  50. lightning_sdk/lightning_cloud/openapi/models/v1_magic_link_login_request.py +29 -3
  51. lightning_sdk/lightning_cloud/openapi/models/v1_magic_link_login_response.py +27 -1
  52. lightning_sdk/lightning_cloud/openapi/models/v1_organization.py +27 -1
  53. lightning_sdk/lightning_cloud/openapi/models/v1_token_usage.py +175 -0
  54. lightning_sdk/lightning_cloud/openapi/models/v1_update_organization_credits_auto_replenish_response.py +97 -0
  55. lightning_sdk/lightning_cloud/openapi/models/v1_update_user_credits_auto_replenish_response.py +97 -0
  56. lightning_sdk/lightning_cloud/openapi/models/v1_update_user_request.py +27 -1
  57. lightning_sdk/lightning_cloud/openapi/models/v1_user_features.py +185 -29
  58. lightning_sdk/llm/llm.py +113 -115
  59. lightning_sdk/llm/public_assistants.json +8 -0
  60. lightning_sdk/pipeline/pipeline.py +17 -2
  61. lightning_sdk/pipeline/printer.py +11 -10
  62. lightning_sdk/pipeline/steps.py +4 -1
  63. lightning_sdk/pipeline/utils.py +29 -4
  64. lightning_sdk/studio.py +3 -0
  65. {lightning_sdk-0.2.23.dist-info → lightning_sdk-0.2.24rc0.dist-info}/METADATA +1 -1
  66. {lightning_sdk-0.2.23.dist-info → lightning_sdk-0.2.24rc0.dist-info}/RECORD +70 -57
  67. {lightning_sdk-0.2.23.dist-info → lightning_sdk-0.2.24rc0.dist-info}/LICENSE +0 -0
  68. {lightning_sdk-0.2.23.dist-info → lightning_sdk-0.2.24rc0.dist-info}/WHEEL +0 -0
  69. {lightning_sdk-0.2.23.dist-info → lightning_sdk-0.2.24rc0.dist-info}/entry_points.txt +0 -0
  70. {lightning_sdk-0.2.23.dist-info → lightning_sdk-0.2.24rc0.dist-info}/top_level.txt +0 -0
@@ -0,0 +1,129 @@
1
+ # coding: utf-8
2
+
3
+ """
4
+ external/v1/auth_service.proto
5
+
6
+ No description provided (generated by Swagger Codegen https://github.com/swagger-api/swagger-codegen) # noqa: E501
7
+
8
+ OpenAPI spec version: version not set
9
+
10
+ Generated by: https://github.com/swagger-api/swagger-codegen.git
11
+
12
+ NOTE
13
+ ----
14
+ standard swagger-codegen-cli for this python client has been modified
15
+ by custom templates. The purpose of these templates is to include
16
+ typing information in the API and Model code. Please refer to the
17
+ main grid repository for more info
18
+ """
19
+
20
+ from __future__ import absolute_import
21
+
22
+ import re # noqa: F401
23
+ from typing import TYPE_CHECKING, Any
24
+
25
+ # python 2 and python 3 compatibility library
26
+ import six
27
+
28
+ from lightning_sdk.lightning_cloud.openapi.api_client import ApiClient
29
+
30
+ if TYPE_CHECKING:
31
+ from datetime import datetime
32
+ from lightning_sdk.lightning_cloud.openapi.models import *
33
+
34
+ class CloudyServiceApi(object):
35
+ """NOTE: This class is auto generated by the swagger code generator program.
36
+
37
+ Do not edit the class manually.
38
+ Ref: https://github.com/swagger-api/swagger-codegen
39
+ """
40
+
41
+ def __init__(self, api_client=None):
42
+ if api_client is None:
43
+ api_client = ApiClient()
44
+ self.api_client = api_client
45
+
46
+ def cloudy_service_list_cloudy_experts(self, **kwargs) -> 'V1ListCloudyExpertsResponse': # noqa: E501
47
+ """Lists all available Cloudy Experts This endpoint returns a list of all Cloudy Experts that can be used to generate cloudy responses # 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.cloudy_service_list_cloudy_experts(async_req=True)
52
+ >>> result = thread.get()
53
+
54
+ :param async_req bool
55
+ :return: V1ListCloudyExpertsResponse
56
+ If the method is called asynchronously,
57
+ returns the request thread.
58
+ """
59
+ kwargs['_return_http_data_only'] = True
60
+ if kwargs.get('async_req'):
61
+ return self.cloudy_service_list_cloudy_experts_with_http_info(**kwargs) # noqa: E501
62
+ else:
63
+ (data) = self.cloudy_service_list_cloudy_experts_with_http_info(**kwargs) # noqa: E501
64
+ return data
65
+
66
+ def cloudy_service_list_cloudy_experts_with_http_info(self, **kwargs) -> 'V1ListCloudyExpertsResponse': # noqa: E501
67
+ """Lists all available Cloudy Experts This endpoint returns a list of all Cloudy Experts that can be used to generate cloudy responses # noqa: E501
68
+
69
+ This method makes a synchronous HTTP request by default. To make an
70
+ asynchronous HTTP request, please pass async_req=True
71
+ >>> thread = api.cloudy_service_list_cloudy_experts_with_http_info(async_req=True)
72
+ >>> result = thread.get()
73
+
74
+ :param async_req bool
75
+ :return: V1ListCloudyExpertsResponse
76
+ If the method is called asynchronously,
77
+ returns the request thread.
78
+ """
79
+
80
+ all_params = [] # noqa: E501
81
+ all_params.append('async_req')
82
+ all_params.append('_return_http_data_only')
83
+ all_params.append('_preload_content')
84
+ all_params.append('_request_timeout')
85
+
86
+ params = locals()
87
+ for key, val in six.iteritems(params['kwargs']):
88
+ if key not in all_params:
89
+ raise TypeError(
90
+ "Got an unexpected keyword argument '%s'"
91
+ " to method cloudy_service_list_cloudy_experts" % key
92
+ )
93
+ params[key] = val
94
+ del params['kwargs']
95
+
96
+ collection_formats = {}
97
+
98
+ path_params = {}
99
+
100
+ query_params = []
101
+
102
+ header_params = {}
103
+
104
+ form_params = []
105
+ local_var_files = {}
106
+
107
+ body_params = None
108
+ # HTTP header `Accept`
109
+ header_params['Accept'] = self.api_client.select_header_accept(
110
+ ['application/json']) # noqa: E501
111
+
112
+ # Authentication setting
113
+ auth_settings = [] # noqa: E501
114
+
115
+ return self.api_client.call_api(
116
+ '/v1/cloudy/experts', 'GET',
117
+ path_params,
118
+ query_params,
119
+ header_params,
120
+ body=body_params,
121
+ post_params=form_params,
122
+ files=local_var_files,
123
+ response_type='V1ListCloudyExpertsResponse', # noqa: E501
124
+ auth_settings=auth_settings,
125
+ async_req=params.get('async_req'),
126
+ _return_http_data_only=params.get('_return_http_data_only'),
127
+ _preload_content=params.get('_preload_content', True),
128
+ _request_timeout=params.get('_request_timeout'),
129
+ collection_formats=collection_formats)
@@ -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,7 +43,7 @@ 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: 'object', project_id: 'str', id: 'str', **kwargs) -> 'V1Pipeline': # noqa: E501
46
+ def pipelines_service_create_child_pipeline(self, body: 'PipelinesIdBody1', project_id: 'str', id: 'str', **kwargs) -> 'V1Pipeline': # noqa: E501
47
47
  """pipelines_service_create_child_pipeline # noqa: E501
48
48
 
49
49
  This method makes a synchronous HTTP request by default. To make an
@@ -52,7 +52,7 @@ class PipelinesServiceApi(object):
52
52
  >>> result = thread.get()
53
53
 
54
54
  :param async_req bool
55
- :param object body: (required)
55
+ :param PipelinesIdBody1 body: (required)
56
56
  :param str project_id: (required)
57
57
  :param str id: (required)
58
58
  :return: V1Pipeline
@@ -66,7 +66,7 @@ class PipelinesServiceApi(object):
66
66
  (data) = self.pipelines_service_create_child_pipeline_with_http_info(body, project_id, id, **kwargs) # noqa: E501
67
67
  return data
68
68
 
69
- def pipelines_service_create_child_pipeline_with_http_info(self, body: 'object', project_id: 'str', id: 'str', **kwargs) -> 'V1Pipeline': # noqa: E501
69
+ def pipelines_service_create_child_pipeline_with_http_info(self, body: 'PipelinesIdBody1', project_id: 'str', id: 'str', **kwargs) -> 'V1Pipeline': # noqa: E501
70
70
  """pipelines_service_create_child_pipeline # noqa: E501
71
71
 
72
72
  This method makes a synchronous HTTP request by default. To make an
@@ -75,7 +75,7 @@ class PipelinesServiceApi(object):
75
75
  >>> result = thread.get()
76
76
 
77
77
  :param async_req bool
78
- :param object body: (required)
78
+ :param PipelinesIdBody1 body: (required)
79
79
  :param str project_id: (required)
80
80
  :param str id: (required)
81
81
  :return: V1Pipeline
@@ -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
 
@@ -58,6 +58,8 @@ from lightning_sdk.lightning_cloud.openapi.models.conversations_id_body import C
58
58
  from lightning_sdk.lightning_cloud.openapi.models.create import Create
59
59
  from lightning_sdk.lightning_cloud.openapi.models.create_checkout_session_request_wallet_type import CreateCheckoutSessionRequestWalletType
60
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
61
63
  from lightning_sdk.lightning_cloud.openapi.models.data_connection_mount_data_connection_mount_copy_status import DataConnectionMountDataConnectionMountCopyStatus
62
64
  from lightning_sdk.lightning_cloud.openapi.models.dataset_id_versions_body import DatasetIdVersionsBody
63
65
  from lightning_sdk.lightning_cloud.openapi.models.dataset_id_visibility_body import DatasetIdVisibilityBody
@@ -130,6 +132,7 @@ from lightning_sdk.lightning_cloud.openapi.models.org_id_memberships_body import
130
132
  from lightning_sdk.lightning_cloud.openapi.models.org_id_roles_body import OrgIdRolesBody
131
133
  from lightning_sdk.lightning_cloud.openapi.models.orgs_id_body import OrgsIdBody
132
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
133
136
  from lightning_sdk.lightning_cloud.openapi.models.pipelinetemplates_id_body import PipelinetemplatesIdBody
134
137
  from lightning_sdk.lightning_cloud.openapi.models.profiler_captures_body import ProfilerCapturesBody
135
138
  from lightning_sdk.lightning_cloud.openapi.models.profiler_enabled_body import ProfilerEnabledBody
@@ -263,10 +266,12 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_cloud_space_publication_typ
263
266
  from lightning_sdk.lightning_cloud.openapi.models.v1_cloud_space_seed_file import V1CloudSpaceSeedFile
264
267
  from lightning_sdk.lightning_cloud.openapi.models.v1_cloud_space_session import V1CloudSpaceSession
265
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
266
270
  from lightning_sdk.lightning_cloud.openapi.models.v1_cloud_space_state import V1CloudSpaceState
267
271
  from lightning_sdk.lightning_cloud.openapi.models.v1_cloud_space_version import V1CloudSpaceVersion
268
272
  from lightning_sdk.lightning_cloud.openapi.models.v1_cloud_space_version_publication import V1CloudSpaceVersionPublication
269
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
270
275
  from lightning_sdk.lightning_cloud.openapi.models.v1_cluster_accelerator import V1ClusterAccelerator
271
276
  from lightning_sdk.lightning_cloud.openapi.models.v1_cluster_availability import V1ClusterAvailability
272
277
  from lightning_sdk.lightning_cloud.openapi.models.v1_cluster_capacity_reservation import V1ClusterCapacityReservation
@@ -554,6 +559,8 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_job_type import V1JobType
554
559
  from lightning_sdk.lightning_cloud.openapi.models.v1_joinable_organization import V1JoinableOrganization
555
560
  from lightning_sdk.lightning_cloud.openapi.models.v1_keep_alive_cloud_space_instance_response import V1KeepAliveCloudSpaceInstanceResponse
556
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
557
564
  from lightning_sdk.lightning_cloud.openapi.models.v1_lambda_labs_direct_v1 import V1LambdaLabsDirectV1
558
565
  from lightning_sdk.lightning_cloud.openapi.models.v1_lightning_app_user import V1LightningAppUser
559
566
  from lightning_sdk.lightning_cloud.openapi.models.v1_lightning_auth import V1LightningAuth
@@ -589,6 +596,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_list_cloud_space_tags_respo
589
596
  from lightning_sdk.lightning_cloud.openapi.models.v1_list_cloud_space_version_publications_response import V1ListCloudSpaceVersionPublicationsResponse
590
597
  from lightning_sdk.lightning_cloud.openapi.models.v1_list_cloud_space_versions_response import V1ListCloudSpaceVersionsResponse
591
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
592
600
  from lightning_sdk.lightning_cloud.openapi.models.v1_list_cluster_accelerators_response import V1ListClusterAcceleratorsResponse
593
601
  from lightning_sdk.lightning_cloud.openapi.models.v1_list_cluster_availabilities_response import V1ListClusterAvailabilitiesResponse
594
602
  from lightning_sdk.lightning_cloud.openapi.models.v1_list_cluster_capacity_reservations_response import V1ListClusterCapacityReservationsResponse
@@ -859,6 +867,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_system_metrics_aggregated i
859
867
  from lightning_sdk.lightning_cloud.openapi.models.v1_system_metrics_list import V1SystemMetricsList
860
868
  from lightning_sdk.lightning_cloud.openapi.models.v1_telemetry import V1Telemetry
861
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
862
871
  from lightning_sdk.lightning_cloud.openapi.models.v1_transaction import V1Transaction
863
872
  from lightning_sdk.lightning_cloud.openapi.models.v1_transfer_cloud_space_response import V1TransferCloudSpaceResponse
864
873
  from lightning_sdk.lightning_cloud.openapi.models.v1_transfer_org_balance_response import V1TransferOrgBalanceResponse
@@ -884,9 +893,11 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_update_lit_page_response im
884
893
  from lightning_sdk.lightning_cloud.openapi.models.v1_update_lit_repository_response import V1UpdateLitRepositoryResponse
885
894
  from lightning_sdk.lightning_cloud.openapi.models.v1_update_metrics_stream_visibility_response import V1UpdateMetricsStreamVisibilityResponse
886
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
887
897
  from lightning_sdk.lightning_cloud.openapi.models.v1_update_project_cluster_accelerators_response import V1UpdateProjectClusterAcceleratorsResponse
888
898
  from lightning_sdk.lightning_cloud.openapi.models.v1_update_shared_metrics_stream_response import V1UpdateSharedMetricsStreamResponse
889
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
890
901
  from lightning_sdk.lightning_cloud.openapi.models.v1_update_user_request import V1UpdateUserRequest
891
902
  from lightning_sdk.lightning_cloud.openapi.models.v1_update_user_viewed_new_features_request import V1UpdateUserViewedNewFeaturesRequest
892
903
  from lightning_sdk.lightning_cloud.openapi.models.v1_update_user_viewed_new_features_response import V1UpdateUserViewedNewFeaturesResponse
@@ -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
@@ -44,12 +44,14 @@ class AssistantIdConversationsBody(object):
44
44
  'auto_name': 'bool',
45
45
  'billing_project_id': 'str',
46
46
  'conversation_id': 'str',
47
+ 'ephemeral': 'bool',
47
48
  'internal_conversation': 'bool',
48
49
  'max_tokens': 'str',
49
50
  'message': 'V1Message',
50
51
  'metadata': 'dict(str, str)',
51
52
  'name': 'str',
52
53
  'parent_message_id': 'str',
54
+ 'reasoning_effort': 'str',
53
55
  'store': 'bool',
54
56
  'stream': 'bool',
55
57
  'system_prompt': 'str'
@@ -59,28 +61,32 @@ class AssistantIdConversationsBody(object):
59
61
  'auto_name': 'autoName',
60
62
  'billing_project_id': 'billingProjectId',
61
63
  'conversation_id': 'conversationId',
64
+ 'ephemeral': 'ephemeral',
62
65
  'internal_conversation': 'internalConversation',
63
66
  'max_tokens': 'maxTokens',
64
67
  'message': 'message',
65
68
  'metadata': 'metadata',
66
69
  'name': 'name',
67
70
  'parent_message_id': 'parentMessageId',
71
+ 'reasoning_effort': 'reasoningEffort',
68
72
  'store': 'store',
69
73
  'stream': 'stream',
70
74
  'system_prompt': 'systemPrompt'
71
75
  }
72
76
 
73
- def __init__(self, auto_name: 'bool' =None, billing_project_id: 'str' =None, conversation_id: 'str' =None, internal_conversation: 'bool' =None, max_tokens: 'str' =None, message: 'V1Message' =None, metadata: 'dict(str, str)' =None, name: 'str' =None, parent_message_id: 'str' =None, store: 'bool' =None, stream: 'bool' =None, system_prompt: 'str' =None): # noqa: E501
77
+ def __init__(self, auto_name: 'bool' =None, billing_project_id: 'str' =None, conversation_id: 'str' =None, ephemeral: 'bool' =None, internal_conversation: 'bool' =None, max_tokens: 'str' =None, message: 'V1Message' =None, metadata: 'dict(str, str)' =None, name: 'str' =None, parent_message_id: 'str' =None, reasoning_effort: 'str' =None, store: 'bool' =None, stream: 'bool' =None, system_prompt: 'str' =None): # noqa: E501
74
78
  """AssistantIdConversationsBody - a model defined in Swagger""" # noqa: E501
75
79
  self._auto_name = None
76
80
  self._billing_project_id = None
77
81
  self._conversation_id = None
82
+ self._ephemeral = None
78
83
  self._internal_conversation = None
79
84
  self._max_tokens = None
80
85
  self._message = None
81
86
  self._metadata = None
82
87
  self._name = None
83
88
  self._parent_message_id = None
89
+ self._reasoning_effort = None
84
90
  self._store = None
85
91
  self._stream = None
86
92
  self._system_prompt = None
@@ -91,6 +97,8 @@ class AssistantIdConversationsBody(object):
91
97
  self.billing_project_id = billing_project_id
92
98
  if conversation_id is not None:
93
99
  self.conversation_id = conversation_id
100
+ if ephemeral is not None:
101
+ self.ephemeral = ephemeral
94
102
  if internal_conversation is not None:
95
103
  self.internal_conversation = internal_conversation
96
104
  if max_tokens is not None:
@@ -103,6 +111,8 @@ class AssistantIdConversationsBody(object):
103
111
  self.name = name
104
112
  if parent_message_id is not None:
105
113
  self.parent_message_id = parent_message_id
114
+ if reasoning_effort is not None:
115
+ self.reasoning_effort = reasoning_effort
106
116
  if store is not None:
107
117
  self.store = store
108
118
  if stream is not None:
@@ -173,6 +183,27 @@ class AssistantIdConversationsBody(object):
173
183
 
174
184
  self._conversation_id = conversation_id
175
185
 
186
+ @property
187
+ def ephemeral(self) -> 'bool':
188
+ """Gets the ephemeral of this AssistantIdConversationsBody. # noqa: E501
189
+
190
+
191
+ :return: The ephemeral of this AssistantIdConversationsBody. # noqa: E501
192
+ :rtype: bool
193
+ """
194
+ return self._ephemeral
195
+
196
+ @ephemeral.setter
197
+ def ephemeral(self, ephemeral: 'bool'):
198
+ """Sets the ephemeral of this AssistantIdConversationsBody.
199
+
200
+
201
+ :param ephemeral: The ephemeral of this AssistantIdConversationsBody. # noqa: E501
202
+ :type: bool
203
+ """
204
+
205
+ self._ephemeral = ephemeral
206
+
176
207
  @property
177
208
  def internal_conversation(self) -> 'bool':
178
209
  """Gets the internal_conversation of this AssistantIdConversationsBody. # noqa: E501
@@ -299,6 +330,27 @@ class AssistantIdConversationsBody(object):
299
330
 
300
331
  self._parent_message_id = parent_message_id
301
332
 
333
+ @property
334
+ def reasoning_effort(self) -> 'str':
335
+ """Gets the reasoning_effort of this AssistantIdConversationsBody. # noqa: E501
336
+
337
+
338
+ :return: The reasoning_effort of this AssistantIdConversationsBody. # noqa: E501
339
+ :rtype: str
340
+ """
341
+ return self._reasoning_effort
342
+
343
+ @reasoning_effort.setter
344
+ def reasoning_effort(self, reasoning_effort: 'str'):
345
+ """Sets the reasoning_effort of this AssistantIdConversationsBody.
346
+
347
+
348
+ :param reasoning_effort: The reasoning_effort of this AssistantIdConversationsBody. # noqa: E501
349
+ :type: str
350
+ """
351
+
352
+ self._reasoning_effort = reasoning_effort
353
+
302
354
  @property
303
355
  def store(self) -> 'bool':
304
356
  """Gets the store of this AssistantIdConversationsBody. # noqa: E501