orca-sdk 0.1.0__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 (175) hide show
  1. orca_sdk/__init__.py +19 -0
  2. orca_sdk/_generated_api_client/__init__.py +3 -0
  3. orca_sdk/_generated_api_client/api/__init__.py +193 -0
  4. orca_sdk/_generated_api_client/api/auth/__init__.py +0 -0
  5. orca_sdk/_generated_api_client/api/auth/check_authentication_auth_get.py +128 -0
  6. orca_sdk/_generated_api_client/api/auth/create_api_key_auth_api_key_post.py +170 -0
  7. orca_sdk/_generated_api_client/api/auth/delete_api_key_auth_api_key_name_or_id_delete.py +156 -0
  8. orca_sdk/_generated_api_client/api/auth/delete_org_auth_org_delete.py +130 -0
  9. orca_sdk/_generated_api_client/api/auth/list_api_keys_auth_api_key_get.py +127 -0
  10. orca_sdk/_generated_api_client/api/classification_model/__init__.py +0 -0
  11. orca_sdk/_generated_api_client/api/classification_model/create_evaluation_classification_model_model_name_or_id_evaluation_post.py +183 -0
  12. orca_sdk/_generated_api_client/api/classification_model/create_model_classification_model_post.py +170 -0
  13. orca_sdk/_generated_api_client/api/classification_model/delete_evaluation_classification_model_model_name_or_id_evaluation_task_id_delete.py +168 -0
  14. orca_sdk/_generated_api_client/api/classification_model/delete_model_classification_model_name_or_id_delete.py +154 -0
  15. orca_sdk/_generated_api_client/api/classification_model/get_evaluation_classification_model_model_name_or_id_evaluation_task_id_get.py +170 -0
  16. orca_sdk/_generated_api_client/api/classification_model/get_model_classification_model_name_or_id_get.py +156 -0
  17. orca_sdk/_generated_api_client/api/classification_model/list_evaluations_classification_model_model_name_or_id_evaluation_get.py +161 -0
  18. orca_sdk/_generated_api_client/api/classification_model/list_models_classification_model_get.py +127 -0
  19. orca_sdk/_generated_api_client/api/classification_model/predict_gpu_classification_model_name_or_id_prediction_post.py +190 -0
  20. orca_sdk/_generated_api_client/api/datasource/__init__.py +0 -0
  21. orca_sdk/_generated_api_client/api/datasource/create_datasource_datasource_post.py +167 -0
  22. orca_sdk/_generated_api_client/api/datasource/delete_datasource_datasource_name_or_id_delete.py +156 -0
  23. orca_sdk/_generated_api_client/api/datasource/get_datasource_datasource_name_or_id_get.py +156 -0
  24. orca_sdk/_generated_api_client/api/datasource/list_datasources_datasource_get.py +127 -0
  25. orca_sdk/_generated_api_client/api/default/__init__.py +0 -0
  26. orca_sdk/_generated_api_client/api/default/healthcheck_get.py +118 -0
  27. orca_sdk/_generated_api_client/api/default/healthcheck_gpu_get.py +118 -0
  28. orca_sdk/_generated_api_client/api/finetuned_embedding_model/__init__.py +0 -0
  29. orca_sdk/_generated_api_client/api/finetuned_embedding_model/create_finetuned_embedding_model_finetuned_embedding_model_post.py +168 -0
  30. orca_sdk/_generated_api_client/api/finetuned_embedding_model/delete_finetuned_embedding_model_finetuned_embedding_model_name_or_id_delete.py +156 -0
  31. orca_sdk/_generated_api_client/api/finetuned_embedding_model/embed_with_finetuned_model_gpu_finetuned_embedding_model_name_or_id_embedding_post.py +189 -0
  32. orca_sdk/_generated_api_client/api/finetuned_embedding_model/get_finetuned_embedding_model_finetuned_embedding_model_name_or_id_get.py +156 -0
  33. orca_sdk/_generated_api_client/api/finetuned_embedding_model/list_finetuned_embedding_models_finetuned_embedding_model_get.py +127 -0
  34. orca_sdk/_generated_api_client/api/memoryset/__init__.py +0 -0
  35. orca_sdk/_generated_api_client/api/memoryset/clone_memoryset_memoryset_name_or_id_clone_post.py +181 -0
  36. orca_sdk/_generated_api_client/api/memoryset/create_analysis_memoryset_name_or_id_analysis_post.py +183 -0
  37. orca_sdk/_generated_api_client/api/memoryset/create_memoryset_memoryset_post.py +168 -0
  38. orca_sdk/_generated_api_client/api/memoryset/delete_memories_memoryset_name_or_id_memories_delete_post.py +181 -0
  39. orca_sdk/_generated_api_client/api/memoryset/delete_memory_memoryset_name_or_id_memory_memory_id_delete.py +167 -0
  40. orca_sdk/_generated_api_client/api/memoryset/delete_memoryset_memoryset_name_or_id_delete.py +156 -0
  41. orca_sdk/_generated_api_client/api/memoryset/get_analysis_memoryset_name_or_id_analysis_analysis_task_id_get.py +169 -0
  42. orca_sdk/_generated_api_client/api/memoryset/get_memories_memoryset_name_or_id_memories_get_post.py +188 -0
  43. orca_sdk/_generated_api_client/api/memoryset/get_memory_memoryset_name_or_id_memory_memory_id_get.py +169 -0
  44. orca_sdk/_generated_api_client/api/memoryset/get_memoryset_memoryset_name_or_id_get.py +156 -0
  45. orca_sdk/_generated_api_client/api/memoryset/insert_memories_gpu_memoryset_name_or_id_memory_post.py +184 -0
  46. orca_sdk/_generated_api_client/api/memoryset/list_analyses_memoryset_name_or_id_analysis_get.py +260 -0
  47. orca_sdk/_generated_api_client/api/memoryset/list_memorysets_memoryset_get.py +127 -0
  48. orca_sdk/_generated_api_client/api/memoryset/memoryset_lookup_gpu_memoryset_name_or_id_lookup_post.py +193 -0
  49. orca_sdk/_generated_api_client/api/memoryset/query_memoryset_memoryset_name_or_id_memories_post.py +188 -0
  50. orca_sdk/_generated_api_client/api/memoryset/update_memories_gpu_memoryset_name_or_id_memories_patch.py +191 -0
  51. orca_sdk/_generated_api_client/api/memoryset/update_memory_gpu_memoryset_name_or_id_memory_patch.py +187 -0
  52. orca_sdk/_generated_api_client/api/pretrained_embedding_model/__init__.py +0 -0
  53. orca_sdk/_generated_api_client/api/pretrained_embedding_model/embed_with_pretrained_model_gpu_pretrained_embedding_model_model_name_embedding_post.py +188 -0
  54. orca_sdk/_generated_api_client/api/pretrained_embedding_model/get_pretrained_embedding_model_pretrained_embedding_model_model_name_get.py +157 -0
  55. orca_sdk/_generated_api_client/api/pretrained_embedding_model/list_pretrained_embedding_models_pretrained_embedding_model_get.py +127 -0
  56. orca_sdk/_generated_api_client/api/task/__init__.py +0 -0
  57. orca_sdk/_generated_api_client/api/task/abort_task_task_task_id_abort_delete.py +154 -0
  58. orca_sdk/_generated_api_client/api/task/get_task_status_task_task_id_status_get.py +156 -0
  59. orca_sdk/_generated_api_client/api/task/list_tasks_task_get.py +243 -0
  60. orca_sdk/_generated_api_client/api/telemetry/__init__.py +0 -0
  61. orca_sdk/_generated_api_client/api/telemetry/drop_feedback_category_with_data_telemetry_feedback_category_name_or_id_delete.py +162 -0
  62. orca_sdk/_generated_api_client/api/telemetry/get_feedback_category_telemetry_feedback_category_name_or_id_get.py +156 -0
  63. orca_sdk/_generated_api_client/api/telemetry/get_prediction_telemetry_prediction_prediction_id_get.py +157 -0
  64. orca_sdk/_generated_api_client/api/telemetry/list_feedback_categories_telemetry_feedback_category_get.py +127 -0
  65. orca_sdk/_generated_api_client/api/telemetry/list_predictions_telemetry_prediction_post.py +175 -0
  66. orca_sdk/_generated_api_client/api/telemetry/record_prediction_feedback_telemetry_prediction_feedback_put.py +171 -0
  67. orca_sdk/_generated_api_client/api/telemetry/update_prediction_telemetry_prediction_prediction_id_patch.py +181 -0
  68. orca_sdk/_generated_api_client/client.py +216 -0
  69. orca_sdk/_generated_api_client/errors.py +38 -0
  70. orca_sdk/_generated_api_client/models/__init__.py +159 -0
  71. orca_sdk/_generated_api_client/models/analyze_neighbor_labels_result.py +84 -0
  72. orca_sdk/_generated_api_client/models/api_key_metadata.py +118 -0
  73. orca_sdk/_generated_api_client/models/base_model.py +55 -0
  74. orca_sdk/_generated_api_client/models/body_create_datasource_datasource_post.py +176 -0
  75. orca_sdk/_generated_api_client/models/classification_evaluation_result.py +114 -0
  76. orca_sdk/_generated_api_client/models/clone_labeled_memoryset_request.py +150 -0
  77. orca_sdk/_generated_api_client/models/column_info.py +114 -0
  78. orca_sdk/_generated_api_client/models/column_type.py +14 -0
  79. orca_sdk/_generated_api_client/models/conflict_error_response.py +80 -0
  80. orca_sdk/_generated_api_client/models/create_api_key_request.py +99 -0
  81. orca_sdk/_generated_api_client/models/create_api_key_response.py +126 -0
  82. orca_sdk/_generated_api_client/models/create_labeled_memoryset_request.py +259 -0
  83. orca_sdk/_generated_api_client/models/create_rac_model_request.py +209 -0
  84. orca_sdk/_generated_api_client/models/datasource_metadata.py +142 -0
  85. orca_sdk/_generated_api_client/models/delete_memories_request.py +70 -0
  86. orca_sdk/_generated_api_client/models/embed_request.py +127 -0
  87. orca_sdk/_generated_api_client/models/embedding_finetuning_method.py +9 -0
  88. orca_sdk/_generated_api_client/models/evaluation_request.py +180 -0
  89. orca_sdk/_generated_api_client/models/evaluation_response.py +140 -0
  90. orca_sdk/_generated_api_client/models/feedback_type.py +9 -0
  91. orca_sdk/_generated_api_client/models/field_validation_error.py +103 -0
  92. orca_sdk/_generated_api_client/models/filter_item.py +231 -0
  93. orca_sdk/_generated_api_client/models/filter_item_field_type_0_item.py +15 -0
  94. orca_sdk/_generated_api_client/models/filter_item_field_type_2_item_type_1.py +16 -0
  95. orca_sdk/_generated_api_client/models/filter_item_op.py +16 -0
  96. orca_sdk/_generated_api_client/models/find_duplicates_analysis_result.py +70 -0
  97. orca_sdk/_generated_api_client/models/finetune_embedding_model_request.py +259 -0
  98. orca_sdk/_generated_api_client/models/finetune_embedding_model_request_training_args.py +66 -0
  99. orca_sdk/_generated_api_client/models/finetuned_embedding_model_metadata.py +166 -0
  100. orca_sdk/_generated_api_client/models/get_memories_request.py +70 -0
  101. orca_sdk/_generated_api_client/models/internal_server_error_response.py +80 -0
  102. orca_sdk/_generated_api_client/models/label_class_metrics.py +108 -0
  103. orca_sdk/_generated_api_client/models/label_prediction_memory_lookup.py +274 -0
  104. orca_sdk/_generated_api_client/models/label_prediction_memory_lookup_metadata.py +68 -0
  105. orca_sdk/_generated_api_client/models/label_prediction_result.py +101 -0
  106. orca_sdk/_generated_api_client/models/label_prediction_with_memories_and_feedback.py +232 -0
  107. orca_sdk/_generated_api_client/models/labeled_memory.py +197 -0
  108. orca_sdk/_generated_api_client/models/labeled_memory_insert.py +108 -0
  109. orca_sdk/_generated_api_client/models/labeled_memory_insert_metadata.py +68 -0
  110. orca_sdk/_generated_api_client/models/labeled_memory_lookup.py +258 -0
  111. orca_sdk/_generated_api_client/models/labeled_memory_lookup_metadata.py +68 -0
  112. orca_sdk/_generated_api_client/models/labeled_memory_metadata.py +68 -0
  113. orca_sdk/_generated_api_client/models/labeled_memory_metrics.py +277 -0
  114. orca_sdk/_generated_api_client/models/labeled_memory_update.py +171 -0
  115. orca_sdk/_generated_api_client/models/labeled_memory_update_metadata_type_0.py +68 -0
  116. orca_sdk/_generated_api_client/models/labeled_memoryset_metadata.py +195 -0
  117. orca_sdk/_generated_api_client/models/list_analyses_memoryset_name_or_id_analysis_get_type_type_0.py +9 -0
  118. orca_sdk/_generated_api_client/models/list_memories_request.py +104 -0
  119. orca_sdk/_generated_api_client/models/list_predictions_request.py +234 -0
  120. orca_sdk/_generated_api_client/models/list_predictions_request_sort_item_item_type_0.py +9 -0
  121. orca_sdk/_generated_api_client/models/list_predictions_request_sort_item_item_type_1.py +9 -0
  122. orca_sdk/_generated_api_client/models/lookup_request.py +81 -0
  123. orca_sdk/_generated_api_client/models/memoryset_analysis_request.py +83 -0
  124. orca_sdk/_generated_api_client/models/memoryset_analysis_request_type.py +9 -0
  125. orca_sdk/_generated_api_client/models/memoryset_analysis_response.py +180 -0
  126. orca_sdk/_generated_api_client/models/memoryset_analysis_response_config.py +66 -0
  127. orca_sdk/_generated_api_client/models/memoryset_analysis_response_type.py +9 -0
  128. orca_sdk/_generated_api_client/models/not_found_error_response.py +100 -0
  129. orca_sdk/_generated_api_client/models/not_found_error_response_resource_type_0.py +20 -0
  130. orca_sdk/_generated_api_client/models/prediction_feedback.py +157 -0
  131. orca_sdk/_generated_api_client/models/prediction_feedback_category.py +115 -0
  132. orca_sdk/_generated_api_client/models/prediction_feedback_request.py +122 -0
  133. orca_sdk/_generated_api_client/models/prediction_feedback_result.py +102 -0
  134. orca_sdk/_generated_api_client/models/prediction_request.py +169 -0
  135. orca_sdk/_generated_api_client/models/pretrained_embedding_model_metadata.py +97 -0
  136. orca_sdk/_generated_api_client/models/pretrained_embedding_model_name.py +11 -0
  137. orca_sdk/_generated_api_client/models/rac_head_type.py +11 -0
  138. orca_sdk/_generated_api_client/models/rac_model_metadata.py +191 -0
  139. orca_sdk/_generated_api_client/models/service_unavailable_error_response.py +80 -0
  140. orca_sdk/_generated_api_client/models/task.py +198 -0
  141. orca_sdk/_generated_api_client/models/task_status.py +14 -0
  142. orca_sdk/_generated_api_client/models/task_status_info.py +133 -0
  143. orca_sdk/_generated_api_client/models/unauthenticated_error_response.py +72 -0
  144. orca_sdk/_generated_api_client/models/unauthorized_error_response.py +80 -0
  145. orca_sdk/_generated_api_client/models/unprocessable_input_error_response.py +94 -0
  146. orca_sdk/_generated_api_client/models/update_prediction_request.py +93 -0
  147. orca_sdk/_generated_api_client/py.typed +1 -0
  148. orca_sdk/_generated_api_client/types.py +56 -0
  149. orca_sdk/_utils/__init__.py +0 -0
  150. orca_sdk/_utils/analysis_ui.py +194 -0
  151. orca_sdk/_utils/analysis_ui_style.css +54 -0
  152. orca_sdk/_utils/auth.py +63 -0
  153. orca_sdk/_utils/auth_test.py +31 -0
  154. orca_sdk/_utils/common.py +37 -0
  155. orca_sdk/_utils/data_parsing.py +99 -0
  156. orca_sdk/_utils/data_parsing_test.py +244 -0
  157. orca_sdk/_utils/prediction_result_ui.css +18 -0
  158. orca_sdk/_utils/prediction_result_ui.py +64 -0
  159. orca_sdk/_utils/task.py +73 -0
  160. orca_sdk/classification_model.py +499 -0
  161. orca_sdk/classification_model_test.py +266 -0
  162. orca_sdk/conftest.py +117 -0
  163. orca_sdk/datasource.py +333 -0
  164. orca_sdk/datasource_test.py +95 -0
  165. orca_sdk/embedding_model.py +336 -0
  166. orca_sdk/embedding_model_test.py +173 -0
  167. orca_sdk/labeled_memoryset.py +1154 -0
  168. orca_sdk/labeled_memoryset_test.py +271 -0
  169. orca_sdk/orca_credentials.py +75 -0
  170. orca_sdk/orca_credentials_test.py +37 -0
  171. orca_sdk/telemetry.py +386 -0
  172. orca_sdk/telemetry_test.py +100 -0
  173. orca_sdk-0.1.0.dist-info/METADATA +39 -0
  174. orca_sdk-0.1.0.dist-info/RECORD +175 -0
  175. orca_sdk-0.1.0.dist-info/WHEEL +4 -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"/classification_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 == 422:
47
+ raise get_error_for_response(response)
48
+ if response.status_code == 404:
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 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 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 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 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
@@ -0,0 +1,154 @@
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 ...types import Response
21
+
22
+
23
+ def _get_kwargs(
24
+ name_or_id: str,
25
+ ) -> dict[str, Any]:
26
+ _kwargs: dict[str, Any] = {
27
+ "method": "delete",
28
+ "url": f"/classification_model/{name_or_id}",
29
+ }
30
+
31
+ return _kwargs
32
+
33
+
34
+ def _parse_response(*, response: httpx.Response) -> None:
35
+ if response.status_code == 204:
36
+ response_204 = None
37
+ return response_204
38
+ if response.status_code == 401:
39
+ raise get_error_for_response(response)
40
+ if response.status_code == 500:
41
+ raise get_error_for_response(response)
42
+ if response.status_code == 503:
43
+ raise get_error_for_response(response)
44
+ if response.status_code == 422:
45
+ raise get_error_for_response(response)
46
+ if response.status_code == 404:
47
+ raise get_error_for_response(response)
48
+
49
+
50
+ def _build_response(*, response: httpx.Response) -> Response[None]:
51
+ return Response(
52
+ status_code=HTTPStatus(response.status_code),
53
+ content=response.content,
54
+ headers=response.headers,
55
+ parsed=_parse_response(response=response),
56
+ )
57
+
58
+
59
+ def sync_detailed(
60
+ name_or_id: str,
61
+ ) -> Response[None]:
62
+ """Delete Model
63
+
64
+ Args:
65
+ name_or_id (str):
66
+
67
+ Raises:
68
+ errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
69
+ httpx.TimeoutException: If the request takes longer than Client.timeout.
70
+
71
+ Returns:
72
+ Response[None]
73
+ """
74
+
75
+ kwargs = _get_kwargs(
76
+ name_or_id=name_or_id,
77
+ )
78
+
79
+ with _client_context() as client:
80
+ response = client.get_httpx_client().request(
81
+ **kwargs,
82
+ )
83
+
84
+ return _build_response(response=response)
85
+
86
+
87
+ def sync(
88
+ name_or_id: str,
89
+ ) -> None:
90
+ """Delete Model
91
+
92
+ Args:
93
+ name_or_id (str):
94
+
95
+ Raises:
96
+ errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
97
+ httpx.TimeoutException: If the request takes longer than Client.timeout.
98
+
99
+ Returns:
100
+ None
101
+ """
102
+
103
+ return sync_detailed(
104
+ name_or_id=name_or_id,
105
+ ).parsed
106
+
107
+
108
+ async def asyncio_detailed(
109
+ name_or_id: str,
110
+ ) -> Response[None]:
111
+ """Delete Model
112
+
113
+ Args:
114
+ name_or_id (str):
115
+
116
+ Raises:
117
+ errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
118
+ httpx.TimeoutException: If the request takes longer than Client.timeout.
119
+
120
+ Returns:
121
+ Response[None]
122
+ """
123
+
124
+ kwargs = _get_kwargs(
125
+ name_or_id=name_or_id,
126
+ )
127
+
128
+ with _client_context() as client:
129
+ response = await client.get_async_httpx_client().request(**kwargs)
130
+
131
+ return _build_response(response=response)
132
+
133
+
134
+ async def asyncio(
135
+ name_or_id: str,
136
+ ) -> None:
137
+ """Delete Model
138
+
139
+ Args:
140
+ name_or_id (str):
141
+
142
+ Raises:
143
+ errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
144
+ httpx.TimeoutException: If the request takes longer than Client.timeout.
145
+
146
+ Returns:
147
+ None
148
+ """
149
+
150
+ return (
151
+ await asyncio_detailed(
152
+ name_or_id=name_or_id,
153
+ )
154
+ ).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 import EvaluationResponse
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"/classification_model/{model_name_or_id}/evaluation/{task_id}",
32
+ }
33
+
34
+ return _kwargs
35
+
36
+
37
+ def _parse_response(*, response: httpx.Response) -> EvaluationResponse:
38
+ if response.status_code == 200:
39
+ response_200 = EvaluationResponse.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 == 422:
49
+ raise get_error_for_response(response)
50
+ if response.status_code == 404:
51
+ raise get_error_for_response(response)
52
+
53
+
54
+ def _build_response(*, response: httpx.Response) -> Response[EvaluationResponse]:
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[EvaluationResponse]:
67
+ """Get 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[EvaluationResponse]
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
+ ) -> EvaluationResponse:
98
+ """Get 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
+ EvaluationResponse
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[EvaluationResponse]:
122
+ """Get 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[EvaluationResponse]
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
+ ) -> EvaluationResponse:
151
+ """Get 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
+ EvaluationResponse
163
+ """
164
+
165
+ return (
166
+ await asyncio_detailed(
167
+ model_name_or_id=model_name_or_id,
168
+ task_id=task_id,
169
+ )
170
+ ).parsed
@@ -0,0 +1,156 @@
1
+ """
2
+ This file is generated by the openapi-python-client tool via the generate_api_client.py script
3
+
4
+ It is a customized template from the openapi-python-client tool's default template:
5
+ https://github.com/openapi-generators/openapi-python-client/blob/861ef5622f10fc96d240dc9becb0edf94e61446c/openapi_python_client/templates/endpoint_module.py.jinja
6
+
7
+ The main changes are:
8
+ - Update the API call responses to either return the successful response type or raise an error by:
9
+ - Updating the _parse_response function to raise an error if the response status code is not in the 2xx range
10
+ - Inject a client into every method via a context manager
11
+ """
12
+
13
+ from http import HTTPStatus
14
+ from typing import Any
15
+
16
+ import httpx
17
+
18
+ from ...client import _client_context
19
+ from ...errors import get_error_for_response
20
+ from ...models.rac_model_metadata import RACModelMetadata
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) -> RACModelMetadata:
36
+ if response.status_code == 200:
37
+ response_200 = RACModelMetadata.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 == 422:
47
+ raise get_error_for_response(response)
48
+ if response.status_code == 404:
49
+ raise get_error_for_response(response)
50
+
51
+
52
+ def _build_response(*, response: httpx.Response) -> Response[RACModelMetadata]:
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[RACModelMetadata]:
64
+ """Get 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[RACModelMetadata]
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
+ ) -> RACModelMetadata:
92
+ """Get 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
+ RACModelMetadata
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[RACModelMetadata]:
113
+ """Get 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[RACModelMetadata]
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
+ ) -> RACModelMetadata:
139
+ """Get 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
+ RACModelMetadata
150
+ """
151
+
152
+ return (
153
+ await asyncio_detailed(
154
+ name_or_id=name_or_id,
155
+ )
156
+ ).parsed