orca-sdk 0.1.1__py3-none-any.whl → 0.1.2__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 (185) 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 +27 -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/classification_model.py +439 -129
  17. orca_sdk/classification_model_test.py +334 -104
  18. orca_sdk/client.py +3747 -0
  19. orca_sdk/conftest.py +164 -19
  20. orca_sdk/credentials.py +120 -18
  21. orca_sdk/credentials_test.py +20 -0
  22. orca_sdk/datasource.py +259 -68
  23. orca_sdk/datasource_test.py +242 -0
  24. orca_sdk/embedding_model.py +425 -82
  25. orca_sdk/embedding_model_test.py +39 -13
  26. orca_sdk/job.py +337 -0
  27. orca_sdk/job_test.py +108 -0
  28. orca_sdk/memoryset.py +1341 -305
  29. orca_sdk/memoryset_test.py +350 -111
  30. orca_sdk/regression_model.py +684 -0
  31. orca_sdk/regression_model_test.py +369 -0
  32. orca_sdk/telemetry.py +449 -143
  33. orca_sdk/telemetry_test.py +43 -24
  34. {orca_sdk-0.1.1.dist-info → orca_sdk-0.1.2.dist-info}/METADATA +34 -16
  35. orca_sdk-0.1.2.dist-info/RECORD +40 -0
  36. {orca_sdk-0.1.1.dist-info → orca_sdk-0.1.2.dist-info}/WHEEL +1 -1
  37. orca_sdk/_generated_api_client/__init__.py +0 -3
  38. orca_sdk/_generated_api_client/api/__init__.py +0 -193
  39. orca_sdk/_generated_api_client/api/auth/__init__.py +0 -0
  40. orca_sdk/_generated_api_client/api/auth/check_authentication_auth_get.py +0 -128
  41. orca_sdk/_generated_api_client/api/auth/create_api_key_auth_api_key_post.py +0 -170
  42. orca_sdk/_generated_api_client/api/auth/delete_api_key_auth_api_key_name_or_id_delete.py +0 -156
  43. orca_sdk/_generated_api_client/api/auth/delete_org_auth_org_delete.py +0 -130
  44. orca_sdk/_generated_api_client/api/auth/list_api_keys_auth_api_key_get.py +0 -127
  45. orca_sdk/_generated_api_client/api/classification_model/__init__.py +0 -0
  46. orca_sdk/_generated_api_client/api/classification_model/create_evaluation_classification_model_model_name_or_id_evaluation_post.py +0 -183
  47. orca_sdk/_generated_api_client/api/classification_model/create_model_classification_model_post.py +0 -170
  48. orca_sdk/_generated_api_client/api/classification_model/delete_evaluation_classification_model_model_name_or_id_evaluation_task_id_delete.py +0 -168
  49. orca_sdk/_generated_api_client/api/classification_model/delete_model_classification_model_name_or_id_delete.py +0 -154
  50. orca_sdk/_generated_api_client/api/classification_model/get_evaluation_classification_model_model_name_or_id_evaluation_task_id_get.py +0 -170
  51. orca_sdk/_generated_api_client/api/classification_model/get_model_classification_model_name_or_id_get.py +0 -156
  52. orca_sdk/_generated_api_client/api/classification_model/list_evaluations_classification_model_model_name_or_id_evaluation_get.py +0 -161
  53. orca_sdk/_generated_api_client/api/classification_model/list_models_classification_model_get.py +0 -127
  54. orca_sdk/_generated_api_client/api/classification_model/predict_gpu_classification_model_name_or_id_prediction_post.py +0 -190
  55. orca_sdk/_generated_api_client/api/datasource/__init__.py +0 -0
  56. orca_sdk/_generated_api_client/api/datasource/create_datasource_datasource_post.py +0 -167
  57. orca_sdk/_generated_api_client/api/datasource/delete_datasource_datasource_name_or_id_delete.py +0 -156
  58. orca_sdk/_generated_api_client/api/datasource/get_datasource_datasource_name_or_id_get.py +0 -156
  59. orca_sdk/_generated_api_client/api/datasource/list_datasources_datasource_get.py +0 -127
  60. orca_sdk/_generated_api_client/api/default/__init__.py +0 -0
  61. orca_sdk/_generated_api_client/api/default/healthcheck_get.py +0 -118
  62. orca_sdk/_generated_api_client/api/default/healthcheck_gpu_get.py +0 -118
  63. orca_sdk/_generated_api_client/api/finetuned_embedding_model/__init__.py +0 -0
  64. orca_sdk/_generated_api_client/api/finetuned_embedding_model/create_finetuned_embedding_model_finetuned_embedding_model_post.py +0 -168
  65. orca_sdk/_generated_api_client/api/finetuned_embedding_model/delete_finetuned_embedding_model_finetuned_embedding_model_name_or_id_delete.py +0 -156
  66. 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
  67. orca_sdk/_generated_api_client/api/finetuned_embedding_model/get_finetuned_embedding_model_finetuned_embedding_model_name_or_id_get.py +0 -156
  68. orca_sdk/_generated_api_client/api/finetuned_embedding_model/list_finetuned_embedding_models_finetuned_embedding_model_get.py +0 -127
  69. orca_sdk/_generated_api_client/api/memoryset/__init__.py +0 -0
  70. orca_sdk/_generated_api_client/api/memoryset/clone_memoryset_memoryset_name_or_id_clone_post.py +0 -181
  71. orca_sdk/_generated_api_client/api/memoryset/create_analysis_memoryset_name_or_id_analysis_post.py +0 -183
  72. orca_sdk/_generated_api_client/api/memoryset/create_memoryset_memoryset_post.py +0 -168
  73. orca_sdk/_generated_api_client/api/memoryset/delete_memories_memoryset_name_or_id_memories_delete_post.py +0 -181
  74. orca_sdk/_generated_api_client/api/memoryset/delete_memory_memoryset_name_or_id_memory_memory_id_delete.py +0 -167
  75. orca_sdk/_generated_api_client/api/memoryset/delete_memoryset_memoryset_name_or_id_delete.py +0 -156
  76. orca_sdk/_generated_api_client/api/memoryset/get_analysis_memoryset_name_or_id_analysis_analysis_task_id_get.py +0 -169
  77. orca_sdk/_generated_api_client/api/memoryset/get_memories_memoryset_name_or_id_memories_get_post.py +0 -188
  78. orca_sdk/_generated_api_client/api/memoryset/get_memory_memoryset_name_or_id_memory_memory_id_get.py +0 -169
  79. orca_sdk/_generated_api_client/api/memoryset/get_memoryset_memoryset_name_or_id_get.py +0 -156
  80. orca_sdk/_generated_api_client/api/memoryset/insert_memories_gpu_memoryset_name_or_id_memory_post.py +0 -184
  81. orca_sdk/_generated_api_client/api/memoryset/list_analyses_memoryset_name_or_id_analysis_get.py +0 -260
  82. orca_sdk/_generated_api_client/api/memoryset/list_memorysets_memoryset_get.py +0 -127
  83. orca_sdk/_generated_api_client/api/memoryset/memoryset_lookup_gpu_memoryset_name_or_id_lookup_post.py +0 -193
  84. orca_sdk/_generated_api_client/api/memoryset/query_memoryset_memoryset_name_or_id_memories_post.py +0 -188
  85. orca_sdk/_generated_api_client/api/memoryset/update_memories_gpu_memoryset_name_or_id_memories_patch.py +0 -191
  86. orca_sdk/_generated_api_client/api/memoryset/update_memory_gpu_memoryset_name_or_id_memory_patch.py +0 -187
  87. orca_sdk/_generated_api_client/api/pretrained_embedding_model/__init__.py +0 -0
  88. orca_sdk/_generated_api_client/api/pretrained_embedding_model/embed_with_pretrained_model_gpu_pretrained_embedding_model_model_name_embedding_post.py +0 -188
  89. orca_sdk/_generated_api_client/api/pretrained_embedding_model/get_pretrained_embedding_model_pretrained_embedding_model_model_name_get.py +0 -157
  90. orca_sdk/_generated_api_client/api/pretrained_embedding_model/list_pretrained_embedding_models_pretrained_embedding_model_get.py +0 -127
  91. orca_sdk/_generated_api_client/api/task/__init__.py +0 -0
  92. orca_sdk/_generated_api_client/api/task/abort_task_task_task_id_abort_delete.py +0 -154
  93. orca_sdk/_generated_api_client/api/task/get_task_status_task_task_id_status_get.py +0 -156
  94. orca_sdk/_generated_api_client/api/task/list_tasks_task_get.py +0 -243
  95. orca_sdk/_generated_api_client/api/telemetry/__init__.py +0 -0
  96. orca_sdk/_generated_api_client/api/telemetry/drop_feedback_category_with_data_telemetry_feedback_category_name_or_id_delete.py +0 -162
  97. orca_sdk/_generated_api_client/api/telemetry/get_feedback_category_telemetry_feedback_category_name_or_id_get.py +0 -156
  98. orca_sdk/_generated_api_client/api/telemetry/get_prediction_telemetry_prediction_prediction_id_get.py +0 -157
  99. orca_sdk/_generated_api_client/api/telemetry/list_feedback_categories_telemetry_feedback_category_get.py +0 -127
  100. orca_sdk/_generated_api_client/api/telemetry/list_predictions_telemetry_prediction_post.py +0 -175
  101. orca_sdk/_generated_api_client/api/telemetry/record_prediction_feedback_telemetry_prediction_feedback_put.py +0 -171
  102. orca_sdk/_generated_api_client/api/telemetry/update_prediction_telemetry_prediction_prediction_id_patch.py +0 -181
  103. orca_sdk/_generated_api_client/client.py +0 -216
  104. orca_sdk/_generated_api_client/errors.py +0 -38
  105. orca_sdk/_generated_api_client/models/__init__.py +0 -159
  106. orca_sdk/_generated_api_client/models/analyze_neighbor_labels_result.py +0 -84
  107. orca_sdk/_generated_api_client/models/api_key_metadata.py +0 -118
  108. orca_sdk/_generated_api_client/models/base_model.py +0 -55
  109. orca_sdk/_generated_api_client/models/body_create_datasource_datasource_post.py +0 -176
  110. orca_sdk/_generated_api_client/models/classification_evaluation_result.py +0 -114
  111. orca_sdk/_generated_api_client/models/clone_labeled_memoryset_request.py +0 -150
  112. orca_sdk/_generated_api_client/models/column_info.py +0 -114
  113. orca_sdk/_generated_api_client/models/column_type.py +0 -14
  114. orca_sdk/_generated_api_client/models/conflict_error_response.py +0 -80
  115. orca_sdk/_generated_api_client/models/create_api_key_request.py +0 -99
  116. orca_sdk/_generated_api_client/models/create_api_key_response.py +0 -126
  117. orca_sdk/_generated_api_client/models/create_labeled_memoryset_request.py +0 -259
  118. orca_sdk/_generated_api_client/models/create_rac_model_request.py +0 -209
  119. orca_sdk/_generated_api_client/models/datasource_metadata.py +0 -142
  120. orca_sdk/_generated_api_client/models/delete_memories_request.py +0 -70
  121. orca_sdk/_generated_api_client/models/embed_request.py +0 -127
  122. orca_sdk/_generated_api_client/models/embedding_finetuning_method.py +0 -9
  123. orca_sdk/_generated_api_client/models/evaluation_request.py +0 -180
  124. orca_sdk/_generated_api_client/models/evaluation_response.py +0 -140
  125. orca_sdk/_generated_api_client/models/feedback_type.py +0 -9
  126. orca_sdk/_generated_api_client/models/field_validation_error.py +0 -103
  127. orca_sdk/_generated_api_client/models/filter_item.py +0 -231
  128. orca_sdk/_generated_api_client/models/filter_item_field_type_0_item.py +0 -15
  129. orca_sdk/_generated_api_client/models/filter_item_field_type_2_item_type_1.py +0 -16
  130. orca_sdk/_generated_api_client/models/filter_item_op.py +0 -16
  131. orca_sdk/_generated_api_client/models/find_duplicates_analysis_result.py +0 -70
  132. orca_sdk/_generated_api_client/models/finetune_embedding_model_request.py +0 -259
  133. orca_sdk/_generated_api_client/models/finetune_embedding_model_request_training_args.py +0 -66
  134. orca_sdk/_generated_api_client/models/finetuned_embedding_model_metadata.py +0 -166
  135. orca_sdk/_generated_api_client/models/get_memories_request.py +0 -70
  136. orca_sdk/_generated_api_client/models/internal_server_error_response.py +0 -80
  137. orca_sdk/_generated_api_client/models/label_class_metrics.py +0 -108
  138. orca_sdk/_generated_api_client/models/label_prediction_memory_lookup.py +0 -274
  139. orca_sdk/_generated_api_client/models/label_prediction_memory_lookup_metadata.py +0 -68
  140. orca_sdk/_generated_api_client/models/label_prediction_result.py +0 -101
  141. orca_sdk/_generated_api_client/models/label_prediction_with_memories_and_feedback.py +0 -232
  142. orca_sdk/_generated_api_client/models/labeled_memory.py +0 -197
  143. orca_sdk/_generated_api_client/models/labeled_memory_insert.py +0 -108
  144. orca_sdk/_generated_api_client/models/labeled_memory_insert_metadata.py +0 -68
  145. orca_sdk/_generated_api_client/models/labeled_memory_lookup.py +0 -258
  146. orca_sdk/_generated_api_client/models/labeled_memory_lookup_metadata.py +0 -68
  147. orca_sdk/_generated_api_client/models/labeled_memory_metadata.py +0 -68
  148. orca_sdk/_generated_api_client/models/labeled_memory_metrics.py +0 -277
  149. orca_sdk/_generated_api_client/models/labeled_memory_update.py +0 -171
  150. orca_sdk/_generated_api_client/models/labeled_memory_update_metadata_type_0.py +0 -68
  151. orca_sdk/_generated_api_client/models/labeled_memoryset_metadata.py +0 -195
  152. orca_sdk/_generated_api_client/models/list_analyses_memoryset_name_or_id_analysis_get_type_type_0.py +0 -9
  153. orca_sdk/_generated_api_client/models/list_memories_request.py +0 -104
  154. orca_sdk/_generated_api_client/models/list_predictions_request.py +0 -234
  155. orca_sdk/_generated_api_client/models/list_predictions_request_sort_item_item_type_0.py +0 -9
  156. orca_sdk/_generated_api_client/models/list_predictions_request_sort_item_item_type_1.py +0 -9
  157. orca_sdk/_generated_api_client/models/lookup_request.py +0 -81
  158. orca_sdk/_generated_api_client/models/memoryset_analysis_request.py +0 -83
  159. orca_sdk/_generated_api_client/models/memoryset_analysis_request_type.py +0 -9
  160. orca_sdk/_generated_api_client/models/memoryset_analysis_response.py +0 -180
  161. orca_sdk/_generated_api_client/models/memoryset_analysis_response_config.py +0 -66
  162. orca_sdk/_generated_api_client/models/memoryset_analysis_response_type.py +0 -9
  163. orca_sdk/_generated_api_client/models/not_found_error_response.py +0 -100
  164. orca_sdk/_generated_api_client/models/not_found_error_response_resource_type_0.py +0 -20
  165. orca_sdk/_generated_api_client/models/prediction_feedback.py +0 -157
  166. orca_sdk/_generated_api_client/models/prediction_feedback_category.py +0 -115
  167. orca_sdk/_generated_api_client/models/prediction_feedback_request.py +0 -122
  168. orca_sdk/_generated_api_client/models/prediction_feedback_result.py +0 -102
  169. orca_sdk/_generated_api_client/models/prediction_request.py +0 -169
  170. orca_sdk/_generated_api_client/models/pretrained_embedding_model_metadata.py +0 -97
  171. orca_sdk/_generated_api_client/models/pretrained_embedding_model_name.py +0 -11
  172. orca_sdk/_generated_api_client/models/rac_head_type.py +0 -11
  173. orca_sdk/_generated_api_client/models/rac_model_metadata.py +0 -191
  174. orca_sdk/_generated_api_client/models/service_unavailable_error_response.py +0 -80
  175. orca_sdk/_generated_api_client/models/task.py +0 -198
  176. orca_sdk/_generated_api_client/models/task_status.py +0 -14
  177. orca_sdk/_generated_api_client/models/task_status_info.py +0 -133
  178. orca_sdk/_generated_api_client/models/unauthenticated_error_response.py +0 -72
  179. orca_sdk/_generated_api_client/models/unauthorized_error_response.py +0 -80
  180. orca_sdk/_generated_api_client/models/unprocessable_input_error_response.py +0 -94
  181. orca_sdk/_generated_api_client/models/update_prediction_request.py +0 -93
  182. orca_sdk/_generated_api_client/py.typed +0 -1
  183. orca_sdk/_generated_api_client/types.py +0 -56
  184. orca_sdk/_utils/task.py +0 -73
  185. 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