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,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.evaluation_response import EvaluationResponse
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) -> EvaluationResponse:
37
+ def _parse_response(*, response: httpx.Response) -> EvaluationResponseClassificationMetrics:
38
38
  if response.status_code == 200:
39
- response_200 = EvaluationResponse.from_dict(response.json())
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[EvaluationResponse]:
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[EvaluationResponse]:
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[EvaluationResponse]
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
- ) -> EvaluationResponse:
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
- EvaluationResponse
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[EvaluationResponse]:
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[EvaluationResponse]
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
- ) -> EvaluationResponse:
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
- EvaluationResponse
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.evaluation_response import EvaluationResponse
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["EvaluationResponse"]:
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 = EvaluationResponse.from_dict(response_200_item_data)
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["EvaluationResponse"]]:
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["EvaluationResponse"]]:
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['EvaluationResponse']]
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["EvaluationResponse"]:
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['EvaluationResponse']
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["EvaluationResponse"]]:
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['EvaluationResponse']]
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["EvaluationResponse"]:
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['EvaluationResponse']
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.prediction_request import PredictionRequest
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: PredictionRequest,
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: PredictionRequest,
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 (PredictionRequest):
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: PredictionRequest,
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 (PredictionRequest):
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: PredictionRequest,
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 (PredictionRequest):
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: PredictionRequest,
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 (PredictionRequest):
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.