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