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.
- orca_sdk/__init__.py +13 -4
- orca_sdk/_generated_api_client/api/__init__.py +80 -34
- orca_sdk/_generated_api_client/api/classification_model/create_classification_model_classification_model_post.py +170 -0
- 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
- 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
- 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
- orca_sdk/_generated_api_client/api/classification_model/get_classification_model_classification_model_name_or_id_get.py +156 -0
- 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
- 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
- orca_sdk/_generated_api_client/api/classification_model/list_classification_models_classification_model_get.py +127 -0
- 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
- orca_sdk/_generated_api_client/api/classification_model/update_classification_model_classification_model_name_or_id_patch.py +183 -0
- orca_sdk/_generated_api_client/api/datasource/download_datasource_datasource_name_or_id_download_get.py +24 -0
- orca_sdk/_generated_api_client/api/memoryset/clone_memoryset_memoryset_name_or_id_clone_post.py +22 -22
- orca_sdk/_generated_api_client/api/memoryset/create_memoryset_memoryset_post.py +22 -22
- orca_sdk/_generated_api_client/api/memoryset/get_memories_memoryset_name_or_id_memories_get_post.py +38 -16
- orca_sdk/_generated_api_client/api/memoryset/get_memory_memoryset_name_or_id_memory_memory_id_get.py +29 -12
- orca_sdk/_generated_api_client/api/memoryset/get_memoryset_memoryset_name_or_id_get.py +12 -12
- orca_sdk/_generated_api_client/api/memoryset/insert_memories_gpu_memoryset_name_or_id_memory_post.py +17 -14
- orca_sdk/_generated_api_client/api/memoryset/list_memorysets_memoryset_get.py +72 -19
- orca_sdk/_generated_api_client/api/memoryset/memoryset_lookup_gpu_memoryset_name_or_id_lookup_post.py +31 -12
- orca_sdk/_generated_api_client/api/memoryset/potential_duplicate_groups_memoryset_name_or_id_potential_duplicate_groups_get.py +49 -20
- orca_sdk/_generated_api_client/api/memoryset/query_memoryset_memoryset_name_or_id_memories_post.py +38 -16
- orca_sdk/_generated_api_client/api/memoryset/update_memories_gpu_memoryset_name_or_id_memories_patch.py +54 -29
- orca_sdk/_generated_api_client/api/memoryset/update_memory_gpu_memoryset_name_or_id_memory_patch.py +44 -26
- orca_sdk/_generated_api_client/api/memoryset/update_memoryset_memoryset_name_or_id_patch.py +22 -22
- orca_sdk/_generated_api_client/api/predictive_model/__init__.py +0 -0
- orca_sdk/_generated_api_client/api/predictive_model/list_predictive_models_predictive_model_get.py +150 -0
- orca_sdk/_generated_api_client/api/regression_model/__init__.py +0 -0
- 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
- 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
- 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
- orca_sdk/_generated_api_client/api/regression_model/evaluate_regression_model_regression_model_model_name_or_id_evaluation_post.py +183 -0
- 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
- orca_sdk/_generated_api_client/api/regression_model/get_regression_model_regression_model_name_or_id_get.py +156 -0
- orca_sdk/_generated_api_client/api/regression_model/list_regression_model_evaluations_regression_model_model_name_or_id_evaluation_get.py +161 -0
- 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
- orca_sdk/_generated_api_client/api/regression_model/predict_score_gpu_regression_model_name_or_id_prediction_post.py +190 -0
- 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
- orca_sdk/_generated_api_client/api/task/get_task_task_task_id_get.py +156 -0
- orca_sdk/_generated_api_client/api/telemetry/get_prediction_telemetry_prediction_prediction_id_get.py +35 -12
- orca_sdk/_generated_api_client/api/telemetry/list_memories_with_feedback_telemetry_memories_post.py +20 -12
- orca_sdk/_generated_api_client/api/telemetry/list_predictions_telemetry_prediction_post.py +35 -12
- orca_sdk/_generated_api_client/models/__init__.py +84 -24
- orca_sdk/_generated_api_client/models/base_score_prediction_result.py +108 -0
- orca_sdk/_generated_api_client/models/{evaluation_request.py → classification_evaluation_request.py} +13 -45
- orca_sdk/_generated_api_client/models/{classification_evaluation_result.py → classification_metrics.py} +106 -56
- orca_sdk/_generated_api_client/models/{rac_model_metadata.py → classification_model_metadata.py} +51 -43
- orca_sdk/_generated_api_client/models/{prediction_request.py → classification_prediction_request.py} +31 -6
- orca_sdk/_generated_api_client/models/{clone_labeled_memoryset_request.py → clone_memoryset_request.py} +5 -5
- orca_sdk/_generated_api_client/models/column_info.py +31 -0
- orca_sdk/_generated_api_client/models/{create_rac_model_request.py → create_classification_model_request.py} +25 -57
- orca_sdk/_generated_api_client/models/{create_labeled_memoryset_request.py → create_memoryset_request.py} +73 -56
- orca_sdk/_generated_api_client/models/create_memoryset_request_index_params.py +66 -0
- orca_sdk/_generated_api_client/models/create_memoryset_request_index_type.py +13 -0
- orca_sdk/_generated_api_client/models/create_regression_model_request.py +137 -0
- orca_sdk/_generated_api_client/models/embedding_evaluation_payload.py +187 -0
- orca_sdk/_generated_api_client/models/embedding_evaluation_response.py +10 -0
- orca_sdk/_generated_api_client/models/evaluation_response.py +22 -9
- orca_sdk/_generated_api_client/models/evaluation_response_classification_metrics.py +140 -0
- orca_sdk/_generated_api_client/models/evaluation_response_regression_metrics.py +140 -0
- orca_sdk/_generated_api_client/models/memory_type.py +9 -0
- orca_sdk/_generated_api_client/models/{labeled_memoryset_metadata.py → memoryset_metadata.py} +73 -13
- orca_sdk/_generated_api_client/models/memoryset_metadata_index_params.py +55 -0
- orca_sdk/_generated_api_client/models/memoryset_metadata_index_type.py +13 -0
- orca_sdk/_generated_api_client/models/{labeled_memoryset_update.py → memoryset_update.py} +19 -31
- orca_sdk/_generated_api_client/models/not_found_error_response_resource_type_0.py +1 -0
- 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
- orca_sdk/_generated_api_client/models/{precision_recall_curve.py → pr_curve.py} +5 -13
- orca_sdk/_generated_api_client/models/{rac_model_update.py → predictive_model_update.py} +14 -5
- orca_sdk/_generated_api_client/models/pretrained_embedding_model_metadata.py +11 -1
- orca_sdk/_generated_api_client/models/rar_head_type.py +8 -0
- orca_sdk/_generated_api_client/models/regression_evaluation_request.py +148 -0
- orca_sdk/_generated_api_client/models/regression_metrics.py +172 -0
- orca_sdk/_generated_api_client/models/regression_model_metadata.py +177 -0
- orca_sdk/_generated_api_client/models/regression_prediction_request.py +195 -0
- orca_sdk/_generated_api_client/models/roc_curve.py +0 -8
- orca_sdk/_generated_api_client/models/score_prediction_memory_lookup.py +196 -0
- orca_sdk/_generated_api_client/models/score_prediction_memory_lookup_metadata.py +68 -0
- orca_sdk/_generated_api_client/models/score_prediction_with_memories_and_feedback.py +252 -0
- orca_sdk/_generated_api_client/models/scored_memory.py +172 -0
- orca_sdk/_generated_api_client/models/scored_memory_insert.py +128 -0
- orca_sdk/_generated_api_client/models/scored_memory_insert_metadata.py +68 -0
- orca_sdk/_generated_api_client/models/scored_memory_lookup.py +180 -0
- orca_sdk/_generated_api_client/models/scored_memory_lookup_metadata.py +68 -0
- orca_sdk/_generated_api_client/models/scored_memory_metadata.py +68 -0
- orca_sdk/_generated_api_client/models/scored_memory_update.py +171 -0
- orca_sdk/_generated_api_client/models/scored_memory_update_metadata_type_0.py +68 -0
- orca_sdk/_generated_api_client/models/scored_memory_with_feedback_metrics.py +193 -0
- orca_sdk/_generated_api_client/models/scored_memory_with_feedback_metrics_feedback_metrics.py +68 -0
- orca_sdk/_generated_api_client/models/scored_memory_with_feedback_metrics_metadata.py +68 -0
- orca_sdk/_generated_api_client/models/update_prediction_request.py +20 -0
- orca_sdk/_shared/__init__.py +9 -1
- orca_sdk/_shared/metrics.py +257 -87
- orca_sdk/_shared/metrics_test.py +136 -77
- orca_sdk/_utils/data_parsing.py +0 -3
- orca_sdk/_utils/data_parsing_test.py +0 -3
- orca_sdk/_utils/prediction_result_ui.py +55 -23
- orca_sdk/classification_model.py +183 -175
- orca_sdk/classification_model_test.py +147 -157
- orca_sdk/conftest.py +76 -26
- orca_sdk/datasource_test.py +0 -1
- orca_sdk/embedding_model.py +136 -14
- orca_sdk/embedding_model_test.py +10 -6
- orca_sdk/job.py +329 -0
- orca_sdk/job_test.py +48 -0
- orca_sdk/memoryset.py +882 -161
- orca_sdk/memoryset_test.py +56 -23
- orca_sdk/regression_model.py +647 -0
- orca_sdk/regression_model_test.py +338 -0
- orca_sdk/telemetry.py +223 -106
- orca_sdk/telemetry_test.py +34 -30
- {orca_sdk-0.0.94.dist-info → orca_sdk-0.0.95.dist-info}/METADATA +2 -4
- {orca_sdk-0.0.94.dist-info → orca_sdk-0.0.95.dist-info}/RECORD +115 -69
- orca_sdk/_utils/task.py +0 -73
- {orca_sdk-0.0.94.dist-info → orca_sdk-0.0.95.dist-info}/WHEEL +0 -0
|
@@ -17,15 +17,15 @@ import httpx
|
|
|
17
17
|
|
|
18
18
|
from ...client import _client_context
|
|
19
19
|
from ...errors import get_error_for_response
|
|
20
|
-
from ...models.
|
|
21
|
-
from ...models.
|
|
20
|
+
from ...models.memoryset_metadata import MemorysetMetadata
|
|
21
|
+
from ...models.memoryset_update import MemorysetUpdate
|
|
22
22
|
from ...types import Response
|
|
23
23
|
|
|
24
24
|
|
|
25
25
|
def _get_kwargs(
|
|
26
26
|
name_or_id: str,
|
|
27
27
|
*,
|
|
28
|
-
body:
|
|
28
|
+
body: MemorysetUpdate,
|
|
29
29
|
) -> dict[str, Any]:
|
|
30
30
|
headers: Dict[str, Any] = {}
|
|
31
31
|
|
|
@@ -43,9 +43,9 @@ def _get_kwargs(
|
|
|
43
43
|
return _kwargs
|
|
44
44
|
|
|
45
45
|
|
|
46
|
-
def _parse_response(*, response: httpx.Response) ->
|
|
46
|
+
def _parse_response(*, response: httpx.Response) -> MemorysetMetadata:
|
|
47
47
|
if response.status_code == 200:
|
|
48
|
-
response_200 =
|
|
48
|
+
response_200 = MemorysetMetadata.from_dict(response.json())
|
|
49
49
|
|
|
50
50
|
return response_200
|
|
51
51
|
if response.status_code == 401:
|
|
@@ -60,7 +60,7 @@ def _parse_response(*, response: httpx.Response) -> LabeledMemorysetMetadata:
|
|
|
60
60
|
raise get_error_for_response(response)
|
|
61
61
|
|
|
62
62
|
|
|
63
|
-
def _build_response(*, response: httpx.Response) -> Response[
|
|
63
|
+
def _build_response(*, response: httpx.Response) -> Response[MemorysetMetadata]:
|
|
64
64
|
return Response(
|
|
65
65
|
status_code=HTTPStatus(response.status_code),
|
|
66
66
|
content=response.content,
|
|
@@ -72,20 +72,20 @@ def _build_response(*, response: httpx.Response) -> Response[LabeledMemorysetMet
|
|
|
72
72
|
def sync_detailed(
|
|
73
73
|
name_or_id: str,
|
|
74
74
|
*,
|
|
75
|
-
body:
|
|
76
|
-
) -> Response[
|
|
75
|
+
body: MemorysetUpdate,
|
|
76
|
+
) -> Response[MemorysetMetadata]:
|
|
77
77
|
"""Update Memoryset
|
|
78
78
|
|
|
79
79
|
Args:
|
|
80
80
|
name_or_id (str):
|
|
81
|
-
body (
|
|
81
|
+
body (MemorysetUpdate):
|
|
82
82
|
|
|
83
83
|
Raises:
|
|
84
84
|
errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
|
|
85
85
|
httpx.TimeoutException: If the request takes longer than Client.timeout.
|
|
86
86
|
|
|
87
87
|
Returns:
|
|
88
|
-
Response[
|
|
88
|
+
Response[MemorysetMetadata]
|
|
89
89
|
"""
|
|
90
90
|
|
|
91
91
|
kwargs = _get_kwargs(
|
|
@@ -104,20 +104,20 @@ def sync_detailed(
|
|
|
104
104
|
def sync(
|
|
105
105
|
name_or_id: str,
|
|
106
106
|
*,
|
|
107
|
-
body:
|
|
108
|
-
) ->
|
|
107
|
+
body: MemorysetUpdate,
|
|
108
|
+
) -> MemorysetMetadata:
|
|
109
109
|
"""Update Memoryset
|
|
110
110
|
|
|
111
111
|
Args:
|
|
112
112
|
name_or_id (str):
|
|
113
|
-
body (
|
|
113
|
+
body (MemorysetUpdate):
|
|
114
114
|
|
|
115
115
|
Raises:
|
|
116
116
|
errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
|
|
117
117
|
httpx.TimeoutException: If the request takes longer than Client.timeout.
|
|
118
118
|
|
|
119
119
|
Returns:
|
|
120
|
-
|
|
120
|
+
MemorysetMetadata
|
|
121
121
|
"""
|
|
122
122
|
|
|
123
123
|
return sync_detailed(
|
|
@@ -129,20 +129,20 @@ def sync(
|
|
|
129
129
|
async def asyncio_detailed(
|
|
130
130
|
name_or_id: str,
|
|
131
131
|
*,
|
|
132
|
-
body:
|
|
133
|
-
) -> Response[
|
|
132
|
+
body: MemorysetUpdate,
|
|
133
|
+
) -> Response[MemorysetMetadata]:
|
|
134
134
|
"""Update Memoryset
|
|
135
135
|
|
|
136
136
|
Args:
|
|
137
137
|
name_or_id (str):
|
|
138
|
-
body (
|
|
138
|
+
body (MemorysetUpdate):
|
|
139
139
|
|
|
140
140
|
Raises:
|
|
141
141
|
errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
|
|
142
142
|
httpx.TimeoutException: If the request takes longer than Client.timeout.
|
|
143
143
|
|
|
144
144
|
Returns:
|
|
145
|
-
Response[
|
|
145
|
+
Response[MemorysetMetadata]
|
|
146
146
|
"""
|
|
147
147
|
|
|
148
148
|
kwargs = _get_kwargs(
|
|
@@ -159,20 +159,20 @@ async def asyncio_detailed(
|
|
|
159
159
|
async def asyncio(
|
|
160
160
|
name_or_id: str,
|
|
161
161
|
*,
|
|
162
|
-
body:
|
|
163
|
-
) ->
|
|
162
|
+
body: MemorysetUpdate,
|
|
163
|
+
) -> MemorysetMetadata:
|
|
164
164
|
"""Update Memoryset
|
|
165
165
|
|
|
166
166
|
Args:
|
|
167
167
|
name_or_id (str):
|
|
168
|
-
body (
|
|
168
|
+
body (MemorysetUpdate):
|
|
169
169
|
|
|
170
170
|
Raises:
|
|
171
171
|
errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
|
|
172
172
|
httpx.TimeoutException: If the request takes longer than Client.timeout.
|
|
173
173
|
|
|
174
174
|
Returns:
|
|
175
|
-
|
|
175
|
+
MemorysetMetadata
|
|
176
176
|
"""
|
|
177
177
|
|
|
178
178
|
return (
|
|
File without changes
|
orca_sdk/_generated_api_client/api/predictive_model/list_predictive_models_predictive_model_get.py
ADDED
|
@@ -0,0 +1,150 @@
|
|
|
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, Union
|
|
15
|
+
|
|
16
|
+
import httpx
|
|
17
|
+
|
|
18
|
+
from ...client import _client_context
|
|
19
|
+
from ...errors import get_error_for_response
|
|
20
|
+
from ...models.classification_model_metadata import ClassificationModelMetadata
|
|
21
|
+
from ...models.regression_model_metadata import RegressionModelMetadata
|
|
22
|
+
from ...types import Response
|
|
23
|
+
|
|
24
|
+
|
|
25
|
+
def _get_kwargs() -> dict[str, Any]:
|
|
26
|
+
_kwargs: dict[str, Any] = {
|
|
27
|
+
"method": "get",
|
|
28
|
+
"url": "/predictive_model/",
|
|
29
|
+
}
|
|
30
|
+
|
|
31
|
+
return _kwargs
|
|
32
|
+
|
|
33
|
+
|
|
34
|
+
def _parse_response(
|
|
35
|
+
*, response: httpx.Response
|
|
36
|
+
) -> List[Union["ClassificationModelMetadata", "RegressionModelMetadata"]]:
|
|
37
|
+
if response.status_code == 200:
|
|
38
|
+
response_200 = []
|
|
39
|
+
_response_200 = response.json()
|
|
40
|
+
for response_200_item_data in _response_200:
|
|
41
|
+
|
|
42
|
+
def _parse_response_200_item(
|
|
43
|
+
data: object,
|
|
44
|
+
) -> Union["ClassificationModelMetadata", "RegressionModelMetadata"]:
|
|
45
|
+
try:
|
|
46
|
+
if not isinstance(data, dict):
|
|
47
|
+
raise TypeError()
|
|
48
|
+
response_200_item_type_0 = ClassificationModelMetadata.from_dict(data)
|
|
49
|
+
|
|
50
|
+
return response_200_item_type_0
|
|
51
|
+
except: # noqa: E722
|
|
52
|
+
pass
|
|
53
|
+
if not isinstance(data, dict):
|
|
54
|
+
raise TypeError()
|
|
55
|
+
response_200_item_type_1 = RegressionModelMetadata.from_dict(data)
|
|
56
|
+
|
|
57
|
+
return response_200_item_type_1
|
|
58
|
+
|
|
59
|
+
response_200_item = _parse_response_200_item(response_200_item_data)
|
|
60
|
+
|
|
61
|
+
response_200.append(response_200_item)
|
|
62
|
+
|
|
63
|
+
return response_200
|
|
64
|
+
if response.status_code == 401:
|
|
65
|
+
raise get_error_for_response(response)
|
|
66
|
+
if response.status_code == 500:
|
|
67
|
+
raise get_error_for_response(response)
|
|
68
|
+
if response.status_code == 503:
|
|
69
|
+
raise get_error_for_response(response)
|
|
70
|
+
else:
|
|
71
|
+
raise get_error_for_response(response)
|
|
72
|
+
|
|
73
|
+
|
|
74
|
+
def _build_response(
|
|
75
|
+
*, response: httpx.Response
|
|
76
|
+
) -> Response[List[Union["ClassificationModelMetadata", "RegressionModelMetadata"]]]:
|
|
77
|
+
return Response(
|
|
78
|
+
status_code=HTTPStatus(response.status_code),
|
|
79
|
+
content=response.content,
|
|
80
|
+
headers=response.headers,
|
|
81
|
+
parsed=_parse_response(response=response),
|
|
82
|
+
)
|
|
83
|
+
|
|
84
|
+
|
|
85
|
+
def sync_detailed() -> Response[List[Union["ClassificationModelMetadata", "RegressionModelMetadata"]]]:
|
|
86
|
+
"""List Predictive Models
|
|
87
|
+
|
|
88
|
+
Raises:
|
|
89
|
+
errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
|
|
90
|
+
httpx.TimeoutException: If the request takes longer than Client.timeout.
|
|
91
|
+
|
|
92
|
+
Returns:
|
|
93
|
+
Response[List[Union['ClassificationModelMetadata', 'RegressionModelMetadata']]]
|
|
94
|
+
"""
|
|
95
|
+
|
|
96
|
+
kwargs = _get_kwargs()
|
|
97
|
+
|
|
98
|
+
with _client_context() as client:
|
|
99
|
+
response = client.get_httpx_client().request(
|
|
100
|
+
**kwargs,
|
|
101
|
+
)
|
|
102
|
+
|
|
103
|
+
return _build_response(response=response)
|
|
104
|
+
|
|
105
|
+
|
|
106
|
+
def sync() -> List[Union["ClassificationModelMetadata", "RegressionModelMetadata"]]:
|
|
107
|
+
"""List Predictive Models
|
|
108
|
+
|
|
109
|
+
Raises:
|
|
110
|
+
errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
|
|
111
|
+
httpx.TimeoutException: If the request takes longer than Client.timeout.
|
|
112
|
+
|
|
113
|
+
Returns:
|
|
114
|
+
List[Union['ClassificationModelMetadata', 'RegressionModelMetadata']]
|
|
115
|
+
"""
|
|
116
|
+
|
|
117
|
+
return sync_detailed().parsed
|
|
118
|
+
|
|
119
|
+
|
|
120
|
+
async def asyncio_detailed() -> Response[List[Union["ClassificationModelMetadata", "RegressionModelMetadata"]]]:
|
|
121
|
+
"""List Predictive Models
|
|
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[Union['ClassificationModelMetadata', 'RegressionModelMetadata']]]
|
|
129
|
+
"""
|
|
130
|
+
|
|
131
|
+
kwargs = _get_kwargs()
|
|
132
|
+
|
|
133
|
+
with _client_context() as client:
|
|
134
|
+
response = await client.get_async_httpx_client().request(**kwargs)
|
|
135
|
+
|
|
136
|
+
return _build_response(response=response)
|
|
137
|
+
|
|
138
|
+
|
|
139
|
+
async def asyncio() -> List[Union["ClassificationModelMetadata", "RegressionModelMetadata"]]:
|
|
140
|
+
"""List Predictive Models
|
|
141
|
+
|
|
142
|
+
Raises:
|
|
143
|
+
errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
|
|
144
|
+
httpx.TimeoutException: If the request takes longer than Client.timeout.
|
|
145
|
+
|
|
146
|
+
Returns:
|
|
147
|
+
List[Union['ClassificationModelMetadata', 'RegressionModelMetadata']]
|
|
148
|
+
"""
|
|
149
|
+
|
|
150
|
+
return (await asyncio_detailed()).parsed
|
|
File without changes
|
|
@@ -17,20 +17,20 @@ import httpx
|
|
|
17
17
|
|
|
18
18
|
from ...client import _client_context
|
|
19
19
|
from ...errors import get_error_for_response
|
|
20
|
-
from ...models.
|
|
21
|
-
from ...models.
|
|
20
|
+
from ...models.create_regression_model_request import CreateRegressionModelRequest
|
|
21
|
+
from ...models.regression_model_metadata import RegressionModelMetadata
|
|
22
22
|
from ...types import Response
|
|
23
23
|
|
|
24
24
|
|
|
25
25
|
def _get_kwargs(
|
|
26
26
|
*,
|
|
27
|
-
body:
|
|
27
|
+
body: CreateRegressionModelRequest,
|
|
28
28
|
) -> dict[str, Any]:
|
|
29
29
|
headers: Dict[str, Any] = {}
|
|
30
30
|
|
|
31
31
|
_kwargs: dict[str, Any] = {
|
|
32
32
|
"method": "post",
|
|
33
|
-
"url": "/
|
|
33
|
+
"url": "/regression_model/",
|
|
34
34
|
}
|
|
35
35
|
|
|
36
36
|
_body = body.to_dict()
|
|
@@ -42,9 +42,9 @@ def _get_kwargs(
|
|
|
42
42
|
return _kwargs
|
|
43
43
|
|
|
44
44
|
|
|
45
|
-
def _parse_response(*, response: httpx.Response) ->
|
|
45
|
+
def _parse_response(*, response: httpx.Response) -> RegressionModelMetadata:
|
|
46
46
|
if response.status_code == 201:
|
|
47
|
-
response_201 =
|
|
47
|
+
response_201 = RegressionModelMetadata.from_dict(response.json())
|
|
48
48
|
|
|
49
49
|
return response_201
|
|
50
50
|
if response.status_code == 401:
|
|
@@ -59,7 +59,7 @@ def _parse_response(*, response: httpx.Response) -> RACModelMetadata:
|
|
|
59
59
|
raise get_error_for_response(response)
|
|
60
60
|
|
|
61
61
|
|
|
62
|
-
def _build_response(*, response: httpx.Response) -> Response[
|
|
62
|
+
def _build_response(*, response: httpx.Response) -> Response[RegressionModelMetadata]:
|
|
63
63
|
return Response(
|
|
64
64
|
status_code=HTTPStatus(response.status_code),
|
|
65
65
|
content=response.content,
|
|
@@ -70,19 +70,19 @@ def _build_response(*, response: httpx.Response) -> Response[RACModelMetadata]:
|
|
|
70
70
|
|
|
71
71
|
def sync_detailed(
|
|
72
72
|
*,
|
|
73
|
-
body:
|
|
74
|
-
) -> Response[
|
|
75
|
-
"""Create Model
|
|
73
|
+
body: CreateRegressionModelRequest,
|
|
74
|
+
) -> Response[RegressionModelMetadata]:
|
|
75
|
+
"""Create Regression Model
|
|
76
76
|
|
|
77
77
|
Args:
|
|
78
|
-
body (
|
|
78
|
+
body (CreateRegressionModelRequest):
|
|
79
79
|
|
|
80
80
|
Raises:
|
|
81
81
|
errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
|
|
82
82
|
httpx.TimeoutException: If the request takes longer than Client.timeout.
|
|
83
83
|
|
|
84
84
|
Returns:
|
|
85
|
-
Response[
|
|
85
|
+
Response[RegressionModelMetadata]
|
|
86
86
|
"""
|
|
87
87
|
|
|
88
88
|
kwargs = _get_kwargs(
|
|
@@ -99,19 +99,19 @@ def sync_detailed(
|
|
|
99
99
|
|
|
100
100
|
def sync(
|
|
101
101
|
*,
|
|
102
|
-
body:
|
|
103
|
-
) ->
|
|
104
|
-
"""Create Model
|
|
102
|
+
body: CreateRegressionModelRequest,
|
|
103
|
+
) -> RegressionModelMetadata:
|
|
104
|
+
"""Create Regression Model
|
|
105
105
|
|
|
106
106
|
Args:
|
|
107
|
-
body (
|
|
107
|
+
body (CreateRegressionModelRequest):
|
|
108
108
|
|
|
109
109
|
Raises:
|
|
110
110
|
errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
|
|
111
111
|
httpx.TimeoutException: If the request takes longer than Client.timeout.
|
|
112
112
|
|
|
113
113
|
Returns:
|
|
114
|
-
|
|
114
|
+
RegressionModelMetadata
|
|
115
115
|
"""
|
|
116
116
|
|
|
117
117
|
return sync_detailed(
|
|
@@ -121,19 +121,19 @@ def sync(
|
|
|
121
121
|
|
|
122
122
|
async def asyncio_detailed(
|
|
123
123
|
*,
|
|
124
|
-
body:
|
|
125
|
-
) -> Response[
|
|
126
|
-
"""Create Model
|
|
124
|
+
body: CreateRegressionModelRequest,
|
|
125
|
+
) -> Response[RegressionModelMetadata]:
|
|
126
|
+
"""Create Regression Model
|
|
127
127
|
|
|
128
128
|
Args:
|
|
129
|
-
body (
|
|
129
|
+
body (CreateRegressionModelRequest):
|
|
130
130
|
|
|
131
131
|
Raises:
|
|
132
132
|
errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
|
|
133
133
|
httpx.TimeoutException: If the request takes longer than Client.timeout.
|
|
134
134
|
|
|
135
135
|
Returns:
|
|
136
|
-
Response[
|
|
136
|
+
Response[RegressionModelMetadata]
|
|
137
137
|
"""
|
|
138
138
|
|
|
139
139
|
kwargs = _get_kwargs(
|
|
@@ -148,19 +148,19 @@ async def asyncio_detailed(
|
|
|
148
148
|
|
|
149
149
|
async def asyncio(
|
|
150
150
|
*,
|
|
151
|
-
body:
|
|
152
|
-
) ->
|
|
153
|
-
"""Create Model
|
|
151
|
+
body: CreateRegressionModelRequest,
|
|
152
|
+
) -> RegressionModelMetadata:
|
|
153
|
+
"""Create Regression Model
|
|
154
154
|
|
|
155
155
|
Args:
|
|
156
|
-
body (
|
|
156
|
+
body (CreateRegressionModelRequest):
|
|
157
157
|
|
|
158
158
|
Raises:
|
|
159
159
|
errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
|
|
160
160
|
httpx.TimeoutException: If the request takes longer than Client.timeout.
|
|
161
161
|
|
|
162
162
|
Returns:
|
|
163
|
-
|
|
163
|
+
RegressionModelMetadata
|
|
164
164
|
"""
|
|
165
165
|
|
|
166
166
|
return (
|
|
@@ -0,0 +1,168 @@
|
|
|
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 ...types import Response
|
|
22
|
+
|
|
23
|
+
|
|
24
|
+
def _get_kwargs(
|
|
25
|
+
model_name_or_id: str,
|
|
26
|
+
task_id: UUID,
|
|
27
|
+
) -> dict[str, Any]:
|
|
28
|
+
_kwargs: dict[str, Any] = {
|
|
29
|
+
"method": "delete",
|
|
30
|
+
"url": f"/regression_model/{model_name_or_id}/evaluation/{task_id}",
|
|
31
|
+
}
|
|
32
|
+
|
|
33
|
+
return _kwargs
|
|
34
|
+
|
|
35
|
+
|
|
36
|
+
def _parse_response(*, response: httpx.Response) -> None:
|
|
37
|
+
if response.status_code == 204:
|
|
38
|
+
response_204 = None
|
|
39
|
+
return response_204
|
|
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[None]:
|
|
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
|
+
model_name_or_id: str,
|
|
63
|
+
task_id: UUID,
|
|
64
|
+
) -> Response[None]:
|
|
65
|
+
"""Delete Regression Model Evaluation
|
|
66
|
+
|
|
67
|
+
Args:
|
|
68
|
+
model_name_or_id (str):
|
|
69
|
+
task_id (UUID):
|
|
70
|
+
|
|
71
|
+
Raises:
|
|
72
|
+
errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
|
|
73
|
+
httpx.TimeoutException: If the request takes longer than Client.timeout.
|
|
74
|
+
|
|
75
|
+
Returns:
|
|
76
|
+
Response[None]
|
|
77
|
+
"""
|
|
78
|
+
|
|
79
|
+
kwargs = _get_kwargs(
|
|
80
|
+
model_name_or_id=model_name_or_id,
|
|
81
|
+
task_id=task_id,
|
|
82
|
+
)
|
|
83
|
+
|
|
84
|
+
with _client_context() as client:
|
|
85
|
+
response = client.get_httpx_client().request(
|
|
86
|
+
**kwargs,
|
|
87
|
+
)
|
|
88
|
+
|
|
89
|
+
return _build_response(response=response)
|
|
90
|
+
|
|
91
|
+
|
|
92
|
+
def sync(
|
|
93
|
+
model_name_or_id: str,
|
|
94
|
+
task_id: UUID,
|
|
95
|
+
) -> None:
|
|
96
|
+
"""Delete Regression Model Evaluation
|
|
97
|
+
|
|
98
|
+
Args:
|
|
99
|
+
model_name_or_id (str):
|
|
100
|
+
task_id (UUID):
|
|
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
|
+
None
|
|
108
|
+
"""
|
|
109
|
+
|
|
110
|
+
return sync_detailed(
|
|
111
|
+
model_name_or_id=model_name_or_id,
|
|
112
|
+
task_id=task_id,
|
|
113
|
+
).parsed
|
|
114
|
+
|
|
115
|
+
|
|
116
|
+
async def asyncio_detailed(
|
|
117
|
+
model_name_or_id: str,
|
|
118
|
+
task_id: UUID,
|
|
119
|
+
) -> Response[None]:
|
|
120
|
+
"""Delete Regression Model Evaluation
|
|
121
|
+
|
|
122
|
+
Args:
|
|
123
|
+
model_name_or_id (str):
|
|
124
|
+
task_id (UUID):
|
|
125
|
+
|
|
126
|
+
Raises:
|
|
127
|
+
errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
|
|
128
|
+
httpx.TimeoutException: If the request takes longer than Client.timeout.
|
|
129
|
+
|
|
130
|
+
Returns:
|
|
131
|
+
Response[None]
|
|
132
|
+
"""
|
|
133
|
+
|
|
134
|
+
kwargs = _get_kwargs(
|
|
135
|
+
model_name_or_id=model_name_or_id,
|
|
136
|
+
task_id=task_id,
|
|
137
|
+
)
|
|
138
|
+
|
|
139
|
+
with _client_context() as client:
|
|
140
|
+
response = await client.get_async_httpx_client().request(**kwargs)
|
|
141
|
+
|
|
142
|
+
return _build_response(response=response)
|
|
143
|
+
|
|
144
|
+
|
|
145
|
+
async def asyncio(
|
|
146
|
+
model_name_or_id: str,
|
|
147
|
+
task_id: UUID,
|
|
148
|
+
) -> None:
|
|
149
|
+
"""Delete Regression Model Evaluation
|
|
150
|
+
|
|
151
|
+
Args:
|
|
152
|
+
model_name_or_id (str):
|
|
153
|
+
task_id (UUID):
|
|
154
|
+
|
|
155
|
+
Raises:
|
|
156
|
+
errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
|
|
157
|
+
httpx.TimeoutException: If the request takes longer than Client.timeout.
|
|
158
|
+
|
|
159
|
+
Returns:
|
|
160
|
+
None
|
|
161
|
+
"""
|
|
162
|
+
|
|
163
|
+
return (
|
|
164
|
+
await asyncio_detailed(
|
|
165
|
+
model_name_or_id=model_name_or_id,
|
|
166
|
+
task_id=task_id,
|
|
167
|
+
)
|
|
168
|
+
).parsed
|
|
@@ -25,7 +25,7 @@ def _get_kwargs(
|
|
|
25
25
|
) -> dict[str, Any]:
|
|
26
26
|
_kwargs: dict[str, Any] = {
|
|
27
27
|
"method": "delete",
|
|
28
|
-
"url": f"/
|
|
28
|
+
"url": f"/regression_model/{name_or_id}",
|
|
29
29
|
}
|
|
30
30
|
|
|
31
31
|
return _kwargs
|
|
@@ -59,7 +59,7 @@ def _build_response(*, response: httpx.Response) -> Response[None]:
|
|
|
59
59
|
def sync_detailed(
|
|
60
60
|
name_or_id: str,
|
|
61
61
|
) -> Response[None]:
|
|
62
|
-
"""Delete Model
|
|
62
|
+
"""Delete Regression Model
|
|
63
63
|
|
|
64
64
|
Args:
|
|
65
65
|
name_or_id (str):
|
|
@@ -87,7 +87,7 @@ def sync_detailed(
|
|
|
87
87
|
def sync(
|
|
88
88
|
name_or_id: str,
|
|
89
89
|
) -> None:
|
|
90
|
-
"""Delete Model
|
|
90
|
+
"""Delete Regression Model
|
|
91
91
|
|
|
92
92
|
Args:
|
|
93
93
|
name_or_id (str):
|
|
@@ -108,7 +108,7 @@ def sync(
|
|
|
108
108
|
async def asyncio_detailed(
|
|
109
109
|
name_or_id: str,
|
|
110
110
|
) -> Response[None]:
|
|
111
|
-
"""Delete Model
|
|
111
|
+
"""Delete Regression Model
|
|
112
112
|
|
|
113
113
|
Args:
|
|
114
114
|
name_or_id (str):
|
|
@@ -134,7 +134,7 @@ async def asyncio_detailed(
|
|
|
134
134
|
async def asyncio(
|
|
135
135
|
name_or_id: str,
|
|
136
136
|
) -> None:
|
|
137
|
-
"""Delete Model
|
|
137
|
+
"""Delete Regression Model
|
|
138
138
|
|
|
139
139
|
Args:
|
|
140
140
|
name_or_id (str):
|