lightning-sdk 0.2.23__py3-none-any.whl → 0.2.24rc1__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 (83) 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 +23 -13
  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 +18 -0
  10. lightning_sdk/lightning_cloud/openapi/api/__init__.py +2 -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/cluster_service_api.py +9 -1
  14. lightning_sdk/lightning_cloud/openapi/api/lit_logger_service_api.py +13 -1
  15. lightning_sdk/lightning_cloud/openapi/api/organizations_service_api.py +105 -0
  16. lightning_sdk/lightning_cloud/openapi/api/pipelines_service_api.py +4 -4
  17. lightning_sdk/lightning_cloud/openapi/api/user_service_api.py +105 -0
  18. lightning_sdk/lightning_cloud/openapi/api/volume_service_api.py +258 -0
  19. lightning_sdk/lightning_cloud/openapi/models/__init__.py +16 -0
  20. lightning_sdk/lightning_cloud/openapi/models/agents_id_body.py +27 -1
  21. lightning_sdk/lightning_cloud/openapi/models/assistant_id_conversations_body.py +79 -1
  22. lightning_sdk/lightning_cloud/openapi/models/credits_autoreplenish_body.py +175 -0
  23. lightning_sdk/lightning_cloud/openapi/models/credits_autoreplenish_body1.py +175 -0
  24. lightning_sdk/lightning_cloud/openapi/models/externalv1_user_status.py +53 -1
  25. lightning_sdk/lightning_cloud/openapi/models/orgs_id_body.py +27 -1
  26. lightning_sdk/lightning_cloud/openapi/models/pipelines_id_body1.py +123 -0
  27. lightning_sdk/lightning_cloud/openapi/models/project_id_agents_body.py +27 -1
  28. lightning_sdk/lightning_cloud/openapi/models/project_id_cloudspaces_body.py +27 -1
  29. lightning_sdk/lightning_cloud/openapi/models/project_id_schedules_body.py +27 -1
  30. lightning_sdk/lightning_cloud/openapi/models/schedules_id_body.py +27 -1
  31. lightning_sdk/lightning_cloud/openapi/models/update.py +29 -3
  32. lightning_sdk/lightning_cloud/openapi/models/v1_assistant.py +27 -1
  33. lightning_sdk/lightning_cloud/openapi/models/v1_billing_tier.py +1 -0
  34. lightning_sdk/lightning_cloud/openapi/models/v1_cloud_provider.py +1 -0
  35. lightning_sdk/lightning_cloud/openapi/models/v1_cloud_space.py +27 -1
  36. lightning_sdk/lightning_cloud/openapi/models/v1_cloud_space_environment_config.py +29 -3
  37. lightning_sdk/lightning_cloud/openapi/models/v1_cloud_space_environment_template_config.py +29 -3
  38. lightning_sdk/lightning_cloud/openapi/models/v1_cloud_space_environment_type.py +1 -0
  39. lightning_sdk/lightning_cloud/openapi/models/v1_cloud_space_specialized_view.py +104 -0
  40. lightning_sdk/lightning_cloud/openapi/models/v1_cloudy_expert.py +279 -0
  41. lightning_sdk/lightning_cloud/openapi/models/v1_cluster_accelerator.py +79 -1
  42. lightning_sdk/lightning_cloud/openapi/models/v1_cluster_capacity_reservation.py +53 -1
  43. lightning_sdk/lightning_cloud/openapi/models/v1_cluster_spec.py +27 -1
  44. lightning_sdk/lightning_cloud/openapi/models/v1_cluster_status.py +27 -1
  45. lightning_sdk/lightning_cloud/openapi/models/v1_conversation_response_chunk.py +29 -3
  46. lightning_sdk/lightning_cloud/openapi/models/v1_create_cloud_space_environment_template_request.py +29 -3
  47. lightning_sdk/lightning_cloud/openapi/models/v1_create_organization_request.py +79 -1
  48. lightning_sdk/lightning_cloud/openapi/models/v1_data_connection_tier.py +103 -0
  49. lightning_sdk/lightning_cloud/openapi/models/v1_deployment_status.py +47 -21
  50. lightning_sdk/lightning_cloud/openapi/models/v1_external_cluster_spec.py +27 -1
  51. lightning_sdk/lightning_cloud/openapi/models/v1_filestore_data_connection.py +29 -3
  52. lightning_sdk/lightning_cloud/openapi/models/v1_get_job_stats_response.py +53 -1
  53. lightning_sdk/lightning_cloud/openapi/models/v1_get_user_response.py +27 -1
  54. lightning_sdk/lightning_cloud/openapi/models/v1_get_volume_response.py +123 -0
  55. lightning_sdk/lightning_cloud/openapi/models/v1_instance_overprovisioning_spec.py +1 -27
  56. lightning_sdk/lightning_cloud/openapi/models/v1_kubernetes_direct_v1.py +149 -0
  57. lightning_sdk/lightning_cloud/openapi/models/v1_kubernetes_direct_v1_status.py +149 -0
  58. lightning_sdk/lightning_cloud/openapi/models/v1_list_cloudy_experts_response.py +123 -0
  59. lightning_sdk/lightning_cloud/openapi/models/v1_magic_link_login_response.py +27 -1
  60. lightning_sdk/lightning_cloud/openapi/models/v1_organization.py +27 -1
  61. lightning_sdk/lightning_cloud/openapi/models/v1_rule_resource.py +1 -0
  62. lightning_sdk/lightning_cloud/openapi/models/v1_schedule.py +27 -1
  63. lightning_sdk/lightning_cloud/openapi/models/v1_schedule_action_type.py +104 -0
  64. lightning_sdk/lightning_cloud/openapi/models/v1_schedule_resource_type.py +1 -0
  65. lightning_sdk/lightning_cloud/openapi/models/v1_token_usage.py +175 -0
  66. lightning_sdk/lightning_cloud/openapi/models/v1_update_organization_credits_auto_replenish_response.py +97 -0
  67. lightning_sdk/lightning_cloud/openapi/models/v1_update_user_credits_auto_replenish_response.py +97 -0
  68. lightning_sdk/lightning_cloud/openapi/models/v1_update_volume_response.py +123 -0
  69. lightning_sdk/lightning_cloud/openapi/models/v1_user_features.py +317 -31
  70. lightning_sdk/lightning_cloud/openapi/models/volumes_id_body.py +123 -0
  71. lightning_sdk/llm/llm.py +118 -115
  72. lightning_sdk/llm/public_assistants.json +8 -0
  73. lightning_sdk/pipeline/pipeline.py +17 -2
  74. lightning_sdk/pipeline/printer.py +11 -10
  75. lightning_sdk/pipeline/steps.py +4 -1
  76. lightning_sdk/pipeline/utils.py +29 -4
  77. lightning_sdk/studio.py +3 -0
  78. {lightning_sdk-0.2.23.dist-info → lightning_sdk-0.2.24rc1.dist-info}/METADATA +1 -1
  79. {lightning_sdk-0.2.23.dist-info → lightning_sdk-0.2.24rc1.dist-info}/RECORD +83 -64
  80. {lightning_sdk-0.2.23.dist-info → lightning_sdk-0.2.24rc1.dist-info}/LICENSE +0 -0
  81. {lightning_sdk-0.2.23.dist-info → lightning_sdk-0.2.24rc1.dist-info}/WHEEL +0 -0
  82. {lightning_sdk-0.2.23.dist-info → lightning_sdk-0.2.24rc1.dist-info}/entry_points.txt +0 -0
  83. {lightning_sdk-0.2.23.dist-info → lightning_sdk-0.2.24rc1.dist-info}/top_level.txt +0 -0
@@ -0,0 +1,258 @@
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 VolumeServiceApi(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 volume_service_get_volume(self, project_id: 'str', id: 'str', **kwargs) -> 'V1GetVolumeResponse': # noqa: E501
47
+ """volume_service_get_volume # 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.volume_service_get_volume(project_id, id, async_req=True)
52
+ >>> result = thread.get()
53
+
54
+ :param async_req bool
55
+ :param str project_id: (required)
56
+ :param str id: (required)
57
+ :return: V1GetVolumeResponse
58
+ If the method is called asynchronously,
59
+ returns the request thread.
60
+ """
61
+ kwargs['_return_http_data_only'] = True
62
+ if kwargs.get('async_req'):
63
+ return self.volume_service_get_volume_with_http_info(project_id, id, **kwargs) # noqa: E501
64
+ else:
65
+ (data) = self.volume_service_get_volume_with_http_info(project_id, id, **kwargs) # noqa: E501
66
+ return data
67
+
68
+ def volume_service_get_volume_with_http_info(self, project_id: 'str', id: 'str', **kwargs) -> 'V1GetVolumeResponse': # noqa: E501
69
+ """volume_service_get_volume # noqa: E501
70
+
71
+ This method makes a synchronous HTTP request by default. To make an
72
+ asynchronous HTTP request, please pass async_req=True
73
+ >>> thread = api.volume_service_get_volume_with_http_info(project_id, id, async_req=True)
74
+ >>> result = thread.get()
75
+
76
+ :param async_req bool
77
+ :param str project_id: (required)
78
+ :param str id: (required)
79
+ :return: V1GetVolumeResponse
80
+ If the method is called asynchronously,
81
+ returns the request thread.
82
+ """
83
+
84
+ all_params = ['project_id', 'id'] # noqa: E501
85
+ all_params.append('async_req')
86
+ all_params.append('_return_http_data_only')
87
+ all_params.append('_preload_content')
88
+ all_params.append('_request_timeout')
89
+
90
+ params = locals()
91
+ for key, val in six.iteritems(params['kwargs']):
92
+ if key not in all_params:
93
+ raise TypeError(
94
+ "Got an unexpected keyword argument '%s'"
95
+ " to method volume_service_get_volume" % key
96
+ )
97
+ params[key] = val
98
+ del params['kwargs']
99
+ # verify the required parameter 'project_id' is set
100
+ if ('project_id' not in params or
101
+ params['project_id'] is None):
102
+ raise ValueError("Missing the required parameter `project_id` when calling `volume_service_get_volume`") # noqa: E501
103
+ # verify the required parameter 'id' is set
104
+ if ('id' not in params or
105
+ params['id'] is None):
106
+ raise ValueError("Missing the required parameter `id` when calling `volume_service_get_volume`") # noqa: E501
107
+
108
+ collection_formats = {}
109
+
110
+ path_params = {}
111
+ if 'project_id' in params:
112
+ path_params['projectId'] = params['project_id'] # noqa: E501
113
+ if 'id' in params:
114
+ path_params['id'] = params['id'] # noqa: E501
115
+
116
+ query_params = []
117
+
118
+ header_params = {}
119
+
120
+ form_params = []
121
+ local_var_files = {}
122
+
123
+ body_params = None
124
+ # HTTP header `Accept`
125
+ header_params['Accept'] = self.api_client.select_header_accept(
126
+ ['application/json']) # noqa: E501
127
+
128
+ # Authentication setting
129
+ auth_settings = [] # noqa: E501
130
+
131
+ return self.api_client.call_api(
132
+ '/v1/projects/{projectId}/volumes/{id}', 'GET',
133
+ path_params,
134
+ query_params,
135
+ header_params,
136
+ body=body_params,
137
+ post_params=form_params,
138
+ files=local_var_files,
139
+ response_type='V1GetVolumeResponse', # noqa: E501
140
+ auth_settings=auth_settings,
141
+ async_req=params.get('async_req'),
142
+ _return_http_data_only=params.get('_return_http_data_only'),
143
+ _preload_content=params.get('_preload_content', True),
144
+ _request_timeout=params.get('_request_timeout'),
145
+ collection_formats=collection_formats)
146
+
147
+ def volume_service_update_volume(self, body: 'VolumesIdBody', project_id: 'str', id: 'str', **kwargs) -> 'V1UpdateVolumeResponse': # noqa: E501
148
+ """volume_service_update_volume # noqa: E501
149
+
150
+ This method makes a synchronous HTTP request by default. To make an
151
+ asynchronous HTTP request, please pass async_req=True
152
+ >>> thread = api.volume_service_update_volume(body, project_id, id, async_req=True)
153
+ >>> result = thread.get()
154
+
155
+ :param async_req bool
156
+ :param VolumesIdBody body: (required)
157
+ :param str project_id: (required)
158
+ :param str id: (required)
159
+ :return: V1UpdateVolumeResponse
160
+ If the method is called asynchronously,
161
+ returns the request thread.
162
+ """
163
+ kwargs['_return_http_data_only'] = True
164
+ if kwargs.get('async_req'):
165
+ return self.volume_service_update_volume_with_http_info(body, project_id, id, **kwargs) # noqa: E501
166
+ else:
167
+ (data) = self.volume_service_update_volume_with_http_info(body, project_id, id, **kwargs) # noqa: E501
168
+ return data
169
+
170
+ def volume_service_update_volume_with_http_info(self, body: 'VolumesIdBody', project_id: 'str', id: 'str', **kwargs) -> 'V1UpdateVolumeResponse': # noqa: E501
171
+ """volume_service_update_volume # noqa: E501
172
+
173
+ This method makes a synchronous HTTP request by default. To make an
174
+ asynchronous HTTP request, please pass async_req=True
175
+ >>> thread = api.volume_service_update_volume_with_http_info(body, project_id, id, async_req=True)
176
+ >>> result = thread.get()
177
+
178
+ :param async_req bool
179
+ :param VolumesIdBody body: (required)
180
+ :param str project_id: (required)
181
+ :param str id: (required)
182
+ :return: V1UpdateVolumeResponse
183
+ If the method is called asynchronously,
184
+ returns the request thread.
185
+ """
186
+
187
+ all_params = ['body', 'project_id', 'id'] # noqa: E501
188
+ all_params.append('async_req')
189
+ all_params.append('_return_http_data_only')
190
+ all_params.append('_preload_content')
191
+ all_params.append('_request_timeout')
192
+
193
+ params = locals()
194
+ for key, val in six.iteritems(params['kwargs']):
195
+ if key not in all_params:
196
+ raise TypeError(
197
+ "Got an unexpected keyword argument '%s'"
198
+ " to method volume_service_update_volume" % key
199
+ )
200
+ params[key] = val
201
+ del params['kwargs']
202
+ # verify the required parameter 'body' is set
203
+ if ('body' not in params or
204
+ params['body'] is None):
205
+ raise ValueError("Missing the required parameter `body` when calling `volume_service_update_volume`") # noqa: E501
206
+ # verify the required parameter 'project_id' is set
207
+ if ('project_id' not in params or
208
+ params['project_id'] is None):
209
+ raise ValueError("Missing the required parameter `project_id` when calling `volume_service_update_volume`") # noqa: E501
210
+ # verify the required parameter 'id' is set
211
+ if ('id' not in params or
212
+ params['id'] is None):
213
+ raise ValueError("Missing the required parameter `id` when calling `volume_service_update_volume`") # noqa: E501
214
+
215
+ collection_formats = {}
216
+
217
+ path_params = {}
218
+ if 'project_id' in params:
219
+ path_params['projectId'] = params['project_id'] # noqa: E501
220
+ if 'id' in params:
221
+ path_params['id'] = params['id'] # noqa: E501
222
+
223
+ query_params = []
224
+
225
+ header_params = {}
226
+
227
+ form_params = []
228
+ local_var_files = {}
229
+
230
+ body_params = None
231
+ if 'body' in params:
232
+ body_params = params['body']
233
+ # HTTP header `Accept`
234
+ header_params['Accept'] = self.api_client.select_header_accept(
235
+ ['application/json']) # noqa: E501
236
+
237
+ # HTTP header `Content-Type`
238
+ header_params['Content-Type'] = self.api_client.select_header_content_type( # noqa: E501
239
+ ['application/json']) # noqa: E501
240
+
241
+ # Authentication setting
242
+ auth_settings = [] # noqa: E501
243
+
244
+ return self.api_client.call_api(
245
+ '/v1/projects/{projectId}/volumes/{id}', 'PUT',
246
+ path_params,
247
+ query_params,
248
+ header_params,
249
+ body=body_params,
250
+ post_params=form_params,
251
+ files=local_var_files,
252
+ response_type='V1UpdateVolumeResponse', # noqa: E501
253
+ auth_settings=auth_settings,
254
+ async_req=params.get('async_req'),
255
+ _return_http_data_only=params.get('_return_http_data_only'),
256
+ _preload_content=params.get('_preload_content', True),
257
+ _request_timeout=params.get('_request_timeout'),
258
+ collection_formats=collection_formats)
@@ -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
@@ -343,6 +348,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_data_connection import V1Da
343
348
  from lightning_sdk.lightning_cloud.openapi.models.v1_data_connection_artifact import V1DataConnectionArtifact
344
349
  from lightning_sdk.lightning_cloud.openapi.models.v1_data_connection_mount import V1DataConnectionMount
345
350
  from lightning_sdk.lightning_cloud.openapi.models.v1_data_connection_state import V1DataConnectionState
351
+ from lightning_sdk.lightning_cloud.openapi.models.v1_data_connection_tier import V1DataConnectionTier
346
352
  from lightning_sdk.lightning_cloud.openapi.models.v1_data_path import V1DataPath
347
353
  from lightning_sdk.lightning_cloud.openapi.models.v1_dataset import V1Dataset
348
354
  from lightning_sdk.lightning_cloud.openapi.models.v1_dataset_type import V1DatasetType
@@ -524,6 +530,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_get_user_balance_response i
524
530
  from lightning_sdk.lightning_cloud.openapi.models.v1_get_user_notification_preferences_response import V1GetUserNotificationPreferencesResponse
525
531
  from lightning_sdk.lightning_cloud.openapi.models.v1_get_user_response import V1GetUserResponse
526
532
  from lightning_sdk.lightning_cloud.openapi.models.v1_get_user_storage_breakdown_response import V1GetUserStorageBreakdownResponse
533
+ from lightning_sdk.lightning_cloud.openapi.models.v1_get_volume_response import V1GetVolumeResponse
527
534
  from lightning_sdk.lightning_cloud.openapi.models.v1_git_credentials import V1GitCredentials
528
535
  from lightning_sdk.lightning_cloud.openapi.models.v1_google_cloud_direct_v1 import V1GoogleCloudDirectV1
529
536
  from lightning_sdk.lightning_cloud.openapi.models.v1_google_cloud_direct_v1_status import V1GoogleCloudDirectV1Status
@@ -554,6 +561,8 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_job_type import V1JobType
554
561
  from lightning_sdk.lightning_cloud.openapi.models.v1_joinable_organization import V1JoinableOrganization
555
562
  from lightning_sdk.lightning_cloud.openapi.models.v1_keep_alive_cloud_space_instance_response import V1KeepAliveCloudSpaceInstanceResponse
556
563
  from lightning_sdk.lightning_cloud.openapi.models.v1_knowledge_configuration import V1KnowledgeConfiguration
564
+ from lightning_sdk.lightning_cloud.openapi.models.v1_kubernetes_direct_v1 import V1KubernetesDirectV1
565
+ from lightning_sdk.lightning_cloud.openapi.models.v1_kubernetes_direct_v1_status import V1KubernetesDirectV1Status
557
566
  from lightning_sdk.lightning_cloud.openapi.models.v1_lambda_labs_direct_v1 import V1LambdaLabsDirectV1
558
567
  from lightning_sdk.lightning_cloud.openapi.models.v1_lightning_app_user import V1LightningAppUser
559
568
  from lightning_sdk.lightning_cloud.openapi.models.v1_lightning_auth import V1LightningAuth
@@ -589,6 +598,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_list_cloud_space_tags_respo
589
598
  from lightning_sdk.lightning_cloud.openapi.models.v1_list_cloud_space_version_publications_response import V1ListCloudSpaceVersionPublicationsResponse
590
599
  from lightning_sdk.lightning_cloud.openapi.models.v1_list_cloud_space_versions_response import V1ListCloudSpaceVersionsResponse
591
600
  from lightning_sdk.lightning_cloud.openapi.models.v1_list_cloud_spaces_response import V1ListCloudSpacesResponse
601
+ from lightning_sdk.lightning_cloud.openapi.models.v1_list_cloudy_experts_response import V1ListCloudyExpertsResponse
592
602
  from lightning_sdk.lightning_cloud.openapi.models.v1_list_cluster_accelerators_response import V1ListClusterAcceleratorsResponse
593
603
  from lightning_sdk.lightning_cloud.openapi.models.v1_list_cluster_availabilities_response import V1ListClusterAvailabilitiesResponse
594
604
  from lightning_sdk.lightning_cloud.openapi.models.v1_list_cluster_capacity_reservations_response import V1ListClusterCapacityReservationsResponse
@@ -815,6 +825,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_slurm_job import V1SLURMJob
815
825
  from lightning_sdk.lightning_cloud.openapi.models.v1_ssh_key_pair import V1SSHKeyPair
816
826
  from lightning_sdk.lightning_cloud.openapi.models.v1_ssh_public_key import V1SSHPublicKey
817
827
  from lightning_sdk.lightning_cloud.openapi.models.v1_schedule import V1Schedule
828
+ from lightning_sdk.lightning_cloud.openapi.models.v1_schedule_action_type import V1ScheduleActionType
818
829
  from lightning_sdk.lightning_cloud.openapi.models.v1_schedule_resource_type import V1ScheduleResourceType
819
830
  from lightning_sdk.lightning_cloud.openapi.models.v1_search_job_logs_response import V1SearchJobLogsResponse
820
831
  from lightning_sdk.lightning_cloud.openapi.models.v1_search_user import V1SearchUser
@@ -859,6 +870,7 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_system_metrics_aggregated i
859
870
  from lightning_sdk.lightning_cloud.openapi.models.v1_system_metrics_list import V1SystemMetricsList
860
871
  from lightning_sdk.lightning_cloud.openapi.models.v1_telemetry import V1Telemetry
861
872
  from lightning_sdk.lightning_cloud.openapi.models.v1_timestamp_code_telemetry import V1TimestampCodeTelemetry
873
+ from lightning_sdk.lightning_cloud.openapi.models.v1_token_usage import V1TokenUsage
862
874
  from lightning_sdk.lightning_cloud.openapi.models.v1_transaction import V1Transaction
863
875
  from lightning_sdk.lightning_cloud.openapi.models.v1_transfer_cloud_space_response import V1TransferCloudSpaceResponse
864
876
  from lightning_sdk.lightning_cloud.openapi.models.v1_transfer_org_balance_response import V1TransferOrgBalanceResponse
@@ -884,12 +896,15 @@ from lightning_sdk.lightning_cloud.openapi.models.v1_update_lit_page_response im
884
896
  from lightning_sdk.lightning_cloud.openapi.models.v1_update_lit_repository_response import V1UpdateLitRepositoryResponse
885
897
  from lightning_sdk.lightning_cloud.openapi.models.v1_update_metrics_stream_visibility_response import V1UpdateMetricsStreamVisibilityResponse
886
898
  from lightning_sdk.lightning_cloud.openapi.models.v1_update_model_visibility_response import V1UpdateModelVisibilityResponse
899
+ from lightning_sdk.lightning_cloud.openapi.models.v1_update_organization_credits_auto_replenish_response import V1UpdateOrganizationCreditsAutoReplenishResponse
887
900
  from lightning_sdk.lightning_cloud.openapi.models.v1_update_project_cluster_accelerators_response import V1UpdateProjectClusterAcceleratorsResponse
888
901
  from lightning_sdk.lightning_cloud.openapi.models.v1_update_shared_metrics_stream_response import V1UpdateSharedMetricsStreamResponse
889
902
  from lightning_sdk.lightning_cloud.openapi.models.v1_update_snowflake_query_response import V1UpdateSnowflakeQueryResponse
903
+ from lightning_sdk.lightning_cloud.openapi.models.v1_update_user_credits_auto_replenish_response import V1UpdateUserCreditsAutoReplenishResponse
890
904
  from lightning_sdk.lightning_cloud.openapi.models.v1_update_user_request import V1UpdateUserRequest
891
905
  from lightning_sdk.lightning_cloud.openapi.models.v1_update_user_viewed_new_features_request import V1UpdateUserViewedNewFeaturesRequest
892
906
  from lightning_sdk.lightning_cloud.openapi.models.v1_update_user_viewed_new_features_response import V1UpdateUserViewedNewFeaturesResponse
907
+ from lightning_sdk.lightning_cloud.openapi.models.v1_update_volume_response import V1UpdateVolumeResponse
893
908
  from lightning_sdk.lightning_cloud.openapi.models.v1_upload_agent_job_artifact_response import V1UploadAgentJobArtifactResponse
894
909
  from lightning_sdk.lightning_cloud.openapi.models.v1_upload_agent_job_output_response import V1UploadAgentJobOutputResponse
895
910
  from lightning_sdk.lightning_cloud.openapi.models.v1_upload_lightningapp_instance_artifact_response import V1UploadLightningappInstanceArtifactResponse
@@ -934,4 +949,5 @@ from lightning_sdk.lightning_cloud.openapi.models.version_uploads_body1 import V
934
949
  from lightning_sdk.lightning_cloud.openapi.models.versions_id_body import VersionsIdBody
935
950
  from lightning_sdk.lightning_cloud.openapi.models.versions_version_body import VersionsVersionBody
936
951
  from lightning_sdk.lightning_cloud.openapi.models.versions_version_body1 import VersionsVersionBody1
952
+ from lightning_sdk.lightning_cloud.openapi.models.volumes_id_body import VolumesIdBody
937
953
  from lightning_sdk.lightning_cloud.openapi.models.works_id_body import WorksIdBody
@@ -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,15 @@ 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',
53
+ 'parent_conversation_id': 'str',
52
54
  'parent_message_id': 'str',
55
+ 'reasoning_effort': 'str',
53
56
  'store': 'bool',
54
57
  'stream': 'bool',
55
58
  'system_prompt': 'str'
@@ -59,28 +62,34 @@ class AssistantIdConversationsBody(object):
59
62
  'auto_name': 'autoName',
60
63
  'billing_project_id': 'billingProjectId',
61
64
  'conversation_id': 'conversationId',
65
+ 'ephemeral': 'ephemeral',
62
66
  'internal_conversation': 'internalConversation',
63
67
  'max_tokens': 'maxTokens',
64
68
  'message': 'message',
65
69
  'metadata': 'metadata',
66
70
  'name': 'name',
71
+ 'parent_conversation_id': 'parentConversationId',
67
72
  'parent_message_id': 'parentMessageId',
73
+ 'reasoning_effort': 'reasoningEffort',
68
74
  'store': 'store',
69
75
  'stream': 'stream',
70
76
  'system_prompt': 'systemPrompt'
71
77
  }
72
78
 
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
79
+ 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_conversation_id: 'str' =None, parent_message_id: 'str' =None, reasoning_effort: 'str' =None, store: 'bool' =None, stream: 'bool' =None, system_prompt: 'str' =None): # noqa: E501
74
80
  """AssistantIdConversationsBody - a model defined in Swagger""" # noqa: E501
75
81
  self._auto_name = None
76
82
  self._billing_project_id = None
77
83
  self._conversation_id = None
84
+ self._ephemeral = None
78
85
  self._internal_conversation = None
79
86
  self._max_tokens = None
80
87
  self._message = None
81
88
  self._metadata = None
82
89
  self._name = None
90
+ self._parent_conversation_id = None
83
91
  self._parent_message_id = None
92
+ self._reasoning_effort = None
84
93
  self._store = None
85
94
  self._stream = None
86
95
  self._system_prompt = None
@@ -91,6 +100,8 @@ class AssistantIdConversationsBody(object):
91
100
  self.billing_project_id = billing_project_id
92
101
  if conversation_id is not None:
93
102
  self.conversation_id = conversation_id
103
+ if ephemeral is not None:
104
+ self.ephemeral = ephemeral
94
105
  if internal_conversation is not None:
95
106
  self.internal_conversation = internal_conversation
96
107
  if max_tokens is not None:
@@ -101,8 +112,12 @@ class AssistantIdConversationsBody(object):
101
112
  self.metadata = metadata
102
113
  if name is not None:
103
114
  self.name = name
115
+ if parent_conversation_id is not None:
116
+ self.parent_conversation_id = parent_conversation_id
104
117
  if parent_message_id is not None:
105
118
  self.parent_message_id = parent_message_id
119
+ if reasoning_effort is not None:
120
+ self.reasoning_effort = reasoning_effort
106
121
  if store is not None:
107
122
  self.store = store
108
123
  if stream is not None:
@@ -173,6 +188,27 @@ class AssistantIdConversationsBody(object):
173
188
 
174
189
  self._conversation_id = conversation_id
175
190
 
191
+ @property
192
+ def ephemeral(self) -> 'bool':
193
+ """Gets the ephemeral of this AssistantIdConversationsBody. # noqa: E501
194
+
195
+
196
+ :return: The ephemeral of this AssistantIdConversationsBody. # noqa: E501
197
+ :rtype: bool
198
+ """
199
+ return self._ephemeral
200
+
201
+ @ephemeral.setter
202
+ def ephemeral(self, ephemeral: 'bool'):
203
+ """Sets the ephemeral of this AssistantIdConversationsBody.
204
+
205
+
206
+ :param ephemeral: The ephemeral of this AssistantIdConversationsBody. # noqa: E501
207
+ :type: bool
208
+ """
209
+
210
+ self._ephemeral = ephemeral
211
+
176
212
  @property
177
213
  def internal_conversation(self) -> 'bool':
178
214
  """Gets the internal_conversation of this AssistantIdConversationsBody. # noqa: E501
@@ -278,6 +314,27 @@ class AssistantIdConversationsBody(object):
278
314
 
279
315
  self._name = name
280
316
 
317
+ @property
318
+ def parent_conversation_id(self) -> 'str':
319
+ """Gets the parent_conversation_id of this AssistantIdConversationsBody. # noqa: E501
320
+
321
+
322
+ :return: The parent_conversation_id of this AssistantIdConversationsBody. # noqa: E501
323
+ :rtype: str
324
+ """
325
+ return self._parent_conversation_id
326
+
327
+ @parent_conversation_id.setter
328
+ def parent_conversation_id(self, parent_conversation_id: 'str'):
329
+ """Sets the parent_conversation_id of this AssistantIdConversationsBody.
330
+
331
+
332
+ :param parent_conversation_id: The parent_conversation_id of this AssistantIdConversationsBody. # noqa: E501
333
+ :type: str
334
+ """
335
+
336
+ self._parent_conversation_id = parent_conversation_id
337
+
281
338
  @property
282
339
  def parent_message_id(self) -> 'str':
283
340
  """Gets the parent_message_id of this AssistantIdConversationsBody. # noqa: E501
@@ -299,6 +356,27 @@ class AssistantIdConversationsBody(object):
299
356
 
300
357
  self._parent_message_id = parent_message_id
301
358
 
359
+ @property
360
+ def reasoning_effort(self) -> 'str':
361
+ """Gets the reasoning_effort of this AssistantIdConversationsBody. # noqa: E501
362
+
363
+
364
+ :return: The reasoning_effort of this AssistantIdConversationsBody. # noqa: E501
365
+ :rtype: str
366
+ """
367
+ return self._reasoning_effort
368
+
369
+ @reasoning_effort.setter
370
+ def reasoning_effort(self, reasoning_effort: 'str'):
371
+ """Sets the reasoning_effort of this AssistantIdConversationsBody.
372
+
373
+
374
+ :param reasoning_effort: The reasoning_effort of this AssistantIdConversationsBody. # noqa: E501
375
+ :type: str
376
+ """
377
+
378
+ self._reasoning_effort = reasoning_effort
379
+
302
380
  @property
303
381
  def store(self) -> 'bool':
304
382
  """Gets the store of this AssistantIdConversationsBody. # noqa: E501