orca-sdk 0.1.1__py3-none-any.whl → 0.1.3__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 (186) hide show
  1. orca_sdk/__init__.py +10 -4
  2. orca_sdk/_shared/__init__.py +10 -0
  3. orca_sdk/_shared/metrics.py +393 -0
  4. orca_sdk/_shared/metrics_test.py +273 -0
  5. orca_sdk/_utils/analysis_ui.py +12 -10
  6. orca_sdk/_utils/analysis_ui_style.css +0 -3
  7. orca_sdk/_utils/auth.py +31 -29
  8. orca_sdk/_utils/data_parsing.py +28 -2
  9. orca_sdk/_utils/data_parsing_test.py +15 -15
  10. orca_sdk/_utils/pagination.py +126 -0
  11. orca_sdk/_utils/pagination_test.py +132 -0
  12. orca_sdk/_utils/prediction_result_ui.py +67 -21
  13. orca_sdk/_utils/tqdm_file_reader.py +12 -0
  14. orca_sdk/_utils/value_parser.py +45 -0
  15. orca_sdk/_utils/value_parser_test.py +39 -0
  16. orca_sdk/async_client.py +3795 -0
  17. orca_sdk/classification_model.py +601 -129
  18. orca_sdk/classification_model_test.py +415 -117
  19. orca_sdk/client.py +3787 -0
  20. orca_sdk/conftest.py +184 -38
  21. orca_sdk/credentials.py +162 -20
  22. orca_sdk/credentials_test.py +100 -16
  23. orca_sdk/datasource.py +268 -68
  24. orca_sdk/datasource_test.py +266 -18
  25. orca_sdk/embedding_model.py +434 -82
  26. orca_sdk/embedding_model_test.py +66 -33
  27. orca_sdk/job.py +343 -0
  28. orca_sdk/job_test.py +108 -0
  29. orca_sdk/memoryset.py +1690 -324
  30. orca_sdk/memoryset_test.py +456 -119
  31. orca_sdk/regression_model.py +694 -0
  32. orca_sdk/regression_model_test.py +378 -0
  33. orca_sdk/telemetry.py +460 -143
  34. orca_sdk/telemetry_test.py +43 -24
  35. {orca_sdk-0.1.1.dist-info → orca_sdk-0.1.3.dist-info}/METADATA +34 -16
  36. orca_sdk-0.1.3.dist-info/RECORD +41 -0
  37. {orca_sdk-0.1.1.dist-info → orca_sdk-0.1.3.dist-info}/WHEEL +1 -1
  38. orca_sdk/_generated_api_client/__init__.py +0 -3
  39. orca_sdk/_generated_api_client/api/__init__.py +0 -193
  40. orca_sdk/_generated_api_client/api/auth/__init__.py +0 -0
  41. orca_sdk/_generated_api_client/api/auth/check_authentication_auth_get.py +0 -128
  42. orca_sdk/_generated_api_client/api/auth/create_api_key_auth_api_key_post.py +0 -170
  43. orca_sdk/_generated_api_client/api/auth/delete_api_key_auth_api_key_name_or_id_delete.py +0 -156
  44. orca_sdk/_generated_api_client/api/auth/delete_org_auth_org_delete.py +0 -130
  45. orca_sdk/_generated_api_client/api/auth/list_api_keys_auth_api_key_get.py +0 -127
  46. orca_sdk/_generated_api_client/api/classification_model/__init__.py +0 -0
  47. orca_sdk/_generated_api_client/api/classification_model/create_evaluation_classification_model_model_name_or_id_evaluation_post.py +0 -183
  48. orca_sdk/_generated_api_client/api/classification_model/create_model_classification_model_post.py +0 -170
  49. orca_sdk/_generated_api_client/api/classification_model/delete_evaluation_classification_model_model_name_or_id_evaluation_task_id_delete.py +0 -168
  50. orca_sdk/_generated_api_client/api/classification_model/delete_model_classification_model_name_or_id_delete.py +0 -154
  51. orca_sdk/_generated_api_client/api/classification_model/get_evaluation_classification_model_model_name_or_id_evaluation_task_id_get.py +0 -170
  52. orca_sdk/_generated_api_client/api/classification_model/get_model_classification_model_name_or_id_get.py +0 -156
  53. orca_sdk/_generated_api_client/api/classification_model/list_evaluations_classification_model_model_name_or_id_evaluation_get.py +0 -161
  54. orca_sdk/_generated_api_client/api/classification_model/list_models_classification_model_get.py +0 -127
  55. orca_sdk/_generated_api_client/api/classification_model/predict_gpu_classification_model_name_or_id_prediction_post.py +0 -190
  56. orca_sdk/_generated_api_client/api/datasource/__init__.py +0 -0
  57. orca_sdk/_generated_api_client/api/datasource/create_datasource_datasource_post.py +0 -167
  58. orca_sdk/_generated_api_client/api/datasource/delete_datasource_datasource_name_or_id_delete.py +0 -156
  59. orca_sdk/_generated_api_client/api/datasource/get_datasource_datasource_name_or_id_get.py +0 -156
  60. orca_sdk/_generated_api_client/api/datasource/list_datasources_datasource_get.py +0 -127
  61. orca_sdk/_generated_api_client/api/default/__init__.py +0 -0
  62. orca_sdk/_generated_api_client/api/default/healthcheck_get.py +0 -118
  63. orca_sdk/_generated_api_client/api/default/healthcheck_gpu_get.py +0 -118
  64. orca_sdk/_generated_api_client/api/finetuned_embedding_model/__init__.py +0 -0
  65. orca_sdk/_generated_api_client/api/finetuned_embedding_model/create_finetuned_embedding_model_finetuned_embedding_model_post.py +0 -168
  66. orca_sdk/_generated_api_client/api/finetuned_embedding_model/delete_finetuned_embedding_model_finetuned_embedding_model_name_or_id_delete.py +0 -156
  67. orca_sdk/_generated_api_client/api/finetuned_embedding_model/embed_with_finetuned_model_gpu_finetuned_embedding_model_name_or_id_embedding_post.py +0 -189
  68. orca_sdk/_generated_api_client/api/finetuned_embedding_model/get_finetuned_embedding_model_finetuned_embedding_model_name_or_id_get.py +0 -156
  69. orca_sdk/_generated_api_client/api/finetuned_embedding_model/list_finetuned_embedding_models_finetuned_embedding_model_get.py +0 -127
  70. orca_sdk/_generated_api_client/api/memoryset/__init__.py +0 -0
  71. orca_sdk/_generated_api_client/api/memoryset/clone_memoryset_memoryset_name_or_id_clone_post.py +0 -181
  72. orca_sdk/_generated_api_client/api/memoryset/create_analysis_memoryset_name_or_id_analysis_post.py +0 -183
  73. orca_sdk/_generated_api_client/api/memoryset/create_memoryset_memoryset_post.py +0 -168
  74. orca_sdk/_generated_api_client/api/memoryset/delete_memories_memoryset_name_or_id_memories_delete_post.py +0 -181
  75. orca_sdk/_generated_api_client/api/memoryset/delete_memory_memoryset_name_or_id_memory_memory_id_delete.py +0 -167
  76. orca_sdk/_generated_api_client/api/memoryset/delete_memoryset_memoryset_name_or_id_delete.py +0 -156
  77. orca_sdk/_generated_api_client/api/memoryset/get_analysis_memoryset_name_or_id_analysis_analysis_task_id_get.py +0 -169
  78. orca_sdk/_generated_api_client/api/memoryset/get_memories_memoryset_name_or_id_memories_get_post.py +0 -188
  79. orca_sdk/_generated_api_client/api/memoryset/get_memory_memoryset_name_or_id_memory_memory_id_get.py +0 -169
  80. orca_sdk/_generated_api_client/api/memoryset/get_memoryset_memoryset_name_or_id_get.py +0 -156
  81. orca_sdk/_generated_api_client/api/memoryset/insert_memories_gpu_memoryset_name_or_id_memory_post.py +0 -184
  82. orca_sdk/_generated_api_client/api/memoryset/list_analyses_memoryset_name_or_id_analysis_get.py +0 -260
  83. orca_sdk/_generated_api_client/api/memoryset/list_memorysets_memoryset_get.py +0 -127
  84. orca_sdk/_generated_api_client/api/memoryset/memoryset_lookup_gpu_memoryset_name_or_id_lookup_post.py +0 -193
  85. orca_sdk/_generated_api_client/api/memoryset/query_memoryset_memoryset_name_or_id_memories_post.py +0 -188
  86. orca_sdk/_generated_api_client/api/memoryset/update_memories_gpu_memoryset_name_or_id_memories_patch.py +0 -191
  87. orca_sdk/_generated_api_client/api/memoryset/update_memory_gpu_memoryset_name_or_id_memory_patch.py +0 -187
  88. orca_sdk/_generated_api_client/api/pretrained_embedding_model/__init__.py +0 -0
  89. orca_sdk/_generated_api_client/api/pretrained_embedding_model/embed_with_pretrained_model_gpu_pretrained_embedding_model_model_name_embedding_post.py +0 -188
  90. orca_sdk/_generated_api_client/api/pretrained_embedding_model/get_pretrained_embedding_model_pretrained_embedding_model_model_name_get.py +0 -157
  91. orca_sdk/_generated_api_client/api/pretrained_embedding_model/list_pretrained_embedding_models_pretrained_embedding_model_get.py +0 -127
  92. orca_sdk/_generated_api_client/api/task/__init__.py +0 -0
  93. orca_sdk/_generated_api_client/api/task/abort_task_task_task_id_abort_delete.py +0 -154
  94. orca_sdk/_generated_api_client/api/task/get_task_status_task_task_id_status_get.py +0 -156
  95. orca_sdk/_generated_api_client/api/task/list_tasks_task_get.py +0 -243
  96. orca_sdk/_generated_api_client/api/telemetry/__init__.py +0 -0
  97. orca_sdk/_generated_api_client/api/telemetry/drop_feedback_category_with_data_telemetry_feedback_category_name_or_id_delete.py +0 -162
  98. orca_sdk/_generated_api_client/api/telemetry/get_feedback_category_telemetry_feedback_category_name_or_id_get.py +0 -156
  99. orca_sdk/_generated_api_client/api/telemetry/get_prediction_telemetry_prediction_prediction_id_get.py +0 -157
  100. orca_sdk/_generated_api_client/api/telemetry/list_feedback_categories_telemetry_feedback_category_get.py +0 -127
  101. orca_sdk/_generated_api_client/api/telemetry/list_predictions_telemetry_prediction_post.py +0 -175
  102. orca_sdk/_generated_api_client/api/telemetry/record_prediction_feedback_telemetry_prediction_feedback_put.py +0 -171
  103. orca_sdk/_generated_api_client/api/telemetry/update_prediction_telemetry_prediction_prediction_id_patch.py +0 -181
  104. orca_sdk/_generated_api_client/client.py +0 -216
  105. orca_sdk/_generated_api_client/errors.py +0 -38
  106. orca_sdk/_generated_api_client/models/__init__.py +0 -159
  107. orca_sdk/_generated_api_client/models/analyze_neighbor_labels_result.py +0 -84
  108. orca_sdk/_generated_api_client/models/api_key_metadata.py +0 -118
  109. orca_sdk/_generated_api_client/models/base_model.py +0 -55
  110. orca_sdk/_generated_api_client/models/body_create_datasource_datasource_post.py +0 -176
  111. orca_sdk/_generated_api_client/models/classification_evaluation_result.py +0 -114
  112. orca_sdk/_generated_api_client/models/clone_labeled_memoryset_request.py +0 -150
  113. orca_sdk/_generated_api_client/models/column_info.py +0 -114
  114. orca_sdk/_generated_api_client/models/column_type.py +0 -14
  115. orca_sdk/_generated_api_client/models/conflict_error_response.py +0 -80
  116. orca_sdk/_generated_api_client/models/create_api_key_request.py +0 -99
  117. orca_sdk/_generated_api_client/models/create_api_key_response.py +0 -126
  118. orca_sdk/_generated_api_client/models/create_labeled_memoryset_request.py +0 -259
  119. orca_sdk/_generated_api_client/models/create_rac_model_request.py +0 -209
  120. orca_sdk/_generated_api_client/models/datasource_metadata.py +0 -142
  121. orca_sdk/_generated_api_client/models/delete_memories_request.py +0 -70
  122. orca_sdk/_generated_api_client/models/embed_request.py +0 -127
  123. orca_sdk/_generated_api_client/models/embedding_finetuning_method.py +0 -9
  124. orca_sdk/_generated_api_client/models/evaluation_request.py +0 -180
  125. orca_sdk/_generated_api_client/models/evaluation_response.py +0 -140
  126. orca_sdk/_generated_api_client/models/feedback_type.py +0 -9
  127. orca_sdk/_generated_api_client/models/field_validation_error.py +0 -103
  128. orca_sdk/_generated_api_client/models/filter_item.py +0 -231
  129. orca_sdk/_generated_api_client/models/filter_item_field_type_0_item.py +0 -15
  130. orca_sdk/_generated_api_client/models/filter_item_field_type_2_item_type_1.py +0 -16
  131. orca_sdk/_generated_api_client/models/filter_item_op.py +0 -16
  132. orca_sdk/_generated_api_client/models/find_duplicates_analysis_result.py +0 -70
  133. orca_sdk/_generated_api_client/models/finetune_embedding_model_request.py +0 -259
  134. orca_sdk/_generated_api_client/models/finetune_embedding_model_request_training_args.py +0 -66
  135. orca_sdk/_generated_api_client/models/finetuned_embedding_model_metadata.py +0 -166
  136. orca_sdk/_generated_api_client/models/get_memories_request.py +0 -70
  137. orca_sdk/_generated_api_client/models/internal_server_error_response.py +0 -80
  138. orca_sdk/_generated_api_client/models/label_class_metrics.py +0 -108
  139. orca_sdk/_generated_api_client/models/label_prediction_memory_lookup.py +0 -274
  140. orca_sdk/_generated_api_client/models/label_prediction_memory_lookup_metadata.py +0 -68
  141. orca_sdk/_generated_api_client/models/label_prediction_result.py +0 -101
  142. orca_sdk/_generated_api_client/models/label_prediction_with_memories_and_feedback.py +0 -232
  143. orca_sdk/_generated_api_client/models/labeled_memory.py +0 -197
  144. orca_sdk/_generated_api_client/models/labeled_memory_insert.py +0 -108
  145. orca_sdk/_generated_api_client/models/labeled_memory_insert_metadata.py +0 -68
  146. orca_sdk/_generated_api_client/models/labeled_memory_lookup.py +0 -258
  147. orca_sdk/_generated_api_client/models/labeled_memory_lookup_metadata.py +0 -68
  148. orca_sdk/_generated_api_client/models/labeled_memory_metadata.py +0 -68
  149. orca_sdk/_generated_api_client/models/labeled_memory_metrics.py +0 -277
  150. orca_sdk/_generated_api_client/models/labeled_memory_update.py +0 -171
  151. orca_sdk/_generated_api_client/models/labeled_memory_update_metadata_type_0.py +0 -68
  152. orca_sdk/_generated_api_client/models/labeled_memoryset_metadata.py +0 -195
  153. orca_sdk/_generated_api_client/models/list_analyses_memoryset_name_or_id_analysis_get_type_type_0.py +0 -9
  154. orca_sdk/_generated_api_client/models/list_memories_request.py +0 -104
  155. orca_sdk/_generated_api_client/models/list_predictions_request.py +0 -234
  156. orca_sdk/_generated_api_client/models/list_predictions_request_sort_item_item_type_0.py +0 -9
  157. orca_sdk/_generated_api_client/models/list_predictions_request_sort_item_item_type_1.py +0 -9
  158. orca_sdk/_generated_api_client/models/lookup_request.py +0 -81
  159. orca_sdk/_generated_api_client/models/memoryset_analysis_request.py +0 -83
  160. orca_sdk/_generated_api_client/models/memoryset_analysis_request_type.py +0 -9
  161. orca_sdk/_generated_api_client/models/memoryset_analysis_response.py +0 -180
  162. orca_sdk/_generated_api_client/models/memoryset_analysis_response_config.py +0 -66
  163. orca_sdk/_generated_api_client/models/memoryset_analysis_response_type.py +0 -9
  164. orca_sdk/_generated_api_client/models/not_found_error_response.py +0 -100
  165. orca_sdk/_generated_api_client/models/not_found_error_response_resource_type_0.py +0 -20
  166. orca_sdk/_generated_api_client/models/prediction_feedback.py +0 -157
  167. orca_sdk/_generated_api_client/models/prediction_feedback_category.py +0 -115
  168. orca_sdk/_generated_api_client/models/prediction_feedback_request.py +0 -122
  169. orca_sdk/_generated_api_client/models/prediction_feedback_result.py +0 -102
  170. orca_sdk/_generated_api_client/models/prediction_request.py +0 -169
  171. orca_sdk/_generated_api_client/models/pretrained_embedding_model_metadata.py +0 -97
  172. orca_sdk/_generated_api_client/models/pretrained_embedding_model_name.py +0 -11
  173. orca_sdk/_generated_api_client/models/rac_head_type.py +0 -11
  174. orca_sdk/_generated_api_client/models/rac_model_metadata.py +0 -191
  175. orca_sdk/_generated_api_client/models/service_unavailable_error_response.py +0 -80
  176. orca_sdk/_generated_api_client/models/task.py +0 -198
  177. orca_sdk/_generated_api_client/models/task_status.py +0 -14
  178. orca_sdk/_generated_api_client/models/task_status_info.py +0 -133
  179. orca_sdk/_generated_api_client/models/unauthenticated_error_response.py +0 -72
  180. orca_sdk/_generated_api_client/models/unauthorized_error_response.py +0 -80
  181. orca_sdk/_generated_api_client/models/unprocessable_input_error_response.py +0 -94
  182. orca_sdk/_generated_api_client/models/update_prediction_request.py +0 -93
  183. orca_sdk/_generated_api_client/py.typed +0 -1
  184. orca_sdk/_generated_api_client/types.py +0 -56
  185. orca_sdk/_utils/task.py +0 -73
  186. orca_sdk-0.1.1.dist-info/RECORD +0 -175
@@ -1,127 +0,0 @@
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.labeled_memoryset_metadata import LabeledMemorysetMetadata
21
- from ...types import Response
22
-
23
-
24
- def _get_kwargs() -> dict[str, Any]:
25
- _kwargs: dict[str, Any] = {
26
- "method": "get",
27
- "url": "/memoryset/",
28
- }
29
-
30
- return _kwargs
31
-
32
-
33
- def _parse_response(*, response: httpx.Response) -> List["LabeledMemorysetMetadata"]:
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 = LabeledMemorysetMetadata.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
-
52
-
53
- def _build_response(*, response: httpx.Response) -> Response[List["LabeledMemorysetMetadata"]]:
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() -> Response[List["LabeledMemorysetMetadata"]]:
63
- """List Memorysets
64
-
65
- Raises:
66
- errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
67
- httpx.TimeoutException: If the request takes longer than Client.timeout.
68
-
69
- Returns:
70
- Response[List['LabeledMemorysetMetadata']]
71
- """
72
-
73
- kwargs = _get_kwargs()
74
-
75
- with _client_context() as client:
76
- response = client.get_httpx_client().request(
77
- **kwargs,
78
- )
79
-
80
- return _build_response(response=response)
81
-
82
-
83
- def sync() -> List["LabeledMemorysetMetadata"]:
84
- """List Memorysets
85
-
86
- Raises:
87
- errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
88
- httpx.TimeoutException: If the request takes longer than Client.timeout.
89
-
90
- Returns:
91
- List['LabeledMemorysetMetadata']
92
- """
93
-
94
- return sync_detailed().parsed
95
-
96
-
97
- async def asyncio_detailed() -> Response[List["LabeledMemorysetMetadata"]]:
98
- """List Memorysets
99
-
100
- Raises:
101
- errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
102
- httpx.TimeoutException: If the request takes longer than Client.timeout.
103
-
104
- Returns:
105
- Response[List['LabeledMemorysetMetadata']]
106
- """
107
-
108
- kwargs = _get_kwargs()
109
-
110
- with _client_context() as client:
111
- response = await client.get_async_httpx_client().request(**kwargs)
112
-
113
- return _build_response(response=response)
114
-
115
-
116
- async def asyncio() -> List["LabeledMemorysetMetadata"]:
117
- """List Memorysets
118
-
119
- Raises:
120
- errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
121
- httpx.TimeoutException: If the request takes longer than Client.timeout.
122
-
123
- Returns:
124
- List['LabeledMemorysetMetadata']
125
- """
126
-
127
- return (await asyncio_detailed()).parsed
@@ -1,193 +0,0 @@
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, List
15
-
16
- import httpx
17
-
18
- from ...client import _client_context
19
- from ...errors import get_error_for_response
20
- from ...models.labeled_memory_lookup import LabeledMemoryLookup
21
- from ...models.lookup_request import LookupRequest
22
- from ...types import Response
23
-
24
-
25
- def _get_kwargs(
26
- name_or_id: str,
27
- *,
28
- body: LookupRequest,
29
- ) -> dict[str, Any]:
30
- headers: Dict[str, Any] = {}
31
-
32
- _kwargs: dict[str, Any] = {
33
- "method": "post",
34
- "url": f"/gpu/memoryset/{name_or_id}/lookup",
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) -> List[List["LabeledMemoryLookup"]]:
47
- if response.status_code == 200:
48
- response_200 = []
49
- _response_200 = response.json()
50
- for response_200_item_data in _response_200:
51
- response_200_item = []
52
- _response_200_item = response_200_item_data
53
- for response_200_item_item_data in _response_200_item:
54
- response_200_item_item = LabeledMemoryLookup.from_dict(response_200_item_item_data)
55
-
56
- response_200_item.append(response_200_item_item)
57
-
58
- response_200.append(response_200_item)
59
-
60
- return response_200
61
- if response.status_code == 401:
62
- raise get_error_for_response(response)
63
- if response.status_code == 500:
64
- raise get_error_for_response(response)
65
- if response.status_code == 503:
66
- raise get_error_for_response(response)
67
- if response.status_code == 422:
68
- raise get_error_for_response(response)
69
- if response.status_code == 404:
70
- raise get_error_for_response(response)
71
-
72
-
73
- def _build_response(*, response: httpx.Response) -> Response[List[List["LabeledMemoryLookup"]]]:
74
- return Response(
75
- status_code=HTTPStatus(response.status_code),
76
- content=response.content,
77
- headers=response.headers,
78
- parsed=_parse_response(response=response),
79
- )
80
-
81
-
82
- def sync_detailed(
83
- name_or_id: str,
84
- *,
85
- body: LookupRequest,
86
- ) -> Response[List[List["LabeledMemoryLookup"]]]:
87
- """Memoryset Lookup
88
-
89
- Args:
90
- name_or_id (str):
91
- body (LookupRequest):
92
-
93
- Raises:
94
- errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
95
- httpx.TimeoutException: If the request takes longer than Client.timeout.
96
-
97
- Returns:
98
- Response[List[List['LabeledMemoryLookup']]]
99
- """
100
-
101
- kwargs = _get_kwargs(
102
- name_or_id=name_or_id,
103
- body=body,
104
- )
105
-
106
- with _client_context() as client:
107
- response = client.get_httpx_client().request(
108
- **kwargs,
109
- )
110
-
111
- return _build_response(response=response)
112
-
113
-
114
- def sync(
115
- name_or_id: str,
116
- *,
117
- body: LookupRequest,
118
- ) -> List[List["LabeledMemoryLookup"]]:
119
- """Memoryset Lookup
120
-
121
- Args:
122
- name_or_id (str):
123
- body (LookupRequest):
124
-
125
- Raises:
126
- errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
127
- httpx.TimeoutException: If the request takes longer than Client.timeout.
128
-
129
- Returns:
130
- List[List['LabeledMemoryLookup']]
131
- """
132
-
133
- return sync_detailed(
134
- name_or_id=name_or_id,
135
- body=body,
136
- ).parsed
137
-
138
-
139
- async def asyncio_detailed(
140
- name_or_id: str,
141
- *,
142
- body: LookupRequest,
143
- ) -> Response[List[List["LabeledMemoryLookup"]]]:
144
- """Memoryset Lookup
145
-
146
- Args:
147
- name_or_id (str):
148
- body (LookupRequest):
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
- Response[List[List['LabeledMemoryLookup']]]
156
- """
157
-
158
- kwargs = _get_kwargs(
159
- name_or_id=name_or_id,
160
- body=body,
161
- )
162
-
163
- with _client_context() as client:
164
- response = await client.get_async_httpx_client().request(**kwargs)
165
-
166
- return _build_response(response=response)
167
-
168
-
169
- async def asyncio(
170
- name_or_id: str,
171
- *,
172
- body: LookupRequest,
173
- ) -> List[List["LabeledMemoryLookup"]]:
174
- """Memoryset Lookup
175
-
176
- Args:
177
- name_or_id (str):
178
- body (LookupRequest):
179
-
180
- Raises:
181
- errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
182
- httpx.TimeoutException: If the request takes longer than Client.timeout.
183
-
184
- Returns:
185
- List[List['LabeledMemoryLookup']]
186
- """
187
-
188
- return (
189
- await asyncio_detailed(
190
- name_or_id=name_or_id,
191
- body=body,
192
- )
193
- ).parsed
@@ -1,188 +0,0 @@
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, List
15
-
16
- import httpx
17
-
18
- from ...client import _client_context
19
- from ...errors import get_error_for_response
20
- from ...models.labeled_memory import LabeledMemory
21
- from ...models.list_memories_request import ListMemoriesRequest
22
- from ...types import Response
23
-
24
-
25
- def _get_kwargs(
26
- name_or_id: str,
27
- *,
28
- body: ListMemoriesRequest,
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}/memories",
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) -> List["LabeledMemory"]:
47
- if response.status_code == 200:
48
- response_200 = []
49
- _response_200 = response.json()
50
- for response_200_item_data in _response_200:
51
- response_200_item = LabeledMemory.from_dict(response_200_item_data)
52
-
53
- response_200.append(response_200_item)
54
-
55
- return response_200
56
- if response.status_code == 401:
57
- raise get_error_for_response(response)
58
- if response.status_code == 500:
59
- raise get_error_for_response(response)
60
- if response.status_code == 503:
61
- raise get_error_for_response(response)
62
- if response.status_code == 422:
63
- raise get_error_for_response(response)
64
- if response.status_code == 404:
65
- raise get_error_for_response(response)
66
-
67
-
68
- def _build_response(*, response: httpx.Response) -> Response[List["LabeledMemory"]]:
69
- return Response(
70
- status_code=HTTPStatus(response.status_code),
71
- content=response.content,
72
- headers=response.headers,
73
- parsed=_parse_response(response=response),
74
- )
75
-
76
-
77
- def sync_detailed(
78
- name_or_id: str,
79
- *,
80
- body: ListMemoriesRequest,
81
- ) -> Response[List["LabeledMemory"]]:
82
- """Query Memoryset
83
-
84
- Args:
85
- name_or_id (str):
86
- body (ListMemoriesRequest):
87
-
88
- Raises:
89
- errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
90
- httpx.TimeoutException: If the request takes longer than Client.timeout.
91
-
92
- Returns:
93
- Response[List['LabeledMemory']]
94
- """
95
-
96
- kwargs = _get_kwargs(
97
- name_or_id=name_or_id,
98
- body=body,
99
- )
100
-
101
- with _client_context() as client:
102
- response = client.get_httpx_client().request(
103
- **kwargs,
104
- )
105
-
106
- return _build_response(response=response)
107
-
108
-
109
- def sync(
110
- name_or_id: str,
111
- *,
112
- body: ListMemoriesRequest,
113
- ) -> List["LabeledMemory"]:
114
- """Query Memoryset
115
-
116
- Args:
117
- name_or_id (str):
118
- body (ListMemoriesRequest):
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
- List['LabeledMemory']
126
- """
127
-
128
- return sync_detailed(
129
- name_or_id=name_or_id,
130
- body=body,
131
- ).parsed
132
-
133
-
134
- async def asyncio_detailed(
135
- name_or_id: str,
136
- *,
137
- body: ListMemoriesRequest,
138
- ) -> Response[List["LabeledMemory"]]:
139
- """Query Memoryset
140
-
141
- Args:
142
- name_or_id (str):
143
- body (ListMemoriesRequest):
144
-
145
- Raises:
146
- errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
147
- httpx.TimeoutException: If the request takes longer than Client.timeout.
148
-
149
- Returns:
150
- Response[List['LabeledMemory']]
151
- """
152
-
153
- kwargs = _get_kwargs(
154
- name_or_id=name_or_id,
155
- body=body,
156
- )
157
-
158
- with _client_context() as client:
159
- response = await client.get_async_httpx_client().request(**kwargs)
160
-
161
- return _build_response(response=response)
162
-
163
-
164
- async def asyncio(
165
- name_or_id: str,
166
- *,
167
- body: ListMemoriesRequest,
168
- ) -> List["LabeledMemory"]:
169
- """Query Memoryset
170
-
171
- Args:
172
- name_or_id (str):
173
- body (ListMemoriesRequest):
174
-
175
- Raises:
176
- errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
177
- httpx.TimeoutException: If the request takes longer than Client.timeout.
178
-
179
- Returns:
180
- List['LabeledMemory']
181
- """
182
-
183
- return (
184
- await asyncio_detailed(
185
- name_or_id=name_or_id,
186
- body=body,
187
- )
188
- ).parsed
@@ -1,191 +0,0 @@
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, List
15
-
16
- import httpx
17
-
18
- from ...client import _client_context
19
- from ...errors import get_error_for_response
20
- from ...models.labeled_memory import LabeledMemory
21
- from ...models.labeled_memory_update import LabeledMemoryUpdate
22
- from ...types import Response
23
-
24
-
25
- def _get_kwargs(
26
- name_or_id: str,
27
- *,
28
- body: List["LabeledMemoryUpdate"],
29
- ) -> dict[str, Any]:
30
- headers: Dict[str, Any] = {}
31
-
32
- _kwargs: dict[str, Any] = {
33
- "method": "patch",
34
- "url": f"/gpu/memoryset/{name_or_id}/memories",
35
- }
36
-
37
- _body = []
38
- for body_item_data in body:
39
- body_item = body_item_data.to_dict()
40
- _body.append(body_item)
41
-
42
- _kwargs["json"] = _body
43
- headers["Content-Type"] = "application/json"
44
-
45
- _kwargs["headers"] = headers
46
- return _kwargs
47
-
48
-
49
- def _parse_response(*, response: httpx.Response) -> List["LabeledMemory"]:
50
- if response.status_code == 200:
51
- response_200 = []
52
- _response_200 = response.json()
53
- for response_200_item_data in _response_200:
54
- response_200_item = LabeledMemory.from_dict(response_200_item_data)
55
-
56
- response_200.append(response_200_item)
57
-
58
- return response_200
59
- if response.status_code == 401:
60
- raise get_error_for_response(response)
61
- if response.status_code == 500:
62
- raise get_error_for_response(response)
63
- if response.status_code == 503:
64
- raise get_error_for_response(response)
65
- if response.status_code == 422:
66
- raise get_error_for_response(response)
67
- if response.status_code == 404:
68
- raise get_error_for_response(response)
69
-
70
-
71
- def _build_response(*, response: httpx.Response) -> Response[List["LabeledMemory"]]:
72
- return Response(
73
- status_code=HTTPStatus(response.status_code),
74
- content=response.content,
75
- headers=response.headers,
76
- parsed=_parse_response(response=response),
77
- )
78
-
79
-
80
- def sync_detailed(
81
- name_or_id: str,
82
- *,
83
- body: List["LabeledMemoryUpdate"],
84
- ) -> Response[List["LabeledMemory"]]:
85
- """Update Memories
86
-
87
- Args:
88
- name_or_id (str):
89
- body (List['LabeledMemoryUpdate']):
90
-
91
- Raises:
92
- errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
93
- httpx.TimeoutException: If the request takes longer than Client.timeout.
94
-
95
- Returns:
96
- Response[List['LabeledMemory']]
97
- """
98
-
99
- kwargs = _get_kwargs(
100
- name_or_id=name_or_id,
101
- body=body,
102
- )
103
-
104
- with _client_context() as client:
105
- response = client.get_httpx_client().request(
106
- **kwargs,
107
- )
108
-
109
- return _build_response(response=response)
110
-
111
-
112
- def sync(
113
- name_or_id: str,
114
- *,
115
- body: List["LabeledMemoryUpdate"],
116
- ) -> List["LabeledMemory"]:
117
- """Update Memories
118
-
119
- Args:
120
- name_or_id (str):
121
- body (List['LabeledMemoryUpdate']):
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
- List['LabeledMemory']
129
- """
130
-
131
- return sync_detailed(
132
- name_or_id=name_or_id,
133
- body=body,
134
- ).parsed
135
-
136
-
137
- async def asyncio_detailed(
138
- name_or_id: str,
139
- *,
140
- body: List["LabeledMemoryUpdate"],
141
- ) -> Response[List["LabeledMemory"]]:
142
- """Update Memories
143
-
144
- Args:
145
- name_or_id (str):
146
- body (List['LabeledMemoryUpdate']):
147
-
148
- Raises:
149
- errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
150
- httpx.TimeoutException: If the request takes longer than Client.timeout.
151
-
152
- Returns:
153
- Response[List['LabeledMemory']]
154
- """
155
-
156
- kwargs = _get_kwargs(
157
- name_or_id=name_or_id,
158
- body=body,
159
- )
160
-
161
- with _client_context() as client:
162
- response = await client.get_async_httpx_client().request(**kwargs)
163
-
164
- return _build_response(response=response)
165
-
166
-
167
- async def asyncio(
168
- name_or_id: str,
169
- *,
170
- body: List["LabeledMemoryUpdate"],
171
- ) -> List["LabeledMemory"]:
172
- """Update Memories
173
-
174
- Args:
175
- name_or_id (str):
176
- body (List['LabeledMemoryUpdate']):
177
-
178
- Raises:
179
- errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
180
- httpx.TimeoutException: If the request takes longer than Client.timeout.
181
-
182
- Returns:
183
- List['LabeledMemory']
184
- """
185
-
186
- return (
187
- await asyncio_detailed(
188
- name_or_id=name_or_id,
189
- body=body,
190
- )
191
- ).parsed