lightning-sdk 2025.8.1__py3-none-any.whl → 2025.8.6rc1__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 (65) hide show
  1. lightning_sdk/__init__.py +1 -1
  2. lightning_sdk/api/deployment_api.py +6 -4
  3. lightning_sdk/api/job_api.py +2 -0
  4. lightning_sdk/api/mmt_api.py +2 -0
  5. lightning_sdk/deployment/deployment.py +5 -0
  6. lightning_sdk/lightning_cloud/login.py +1 -7
  7. lightning_sdk/lightning_cloud/openapi/__init__.py +15 -23
  8. lightning_sdk/lightning_cloud/openapi/api/__init__.py +1 -0
  9. lightning_sdk/lightning_cloud/openapi/api/assistants_service_api.py +5 -1
  10. lightning_sdk/lightning_cloud/openapi/api/cloud_space_service_api.py +228 -1
  11. lightning_sdk/lightning_cloud/openapi/api/endpoint_service_api.py +126 -2129
  12. lightning_sdk/lightning_cloud/openapi/api/k8_s_cluster_service_api.py +625 -0
  13. lightning_sdk/lightning_cloud/openapi/api/storage_service_api.py +5 -1
  14. lightning_sdk/lightning_cloud/openapi/models/__init__.py +14 -23
  15. lightning_sdk/lightning_cloud/openapi/models/cluster_id_metrics_body.py +175 -0
  16. lightning_sdk/lightning_cloud/openapi/models/orgs_id_body.py +53 -1
  17. lightning_sdk/lightning_cloud/openapi/models/projects_id_body.py +53 -1
  18. lightning_sdk/lightning_cloud/openapi/models/v1_cloud_space.py +27 -1
  19. lightning_sdk/lightning_cloud/openapi/models/v1_cloud_space_transfer_metadata.py +201 -0
  20. lightning_sdk/lightning_cloud/openapi/models/v1_cloudflare_v1.py +29 -3
  21. lightning_sdk/lightning_cloud/openapi/models/{v1_complete_upload_service_execution_artifact_response.py → v1_complete_running_cloud_space_instance_transfer_response.py} +6 -6
  22. lightning_sdk/lightning_cloud/openapi/models/v1_container_metrics.py +461 -0
  23. lightning_sdk/lightning_cloud/openapi/models/v1_filesystem_metrics.py +227 -0
  24. lightning_sdk/lightning_cloud/openapi/models/{v1_list_file_endpoints_response.py → v1_list_container_metrics_response.py} +23 -23
  25. lightning_sdk/lightning_cloud/openapi/models/{service_artifact_artifact_kind.py → v1_list_node_file_system_metrics_response.py} +6 -13
  26. lightning_sdk/lightning_cloud/openapi/models/{v1_upload_service_execution_artifact_parts_response.py → v1_list_node_metrics_response.py} +23 -23
  27. lightning_sdk/lightning_cloud/openapi/models/v1_list_pod_metrics_response.py +123 -0
  28. lightning_sdk/lightning_cloud/openapi/models/v1_managed_model.py +79 -1
  29. lightning_sdk/lightning_cloud/openapi/models/v1_node_metrics.py +357 -0
  30. lightning_sdk/lightning_cloud/openapi/models/v1_organization.py +53 -1
  31. lightning_sdk/lightning_cloud/openapi/models/v1_pod_metrics.py +435 -0
  32. lightning_sdk/lightning_cloud/openapi/models/v1_project_settings.py +53 -1
  33. lightning_sdk/lightning_cloud/openapi/models/{v1_delete_service_execution_response.py → v1_report_k8s_cluster_metrics_response.py} +6 -6
  34. lightning_sdk/lightning_cloud/openapi/models/{command_argument_command_argument_type.py → v1_request_cloud_space_access_response.py} +6 -12
  35. lightning_sdk/lightning_cloud/openapi/models/v1_update_cloud_space_instance_config_request.py +331 -0
  36. lightning_sdk/lightning_cloud/openapi/models/v1_user_features.py +157 -53
  37. lightning_sdk/lightning_cloud/utils/data_connection.py +56 -1
  38. lightning_sdk/llm/llm.py +2 -13
  39. lightning_sdk/llm/public_assistants.py +17 -0
  40. lightning_sdk/pipeline/steps.py +30 -1
  41. lightning_sdk/services/file_endpoint.py +3 -4
  42. {lightning_sdk-2025.8.1.dist-info → lightning_sdk-2025.8.6rc1.dist-info}/METADATA +1 -1
  43. {lightning_sdk-2025.8.1.dist-info → lightning_sdk-2025.8.6rc1.dist-info}/RECORD +47 -55
  44. lightning_sdk/lightning_cloud/openapi/models/fileendpoints_id_body.py +0 -409
  45. lightning_sdk/lightning_cloud/openapi/models/id_complete_body.py +0 -175
  46. lightning_sdk/lightning_cloud/openapi/models/id_storage_body.py +0 -175
  47. lightning_sdk/lightning_cloud/openapi/models/id_uploads_body1.py +0 -175
  48. lightning_sdk/lightning_cloud/openapi/models/project_id_fileendpoints_body.py +0 -357
  49. lightning_sdk/lightning_cloud/openapi/models/project_id_serviceexecution_body.py +0 -175
  50. lightning_sdk/lightning_cloud/openapi/models/serviceexecution_id_body.py +0 -331
  51. lightning_sdk/lightning_cloud/openapi/models/v1_command_argument.py +0 -305
  52. lightning_sdk/lightning_cloud/openapi/models/v1_delete_file_endpoint_response.py +0 -149
  53. lightning_sdk/lightning_cloud/openapi/models/v1_download_service_execution_artifact_response.py +0 -175
  54. lightning_sdk/lightning_cloud/openapi/models/v1_file_endpoint.py +0 -461
  55. lightning_sdk/lightning_cloud/openapi/models/v1_get_service_execution_status_response.py +0 -175
  56. lightning_sdk/lightning_cloud/openapi/models/v1_list_service_execution_lightningapp_instances_response.py +0 -175
  57. lightning_sdk/lightning_cloud/openapi/models/v1_list_service_executions_response.py +0 -123
  58. lightning_sdk/lightning_cloud/openapi/models/v1_service_artifact.py +0 -201
  59. lightning_sdk/lightning_cloud/openapi/models/v1_service_execution.py +0 -383
  60. lightning_sdk/lightning_cloud/openapi/models/v1_upload_service_execution_artifact_response.py +0 -149
  61. lightning_sdk/llm/public_assistants.json +0 -34
  62. {lightning_sdk-2025.8.1.dist-info → lightning_sdk-2025.8.6rc1.dist-info}/LICENSE +0 -0
  63. {lightning_sdk-2025.8.1.dist-info → lightning_sdk-2025.8.6rc1.dist-info}/WHEEL +0 -0
  64. {lightning_sdk-2025.8.1.dist-info → lightning_sdk-2025.8.6rc1.dist-info}/entry_points.txt +0 -0
  65. {lightning_sdk-2025.8.1.dist-info → lightning_sdk-2025.8.6rc1.dist-info}/top_level.txt +0 -0
@@ -28,7 +28,7 @@ if TYPE_CHECKING:
28
28
  from datetime import datetime
29
29
  from lightning_sdk.lightning_cloud.openapi.models import *
30
30
 
31
- class V1ListFileEndpointsResponse(object):
31
+ class V1ListContainerMetricsResponse(object):
32
32
  """NOTE: This class is auto generated by the swagger code generator program.
33
33
 
34
34
  Do not edit the class manually.
@@ -41,40 +41,40 @@ class V1ListFileEndpointsResponse(object):
41
41
  and the value is json key in definition.
42
42
  """
43
43
  swagger_types = {
44
- 'file_endpoints': 'list[V1FileEndpoint]'
44
+ 'container_metrics': 'list[V1ContainerMetrics]'
45
45
  }
46
46
 
47
47
  attribute_map = {
48
- 'file_endpoints': 'fileEndpoints'
48
+ 'container_metrics': 'containerMetrics'
49
49
  }
50
50
 
51
- def __init__(self, file_endpoints: 'list[V1FileEndpoint]' =None): # noqa: E501
52
- """V1ListFileEndpointsResponse - a model defined in Swagger""" # noqa: E501
53
- self._file_endpoints = None
51
+ def __init__(self, container_metrics: 'list[V1ContainerMetrics]' =None): # noqa: E501
52
+ """V1ListContainerMetricsResponse - a model defined in Swagger""" # noqa: E501
53
+ self._container_metrics = None
54
54
  self.discriminator = None
55
- if file_endpoints is not None:
56
- self.file_endpoints = file_endpoints
55
+ if container_metrics is not None:
56
+ self.container_metrics = container_metrics
57
57
 
58
58
  @property
59
- def file_endpoints(self) -> 'list[V1FileEndpoint]':
60
- """Gets the file_endpoints of this V1ListFileEndpointsResponse. # noqa: E501
59
+ def container_metrics(self) -> 'list[V1ContainerMetrics]':
60
+ """Gets the container_metrics of this V1ListContainerMetricsResponse. # noqa: E501
61
61
 
62
62
 
63
- :return: The file_endpoints of this V1ListFileEndpointsResponse. # noqa: E501
64
- :rtype: list[V1FileEndpoint]
63
+ :return: The container_metrics of this V1ListContainerMetricsResponse. # noqa: E501
64
+ :rtype: list[V1ContainerMetrics]
65
65
  """
66
- return self._file_endpoints
66
+ return self._container_metrics
67
67
 
68
- @file_endpoints.setter
69
- def file_endpoints(self, file_endpoints: 'list[V1FileEndpoint]'):
70
- """Sets the file_endpoints of this V1ListFileEndpointsResponse.
68
+ @container_metrics.setter
69
+ def container_metrics(self, container_metrics: 'list[V1ContainerMetrics]'):
70
+ """Sets the container_metrics of this V1ListContainerMetricsResponse.
71
71
 
72
72
 
73
- :param file_endpoints: The file_endpoints of this V1ListFileEndpointsResponse. # noqa: E501
74
- :type: list[V1FileEndpoint]
73
+ :param container_metrics: The container_metrics of this V1ListContainerMetricsResponse. # noqa: E501
74
+ :type: list[V1ContainerMetrics]
75
75
  """
76
76
 
77
- self._file_endpoints = file_endpoints
77
+ self._container_metrics = container_metrics
78
78
 
79
79
  def to_dict(self) -> dict:
80
80
  """Returns the model properties as a dict"""
@@ -97,7 +97,7 @@ class V1ListFileEndpointsResponse(object):
97
97
  ))
98
98
  else:
99
99
  result[attr] = value
100
- if issubclass(V1ListFileEndpointsResponse, dict):
100
+ if issubclass(V1ListContainerMetricsResponse, dict):
101
101
  for key, value in self.items():
102
102
  result[key] = value
103
103
 
@@ -111,13 +111,13 @@ class V1ListFileEndpointsResponse(object):
111
111
  """For `print` and `pprint`"""
112
112
  return self.to_str()
113
113
 
114
- def __eq__(self, other: 'V1ListFileEndpointsResponse') -> bool:
114
+ def __eq__(self, other: 'V1ListContainerMetricsResponse') -> bool:
115
115
  """Returns true if both objects are equal"""
116
- if not isinstance(other, V1ListFileEndpointsResponse):
116
+ if not isinstance(other, V1ListContainerMetricsResponse):
117
117
  return False
118
118
 
119
119
  return self.__dict__ == other.__dict__
120
120
 
121
- def __ne__(self, other: 'V1ListFileEndpointsResponse') -> bool:
121
+ def __ne__(self, other: 'V1ListContainerMetricsResponse') -> bool:
122
122
  """Returns true if both objects are not equal"""
123
123
  return not self == other
@@ -28,18 +28,11 @@ if TYPE_CHECKING:
28
28
  from datetime import datetime
29
29
  from lightning_sdk.lightning_cloud.openapi.models import *
30
30
 
31
- class ServiceArtifactArtifactKind(object):
31
+ class V1ListNodeFileSystemMetricsResponse(object):
32
32
  """NOTE: This class is auto generated by the swagger code generator program.
33
33
 
34
34
  Do not edit the class manually.
35
35
  """
36
-
37
- """
38
- allowed enum values
39
- """
40
- FILE = "FILE"
41
- FOLDER = "FOLDER"
42
- JSON = "JSON"
43
36
  """
44
37
  Attributes:
45
38
  swagger_types (dict): The key is attribute name
@@ -54,7 +47,7 @@ class ServiceArtifactArtifactKind(object):
54
47
  }
55
48
 
56
49
  def __init__(self): # noqa: E501
57
- """ServiceArtifactArtifactKind - a model defined in Swagger""" # noqa: E501
50
+ """V1ListNodeFileSystemMetricsResponse - a model defined in Swagger""" # noqa: E501
58
51
  self.discriminator = None
59
52
 
60
53
  def to_dict(self) -> dict:
@@ -78,7 +71,7 @@ class ServiceArtifactArtifactKind(object):
78
71
  ))
79
72
  else:
80
73
  result[attr] = value
81
- if issubclass(ServiceArtifactArtifactKind, dict):
74
+ if issubclass(V1ListNodeFileSystemMetricsResponse, dict):
82
75
  for key, value in self.items():
83
76
  result[key] = value
84
77
 
@@ -92,13 +85,13 @@ class ServiceArtifactArtifactKind(object):
92
85
  """For `print` and `pprint`"""
93
86
  return self.to_str()
94
87
 
95
- def __eq__(self, other: 'ServiceArtifactArtifactKind') -> bool:
88
+ def __eq__(self, other: 'V1ListNodeFileSystemMetricsResponse') -> bool:
96
89
  """Returns true if both objects are equal"""
97
- if not isinstance(other, ServiceArtifactArtifactKind):
90
+ if not isinstance(other, V1ListNodeFileSystemMetricsResponse):
98
91
  return False
99
92
 
100
93
  return self.__dict__ == other.__dict__
101
94
 
102
- def __ne__(self, other: 'ServiceArtifactArtifactKind') -> bool:
95
+ def __ne__(self, other: 'V1ListNodeFileSystemMetricsResponse') -> bool:
103
96
  """Returns true if both objects are not equal"""
104
97
  return not self == other
@@ -28,7 +28,7 @@ if TYPE_CHECKING:
28
28
  from datetime import datetime
29
29
  from lightning_sdk.lightning_cloud.openapi.models import *
30
30
 
31
- class V1UploadServiceExecutionArtifactPartsResponse(object):
31
+ class V1ListNodeMetricsResponse(object):
32
32
  """NOTE: This class is auto generated by the swagger code generator program.
33
33
 
34
34
  Do not edit the class manually.
@@ -41,40 +41,40 @@ class V1UploadServiceExecutionArtifactPartsResponse(object):
41
41
  and the value is json key in definition.
42
42
  """
43
43
  swagger_types = {
44
- 'urls': 'list[V1PresignedUrl]'
44
+ 'node_metrics': 'list[V1NodeMetrics]'
45
45
  }
46
46
 
47
47
  attribute_map = {
48
- 'urls': 'urls'
48
+ 'node_metrics': 'nodeMetrics'
49
49
  }
50
50
 
51
- def __init__(self, urls: 'list[V1PresignedUrl]' =None): # noqa: E501
52
- """V1UploadServiceExecutionArtifactPartsResponse - a model defined in Swagger""" # noqa: E501
53
- self._urls = None
51
+ def __init__(self, node_metrics: 'list[V1NodeMetrics]' =None): # noqa: E501
52
+ """V1ListNodeMetricsResponse - a model defined in Swagger""" # noqa: E501
53
+ self._node_metrics = None
54
54
  self.discriminator = None
55
- if urls is not None:
56
- self.urls = urls
55
+ if node_metrics is not None:
56
+ self.node_metrics = node_metrics
57
57
 
58
58
  @property
59
- def urls(self) -> 'list[V1PresignedUrl]':
60
- """Gets the urls of this V1UploadServiceExecutionArtifactPartsResponse. # noqa: E501
59
+ def node_metrics(self) -> 'list[V1NodeMetrics]':
60
+ """Gets the node_metrics of this V1ListNodeMetricsResponse. # noqa: E501
61
61
 
62
62
 
63
- :return: The urls of this V1UploadServiceExecutionArtifactPartsResponse. # noqa: E501
64
- :rtype: list[V1PresignedUrl]
63
+ :return: The node_metrics of this V1ListNodeMetricsResponse. # noqa: E501
64
+ :rtype: list[V1NodeMetrics]
65
65
  """
66
- return self._urls
66
+ return self._node_metrics
67
67
 
68
- @urls.setter
69
- def urls(self, urls: 'list[V1PresignedUrl]'):
70
- """Sets the urls of this V1UploadServiceExecutionArtifactPartsResponse.
68
+ @node_metrics.setter
69
+ def node_metrics(self, node_metrics: 'list[V1NodeMetrics]'):
70
+ """Sets the node_metrics of this V1ListNodeMetricsResponse.
71
71
 
72
72
 
73
- :param urls: The urls of this V1UploadServiceExecutionArtifactPartsResponse. # noqa: E501
74
- :type: list[V1PresignedUrl]
73
+ :param node_metrics: The node_metrics of this V1ListNodeMetricsResponse. # noqa: E501
74
+ :type: list[V1NodeMetrics]
75
75
  """
76
76
 
77
- self._urls = urls
77
+ self._node_metrics = node_metrics
78
78
 
79
79
  def to_dict(self) -> dict:
80
80
  """Returns the model properties as a dict"""
@@ -97,7 +97,7 @@ class V1UploadServiceExecutionArtifactPartsResponse(object):
97
97
  ))
98
98
  else:
99
99
  result[attr] = value
100
- if issubclass(V1UploadServiceExecutionArtifactPartsResponse, dict):
100
+ if issubclass(V1ListNodeMetricsResponse, dict):
101
101
  for key, value in self.items():
102
102
  result[key] = value
103
103
 
@@ -111,13 +111,13 @@ class V1UploadServiceExecutionArtifactPartsResponse(object):
111
111
  """For `print` and `pprint`"""
112
112
  return self.to_str()
113
113
 
114
- def __eq__(self, other: 'V1UploadServiceExecutionArtifactPartsResponse') -> bool:
114
+ def __eq__(self, other: 'V1ListNodeMetricsResponse') -> bool:
115
115
  """Returns true if both objects are equal"""
116
- if not isinstance(other, V1UploadServiceExecutionArtifactPartsResponse):
116
+ if not isinstance(other, V1ListNodeMetricsResponse):
117
117
  return False
118
118
 
119
119
  return self.__dict__ == other.__dict__
120
120
 
121
- def __ne__(self, other: 'V1UploadServiceExecutionArtifactPartsResponse') -> bool:
121
+ def __ne__(self, other: 'V1ListNodeMetricsResponse') -> bool:
122
122
  """Returns true if both objects are not equal"""
123
123
  return not self == other
@@ -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 V1ListPodMetricsResponse(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
+ 'pod_metrics': 'list[V1PodMetrics]'
45
+ }
46
+
47
+ attribute_map = {
48
+ 'pod_metrics': 'podMetrics'
49
+ }
50
+
51
+ def __init__(self, pod_metrics: 'list[V1PodMetrics]' =None): # noqa: E501
52
+ """V1ListPodMetricsResponse - a model defined in Swagger""" # noqa: E501
53
+ self._pod_metrics = None
54
+ self.discriminator = None
55
+ if pod_metrics is not None:
56
+ self.pod_metrics = pod_metrics
57
+
58
+ @property
59
+ def pod_metrics(self) -> 'list[V1PodMetrics]':
60
+ """Gets the pod_metrics of this V1ListPodMetricsResponse. # noqa: E501
61
+
62
+
63
+ :return: The pod_metrics of this V1ListPodMetricsResponse. # noqa: E501
64
+ :rtype: list[V1PodMetrics]
65
+ """
66
+ return self._pod_metrics
67
+
68
+ @pod_metrics.setter
69
+ def pod_metrics(self, pod_metrics: 'list[V1PodMetrics]'):
70
+ """Sets the pod_metrics of this V1ListPodMetricsResponse.
71
+
72
+
73
+ :param pod_metrics: The pod_metrics of this V1ListPodMetricsResponse. # noqa: E501
74
+ :type: list[V1PodMetrics]
75
+ """
76
+
77
+ self._pod_metrics = pod_metrics
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(V1ListPodMetricsResponse, 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: 'V1ListPodMetricsResponse') -> bool:
115
+ """Returns true if both objects are equal"""
116
+ if not isinstance(other, V1ListPodMetricsResponse):
117
+ return False
118
+
119
+ return self.__dict__ == other.__dict__
120
+
121
+ def __ne__(self, other: 'V1ListPodMetricsResponse') -> bool:
122
+ """Returns true if both objects are not equal"""
123
+ return not self == other
@@ -44,6 +44,7 @@ class V1ManagedModel(object):
44
44
  'abilities': 'V1ManagedModelAbilities',
45
45
  'assistant_id': 'str',
46
46
  'completion_token_price': 'float',
47
+ 'completion_token_price_above_threshold': 'float',
47
48
  'context_length': 'str',
48
49
  'deployment_details': 'V1DeploymentDetails',
49
50
  'description': 'str',
@@ -53,10 +54,12 @@ class V1ManagedModel(object):
53
54
  'max_completion_tokens': 'str',
54
55
  'name': 'str',
55
56
  'prompt_token_price': 'float',
57
+ 'prompt_token_price_above_threshold': 'float',
56
58
  'status': 'V1AssistantModelStatus',
57
59
  'temperature': 'float',
58
60
  'throughput': 'float',
59
61
  'time_to_first_token': 'float',
62
+ 'token_threshold': 'str',
60
63
  'top_k': 'str'
61
64
  }
62
65
 
@@ -64,6 +67,7 @@ class V1ManagedModel(object):
64
67
  'abilities': 'abilities',
65
68
  'assistant_id': 'assistantId',
66
69
  'completion_token_price': 'completionTokenPrice',
70
+ 'completion_token_price_above_threshold': 'completionTokenPriceAboveThreshold',
67
71
  'context_length': 'contextLength',
68
72
  'deployment_details': 'deploymentDetails',
69
73
  'description': 'description',
@@ -73,18 +77,21 @@ class V1ManagedModel(object):
73
77
  'max_completion_tokens': 'maxCompletionTokens',
74
78
  'name': 'name',
75
79
  'prompt_token_price': 'promptTokenPrice',
80
+ 'prompt_token_price_above_threshold': 'promptTokenPriceAboveThreshold',
76
81
  'status': 'status',
77
82
  'temperature': 'temperature',
78
83
  'throughput': 'throughput',
79
84
  'time_to_first_token': 'timeToFirstToken',
85
+ 'token_threshold': 'tokenThreshold',
80
86
  'top_k': 'topK'
81
87
  }
82
88
 
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
89
+ def __init__(self, abilities: 'V1ManagedModelAbilities' =None, assistant_id: 'str' =None, completion_token_price: 'float' =None, completion_token_price_above_threshold: '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, prompt_token_price_above_threshold: 'float' =None, status: 'V1AssistantModelStatus' =None, temperature: 'float' =None, throughput: 'float' =None, time_to_first_token: 'float' =None, token_threshold: 'str' =None, top_k: 'str' =None): # noqa: E501
84
90
  """V1ManagedModel - a model defined in Swagger""" # noqa: E501
85
91
  self._abilities = None
86
92
  self._assistant_id = None
87
93
  self._completion_token_price = None
94
+ self._completion_token_price_above_threshold = None
88
95
  self._context_length = None
89
96
  self._deployment_details = None
90
97
  self._description = None
@@ -94,10 +101,12 @@ class V1ManagedModel(object):
94
101
  self._max_completion_tokens = None
95
102
  self._name = None
96
103
  self._prompt_token_price = None
104
+ self._prompt_token_price_above_threshold = None
97
105
  self._status = None
98
106
  self._temperature = None
99
107
  self._throughput = None
100
108
  self._time_to_first_token = None
109
+ self._token_threshold = None
101
110
  self._top_k = None
102
111
  self.discriminator = None
103
112
  if abilities is not None:
@@ -106,6 +115,8 @@ class V1ManagedModel(object):
106
115
  self.assistant_id = assistant_id
107
116
  if completion_token_price is not None:
108
117
  self.completion_token_price = completion_token_price
118
+ if completion_token_price_above_threshold is not None:
119
+ self.completion_token_price_above_threshold = completion_token_price_above_threshold
109
120
  if context_length is not None:
110
121
  self.context_length = context_length
111
122
  if deployment_details is not None:
@@ -124,6 +135,8 @@ class V1ManagedModel(object):
124
135
  self.name = name
125
136
  if prompt_token_price is not None:
126
137
  self.prompt_token_price = prompt_token_price
138
+ if prompt_token_price_above_threshold is not None:
139
+ self.prompt_token_price_above_threshold = prompt_token_price_above_threshold
127
140
  if status is not None:
128
141
  self.status = status
129
142
  if temperature is not None:
@@ -132,6 +145,8 @@ class V1ManagedModel(object):
132
145
  self.throughput = throughput
133
146
  if time_to_first_token is not None:
134
147
  self.time_to_first_token = time_to_first_token
148
+ if token_threshold is not None:
149
+ self.token_threshold = token_threshold
135
150
  if top_k is not None:
136
151
  self.top_k = top_k
137
152
 
@@ -198,6 +213,27 @@ class V1ManagedModel(object):
198
213
 
199
214
  self._completion_token_price = completion_token_price
200
215
 
216
+ @property
217
+ def completion_token_price_above_threshold(self) -> 'float':
218
+ """Gets the completion_token_price_above_threshold of this V1ManagedModel. # noqa: E501
219
+
220
+
221
+ :return: The completion_token_price_above_threshold of this V1ManagedModel. # noqa: E501
222
+ :rtype: float
223
+ """
224
+ return self._completion_token_price_above_threshold
225
+
226
+ @completion_token_price_above_threshold.setter
227
+ def completion_token_price_above_threshold(self, completion_token_price_above_threshold: 'float'):
228
+ """Sets the completion_token_price_above_threshold of this V1ManagedModel.
229
+
230
+
231
+ :param completion_token_price_above_threshold: The completion_token_price_above_threshold of this V1ManagedModel. # noqa: E501
232
+ :type: float
233
+ """
234
+
235
+ self._completion_token_price_above_threshold = completion_token_price_above_threshold
236
+
201
237
  @property
202
238
  def context_length(self) -> 'str':
203
239
  """Gets the context_length of this V1ManagedModel. # noqa: E501
@@ -387,6 +423,27 @@ class V1ManagedModel(object):
387
423
 
388
424
  self._prompt_token_price = prompt_token_price
389
425
 
426
+ @property
427
+ def prompt_token_price_above_threshold(self) -> 'float':
428
+ """Gets the prompt_token_price_above_threshold of this V1ManagedModel. # noqa: E501
429
+
430
+
431
+ :return: The prompt_token_price_above_threshold of this V1ManagedModel. # noqa: E501
432
+ :rtype: float
433
+ """
434
+ return self._prompt_token_price_above_threshold
435
+
436
+ @prompt_token_price_above_threshold.setter
437
+ def prompt_token_price_above_threshold(self, prompt_token_price_above_threshold: 'float'):
438
+ """Sets the prompt_token_price_above_threshold of this V1ManagedModel.
439
+
440
+
441
+ :param prompt_token_price_above_threshold: The prompt_token_price_above_threshold of this V1ManagedModel. # noqa: E501
442
+ :type: float
443
+ """
444
+
445
+ self._prompt_token_price_above_threshold = prompt_token_price_above_threshold
446
+
390
447
  @property
391
448
  def status(self) -> 'V1AssistantModelStatus':
392
449
  """Gets the status of this V1ManagedModel. # noqa: E501
@@ -471,6 +528,27 @@ class V1ManagedModel(object):
471
528
 
472
529
  self._time_to_first_token = time_to_first_token
473
530
 
531
+ @property
532
+ def token_threshold(self) -> 'str':
533
+ """Gets the token_threshold of this V1ManagedModel. # noqa: E501
534
+
535
+
536
+ :return: The token_threshold of this V1ManagedModel. # noqa: E501
537
+ :rtype: str
538
+ """
539
+ return self._token_threshold
540
+
541
+ @token_threshold.setter
542
+ def token_threshold(self, token_threshold: 'str'):
543
+ """Sets the token_threshold of this V1ManagedModel.
544
+
545
+
546
+ :param token_threshold: The token_threshold of this V1ManagedModel. # noqa: E501
547
+ :type: str
548
+ """
549
+
550
+ self._token_threshold = token_threshold
551
+
474
552
  @property
475
553
  def top_k(self) -> 'str':
476
554
  """Gets the top_k of this V1ManagedModel. # noqa: E501