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
|
@@ -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.classification_model_metadata import ClassificationModelMetadata
|
|
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"/classification_model/{name_or_id}",
|
|
30
|
+
}
|
|
31
|
+
|
|
32
|
+
return _kwargs
|
|
33
|
+
|
|
34
|
+
|
|
35
|
+
def _parse_response(*, response: httpx.Response) -> ClassificationModelMetadata:
|
|
36
|
+
if response.status_code == 200:
|
|
37
|
+
response_200 = ClassificationModelMetadata.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[ClassificationModelMetadata]:
|
|
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[ClassificationModelMetadata]:
|
|
64
|
+
"""Get Classification 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[ClassificationModelMetadata]
|
|
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
|
+
) -> ClassificationModelMetadata:
|
|
92
|
+
"""Get Classification 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
|
+
ClassificationModelMetadata
|
|
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[ClassificationModelMetadata]:
|
|
113
|
+
"""Get Classification 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[ClassificationModelMetadata]
|
|
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
|
+
) -> ClassificationModelMetadata:
|
|
139
|
+
"""Get Classification 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
|
+
ClassificationModelMetadata
|
|
150
|
+
"""
|
|
151
|
+
|
|
152
|
+
return (
|
|
153
|
+
await asyncio_detailed(
|
|
154
|
+
name_or_id=name_or_id,
|
|
155
|
+
)
|
|
156
|
+
).parsed
|
|
@@ -18,7 +18,7 @@ import httpx
|
|
|
18
18
|
|
|
19
19
|
from ...client import _client_context
|
|
20
20
|
from ...errors import get_error_for_response
|
|
21
|
-
from ...models.
|
|
21
|
+
from ...models.evaluation_response_classification_metrics import EvaluationResponseClassificationMetrics
|
|
22
22
|
from ...types import Response
|
|
23
23
|
|
|
24
24
|
|
|
@@ -34,9 +34,9 @@ def _get_kwargs(
|
|
|
34
34
|
return _kwargs
|
|
35
35
|
|
|
36
36
|
|
|
37
|
-
def _parse_response(*, response: httpx.Response) ->
|
|
37
|
+
def _parse_response(*, response: httpx.Response) -> EvaluationResponseClassificationMetrics:
|
|
38
38
|
if response.status_code == 200:
|
|
39
|
-
response_200 =
|
|
39
|
+
response_200 = EvaluationResponseClassificationMetrics.from_dict(response.json())
|
|
40
40
|
|
|
41
41
|
return response_200
|
|
42
42
|
if response.status_code == 401:
|
|
@@ -51,7 +51,7 @@ def _parse_response(*, response: httpx.Response) -> EvaluationResponse:
|
|
|
51
51
|
raise get_error_for_response(response)
|
|
52
52
|
|
|
53
53
|
|
|
54
|
-
def _build_response(*, response: httpx.Response) -> Response[
|
|
54
|
+
def _build_response(*, response: httpx.Response) -> Response[EvaluationResponseClassificationMetrics]:
|
|
55
55
|
return Response(
|
|
56
56
|
status_code=HTTPStatus(response.status_code),
|
|
57
57
|
content=response.content,
|
|
@@ -63,8 +63,8 @@ def _build_response(*, response: httpx.Response) -> Response[EvaluationResponse]
|
|
|
63
63
|
def sync_detailed(
|
|
64
64
|
model_name_or_id: str,
|
|
65
65
|
task_id: UUID,
|
|
66
|
-
) -> Response[
|
|
67
|
-
"""Get Evaluation
|
|
66
|
+
) -> Response[EvaluationResponseClassificationMetrics]:
|
|
67
|
+
"""Get Classification Model Evaluation
|
|
68
68
|
|
|
69
69
|
Args:
|
|
70
70
|
model_name_or_id (str):
|
|
@@ -75,7 +75,7 @@ def sync_detailed(
|
|
|
75
75
|
httpx.TimeoutException: If the request takes longer than Client.timeout.
|
|
76
76
|
|
|
77
77
|
Returns:
|
|
78
|
-
Response[
|
|
78
|
+
Response[EvaluationResponseClassificationMetrics]
|
|
79
79
|
"""
|
|
80
80
|
|
|
81
81
|
kwargs = _get_kwargs(
|
|
@@ -94,8 +94,8 @@ def sync_detailed(
|
|
|
94
94
|
def sync(
|
|
95
95
|
model_name_or_id: str,
|
|
96
96
|
task_id: UUID,
|
|
97
|
-
) ->
|
|
98
|
-
"""Get Evaluation
|
|
97
|
+
) -> EvaluationResponseClassificationMetrics:
|
|
98
|
+
"""Get Classification Model Evaluation
|
|
99
99
|
|
|
100
100
|
Args:
|
|
101
101
|
model_name_or_id (str):
|
|
@@ -106,7 +106,7 @@ def sync(
|
|
|
106
106
|
httpx.TimeoutException: If the request takes longer than Client.timeout.
|
|
107
107
|
|
|
108
108
|
Returns:
|
|
109
|
-
|
|
109
|
+
EvaluationResponseClassificationMetrics
|
|
110
110
|
"""
|
|
111
111
|
|
|
112
112
|
return sync_detailed(
|
|
@@ -118,8 +118,8 @@ def sync(
|
|
|
118
118
|
async def asyncio_detailed(
|
|
119
119
|
model_name_or_id: str,
|
|
120
120
|
task_id: UUID,
|
|
121
|
-
) -> Response[
|
|
122
|
-
"""Get Evaluation
|
|
121
|
+
) -> Response[EvaluationResponseClassificationMetrics]:
|
|
122
|
+
"""Get Classification Model Evaluation
|
|
123
123
|
|
|
124
124
|
Args:
|
|
125
125
|
model_name_or_id (str):
|
|
@@ -130,7 +130,7 @@ async def asyncio_detailed(
|
|
|
130
130
|
httpx.TimeoutException: If the request takes longer than Client.timeout.
|
|
131
131
|
|
|
132
132
|
Returns:
|
|
133
|
-
Response[
|
|
133
|
+
Response[EvaluationResponseClassificationMetrics]
|
|
134
134
|
"""
|
|
135
135
|
|
|
136
136
|
kwargs = _get_kwargs(
|
|
@@ -147,8 +147,8 @@ async def asyncio_detailed(
|
|
|
147
147
|
async def asyncio(
|
|
148
148
|
model_name_or_id: str,
|
|
149
149
|
task_id: UUID,
|
|
150
|
-
) ->
|
|
151
|
-
"""Get Evaluation
|
|
150
|
+
) -> EvaluationResponseClassificationMetrics:
|
|
151
|
+
"""Get Classification Model Evaluation
|
|
152
152
|
|
|
153
153
|
Args:
|
|
154
154
|
model_name_or_id (str):
|
|
@@ -159,7 +159,7 @@ async def asyncio(
|
|
|
159
159
|
httpx.TimeoutException: If the request takes longer than Client.timeout.
|
|
160
160
|
|
|
161
161
|
Returns:
|
|
162
|
-
|
|
162
|
+
EvaluationResponseClassificationMetrics
|
|
163
163
|
"""
|
|
164
164
|
|
|
165
165
|
return (
|
|
@@ -17,7 +17,7 @@ import httpx
|
|
|
17
17
|
|
|
18
18
|
from ...client import _client_context
|
|
19
19
|
from ...errors import get_error_for_response
|
|
20
|
-
from ...models.
|
|
20
|
+
from ...models.evaluation_response_classification_metrics import EvaluationResponseClassificationMetrics
|
|
21
21
|
from ...types import Response
|
|
22
22
|
|
|
23
23
|
|
|
@@ -32,12 +32,12 @@ def _get_kwargs(
|
|
|
32
32
|
return _kwargs
|
|
33
33
|
|
|
34
34
|
|
|
35
|
-
def _parse_response(*, response: httpx.Response) -> List["
|
|
35
|
+
def _parse_response(*, response: httpx.Response) -> List["EvaluationResponseClassificationMetrics"]:
|
|
36
36
|
if response.status_code == 200:
|
|
37
37
|
response_200 = []
|
|
38
38
|
_response_200 = response.json()
|
|
39
39
|
for response_200_item_data in _response_200:
|
|
40
|
-
response_200_item =
|
|
40
|
+
response_200_item = EvaluationResponseClassificationMetrics.from_dict(response_200_item_data)
|
|
41
41
|
|
|
42
42
|
response_200.append(response_200_item)
|
|
43
43
|
|
|
@@ -54,7 +54,7 @@ def _parse_response(*, response: httpx.Response) -> List["EvaluationResponse"]:
|
|
|
54
54
|
raise get_error_for_response(response)
|
|
55
55
|
|
|
56
56
|
|
|
57
|
-
def _build_response(*, response: httpx.Response) -> Response[List["
|
|
57
|
+
def _build_response(*, response: httpx.Response) -> Response[List["EvaluationResponseClassificationMetrics"]]:
|
|
58
58
|
return Response(
|
|
59
59
|
status_code=HTTPStatus(response.status_code),
|
|
60
60
|
content=response.content,
|
|
@@ -65,8 +65,8 @@ def _build_response(*, response: httpx.Response) -> Response[List["EvaluationRes
|
|
|
65
65
|
|
|
66
66
|
def sync_detailed(
|
|
67
67
|
model_name_or_id: str,
|
|
68
|
-
) -> Response[List["
|
|
69
|
-
"""List Evaluations
|
|
68
|
+
) -> Response[List["EvaluationResponseClassificationMetrics"]]:
|
|
69
|
+
"""List Classification Model Evaluations
|
|
70
70
|
|
|
71
71
|
Args:
|
|
72
72
|
model_name_or_id (str):
|
|
@@ -76,7 +76,7 @@ def sync_detailed(
|
|
|
76
76
|
httpx.TimeoutException: If the request takes longer than Client.timeout.
|
|
77
77
|
|
|
78
78
|
Returns:
|
|
79
|
-
Response[List['
|
|
79
|
+
Response[List['EvaluationResponseClassificationMetrics']]
|
|
80
80
|
"""
|
|
81
81
|
|
|
82
82
|
kwargs = _get_kwargs(
|
|
@@ -93,8 +93,8 @@ def sync_detailed(
|
|
|
93
93
|
|
|
94
94
|
def sync(
|
|
95
95
|
model_name_or_id: str,
|
|
96
|
-
) -> List["
|
|
97
|
-
"""List Evaluations
|
|
96
|
+
) -> List["EvaluationResponseClassificationMetrics"]:
|
|
97
|
+
"""List Classification Model Evaluations
|
|
98
98
|
|
|
99
99
|
Args:
|
|
100
100
|
model_name_or_id (str):
|
|
@@ -104,7 +104,7 @@ def sync(
|
|
|
104
104
|
httpx.TimeoutException: If the request takes longer than Client.timeout.
|
|
105
105
|
|
|
106
106
|
Returns:
|
|
107
|
-
List['
|
|
107
|
+
List['EvaluationResponseClassificationMetrics']
|
|
108
108
|
"""
|
|
109
109
|
|
|
110
110
|
return sync_detailed(
|
|
@@ -114,8 +114,8 @@ def sync(
|
|
|
114
114
|
|
|
115
115
|
async def asyncio_detailed(
|
|
116
116
|
model_name_or_id: str,
|
|
117
|
-
) -> Response[List["
|
|
118
|
-
"""List Evaluations
|
|
117
|
+
) -> Response[List["EvaluationResponseClassificationMetrics"]]:
|
|
118
|
+
"""List Classification Model Evaluations
|
|
119
119
|
|
|
120
120
|
Args:
|
|
121
121
|
model_name_or_id (str):
|
|
@@ -125,7 +125,7 @@ async def asyncio_detailed(
|
|
|
125
125
|
httpx.TimeoutException: If the request takes longer than Client.timeout.
|
|
126
126
|
|
|
127
127
|
Returns:
|
|
128
|
-
Response[List['
|
|
128
|
+
Response[List['EvaluationResponseClassificationMetrics']]
|
|
129
129
|
"""
|
|
130
130
|
|
|
131
131
|
kwargs = _get_kwargs(
|
|
@@ -140,8 +140,8 @@ async def asyncio_detailed(
|
|
|
140
140
|
|
|
141
141
|
async def asyncio(
|
|
142
142
|
model_name_or_id: str,
|
|
143
|
-
) -> List["
|
|
144
|
-
"""List Evaluations
|
|
143
|
+
) -> List["EvaluationResponseClassificationMetrics"]:
|
|
144
|
+
"""List Classification Model Evaluations
|
|
145
145
|
|
|
146
146
|
Args:
|
|
147
147
|
model_name_or_id (str):
|
|
@@ -151,7 +151,7 @@ async def asyncio(
|
|
|
151
151
|
httpx.TimeoutException: If the request takes longer than Client.timeout.
|
|
152
152
|
|
|
153
153
|
Returns:
|
|
154
|
-
List['
|
|
154
|
+
List['EvaluationResponseClassificationMetrics']
|
|
155
155
|
"""
|
|
156
156
|
|
|
157
157
|
return (
|
|
@@ -0,0 +1,127 @@
|
|
|
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.classification_model_metadata import ClassificationModelMetadata
|
|
21
|
+
from ...types import Response
|
|
22
|
+
|
|
23
|
+
|
|
24
|
+
def _get_kwargs() -> dict[str, Any]:
|
|
25
|
+
_kwargs: dict[str, Any] = {
|
|
26
|
+
"method": "get",
|
|
27
|
+
"url": "/classification_model/",
|
|
28
|
+
}
|
|
29
|
+
|
|
30
|
+
return _kwargs
|
|
31
|
+
|
|
32
|
+
|
|
33
|
+
def _parse_response(*, response: httpx.Response) -> List["ClassificationModelMetadata"]:
|
|
34
|
+
if response.status_code == 200:
|
|
35
|
+
response_200 = []
|
|
36
|
+
_response_200 = response.json()
|
|
37
|
+
for response_200_item_data in _response_200:
|
|
38
|
+
response_200_item = ClassificationModelMetadata.from_dict(response_200_item_data)
|
|
39
|
+
|
|
40
|
+
response_200.append(response_200_item)
|
|
41
|
+
|
|
42
|
+
return response_200
|
|
43
|
+
if response.status_code == 401:
|
|
44
|
+
raise get_error_for_response(response)
|
|
45
|
+
if response.status_code == 500:
|
|
46
|
+
raise get_error_for_response(response)
|
|
47
|
+
if response.status_code == 503:
|
|
48
|
+
raise get_error_for_response(response)
|
|
49
|
+
else:
|
|
50
|
+
raise get_error_for_response(response)
|
|
51
|
+
|
|
52
|
+
|
|
53
|
+
def _build_response(*, response: httpx.Response) -> Response[List["ClassificationModelMetadata"]]:
|
|
54
|
+
return Response(
|
|
55
|
+
status_code=HTTPStatus(response.status_code),
|
|
56
|
+
content=response.content,
|
|
57
|
+
headers=response.headers,
|
|
58
|
+
parsed=_parse_response(response=response),
|
|
59
|
+
)
|
|
60
|
+
|
|
61
|
+
|
|
62
|
+
def sync_detailed() -> Response[List["ClassificationModelMetadata"]]:
|
|
63
|
+
"""List Classification Models
|
|
64
|
+
|
|
65
|
+
Raises:
|
|
66
|
+
errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
|
|
67
|
+
httpx.TimeoutException: If the request takes longer than Client.timeout.
|
|
68
|
+
|
|
69
|
+
Returns:
|
|
70
|
+
Response[List['ClassificationModelMetadata']]
|
|
71
|
+
"""
|
|
72
|
+
|
|
73
|
+
kwargs = _get_kwargs()
|
|
74
|
+
|
|
75
|
+
with _client_context() as client:
|
|
76
|
+
response = client.get_httpx_client().request(
|
|
77
|
+
**kwargs,
|
|
78
|
+
)
|
|
79
|
+
|
|
80
|
+
return _build_response(response=response)
|
|
81
|
+
|
|
82
|
+
|
|
83
|
+
def sync() -> List["ClassificationModelMetadata"]:
|
|
84
|
+
"""List Classification Models
|
|
85
|
+
|
|
86
|
+
Raises:
|
|
87
|
+
errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
|
|
88
|
+
httpx.TimeoutException: If the request takes longer than Client.timeout.
|
|
89
|
+
|
|
90
|
+
Returns:
|
|
91
|
+
List['ClassificationModelMetadata']
|
|
92
|
+
"""
|
|
93
|
+
|
|
94
|
+
return sync_detailed().parsed
|
|
95
|
+
|
|
96
|
+
|
|
97
|
+
async def asyncio_detailed() -> Response[List["ClassificationModelMetadata"]]:
|
|
98
|
+
"""List Classification Models
|
|
99
|
+
|
|
100
|
+
Raises:
|
|
101
|
+
errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
|
|
102
|
+
httpx.TimeoutException: If the request takes longer than Client.timeout.
|
|
103
|
+
|
|
104
|
+
Returns:
|
|
105
|
+
Response[List['ClassificationModelMetadata']]
|
|
106
|
+
"""
|
|
107
|
+
|
|
108
|
+
kwargs = _get_kwargs()
|
|
109
|
+
|
|
110
|
+
with _client_context() as client:
|
|
111
|
+
response = await client.get_async_httpx_client().request(**kwargs)
|
|
112
|
+
|
|
113
|
+
return _build_response(response=response)
|
|
114
|
+
|
|
115
|
+
|
|
116
|
+
async def asyncio() -> List["ClassificationModelMetadata"]:
|
|
117
|
+
"""List Classification Models
|
|
118
|
+
|
|
119
|
+
Raises:
|
|
120
|
+
errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
|
|
121
|
+
httpx.TimeoutException: If the request takes longer than Client.timeout.
|
|
122
|
+
|
|
123
|
+
Returns:
|
|
124
|
+
List['ClassificationModelMetadata']
|
|
125
|
+
"""
|
|
126
|
+
|
|
127
|
+
return (await asyncio_detailed()).parsed
|
|
@@ -18,14 +18,14 @@ import httpx
|
|
|
18
18
|
from ...client import _client_context
|
|
19
19
|
from ...errors import get_error_for_response
|
|
20
20
|
from ...models.base_label_prediction_result import BaseLabelPredictionResult
|
|
21
|
-
from ...models.
|
|
21
|
+
from ...models.classification_prediction_request import ClassificationPredictionRequest
|
|
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: ClassificationPredictionRequest,
|
|
29
29
|
) -> dict[str, Any]:
|
|
30
30
|
headers: Dict[str, Any] = {}
|
|
31
31
|
|
|
@@ -79,13 +79,13 @@ def _build_response(*, response: httpx.Response) -> Response[List["BaseLabelPred
|
|
|
79
79
|
def sync_detailed(
|
|
80
80
|
name_or_id: str,
|
|
81
81
|
*,
|
|
82
|
-
body:
|
|
82
|
+
body: ClassificationPredictionRequest,
|
|
83
83
|
) -> Response[List["BaseLabelPredictionResult"]]:
|
|
84
|
-
"""Predict
|
|
84
|
+
"""Predict Label
|
|
85
85
|
|
|
86
86
|
Args:
|
|
87
87
|
name_or_id (str):
|
|
88
|
-
body (
|
|
88
|
+
body (ClassificationPredictionRequest):
|
|
89
89
|
|
|
90
90
|
Raises:
|
|
91
91
|
errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
|
|
@@ -111,13 +111,13 @@ def sync_detailed(
|
|
|
111
111
|
def sync(
|
|
112
112
|
name_or_id: str,
|
|
113
113
|
*,
|
|
114
|
-
body:
|
|
114
|
+
body: ClassificationPredictionRequest,
|
|
115
115
|
) -> List["BaseLabelPredictionResult"]:
|
|
116
|
-
"""Predict
|
|
116
|
+
"""Predict Label
|
|
117
117
|
|
|
118
118
|
Args:
|
|
119
119
|
name_or_id (str):
|
|
120
|
-
body (
|
|
120
|
+
body (ClassificationPredictionRequest):
|
|
121
121
|
|
|
122
122
|
Raises:
|
|
123
123
|
errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
|
|
@@ -136,13 +136,13 @@ def sync(
|
|
|
136
136
|
async def asyncio_detailed(
|
|
137
137
|
name_or_id: str,
|
|
138
138
|
*,
|
|
139
|
-
body:
|
|
139
|
+
body: ClassificationPredictionRequest,
|
|
140
140
|
) -> Response[List["BaseLabelPredictionResult"]]:
|
|
141
|
-
"""Predict
|
|
141
|
+
"""Predict Label
|
|
142
142
|
|
|
143
143
|
Args:
|
|
144
144
|
name_or_id (str):
|
|
145
|
-
body (
|
|
145
|
+
body (ClassificationPredictionRequest):
|
|
146
146
|
|
|
147
147
|
Raises:
|
|
148
148
|
errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
|
|
@@ -166,13 +166,13 @@ async def asyncio_detailed(
|
|
|
166
166
|
async def asyncio(
|
|
167
167
|
name_or_id: str,
|
|
168
168
|
*,
|
|
169
|
-
body:
|
|
169
|
+
body: ClassificationPredictionRequest,
|
|
170
170
|
) -> List["BaseLabelPredictionResult"]:
|
|
171
|
-
"""Predict
|
|
171
|
+
"""Predict Label
|
|
172
172
|
|
|
173
173
|
Args:
|
|
174
174
|
name_or_id (str):
|
|
175
|
-
body (
|
|
175
|
+
body (ClassificationPredictionRequest):
|
|
176
176
|
|
|
177
177
|
Raises:
|
|
178
178
|
errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
|