orca-sdk 0.0.94__py3-none-any.whl → 0.0.95__py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (116) hide show
  1. orca_sdk/__init__.py +13 -4
  2. orca_sdk/_generated_api_client/api/__init__.py +80 -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 +24 -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/telemetry/get_prediction_telemetry_prediction_prediction_id_get.py +35 -12
  42. orca_sdk/_generated_api_client/api/telemetry/list_memories_with_feedback_telemetry_memories_post.py +20 -12
  43. orca_sdk/_generated_api_client/api/telemetry/list_predictions_telemetry_prediction_post.py +35 -12
  44. orca_sdk/_generated_api_client/models/__init__.py +84 -24
  45. orca_sdk/_generated_api_client/models/base_score_prediction_result.py +108 -0
  46. orca_sdk/_generated_api_client/models/{evaluation_request.py → classification_evaluation_request.py} +13 -45
  47. orca_sdk/_generated_api_client/models/{classification_evaluation_result.py → classification_metrics.py} +106 -56
  48. orca_sdk/_generated_api_client/models/{rac_model_metadata.py → classification_model_metadata.py} +51 -43
  49. orca_sdk/_generated_api_client/models/{prediction_request.py → classification_prediction_request.py} +31 -6
  50. orca_sdk/_generated_api_client/models/{clone_labeled_memoryset_request.py → clone_memoryset_request.py} +5 -5
  51. orca_sdk/_generated_api_client/models/column_info.py +31 -0
  52. orca_sdk/_generated_api_client/models/{create_rac_model_request.py → create_classification_model_request.py} +25 -57
  53. orca_sdk/_generated_api_client/models/{create_labeled_memoryset_request.py → create_memoryset_request.py} +73 -56
  54. orca_sdk/_generated_api_client/models/create_memoryset_request_index_params.py +66 -0
  55. orca_sdk/_generated_api_client/models/create_memoryset_request_index_type.py +13 -0
  56. orca_sdk/_generated_api_client/models/create_regression_model_request.py +137 -0
  57. orca_sdk/_generated_api_client/models/embedding_evaluation_payload.py +187 -0
  58. orca_sdk/_generated_api_client/models/embedding_evaluation_response.py +10 -0
  59. orca_sdk/_generated_api_client/models/evaluation_response.py +22 -9
  60. orca_sdk/_generated_api_client/models/evaluation_response_classification_metrics.py +140 -0
  61. orca_sdk/_generated_api_client/models/evaluation_response_regression_metrics.py +140 -0
  62. orca_sdk/_generated_api_client/models/memory_type.py +9 -0
  63. orca_sdk/_generated_api_client/models/{labeled_memoryset_metadata.py → memoryset_metadata.py} +73 -13
  64. orca_sdk/_generated_api_client/models/memoryset_metadata_index_params.py +55 -0
  65. orca_sdk/_generated_api_client/models/memoryset_metadata_index_type.py +13 -0
  66. orca_sdk/_generated_api_client/models/{labeled_memoryset_update.py → memoryset_update.py} +19 -31
  67. orca_sdk/_generated_api_client/models/not_found_error_response_resource_type_0.py +1 -0
  68. 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
  69. orca_sdk/_generated_api_client/models/{precision_recall_curve.py → pr_curve.py} +5 -13
  70. orca_sdk/_generated_api_client/models/{rac_model_update.py → predictive_model_update.py} +14 -5
  71. orca_sdk/_generated_api_client/models/pretrained_embedding_model_metadata.py +11 -1
  72. orca_sdk/_generated_api_client/models/rar_head_type.py +8 -0
  73. orca_sdk/_generated_api_client/models/regression_evaluation_request.py +148 -0
  74. orca_sdk/_generated_api_client/models/regression_metrics.py +172 -0
  75. orca_sdk/_generated_api_client/models/regression_model_metadata.py +177 -0
  76. orca_sdk/_generated_api_client/models/regression_prediction_request.py +195 -0
  77. orca_sdk/_generated_api_client/models/roc_curve.py +0 -8
  78. orca_sdk/_generated_api_client/models/score_prediction_memory_lookup.py +196 -0
  79. orca_sdk/_generated_api_client/models/score_prediction_memory_lookup_metadata.py +68 -0
  80. orca_sdk/_generated_api_client/models/score_prediction_with_memories_and_feedback.py +252 -0
  81. orca_sdk/_generated_api_client/models/scored_memory.py +172 -0
  82. orca_sdk/_generated_api_client/models/scored_memory_insert.py +128 -0
  83. orca_sdk/_generated_api_client/models/scored_memory_insert_metadata.py +68 -0
  84. orca_sdk/_generated_api_client/models/scored_memory_lookup.py +180 -0
  85. orca_sdk/_generated_api_client/models/scored_memory_lookup_metadata.py +68 -0
  86. orca_sdk/_generated_api_client/models/scored_memory_metadata.py +68 -0
  87. orca_sdk/_generated_api_client/models/scored_memory_update.py +171 -0
  88. orca_sdk/_generated_api_client/models/scored_memory_update_metadata_type_0.py +68 -0
  89. orca_sdk/_generated_api_client/models/scored_memory_with_feedback_metrics.py +193 -0
  90. orca_sdk/_generated_api_client/models/scored_memory_with_feedback_metrics_feedback_metrics.py +68 -0
  91. orca_sdk/_generated_api_client/models/scored_memory_with_feedback_metrics_metadata.py +68 -0
  92. orca_sdk/_generated_api_client/models/update_prediction_request.py +20 -0
  93. orca_sdk/_shared/__init__.py +9 -1
  94. orca_sdk/_shared/metrics.py +257 -87
  95. orca_sdk/_shared/metrics_test.py +136 -77
  96. orca_sdk/_utils/data_parsing.py +0 -3
  97. orca_sdk/_utils/data_parsing_test.py +0 -3
  98. orca_sdk/_utils/prediction_result_ui.py +55 -23
  99. orca_sdk/classification_model.py +183 -175
  100. orca_sdk/classification_model_test.py +147 -157
  101. orca_sdk/conftest.py +76 -26
  102. orca_sdk/datasource_test.py +0 -1
  103. orca_sdk/embedding_model.py +136 -14
  104. orca_sdk/embedding_model_test.py +10 -6
  105. orca_sdk/job.py +329 -0
  106. orca_sdk/job_test.py +48 -0
  107. orca_sdk/memoryset.py +882 -161
  108. orca_sdk/memoryset_test.py +56 -23
  109. orca_sdk/regression_model.py +647 -0
  110. orca_sdk/regression_model_test.py +338 -0
  111. orca_sdk/telemetry.py +223 -106
  112. orca_sdk/telemetry_test.py +34 -30
  113. {orca_sdk-0.0.94.dist-info → orca_sdk-0.0.95.dist-info}/METADATA +2 -4
  114. {orca_sdk-0.0.94.dist-info → orca_sdk-0.0.95.dist-info}/RECORD +115 -69
  115. orca_sdk/_utils/task.py +0 -73
  116. {orca_sdk-0.0.94.dist-info → orca_sdk-0.0.95.dist-info}/WHEEL +0 -0
@@ -17,15 +17,15 @@ import httpx
17
17
 
18
18
  from ...client import _client_context
19
19
  from ...errors import get_error_for_response
20
- from ...models.labeled_memoryset_metadata import LabeledMemorysetMetadata
21
- from ...models.labeled_memoryset_update import LabeledMemorysetUpdate
20
+ from ...models.memoryset_metadata import MemorysetMetadata
21
+ from ...models.memoryset_update import MemorysetUpdate
22
22
  from ...types import Response
23
23
 
24
24
 
25
25
  def _get_kwargs(
26
26
  name_or_id: str,
27
27
  *,
28
- body: LabeledMemorysetUpdate,
28
+ body: MemorysetUpdate,
29
29
  ) -> dict[str, Any]:
30
30
  headers: Dict[str, Any] = {}
31
31
 
@@ -43,9 +43,9 @@ def _get_kwargs(
43
43
  return _kwargs
44
44
 
45
45
 
46
- def _parse_response(*, response: httpx.Response) -> LabeledMemorysetMetadata:
46
+ def _parse_response(*, response: httpx.Response) -> MemorysetMetadata:
47
47
  if response.status_code == 200:
48
- response_200 = LabeledMemorysetMetadata.from_dict(response.json())
48
+ response_200 = MemorysetMetadata.from_dict(response.json())
49
49
 
50
50
  return response_200
51
51
  if response.status_code == 401:
@@ -60,7 +60,7 @@ def _parse_response(*, response: httpx.Response) -> LabeledMemorysetMetadata:
60
60
  raise get_error_for_response(response)
61
61
 
62
62
 
63
- def _build_response(*, response: httpx.Response) -> Response[LabeledMemorysetMetadata]:
63
+ def _build_response(*, response: httpx.Response) -> Response[MemorysetMetadata]:
64
64
  return Response(
65
65
  status_code=HTTPStatus(response.status_code),
66
66
  content=response.content,
@@ -72,20 +72,20 @@ def _build_response(*, response: httpx.Response) -> Response[LabeledMemorysetMet
72
72
  def sync_detailed(
73
73
  name_or_id: str,
74
74
  *,
75
- body: LabeledMemorysetUpdate,
76
- ) -> Response[LabeledMemorysetMetadata]:
75
+ body: MemorysetUpdate,
76
+ ) -> Response[MemorysetMetadata]:
77
77
  """Update Memoryset
78
78
 
79
79
  Args:
80
80
  name_or_id (str):
81
- body (LabeledMemorysetUpdate):
81
+ body (MemorysetUpdate):
82
82
 
83
83
  Raises:
84
84
  errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
85
85
  httpx.TimeoutException: If the request takes longer than Client.timeout.
86
86
 
87
87
  Returns:
88
- Response[LabeledMemorysetMetadata]
88
+ Response[MemorysetMetadata]
89
89
  """
90
90
 
91
91
  kwargs = _get_kwargs(
@@ -104,20 +104,20 @@ def sync_detailed(
104
104
  def sync(
105
105
  name_or_id: str,
106
106
  *,
107
- body: LabeledMemorysetUpdate,
108
- ) -> LabeledMemorysetMetadata:
107
+ body: MemorysetUpdate,
108
+ ) -> MemorysetMetadata:
109
109
  """Update Memoryset
110
110
 
111
111
  Args:
112
112
  name_or_id (str):
113
- body (LabeledMemorysetUpdate):
113
+ body (MemorysetUpdate):
114
114
 
115
115
  Raises:
116
116
  errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
117
117
  httpx.TimeoutException: If the request takes longer than Client.timeout.
118
118
 
119
119
  Returns:
120
- LabeledMemorysetMetadata
120
+ MemorysetMetadata
121
121
  """
122
122
 
123
123
  return sync_detailed(
@@ -129,20 +129,20 @@ def sync(
129
129
  async def asyncio_detailed(
130
130
  name_or_id: str,
131
131
  *,
132
- body: LabeledMemorysetUpdate,
133
- ) -> Response[LabeledMemorysetMetadata]:
132
+ body: MemorysetUpdate,
133
+ ) -> Response[MemorysetMetadata]:
134
134
  """Update Memoryset
135
135
 
136
136
  Args:
137
137
  name_or_id (str):
138
- body (LabeledMemorysetUpdate):
138
+ body (MemorysetUpdate):
139
139
 
140
140
  Raises:
141
141
  errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
142
142
  httpx.TimeoutException: If the request takes longer than Client.timeout.
143
143
 
144
144
  Returns:
145
- Response[LabeledMemorysetMetadata]
145
+ Response[MemorysetMetadata]
146
146
  """
147
147
 
148
148
  kwargs = _get_kwargs(
@@ -159,20 +159,20 @@ async def asyncio_detailed(
159
159
  async def asyncio(
160
160
  name_or_id: str,
161
161
  *,
162
- body: LabeledMemorysetUpdate,
163
- ) -> LabeledMemorysetMetadata:
162
+ body: MemorysetUpdate,
163
+ ) -> MemorysetMetadata:
164
164
  """Update Memoryset
165
165
 
166
166
  Args:
167
167
  name_or_id (str):
168
- body (LabeledMemorysetUpdate):
168
+ body (MemorysetUpdate):
169
169
 
170
170
  Raises:
171
171
  errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
172
172
  httpx.TimeoutException: If the request takes longer than Client.timeout.
173
173
 
174
174
  Returns:
175
- LabeledMemorysetMetadata
175
+ MemorysetMetadata
176
176
  """
177
177
 
178
178
  return (
@@ -0,0 +1,150 @@
1
+ """
2
+ This file is generated by the openapi-python-client tool via the generate_api_client.py script
3
+
4
+ It is a customized template from the openapi-python-client tool's default template:
5
+ https://github.com/openapi-generators/openapi-python-client/blob/861ef5622f10fc96d240dc9becb0edf94e61446c/openapi_python_client/templates/endpoint_module.py.jinja
6
+
7
+ The main changes are:
8
+ - Update the API call responses to either return the successful response type or raise an error by:
9
+ - Updating the _parse_response function to raise an error if the response status code is not in the 2xx range
10
+ - Inject a client into every method via a context manager
11
+ """
12
+
13
+ from http import HTTPStatus
14
+ from typing import Any, List, Union
15
+
16
+ import httpx
17
+
18
+ from ...client import _client_context
19
+ from ...errors import get_error_for_response
20
+ from ...models.classification_model_metadata import ClassificationModelMetadata
21
+ from ...models.regression_model_metadata import RegressionModelMetadata
22
+ from ...types import Response
23
+
24
+
25
+ def _get_kwargs() -> dict[str, Any]:
26
+ _kwargs: dict[str, Any] = {
27
+ "method": "get",
28
+ "url": "/predictive_model/",
29
+ }
30
+
31
+ return _kwargs
32
+
33
+
34
+ def _parse_response(
35
+ *, response: httpx.Response
36
+ ) -> List[Union["ClassificationModelMetadata", "RegressionModelMetadata"]]:
37
+ if response.status_code == 200:
38
+ response_200 = []
39
+ _response_200 = response.json()
40
+ for response_200_item_data in _response_200:
41
+
42
+ def _parse_response_200_item(
43
+ data: object,
44
+ ) -> Union["ClassificationModelMetadata", "RegressionModelMetadata"]:
45
+ try:
46
+ if not isinstance(data, dict):
47
+ raise TypeError()
48
+ response_200_item_type_0 = ClassificationModelMetadata.from_dict(data)
49
+
50
+ return response_200_item_type_0
51
+ except: # noqa: E722
52
+ pass
53
+ if not isinstance(data, dict):
54
+ raise TypeError()
55
+ response_200_item_type_1 = RegressionModelMetadata.from_dict(data)
56
+
57
+ return response_200_item_type_1
58
+
59
+ response_200_item = _parse_response_200_item(response_200_item_data)
60
+
61
+ response_200.append(response_200_item)
62
+
63
+ return response_200
64
+ if response.status_code == 401:
65
+ raise get_error_for_response(response)
66
+ if response.status_code == 500:
67
+ raise get_error_for_response(response)
68
+ if response.status_code == 503:
69
+ raise get_error_for_response(response)
70
+ else:
71
+ raise get_error_for_response(response)
72
+
73
+
74
+ def _build_response(
75
+ *, response: httpx.Response
76
+ ) -> Response[List[Union["ClassificationModelMetadata", "RegressionModelMetadata"]]]:
77
+ return Response(
78
+ status_code=HTTPStatus(response.status_code),
79
+ content=response.content,
80
+ headers=response.headers,
81
+ parsed=_parse_response(response=response),
82
+ )
83
+
84
+
85
+ def sync_detailed() -> Response[List[Union["ClassificationModelMetadata", "RegressionModelMetadata"]]]:
86
+ """List Predictive Models
87
+
88
+ Raises:
89
+ errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
90
+ httpx.TimeoutException: If the request takes longer than Client.timeout.
91
+
92
+ Returns:
93
+ Response[List[Union['ClassificationModelMetadata', 'RegressionModelMetadata']]]
94
+ """
95
+
96
+ kwargs = _get_kwargs()
97
+
98
+ with _client_context() as client:
99
+ response = client.get_httpx_client().request(
100
+ **kwargs,
101
+ )
102
+
103
+ return _build_response(response=response)
104
+
105
+
106
+ def sync() -> List[Union["ClassificationModelMetadata", "RegressionModelMetadata"]]:
107
+ """List Predictive Models
108
+
109
+ Raises:
110
+ errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
111
+ httpx.TimeoutException: If the request takes longer than Client.timeout.
112
+
113
+ Returns:
114
+ List[Union['ClassificationModelMetadata', 'RegressionModelMetadata']]
115
+ """
116
+
117
+ return sync_detailed().parsed
118
+
119
+
120
+ async def asyncio_detailed() -> Response[List[Union["ClassificationModelMetadata", "RegressionModelMetadata"]]]:
121
+ """List Predictive Models
122
+
123
+ Raises:
124
+ errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
125
+ httpx.TimeoutException: If the request takes longer than Client.timeout.
126
+
127
+ Returns:
128
+ Response[List[Union['ClassificationModelMetadata', 'RegressionModelMetadata']]]
129
+ """
130
+
131
+ kwargs = _get_kwargs()
132
+
133
+ with _client_context() as client:
134
+ response = await client.get_async_httpx_client().request(**kwargs)
135
+
136
+ return _build_response(response=response)
137
+
138
+
139
+ async def asyncio() -> List[Union["ClassificationModelMetadata", "RegressionModelMetadata"]]:
140
+ """List Predictive Models
141
+
142
+ Raises:
143
+ errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
144
+ httpx.TimeoutException: If the request takes longer than Client.timeout.
145
+
146
+ Returns:
147
+ List[Union['ClassificationModelMetadata', 'RegressionModelMetadata']]
148
+ """
149
+
150
+ return (await asyncio_detailed()).parsed
@@ -17,20 +17,20 @@ import httpx
17
17
 
18
18
  from ...client import _client_context
19
19
  from ...errors import get_error_for_response
20
- from ...models.create_rac_model_request import CreateRACModelRequest
21
- from ...models.rac_model_metadata import RACModelMetadata
20
+ from ...models.create_regression_model_request import CreateRegressionModelRequest
21
+ from ...models.regression_model_metadata import RegressionModelMetadata
22
22
  from ...types import Response
23
23
 
24
24
 
25
25
  def _get_kwargs(
26
26
  *,
27
- body: CreateRACModelRequest,
27
+ body: CreateRegressionModelRequest,
28
28
  ) -> dict[str, Any]:
29
29
  headers: Dict[str, Any] = {}
30
30
 
31
31
  _kwargs: dict[str, Any] = {
32
32
  "method": "post",
33
- "url": "/classification_model/",
33
+ "url": "/regression_model/",
34
34
  }
35
35
 
36
36
  _body = body.to_dict()
@@ -42,9 +42,9 @@ def _get_kwargs(
42
42
  return _kwargs
43
43
 
44
44
 
45
- def _parse_response(*, response: httpx.Response) -> RACModelMetadata:
45
+ def _parse_response(*, response: httpx.Response) -> RegressionModelMetadata:
46
46
  if response.status_code == 201:
47
- response_201 = RACModelMetadata.from_dict(response.json())
47
+ response_201 = RegressionModelMetadata.from_dict(response.json())
48
48
 
49
49
  return response_201
50
50
  if response.status_code == 401:
@@ -59,7 +59,7 @@ def _parse_response(*, response: httpx.Response) -> RACModelMetadata:
59
59
  raise get_error_for_response(response)
60
60
 
61
61
 
62
- def _build_response(*, response: httpx.Response) -> Response[RACModelMetadata]:
62
+ def _build_response(*, response: httpx.Response) -> Response[RegressionModelMetadata]:
63
63
  return Response(
64
64
  status_code=HTTPStatus(response.status_code),
65
65
  content=response.content,
@@ -70,19 +70,19 @@ def _build_response(*, response: httpx.Response) -> Response[RACModelMetadata]:
70
70
 
71
71
  def sync_detailed(
72
72
  *,
73
- body: CreateRACModelRequest,
74
- ) -> Response[RACModelMetadata]:
75
- """Create Model
73
+ body: CreateRegressionModelRequest,
74
+ ) -> Response[RegressionModelMetadata]:
75
+ """Create Regression Model
76
76
 
77
77
  Args:
78
- body (CreateRACModelRequest):
78
+ body (CreateRegressionModelRequest):
79
79
 
80
80
  Raises:
81
81
  errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
82
82
  httpx.TimeoutException: If the request takes longer than Client.timeout.
83
83
 
84
84
  Returns:
85
- Response[RACModelMetadata]
85
+ Response[RegressionModelMetadata]
86
86
  """
87
87
 
88
88
  kwargs = _get_kwargs(
@@ -99,19 +99,19 @@ def sync_detailed(
99
99
 
100
100
  def sync(
101
101
  *,
102
- body: CreateRACModelRequest,
103
- ) -> RACModelMetadata:
104
- """Create Model
102
+ body: CreateRegressionModelRequest,
103
+ ) -> RegressionModelMetadata:
104
+ """Create Regression Model
105
105
 
106
106
  Args:
107
- body (CreateRACModelRequest):
107
+ body (CreateRegressionModelRequest):
108
108
 
109
109
  Raises:
110
110
  errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
111
111
  httpx.TimeoutException: If the request takes longer than Client.timeout.
112
112
 
113
113
  Returns:
114
- RACModelMetadata
114
+ RegressionModelMetadata
115
115
  """
116
116
 
117
117
  return sync_detailed(
@@ -121,19 +121,19 @@ def sync(
121
121
 
122
122
  async def asyncio_detailed(
123
123
  *,
124
- body: CreateRACModelRequest,
125
- ) -> Response[RACModelMetadata]:
126
- """Create Model
124
+ body: CreateRegressionModelRequest,
125
+ ) -> Response[RegressionModelMetadata]:
126
+ """Create Regression Model
127
127
 
128
128
  Args:
129
- body (CreateRACModelRequest):
129
+ body (CreateRegressionModelRequest):
130
130
 
131
131
  Raises:
132
132
  errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
133
133
  httpx.TimeoutException: If the request takes longer than Client.timeout.
134
134
 
135
135
  Returns:
136
- Response[RACModelMetadata]
136
+ Response[RegressionModelMetadata]
137
137
  """
138
138
 
139
139
  kwargs = _get_kwargs(
@@ -148,19 +148,19 @@ async def asyncio_detailed(
148
148
 
149
149
  async def asyncio(
150
150
  *,
151
- body: CreateRACModelRequest,
152
- ) -> RACModelMetadata:
153
- """Create Model
151
+ body: CreateRegressionModelRequest,
152
+ ) -> RegressionModelMetadata:
153
+ """Create Regression Model
154
154
 
155
155
  Args:
156
- body (CreateRACModelRequest):
156
+ body (CreateRegressionModelRequest):
157
157
 
158
158
  Raises:
159
159
  errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
160
160
  httpx.TimeoutException: If the request takes longer than Client.timeout.
161
161
 
162
162
  Returns:
163
- RACModelMetadata
163
+ RegressionModelMetadata
164
164
  """
165
165
 
166
166
  return (
@@ -0,0 +1,168 @@
1
+ """
2
+ This file is generated by the openapi-python-client tool via the generate_api_client.py script
3
+
4
+ It is a customized template from the openapi-python-client tool's default template:
5
+ https://github.com/openapi-generators/openapi-python-client/blob/861ef5622f10fc96d240dc9becb0edf94e61446c/openapi_python_client/templates/endpoint_module.py.jinja
6
+
7
+ The main changes are:
8
+ - Update the API call responses to either return the successful response type or raise an error by:
9
+ - Updating the _parse_response function to raise an error if the response status code is not in the 2xx range
10
+ - Inject a client into every method via a context manager
11
+ """
12
+
13
+ from http import HTTPStatus
14
+ from typing import Any
15
+ from uuid import UUID
16
+
17
+ import httpx
18
+
19
+ from ...client import _client_context
20
+ from ...errors import get_error_for_response
21
+ from ...types import Response
22
+
23
+
24
+ def _get_kwargs(
25
+ model_name_or_id: str,
26
+ task_id: UUID,
27
+ ) -> dict[str, Any]:
28
+ _kwargs: dict[str, Any] = {
29
+ "method": "delete",
30
+ "url": f"/regression_model/{model_name_or_id}/evaluation/{task_id}",
31
+ }
32
+
33
+ return _kwargs
34
+
35
+
36
+ def _parse_response(*, response: httpx.Response) -> None:
37
+ if response.status_code == 204:
38
+ response_204 = None
39
+ return response_204
40
+ if response.status_code == 401:
41
+ raise get_error_for_response(response)
42
+ if response.status_code == 500:
43
+ raise get_error_for_response(response)
44
+ if response.status_code == 503:
45
+ raise get_error_for_response(response)
46
+ if response.status_code == 404:
47
+ raise get_error_for_response(response)
48
+ else:
49
+ raise get_error_for_response(response)
50
+
51
+
52
+ def _build_response(*, response: httpx.Response) -> Response[None]:
53
+ return Response(
54
+ status_code=HTTPStatus(response.status_code),
55
+ content=response.content,
56
+ headers=response.headers,
57
+ parsed=_parse_response(response=response),
58
+ )
59
+
60
+
61
+ def sync_detailed(
62
+ model_name_or_id: str,
63
+ task_id: UUID,
64
+ ) -> Response[None]:
65
+ """Delete Regression Model Evaluation
66
+
67
+ Args:
68
+ model_name_or_id (str):
69
+ task_id (UUID):
70
+
71
+ Raises:
72
+ errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
73
+ httpx.TimeoutException: If the request takes longer than Client.timeout.
74
+
75
+ Returns:
76
+ Response[None]
77
+ """
78
+
79
+ kwargs = _get_kwargs(
80
+ model_name_or_id=model_name_or_id,
81
+ task_id=task_id,
82
+ )
83
+
84
+ with _client_context() as client:
85
+ response = client.get_httpx_client().request(
86
+ **kwargs,
87
+ )
88
+
89
+ return _build_response(response=response)
90
+
91
+
92
+ def sync(
93
+ model_name_or_id: str,
94
+ task_id: UUID,
95
+ ) -> None:
96
+ """Delete Regression Model Evaluation
97
+
98
+ Args:
99
+ model_name_or_id (str):
100
+ task_id (UUID):
101
+
102
+ Raises:
103
+ errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
104
+ httpx.TimeoutException: If the request takes longer than Client.timeout.
105
+
106
+ Returns:
107
+ None
108
+ """
109
+
110
+ return sync_detailed(
111
+ model_name_or_id=model_name_or_id,
112
+ task_id=task_id,
113
+ ).parsed
114
+
115
+
116
+ async def asyncio_detailed(
117
+ model_name_or_id: str,
118
+ task_id: UUID,
119
+ ) -> Response[None]:
120
+ """Delete Regression Model Evaluation
121
+
122
+ Args:
123
+ model_name_or_id (str):
124
+ task_id (UUID):
125
+
126
+ Raises:
127
+ errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
128
+ httpx.TimeoutException: If the request takes longer than Client.timeout.
129
+
130
+ Returns:
131
+ Response[None]
132
+ """
133
+
134
+ kwargs = _get_kwargs(
135
+ model_name_or_id=model_name_or_id,
136
+ task_id=task_id,
137
+ )
138
+
139
+ with _client_context() as client:
140
+ response = await client.get_async_httpx_client().request(**kwargs)
141
+
142
+ return _build_response(response=response)
143
+
144
+
145
+ async def asyncio(
146
+ model_name_or_id: str,
147
+ task_id: UUID,
148
+ ) -> None:
149
+ """Delete Regression Model Evaluation
150
+
151
+ Args:
152
+ model_name_or_id (str):
153
+ task_id (UUID):
154
+
155
+ Raises:
156
+ errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
157
+ httpx.TimeoutException: If the request takes longer than Client.timeout.
158
+
159
+ Returns:
160
+ None
161
+ """
162
+
163
+ return (
164
+ await asyncio_detailed(
165
+ model_name_or_id=model_name_or_id,
166
+ task_id=task_id,
167
+ )
168
+ ).parsed
@@ -25,7 +25,7 @@ def _get_kwargs(
25
25
  ) -> dict[str, Any]:
26
26
  _kwargs: dict[str, Any] = {
27
27
  "method": "delete",
28
- "url": f"/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):