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.
- orca_sdk/__init__.py +10 -4
- orca_sdk/_shared/__init__.py +10 -0
- orca_sdk/_shared/metrics.py +393 -0
- orca_sdk/_shared/metrics_test.py +273 -0
- orca_sdk/_utils/analysis_ui.py +12 -10
- orca_sdk/_utils/analysis_ui_style.css +0 -3
- orca_sdk/_utils/auth.py +27 -29
- orca_sdk/_utils/data_parsing.py +28 -2
- orca_sdk/_utils/data_parsing_test.py +15 -15
- orca_sdk/_utils/pagination.py +126 -0
- orca_sdk/_utils/pagination_test.py +132 -0
- orca_sdk/_utils/prediction_result_ui.py +67 -21
- orca_sdk/_utils/tqdm_file_reader.py +12 -0
- orca_sdk/_utils/value_parser.py +45 -0
- orca_sdk/_utils/value_parser_test.py +39 -0
- orca_sdk/classification_model.py +439 -129
- orca_sdk/classification_model_test.py +334 -104
- orca_sdk/client.py +3747 -0
- orca_sdk/conftest.py +164 -19
- orca_sdk/credentials.py +120 -18
- orca_sdk/credentials_test.py +20 -0
- orca_sdk/datasource.py +259 -68
- orca_sdk/datasource_test.py +242 -0
- orca_sdk/embedding_model.py +425 -82
- orca_sdk/embedding_model_test.py +39 -13
- orca_sdk/job.py +337 -0
- orca_sdk/job_test.py +108 -0
- orca_sdk/memoryset.py +1341 -305
- orca_sdk/memoryset_test.py +350 -111
- orca_sdk/regression_model.py +684 -0
- orca_sdk/regression_model_test.py +369 -0
- orca_sdk/telemetry.py +449 -143
- orca_sdk/telemetry_test.py +43 -24
- {orca_sdk-0.1.1.dist-info → orca_sdk-0.1.2.dist-info}/METADATA +34 -16
- orca_sdk-0.1.2.dist-info/RECORD +40 -0
- {orca_sdk-0.1.1.dist-info → orca_sdk-0.1.2.dist-info}/WHEEL +1 -1
- orca_sdk/_generated_api_client/__init__.py +0 -3
- orca_sdk/_generated_api_client/api/__init__.py +0 -193
- orca_sdk/_generated_api_client/api/auth/__init__.py +0 -0
- orca_sdk/_generated_api_client/api/auth/check_authentication_auth_get.py +0 -128
- orca_sdk/_generated_api_client/api/auth/create_api_key_auth_api_key_post.py +0 -170
- orca_sdk/_generated_api_client/api/auth/delete_api_key_auth_api_key_name_or_id_delete.py +0 -156
- orca_sdk/_generated_api_client/api/auth/delete_org_auth_org_delete.py +0 -130
- orca_sdk/_generated_api_client/api/auth/list_api_keys_auth_api_key_get.py +0 -127
- orca_sdk/_generated_api_client/api/classification_model/__init__.py +0 -0
- orca_sdk/_generated_api_client/api/classification_model/create_evaluation_classification_model_model_name_or_id_evaluation_post.py +0 -183
- orca_sdk/_generated_api_client/api/classification_model/create_model_classification_model_post.py +0 -170
- orca_sdk/_generated_api_client/api/classification_model/delete_evaluation_classification_model_model_name_or_id_evaluation_task_id_delete.py +0 -168
- orca_sdk/_generated_api_client/api/classification_model/delete_model_classification_model_name_or_id_delete.py +0 -154
- orca_sdk/_generated_api_client/api/classification_model/get_evaluation_classification_model_model_name_or_id_evaluation_task_id_get.py +0 -170
- orca_sdk/_generated_api_client/api/classification_model/get_model_classification_model_name_or_id_get.py +0 -156
- orca_sdk/_generated_api_client/api/classification_model/list_evaluations_classification_model_model_name_or_id_evaluation_get.py +0 -161
- orca_sdk/_generated_api_client/api/classification_model/list_models_classification_model_get.py +0 -127
- orca_sdk/_generated_api_client/api/classification_model/predict_gpu_classification_model_name_or_id_prediction_post.py +0 -190
- orca_sdk/_generated_api_client/api/datasource/__init__.py +0 -0
- orca_sdk/_generated_api_client/api/datasource/create_datasource_datasource_post.py +0 -167
- orca_sdk/_generated_api_client/api/datasource/delete_datasource_datasource_name_or_id_delete.py +0 -156
- orca_sdk/_generated_api_client/api/datasource/get_datasource_datasource_name_or_id_get.py +0 -156
- orca_sdk/_generated_api_client/api/datasource/list_datasources_datasource_get.py +0 -127
- orca_sdk/_generated_api_client/api/default/__init__.py +0 -0
- orca_sdk/_generated_api_client/api/default/healthcheck_get.py +0 -118
- orca_sdk/_generated_api_client/api/default/healthcheck_gpu_get.py +0 -118
- orca_sdk/_generated_api_client/api/finetuned_embedding_model/__init__.py +0 -0
- orca_sdk/_generated_api_client/api/finetuned_embedding_model/create_finetuned_embedding_model_finetuned_embedding_model_post.py +0 -168
- orca_sdk/_generated_api_client/api/finetuned_embedding_model/delete_finetuned_embedding_model_finetuned_embedding_model_name_or_id_delete.py +0 -156
- 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
- orca_sdk/_generated_api_client/api/finetuned_embedding_model/get_finetuned_embedding_model_finetuned_embedding_model_name_or_id_get.py +0 -156
- orca_sdk/_generated_api_client/api/finetuned_embedding_model/list_finetuned_embedding_models_finetuned_embedding_model_get.py +0 -127
- orca_sdk/_generated_api_client/api/memoryset/__init__.py +0 -0
- orca_sdk/_generated_api_client/api/memoryset/clone_memoryset_memoryset_name_or_id_clone_post.py +0 -181
- orca_sdk/_generated_api_client/api/memoryset/create_analysis_memoryset_name_or_id_analysis_post.py +0 -183
- orca_sdk/_generated_api_client/api/memoryset/create_memoryset_memoryset_post.py +0 -168
- orca_sdk/_generated_api_client/api/memoryset/delete_memories_memoryset_name_or_id_memories_delete_post.py +0 -181
- orca_sdk/_generated_api_client/api/memoryset/delete_memory_memoryset_name_or_id_memory_memory_id_delete.py +0 -167
- orca_sdk/_generated_api_client/api/memoryset/delete_memoryset_memoryset_name_or_id_delete.py +0 -156
- orca_sdk/_generated_api_client/api/memoryset/get_analysis_memoryset_name_or_id_analysis_analysis_task_id_get.py +0 -169
- orca_sdk/_generated_api_client/api/memoryset/get_memories_memoryset_name_or_id_memories_get_post.py +0 -188
- orca_sdk/_generated_api_client/api/memoryset/get_memory_memoryset_name_or_id_memory_memory_id_get.py +0 -169
- orca_sdk/_generated_api_client/api/memoryset/get_memoryset_memoryset_name_or_id_get.py +0 -156
- orca_sdk/_generated_api_client/api/memoryset/insert_memories_gpu_memoryset_name_or_id_memory_post.py +0 -184
- orca_sdk/_generated_api_client/api/memoryset/list_analyses_memoryset_name_or_id_analysis_get.py +0 -260
- orca_sdk/_generated_api_client/api/memoryset/list_memorysets_memoryset_get.py +0 -127
- orca_sdk/_generated_api_client/api/memoryset/memoryset_lookup_gpu_memoryset_name_or_id_lookup_post.py +0 -193
- orca_sdk/_generated_api_client/api/memoryset/query_memoryset_memoryset_name_or_id_memories_post.py +0 -188
- orca_sdk/_generated_api_client/api/memoryset/update_memories_gpu_memoryset_name_or_id_memories_patch.py +0 -191
- orca_sdk/_generated_api_client/api/memoryset/update_memory_gpu_memoryset_name_or_id_memory_patch.py +0 -187
- orca_sdk/_generated_api_client/api/pretrained_embedding_model/__init__.py +0 -0
- orca_sdk/_generated_api_client/api/pretrained_embedding_model/embed_with_pretrained_model_gpu_pretrained_embedding_model_model_name_embedding_post.py +0 -188
- orca_sdk/_generated_api_client/api/pretrained_embedding_model/get_pretrained_embedding_model_pretrained_embedding_model_model_name_get.py +0 -157
- orca_sdk/_generated_api_client/api/pretrained_embedding_model/list_pretrained_embedding_models_pretrained_embedding_model_get.py +0 -127
- orca_sdk/_generated_api_client/api/task/__init__.py +0 -0
- orca_sdk/_generated_api_client/api/task/abort_task_task_task_id_abort_delete.py +0 -154
- orca_sdk/_generated_api_client/api/task/get_task_status_task_task_id_status_get.py +0 -156
- orca_sdk/_generated_api_client/api/task/list_tasks_task_get.py +0 -243
- orca_sdk/_generated_api_client/api/telemetry/__init__.py +0 -0
- orca_sdk/_generated_api_client/api/telemetry/drop_feedback_category_with_data_telemetry_feedback_category_name_or_id_delete.py +0 -162
- orca_sdk/_generated_api_client/api/telemetry/get_feedback_category_telemetry_feedback_category_name_or_id_get.py +0 -156
- orca_sdk/_generated_api_client/api/telemetry/get_prediction_telemetry_prediction_prediction_id_get.py +0 -157
- orca_sdk/_generated_api_client/api/telemetry/list_feedback_categories_telemetry_feedback_category_get.py +0 -127
- orca_sdk/_generated_api_client/api/telemetry/list_predictions_telemetry_prediction_post.py +0 -175
- orca_sdk/_generated_api_client/api/telemetry/record_prediction_feedback_telemetry_prediction_feedback_put.py +0 -171
- orca_sdk/_generated_api_client/api/telemetry/update_prediction_telemetry_prediction_prediction_id_patch.py +0 -181
- orca_sdk/_generated_api_client/client.py +0 -216
- orca_sdk/_generated_api_client/errors.py +0 -38
- orca_sdk/_generated_api_client/models/__init__.py +0 -159
- orca_sdk/_generated_api_client/models/analyze_neighbor_labels_result.py +0 -84
- orca_sdk/_generated_api_client/models/api_key_metadata.py +0 -118
- orca_sdk/_generated_api_client/models/base_model.py +0 -55
- orca_sdk/_generated_api_client/models/body_create_datasource_datasource_post.py +0 -176
- orca_sdk/_generated_api_client/models/classification_evaluation_result.py +0 -114
- orca_sdk/_generated_api_client/models/clone_labeled_memoryset_request.py +0 -150
- orca_sdk/_generated_api_client/models/column_info.py +0 -114
- orca_sdk/_generated_api_client/models/column_type.py +0 -14
- orca_sdk/_generated_api_client/models/conflict_error_response.py +0 -80
- orca_sdk/_generated_api_client/models/create_api_key_request.py +0 -99
- orca_sdk/_generated_api_client/models/create_api_key_response.py +0 -126
- orca_sdk/_generated_api_client/models/create_labeled_memoryset_request.py +0 -259
- orca_sdk/_generated_api_client/models/create_rac_model_request.py +0 -209
- orca_sdk/_generated_api_client/models/datasource_metadata.py +0 -142
- orca_sdk/_generated_api_client/models/delete_memories_request.py +0 -70
- orca_sdk/_generated_api_client/models/embed_request.py +0 -127
- orca_sdk/_generated_api_client/models/embedding_finetuning_method.py +0 -9
- orca_sdk/_generated_api_client/models/evaluation_request.py +0 -180
- orca_sdk/_generated_api_client/models/evaluation_response.py +0 -140
- orca_sdk/_generated_api_client/models/feedback_type.py +0 -9
- orca_sdk/_generated_api_client/models/field_validation_error.py +0 -103
- orca_sdk/_generated_api_client/models/filter_item.py +0 -231
- orca_sdk/_generated_api_client/models/filter_item_field_type_0_item.py +0 -15
- orca_sdk/_generated_api_client/models/filter_item_field_type_2_item_type_1.py +0 -16
- orca_sdk/_generated_api_client/models/filter_item_op.py +0 -16
- orca_sdk/_generated_api_client/models/find_duplicates_analysis_result.py +0 -70
- orca_sdk/_generated_api_client/models/finetune_embedding_model_request.py +0 -259
- orca_sdk/_generated_api_client/models/finetune_embedding_model_request_training_args.py +0 -66
- orca_sdk/_generated_api_client/models/finetuned_embedding_model_metadata.py +0 -166
- orca_sdk/_generated_api_client/models/get_memories_request.py +0 -70
- orca_sdk/_generated_api_client/models/internal_server_error_response.py +0 -80
- orca_sdk/_generated_api_client/models/label_class_metrics.py +0 -108
- orca_sdk/_generated_api_client/models/label_prediction_memory_lookup.py +0 -274
- orca_sdk/_generated_api_client/models/label_prediction_memory_lookup_metadata.py +0 -68
- orca_sdk/_generated_api_client/models/label_prediction_result.py +0 -101
- orca_sdk/_generated_api_client/models/label_prediction_with_memories_and_feedback.py +0 -232
- orca_sdk/_generated_api_client/models/labeled_memory.py +0 -197
- orca_sdk/_generated_api_client/models/labeled_memory_insert.py +0 -108
- orca_sdk/_generated_api_client/models/labeled_memory_insert_metadata.py +0 -68
- orca_sdk/_generated_api_client/models/labeled_memory_lookup.py +0 -258
- orca_sdk/_generated_api_client/models/labeled_memory_lookup_metadata.py +0 -68
- orca_sdk/_generated_api_client/models/labeled_memory_metadata.py +0 -68
- orca_sdk/_generated_api_client/models/labeled_memory_metrics.py +0 -277
- orca_sdk/_generated_api_client/models/labeled_memory_update.py +0 -171
- orca_sdk/_generated_api_client/models/labeled_memory_update_metadata_type_0.py +0 -68
- orca_sdk/_generated_api_client/models/labeled_memoryset_metadata.py +0 -195
- orca_sdk/_generated_api_client/models/list_analyses_memoryset_name_or_id_analysis_get_type_type_0.py +0 -9
- orca_sdk/_generated_api_client/models/list_memories_request.py +0 -104
- orca_sdk/_generated_api_client/models/list_predictions_request.py +0 -234
- orca_sdk/_generated_api_client/models/list_predictions_request_sort_item_item_type_0.py +0 -9
- orca_sdk/_generated_api_client/models/list_predictions_request_sort_item_item_type_1.py +0 -9
- orca_sdk/_generated_api_client/models/lookup_request.py +0 -81
- orca_sdk/_generated_api_client/models/memoryset_analysis_request.py +0 -83
- orca_sdk/_generated_api_client/models/memoryset_analysis_request_type.py +0 -9
- orca_sdk/_generated_api_client/models/memoryset_analysis_response.py +0 -180
- orca_sdk/_generated_api_client/models/memoryset_analysis_response_config.py +0 -66
- orca_sdk/_generated_api_client/models/memoryset_analysis_response_type.py +0 -9
- orca_sdk/_generated_api_client/models/not_found_error_response.py +0 -100
- orca_sdk/_generated_api_client/models/not_found_error_response_resource_type_0.py +0 -20
- orca_sdk/_generated_api_client/models/prediction_feedback.py +0 -157
- orca_sdk/_generated_api_client/models/prediction_feedback_category.py +0 -115
- orca_sdk/_generated_api_client/models/prediction_feedback_request.py +0 -122
- orca_sdk/_generated_api_client/models/prediction_feedback_result.py +0 -102
- orca_sdk/_generated_api_client/models/prediction_request.py +0 -169
- orca_sdk/_generated_api_client/models/pretrained_embedding_model_metadata.py +0 -97
- orca_sdk/_generated_api_client/models/pretrained_embedding_model_name.py +0 -11
- orca_sdk/_generated_api_client/models/rac_head_type.py +0 -11
- orca_sdk/_generated_api_client/models/rac_model_metadata.py +0 -191
- orca_sdk/_generated_api_client/models/service_unavailable_error_response.py +0 -80
- orca_sdk/_generated_api_client/models/task.py +0 -198
- orca_sdk/_generated_api_client/models/task_status.py +0 -14
- orca_sdk/_generated_api_client/models/task_status_info.py +0 -133
- orca_sdk/_generated_api_client/models/unauthenticated_error_response.py +0 -72
- orca_sdk/_generated_api_client/models/unauthorized_error_response.py +0 -80
- orca_sdk/_generated_api_client/models/unprocessable_input_error_response.py +0 -94
- orca_sdk/_generated_api_client/models/update_prediction_request.py +0 -93
- orca_sdk/_generated_api_client/py.typed +0 -1
- orca_sdk/_generated_api_client/types.py +0 -56
- orca_sdk/_utils/task.py +0 -73
- orca_sdk-0.1.1.dist-info/RECORD +0 -175
|
@@ -1,156 +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
|
|
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(
|
|
25
|
-
name_or_id: str,
|
|
26
|
-
) -> dict[str, Any]:
|
|
27
|
-
_kwargs: dict[str, Any] = {
|
|
28
|
-
"method": "get",
|
|
29
|
-
"url": f"/memoryset/{name_or_id}",
|
|
30
|
-
}
|
|
31
|
-
|
|
32
|
-
return _kwargs
|
|
33
|
-
|
|
34
|
-
|
|
35
|
-
def _parse_response(*, response: httpx.Response) -> LabeledMemorysetMetadata:
|
|
36
|
-
if response.status_code == 200:
|
|
37
|
-
response_200 = LabeledMemorysetMetadata.from_dict(response.json())
|
|
38
|
-
|
|
39
|
-
return response_200
|
|
40
|
-
if response.status_code == 401:
|
|
41
|
-
raise get_error_for_response(response)
|
|
42
|
-
if response.status_code == 500:
|
|
43
|
-
raise get_error_for_response(response)
|
|
44
|
-
if response.status_code == 503:
|
|
45
|
-
raise get_error_for_response(response)
|
|
46
|
-
if response.status_code == 422:
|
|
47
|
-
raise get_error_for_response(response)
|
|
48
|
-
if response.status_code == 404:
|
|
49
|
-
raise get_error_for_response(response)
|
|
50
|
-
|
|
51
|
-
|
|
52
|
-
def _build_response(*, response: httpx.Response) -> Response[LabeledMemorysetMetadata]:
|
|
53
|
-
return Response(
|
|
54
|
-
status_code=HTTPStatus(response.status_code),
|
|
55
|
-
content=response.content,
|
|
56
|
-
headers=response.headers,
|
|
57
|
-
parsed=_parse_response(response=response),
|
|
58
|
-
)
|
|
59
|
-
|
|
60
|
-
|
|
61
|
-
def sync_detailed(
|
|
62
|
-
name_or_id: str,
|
|
63
|
-
) -> Response[LabeledMemorysetMetadata]:
|
|
64
|
-
"""Get Memoryset
|
|
65
|
-
|
|
66
|
-
Args:
|
|
67
|
-
name_or_id (str):
|
|
68
|
-
|
|
69
|
-
Raises:
|
|
70
|
-
errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
|
|
71
|
-
httpx.TimeoutException: If the request takes longer than Client.timeout.
|
|
72
|
-
|
|
73
|
-
Returns:
|
|
74
|
-
Response[LabeledMemorysetMetadata]
|
|
75
|
-
"""
|
|
76
|
-
|
|
77
|
-
kwargs = _get_kwargs(
|
|
78
|
-
name_or_id=name_or_id,
|
|
79
|
-
)
|
|
80
|
-
|
|
81
|
-
with _client_context() as client:
|
|
82
|
-
response = client.get_httpx_client().request(
|
|
83
|
-
**kwargs,
|
|
84
|
-
)
|
|
85
|
-
|
|
86
|
-
return _build_response(response=response)
|
|
87
|
-
|
|
88
|
-
|
|
89
|
-
def sync(
|
|
90
|
-
name_or_id: str,
|
|
91
|
-
) -> LabeledMemorysetMetadata:
|
|
92
|
-
"""Get Memoryset
|
|
93
|
-
|
|
94
|
-
Args:
|
|
95
|
-
name_or_id (str):
|
|
96
|
-
|
|
97
|
-
Raises:
|
|
98
|
-
errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
|
|
99
|
-
httpx.TimeoutException: If the request takes longer than Client.timeout.
|
|
100
|
-
|
|
101
|
-
Returns:
|
|
102
|
-
LabeledMemorysetMetadata
|
|
103
|
-
"""
|
|
104
|
-
|
|
105
|
-
return sync_detailed(
|
|
106
|
-
name_or_id=name_or_id,
|
|
107
|
-
).parsed
|
|
108
|
-
|
|
109
|
-
|
|
110
|
-
async def asyncio_detailed(
|
|
111
|
-
name_or_id: str,
|
|
112
|
-
) -> Response[LabeledMemorysetMetadata]:
|
|
113
|
-
"""Get Memoryset
|
|
114
|
-
|
|
115
|
-
Args:
|
|
116
|
-
name_or_id (str):
|
|
117
|
-
|
|
118
|
-
Raises:
|
|
119
|
-
errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
|
|
120
|
-
httpx.TimeoutException: If the request takes longer than Client.timeout.
|
|
121
|
-
|
|
122
|
-
Returns:
|
|
123
|
-
Response[LabeledMemorysetMetadata]
|
|
124
|
-
"""
|
|
125
|
-
|
|
126
|
-
kwargs = _get_kwargs(
|
|
127
|
-
name_or_id=name_or_id,
|
|
128
|
-
)
|
|
129
|
-
|
|
130
|
-
with _client_context() as client:
|
|
131
|
-
response = await client.get_async_httpx_client().request(**kwargs)
|
|
132
|
-
|
|
133
|
-
return _build_response(response=response)
|
|
134
|
-
|
|
135
|
-
|
|
136
|
-
async def asyncio(
|
|
137
|
-
name_or_id: str,
|
|
138
|
-
) -> LabeledMemorysetMetadata:
|
|
139
|
-
"""Get Memoryset
|
|
140
|
-
|
|
141
|
-
Args:
|
|
142
|
-
name_or_id (str):
|
|
143
|
-
|
|
144
|
-
Raises:
|
|
145
|
-
errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
|
|
146
|
-
httpx.TimeoutException: If the request takes longer than Client.timeout.
|
|
147
|
-
|
|
148
|
-
Returns:
|
|
149
|
-
LabeledMemorysetMetadata
|
|
150
|
-
"""
|
|
151
|
-
|
|
152
|
-
return (
|
|
153
|
-
await asyncio_detailed(
|
|
154
|
-
name_or_id=name_or_id,
|
|
155
|
-
)
|
|
156
|
-
).parsed
|
orca_sdk/_generated_api_client/api/memoryset/insert_memories_gpu_memoryset_name_or_id_memory_post.py
DELETED
|
@@ -1,184 +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_insert import LabeledMemoryInsert
|
|
21
|
-
from ...types import Response
|
|
22
|
-
|
|
23
|
-
|
|
24
|
-
def _get_kwargs(
|
|
25
|
-
name_or_id: str,
|
|
26
|
-
*,
|
|
27
|
-
body: List["LabeledMemoryInsert"],
|
|
28
|
-
) -> dict[str, Any]:
|
|
29
|
-
headers: Dict[str, Any] = {}
|
|
30
|
-
|
|
31
|
-
_kwargs: dict[str, Any] = {
|
|
32
|
-
"method": "post",
|
|
33
|
-
"url": f"/gpu/memoryset/{name_or_id}/memory",
|
|
34
|
-
}
|
|
35
|
-
|
|
36
|
-
_body = []
|
|
37
|
-
for body_item_data in body:
|
|
38
|
-
body_item = body_item_data.to_dict()
|
|
39
|
-
_body.append(body_item)
|
|
40
|
-
|
|
41
|
-
_kwargs["json"] = _body
|
|
42
|
-
headers["Content-Type"] = "application/json"
|
|
43
|
-
|
|
44
|
-
_kwargs["headers"] = headers
|
|
45
|
-
return _kwargs
|
|
46
|
-
|
|
47
|
-
|
|
48
|
-
def _parse_response(*, response: httpx.Response) -> None:
|
|
49
|
-
if response.status_code == 201:
|
|
50
|
-
response_201 = response.json()
|
|
51
|
-
return response_201
|
|
52
|
-
if response.status_code == 401:
|
|
53
|
-
raise get_error_for_response(response)
|
|
54
|
-
if response.status_code == 500:
|
|
55
|
-
raise get_error_for_response(response)
|
|
56
|
-
if response.status_code == 503:
|
|
57
|
-
raise get_error_for_response(response)
|
|
58
|
-
if response.status_code == 422:
|
|
59
|
-
raise get_error_for_response(response)
|
|
60
|
-
if response.status_code == 404:
|
|
61
|
-
raise get_error_for_response(response)
|
|
62
|
-
|
|
63
|
-
|
|
64
|
-
def _build_response(*, response: httpx.Response) -> Response[None]:
|
|
65
|
-
return Response(
|
|
66
|
-
status_code=HTTPStatus(response.status_code),
|
|
67
|
-
content=response.content,
|
|
68
|
-
headers=response.headers,
|
|
69
|
-
parsed=_parse_response(response=response),
|
|
70
|
-
)
|
|
71
|
-
|
|
72
|
-
|
|
73
|
-
def sync_detailed(
|
|
74
|
-
name_or_id: str,
|
|
75
|
-
*,
|
|
76
|
-
body: List["LabeledMemoryInsert"],
|
|
77
|
-
) -> Response[None]:
|
|
78
|
-
"""Insert Memories
|
|
79
|
-
|
|
80
|
-
Args:
|
|
81
|
-
name_or_id (str):
|
|
82
|
-
body (List['LabeledMemoryInsert']):
|
|
83
|
-
|
|
84
|
-
Raises:
|
|
85
|
-
errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
|
|
86
|
-
httpx.TimeoutException: If the request takes longer than Client.timeout.
|
|
87
|
-
|
|
88
|
-
Returns:
|
|
89
|
-
Response[None]
|
|
90
|
-
"""
|
|
91
|
-
|
|
92
|
-
kwargs = _get_kwargs(
|
|
93
|
-
name_or_id=name_or_id,
|
|
94
|
-
body=body,
|
|
95
|
-
)
|
|
96
|
-
|
|
97
|
-
with _client_context() as client:
|
|
98
|
-
response = client.get_httpx_client().request(
|
|
99
|
-
**kwargs,
|
|
100
|
-
)
|
|
101
|
-
|
|
102
|
-
return _build_response(response=response)
|
|
103
|
-
|
|
104
|
-
|
|
105
|
-
def sync(
|
|
106
|
-
name_or_id: str,
|
|
107
|
-
*,
|
|
108
|
-
body: List["LabeledMemoryInsert"],
|
|
109
|
-
) -> None:
|
|
110
|
-
"""Insert Memories
|
|
111
|
-
|
|
112
|
-
Args:
|
|
113
|
-
name_or_id (str):
|
|
114
|
-
body (List['LabeledMemoryInsert']):
|
|
115
|
-
|
|
116
|
-
Raises:
|
|
117
|
-
errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
|
|
118
|
-
httpx.TimeoutException: If the request takes longer than Client.timeout.
|
|
119
|
-
|
|
120
|
-
Returns:
|
|
121
|
-
None
|
|
122
|
-
"""
|
|
123
|
-
|
|
124
|
-
return sync_detailed(
|
|
125
|
-
name_or_id=name_or_id,
|
|
126
|
-
body=body,
|
|
127
|
-
).parsed
|
|
128
|
-
|
|
129
|
-
|
|
130
|
-
async def asyncio_detailed(
|
|
131
|
-
name_or_id: str,
|
|
132
|
-
*,
|
|
133
|
-
body: List["LabeledMemoryInsert"],
|
|
134
|
-
) -> Response[None]:
|
|
135
|
-
"""Insert Memories
|
|
136
|
-
|
|
137
|
-
Args:
|
|
138
|
-
name_or_id (str):
|
|
139
|
-
body (List['LabeledMemoryInsert']):
|
|
140
|
-
|
|
141
|
-
Raises:
|
|
142
|
-
errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
|
|
143
|
-
httpx.TimeoutException: If the request takes longer than Client.timeout.
|
|
144
|
-
|
|
145
|
-
Returns:
|
|
146
|
-
Response[None]
|
|
147
|
-
"""
|
|
148
|
-
|
|
149
|
-
kwargs = _get_kwargs(
|
|
150
|
-
name_or_id=name_or_id,
|
|
151
|
-
body=body,
|
|
152
|
-
)
|
|
153
|
-
|
|
154
|
-
with _client_context() as client:
|
|
155
|
-
response = await client.get_async_httpx_client().request(**kwargs)
|
|
156
|
-
|
|
157
|
-
return _build_response(response=response)
|
|
158
|
-
|
|
159
|
-
|
|
160
|
-
async def asyncio(
|
|
161
|
-
name_or_id: str,
|
|
162
|
-
*,
|
|
163
|
-
body: List["LabeledMemoryInsert"],
|
|
164
|
-
) -> None:
|
|
165
|
-
"""Insert Memories
|
|
166
|
-
|
|
167
|
-
Args:
|
|
168
|
-
name_or_id (str):
|
|
169
|
-
body (List['LabeledMemoryInsert']):
|
|
170
|
-
|
|
171
|
-
Raises:
|
|
172
|
-
errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
|
|
173
|
-
httpx.TimeoutException: If the request takes longer than Client.timeout.
|
|
174
|
-
|
|
175
|
-
Returns:
|
|
176
|
-
None
|
|
177
|
-
"""
|
|
178
|
-
|
|
179
|
-
return (
|
|
180
|
-
await asyncio_detailed(
|
|
181
|
-
name_or_id=name_or_id,
|
|
182
|
-
body=body,
|
|
183
|
-
)
|
|
184
|
-
).parsed
|
orca_sdk/_generated_api_client/api/memoryset/list_analyses_memoryset_name_or_id_analysis_get.py
DELETED
|
@@ -1,260 +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 enum import Enum
|
|
14
|
-
from http import HTTPStatus
|
|
15
|
-
from typing import Any, Dict, List, Union
|
|
16
|
-
|
|
17
|
-
import httpx
|
|
18
|
-
|
|
19
|
-
from ...client import _client_context
|
|
20
|
-
from ...errors import get_error_for_response
|
|
21
|
-
from ...models.list_analyses_memoryset_name_or_id_analysis_get_type_type_0 import (
|
|
22
|
-
ListAnalysesMemorysetNameOrIdAnalysisGetTypeType0,
|
|
23
|
-
)
|
|
24
|
-
from ...models.memoryset_analysis_response import MemorysetAnalysisResponse
|
|
25
|
-
from ...models.task_status import TaskStatus
|
|
26
|
-
from ...types import UNSET, Response, Unset
|
|
27
|
-
|
|
28
|
-
|
|
29
|
-
def _get_kwargs(
|
|
30
|
-
name_or_id: str,
|
|
31
|
-
*,
|
|
32
|
-
type: Union[ListAnalysesMemorysetNameOrIdAnalysisGetTypeType0, None, Unset] = UNSET,
|
|
33
|
-
status: Union[None, TaskStatus, Unset] = UNSET,
|
|
34
|
-
limit: Union[None, Unset, int] = 100,
|
|
35
|
-
offset: Union[None, Unset, int] = 0,
|
|
36
|
-
) -> dict[str, Any]:
|
|
37
|
-
params: Dict[str, Any] = {}
|
|
38
|
-
|
|
39
|
-
json_type: Union[None, Unset, str]
|
|
40
|
-
if isinstance(type, Unset):
|
|
41
|
-
json_type = UNSET
|
|
42
|
-
elif isinstance(type, ListAnalysesMemorysetNameOrIdAnalysisGetTypeType0):
|
|
43
|
-
json_type = type.value if isinstance(type, Enum) else type
|
|
44
|
-
else:
|
|
45
|
-
json_type = type
|
|
46
|
-
params["type"] = json_type
|
|
47
|
-
|
|
48
|
-
json_status: Union[None, Unset, str]
|
|
49
|
-
if isinstance(status, Unset):
|
|
50
|
-
json_status = UNSET
|
|
51
|
-
elif isinstance(status, TaskStatus):
|
|
52
|
-
json_status = status.value if isinstance(status, Enum) else status
|
|
53
|
-
else:
|
|
54
|
-
json_status = status
|
|
55
|
-
params["status"] = json_status
|
|
56
|
-
|
|
57
|
-
json_limit: Union[None, Unset, int]
|
|
58
|
-
if isinstance(limit, Unset):
|
|
59
|
-
json_limit = UNSET
|
|
60
|
-
else:
|
|
61
|
-
json_limit = limit
|
|
62
|
-
params["limit"] = json_limit
|
|
63
|
-
|
|
64
|
-
json_offset: Union[None, Unset, int]
|
|
65
|
-
if isinstance(offset, Unset):
|
|
66
|
-
json_offset = UNSET
|
|
67
|
-
else:
|
|
68
|
-
json_offset = offset
|
|
69
|
-
params["offset"] = json_offset
|
|
70
|
-
|
|
71
|
-
params = {k: v for k, v in params.items() if v is not UNSET and v is not None}
|
|
72
|
-
|
|
73
|
-
_kwargs: dict[str, Any] = {
|
|
74
|
-
"method": "get",
|
|
75
|
-
"url": f"/memoryset/{name_or_id}/analysis",
|
|
76
|
-
"params": params,
|
|
77
|
-
}
|
|
78
|
-
|
|
79
|
-
return _kwargs
|
|
80
|
-
|
|
81
|
-
|
|
82
|
-
def _parse_response(*, response: httpx.Response) -> List["MemorysetAnalysisResponse"]:
|
|
83
|
-
if response.status_code == 200:
|
|
84
|
-
response_200 = []
|
|
85
|
-
_response_200 = response.json()
|
|
86
|
-
for response_200_item_data in _response_200:
|
|
87
|
-
response_200_item = MemorysetAnalysisResponse.from_dict(response_200_item_data)
|
|
88
|
-
|
|
89
|
-
response_200.append(response_200_item)
|
|
90
|
-
|
|
91
|
-
return response_200
|
|
92
|
-
if response.status_code == 401:
|
|
93
|
-
raise get_error_for_response(response)
|
|
94
|
-
if response.status_code == 500:
|
|
95
|
-
raise get_error_for_response(response)
|
|
96
|
-
if response.status_code == 503:
|
|
97
|
-
raise get_error_for_response(response)
|
|
98
|
-
if response.status_code == 422:
|
|
99
|
-
raise get_error_for_response(response)
|
|
100
|
-
if response.status_code == 404:
|
|
101
|
-
raise get_error_for_response(response)
|
|
102
|
-
|
|
103
|
-
|
|
104
|
-
def _build_response(*, response: httpx.Response) -> Response[List["MemorysetAnalysisResponse"]]:
|
|
105
|
-
return Response(
|
|
106
|
-
status_code=HTTPStatus(response.status_code),
|
|
107
|
-
content=response.content,
|
|
108
|
-
headers=response.headers,
|
|
109
|
-
parsed=_parse_response(response=response),
|
|
110
|
-
)
|
|
111
|
-
|
|
112
|
-
|
|
113
|
-
def sync_detailed(
|
|
114
|
-
name_or_id: str,
|
|
115
|
-
*,
|
|
116
|
-
type: Union[ListAnalysesMemorysetNameOrIdAnalysisGetTypeType0, None, Unset] = UNSET,
|
|
117
|
-
status: Union[None, TaskStatus, Unset] = UNSET,
|
|
118
|
-
limit: Union[None, Unset, int] = 100,
|
|
119
|
-
offset: Union[None, Unset, int] = 0,
|
|
120
|
-
) -> Response[List["MemorysetAnalysisResponse"]]:
|
|
121
|
-
"""List Analyses
|
|
122
|
-
|
|
123
|
-
Args:
|
|
124
|
-
name_or_id (str):
|
|
125
|
-
type (Union[ListAnalysesMemorysetNameOrIdAnalysisGetTypeType0, None, Unset]):
|
|
126
|
-
status (Union[None, TaskStatus, Unset]):
|
|
127
|
-
limit (Union[None, Unset, int]): Default: 100.
|
|
128
|
-
offset (Union[None, Unset, int]): Default: 0.
|
|
129
|
-
|
|
130
|
-
Raises:
|
|
131
|
-
errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
|
|
132
|
-
httpx.TimeoutException: If the request takes longer than Client.timeout.
|
|
133
|
-
|
|
134
|
-
Returns:
|
|
135
|
-
Response[List['MemorysetAnalysisResponse']]
|
|
136
|
-
"""
|
|
137
|
-
|
|
138
|
-
kwargs = _get_kwargs(
|
|
139
|
-
name_or_id=name_or_id,
|
|
140
|
-
type=type,
|
|
141
|
-
status=status,
|
|
142
|
-
limit=limit,
|
|
143
|
-
offset=offset,
|
|
144
|
-
)
|
|
145
|
-
|
|
146
|
-
with _client_context() as client:
|
|
147
|
-
response = client.get_httpx_client().request(
|
|
148
|
-
**kwargs,
|
|
149
|
-
)
|
|
150
|
-
|
|
151
|
-
return _build_response(response=response)
|
|
152
|
-
|
|
153
|
-
|
|
154
|
-
def sync(
|
|
155
|
-
name_or_id: str,
|
|
156
|
-
*,
|
|
157
|
-
type: Union[ListAnalysesMemorysetNameOrIdAnalysisGetTypeType0, None, Unset] = UNSET,
|
|
158
|
-
status: Union[None, TaskStatus, Unset] = UNSET,
|
|
159
|
-
limit: Union[None, Unset, int] = 100,
|
|
160
|
-
offset: Union[None, Unset, int] = 0,
|
|
161
|
-
) -> List["MemorysetAnalysisResponse"]:
|
|
162
|
-
"""List Analyses
|
|
163
|
-
|
|
164
|
-
Args:
|
|
165
|
-
name_or_id (str):
|
|
166
|
-
type (Union[ListAnalysesMemorysetNameOrIdAnalysisGetTypeType0, None, Unset]):
|
|
167
|
-
status (Union[None, TaskStatus, Unset]):
|
|
168
|
-
limit (Union[None, Unset, int]): Default: 100.
|
|
169
|
-
offset (Union[None, Unset, int]): Default: 0.
|
|
170
|
-
|
|
171
|
-
Raises:
|
|
172
|
-
errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
|
|
173
|
-
httpx.TimeoutException: If the request takes longer than Client.timeout.
|
|
174
|
-
|
|
175
|
-
Returns:
|
|
176
|
-
List['MemorysetAnalysisResponse']
|
|
177
|
-
"""
|
|
178
|
-
|
|
179
|
-
return sync_detailed(
|
|
180
|
-
name_or_id=name_or_id,
|
|
181
|
-
type=type,
|
|
182
|
-
status=status,
|
|
183
|
-
limit=limit,
|
|
184
|
-
offset=offset,
|
|
185
|
-
).parsed
|
|
186
|
-
|
|
187
|
-
|
|
188
|
-
async def asyncio_detailed(
|
|
189
|
-
name_or_id: str,
|
|
190
|
-
*,
|
|
191
|
-
type: Union[ListAnalysesMemorysetNameOrIdAnalysisGetTypeType0, None, Unset] = UNSET,
|
|
192
|
-
status: Union[None, TaskStatus, Unset] = UNSET,
|
|
193
|
-
limit: Union[None, Unset, int] = 100,
|
|
194
|
-
offset: Union[None, Unset, int] = 0,
|
|
195
|
-
) -> Response[List["MemorysetAnalysisResponse"]]:
|
|
196
|
-
"""List Analyses
|
|
197
|
-
|
|
198
|
-
Args:
|
|
199
|
-
name_or_id (str):
|
|
200
|
-
type (Union[ListAnalysesMemorysetNameOrIdAnalysisGetTypeType0, None, Unset]):
|
|
201
|
-
status (Union[None, TaskStatus, Unset]):
|
|
202
|
-
limit (Union[None, Unset, int]): Default: 100.
|
|
203
|
-
offset (Union[None, Unset, int]): Default: 0.
|
|
204
|
-
|
|
205
|
-
Raises:
|
|
206
|
-
errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
|
|
207
|
-
httpx.TimeoutException: If the request takes longer than Client.timeout.
|
|
208
|
-
|
|
209
|
-
Returns:
|
|
210
|
-
Response[List['MemorysetAnalysisResponse']]
|
|
211
|
-
"""
|
|
212
|
-
|
|
213
|
-
kwargs = _get_kwargs(
|
|
214
|
-
name_or_id=name_or_id,
|
|
215
|
-
type=type,
|
|
216
|
-
status=status,
|
|
217
|
-
limit=limit,
|
|
218
|
-
offset=offset,
|
|
219
|
-
)
|
|
220
|
-
|
|
221
|
-
with _client_context() as client:
|
|
222
|
-
response = await client.get_async_httpx_client().request(**kwargs)
|
|
223
|
-
|
|
224
|
-
return _build_response(response=response)
|
|
225
|
-
|
|
226
|
-
|
|
227
|
-
async def asyncio(
|
|
228
|
-
name_or_id: str,
|
|
229
|
-
*,
|
|
230
|
-
type: Union[ListAnalysesMemorysetNameOrIdAnalysisGetTypeType0, None, Unset] = UNSET,
|
|
231
|
-
status: Union[None, TaskStatus, Unset] = UNSET,
|
|
232
|
-
limit: Union[None, Unset, int] = 100,
|
|
233
|
-
offset: Union[None, Unset, int] = 0,
|
|
234
|
-
) -> List["MemorysetAnalysisResponse"]:
|
|
235
|
-
"""List Analyses
|
|
236
|
-
|
|
237
|
-
Args:
|
|
238
|
-
name_or_id (str):
|
|
239
|
-
type (Union[ListAnalysesMemorysetNameOrIdAnalysisGetTypeType0, None, Unset]):
|
|
240
|
-
status (Union[None, TaskStatus, Unset]):
|
|
241
|
-
limit (Union[None, Unset, int]): Default: 100.
|
|
242
|
-
offset (Union[None, Unset, int]): Default: 0.
|
|
243
|
-
|
|
244
|
-
Raises:
|
|
245
|
-
errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
|
|
246
|
-
httpx.TimeoutException: If the request takes longer than Client.timeout.
|
|
247
|
-
|
|
248
|
-
Returns:
|
|
249
|
-
List['MemorysetAnalysisResponse']
|
|
250
|
-
"""
|
|
251
|
-
|
|
252
|
-
return (
|
|
253
|
-
await asyncio_detailed(
|
|
254
|
-
name_or_id=name_or_id,
|
|
255
|
-
type=type,
|
|
256
|
-
status=status,
|
|
257
|
-
limit=limit,
|
|
258
|
-
offset=offset,
|
|
259
|
-
)
|
|
260
|
-
).parsed
|