orca-sdk 0.0.94__py3-none-any.whl → 0.0.95__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 (116) hide show
  1. orca_sdk/__init__.py +13 -4
  2. orca_sdk/_generated_api_client/api/__init__.py +80 -34
  3. orca_sdk/_generated_api_client/api/classification_model/create_classification_model_classification_model_post.py +170 -0
  4. orca_sdk/_generated_api_client/api/classification_model/{get_model_classification_model_name_or_id_get.py → delete_classification_model_classification_model_name_or_id_delete.py} +20 -20
  5. orca_sdk/_generated_api_client/api/classification_model/{delete_evaluation_classification_model_model_name_or_id_evaluation_task_id_delete.py → delete_classification_model_evaluation_classification_model_model_name_or_id_evaluation_task_id_delete.py} +4 -4
  6. orca_sdk/_generated_api_client/api/classification_model/{create_evaluation_classification_model_model_name_or_id_evaluation_post.py → evaluate_classification_model_classification_model_model_name_or_id_evaluation_post.py} +14 -14
  7. orca_sdk/_generated_api_client/api/classification_model/get_classification_model_classification_model_name_or_id_get.py +156 -0
  8. orca_sdk/_generated_api_client/api/classification_model/{get_evaluation_classification_model_model_name_or_id_evaluation_task_id_get.py → get_classification_model_evaluation_classification_model_model_name_or_id_evaluation_task_id_get.py} +16 -16
  9. orca_sdk/_generated_api_client/api/classification_model/{list_evaluations_classification_model_model_name_or_id_evaluation_get.py → list_classification_model_evaluations_classification_model_model_name_or_id_evaluation_get.py} +16 -16
  10. orca_sdk/_generated_api_client/api/classification_model/list_classification_models_classification_model_get.py +127 -0
  11. orca_sdk/_generated_api_client/api/classification_model/{predict_gpu_classification_model_name_or_id_prediction_post.py → predict_label_gpu_classification_model_name_or_id_prediction_post.py} +14 -14
  12. orca_sdk/_generated_api_client/api/classification_model/update_classification_model_classification_model_name_or_id_patch.py +183 -0
  13. orca_sdk/_generated_api_client/api/datasource/download_datasource_datasource_name_or_id_download_get.py +24 -0
  14. orca_sdk/_generated_api_client/api/memoryset/clone_memoryset_memoryset_name_or_id_clone_post.py +22 -22
  15. orca_sdk/_generated_api_client/api/memoryset/create_memoryset_memoryset_post.py +22 -22
  16. orca_sdk/_generated_api_client/api/memoryset/get_memories_memoryset_name_or_id_memories_get_post.py +38 -16
  17. orca_sdk/_generated_api_client/api/memoryset/get_memory_memoryset_name_or_id_memory_memory_id_get.py +29 -12
  18. orca_sdk/_generated_api_client/api/memoryset/get_memoryset_memoryset_name_or_id_get.py +12 -12
  19. orca_sdk/_generated_api_client/api/memoryset/insert_memories_gpu_memoryset_name_or_id_memory_post.py +17 -14
  20. orca_sdk/_generated_api_client/api/memoryset/list_memorysets_memoryset_get.py +72 -19
  21. orca_sdk/_generated_api_client/api/memoryset/memoryset_lookup_gpu_memoryset_name_or_id_lookup_post.py +31 -12
  22. orca_sdk/_generated_api_client/api/memoryset/potential_duplicate_groups_memoryset_name_or_id_potential_duplicate_groups_get.py +49 -20
  23. orca_sdk/_generated_api_client/api/memoryset/query_memoryset_memoryset_name_or_id_memories_post.py +38 -16
  24. orca_sdk/_generated_api_client/api/memoryset/update_memories_gpu_memoryset_name_or_id_memories_patch.py +54 -29
  25. orca_sdk/_generated_api_client/api/memoryset/update_memory_gpu_memoryset_name_or_id_memory_patch.py +44 -26
  26. orca_sdk/_generated_api_client/api/memoryset/update_memoryset_memoryset_name_or_id_patch.py +22 -22
  27. orca_sdk/_generated_api_client/api/predictive_model/__init__.py +0 -0
  28. orca_sdk/_generated_api_client/api/predictive_model/list_predictive_models_predictive_model_get.py +150 -0
  29. orca_sdk/_generated_api_client/api/regression_model/__init__.py +0 -0
  30. orca_sdk/_generated_api_client/api/{classification_model/create_model_classification_model_post.py → regression_model/create_regression_model_regression_model_post.py} +27 -27
  31. orca_sdk/_generated_api_client/api/regression_model/delete_regression_model_evaluation_regression_model_model_name_or_id_evaluation_task_id_delete.py +168 -0
  32. orca_sdk/_generated_api_client/api/{classification_model/delete_model_classification_model_name_or_id_delete.py → regression_model/delete_regression_model_regression_model_name_or_id_delete.py} +5 -5
  33. orca_sdk/_generated_api_client/api/regression_model/evaluate_regression_model_regression_model_model_name_or_id_evaluation_post.py +183 -0
  34. orca_sdk/_generated_api_client/api/regression_model/get_regression_model_evaluation_regression_model_model_name_or_id_evaluation_task_id_get.py +170 -0
  35. orca_sdk/_generated_api_client/api/regression_model/get_regression_model_regression_model_name_or_id_get.py +156 -0
  36. orca_sdk/_generated_api_client/api/regression_model/list_regression_model_evaluations_regression_model_model_name_or_id_evaluation_get.py +161 -0
  37. orca_sdk/_generated_api_client/api/{classification_model/list_models_classification_model_get.py → regression_model/list_regression_models_regression_model_get.py} +17 -17
  38. orca_sdk/_generated_api_client/api/regression_model/predict_score_gpu_regression_model_name_or_id_prediction_post.py +190 -0
  39. orca_sdk/_generated_api_client/api/{classification_model/update_model_classification_model_name_or_id_patch.py → regression_model/update_regression_model_regression_model_name_or_id_patch.py} +27 -27
  40. orca_sdk/_generated_api_client/api/task/get_task_task_task_id_get.py +156 -0
  41. orca_sdk/_generated_api_client/api/telemetry/get_prediction_telemetry_prediction_prediction_id_get.py +35 -12
  42. orca_sdk/_generated_api_client/api/telemetry/list_memories_with_feedback_telemetry_memories_post.py +20 -12
  43. orca_sdk/_generated_api_client/api/telemetry/list_predictions_telemetry_prediction_post.py +35 -12
  44. orca_sdk/_generated_api_client/models/__init__.py +84 -24
  45. orca_sdk/_generated_api_client/models/base_score_prediction_result.py +108 -0
  46. orca_sdk/_generated_api_client/models/{evaluation_request.py → classification_evaluation_request.py} +13 -45
  47. orca_sdk/_generated_api_client/models/{classification_evaluation_result.py → classification_metrics.py} +106 -56
  48. orca_sdk/_generated_api_client/models/{rac_model_metadata.py → classification_model_metadata.py} +51 -43
  49. orca_sdk/_generated_api_client/models/{prediction_request.py → classification_prediction_request.py} +31 -6
  50. orca_sdk/_generated_api_client/models/{clone_labeled_memoryset_request.py → clone_memoryset_request.py} +5 -5
  51. orca_sdk/_generated_api_client/models/column_info.py +31 -0
  52. orca_sdk/_generated_api_client/models/{create_rac_model_request.py → create_classification_model_request.py} +25 -57
  53. orca_sdk/_generated_api_client/models/{create_labeled_memoryset_request.py → create_memoryset_request.py} +73 -56
  54. orca_sdk/_generated_api_client/models/create_memoryset_request_index_params.py +66 -0
  55. orca_sdk/_generated_api_client/models/create_memoryset_request_index_type.py +13 -0
  56. orca_sdk/_generated_api_client/models/create_regression_model_request.py +137 -0
  57. orca_sdk/_generated_api_client/models/embedding_evaluation_payload.py +187 -0
  58. orca_sdk/_generated_api_client/models/embedding_evaluation_response.py +10 -0
  59. orca_sdk/_generated_api_client/models/evaluation_response.py +22 -9
  60. orca_sdk/_generated_api_client/models/evaluation_response_classification_metrics.py +140 -0
  61. orca_sdk/_generated_api_client/models/evaluation_response_regression_metrics.py +140 -0
  62. orca_sdk/_generated_api_client/models/memory_type.py +9 -0
  63. orca_sdk/_generated_api_client/models/{labeled_memoryset_metadata.py → memoryset_metadata.py} +73 -13
  64. orca_sdk/_generated_api_client/models/memoryset_metadata_index_params.py +55 -0
  65. orca_sdk/_generated_api_client/models/memoryset_metadata_index_type.py +13 -0
  66. orca_sdk/_generated_api_client/models/{labeled_memoryset_update.py → memoryset_update.py} +19 -31
  67. orca_sdk/_generated_api_client/models/not_found_error_response_resource_type_0.py +1 -0
  68. orca_sdk/_generated_api_client/models/{paginated_labeled_memory_with_feedback_metrics.py → paginated_union_labeled_memory_with_feedback_metrics_scored_memory_with_feedback_metrics.py} +37 -10
  69. orca_sdk/_generated_api_client/models/{precision_recall_curve.py → pr_curve.py} +5 -13
  70. orca_sdk/_generated_api_client/models/{rac_model_update.py → predictive_model_update.py} +14 -5
  71. orca_sdk/_generated_api_client/models/pretrained_embedding_model_metadata.py +11 -1
  72. orca_sdk/_generated_api_client/models/rar_head_type.py +8 -0
  73. orca_sdk/_generated_api_client/models/regression_evaluation_request.py +148 -0
  74. orca_sdk/_generated_api_client/models/regression_metrics.py +172 -0
  75. orca_sdk/_generated_api_client/models/regression_model_metadata.py +177 -0
  76. orca_sdk/_generated_api_client/models/regression_prediction_request.py +195 -0
  77. orca_sdk/_generated_api_client/models/roc_curve.py +0 -8
  78. orca_sdk/_generated_api_client/models/score_prediction_memory_lookup.py +196 -0
  79. orca_sdk/_generated_api_client/models/score_prediction_memory_lookup_metadata.py +68 -0
  80. orca_sdk/_generated_api_client/models/score_prediction_with_memories_and_feedback.py +252 -0
  81. orca_sdk/_generated_api_client/models/scored_memory.py +172 -0
  82. orca_sdk/_generated_api_client/models/scored_memory_insert.py +128 -0
  83. orca_sdk/_generated_api_client/models/scored_memory_insert_metadata.py +68 -0
  84. orca_sdk/_generated_api_client/models/scored_memory_lookup.py +180 -0
  85. orca_sdk/_generated_api_client/models/scored_memory_lookup_metadata.py +68 -0
  86. orca_sdk/_generated_api_client/models/scored_memory_metadata.py +68 -0
  87. orca_sdk/_generated_api_client/models/scored_memory_update.py +171 -0
  88. orca_sdk/_generated_api_client/models/scored_memory_update_metadata_type_0.py +68 -0
  89. orca_sdk/_generated_api_client/models/scored_memory_with_feedback_metrics.py +193 -0
  90. orca_sdk/_generated_api_client/models/scored_memory_with_feedback_metrics_feedback_metrics.py +68 -0
  91. orca_sdk/_generated_api_client/models/scored_memory_with_feedback_metrics_metadata.py +68 -0
  92. orca_sdk/_generated_api_client/models/update_prediction_request.py +20 -0
  93. orca_sdk/_shared/__init__.py +9 -1
  94. orca_sdk/_shared/metrics.py +257 -87
  95. orca_sdk/_shared/metrics_test.py +136 -77
  96. orca_sdk/_utils/data_parsing.py +0 -3
  97. orca_sdk/_utils/data_parsing_test.py +0 -3
  98. orca_sdk/_utils/prediction_result_ui.py +55 -23
  99. orca_sdk/classification_model.py +183 -175
  100. orca_sdk/classification_model_test.py +147 -157
  101. orca_sdk/conftest.py +76 -26
  102. orca_sdk/datasource_test.py +0 -1
  103. orca_sdk/embedding_model.py +136 -14
  104. orca_sdk/embedding_model_test.py +10 -6
  105. orca_sdk/job.py +329 -0
  106. orca_sdk/job_test.py +48 -0
  107. orca_sdk/memoryset.py +882 -161
  108. orca_sdk/memoryset_test.py +56 -23
  109. orca_sdk/regression_model.py +647 -0
  110. orca_sdk/regression_model_test.py +338 -0
  111. orca_sdk/telemetry.py +223 -106
  112. orca_sdk/telemetry_test.py +34 -30
  113. {orca_sdk-0.0.94.dist-info → orca_sdk-0.0.95.dist-info}/METADATA +2 -4
  114. {orca_sdk-0.0.94.dist-info → orca_sdk-0.0.95.dist-info}/RECORD +115 -69
  115. orca_sdk/_utils/task.py +0 -73
  116. {orca_sdk-0.0.94.dist-info → orca_sdk-0.0.95.dist-info}/WHEEL +0 -0
@@ -0,0 +1,183 @@
1
+ """
2
+ This file is generated by the openapi-python-client tool via the generate_api_client.py script
3
+
4
+ It is a customized template from the openapi-python-client tool's default template:
5
+ https://github.com/openapi-generators/openapi-python-client/blob/861ef5622f10fc96d240dc9becb0edf94e61446c/openapi_python_client/templates/endpoint_module.py.jinja
6
+
7
+ The main changes are:
8
+ - Update the API call responses to either return the successful response type or raise an error by:
9
+ - Updating the _parse_response function to raise an error if the response status code is not in the 2xx range
10
+ - Inject a client into every method via a context manager
11
+ """
12
+
13
+ from http import HTTPStatus
14
+ from typing import Any, Dict
15
+
16
+ import httpx
17
+
18
+ from ...client import _client_context
19
+ from ...errors import get_error_for_response
20
+ from ...models.evaluation_response import EvaluationResponse
21
+ from ...models.regression_evaluation_request import RegressionEvaluationRequest
22
+ from ...types import Response
23
+
24
+
25
+ def _get_kwargs(
26
+ model_name_or_id: str,
27
+ *,
28
+ body: RegressionEvaluationRequest,
29
+ ) -> dict[str, Any]:
30
+ headers: Dict[str, Any] = {}
31
+
32
+ _kwargs: dict[str, Any] = {
33
+ "method": "post",
34
+ "url": f"/regression_model/{model_name_or_id}/evaluation",
35
+ }
36
+
37
+ _body = body.to_dict()
38
+
39
+ _kwargs["json"] = _body
40
+ headers["Content-Type"] = "application/json"
41
+
42
+ _kwargs["headers"] = headers
43
+ return _kwargs
44
+
45
+
46
+ def _parse_response(*, response: httpx.Response) -> EvaluationResponse:
47
+ if response.status_code == 201:
48
+ response_201 = EvaluationResponse.from_dict(response.json())
49
+
50
+ return response_201
51
+ if response.status_code == 401:
52
+ raise get_error_for_response(response)
53
+ if response.status_code == 500:
54
+ raise get_error_for_response(response)
55
+ if response.status_code == 503:
56
+ raise get_error_for_response(response)
57
+ if response.status_code == 409:
58
+ raise get_error_for_response(response)
59
+ else:
60
+ raise get_error_for_response(response)
61
+
62
+
63
+ def _build_response(*, response: httpx.Response) -> Response[EvaluationResponse]:
64
+ return Response(
65
+ status_code=HTTPStatus(response.status_code),
66
+ content=response.content,
67
+ headers=response.headers,
68
+ parsed=_parse_response(response=response),
69
+ )
70
+
71
+
72
+ def sync_detailed(
73
+ model_name_or_id: str,
74
+ *,
75
+ body: RegressionEvaluationRequest,
76
+ ) -> Response[EvaluationResponse]:
77
+ """Evaluate Regression Model
78
+
79
+ Args:
80
+ model_name_or_id (str):
81
+ body (RegressionEvaluationRequest):
82
+
83
+ Raises:
84
+ errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
85
+ httpx.TimeoutException: If the request takes longer than Client.timeout.
86
+
87
+ Returns:
88
+ Response[EvaluationResponse]
89
+ """
90
+
91
+ kwargs = _get_kwargs(
92
+ model_name_or_id=model_name_or_id,
93
+ body=body,
94
+ )
95
+
96
+ with _client_context() as client:
97
+ response = client.get_httpx_client().request(
98
+ **kwargs,
99
+ )
100
+
101
+ return _build_response(response=response)
102
+
103
+
104
+ def sync(
105
+ model_name_or_id: str,
106
+ *,
107
+ body: RegressionEvaluationRequest,
108
+ ) -> EvaluationResponse:
109
+ """Evaluate Regression Model
110
+
111
+ Args:
112
+ model_name_or_id (str):
113
+ body (RegressionEvaluationRequest):
114
+
115
+ Raises:
116
+ errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
117
+ httpx.TimeoutException: If the request takes longer than Client.timeout.
118
+
119
+ Returns:
120
+ EvaluationResponse
121
+ """
122
+
123
+ return sync_detailed(
124
+ model_name_or_id=model_name_or_id,
125
+ body=body,
126
+ ).parsed
127
+
128
+
129
+ async def asyncio_detailed(
130
+ model_name_or_id: str,
131
+ *,
132
+ body: RegressionEvaluationRequest,
133
+ ) -> Response[EvaluationResponse]:
134
+ """Evaluate Regression Model
135
+
136
+ Args:
137
+ model_name_or_id (str):
138
+ body (RegressionEvaluationRequest):
139
+
140
+ Raises:
141
+ errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
142
+ httpx.TimeoutException: If the request takes longer than Client.timeout.
143
+
144
+ Returns:
145
+ Response[EvaluationResponse]
146
+ """
147
+
148
+ kwargs = _get_kwargs(
149
+ model_name_or_id=model_name_or_id,
150
+ body=body,
151
+ )
152
+
153
+ with _client_context() as client:
154
+ response = await client.get_async_httpx_client().request(**kwargs)
155
+
156
+ return _build_response(response=response)
157
+
158
+
159
+ async def asyncio(
160
+ model_name_or_id: str,
161
+ *,
162
+ body: RegressionEvaluationRequest,
163
+ ) -> EvaluationResponse:
164
+ """Evaluate Regression Model
165
+
166
+ Args:
167
+ model_name_or_id (str):
168
+ body (RegressionEvaluationRequest):
169
+
170
+ Raises:
171
+ errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
172
+ httpx.TimeoutException: If the request takes longer than Client.timeout.
173
+
174
+ Returns:
175
+ EvaluationResponse
176
+ """
177
+
178
+ return (
179
+ await asyncio_detailed(
180
+ model_name_or_id=model_name_or_id,
181
+ body=body,
182
+ )
183
+ ).parsed
@@ -0,0 +1,170 @@
1
+ """
2
+ This file is generated by the openapi-python-client tool via the generate_api_client.py script
3
+
4
+ It is a customized template from the openapi-python-client tool's default template:
5
+ https://github.com/openapi-generators/openapi-python-client/blob/861ef5622f10fc96d240dc9becb0edf94e61446c/openapi_python_client/templates/endpoint_module.py.jinja
6
+
7
+ The main changes are:
8
+ - Update the API call responses to either return the successful response type or raise an error by:
9
+ - Updating the _parse_response function to raise an error if the response status code is not in the 2xx range
10
+ - Inject a client into every method via a context manager
11
+ """
12
+
13
+ from http import HTTPStatus
14
+ from typing import Any
15
+ from uuid import UUID
16
+
17
+ import httpx
18
+
19
+ from ...client import _client_context
20
+ from ...errors import get_error_for_response
21
+ from ...models.evaluation_response_regression_metrics import EvaluationResponseRegressionMetrics
22
+ from ...types import Response
23
+
24
+
25
+ def _get_kwargs(
26
+ model_name_or_id: str,
27
+ task_id: UUID,
28
+ ) -> dict[str, Any]:
29
+ _kwargs: dict[str, Any] = {
30
+ "method": "get",
31
+ "url": f"/regression_model/{model_name_or_id}/evaluation/{task_id}",
32
+ }
33
+
34
+ return _kwargs
35
+
36
+
37
+ def _parse_response(*, response: httpx.Response) -> EvaluationResponseRegressionMetrics:
38
+ if response.status_code == 200:
39
+ response_200 = EvaluationResponseRegressionMetrics.from_dict(response.json())
40
+
41
+ return response_200
42
+ if response.status_code == 401:
43
+ raise get_error_for_response(response)
44
+ if response.status_code == 500:
45
+ raise get_error_for_response(response)
46
+ if response.status_code == 503:
47
+ raise get_error_for_response(response)
48
+ if response.status_code == 404:
49
+ raise get_error_for_response(response)
50
+ else:
51
+ raise get_error_for_response(response)
52
+
53
+
54
+ def _build_response(*, response: httpx.Response) -> Response[EvaluationResponseRegressionMetrics]:
55
+ return Response(
56
+ status_code=HTTPStatus(response.status_code),
57
+ content=response.content,
58
+ headers=response.headers,
59
+ parsed=_parse_response(response=response),
60
+ )
61
+
62
+
63
+ def sync_detailed(
64
+ model_name_or_id: str,
65
+ task_id: UUID,
66
+ ) -> Response[EvaluationResponseRegressionMetrics]:
67
+ """Get Regression Model Evaluation
68
+
69
+ Args:
70
+ model_name_or_id (str):
71
+ task_id (UUID):
72
+
73
+ Raises:
74
+ errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
75
+ httpx.TimeoutException: If the request takes longer than Client.timeout.
76
+
77
+ Returns:
78
+ Response[EvaluationResponseRegressionMetrics]
79
+ """
80
+
81
+ kwargs = _get_kwargs(
82
+ model_name_or_id=model_name_or_id,
83
+ task_id=task_id,
84
+ )
85
+
86
+ with _client_context() as client:
87
+ response = client.get_httpx_client().request(
88
+ **kwargs,
89
+ )
90
+
91
+ return _build_response(response=response)
92
+
93
+
94
+ def sync(
95
+ model_name_or_id: str,
96
+ task_id: UUID,
97
+ ) -> EvaluationResponseRegressionMetrics:
98
+ """Get Regression Model Evaluation
99
+
100
+ Args:
101
+ model_name_or_id (str):
102
+ task_id (UUID):
103
+
104
+ Raises:
105
+ errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
106
+ httpx.TimeoutException: If the request takes longer than Client.timeout.
107
+
108
+ Returns:
109
+ EvaluationResponseRegressionMetrics
110
+ """
111
+
112
+ return sync_detailed(
113
+ model_name_or_id=model_name_or_id,
114
+ task_id=task_id,
115
+ ).parsed
116
+
117
+
118
+ async def asyncio_detailed(
119
+ model_name_or_id: str,
120
+ task_id: UUID,
121
+ ) -> Response[EvaluationResponseRegressionMetrics]:
122
+ """Get Regression Model Evaluation
123
+
124
+ Args:
125
+ model_name_or_id (str):
126
+ task_id (UUID):
127
+
128
+ Raises:
129
+ errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
130
+ httpx.TimeoutException: If the request takes longer than Client.timeout.
131
+
132
+ Returns:
133
+ Response[EvaluationResponseRegressionMetrics]
134
+ """
135
+
136
+ kwargs = _get_kwargs(
137
+ model_name_or_id=model_name_or_id,
138
+ task_id=task_id,
139
+ )
140
+
141
+ with _client_context() as client:
142
+ response = await client.get_async_httpx_client().request(**kwargs)
143
+
144
+ return _build_response(response=response)
145
+
146
+
147
+ async def asyncio(
148
+ model_name_or_id: str,
149
+ task_id: UUID,
150
+ ) -> EvaluationResponseRegressionMetrics:
151
+ """Get Regression Model Evaluation
152
+
153
+ Args:
154
+ model_name_or_id (str):
155
+ task_id (UUID):
156
+
157
+ Raises:
158
+ errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
159
+ httpx.TimeoutException: If the request takes longer than Client.timeout.
160
+
161
+ Returns:
162
+ EvaluationResponseRegressionMetrics
163
+ """
164
+
165
+ return (
166
+ await asyncio_detailed(
167
+ model_name_or_id=model_name_or_id,
168
+ task_id=task_id,
169
+ )
170
+ ).parsed
@@ -0,0 +1,156 @@
1
+ """
2
+ This file is generated by the openapi-python-client tool via the generate_api_client.py script
3
+
4
+ It is a customized template from the openapi-python-client tool's default template:
5
+ https://github.com/openapi-generators/openapi-python-client/blob/861ef5622f10fc96d240dc9becb0edf94e61446c/openapi_python_client/templates/endpoint_module.py.jinja
6
+
7
+ The main changes are:
8
+ - Update the API call responses to either return the successful response type or raise an error by:
9
+ - Updating the _parse_response function to raise an error if the response status code is not in the 2xx range
10
+ - Inject a client into every method via a context manager
11
+ """
12
+
13
+ from http import HTTPStatus
14
+ from typing import Any
15
+
16
+ import httpx
17
+
18
+ from ...client import _client_context
19
+ from ...errors import get_error_for_response
20
+ from ...models.regression_model_metadata import RegressionModelMetadata
21
+ from ...types import Response
22
+
23
+
24
+ def _get_kwargs(
25
+ name_or_id: str,
26
+ ) -> dict[str, Any]:
27
+ _kwargs: dict[str, Any] = {
28
+ "method": "get",
29
+ "url": f"/regression_model/{name_or_id}",
30
+ }
31
+
32
+ return _kwargs
33
+
34
+
35
+ def _parse_response(*, response: httpx.Response) -> RegressionModelMetadata:
36
+ if response.status_code == 200:
37
+ response_200 = RegressionModelMetadata.from_dict(response.json())
38
+
39
+ return response_200
40
+ if response.status_code == 401:
41
+ raise get_error_for_response(response)
42
+ if response.status_code == 500:
43
+ raise get_error_for_response(response)
44
+ if response.status_code == 503:
45
+ raise get_error_for_response(response)
46
+ if response.status_code == 404:
47
+ raise get_error_for_response(response)
48
+ else:
49
+ raise get_error_for_response(response)
50
+
51
+
52
+ def _build_response(*, response: httpx.Response) -> Response[RegressionModelMetadata]:
53
+ return Response(
54
+ status_code=HTTPStatus(response.status_code),
55
+ content=response.content,
56
+ headers=response.headers,
57
+ parsed=_parse_response(response=response),
58
+ )
59
+
60
+
61
+ def sync_detailed(
62
+ name_or_id: str,
63
+ ) -> Response[RegressionModelMetadata]:
64
+ """Get Regression Model
65
+
66
+ Args:
67
+ name_or_id (str):
68
+
69
+ Raises:
70
+ errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
71
+ httpx.TimeoutException: If the request takes longer than Client.timeout.
72
+
73
+ Returns:
74
+ Response[RegressionModelMetadata]
75
+ """
76
+
77
+ kwargs = _get_kwargs(
78
+ name_or_id=name_or_id,
79
+ )
80
+
81
+ with _client_context() as client:
82
+ response = client.get_httpx_client().request(
83
+ **kwargs,
84
+ )
85
+
86
+ return _build_response(response=response)
87
+
88
+
89
+ def sync(
90
+ name_or_id: str,
91
+ ) -> RegressionModelMetadata:
92
+ """Get Regression Model
93
+
94
+ Args:
95
+ name_or_id (str):
96
+
97
+ Raises:
98
+ errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
99
+ httpx.TimeoutException: If the request takes longer than Client.timeout.
100
+
101
+ Returns:
102
+ RegressionModelMetadata
103
+ """
104
+
105
+ return sync_detailed(
106
+ name_or_id=name_or_id,
107
+ ).parsed
108
+
109
+
110
+ async def asyncio_detailed(
111
+ name_or_id: str,
112
+ ) -> Response[RegressionModelMetadata]:
113
+ """Get Regression Model
114
+
115
+ Args:
116
+ name_or_id (str):
117
+
118
+ Raises:
119
+ errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
120
+ httpx.TimeoutException: If the request takes longer than Client.timeout.
121
+
122
+ Returns:
123
+ Response[RegressionModelMetadata]
124
+ """
125
+
126
+ kwargs = _get_kwargs(
127
+ name_or_id=name_or_id,
128
+ )
129
+
130
+ with _client_context() as client:
131
+ response = await client.get_async_httpx_client().request(**kwargs)
132
+
133
+ return _build_response(response=response)
134
+
135
+
136
+ async def asyncio(
137
+ name_or_id: str,
138
+ ) -> RegressionModelMetadata:
139
+ """Get Regression Model
140
+
141
+ Args:
142
+ name_or_id (str):
143
+
144
+ Raises:
145
+ errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
146
+ httpx.TimeoutException: If the request takes longer than Client.timeout.
147
+
148
+ Returns:
149
+ RegressionModelMetadata
150
+ """
151
+
152
+ return (
153
+ await asyncio_detailed(
154
+ name_or_id=name_or_id,
155
+ )
156
+ ).parsed
@@ -0,0 +1,161 @@
1
+ """
2
+ This file is generated by the openapi-python-client tool via the generate_api_client.py script
3
+
4
+ It is a customized template from the openapi-python-client tool's default template:
5
+ https://github.com/openapi-generators/openapi-python-client/blob/861ef5622f10fc96d240dc9becb0edf94e61446c/openapi_python_client/templates/endpoint_module.py.jinja
6
+
7
+ The main changes are:
8
+ - Update the API call responses to either return the successful response type or raise an error by:
9
+ - Updating the _parse_response function to raise an error if the response status code is not in the 2xx range
10
+ - Inject a client into every method via a context manager
11
+ """
12
+
13
+ from http import HTTPStatus
14
+ from typing import Any, List
15
+
16
+ import httpx
17
+
18
+ from ...client import _client_context
19
+ from ...errors import get_error_for_response
20
+ from ...models.evaluation_response_regression_metrics import EvaluationResponseRegressionMetrics
21
+ from ...types import Response
22
+
23
+
24
+ def _get_kwargs(
25
+ model_name_or_id: str,
26
+ ) -> dict[str, Any]:
27
+ _kwargs: dict[str, Any] = {
28
+ "method": "get",
29
+ "url": f"/regression_model/{model_name_or_id}/evaluation",
30
+ }
31
+
32
+ return _kwargs
33
+
34
+
35
+ def _parse_response(*, response: httpx.Response) -> List["EvaluationResponseRegressionMetrics"]:
36
+ if response.status_code == 200:
37
+ response_200 = []
38
+ _response_200 = response.json()
39
+ for response_200_item_data in _response_200:
40
+ response_200_item = EvaluationResponseRegressionMetrics.from_dict(response_200_item_data)
41
+
42
+ response_200.append(response_200_item)
43
+
44
+ return response_200
45
+ if response.status_code == 401:
46
+ raise get_error_for_response(response)
47
+ if response.status_code == 500:
48
+ raise get_error_for_response(response)
49
+ if response.status_code == 503:
50
+ raise get_error_for_response(response)
51
+ if response.status_code == 404:
52
+ raise get_error_for_response(response)
53
+ else:
54
+ raise get_error_for_response(response)
55
+
56
+
57
+ def _build_response(*, response: httpx.Response) -> Response[List["EvaluationResponseRegressionMetrics"]]:
58
+ return Response(
59
+ status_code=HTTPStatus(response.status_code),
60
+ content=response.content,
61
+ headers=response.headers,
62
+ parsed=_parse_response(response=response),
63
+ )
64
+
65
+
66
+ def sync_detailed(
67
+ model_name_or_id: str,
68
+ ) -> Response[List["EvaluationResponseRegressionMetrics"]]:
69
+ """List Regression Model Evaluations
70
+
71
+ Args:
72
+ model_name_or_id (str):
73
+
74
+ Raises:
75
+ errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
76
+ httpx.TimeoutException: If the request takes longer than Client.timeout.
77
+
78
+ Returns:
79
+ Response[List['EvaluationResponseRegressionMetrics']]
80
+ """
81
+
82
+ kwargs = _get_kwargs(
83
+ model_name_or_id=model_name_or_id,
84
+ )
85
+
86
+ with _client_context() as client:
87
+ response = client.get_httpx_client().request(
88
+ **kwargs,
89
+ )
90
+
91
+ return _build_response(response=response)
92
+
93
+
94
+ def sync(
95
+ model_name_or_id: str,
96
+ ) -> List["EvaluationResponseRegressionMetrics"]:
97
+ """List Regression Model Evaluations
98
+
99
+ Args:
100
+ model_name_or_id (str):
101
+
102
+ Raises:
103
+ errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
104
+ httpx.TimeoutException: If the request takes longer than Client.timeout.
105
+
106
+ Returns:
107
+ List['EvaluationResponseRegressionMetrics']
108
+ """
109
+
110
+ return sync_detailed(
111
+ model_name_or_id=model_name_or_id,
112
+ ).parsed
113
+
114
+
115
+ async def asyncio_detailed(
116
+ model_name_or_id: str,
117
+ ) -> Response[List["EvaluationResponseRegressionMetrics"]]:
118
+ """List Regression Model Evaluations
119
+
120
+ Args:
121
+ model_name_or_id (str):
122
+
123
+ Raises:
124
+ errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
125
+ httpx.TimeoutException: If the request takes longer than Client.timeout.
126
+
127
+ Returns:
128
+ Response[List['EvaluationResponseRegressionMetrics']]
129
+ """
130
+
131
+ kwargs = _get_kwargs(
132
+ model_name_or_id=model_name_or_id,
133
+ )
134
+
135
+ with _client_context() as client:
136
+ response = await client.get_async_httpx_client().request(**kwargs)
137
+
138
+ return _build_response(response=response)
139
+
140
+
141
+ async def asyncio(
142
+ model_name_or_id: str,
143
+ ) -> List["EvaluationResponseRegressionMetrics"]:
144
+ """List Regression Model Evaluations
145
+
146
+ Args:
147
+ model_name_or_id (str):
148
+
149
+ Raises:
150
+ errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
151
+ httpx.TimeoutException: If the request takes longer than Client.timeout.
152
+
153
+ Returns:
154
+ List['EvaluationResponseRegressionMetrics']
155
+ """
156
+
157
+ return (
158
+ await asyncio_detailed(
159
+ model_name_or_id=model_name_or_id,
160
+ )
161
+ ).parsed