lightning-sdk 2025.7.17__py3-none-any.whl → 2025.7.30rc0__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 (99) hide show
  1. lightning_sdk/__init__.py +3 -2
  2. lightning_sdk/api/cloud_account_api.py +204 -0
  3. lightning_sdk/api/deployment_api.py +11 -0
  4. lightning_sdk/api/job_api.py +82 -10
  5. lightning_sdk/api/llm_api.py +1 -1
  6. lightning_sdk/api/mmt_api.py +44 -5
  7. lightning_sdk/api/pipeline_api.py +4 -3
  8. lightning_sdk/api/studio_api.py +51 -8
  9. lightning_sdk/api/utils.py +6 -2
  10. lightning_sdk/cli/clusters_menu.py +3 -3
  11. lightning_sdk/cli/create.py +25 -11
  12. lightning_sdk/cli/deploy/_auth.py +19 -3
  13. lightning_sdk/cli/deploy/serve.py +21 -5
  14. lightning_sdk/cli/download.py +25 -1
  15. lightning_sdk/cli/entrypoint.py +4 -2
  16. lightning_sdk/cli/list.py +5 -1
  17. lightning_sdk/cli/run.py +3 -1
  18. lightning_sdk/cli/start.py +40 -8
  19. lightning_sdk/cli/switch.py +3 -1
  20. lightning_sdk/deployment/deployment.py +8 -0
  21. lightning_sdk/job/base.py +27 -3
  22. lightning_sdk/job/job.py +28 -4
  23. lightning_sdk/job/v1.py +10 -1
  24. lightning_sdk/job/v2.py +22 -2
  25. lightning_sdk/job/work.py +5 -1
  26. lightning_sdk/lightning_cloud/openapi/__init__.py +14 -1
  27. lightning_sdk/lightning_cloud/openapi/api/assistants_service_api.py +428 -0
  28. lightning_sdk/lightning_cloud/openapi/api/billing_service_api.py +153 -48
  29. lightning_sdk/lightning_cloud/openapi/api/cloudy_service_api.py +295 -0
  30. lightning_sdk/lightning_cloud/openapi/api/cluster_service_api.py +93 -0
  31. lightning_sdk/lightning_cloud/openapi/models/__init__.py +14 -1
  32. lightning_sdk/lightning_cloud/openapi/models/agentmanagedendpoints_id_body.py +27 -1
  33. lightning_sdk/lightning_cloud/openapi/models/blogposts_id_body.py +53 -1
  34. lightning_sdk/lightning_cloud/openapi/models/conversations_id_body1.py +123 -0
  35. lightning_sdk/lightning_cloud/openapi/models/messages_id_body.py +123 -0
  36. lightning_sdk/lightning_cloud/openapi/models/metricsstream_id_body.py +27 -1
  37. lightning_sdk/lightning_cloud/openapi/models/project_id_schedules_body.py +81 -3
  38. lightning_sdk/lightning_cloud/openapi/models/schedules_id_body.py +79 -1
  39. lightning_sdk/lightning_cloud/openapi/models/user_id_upgradetrigger_body.py +201 -0
  40. lightning_sdk/lightning_cloud/openapi/models/user_user_id_body.py +201 -0
  41. lightning_sdk/lightning_cloud/openapi/models/v1_billing_subscription.py +27 -1
  42. lightning_sdk/lightning_cloud/openapi/models/v1_blog_post.py +53 -1
  43. lightning_sdk/lightning_cloud/openapi/models/v1_cloudy_settings.py +227 -0
  44. lightning_sdk/lightning_cloud/openapi/models/v1_cluster_spec.py +27 -1
  45. lightning_sdk/lightning_cloud/openapi/models/v1_conversation.py +27 -1
  46. lightning_sdk/lightning_cloud/openapi/models/v1_conversation_response_chunk.py +27 -1
  47. lightning_sdk/lightning_cloud/openapi/models/v1_create_billing_upgrade_trigger_record_response.py +97 -0
  48. lightning_sdk/lightning_cloud/openapi/models/v1_create_blog_post_request.py +53 -1
  49. lightning_sdk/lightning_cloud/openapi/models/v1_create_checkout_session_request.py +27 -1
  50. lightning_sdk/lightning_cloud/openapi/models/v1_create_subscription_checkout_session_request.py +55 -3
  51. lightning_sdk/lightning_cloud/openapi/models/v1_function_call.py +149 -0
  52. lightning_sdk/lightning_cloud/openapi/models/{v1_get_clickhouse_assistant_session_daily_aggregated_response.py → v1_get_assistant_session_daily_aggregated_response.py} +22 -22
  53. lightning_sdk/lightning_cloud/openapi/models/v1_get_cluster_health_response.py +149 -0
  54. lightning_sdk/lightning_cloud/openapi/models/v1_get_user_response.py +27 -1
  55. lightning_sdk/lightning_cloud/openapi/models/v1_job_spec.py +27 -1
  56. lightning_sdk/lightning_cloud/openapi/models/v1_kubernetes_direct_v1.py +105 -1
  57. lightning_sdk/lightning_cloud/openapi/models/v1_like_status.py +104 -0
  58. lightning_sdk/lightning_cloud/openapi/models/v1_list_published_managed_endpoints_response.py +123 -0
  59. lightning_sdk/lightning_cloud/openapi/models/v1_managed_endpoint.py +27 -1
  60. lightning_sdk/lightning_cloud/openapi/models/v1_managed_model.py +95 -17
  61. lightning_sdk/lightning_cloud/openapi/models/v1_message.py +27 -1
  62. lightning_sdk/lightning_cloud/openapi/models/v1_quote_subscription_response.py +27 -1
  63. lightning_sdk/lightning_cloud/openapi/models/v1_resource_visibility.py +27 -1
  64. lightning_sdk/lightning_cloud/openapi/models/v1_response_choice.py +29 -3
  65. lightning_sdk/lightning_cloud/openapi/models/v1_schedule.py +79 -1
  66. lightning_sdk/lightning_cloud/openapi/models/v1_service_health.py +27 -1
  67. lightning_sdk/lightning_cloud/openapi/models/v1_slurm_v1.py +79 -1
  68. lightning_sdk/lightning_cloud/openapi/models/v1_slurm_v1_status.py +79 -1
  69. lightning_sdk/lightning_cloud/openapi/models/v1_tool_call.py +175 -0
  70. lightning_sdk/lightning_cloud/openapi/models/v1_update_conversation_like_response.py +149 -0
  71. lightning_sdk/lightning_cloud/openapi/models/v1_update_conversation_message_like_response.py +149 -0
  72. lightning_sdk/lightning_cloud/openapi/models/v1_user_features.py +79 -313
  73. lightning_sdk/lightning_cloud/openapi/models/v1_volume_state.py +1 -0
  74. lightning_sdk/llm/llm.py +69 -11
  75. lightning_sdk/llm/public_assistants.json +32 -8
  76. lightning_sdk/machine.py +151 -43
  77. lightning_sdk/mmt/base.py +20 -2
  78. lightning_sdk/mmt/mmt.py +25 -3
  79. lightning_sdk/mmt/v1.py +7 -1
  80. lightning_sdk/mmt/v2.py +27 -3
  81. lightning_sdk/models.py +1 -1
  82. lightning_sdk/organization.py +4 -0
  83. lightning_sdk/pipeline/pipeline.py +16 -5
  84. lightning_sdk/pipeline/printer.py +5 -3
  85. lightning_sdk/pipeline/schedule.py +844 -1
  86. lightning_sdk/pipeline/steps.py +19 -4
  87. lightning_sdk/sandbox.py +4 -1
  88. lightning_sdk/serve.py +2 -0
  89. lightning_sdk/studio.py +91 -44
  90. lightning_sdk/teamspace.py +19 -10
  91. lightning_sdk/utils/resolve.py +37 -2
  92. {lightning_sdk-2025.7.17.dist-info → lightning_sdk-2025.7.30rc0.dist-info}/METADATA +7 -5
  93. {lightning_sdk-2025.7.17.dist-info → lightning_sdk-2025.7.30rc0.dist-info}/RECORD +98 -85
  94. lightning_sdk/api/cluster_api.py +0 -119
  95. /lightning_sdk/cli/{inspect.py → inspection.py} +0 -0
  96. {lightning_sdk-2025.7.17.dist-info → lightning_sdk-2025.7.30rc0.dist-info}/LICENSE +0 -0
  97. {lightning_sdk-2025.7.17.dist-info → lightning_sdk-2025.7.30rc0.dist-info}/WHEEL +0 -0
  98. {lightning_sdk-2025.7.17.dist-info → lightning_sdk-2025.7.30rc0.dist-info}/entry_points.txt +0 -0
  99. {lightning_sdk-2025.7.17.dist-info → lightning_sdk-2025.7.30rc0.dist-info}/top_level.txt +0 -0
@@ -0,0 +1,123 @@
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
+ import pprint
21
+ import re # noqa: F401
22
+
23
+ from typing import TYPE_CHECKING
24
+
25
+ import six
26
+
27
+ if TYPE_CHECKING:
28
+ from datetime import datetime
29
+ from lightning_sdk.lightning_cloud.openapi.models import *
30
+
31
+ class V1ListPublishedManagedEndpointsResponse(object):
32
+ """NOTE: This class is auto generated by the swagger code generator program.
33
+
34
+ Do not edit the class manually.
35
+ """
36
+ """
37
+ Attributes:
38
+ swagger_types (dict): The key is attribute name
39
+ and the value is attribute type.
40
+ attribute_map (dict): The key is attribute name
41
+ and the value is json key in definition.
42
+ """
43
+ swagger_types = {
44
+ 'managed_endpoints': 'list[V1ManagedEndpoint]'
45
+ }
46
+
47
+ attribute_map = {
48
+ 'managed_endpoints': 'managedEndpoints'
49
+ }
50
+
51
+ def __init__(self, managed_endpoints: 'list[V1ManagedEndpoint]' =None): # noqa: E501
52
+ """V1ListPublishedManagedEndpointsResponse - a model defined in Swagger""" # noqa: E501
53
+ self._managed_endpoints = None
54
+ self.discriminator = None
55
+ if managed_endpoints is not None:
56
+ self.managed_endpoints = managed_endpoints
57
+
58
+ @property
59
+ def managed_endpoints(self) -> 'list[V1ManagedEndpoint]':
60
+ """Gets the managed_endpoints of this V1ListPublishedManagedEndpointsResponse. # noqa: E501
61
+
62
+
63
+ :return: The managed_endpoints of this V1ListPublishedManagedEndpointsResponse. # noqa: E501
64
+ :rtype: list[V1ManagedEndpoint]
65
+ """
66
+ return self._managed_endpoints
67
+
68
+ @managed_endpoints.setter
69
+ def managed_endpoints(self, managed_endpoints: 'list[V1ManagedEndpoint]'):
70
+ """Sets the managed_endpoints of this V1ListPublishedManagedEndpointsResponse.
71
+
72
+
73
+ :param managed_endpoints: The managed_endpoints of this V1ListPublishedManagedEndpointsResponse. # noqa: E501
74
+ :type: list[V1ManagedEndpoint]
75
+ """
76
+
77
+ self._managed_endpoints = managed_endpoints
78
+
79
+ def to_dict(self) -> dict:
80
+ """Returns the model properties as a dict"""
81
+ result = {}
82
+
83
+ for attr, _ in six.iteritems(self.swagger_types):
84
+ value = getattr(self, attr)
85
+ if isinstance(value, list):
86
+ result[attr] = list(map(
87
+ lambda x: x.to_dict() if hasattr(x, "to_dict") else x,
88
+ value
89
+ ))
90
+ elif hasattr(value, "to_dict"):
91
+ result[attr] = value.to_dict()
92
+ elif isinstance(value, dict):
93
+ result[attr] = dict(map(
94
+ lambda item: (item[0], item[1].to_dict())
95
+ if hasattr(item[1], "to_dict") else item,
96
+ value.items()
97
+ ))
98
+ else:
99
+ result[attr] = value
100
+ if issubclass(V1ListPublishedManagedEndpointsResponse, dict):
101
+ for key, value in self.items():
102
+ result[key] = value
103
+
104
+ return result
105
+
106
+ def to_str(self) -> str:
107
+ """Returns the string representation of the model"""
108
+ return pprint.pformat(self.to_dict())
109
+
110
+ def __repr__(self) -> str:
111
+ """For `print` and `pprint`"""
112
+ return self.to_str()
113
+
114
+ def __eq__(self, other: 'V1ListPublishedManagedEndpointsResponse') -> bool:
115
+ """Returns true if both objects are equal"""
116
+ if not isinstance(other, V1ListPublishedManagedEndpointsResponse):
117
+ return False
118
+
119
+ return self.__dict__ == other.__dict__
120
+
121
+ def __ne__(self, other: 'V1ListPublishedManagedEndpointsResponse') -> bool:
122
+ """Returns true if both objects are not equal"""
123
+ return not self == other
@@ -49,6 +49,7 @@ class V1ManagedEndpoint(object):
49
49
  'name': 'str',
50
50
  'org_id': 'str',
51
51
  'project_id': 'str',
52
+ 'provider_display_name': 'str',
52
53
  'publish_status': 'str',
53
54
  'user_id': 'str'
54
55
  }
@@ -62,11 +63,12 @@ class V1ManagedEndpoint(object):
62
63
  'name': 'name',
63
64
  'org_id': 'orgId',
64
65
  'project_id': 'projectId',
66
+ 'provider_display_name': 'providerDisplayName',
65
67
  'publish_status': 'publishStatus',
66
68
  'user_id': 'userId'
67
69
  }
68
70
 
69
- def __init__(self, api_key: 'str' =None, base_url: 'str' =None, description: 'str' =None, id: 'str' =None, models_metadata: 'list[V1ManagedModel]' =None, name: 'str' =None, org_id: 'str' =None, project_id: 'str' =None, publish_status: 'str' =None, user_id: 'str' =None): # noqa: E501
71
+ def __init__(self, api_key: 'str' =None, base_url: 'str' =None, description: 'str' =None, id: 'str' =None, models_metadata: 'list[V1ManagedModel]' =None, name: 'str' =None, org_id: 'str' =None, project_id: 'str' =None, provider_display_name: 'str' =None, publish_status: 'str' =None, user_id: 'str' =None): # noqa: E501
70
72
  """V1ManagedEndpoint - a model defined in Swagger""" # noqa: E501
71
73
  self._api_key = None
72
74
  self._base_url = None
@@ -76,6 +78,7 @@ class V1ManagedEndpoint(object):
76
78
  self._name = None
77
79
  self._org_id = None
78
80
  self._project_id = None
81
+ self._provider_display_name = None
79
82
  self._publish_status = None
80
83
  self._user_id = None
81
84
  self.discriminator = None
@@ -95,6 +98,8 @@ class V1ManagedEndpoint(object):
95
98
  self.org_id = org_id
96
99
  if project_id is not None:
97
100
  self.project_id = project_id
101
+ if provider_display_name is not None:
102
+ self.provider_display_name = provider_display_name
98
103
  if publish_status is not None:
99
104
  self.publish_status = publish_status
100
105
  if user_id is not None:
@@ -268,6 +273,27 @@ class V1ManagedEndpoint(object):
268
273
 
269
274
  self._project_id = project_id
270
275
 
276
+ @property
277
+ def provider_display_name(self) -> 'str':
278
+ """Gets the provider_display_name of this V1ManagedEndpoint. # noqa: E501
279
+
280
+
281
+ :return: The provider_display_name of this V1ManagedEndpoint. # noqa: E501
282
+ :rtype: str
283
+ """
284
+ return self._provider_display_name
285
+
286
+ @provider_display_name.setter
287
+ def provider_display_name(self, provider_display_name: 'str'):
288
+ """Sets the provider_display_name of this V1ManagedEndpoint.
289
+
290
+
291
+ :param provider_display_name: The provider_display_name of this V1ManagedEndpoint. # noqa: E501
292
+ :type: str
293
+ """
294
+
295
+ self._provider_display_name = provider_display_name
296
+
271
297
  @property
272
298
  def publish_status(self) -> 'str':
273
299
  """Gets the publish_status of this V1ManagedEndpoint. # noqa: E501
@@ -42,11 +42,12 @@ class V1ManagedModel(object):
42
42
  """
43
43
  swagger_types = {
44
44
  'abilities': 'V1ManagedModelAbilities',
45
+ 'assistant_id': 'str',
45
46
  'completion_token_price': 'float',
46
47
  'context_length': 'str',
47
48
  'deployment_details': 'V1DeploymentDetails',
48
49
  'description': 'str',
49
- 'disabled': 'bool',
50
+ 'display_name': 'str',
50
51
  'endpoint_id': 'str',
51
52
  'id': 'str',
52
53
  'max_completion_tokens': 'str',
@@ -54,16 +55,19 @@ class V1ManagedModel(object):
54
55
  'prompt_token_price': 'float',
55
56
  'status': 'V1AssistantModelStatus',
56
57
  'temperature': 'float',
58
+ 'throughput': 'float',
59
+ 'time_to_first_token': 'float',
57
60
  'top_k': 'str'
58
61
  }
59
62
 
60
63
  attribute_map = {
61
64
  'abilities': 'abilities',
65
+ 'assistant_id': 'assistantId',
62
66
  'completion_token_price': 'completionTokenPrice',
63
67
  'context_length': 'contextLength',
64
68
  'deployment_details': 'deploymentDetails',
65
69
  'description': 'description',
66
- 'disabled': 'disabled',
70
+ 'display_name': 'displayName',
67
71
  'endpoint_id': 'endpointId',
68
72
  'id': 'id',
69
73
  'max_completion_tokens': 'maxCompletionTokens',
@@ -71,17 +75,20 @@ class V1ManagedModel(object):
71
75
  'prompt_token_price': 'promptTokenPrice',
72
76
  'status': 'status',
73
77
  'temperature': 'temperature',
78
+ 'throughput': 'throughput',
79
+ 'time_to_first_token': 'timeToFirstToken',
74
80
  'top_k': 'topK'
75
81
  }
76
82
 
77
- def __init__(self, abilities: 'V1ManagedModelAbilities' =None, completion_token_price: 'float' =None, context_length: 'str' =None, deployment_details: 'V1DeploymentDetails' =None, description: 'str' =None, disabled: 'bool' =None, endpoint_id: 'str' =None, id: 'str' =None, max_completion_tokens: 'str' =None, name: 'str' =None, prompt_token_price: 'float' =None, status: 'V1AssistantModelStatus' =None, temperature: 'float' =None, top_k: 'str' =None): # noqa: E501
83
+ def __init__(self, abilities: 'V1ManagedModelAbilities' =None, assistant_id: 'str' =None, completion_token_price: 'float' =None, context_length: 'str' =None, deployment_details: 'V1DeploymentDetails' =None, description: 'str' =None, display_name: 'str' =None, endpoint_id: 'str' =None, id: 'str' =None, max_completion_tokens: 'str' =None, name: 'str' =None, prompt_token_price: 'float' =None, status: 'V1AssistantModelStatus' =None, temperature: 'float' =None, throughput: 'float' =None, time_to_first_token: 'float' =None, top_k: 'str' =None): # noqa: E501
78
84
  """V1ManagedModel - a model defined in Swagger""" # noqa: E501
79
85
  self._abilities = None
86
+ self._assistant_id = None
80
87
  self._completion_token_price = None
81
88
  self._context_length = None
82
89
  self._deployment_details = None
83
90
  self._description = None
84
- self._disabled = None
91
+ self._display_name = None
85
92
  self._endpoint_id = None
86
93
  self._id = None
87
94
  self._max_completion_tokens = None
@@ -89,10 +96,14 @@ class V1ManagedModel(object):
89
96
  self._prompt_token_price = None
90
97
  self._status = None
91
98
  self._temperature = None
99
+ self._throughput = None
100
+ self._time_to_first_token = None
92
101
  self._top_k = None
93
102
  self.discriminator = None
94
103
  if abilities is not None:
95
104
  self.abilities = abilities
105
+ if assistant_id is not None:
106
+ self.assistant_id = assistant_id
96
107
  if completion_token_price is not None:
97
108
  self.completion_token_price = completion_token_price
98
109
  if context_length is not None:
@@ -101,8 +112,8 @@ class V1ManagedModel(object):
101
112
  self.deployment_details = deployment_details
102
113
  if description is not None:
103
114
  self.description = description
104
- if disabled is not None:
105
- self.disabled = disabled
115
+ if display_name is not None:
116
+ self.display_name = display_name
106
117
  if endpoint_id is not None:
107
118
  self.endpoint_id = endpoint_id
108
119
  if id is not None:
@@ -117,6 +128,10 @@ class V1ManagedModel(object):
117
128
  self.status = status
118
129
  if temperature is not None:
119
130
  self.temperature = temperature
131
+ if throughput is not None:
132
+ self.throughput = throughput
133
+ if time_to_first_token is not None:
134
+ self.time_to_first_token = time_to_first_token
120
135
  if top_k is not None:
121
136
  self.top_k = top_k
122
137
 
@@ -141,6 +156,27 @@ class V1ManagedModel(object):
141
156
 
142
157
  self._abilities = abilities
143
158
 
159
+ @property
160
+ def assistant_id(self) -> 'str':
161
+ """Gets the assistant_id of this V1ManagedModel. # noqa: E501
162
+
163
+
164
+ :return: The assistant_id of this V1ManagedModel. # noqa: E501
165
+ :rtype: str
166
+ """
167
+ return self._assistant_id
168
+
169
+ @assistant_id.setter
170
+ def assistant_id(self, assistant_id: 'str'):
171
+ """Sets the assistant_id of this V1ManagedModel.
172
+
173
+
174
+ :param assistant_id: The assistant_id of this V1ManagedModel. # noqa: E501
175
+ :type: str
176
+ """
177
+
178
+ self._assistant_id = assistant_id
179
+
144
180
  @property
145
181
  def completion_token_price(self) -> 'float':
146
182
  """Gets the completion_token_price of this V1ManagedModel. # noqa: E501
@@ -226,25 +262,25 @@ class V1ManagedModel(object):
226
262
  self._description = description
227
263
 
228
264
  @property
229
- def disabled(self) -> 'bool':
230
- """Gets the disabled of this V1ManagedModel. # noqa: E501
265
+ def display_name(self) -> 'str':
266
+ """Gets the display_name of this V1ManagedModel. # noqa: E501
231
267
 
232
268
 
233
- :return: The disabled of this V1ManagedModel. # noqa: E501
234
- :rtype: bool
269
+ :return: The display_name of this V1ManagedModel. # noqa: E501
270
+ :rtype: str
235
271
  """
236
- return self._disabled
272
+ return self._display_name
237
273
 
238
- @disabled.setter
239
- def disabled(self, disabled: 'bool'):
240
- """Sets the disabled of this V1ManagedModel.
274
+ @display_name.setter
275
+ def display_name(self, display_name: 'str'):
276
+ """Sets the display_name of this V1ManagedModel.
241
277
 
242
278
 
243
- :param disabled: The disabled of this V1ManagedModel. # noqa: E501
244
- :type: bool
279
+ :param display_name: The display_name of this V1ManagedModel. # noqa: E501
280
+ :type: str
245
281
  """
246
282
 
247
- self._disabled = disabled
283
+ self._display_name = display_name
248
284
 
249
285
  @property
250
286
  def endpoint_id(self) -> 'str':
@@ -393,6 +429,48 @@ class V1ManagedModel(object):
393
429
 
394
430
  self._temperature = temperature
395
431
 
432
+ @property
433
+ def throughput(self) -> 'float':
434
+ """Gets the throughput of this V1ManagedModel. # noqa: E501
435
+
436
+
437
+ :return: The throughput of this V1ManagedModel. # noqa: E501
438
+ :rtype: float
439
+ """
440
+ return self._throughput
441
+
442
+ @throughput.setter
443
+ def throughput(self, throughput: 'float'):
444
+ """Sets the throughput of this V1ManagedModel.
445
+
446
+
447
+ :param throughput: The throughput of this V1ManagedModel. # noqa: E501
448
+ :type: float
449
+ """
450
+
451
+ self._throughput = throughput
452
+
453
+ @property
454
+ def time_to_first_token(self) -> 'float':
455
+ """Gets the time_to_first_token of this V1ManagedModel. # noqa: E501
456
+
457
+
458
+ :return: The time_to_first_token of this V1ManagedModel. # noqa: E501
459
+ :rtype: float
460
+ """
461
+ return self._time_to_first_token
462
+
463
+ @time_to_first_token.setter
464
+ def time_to_first_token(self, time_to_first_token: 'float'):
465
+ """Sets the time_to_first_token of this V1ManagedModel.
466
+
467
+
468
+ :param time_to_first_token: The time_to_first_token of this V1ManagedModel. # noqa: E501
469
+ :type: float
470
+ """
471
+
472
+ self._time_to_first_token = time_to_first_token
473
+
396
474
  @property
397
475
  def top_k(self) -> 'str':
398
476
  """Gets the top_k of this V1ManagedModel. # noqa: E501
@@ -51,6 +51,7 @@ class V1Message(object):
51
51
  'executable': 'bool',
52
52
  'id': 'str',
53
53
  'internal_message': 'bool',
54
+ 'like': 'V1LikeStatus',
54
55
  'metadata': 'dict(str, str)',
55
56
  'model': 'str',
56
57
  'parent_message_id': 'str',
@@ -69,6 +70,7 @@ class V1Message(object):
69
70
  'executable': 'executable',
70
71
  'id': 'id',
71
72
  'internal_message': 'internalMessage',
73
+ 'like': 'like',
72
74
  'metadata': 'metadata',
73
75
  'model': 'model',
74
76
  'parent_message_id': 'parentMessageId',
@@ -76,7 +78,7 @@ class V1Message(object):
76
78
  'throughput': 'throughput'
77
79
  }
78
80
 
79
- def __init__(self, account_id: 'str' =None, assistant_id: 'str' =None, author: 'V1MessageAuthor' =None, completion_tokens: 'str' =None, content: 'list[V1MessageContent]' =None, conversation_id: 'str' =None, created_at: 'datetime' =None, executable: 'bool' =None, id: 'str' =None, internal_message: 'bool' =None, metadata: 'dict(str, str)' =None, model: 'str' =None, parent_message_id: 'str' =None, prompt_tokens: 'str' =None, throughput: 'float' =None): # noqa: E501
81
+ def __init__(self, account_id: 'str' =None, assistant_id: 'str' =None, author: 'V1MessageAuthor' =None, completion_tokens: 'str' =None, content: 'list[V1MessageContent]' =None, conversation_id: 'str' =None, created_at: 'datetime' =None, executable: 'bool' =None, id: 'str' =None, internal_message: 'bool' =None, like: 'V1LikeStatus' =None, metadata: 'dict(str, str)' =None, model: 'str' =None, parent_message_id: 'str' =None, prompt_tokens: 'str' =None, throughput: 'float' =None): # noqa: E501
80
82
  """V1Message - a model defined in Swagger""" # noqa: E501
81
83
  self._account_id = None
82
84
  self._assistant_id = None
@@ -88,6 +90,7 @@ class V1Message(object):
88
90
  self._executable = None
89
91
  self._id = None
90
92
  self._internal_message = None
93
+ self._like = None
91
94
  self._metadata = None
92
95
  self._model = None
93
96
  self._parent_message_id = None
@@ -114,6 +117,8 @@ class V1Message(object):
114
117
  self.id = id
115
118
  if internal_message is not None:
116
119
  self.internal_message = internal_message
120
+ if like is not None:
121
+ self.like = like
117
122
  if metadata is not None:
118
123
  self.metadata = metadata
119
124
  if model is not None:
@@ -335,6 +340,27 @@ class V1Message(object):
335
340
 
336
341
  self._internal_message = internal_message
337
342
 
343
+ @property
344
+ def like(self) -> 'V1LikeStatus':
345
+ """Gets the like of this V1Message. # noqa: E501
346
+
347
+
348
+ :return: The like of this V1Message. # noqa: E501
349
+ :rtype: V1LikeStatus
350
+ """
351
+ return self._like
352
+
353
+ @like.setter
354
+ def like(self, like: 'V1LikeStatus'):
355
+ """Sets the like of this V1Message.
356
+
357
+
358
+ :param like: The like of this V1Message. # noqa: E501
359
+ :type: V1LikeStatus
360
+ """
361
+
362
+ self._like = like
363
+
338
364
  @property
339
365
  def metadata(self) -> 'dict(str, str)':
340
366
  """Gets the metadata of this V1Message. # noqa: E501
@@ -43,6 +43,7 @@ class V1QuoteSubscriptionResponse(object):
43
43
  swagger_types = {
44
44
  'annual_amount': 'int',
45
45
  'annual_tax': 'int',
46
+ 'eligible': 'bool',
46
47
  'features': 'list[V1BillingFeature]',
47
48
  'monthly_amount': 'int',
48
49
  'monthly_tax': 'int',
@@ -54,6 +55,7 @@ class V1QuoteSubscriptionResponse(object):
54
55
  attribute_map = {
55
56
  'annual_amount': 'annualAmount',
56
57
  'annual_tax': 'annualTax',
58
+ 'eligible': 'eligible',
57
59
  'features': 'features',
58
60
  'monthly_amount': 'monthlyAmount',
59
61
  'monthly_tax': 'monthlyTax',
@@ -62,10 +64,11 @@ class V1QuoteSubscriptionResponse(object):
62
64
  'trial_days': 'trialDays'
63
65
  }
64
66
 
65
- def __init__(self, annual_amount: 'int' =None, annual_tax: 'int' =None, features: 'list[V1BillingFeature]' =None, monthly_amount: 'int' =None, monthly_tax: 'int' =None, prorated_amount: 'int' =None, prorated_tax: 'int' =None, trial_days: 'int' =None): # noqa: E501
67
+ def __init__(self, annual_amount: 'int' =None, annual_tax: 'int' =None, eligible: 'bool' =None, features: 'list[V1BillingFeature]' =None, monthly_amount: 'int' =None, monthly_tax: 'int' =None, prorated_amount: 'int' =None, prorated_tax: 'int' =None, trial_days: 'int' =None): # noqa: E501
66
68
  """V1QuoteSubscriptionResponse - a model defined in Swagger""" # noqa: E501
67
69
  self._annual_amount = None
68
70
  self._annual_tax = None
71
+ self._eligible = None
69
72
  self._features = None
70
73
  self._monthly_amount = None
71
74
  self._monthly_tax = None
@@ -77,6 +80,8 @@ class V1QuoteSubscriptionResponse(object):
77
80
  self.annual_amount = annual_amount
78
81
  if annual_tax is not None:
79
82
  self.annual_tax = annual_tax
83
+ if eligible is not None:
84
+ self.eligible = eligible
80
85
  if features is not None:
81
86
  self.features = features
82
87
  if monthly_amount is not None:
@@ -132,6 +137,27 @@ class V1QuoteSubscriptionResponse(object):
132
137
 
133
138
  self._annual_tax = annual_tax
134
139
 
140
+ @property
141
+ def eligible(self) -> 'bool':
142
+ """Gets the eligible of this V1QuoteSubscriptionResponse. # noqa: E501
143
+
144
+
145
+ :return: The eligible of this V1QuoteSubscriptionResponse. # noqa: E501
146
+ :rtype: bool
147
+ """
148
+ return self._eligible
149
+
150
+ @eligible.setter
151
+ def eligible(self, eligible: 'bool'):
152
+ """Sets the eligible of this V1QuoteSubscriptionResponse.
153
+
154
+
155
+ :param eligible: The eligible of this V1QuoteSubscriptionResponse. # noqa: E501
156
+ :type: bool
157
+ """
158
+
159
+ self._eligible = eligible
160
+
135
161
  @property
136
162
  def features(self) -> 'list[V1BillingFeature]':
137
163
  """Gets the features of this V1QuoteSubscriptionResponse. # noqa: E501
@@ -43,25 +43,30 @@ class V1ResourceVisibility(object):
43
43
  swagger_types = {
44
44
  'all_org_users': 'bool',
45
45
  'all_users': 'bool',
46
+ 'public': 'bool',
46
47
  'user_ids': 'list[str]'
47
48
  }
48
49
 
49
50
  attribute_map = {
50
51
  'all_org_users': 'allOrgUsers',
51
52
  'all_users': 'allUsers',
53
+ 'public': 'public',
52
54
  'user_ids': 'userIds'
53
55
  }
54
56
 
55
- def __init__(self, all_org_users: 'bool' =None, all_users: 'bool' =None, user_ids: 'list[str]' =None): # noqa: E501
57
+ def __init__(self, all_org_users: 'bool' =None, all_users: 'bool' =None, public: 'bool' =None, user_ids: 'list[str]' =None): # noqa: E501
56
58
  """V1ResourceVisibility - a model defined in Swagger""" # noqa: E501
57
59
  self._all_org_users = None
58
60
  self._all_users = None
61
+ self._public = None
59
62
  self._user_ids = None
60
63
  self.discriminator = None
61
64
  if all_org_users is not None:
62
65
  self.all_org_users = all_org_users
63
66
  if all_users is not None:
64
67
  self.all_users = all_users
68
+ if public is not None:
69
+ self.public = public
65
70
  if user_ids is not None:
66
71
  self.user_ids = user_ids
67
72
 
@@ -107,6 +112,27 @@ class V1ResourceVisibility(object):
107
112
 
108
113
  self._all_users = all_users
109
114
 
115
+ @property
116
+ def public(self) -> 'bool':
117
+ """Gets the public of this V1ResourceVisibility. # noqa: E501
118
+
119
+
120
+ :return: The public of this V1ResourceVisibility. # noqa: E501
121
+ :rtype: bool
122
+ """
123
+ return self._public
124
+
125
+ @public.setter
126
+ def public(self, public: 'bool'):
127
+ """Sets the public of this V1ResourceVisibility.
128
+
129
+
130
+ :param public: The public of this V1ResourceVisibility. # noqa: E501
131
+ :type: bool
132
+ """
133
+
134
+ self._public = public
135
+
110
136
  @property
111
137
  def user_ids(self) -> 'list[str]':
112
138
  """Gets the user_ids of this V1ResourceVisibility. # noqa: E501
@@ -43,20 +43,23 @@ class V1ResponseChoice(object):
43
43
  swagger_types = {
44
44
  'delta': 'V1ResponseChoiceDelta',
45
45
  'finish_reason': 'str',
46
- 'index': 'int'
46
+ 'index': 'int',
47
+ 'tool_calls': 'list[V1ToolCall]'
47
48
  }
48
49
 
49
50
  attribute_map = {
50
51
  'delta': 'delta',
51
52
  'finish_reason': 'finishReason',
52
- 'index': 'index'
53
+ 'index': 'index',
54
+ 'tool_calls': 'toolCalls'
53
55
  }
54
56
 
55
- def __init__(self, delta: 'V1ResponseChoiceDelta' =None, finish_reason: 'str' =None, index: 'int' =None): # noqa: E501
57
+ def __init__(self, delta: 'V1ResponseChoiceDelta' =None, finish_reason: 'str' =None, index: 'int' =None, tool_calls: 'list[V1ToolCall]' =None): # noqa: E501
56
58
  """V1ResponseChoice - a model defined in Swagger""" # noqa: E501
57
59
  self._delta = None
58
60
  self._finish_reason = None
59
61
  self._index = None
62
+ self._tool_calls = None
60
63
  self.discriminator = None
61
64
  if delta is not None:
62
65
  self.delta = delta
@@ -64,6 +67,8 @@ class V1ResponseChoice(object):
64
67
  self.finish_reason = finish_reason
65
68
  if index is not None:
66
69
  self.index = index
70
+ if tool_calls is not None:
71
+ self.tool_calls = tool_calls
67
72
 
68
73
  @property
69
74
  def delta(self) -> 'V1ResponseChoiceDelta':
@@ -128,6 +133,27 @@ class V1ResponseChoice(object):
128
133
 
129
134
  self._index = index
130
135
 
136
+ @property
137
+ def tool_calls(self) -> 'list[V1ToolCall]':
138
+ """Gets the tool_calls of this V1ResponseChoice. # noqa: E501
139
+
140
+
141
+ :return: The tool_calls of this V1ResponseChoice. # noqa: E501
142
+ :rtype: list[V1ToolCall]
143
+ """
144
+ return self._tool_calls
145
+
146
+ @tool_calls.setter
147
+ def tool_calls(self, tool_calls: 'list[V1ToolCall]'):
148
+ """Sets the tool_calls of this V1ResponseChoice.
149
+
150
+
151
+ :param tool_calls: The tool_calls of this V1ResponseChoice. # noqa: E501
152
+ :type: list[V1ToolCall]
153
+ """
154
+
155
+ self._tool_calls = tool_calls
156
+
131
157
  def to_dict(self) -> dict:
132
158
  """Returns the model properties as a dict"""
133
159
  result = {}