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.
- 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 +31 -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/async_client.py +3795 -0
- orca_sdk/classification_model.py +601 -129
- orca_sdk/classification_model_test.py +415 -117
- orca_sdk/client.py +3787 -0
- orca_sdk/conftest.py +184 -38
- orca_sdk/credentials.py +162 -20
- orca_sdk/credentials_test.py +100 -16
- orca_sdk/datasource.py +268 -68
- orca_sdk/datasource_test.py +266 -18
- orca_sdk/embedding_model.py +434 -82
- orca_sdk/embedding_model_test.py +66 -33
- orca_sdk/job.py +343 -0
- orca_sdk/job_test.py +108 -0
- orca_sdk/memoryset.py +1690 -324
- orca_sdk/memoryset_test.py +456 -119
- orca_sdk/regression_model.py +694 -0
- orca_sdk/regression_model_test.py +378 -0
- orca_sdk/telemetry.py +460 -143
- orca_sdk/telemetry_test.py +43 -24
- {orca_sdk-0.1.1.dist-info → orca_sdk-0.1.3.dist-info}/METADATA +34 -16
- orca_sdk-0.1.3.dist-info/RECORD +41 -0
- {orca_sdk-0.1.1.dist-info → orca_sdk-0.1.3.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,68 +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/model.py.jinja
|
|
6
|
-
|
|
7
|
-
The main change is:
|
|
8
|
-
- Fix typing issues
|
|
9
|
-
"""
|
|
10
|
-
|
|
11
|
-
# flake8: noqa: C901
|
|
12
|
-
|
|
13
|
-
from typing import Any, Type, TypeVar, Union, cast
|
|
14
|
-
|
|
15
|
-
from attrs import define as _attrs_define
|
|
16
|
-
from attrs import field as _attrs_field
|
|
17
|
-
|
|
18
|
-
T = TypeVar("T", bound="LabeledMemoryInsertMetadata")
|
|
19
|
-
|
|
20
|
-
|
|
21
|
-
@_attrs_define
|
|
22
|
-
class LabeledMemoryInsertMetadata:
|
|
23
|
-
""" """
|
|
24
|
-
|
|
25
|
-
additional_properties: dict[str, Union[None, bool, float, int, str]] = _attrs_field(init=False, factory=dict)
|
|
26
|
-
|
|
27
|
-
def to_dict(self) -> dict[str, Any]:
|
|
28
|
-
field_dict: dict[str, Any] = {}
|
|
29
|
-
for prop_name, prop in self.additional_properties.items():
|
|
30
|
-
field_dict[prop_name] = prop
|
|
31
|
-
|
|
32
|
-
return field_dict
|
|
33
|
-
|
|
34
|
-
@classmethod
|
|
35
|
-
def from_dict(cls: Type[T], src_dict: dict[str, Any]) -> T:
|
|
36
|
-
d = src_dict.copy()
|
|
37
|
-
labeled_memory_insert_metadata = cls()
|
|
38
|
-
|
|
39
|
-
additional_properties = {}
|
|
40
|
-
for prop_name, prop_dict in d.items():
|
|
41
|
-
|
|
42
|
-
def _parse_additional_property(data: object) -> Union[None, bool, float, int, str]:
|
|
43
|
-
if data is None:
|
|
44
|
-
return data
|
|
45
|
-
return cast(Union[None, bool, float, int, str], data)
|
|
46
|
-
|
|
47
|
-
additional_property = _parse_additional_property(prop_dict)
|
|
48
|
-
|
|
49
|
-
additional_properties[prop_name] = additional_property
|
|
50
|
-
|
|
51
|
-
labeled_memory_insert_metadata.additional_properties = additional_properties
|
|
52
|
-
return labeled_memory_insert_metadata
|
|
53
|
-
|
|
54
|
-
@property
|
|
55
|
-
def additional_keys(self) -> list[str]:
|
|
56
|
-
return list(self.additional_properties.keys())
|
|
57
|
-
|
|
58
|
-
def __getitem__(self, key: str) -> Union[None, bool, float, int, str]:
|
|
59
|
-
return self.additional_properties[key]
|
|
60
|
-
|
|
61
|
-
def __setitem__(self, key: str, value: Union[None, bool, float, int, str]) -> None:
|
|
62
|
-
self.additional_properties[key] = value
|
|
63
|
-
|
|
64
|
-
def __delitem__(self, key: str) -> None:
|
|
65
|
-
del self.additional_properties[key]
|
|
66
|
-
|
|
67
|
-
def __contains__(self, key: str) -> bool:
|
|
68
|
-
return key in self.additional_properties
|
|
@@ -1,258 +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/model.py.jinja
|
|
6
|
-
|
|
7
|
-
The main change is:
|
|
8
|
-
- Fix typing issues
|
|
9
|
-
"""
|
|
10
|
-
|
|
11
|
-
# flake8: noqa: C901
|
|
12
|
-
|
|
13
|
-
import datetime
|
|
14
|
-
from typing import TYPE_CHECKING, Any, Dict, List, Type, TypeVar, Union, cast
|
|
15
|
-
|
|
16
|
-
from attrs import define as _attrs_define
|
|
17
|
-
from attrs import field as _attrs_field
|
|
18
|
-
from dateutil.parser import isoparse
|
|
19
|
-
|
|
20
|
-
from ..types import UNSET, Unset
|
|
21
|
-
|
|
22
|
-
if TYPE_CHECKING:
|
|
23
|
-
from ..models.labeled_memory_lookup_metadata import LabeledMemoryLookupMetadata
|
|
24
|
-
from ..models.labeled_memory_metrics import LabeledMemoryMetrics
|
|
25
|
-
|
|
26
|
-
|
|
27
|
-
T = TypeVar("T", bound="LabeledMemoryLookup")
|
|
28
|
-
|
|
29
|
-
|
|
30
|
-
@_attrs_define
|
|
31
|
-
class LabeledMemoryLookup:
|
|
32
|
-
"""Single labeled memory lookup result.
|
|
33
|
-
|
|
34
|
-
Attributes:
|
|
35
|
-
value (str):
|
|
36
|
-
embedding (List[float]):
|
|
37
|
-
label (int):
|
|
38
|
-
label_name (Union[None, str]):
|
|
39
|
-
source_id (Union[None, str]):
|
|
40
|
-
metadata (LabeledMemoryLookupMetadata):
|
|
41
|
-
memory_id (str):
|
|
42
|
-
memory_version (int):
|
|
43
|
-
created_at (datetime.datetime):
|
|
44
|
-
updated_at (datetime.datetime):
|
|
45
|
-
metrics (Union['LabeledMemoryMetrics', None]):
|
|
46
|
-
lookup_score (float):
|
|
47
|
-
reranker_score (Union[None, Unset, float]):
|
|
48
|
-
reranker_embedding (Union[List[float], None, Unset]):
|
|
49
|
-
"""
|
|
50
|
-
|
|
51
|
-
value: str
|
|
52
|
-
embedding: List[float]
|
|
53
|
-
label: int
|
|
54
|
-
label_name: Union[None, str]
|
|
55
|
-
source_id: Union[None, str]
|
|
56
|
-
metadata: "LabeledMemoryLookupMetadata"
|
|
57
|
-
memory_id: str
|
|
58
|
-
memory_version: int
|
|
59
|
-
created_at: datetime.datetime
|
|
60
|
-
updated_at: datetime.datetime
|
|
61
|
-
metrics: Union["LabeledMemoryMetrics", None]
|
|
62
|
-
lookup_score: float
|
|
63
|
-
reranker_score: Union[None, Unset, float] = UNSET
|
|
64
|
-
reranker_embedding: Union[List[float], None, Unset] = UNSET
|
|
65
|
-
additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
|
|
66
|
-
|
|
67
|
-
def to_dict(self) -> dict[str, Any]:
|
|
68
|
-
from ..models.labeled_memory_metrics import LabeledMemoryMetrics
|
|
69
|
-
|
|
70
|
-
value: str
|
|
71
|
-
value = self.value
|
|
72
|
-
|
|
73
|
-
embedding = self.embedding
|
|
74
|
-
|
|
75
|
-
label = self.label
|
|
76
|
-
|
|
77
|
-
label_name: Union[None, str]
|
|
78
|
-
label_name = self.label_name
|
|
79
|
-
|
|
80
|
-
source_id: Union[None, str]
|
|
81
|
-
source_id = self.source_id
|
|
82
|
-
|
|
83
|
-
metadata = self.metadata.to_dict()
|
|
84
|
-
|
|
85
|
-
memory_id = self.memory_id
|
|
86
|
-
|
|
87
|
-
memory_version = self.memory_version
|
|
88
|
-
|
|
89
|
-
created_at = self.created_at.isoformat()
|
|
90
|
-
|
|
91
|
-
updated_at = self.updated_at.isoformat()
|
|
92
|
-
|
|
93
|
-
metrics: Union[Dict[str, Any], None]
|
|
94
|
-
if isinstance(self.metrics, LabeledMemoryMetrics):
|
|
95
|
-
metrics = self.metrics.to_dict()
|
|
96
|
-
else:
|
|
97
|
-
metrics = self.metrics
|
|
98
|
-
|
|
99
|
-
lookup_score = self.lookup_score
|
|
100
|
-
|
|
101
|
-
reranker_score: Union[None, Unset, float]
|
|
102
|
-
if isinstance(self.reranker_score, Unset):
|
|
103
|
-
reranker_score = UNSET
|
|
104
|
-
else:
|
|
105
|
-
reranker_score = self.reranker_score
|
|
106
|
-
|
|
107
|
-
reranker_embedding: Union[List[float], None, Unset]
|
|
108
|
-
if isinstance(self.reranker_embedding, Unset):
|
|
109
|
-
reranker_embedding = UNSET
|
|
110
|
-
elif isinstance(self.reranker_embedding, list):
|
|
111
|
-
reranker_embedding = self.reranker_embedding
|
|
112
|
-
|
|
113
|
-
else:
|
|
114
|
-
reranker_embedding = self.reranker_embedding
|
|
115
|
-
|
|
116
|
-
field_dict: dict[str, Any] = {}
|
|
117
|
-
field_dict.update(self.additional_properties)
|
|
118
|
-
field_dict.update(
|
|
119
|
-
{
|
|
120
|
-
"value": value,
|
|
121
|
-
"embedding": embedding,
|
|
122
|
-
"label": label,
|
|
123
|
-
"label_name": label_name,
|
|
124
|
-
"source_id": source_id,
|
|
125
|
-
"metadata": metadata,
|
|
126
|
-
"memory_id": memory_id,
|
|
127
|
-
"memory_version": memory_version,
|
|
128
|
-
"created_at": created_at,
|
|
129
|
-
"updated_at": updated_at,
|
|
130
|
-
"metrics": metrics,
|
|
131
|
-
"lookup_score": lookup_score,
|
|
132
|
-
}
|
|
133
|
-
)
|
|
134
|
-
if reranker_score is not UNSET:
|
|
135
|
-
field_dict["reranker_score"] = reranker_score
|
|
136
|
-
if reranker_embedding is not UNSET:
|
|
137
|
-
field_dict["reranker_embedding"] = reranker_embedding
|
|
138
|
-
|
|
139
|
-
return field_dict
|
|
140
|
-
|
|
141
|
-
@classmethod
|
|
142
|
-
def from_dict(cls: Type[T], src_dict: dict[str, Any]) -> T:
|
|
143
|
-
from ..models.labeled_memory_lookup_metadata import LabeledMemoryLookupMetadata
|
|
144
|
-
from ..models.labeled_memory_metrics import LabeledMemoryMetrics
|
|
145
|
-
|
|
146
|
-
d = src_dict.copy()
|
|
147
|
-
|
|
148
|
-
def _parse_value(data: object) -> str:
|
|
149
|
-
return cast(str, data)
|
|
150
|
-
|
|
151
|
-
value = _parse_value(d.pop("value"))
|
|
152
|
-
|
|
153
|
-
embedding = cast(List[float], d.pop("embedding"))
|
|
154
|
-
|
|
155
|
-
label = d.pop("label")
|
|
156
|
-
|
|
157
|
-
def _parse_label_name(data: object) -> Union[None, str]:
|
|
158
|
-
if data is None:
|
|
159
|
-
return data
|
|
160
|
-
return cast(Union[None, str], data)
|
|
161
|
-
|
|
162
|
-
label_name = _parse_label_name(d.pop("label_name"))
|
|
163
|
-
|
|
164
|
-
def _parse_source_id(data: object) -> Union[None, str]:
|
|
165
|
-
if data is None:
|
|
166
|
-
return data
|
|
167
|
-
return cast(Union[None, str], data)
|
|
168
|
-
|
|
169
|
-
source_id = _parse_source_id(d.pop("source_id"))
|
|
170
|
-
|
|
171
|
-
metadata = LabeledMemoryLookupMetadata.from_dict(d.pop("metadata"))
|
|
172
|
-
|
|
173
|
-
memory_id = d.pop("memory_id")
|
|
174
|
-
|
|
175
|
-
memory_version = d.pop("memory_version")
|
|
176
|
-
|
|
177
|
-
created_at = isoparse(d.pop("created_at"))
|
|
178
|
-
|
|
179
|
-
updated_at = isoparse(d.pop("updated_at"))
|
|
180
|
-
|
|
181
|
-
def _parse_metrics(data: object) -> Union["LabeledMemoryMetrics", None]:
|
|
182
|
-
if data is None:
|
|
183
|
-
return data
|
|
184
|
-
try:
|
|
185
|
-
if not isinstance(data, dict):
|
|
186
|
-
raise TypeError()
|
|
187
|
-
metrics_type_0 = LabeledMemoryMetrics.from_dict(data)
|
|
188
|
-
|
|
189
|
-
return metrics_type_0
|
|
190
|
-
except: # noqa: E722
|
|
191
|
-
pass
|
|
192
|
-
return cast(Union["LabeledMemoryMetrics", None], data)
|
|
193
|
-
|
|
194
|
-
metrics = _parse_metrics(d.pop("metrics"))
|
|
195
|
-
|
|
196
|
-
lookup_score = d.pop("lookup_score")
|
|
197
|
-
|
|
198
|
-
def _parse_reranker_score(data: object) -> Union[None, Unset, float]:
|
|
199
|
-
if data is None:
|
|
200
|
-
return data
|
|
201
|
-
if isinstance(data, Unset):
|
|
202
|
-
return data
|
|
203
|
-
return cast(Union[None, Unset, float], data)
|
|
204
|
-
|
|
205
|
-
reranker_score = _parse_reranker_score(d.pop("reranker_score", UNSET))
|
|
206
|
-
|
|
207
|
-
def _parse_reranker_embedding(data: object) -> Union[List[float], None, Unset]:
|
|
208
|
-
if data is None:
|
|
209
|
-
return data
|
|
210
|
-
if isinstance(data, Unset):
|
|
211
|
-
return data
|
|
212
|
-
try:
|
|
213
|
-
if not isinstance(data, list):
|
|
214
|
-
raise TypeError()
|
|
215
|
-
reranker_embedding_type_0 = cast(List[float], data)
|
|
216
|
-
|
|
217
|
-
return reranker_embedding_type_0
|
|
218
|
-
except: # noqa: E722
|
|
219
|
-
pass
|
|
220
|
-
return cast(Union[List[float], None, Unset], data)
|
|
221
|
-
|
|
222
|
-
reranker_embedding = _parse_reranker_embedding(d.pop("reranker_embedding", UNSET))
|
|
223
|
-
|
|
224
|
-
labeled_memory_lookup = cls(
|
|
225
|
-
value=value,
|
|
226
|
-
embedding=embedding,
|
|
227
|
-
label=label,
|
|
228
|
-
label_name=label_name,
|
|
229
|
-
source_id=source_id,
|
|
230
|
-
metadata=metadata,
|
|
231
|
-
memory_id=memory_id,
|
|
232
|
-
memory_version=memory_version,
|
|
233
|
-
created_at=created_at,
|
|
234
|
-
updated_at=updated_at,
|
|
235
|
-
metrics=metrics,
|
|
236
|
-
lookup_score=lookup_score,
|
|
237
|
-
reranker_score=reranker_score,
|
|
238
|
-
reranker_embedding=reranker_embedding,
|
|
239
|
-
)
|
|
240
|
-
|
|
241
|
-
labeled_memory_lookup.additional_properties = d
|
|
242
|
-
return labeled_memory_lookup
|
|
243
|
-
|
|
244
|
-
@property
|
|
245
|
-
def additional_keys(self) -> list[str]:
|
|
246
|
-
return list(self.additional_properties.keys())
|
|
247
|
-
|
|
248
|
-
def __getitem__(self, key: str) -> Any:
|
|
249
|
-
return self.additional_properties[key]
|
|
250
|
-
|
|
251
|
-
def __setitem__(self, key: str, value: Any) -> None:
|
|
252
|
-
self.additional_properties[key] = value
|
|
253
|
-
|
|
254
|
-
def __delitem__(self, key: str) -> None:
|
|
255
|
-
del self.additional_properties[key]
|
|
256
|
-
|
|
257
|
-
def __contains__(self, key: str) -> bool:
|
|
258
|
-
return key in self.additional_properties
|
|
@@ -1,68 +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/model.py.jinja
|
|
6
|
-
|
|
7
|
-
The main change is:
|
|
8
|
-
- Fix typing issues
|
|
9
|
-
"""
|
|
10
|
-
|
|
11
|
-
# flake8: noqa: C901
|
|
12
|
-
|
|
13
|
-
from typing import Any, Type, TypeVar, Union, cast
|
|
14
|
-
|
|
15
|
-
from attrs import define as _attrs_define
|
|
16
|
-
from attrs import field as _attrs_field
|
|
17
|
-
|
|
18
|
-
T = TypeVar("T", bound="LabeledMemoryLookupMetadata")
|
|
19
|
-
|
|
20
|
-
|
|
21
|
-
@_attrs_define
|
|
22
|
-
class LabeledMemoryLookupMetadata:
|
|
23
|
-
""" """
|
|
24
|
-
|
|
25
|
-
additional_properties: dict[str, Union[None, bool, float, int, str]] = _attrs_field(init=False, factory=dict)
|
|
26
|
-
|
|
27
|
-
def to_dict(self) -> dict[str, Any]:
|
|
28
|
-
field_dict: dict[str, Any] = {}
|
|
29
|
-
for prop_name, prop in self.additional_properties.items():
|
|
30
|
-
field_dict[prop_name] = prop
|
|
31
|
-
|
|
32
|
-
return field_dict
|
|
33
|
-
|
|
34
|
-
@classmethod
|
|
35
|
-
def from_dict(cls: Type[T], src_dict: dict[str, Any]) -> T:
|
|
36
|
-
d = src_dict.copy()
|
|
37
|
-
labeled_memory_lookup_metadata = cls()
|
|
38
|
-
|
|
39
|
-
additional_properties = {}
|
|
40
|
-
for prop_name, prop_dict in d.items():
|
|
41
|
-
|
|
42
|
-
def _parse_additional_property(data: object) -> Union[None, bool, float, int, str]:
|
|
43
|
-
if data is None:
|
|
44
|
-
return data
|
|
45
|
-
return cast(Union[None, bool, float, int, str], data)
|
|
46
|
-
|
|
47
|
-
additional_property = _parse_additional_property(prop_dict)
|
|
48
|
-
|
|
49
|
-
additional_properties[prop_name] = additional_property
|
|
50
|
-
|
|
51
|
-
labeled_memory_lookup_metadata.additional_properties = additional_properties
|
|
52
|
-
return labeled_memory_lookup_metadata
|
|
53
|
-
|
|
54
|
-
@property
|
|
55
|
-
def additional_keys(self) -> list[str]:
|
|
56
|
-
return list(self.additional_properties.keys())
|
|
57
|
-
|
|
58
|
-
def __getitem__(self, key: str) -> Union[None, bool, float, int, str]:
|
|
59
|
-
return self.additional_properties[key]
|
|
60
|
-
|
|
61
|
-
def __setitem__(self, key: str, value: Union[None, bool, float, int, str]) -> None:
|
|
62
|
-
self.additional_properties[key] = value
|
|
63
|
-
|
|
64
|
-
def __delitem__(self, key: str) -> None:
|
|
65
|
-
del self.additional_properties[key]
|
|
66
|
-
|
|
67
|
-
def __contains__(self, key: str) -> bool:
|
|
68
|
-
return key in self.additional_properties
|
|
@@ -1,68 +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/model.py.jinja
|
|
6
|
-
|
|
7
|
-
The main change is:
|
|
8
|
-
- Fix typing issues
|
|
9
|
-
"""
|
|
10
|
-
|
|
11
|
-
# flake8: noqa: C901
|
|
12
|
-
|
|
13
|
-
from typing import Any, Type, TypeVar, Union, cast
|
|
14
|
-
|
|
15
|
-
from attrs import define as _attrs_define
|
|
16
|
-
from attrs import field as _attrs_field
|
|
17
|
-
|
|
18
|
-
T = TypeVar("T", bound="LabeledMemoryMetadata")
|
|
19
|
-
|
|
20
|
-
|
|
21
|
-
@_attrs_define
|
|
22
|
-
class LabeledMemoryMetadata:
|
|
23
|
-
""" """
|
|
24
|
-
|
|
25
|
-
additional_properties: dict[str, Union[None, bool, float, int, str]] = _attrs_field(init=False, factory=dict)
|
|
26
|
-
|
|
27
|
-
def to_dict(self) -> dict[str, Any]:
|
|
28
|
-
field_dict: dict[str, Any] = {}
|
|
29
|
-
for prop_name, prop in self.additional_properties.items():
|
|
30
|
-
field_dict[prop_name] = prop
|
|
31
|
-
|
|
32
|
-
return field_dict
|
|
33
|
-
|
|
34
|
-
@classmethod
|
|
35
|
-
def from_dict(cls: Type[T], src_dict: dict[str, Any]) -> T:
|
|
36
|
-
d = src_dict.copy()
|
|
37
|
-
labeled_memory_metadata = cls()
|
|
38
|
-
|
|
39
|
-
additional_properties = {}
|
|
40
|
-
for prop_name, prop_dict in d.items():
|
|
41
|
-
|
|
42
|
-
def _parse_additional_property(data: object) -> Union[None, bool, float, int, str]:
|
|
43
|
-
if data is None:
|
|
44
|
-
return data
|
|
45
|
-
return cast(Union[None, bool, float, int, str], data)
|
|
46
|
-
|
|
47
|
-
additional_property = _parse_additional_property(prop_dict)
|
|
48
|
-
|
|
49
|
-
additional_properties[prop_name] = additional_property
|
|
50
|
-
|
|
51
|
-
labeled_memory_metadata.additional_properties = additional_properties
|
|
52
|
-
return labeled_memory_metadata
|
|
53
|
-
|
|
54
|
-
@property
|
|
55
|
-
def additional_keys(self) -> list[str]:
|
|
56
|
-
return list(self.additional_properties.keys())
|
|
57
|
-
|
|
58
|
-
def __getitem__(self, key: str) -> Union[None, bool, float, int, str]:
|
|
59
|
-
return self.additional_properties[key]
|
|
60
|
-
|
|
61
|
-
def __setitem__(self, key: str, value: Union[None, bool, float, int, str]) -> None:
|
|
62
|
-
self.additional_properties[key] = value
|
|
63
|
-
|
|
64
|
-
def __delitem__(self, key: str) -> None:
|
|
65
|
-
del self.additional_properties[key]
|
|
66
|
-
|
|
67
|
-
def __contains__(self, key: str) -> bool:
|
|
68
|
-
return key in self.additional_properties
|