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,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.task_status_info import TaskStatusInfo
21
+ from ...types import Response
22
+
23
+
24
+ def _get_kwargs(
25
+ task_id: str,
26
+ ) -> dict[str, Any]:
27
+ _kwargs: dict[str, Any] = {
28
+ "method": "get",
29
+ "url": f"/task/{task_id}/status",
30
+ }
31
+
32
+ return _kwargs
33
+
34
+
35
+ def _parse_response(*, response: httpx.Response) -> TaskStatusInfo:
36
+ if response.status_code == 200:
37
+ response_200 = TaskStatusInfo.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[TaskStatusInfo]:
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
+ task_id: str,
63
+ ) -> Response[TaskStatusInfo]:
64
+ """Get Task Status
65
+
66
+ Args:
67
+ task_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[TaskStatusInfo]
75
+ """
76
+
77
+ kwargs = _get_kwargs(
78
+ task_id=task_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
+ task_id: str,
91
+ ) -> TaskStatusInfo:
92
+ """Get Task Status
93
+
94
+ Args:
95
+ task_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
+ TaskStatusInfo
103
+ """
104
+
105
+ return sync_detailed(
106
+ task_id=task_id,
107
+ ).parsed
108
+
109
+
110
+ async def asyncio_detailed(
111
+ task_id: str,
112
+ ) -> Response[TaskStatusInfo]:
113
+ """Get Task Status
114
+
115
+ Args:
116
+ task_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[TaskStatusInfo]
124
+ """
125
+
126
+ kwargs = _get_kwargs(
127
+ task_id=task_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
+ task_id: str,
138
+ ) -> TaskStatusInfo:
139
+ """Get Task Status
140
+
141
+ Args:
142
+ task_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
+ TaskStatusInfo
150
+ """
151
+
152
+ return (
153
+ await asyncio_detailed(
154
+ task_id=task_id,
155
+ )
156
+ ).parsed
@@ -0,0 +1,243 @@
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 enum import Enum
14
+ from http import HTTPStatus
15
+ from typing import Any, Dict, List, Union
16
+
17
+ import httpx
18
+
19
+ from ...client import _client_context
20
+ from ...errors import get_error_for_response
21
+ from ...models.task import Task
22
+ from ...models.task_status import TaskStatus
23
+ from ...types import UNSET, Response, Unset
24
+
25
+
26
+ def _get_kwargs(
27
+ *,
28
+ status: Union[List[TaskStatus], None, TaskStatus, Unset] = UNSET,
29
+ type: Union[List[str], None, Unset, str] = UNSET,
30
+ limit: Union[Unset, int] = 100,
31
+ offset: Union[Unset, int] = 0,
32
+ ) -> dict[str, Any]:
33
+ params: Dict[str, Any] = {}
34
+
35
+ json_status: Union[List[str], None, Unset, str]
36
+ if isinstance(status, Unset):
37
+ json_status = UNSET
38
+ elif isinstance(status, TaskStatus):
39
+ json_status = status.value if isinstance(status, Enum) else status
40
+ elif isinstance(status, list):
41
+ json_status = []
42
+ for status_type_1_item_data in status:
43
+ status_type_1_item = (
44
+ status_type_1_item_data.value if isinstance(status_type_1_item_data, Enum) else status_type_1_item_data
45
+ )
46
+ json_status.append(status_type_1_item)
47
+
48
+ else:
49
+ json_status = status
50
+ params["status"] = json_status
51
+
52
+ json_type: Union[List[str], None, Unset, str]
53
+ if isinstance(type, Unset):
54
+ json_type = UNSET
55
+ elif isinstance(type, list):
56
+ json_type = type
57
+
58
+ else:
59
+ json_type = type
60
+ params["type"] = json_type
61
+
62
+ params["limit"] = limit
63
+
64
+ params["offset"] = offset
65
+
66
+ params = {k: v for k, v in params.items() if v is not UNSET and v is not None}
67
+
68
+ _kwargs: dict[str, Any] = {
69
+ "method": "get",
70
+ "url": "/task",
71
+ "params": params,
72
+ }
73
+
74
+ return _kwargs
75
+
76
+
77
+ def _parse_response(*, response: httpx.Response) -> List["Task"]:
78
+ if response.status_code == 200:
79
+ response_200 = []
80
+ _response_200 = response.json()
81
+ for response_200_item_data in _response_200:
82
+ response_200_item = Task.from_dict(response_200_item_data)
83
+
84
+ response_200.append(response_200_item)
85
+
86
+ return response_200
87
+ if response.status_code == 401:
88
+ raise get_error_for_response(response)
89
+ if response.status_code == 500:
90
+ raise get_error_for_response(response)
91
+ if response.status_code == 503:
92
+ raise get_error_for_response(response)
93
+ if response.status_code == 422:
94
+ raise get_error_for_response(response)
95
+ if response.status_code == 404:
96
+ raise get_error_for_response(response)
97
+
98
+
99
+ def _build_response(*, response: httpx.Response) -> Response[List["Task"]]:
100
+ return Response(
101
+ status_code=HTTPStatus(response.status_code),
102
+ content=response.content,
103
+ headers=response.headers,
104
+ parsed=_parse_response(response=response),
105
+ )
106
+
107
+
108
+ def sync_detailed(
109
+ *,
110
+ status: Union[List[TaskStatus], None, TaskStatus, Unset] = UNSET,
111
+ type: Union[List[str], None, Unset, str] = UNSET,
112
+ limit: Union[Unset, int] = 100,
113
+ offset: Union[Unset, int] = 0,
114
+ ) -> Response[List["Task"]]:
115
+ """List Tasks
116
+
117
+ Args:
118
+ status (Union[List[TaskStatus], None, TaskStatus, Unset]):
119
+ type (Union[List[str], None, Unset, str]):
120
+ limit (Union[Unset, int]): Default: 100.
121
+ offset (Union[Unset, int]): Default: 0.
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['Task']]
129
+ """
130
+
131
+ kwargs = _get_kwargs(
132
+ status=status,
133
+ type=type,
134
+ limit=limit,
135
+ offset=offset,
136
+ )
137
+
138
+ with _client_context() as client:
139
+ response = client.get_httpx_client().request(
140
+ **kwargs,
141
+ )
142
+
143
+ return _build_response(response=response)
144
+
145
+
146
+ def sync(
147
+ *,
148
+ status: Union[List[TaskStatus], None, TaskStatus, Unset] = UNSET,
149
+ type: Union[List[str], None, Unset, str] = UNSET,
150
+ limit: Union[Unset, int] = 100,
151
+ offset: Union[Unset, int] = 0,
152
+ ) -> List["Task"]:
153
+ """List Tasks
154
+
155
+ Args:
156
+ status (Union[List[TaskStatus], None, TaskStatus, Unset]):
157
+ type (Union[List[str], None, Unset, str]):
158
+ limit (Union[Unset, int]): Default: 100.
159
+ offset (Union[Unset, int]): Default: 0.
160
+
161
+ Raises:
162
+ errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
163
+ httpx.TimeoutException: If the request takes longer than Client.timeout.
164
+
165
+ Returns:
166
+ List['Task']
167
+ """
168
+
169
+ return sync_detailed(
170
+ status=status,
171
+ type=type,
172
+ limit=limit,
173
+ offset=offset,
174
+ ).parsed
175
+
176
+
177
+ async def asyncio_detailed(
178
+ *,
179
+ status: Union[List[TaskStatus], None, TaskStatus, Unset] = UNSET,
180
+ type: Union[List[str], None, Unset, str] = UNSET,
181
+ limit: Union[Unset, int] = 100,
182
+ offset: Union[Unset, int] = 0,
183
+ ) -> Response[List["Task"]]:
184
+ """List Tasks
185
+
186
+ Args:
187
+ status (Union[List[TaskStatus], None, TaskStatus, Unset]):
188
+ type (Union[List[str], None, Unset, str]):
189
+ limit (Union[Unset, int]): Default: 100.
190
+ offset (Union[Unset, int]): Default: 0.
191
+
192
+ Raises:
193
+ errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
194
+ httpx.TimeoutException: If the request takes longer than Client.timeout.
195
+
196
+ Returns:
197
+ Response[List['Task']]
198
+ """
199
+
200
+ kwargs = _get_kwargs(
201
+ status=status,
202
+ type=type,
203
+ limit=limit,
204
+ offset=offset,
205
+ )
206
+
207
+ with _client_context() as client:
208
+ response = await client.get_async_httpx_client().request(**kwargs)
209
+
210
+ return _build_response(response=response)
211
+
212
+
213
+ async def asyncio(
214
+ *,
215
+ status: Union[List[TaskStatus], None, TaskStatus, Unset] = UNSET,
216
+ type: Union[List[str], None, Unset, str] = UNSET,
217
+ limit: Union[Unset, int] = 100,
218
+ offset: Union[Unset, int] = 0,
219
+ ) -> List["Task"]:
220
+ """List Tasks
221
+
222
+ Args:
223
+ status (Union[List[TaskStatus], None, TaskStatus, Unset]):
224
+ type (Union[List[str], None, Unset, str]):
225
+ limit (Union[Unset, int]): Default: 100.
226
+ offset (Union[Unset, int]): Default: 0.
227
+
228
+ Raises:
229
+ errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
230
+ httpx.TimeoutException: If the request takes longer than Client.timeout.
231
+
232
+ Returns:
233
+ List['Task']
234
+ """
235
+
236
+ return (
237
+ await asyncio_detailed(
238
+ status=status,
239
+ type=type,
240
+ limit=limit,
241
+ offset=offset,
242
+ )
243
+ ).parsed
@@ -0,0 +1,162 @@
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"/telemetry/feedback_category/{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
+ """Drop Feedback Category With Data
63
+
64
+ Delete a feedback category and all associated feedback records.
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[None]
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
+ ) -> None:
92
+ """Drop Feedback Category With Data
93
+
94
+ Delete a feedback category and all associated feedback records.
95
+
96
+ Args:
97
+ name_or_id (str):
98
+
99
+ Raises:
100
+ errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
101
+ httpx.TimeoutException: If the request takes longer than Client.timeout.
102
+
103
+ Returns:
104
+ None
105
+ """
106
+
107
+ return sync_detailed(
108
+ name_or_id=name_or_id,
109
+ ).parsed
110
+
111
+
112
+ async def asyncio_detailed(
113
+ name_or_id: str,
114
+ ) -> Response[None]:
115
+ """Drop Feedback Category With Data
116
+
117
+ Delete a feedback category and all associated feedback records.
118
+
119
+ Args:
120
+ name_or_id (str):
121
+
122
+ Raises:
123
+ errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
124
+ httpx.TimeoutException: If the request takes longer than Client.timeout.
125
+
126
+ Returns:
127
+ Response[None]
128
+ """
129
+
130
+ kwargs = _get_kwargs(
131
+ name_or_id=name_or_id,
132
+ )
133
+
134
+ with _client_context() as client:
135
+ response = await client.get_async_httpx_client().request(**kwargs)
136
+
137
+ return _build_response(response=response)
138
+
139
+
140
+ async def asyncio(
141
+ name_or_id: str,
142
+ ) -> None:
143
+ """Drop Feedback Category With Data
144
+
145
+ Delete a feedback category and all associated feedback records.
146
+
147
+ Args:
148
+ name_or_id (str):
149
+
150
+ Raises:
151
+ errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
152
+ httpx.TimeoutException: If the request takes longer than Client.timeout.
153
+
154
+ Returns:
155
+ None
156
+ """
157
+
158
+ return (
159
+ await asyncio_detailed(
160
+ name_or_id=name_or_id,
161
+ )
162
+ ).parsed