orca-sdk 0.0.93__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 (125) hide show
  1. orca_sdk/__init__.py +13 -4
  2. orca_sdk/_generated_api_client/api/__init__.py +84 -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 +172 -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/task/list_tasks_task_get.py +60 -10
  42. orca_sdk/_generated_api_client/api/telemetry/count_predictions_telemetry_prediction_count_post.py +10 -10
  43. orca_sdk/_generated_api_client/api/telemetry/get_prediction_telemetry_prediction_prediction_id_get.py +35 -12
  44. orca_sdk/_generated_api_client/api/telemetry/list_memories_with_feedback_telemetry_memories_post.py +20 -12
  45. orca_sdk/_generated_api_client/api/telemetry/list_predictions_telemetry_prediction_post.py +35 -12
  46. orca_sdk/_generated_api_client/models/__init__.py +90 -24
  47. orca_sdk/_generated_api_client/models/base_score_prediction_result.py +108 -0
  48. orca_sdk/_generated_api_client/models/{evaluation_request.py → classification_evaluation_request.py} +13 -45
  49. orca_sdk/_generated_api_client/models/{classification_evaluation_result.py → classification_metrics.py} +106 -56
  50. orca_sdk/_generated_api_client/models/{rac_model_metadata.py → classification_model_metadata.py} +51 -43
  51. orca_sdk/_generated_api_client/models/{prediction_request.py → classification_prediction_request.py} +31 -6
  52. orca_sdk/_generated_api_client/models/{clone_labeled_memoryset_request.py → clone_memoryset_request.py} +5 -5
  53. orca_sdk/_generated_api_client/models/column_info.py +31 -0
  54. orca_sdk/_generated_api_client/models/count_predictions_request.py +195 -0
  55. orca_sdk/_generated_api_client/models/{create_rac_model_request.py → create_classification_model_request.py} +25 -57
  56. orca_sdk/_generated_api_client/models/{create_labeled_memoryset_request.py → create_memoryset_request.py} +73 -56
  57. orca_sdk/_generated_api_client/models/create_memoryset_request_index_params.py +66 -0
  58. orca_sdk/_generated_api_client/models/create_memoryset_request_index_type.py +13 -0
  59. orca_sdk/_generated_api_client/models/create_regression_model_request.py +137 -0
  60. orca_sdk/_generated_api_client/models/embedding_evaluation_payload.py +187 -0
  61. orca_sdk/_generated_api_client/models/embedding_evaluation_response.py +10 -0
  62. orca_sdk/_generated_api_client/models/evaluation_response.py +22 -9
  63. orca_sdk/_generated_api_client/models/evaluation_response_classification_metrics.py +140 -0
  64. orca_sdk/_generated_api_client/models/evaluation_response_regression_metrics.py +140 -0
  65. orca_sdk/_generated_api_client/models/http_validation_error.py +86 -0
  66. orca_sdk/_generated_api_client/models/list_predictions_request.py +62 -0
  67. orca_sdk/_generated_api_client/models/memory_type.py +9 -0
  68. orca_sdk/_generated_api_client/models/memoryset_analysis_configs.py +0 -20
  69. orca_sdk/_generated_api_client/models/{labeled_memoryset_metadata.py → memoryset_metadata.py} +73 -13
  70. orca_sdk/_generated_api_client/models/memoryset_metadata_index_params.py +55 -0
  71. orca_sdk/_generated_api_client/models/memoryset_metadata_index_type.py +13 -0
  72. orca_sdk/_generated_api_client/models/{labeled_memoryset_update.py → memoryset_update.py} +19 -31
  73. orca_sdk/_generated_api_client/models/not_found_error_response_resource_type_0.py +1 -0
  74. 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
  75. orca_sdk/_generated_api_client/models/{precision_recall_curve.py → pr_curve.py} +5 -13
  76. orca_sdk/_generated_api_client/models/{rac_model_update.py → predictive_model_update.py} +14 -5
  77. orca_sdk/_generated_api_client/models/pretrained_embedding_model_metadata.py +11 -1
  78. orca_sdk/_generated_api_client/models/pretrained_embedding_model_name.py +5 -0
  79. orca_sdk/_generated_api_client/models/rar_head_type.py +8 -0
  80. orca_sdk/_generated_api_client/models/regression_evaluation_request.py +148 -0
  81. orca_sdk/_generated_api_client/models/regression_metrics.py +172 -0
  82. orca_sdk/_generated_api_client/models/regression_model_metadata.py +177 -0
  83. orca_sdk/_generated_api_client/models/regression_prediction_request.py +195 -0
  84. orca_sdk/_generated_api_client/models/roc_curve.py +0 -8
  85. orca_sdk/_generated_api_client/models/score_prediction_memory_lookup.py +196 -0
  86. orca_sdk/_generated_api_client/models/score_prediction_memory_lookup_metadata.py +68 -0
  87. orca_sdk/_generated_api_client/models/score_prediction_with_memories_and_feedback.py +252 -0
  88. orca_sdk/_generated_api_client/models/scored_memory.py +172 -0
  89. orca_sdk/_generated_api_client/models/scored_memory_insert.py +128 -0
  90. orca_sdk/_generated_api_client/models/scored_memory_insert_metadata.py +68 -0
  91. orca_sdk/_generated_api_client/models/scored_memory_lookup.py +180 -0
  92. orca_sdk/_generated_api_client/models/scored_memory_lookup_metadata.py +68 -0
  93. orca_sdk/_generated_api_client/models/scored_memory_metadata.py +68 -0
  94. orca_sdk/_generated_api_client/models/scored_memory_update.py +171 -0
  95. orca_sdk/_generated_api_client/models/scored_memory_update_metadata_type_0.py +68 -0
  96. orca_sdk/_generated_api_client/models/scored_memory_with_feedback_metrics.py +193 -0
  97. orca_sdk/_generated_api_client/models/scored_memory_with_feedback_metrics_feedback_metrics.py +68 -0
  98. orca_sdk/_generated_api_client/models/scored_memory_with_feedback_metrics_metadata.py +68 -0
  99. orca_sdk/_generated_api_client/models/update_prediction_request.py +20 -0
  100. orca_sdk/_generated_api_client/models/validation_error.py +99 -0
  101. orca_sdk/_shared/__init__.py +9 -1
  102. orca_sdk/_shared/metrics.py +257 -87
  103. orca_sdk/_shared/metrics_test.py +136 -77
  104. orca_sdk/_utils/data_parsing.py +0 -3
  105. orca_sdk/_utils/data_parsing_test.py +0 -3
  106. orca_sdk/_utils/prediction_result_ui.py +55 -23
  107. orca_sdk/classification_model.py +184 -174
  108. orca_sdk/classification_model_test.py +178 -142
  109. orca_sdk/conftest.py +77 -26
  110. orca_sdk/datasource.py +34 -0
  111. orca_sdk/datasource_test.py +9 -1
  112. orca_sdk/embedding_model.py +136 -14
  113. orca_sdk/embedding_model_test.py +10 -6
  114. orca_sdk/job.py +329 -0
  115. orca_sdk/job_test.py +48 -0
  116. orca_sdk/memoryset.py +882 -161
  117. orca_sdk/memoryset_test.py +58 -23
  118. orca_sdk/regression_model.py +647 -0
  119. orca_sdk/regression_model_test.py +338 -0
  120. orca_sdk/telemetry.py +225 -106
  121. orca_sdk/telemetry_test.py +34 -30
  122. {orca_sdk-0.0.93.dist-info → orca_sdk-0.0.95.dist-info}/METADATA +2 -4
  123. {orca_sdk-0.0.93.dist-info → orca_sdk-0.0.95.dist-info}/RECORD +124 -74
  124. orca_sdk/_utils/task.py +0 -73
  125. {orca_sdk-0.0.93.dist-info → orca_sdk-0.0.95.dist-info}/WHEEL +0 -0
@@ -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"/classification_model/{name_or_id}",
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):
@@ -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