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,129 @@
1
+ """
2
+ This file is generated by the openapi-python-client tool via the generate_api_client.py script
3
+
4
+ It is a customized template from the openapi-python-client tool's default template:
5
+ https://github.com/openapi-generators/openapi-python-client/blob/861ef5622f10fc96d240dc9becb0edf94e61446c/openapi_python_client/templates/endpoint_module.py.jinja
6
+
7
+ The main changes are:
8
+ - Update the API call responses to either return the successful response type or raise an error by:
9
+ - Updating the _parse_response function to raise an error if the response status code is not in the 2xx range
10
+ - Inject a client into every method via a context manager
11
+ """
12
+
13
+ from http import HTTPStatus
14
+ from typing import Any, List
15
+
16
+ import httpx
17
+
18
+ from ...client import _client_context
19
+ from ...errors import get_error_for_response
20
+ from ...models.finetuned_embedding_model_metadata import FinetunedEmbeddingModelMetadata
21
+ from ...types import Response
22
+
23
+
24
+ def _get_kwargs() -> dict[str, Any]:
25
+ _kwargs: dict[str, Any] = {
26
+ "method": "get",
27
+ "url": "/finetuned_embedding_model/",
28
+ }
29
+
30
+ return _kwargs
31
+
32
+
33
+ def _parse_response(*, response: httpx.Response) -> List["FinetunedEmbeddingModelMetadata"]:
34
+ if response.status_code == 200:
35
+ response_200 = []
36
+ _response_200 = response.json()
37
+ for response_200_item_data in _response_200:
38
+ response_200_item = FinetunedEmbeddingModelMetadata.from_dict(response_200_item_data)
39
+
40
+ response_200.append(response_200_item)
41
+
42
+ return response_200
43
+ if response.status_code == 401:
44
+ raise get_error_for_response(response)
45
+ if response.status_code == 500:
46
+ raise get_error_for_response(response)
47
+ if response.status_code == 503:
48
+ raise get_error_for_response(response)
49
+ if response.status_code == 422:
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[List["FinetunedEmbeddingModelMetadata"]]:
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() -> Response[List["FinetunedEmbeddingModelMetadata"]]:
65
+ """List Finetuned Embedding Models
66
+
67
+ Raises:
68
+ errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
69
+ httpx.TimeoutException: If the request takes longer than Client.timeout.
70
+
71
+ Returns:
72
+ Response[List['FinetunedEmbeddingModelMetadata']]
73
+ """
74
+
75
+ kwargs = _get_kwargs()
76
+
77
+ with _client_context() as client:
78
+ response = client.get_httpx_client().request(
79
+ **kwargs,
80
+ )
81
+
82
+ return _build_response(response=response)
83
+
84
+
85
+ def sync() -> List["FinetunedEmbeddingModelMetadata"]:
86
+ """List Finetuned Embedding 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
+ List['FinetunedEmbeddingModelMetadata']
94
+ """
95
+
96
+ return sync_detailed().parsed
97
+
98
+
99
+ async def asyncio_detailed() -> Response[List["FinetunedEmbeddingModelMetadata"]]:
100
+ """List Finetuned Embedding Models
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
+ Response[List['FinetunedEmbeddingModelMetadata']]
108
+ """
109
+
110
+ kwargs = _get_kwargs()
111
+
112
+ with _client_context() as client:
113
+ response = await client.get_async_httpx_client().request(**kwargs)
114
+
115
+ return _build_response(response=response)
116
+
117
+
118
+ async def asyncio() -> List["FinetunedEmbeddingModelMetadata"]:
119
+ """List Finetuned Embedding Models
120
+
121
+ Raises:
122
+ errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
123
+ httpx.TimeoutException: If the request takes longer than Client.timeout.
124
+
125
+ Returns:
126
+ List['FinetunedEmbeddingModelMetadata']
127
+ """
128
+
129
+ return (await asyncio_detailed()).parsed
@@ -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.clone_labeled_memoryset_request import CloneLabeledMemorysetRequest
21
+ from ...models.labeled_memoryset_metadata import LabeledMemorysetMetadata
22
+ from ...types import Response
23
+
24
+
25
+ def _get_kwargs(
26
+ name_or_id: str,
27
+ *,
28
+ body: CloneLabeledMemorysetRequest,
29
+ ) -> dict[str, Any]:
30
+ headers: Dict[str, Any] = {}
31
+
32
+ _kwargs: dict[str, Any] = {
33
+ "method": "post",
34
+ "url": f"/memoryset/{name_or_id}/clone",
35
+ }
36
+
37
+ _body = body.to_dict()
38
+
39
+ _kwargs["json"] = _body
40
+ headers["Content-Type"] = "application/json"
41
+
42
+ _kwargs["headers"] = headers
43
+ return _kwargs
44
+
45
+
46
+ def _parse_response(*, response: httpx.Response) -> LabeledMemorysetMetadata:
47
+ if response.status_code == 201:
48
+ response_201 = LabeledMemorysetMetadata.from_dict(response.json())
49
+
50
+ return response_201
51
+ if response.status_code == 401:
52
+ raise get_error_for_response(response)
53
+ if response.status_code == 500:
54
+ raise get_error_for_response(response)
55
+ if response.status_code == 503:
56
+ raise get_error_for_response(response)
57
+ if response.status_code == 422:
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[LabeledMemorysetMetadata]:
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: CloneLabeledMemorysetRequest,
76
+ ) -> Response[LabeledMemorysetMetadata]:
77
+ """Clone Memoryset
78
+
79
+ Args:
80
+ name_or_id (str):
81
+ body (CloneLabeledMemorysetRequest):
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[LabeledMemorysetMetadata]
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: CloneLabeledMemorysetRequest,
108
+ ) -> LabeledMemorysetMetadata:
109
+ """Clone Memoryset
110
+
111
+ Args:
112
+ name_or_id (str):
113
+ body (CloneLabeledMemorysetRequest):
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
+ LabeledMemorysetMetadata
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: CloneLabeledMemorysetRequest,
133
+ ) -> Response[LabeledMemorysetMetadata]:
134
+ """Clone Memoryset
135
+
136
+ Args:
137
+ name_or_id (str):
138
+ body (CloneLabeledMemorysetRequest):
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[LabeledMemorysetMetadata]
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: CloneLabeledMemorysetRequest,
163
+ ) -> LabeledMemorysetMetadata:
164
+ """Clone Memoryset
165
+
166
+ Args:
167
+ name_or_id (str):
168
+ body (CloneLabeledMemorysetRequest):
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
+ LabeledMemorysetMetadata
176
+ """
177
+
178
+ return (
179
+ await asyncio_detailed(
180
+ name_or_id=name_or_id,
181
+ body=body,
182
+ )
183
+ ).parsed
@@ -0,0 +1,185 @@
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.memoryset_analysis_request import MemorysetAnalysisRequest
21
+ from ...models.memoryset_analysis_response import MemorysetAnalysisResponse
22
+ from ...types import Response
23
+
24
+
25
+ def _get_kwargs(
26
+ name_or_id: str,
27
+ *,
28
+ body: MemorysetAnalysisRequest,
29
+ ) -> dict[str, Any]:
30
+ headers: Dict[str, Any] = {}
31
+
32
+ _kwargs: dict[str, Any] = {
33
+ "method": "post",
34
+ "url": f"/memoryset/{name_or_id}/analysis",
35
+ }
36
+
37
+ _body = body.to_dict()
38
+
39
+ _kwargs["json"] = _body
40
+ headers["Content-Type"] = "application/json"
41
+
42
+ _kwargs["headers"] = headers
43
+ return _kwargs
44
+
45
+
46
+ def _parse_response(*, response: httpx.Response) -> MemorysetAnalysisResponse:
47
+ if response.status_code == 201:
48
+ response_201 = MemorysetAnalysisResponse.from_dict(response.json())
49
+
50
+ return response_201
51
+ if response.status_code == 401:
52
+ raise get_error_for_response(response)
53
+ if response.status_code == 500:
54
+ raise get_error_for_response(response)
55
+ if response.status_code == 503:
56
+ raise get_error_for_response(response)
57
+ if response.status_code == 422:
58
+ raise get_error_for_response(response)
59
+ if response.status_code == 409:
60
+ raise get_error_for_response(response)
61
+ else:
62
+ raise get_error_for_response(response)
63
+
64
+
65
+ def _build_response(*, response: httpx.Response) -> Response[MemorysetAnalysisResponse]:
66
+ return Response(
67
+ status_code=HTTPStatus(response.status_code),
68
+ content=response.content,
69
+ headers=response.headers,
70
+ parsed=_parse_response(response=response),
71
+ )
72
+
73
+
74
+ def sync_detailed(
75
+ name_or_id: str,
76
+ *,
77
+ body: MemorysetAnalysisRequest,
78
+ ) -> Response[MemorysetAnalysisResponse]:
79
+ """Create Analysis
80
+
81
+ Args:
82
+ name_or_id (str):
83
+ body (MemorysetAnalysisRequest):
84
+
85
+ Raises:
86
+ errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
87
+ httpx.TimeoutException: If the request takes longer than Client.timeout.
88
+
89
+ Returns:
90
+ Response[MemorysetAnalysisResponse]
91
+ """
92
+
93
+ kwargs = _get_kwargs(
94
+ name_or_id=name_or_id,
95
+ body=body,
96
+ )
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(
107
+ name_or_id: str,
108
+ *,
109
+ body: MemorysetAnalysisRequest,
110
+ ) -> MemorysetAnalysisResponse:
111
+ """Create Analysis
112
+
113
+ Args:
114
+ name_or_id (str):
115
+ body (MemorysetAnalysisRequest):
116
+
117
+ Raises:
118
+ errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
119
+ httpx.TimeoutException: If the request takes longer than Client.timeout.
120
+
121
+ Returns:
122
+ MemorysetAnalysisResponse
123
+ """
124
+
125
+ return sync_detailed(
126
+ name_or_id=name_or_id,
127
+ body=body,
128
+ ).parsed
129
+
130
+
131
+ async def asyncio_detailed(
132
+ name_or_id: str,
133
+ *,
134
+ body: MemorysetAnalysisRequest,
135
+ ) -> Response[MemorysetAnalysisResponse]:
136
+ """Create Analysis
137
+
138
+ Args:
139
+ name_or_id (str):
140
+ body (MemorysetAnalysisRequest):
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
+ Response[MemorysetAnalysisResponse]
148
+ """
149
+
150
+ kwargs = _get_kwargs(
151
+ name_or_id=name_or_id,
152
+ body=body,
153
+ )
154
+
155
+ with _client_context() as client:
156
+ response = await client.get_async_httpx_client().request(**kwargs)
157
+
158
+ return _build_response(response=response)
159
+
160
+
161
+ async def asyncio(
162
+ name_or_id: str,
163
+ *,
164
+ body: MemorysetAnalysisRequest,
165
+ ) -> MemorysetAnalysisResponse:
166
+ """Create Analysis
167
+
168
+ Args:
169
+ name_or_id (str):
170
+ body (MemorysetAnalysisRequest):
171
+
172
+ Raises:
173
+ errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
174
+ httpx.TimeoutException: If the request takes longer than Client.timeout.
175
+
176
+ Returns:
177
+ MemorysetAnalysisResponse
178
+ """
179
+
180
+ return (
181
+ await asyncio_detailed(
182
+ name_or_id=name_or_id,
183
+ body=body,
184
+ )
185
+ ).parsed
@@ -0,0 +1,170 @@
1
+ """
2
+ This file is generated by the openapi-python-client tool via the generate_api_client.py script
3
+
4
+ It is a customized template from the openapi-python-client tool's default template:
5
+ https://github.com/openapi-generators/openapi-python-client/blob/861ef5622f10fc96d240dc9becb0edf94e61446c/openapi_python_client/templates/endpoint_module.py.jinja
6
+
7
+ The main changes are:
8
+ - Update the API call responses to either return the successful response type or raise an error by:
9
+ - Updating the _parse_response function to raise an error if the response status code is not in the 2xx range
10
+ - Inject a client into every method via a context manager
11
+ """
12
+
13
+ from http import HTTPStatus
14
+ from typing import Any, Dict
15
+
16
+ import httpx
17
+
18
+ from ...client import _client_context
19
+ from ...errors import get_error_for_response
20
+ from ...models.create_labeled_memoryset_request import CreateLabeledMemorysetRequest
21
+ from ...models.labeled_memoryset_metadata import LabeledMemorysetMetadata
22
+ from ...types import Response
23
+
24
+
25
+ def _get_kwargs(
26
+ *,
27
+ body: CreateLabeledMemorysetRequest,
28
+ ) -> dict[str, Any]:
29
+ headers: Dict[str, Any] = {}
30
+
31
+ _kwargs: dict[str, Any] = {
32
+ "method": "post",
33
+ "url": "/memoryset/",
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) -> LabeledMemorysetMetadata:
46
+ if response.status_code == 201:
47
+ response_201 = LabeledMemorysetMetadata.from_dict(response.json())
48
+
49
+ return response_201
50
+ if response.status_code == 401:
51
+ raise get_error_for_response(response)
52
+ if response.status_code == 500:
53
+ raise get_error_for_response(response)
54
+ if response.status_code == 503:
55
+ raise get_error_for_response(response)
56
+ if response.status_code == 422:
57
+ raise get_error_for_response(response)
58
+ else:
59
+ raise get_error_for_response(response)
60
+
61
+
62
+ def _build_response(*, response: httpx.Response) -> Response[LabeledMemorysetMetadata]:
63
+ return Response(
64
+ status_code=HTTPStatus(response.status_code),
65
+ content=response.content,
66
+ headers=response.headers,
67
+ parsed=_parse_response(response=response),
68
+ )
69
+
70
+
71
+ def sync_detailed(
72
+ *,
73
+ body: CreateLabeledMemorysetRequest,
74
+ ) -> Response[LabeledMemorysetMetadata]:
75
+ """Create Memoryset
76
+
77
+ Args:
78
+ body (CreateLabeledMemorysetRequest):
79
+
80
+ Raises:
81
+ errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
82
+ httpx.TimeoutException: If the request takes longer than Client.timeout.
83
+
84
+ Returns:
85
+ Response[LabeledMemorysetMetadata]
86
+ """
87
+
88
+ kwargs = _get_kwargs(
89
+ body=body,
90
+ )
91
+
92
+ with _client_context() as client:
93
+ response = client.get_httpx_client().request(
94
+ **kwargs,
95
+ )
96
+
97
+ return _build_response(response=response)
98
+
99
+
100
+ def sync(
101
+ *,
102
+ body: CreateLabeledMemorysetRequest,
103
+ ) -> LabeledMemorysetMetadata:
104
+ """Create Memoryset
105
+
106
+ Args:
107
+ body (CreateLabeledMemorysetRequest):
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
+ LabeledMemorysetMetadata
115
+ """
116
+
117
+ return sync_detailed(
118
+ body=body,
119
+ ).parsed
120
+
121
+
122
+ async def asyncio_detailed(
123
+ *,
124
+ body: CreateLabeledMemorysetRequest,
125
+ ) -> Response[LabeledMemorysetMetadata]:
126
+ """Create Memoryset
127
+
128
+ Args:
129
+ body (CreateLabeledMemorysetRequest):
130
+
131
+ Raises:
132
+ errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
133
+ httpx.TimeoutException: If the request takes longer than Client.timeout.
134
+
135
+ Returns:
136
+ Response[LabeledMemorysetMetadata]
137
+ """
138
+
139
+ kwargs = _get_kwargs(
140
+ body=body,
141
+ )
142
+
143
+ with _client_context() as client:
144
+ response = await client.get_async_httpx_client().request(**kwargs)
145
+
146
+ return _build_response(response=response)
147
+
148
+
149
+ async def asyncio(
150
+ *,
151
+ body: CreateLabeledMemorysetRequest,
152
+ ) -> LabeledMemorysetMetadata:
153
+ """Create Memoryset
154
+
155
+ Args:
156
+ body (CreateLabeledMemorysetRequest):
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
+ LabeledMemorysetMetadata
164
+ """
165
+
166
+ return (
167
+ await asyncio_detailed(
168
+ body=body,
169
+ )
170
+ ).parsed