orca-sdk 0.0.78__py3-none-any.whl
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- orca_sdk/__init__.py +24 -0
- orca_sdk/_generated_api_client/__init__.py +3 -0
- orca_sdk/_generated_api_client/api/__init__.py +205 -0
- orca_sdk/_generated_api_client/api/auth/__init__.py +0 -0
- orca_sdk/_generated_api_client/api/auth/check_authentication_auth_get.py +130 -0
- orca_sdk/_generated_api_client/api/auth/create_api_key_auth_api_key_post.py +172 -0
- orca_sdk/_generated_api_client/api/auth/delete_api_key_auth_api_key_name_or_id_delete.py +158 -0
- orca_sdk/_generated_api_client/api/auth/delete_org_auth_org_delete.py +132 -0
- orca_sdk/_generated_api_client/api/auth/list_api_keys_auth_api_key_get.py +129 -0
- 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 +185 -0
- orca_sdk/_generated_api_client/api/classification_model/create_model_classification_model_post.py +172 -0
- orca_sdk/_generated_api_client/api/classification_model/delete_evaluation_classification_model_model_name_or_id_evaluation_task_id_delete.py +170 -0
- orca_sdk/_generated_api_client/api/classification_model/delete_model_classification_model_name_or_id_delete.py +156 -0
- orca_sdk/_generated_api_client/api/classification_model/get_evaluation_classification_model_model_name_or_id_evaluation_task_id_get.py +172 -0
- orca_sdk/_generated_api_client/api/classification_model/get_model_classification_model_name_or_id_get.py +158 -0
- orca_sdk/_generated_api_client/api/classification_model/list_evaluations_classification_model_model_name_or_id_evaluation_get.py +163 -0
- orca_sdk/_generated_api_client/api/classification_model/list_models_classification_model_get.py +129 -0
- orca_sdk/_generated_api_client/api/classification_model/predict_gpu_classification_model_name_or_id_prediction_post.py +192 -0
- orca_sdk/_generated_api_client/api/datasource/__init__.py +0 -0
- orca_sdk/_generated_api_client/api/datasource/create_datasource_datasource_post.py +169 -0
- orca_sdk/_generated_api_client/api/datasource/create_embedding_evaluation_datasource_name_or_id_embedding_evaluation_post.py +185 -0
- orca_sdk/_generated_api_client/api/datasource/delete_datasource_datasource_name_or_id_delete.py +158 -0
- orca_sdk/_generated_api_client/api/datasource/get_datasource_datasource_name_or_id_get.py +158 -0
- orca_sdk/_generated_api_client/api/datasource/get_embedding_evaluation_datasource_name_or_id_embedding_evaluation_task_id_get.py +171 -0
- orca_sdk/_generated_api_client/api/datasource/list_datasources_datasource_get.py +129 -0
- orca_sdk/_generated_api_client/api/datasource/list_embedding_evaluations_datasource_name_or_id_embedding_evaluation_get.py +237 -0
- orca_sdk/_generated_api_client/api/default/__init__.py +0 -0
- orca_sdk/_generated_api_client/api/default/healthcheck_get.py +120 -0
- orca_sdk/_generated_api_client/api/default/healthcheck_gpu_get.py +120 -0
- 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 +170 -0
- orca_sdk/_generated_api_client/api/finetuned_embedding_model/delete_finetuned_embedding_model_finetuned_embedding_model_name_or_id_delete.py +158 -0
- orca_sdk/_generated_api_client/api/finetuned_embedding_model/embed_with_finetuned_model_gpu_finetuned_embedding_model_name_or_id_embedding_post.py +191 -0
- orca_sdk/_generated_api_client/api/finetuned_embedding_model/get_finetuned_embedding_model_finetuned_embedding_model_name_or_id_get.py +158 -0
- orca_sdk/_generated_api_client/api/finetuned_embedding_model/list_finetuned_embedding_models_finetuned_embedding_model_get.py +129 -0
- 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 +183 -0
- orca_sdk/_generated_api_client/api/memoryset/create_analysis_memoryset_name_or_id_analysis_post.py +185 -0
- orca_sdk/_generated_api_client/api/memoryset/create_memoryset_memoryset_post.py +170 -0
- orca_sdk/_generated_api_client/api/memoryset/delete_memories_memoryset_name_or_id_memories_delete_post.py +183 -0
- orca_sdk/_generated_api_client/api/memoryset/delete_memory_memoryset_name_or_id_memory_memory_id_delete.py +169 -0
- orca_sdk/_generated_api_client/api/memoryset/delete_memoryset_memoryset_name_or_id_delete.py +158 -0
- orca_sdk/_generated_api_client/api/memoryset/get_analysis_memoryset_name_or_id_analysis_analysis_task_id_get.py +171 -0
- orca_sdk/_generated_api_client/api/memoryset/get_memories_memoryset_name_or_id_memories_get_post.py +190 -0
- orca_sdk/_generated_api_client/api/memoryset/get_memory_memoryset_name_or_id_memory_memory_id_get.py +171 -0
- orca_sdk/_generated_api_client/api/memoryset/get_memoryset_memoryset_name_or_id_get.py +158 -0
- orca_sdk/_generated_api_client/api/memoryset/insert_memories_gpu_memoryset_name_or_id_memory_post.py +186 -0
- orca_sdk/_generated_api_client/api/memoryset/list_analyses_memoryset_name_or_id_analysis_get.py +262 -0
- orca_sdk/_generated_api_client/api/memoryset/list_memorysets_memoryset_get.py +129 -0
- orca_sdk/_generated_api_client/api/memoryset/memoryset_lookup_gpu_memoryset_name_or_id_lookup_post.py +195 -0
- orca_sdk/_generated_api_client/api/memoryset/query_memoryset_memoryset_name_or_id_memories_post.py +190 -0
- orca_sdk/_generated_api_client/api/memoryset/update_memories_gpu_memoryset_name_or_id_memories_patch.py +193 -0
- orca_sdk/_generated_api_client/api/memoryset/update_memory_gpu_memoryset_name_or_id_memory_patch.py +189 -0
- 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 +194 -0
- orca_sdk/_generated_api_client/api/pretrained_embedding_model/get_pretrained_embedding_model_pretrained_embedding_model_model_name_get.py +163 -0
- orca_sdk/_generated_api_client/api/pretrained_embedding_model/list_pretrained_embedding_models_pretrained_embedding_model_get.py +129 -0
- 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 +156 -0
- orca_sdk/_generated_api_client/api/task/get_task_status_task_task_id_status_get.py +158 -0
- orca_sdk/_generated_api_client/api/task/list_tasks_task_get.py +245 -0
- 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 +164 -0
- orca_sdk/_generated_api_client/api/telemetry/get_feedback_category_telemetry_feedback_category_name_or_id_get.py +158 -0
- orca_sdk/_generated_api_client/api/telemetry/get_prediction_telemetry_prediction_prediction_id_get.py +159 -0
- orca_sdk/_generated_api_client/api/telemetry/list_feedback_categories_telemetry_feedback_category_get.py +129 -0
- orca_sdk/_generated_api_client/api/telemetry/list_predictions_telemetry_prediction_post.py +177 -0
- orca_sdk/_generated_api_client/api/telemetry/record_prediction_feedback_telemetry_prediction_feedback_put.py +173 -0
- orca_sdk/_generated_api_client/api/telemetry/update_prediction_telemetry_prediction_prediction_id_patch.py +183 -0
- orca_sdk/_generated_api_client/client.py +216 -0
- orca_sdk/_generated_api_client/errors.py +38 -0
- orca_sdk/_generated_api_client/models/__init__.py +179 -0
- orca_sdk/_generated_api_client/models/analyze_neighbor_labels_result.py +116 -0
- orca_sdk/_generated_api_client/models/api_key_metadata.py +137 -0
- orca_sdk/_generated_api_client/models/api_key_metadata_scope_item.py +9 -0
- orca_sdk/_generated_api_client/models/base_model.py +55 -0
- orca_sdk/_generated_api_client/models/body_create_datasource_datasource_post.py +176 -0
- orca_sdk/_generated_api_client/models/classification_evaluation_result.py +147 -0
- orca_sdk/_generated_api_client/models/clone_labeled_memoryset_request.py +150 -0
- orca_sdk/_generated_api_client/models/column_info.py +114 -0
- orca_sdk/_generated_api_client/models/column_type.py +14 -0
- orca_sdk/_generated_api_client/models/conflict_error_response.py +80 -0
- orca_sdk/_generated_api_client/models/create_api_key_request.py +120 -0
- orca_sdk/_generated_api_client/models/create_api_key_request_scope_item.py +9 -0
- orca_sdk/_generated_api_client/models/create_api_key_response.py +145 -0
- orca_sdk/_generated_api_client/models/create_api_key_response_scope_item.py +9 -0
- orca_sdk/_generated_api_client/models/create_labeled_memoryset_request.py +279 -0
- orca_sdk/_generated_api_client/models/create_rac_model_request.py +209 -0
- orca_sdk/_generated_api_client/models/datasource_metadata.py +142 -0
- orca_sdk/_generated_api_client/models/delete_memories_request.py +70 -0
- orca_sdk/_generated_api_client/models/embed_request.py +127 -0
- orca_sdk/_generated_api_client/models/embedding_evaluation_request.py +179 -0
- orca_sdk/_generated_api_client/models/embedding_evaluation_response.py +148 -0
- orca_sdk/_generated_api_client/models/embedding_evaluation_result.py +86 -0
- orca_sdk/_generated_api_client/models/embedding_finetuning_method.py +9 -0
- orca_sdk/_generated_api_client/models/embedding_model_result.py +114 -0
- orca_sdk/_generated_api_client/models/evaluation_request.py +180 -0
- orca_sdk/_generated_api_client/models/evaluation_response.py +140 -0
- orca_sdk/_generated_api_client/models/feedback_type.py +9 -0
- orca_sdk/_generated_api_client/models/field_validation_error.py +103 -0
- orca_sdk/_generated_api_client/models/filter_item.py +231 -0
- orca_sdk/_generated_api_client/models/filter_item_field_type_0_item.py +15 -0
- orca_sdk/_generated_api_client/models/filter_item_field_type_2_item_type_1.py +20 -0
- orca_sdk/_generated_api_client/models/filter_item_op.py +16 -0
- orca_sdk/_generated_api_client/models/find_duplicates_analysis_result.py +70 -0
- orca_sdk/_generated_api_client/models/finetune_embedding_model_request.py +259 -0
- orca_sdk/_generated_api_client/models/finetune_embedding_model_request_training_args.py +66 -0
- orca_sdk/_generated_api_client/models/finetuned_embedding_model_metadata.py +166 -0
- orca_sdk/_generated_api_client/models/get_memories_request.py +70 -0
- orca_sdk/_generated_api_client/models/internal_server_error_response.py +80 -0
- orca_sdk/_generated_api_client/models/label_class_metrics.py +108 -0
- orca_sdk/_generated_api_client/models/label_prediction_memory_lookup.py +274 -0
- orca_sdk/_generated_api_client/models/label_prediction_memory_lookup_metadata.py +68 -0
- orca_sdk/_generated_api_client/models/label_prediction_result.py +115 -0
- orca_sdk/_generated_api_client/models/label_prediction_with_memories_and_feedback.py +246 -0
- orca_sdk/_generated_api_client/models/labeled_memory.py +197 -0
- orca_sdk/_generated_api_client/models/labeled_memory_insert.py +128 -0
- orca_sdk/_generated_api_client/models/labeled_memory_insert_metadata.py +68 -0
- orca_sdk/_generated_api_client/models/labeled_memory_lookup.py +258 -0
- orca_sdk/_generated_api_client/models/labeled_memory_lookup_metadata.py +68 -0
- orca_sdk/_generated_api_client/models/labeled_memory_metadata.py +68 -0
- orca_sdk/_generated_api_client/models/labeled_memory_metrics.py +237 -0
- orca_sdk/_generated_api_client/models/labeled_memory_update.py +171 -0
- orca_sdk/_generated_api_client/models/labeled_memory_update_metadata_type_0.py +68 -0
- orca_sdk/_generated_api_client/models/labeled_memoryset_metadata.py +195 -0
- orca_sdk/_generated_api_client/models/list_analyses_memoryset_name_or_id_analysis_get_type_type_0.py +9 -0
- orca_sdk/_generated_api_client/models/list_memories_request.py +104 -0
- orca_sdk/_generated_api_client/models/list_predictions_request.py +257 -0
- orca_sdk/_generated_api_client/models/lookup_request.py +81 -0
- orca_sdk/_generated_api_client/models/memory_metrics.py +156 -0
- orca_sdk/_generated_api_client/models/memoryset_analysis_request.py +83 -0
- orca_sdk/_generated_api_client/models/memoryset_analysis_request_type.py +9 -0
- orca_sdk/_generated_api_client/models/memoryset_analysis_response.py +180 -0
- orca_sdk/_generated_api_client/models/memoryset_analysis_response_config.py +66 -0
- orca_sdk/_generated_api_client/models/memoryset_analysis_response_type.py +9 -0
- orca_sdk/_generated_api_client/models/not_found_error_response.py +100 -0
- orca_sdk/_generated_api_client/models/not_found_error_response_resource_type_0.py +21 -0
- orca_sdk/_generated_api_client/models/precision_recall_curve.py +94 -0
- orca_sdk/_generated_api_client/models/prediction_feedback.py +157 -0
- orca_sdk/_generated_api_client/models/prediction_feedback_category.py +115 -0
- orca_sdk/_generated_api_client/models/prediction_feedback_request.py +122 -0
- orca_sdk/_generated_api_client/models/prediction_feedback_result.py +102 -0
- orca_sdk/_generated_api_client/models/prediction_request.py +169 -0
- orca_sdk/_generated_api_client/models/prediction_sort_item_item_type_0.py +10 -0
- orca_sdk/_generated_api_client/models/prediction_sort_item_item_type_1.py +9 -0
- orca_sdk/_generated_api_client/models/pretrained_embedding_model_metadata.py +97 -0
- orca_sdk/_generated_api_client/models/pretrained_embedding_model_name.py +12 -0
- orca_sdk/_generated_api_client/models/rac_head_type.py +11 -0
- orca_sdk/_generated_api_client/models/rac_model_metadata.py +191 -0
- orca_sdk/_generated_api_client/models/roc_curve.py +94 -0
- orca_sdk/_generated_api_client/models/service_unavailable_error_response.py +80 -0
- orca_sdk/_generated_api_client/models/task.py +198 -0
- orca_sdk/_generated_api_client/models/task_status.py +14 -0
- orca_sdk/_generated_api_client/models/task_status_info.py +133 -0
- orca_sdk/_generated_api_client/models/unauthenticated_error_response.py +72 -0
- orca_sdk/_generated_api_client/models/unauthorized_error_response.py +80 -0
- orca_sdk/_generated_api_client/models/unprocessable_input_error_response.py +94 -0
- orca_sdk/_generated_api_client/models/update_prediction_request.py +93 -0
- orca_sdk/_generated_api_client/py.typed +1 -0
- orca_sdk/_generated_api_client/types.py +56 -0
- orca_sdk/_utils/__init__.py +0 -0
- orca_sdk/_utils/analysis_ui.py +192 -0
- orca_sdk/_utils/analysis_ui_style.css +54 -0
- orca_sdk/_utils/auth.py +68 -0
- orca_sdk/_utils/auth_test.py +31 -0
- orca_sdk/_utils/common.py +37 -0
- orca_sdk/_utils/data_parsing.py +99 -0
- orca_sdk/_utils/data_parsing_test.py +244 -0
- orca_sdk/_utils/prediction_result_ui.css +18 -0
- orca_sdk/_utils/prediction_result_ui.py +64 -0
- orca_sdk/_utils/task.py +73 -0
- orca_sdk/classification_model.py +508 -0
- orca_sdk/classification_model_test.py +272 -0
- orca_sdk/conftest.py +116 -0
- orca_sdk/credentials.py +126 -0
- orca_sdk/credentials_test.py +37 -0
- orca_sdk/datasource.py +333 -0
- orca_sdk/datasource_test.py +96 -0
- orca_sdk/embedding_model.py +347 -0
- orca_sdk/embedding_model_test.py +176 -0
- orca_sdk/memoryset.py +1209 -0
- orca_sdk/memoryset_test.py +287 -0
- orca_sdk/telemetry.py +398 -0
- orca_sdk/telemetry_test.py +109 -0
- orca_sdk-0.0.78.dist-info/METADATA +79 -0
- orca_sdk-0.0.78.dist-info/RECORD +188 -0
- orca_sdk-0.0.78.dist-info/WHEEL +4 -0
|
@@ -0,0 +1,237 @@
|
|
|
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, List, Type, TypeVar, Union, cast
|
|
14
|
+
|
|
15
|
+
from attrs import define as _attrs_define
|
|
16
|
+
from attrs import field as _attrs_field
|
|
17
|
+
|
|
18
|
+
from ..types import UNSET, Unset
|
|
19
|
+
|
|
20
|
+
T = TypeVar("T", bound="LabeledMemoryMetrics")
|
|
21
|
+
|
|
22
|
+
|
|
23
|
+
@_attrs_define
|
|
24
|
+
class LabeledMemoryMetrics:
|
|
25
|
+
"""Metrics computed for a labeled memory.
|
|
26
|
+
|
|
27
|
+
Attributes:
|
|
28
|
+
is_duplicate (Union[Unset, bool]):
|
|
29
|
+
duplicate_memory_ids (Union[Unset, List[str]]):
|
|
30
|
+
has_possible_duplicates (Union[Unset, bool]):
|
|
31
|
+
possible_duplicate_memory_ids (Union[List[str], None, Unset]):
|
|
32
|
+
cluster (Union[Unset, int]):
|
|
33
|
+
embedding_2d (Union[Unset, List[float]]):
|
|
34
|
+
neighbor_label_logits (Union[Unset, List[float]]):
|
|
35
|
+
neighbor_predicted_label (Union[Unset, int]):
|
|
36
|
+
neighbor_predicted_label_ambiguity (Union[Unset, float]):
|
|
37
|
+
neighbor_predicted_label_confidence (Union[Unset, float]):
|
|
38
|
+
current_label_neighbor_confidence (Union[Unset, float]):
|
|
39
|
+
normalized_neighbor_label_entropy (Union[Unset, float]):
|
|
40
|
+
neighbor_predicted_label_matches_current_label (Union[None, Unset, bool]):
|
|
41
|
+
"""
|
|
42
|
+
|
|
43
|
+
is_duplicate: Union[Unset, bool] = UNSET
|
|
44
|
+
duplicate_memory_ids: Union[Unset, List[str]] = UNSET
|
|
45
|
+
has_possible_duplicates: Union[Unset, bool] = UNSET
|
|
46
|
+
possible_duplicate_memory_ids: Union[List[str], None, Unset] = UNSET
|
|
47
|
+
cluster: Union[Unset, int] = UNSET
|
|
48
|
+
embedding_2d: Union[Unset, List[float]] = UNSET
|
|
49
|
+
neighbor_label_logits: Union[Unset, List[float]] = UNSET
|
|
50
|
+
neighbor_predicted_label: Union[Unset, int] = UNSET
|
|
51
|
+
neighbor_predicted_label_ambiguity: Union[Unset, float] = UNSET
|
|
52
|
+
neighbor_predicted_label_confidence: Union[Unset, float] = UNSET
|
|
53
|
+
current_label_neighbor_confidence: Union[Unset, float] = UNSET
|
|
54
|
+
normalized_neighbor_label_entropy: Union[Unset, float] = UNSET
|
|
55
|
+
neighbor_predicted_label_matches_current_label: Union[None, Unset, bool] = UNSET
|
|
56
|
+
additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
|
|
57
|
+
|
|
58
|
+
def to_dict(self) -> dict[str, Any]:
|
|
59
|
+
is_duplicate = self.is_duplicate
|
|
60
|
+
|
|
61
|
+
duplicate_memory_ids: Union[Unset, List[str]] = UNSET
|
|
62
|
+
if not isinstance(self.duplicate_memory_ids, Unset):
|
|
63
|
+
duplicate_memory_ids = self.duplicate_memory_ids
|
|
64
|
+
|
|
65
|
+
has_possible_duplicates = self.has_possible_duplicates
|
|
66
|
+
|
|
67
|
+
possible_duplicate_memory_ids: Union[List[str], None, Unset]
|
|
68
|
+
if isinstance(self.possible_duplicate_memory_ids, Unset):
|
|
69
|
+
possible_duplicate_memory_ids = UNSET
|
|
70
|
+
elif isinstance(self.possible_duplicate_memory_ids, list):
|
|
71
|
+
possible_duplicate_memory_ids = self.possible_duplicate_memory_ids
|
|
72
|
+
|
|
73
|
+
else:
|
|
74
|
+
possible_duplicate_memory_ids = self.possible_duplicate_memory_ids
|
|
75
|
+
|
|
76
|
+
cluster = self.cluster
|
|
77
|
+
|
|
78
|
+
embedding_2d: Union[Unset, List[float]] = UNSET
|
|
79
|
+
if not isinstance(self.embedding_2d, Unset):
|
|
80
|
+
embedding_2d = []
|
|
81
|
+
for embedding_2d_item_data in self.embedding_2d:
|
|
82
|
+
embedding_2d_item: float
|
|
83
|
+
embedding_2d_item = embedding_2d_item_data
|
|
84
|
+
embedding_2d.append(embedding_2d_item)
|
|
85
|
+
|
|
86
|
+
neighbor_label_logits: Union[Unset, List[float]] = UNSET
|
|
87
|
+
if not isinstance(self.neighbor_label_logits, Unset):
|
|
88
|
+
neighbor_label_logits = self.neighbor_label_logits
|
|
89
|
+
|
|
90
|
+
neighbor_predicted_label = self.neighbor_predicted_label
|
|
91
|
+
|
|
92
|
+
neighbor_predicted_label_ambiguity = self.neighbor_predicted_label_ambiguity
|
|
93
|
+
|
|
94
|
+
neighbor_predicted_label_confidence = self.neighbor_predicted_label_confidence
|
|
95
|
+
|
|
96
|
+
current_label_neighbor_confidence = self.current_label_neighbor_confidence
|
|
97
|
+
|
|
98
|
+
normalized_neighbor_label_entropy = self.normalized_neighbor_label_entropy
|
|
99
|
+
|
|
100
|
+
neighbor_predicted_label_matches_current_label: Union[None, Unset, bool]
|
|
101
|
+
if isinstance(self.neighbor_predicted_label_matches_current_label, Unset):
|
|
102
|
+
neighbor_predicted_label_matches_current_label = UNSET
|
|
103
|
+
else:
|
|
104
|
+
neighbor_predicted_label_matches_current_label = self.neighbor_predicted_label_matches_current_label
|
|
105
|
+
|
|
106
|
+
field_dict: dict[str, Any] = {}
|
|
107
|
+
field_dict.update(self.additional_properties)
|
|
108
|
+
field_dict.update({})
|
|
109
|
+
if is_duplicate is not UNSET:
|
|
110
|
+
field_dict["is_duplicate"] = is_duplicate
|
|
111
|
+
if duplicate_memory_ids is not UNSET:
|
|
112
|
+
field_dict["duplicate_memory_ids"] = duplicate_memory_ids
|
|
113
|
+
if has_possible_duplicates is not UNSET:
|
|
114
|
+
field_dict["has_possible_duplicates"] = has_possible_duplicates
|
|
115
|
+
if possible_duplicate_memory_ids is not UNSET:
|
|
116
|
+
field_dict["possible_duplicate_memory_ids"] = possible_duplicate_memory_ids
|
|
117
|
+
if cluster is not UNSET:
|
|
118
|
+
field_dict["cluster"] = cluster
|
|
119
|
+
if embedding_2d is not UNSET:
|
|
120
|
+
field_dict["embedding_2d"] = embedding_2d
|
|
121
|
+
if neighbor_label_logits is not UNSET:
|
|
122
|
+
field_dict["neighbor_label_logits"] = neighbor_label_logits
|
|
123
|
+
if neighbor_predicted_label is not UNSET:
|
|
124
|
+
field_dict["neighbor_predicted_label"] = neighbor_predicted_label
|
|
125
|
+
if neighbor_predicted_label_ambiguity is not UNSET:
|
|
126
|
+
field_dict["neighbor_predicted_label_ambiguity"] = neighbor_predicted_label_ambiguity
|
|
127
|
+
if neighbor_predicted_label_confidence is not UNSET:
|
|
128
|
+
field_dict["neighbor_predicted_label_confidence"] = neighbor_predicted_label_confidence
|
|
129
|
+
if current_label_neighbor_confidence is not UNSET:
|
|
130
|
+
field_dict["current_label_neighbor_confidence"] = current_label_neighbor_confidence
|
|
131
|
+
if normalized_neighbor_label_entropy is not UNSET:
|
|
132
|
+
field_dict["normalized_neighbor_label_entropy"] = normalized_neighbor_label_entropy
|
|
133
|
+
if neighbor_predicted_label_matches_current_label is not UNSET:
|
|
134
|
+
field_dict["neighbor_predicted_label_matches_current_label"] = (
|
|
135
|
+
neighbor_predicted_label_matches_current_label
|
|
136
|
+
)
|
|
137
|
+
|
|
138
|
+
return field_dict
|
|
139
|
+
|
|
140
|
+
@classmethod
|
|
141
|
+
def from_dict(cls: Type[T], src_dict: dict[str, Any]) -> T:
|
|
142
|
+
d = src_dict.copy()
|
|
143
|
+
is_duplicate = d.pop("is_duplicate", UNSET)
|
|
144
|
+
|
|
145
|
+
duplicate_memory_ids = cast(List[str], d.pop("duplicate_memory_ids", UNSET))
|
|
146
|
+
|
|
147
|
+
has_possible_duplicates = d.pop("has_possible_duplicates", UNSET)
|
|
148
|
+
|
|
149
|
+
def _parse_possible_duplicate_memory_ids(data: object) -> Union[List[str], None, Unset]:
|
|
150
|
+
if data is None:
|
|
151
|
+
return data
|
|
152
|
+
if isinstance(data, Unset):
|
|
153
|
+
return data
|
|
154
|
+
try:
|
|
155
|
+
if not isinstance(data, list):
|
|
156
|
+
raise TypeError()
|
|
157
|
+
possible_duplicate_memory_ids_type_0 = cast(List[str], data)
|
|
158
|
+
|
|
159
|
+
return possible_duplicate_memory_ids_type_0
|
|
160
|
+
except: # noqa: E722
|
|
161
|
+
pass
|
|
162
|
+
return cast(Union[List[str], None, Unset], data)
|
|
163
|
+
|
|
164
|
+
possible_duplicate_memory_ids = _parse_possible_duplicate_memory_ids(
|
|
165
|
+
d.pop("possible_duplicate_memory_ids", UNSET)
|
|
166
|
+
)
|
|
167
|
+
|
|
168
|
+
cluster = d.pop("cluster", UNSET)
|
|
169
|
+
|
|
170
|
+
embedding_2d = []
|
|
171
|
+
_embedding_2d = d.pop("embedding_2d", UNSET)
|
|
172
|
+
for embedding_2d_item_data in _embedding_2d or []:
|
|
173
|
+
|
|
174
|
+
def _parse_embedding_2d_item(data: object) -> float:
|
|
175
|
+
return cast(float, data)
|
|
176
|
+
|
|
177
|
+
embedding_2d_item = _parse_embedding_2d_item(embedding_2d_item_data)
|
|
178
|
+
|
|
179
|
+
embedding_2d.append(embedding_2d_item)
|
|
180
|
+
|
|
181
|
+
neighbor_label_logits = cast(List[float], d.pop("neighbor_label_logits", UNSET))
|
|
182
|
+
|
|
183
|
+
neighbor_predicted_label = d.pop("neighbor_predicted_label", UNSET)
|
|
184
|
+
|
|
185
|
+
neighbor_predicted_label_ambiguity = d.pop("neighbor_predicted_label_ambiguity", UNSET)
|
|
186
|
+
|
|
187
|
+
neighbor_predicted_label_confidence = d.pop("neighbor_predicted_label_confidence", UNSET)
|
|
188
|
+
|
|
189
|
+
current_label_neighbor_confidence = d.pop("current_label_neighbor_confidence", UNSET)
|
|
190
|
+
|
|
191
|
+
normalized_neighbor_label_entropy = d.pop("normalized_neighbor_label_entropy", UNSET)
|
|
192
|
+
|
|
193
|
+
def _parse_neighbor_predicted_label_matches_current_label(data: object) -> Union[None, Unset, bool]:
|
|
194
|
+
if data is None:
|
|
195
|
+
return data
|
|
196
|
+
if isinstance(data, Unset):
|
|
197
|
+
return data
|
|
198
|
+
return cast(Union[None, Unset, bool], data)
|
|
199
|
+
|
|
200
|
+
neighbor_predicted_label_matches_current_label = _parse_neighbor_predicted_label_matches_current_label(
|
|
201
|
+
d.pop("neighbor_predicted_label_matches_current_label", UNSET)
|
|
202
|
+
)
|
|
203
|
+
|
|
204
|
+
labeled_memory_metrics = cls(
|
|
205
|
+
is_duplicate=is_duplicate,
|
|
206
|
+
duplicate_memory_ids=duplicate_memory_ids,
|
|
207
|
+
has_possible_duplicates=has_possible_duplicates,
|
|
208
|
+
possible_duplicate_memory_ids=possible_duplicate_memory_ids,
|
|
209
|
+
cluster=cluster,
|
|
210
|
+
embedding_2d=embedding_2d,
|
|
211
|
+
neighbor_label_logits=neighbor_label_logits,
|
|
212
|
+
neighbor_predicted_label=neighbor_predicted_label,
|
|
213
|
+
neighbor_predicted_label_ambiguity=neighbor_predicted_label_ambiguity,
|
|
214
|
+
neighbor_predicted_label_confidence=neighbor_predicted_label_confidence,
|
|
215
|
+
current_label_neighbor_confidence=current_label_neighbor_confidence,
|
|
216
|
+
normalized_neighbor_label_entropy=normalized_neighbor_label_entropy,
|
|
217
|
+
neighbor_predicted_label_matches_current_label=neighbor_predicted_label_matches_current_label,
|
|
218
|
+
)
|
|
219
|
+
|
|
220
|
+
labeled_memory_metrics.additional_properties = d
|
|
221
|
+
return labeled_memory_metrics
|
|
222
|
+
|
|
223
|
+
@property
|
|
224
|
+
def additional_keys(self) -> list[str]:
|
|
225
|
+
return list(self.additional_properties.keys())
|
|
226
|
+
|
|
227
|
+
def __getitem__(self, key: str) -> Any:
|
|
228
|
+
return self.additional_properties[key]
|
|
229
|
+
|
|
230
|
+
def __setitem__(self, key: str, value: Any) -> None:
|
|
231
|
+
self.additional_properties[key] = value
|
|
232
|
+
|
|
233
|
+
def __delitem__(self, key: str) -> None:
|
|
234
|
+
del self.additional_properties[key]
|
|
235
|
+
|
|
236
|
+
def __contains__(self, key: str) -> bool:
|
|
237
|
+
return key in self.additional_properties
|
|
@@ -0,0 +1,171 @@
|
|
|
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 TYPE_CHECKING, Any, Dict, Type, TypeVar, Union, cast
|
|
14
|
+
|
|
15
|
+
from attrs import define as _attrs_define
|
|
16
|
+
|
|
17
|
+
from ..types import UNSET, Unset
|
|
18
|
+
|
|
19
|
+
if TYPE_CHECKING:
|
|
20
|
+
from ..models.labeled_memory_metrics import LabeledMemoryMetrics
|
|
21
|
+
from ..models.labeled_memory_update_metadata_type_0 import LabeledMemoryUpdateMetadataType0
|
|
22
|
+
|
|
23
|
+
|
|
24
|
+
T = TypeVar("T", bound="LabeledMemoryUpdate")
|
|
25
|
+
|
|
26
|
+
|
|
27
|
+
@_attrs_define
|
|
28
|
+
class LabeledMemoryUpdate:
|
|
29
|
+
"""Model for updating a labeled memory (unset fields are kept as is)
|
|
30
|
+
|
|
31
|
+
Attributes:
|
|
32
|
+
memory_id (str):
|
|
33
|
+
value (Union[Unset, str]):
|
|
34
|
+
metadata (Union['LabeledMemoryUpdateMetadataType0', None, Unset]):
|
|
35
|
+
source_id (Union[None, Unset, str]):
|
|
36
|
+
label (Union[Unset, int]):
|
|
37
|
+
metrics (Union['LabeledMemoryMetrics', None, Unset]):
|
|
38
|
+
"""
|
|
39
|
+
|
|
40
|
+
memory_id: str
|
|
41
|
+
value: Union[Unset, str] = UNSET
|
|
42
|
+
metadata: Union["LabeledMemoryUpdateMetadataType0", None, Unset] = UNSET
|
|
43
|
+
source_id: Union[None, Unset, str] = UNSET
|
|
44
|
+
label: Union[Unset, int] = UNSET
|
|
45
|
+
metrics: Union["LabeledMemoryMetrics", None, Unset] = UNSET
|
|
46
|
+
|
|
47
|
+
def to_dict(self) -> dict[str, Any]:
|
|
48
|
+
from ..models.labeled_memory_metrics import LabeledMemoryMetrics
|
|
49
|
+
from ..models.labeled_memory_update_metadata_type_0 import LabeledMemoryUpdateMetadataType0
|
|
50
|
+
|
|
51
|
+
memory_id = self.memory_id
|
|
52
|
+
|
|
53
|
+
value: Union[Unset, str]
|
|
54
|
+
if isinstance(self.value, Unset):
|
|
55
|
+
value = UNSET
|
|
56
|
+
else:
|
|
57
|
+
value = self.value
|
|
58
|
+
|
|
59
|
+
metadata: Union[Dict[str, Any], None, Unset]
|
|
60
|
+
if isinstance(self.metadata, Unset):
|
|
61
|
+
metadata = UNSET
|
|
62
|
+
elif isinstance(self.metadata, LabeledMemoryUpdateMetadataType0):
|
|
63
|
+
metadata = self.metadata.to_dict()
|
|
64
|
+
else:
|
|
65
|
+
metadata = self.metadata
|
|
66
|
+
|
|
67
|
+
source_id: Union[None, Unset, str]
|
|
68
|
+
if isinstance(self.source_id, Unset):
|
|
69
|
+
source_id = UNSET
|
|
70
|
+
else:
|
|
71
|
+
source_id = self.source_id
|
|
72
|
+
|
|
73
|
+
label = self.label
|
|
74
|
+
|
|
75
|
+
metrics: Union[Dict[str, Any], None, Unset]
|
|
76
|
+
if isinstance(self.metrics, Unset):
|
|
77
|
+
metrics = UNSET
|
|
78
|
+
elif isinstance(self.metrics, LabeledMemoryMetrics):
|
|
79
|
+
metrics = self.metrics.to_dict()
|
|
80
|
+
else:
|
|
81
|
+
metrics = self.metrics
|
|
82
|
+
|
|
83
|
+
field_dict: dict[str, Any] = {}
|
|
84
|
+
field_dict.update(
|
|
85
|
+
{
|
|
86
|
+
"memory_id": memory_id,
|
|
87
|
+
}
|
|
88
|
+
)
|
|
89
|
+
if value is not UNSET:
|
|
90
|
+
field_dict["value"] = value
|
|
91
|
+
if metadata is not UNSET:
|
|
92
|
+
field_dict["metadata"] = metadata
|
|
93
|
+
if source_id is not UNSET:
|
|
94
|
+
field_dict["source_id"] = source_id
|
|
95
|
+
if label is not UNSET:
|
|
96
|
+
field_dict["label"] = label
|
|
97
|
+
if metrics is not UNSET:
|
|
98
|
+
field_dict["metrics"] = metrics
|
|
99
|
+
|
|
100
|
+
return field_dict
|
|
101
|
+
|
|
102
|
+
@classmethod
|
|
103
|
+
def from_dict(cls: Type[T], src_dict: dict[str, Any]) -> T:
|
|
104
|
+
from ..models.labeled_memory_metrics import LabeledMemoryMetrics
|
|
105
|
+
from ..models.labeled_memory_update_metadata_type_0 import LabeledMemoryUpdateMetadataType0
|
|
106
|
+
|
|
107
|
+
d = src_dict.copy()
|
|
108
|
+
memory_id = d.pop("memory_id")
|
|
109
|
+
|
|
110
|
+
def _parse_value(data: object) -> Union[Unset, str]:
|
|
111
|
+
if isinstance(data, Unset):
|
|
112
|
+
return data
|
|
113
|
+
return cast(Union[Unset, str], data)
|
|
114
|
+
|
|
115
|
+
value = _parse_value(d.pop("value", UNSET))
|
|
116
|
+
|
|
117
|
+
def _parse_metadata(data: object) -> Union["LabeledMemoryUpdateMetadataType0", None, Unset]:
|
|
118
|
+
if data is None:
|
|
119
|
+
return data
|
|
120
|
+
if isinstance(data, Unset):
|
|
121
|
+
return data
|
|
122
|
+
try:
|
|
123
|
+
if not isinstance(data, dict):
|
|
124
|
+
raise TypeError()
|
|
125
|
+
metadata_type_0 = LabeledMemoryUpdateMetadataType0.from_dict(data)
|
|
126
|
+
|
|
127
|
+
return metadata_type_0
|
|
128
|
+
except: # noqa: E722
|
|
129
|
+
pass
|
|
130
|
+
return cast(Union["LabeledMemoryUpdateMetadataType0", None, Unset], data)
|
|
131
|
+
|
|
132
|
+
metadata = _parse_metadata(d.pop("metadata", UNSET))
|
|
133
|
+
|
|
134
|
+
def _parse_source_id(data: object) -> Union[None, Unset, str]:
|
|
135
|
+
if data is None:
|
|
136
|
+
return data
|
|
137
|
+
if isinstance(data, Unset):
|
|
138
|
+
return data
|
|
139
|
+
return cast(Union[None, Unset, str], data)
|
|
140
|
+
|
|
141
|
+
source_id = _parse_source_id(d.pop("source_id", UNSET))
|
|
142
|
+
|
|
143
|
+
label = d.pop("label", UNSET)
|
|
144
|
+
|
|
145
|
+
def _parse_metrics(data: object) -> Union["LabeledMemoryMetrics", None, Unset]:
|
|
146
|
+
if data is None:
|
|
147
|
+
return data
|
|
148
|
+
if isinstance(data, Unset):
|
|
149
|
+
return data
|
|
150
|
+
try:
|
|
151
|
+
if not isinstance(data, dict):
|
|
152
|
+
raise TypeError()
|
|
153
|
+
metrics_type_0 = LabeledMemoryMetrics.from_dict(data)
|
|
154
|
+
|
|
155
|
+
return metrics_type_0
|
|
156
|
+
except: # noqa: E722
|
|
157
|
+
pass
|
|
158
|
+
return cast(Union["LabeledMemoryMetrics", None, Unset], data)
|
|
159
|
+
|
|
160
|
+
metrics = _parse_metrics(d.pop("metrics", UNSET))
|
|
161
|
+
|
|
162
|
+
labeled_memory_update = cls(
|
|
163
|
+
memory_id=memory_id,
|
|
164
|
+
value=value,
|
|
165
|
+
metadata=metadata,
|
|
166
|
+
source_id=source_id,
|
|
167
|
+
label=label,
|
|
168
|
+
metrics=metrics,
|
|
169
|
+
)
|
|
170
|
+
|
|
171
|
+
return labeled_memory_update
|
|
@@ -0,0 +1,68 @@
|
|
|
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="LabeledMemoryUpdateMetadataType0")
|
|
19
|
+
|
|
20
|
+
|
|
21
|
+
@_attrs_define
|
|
22
|
+
class LabeledMemoryUpdateMetadataType0:
|
|
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_update_metadata_type_0 = 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_update_metadata_type_0.additional_properties = additional_properties
|
|
52
|
+
return labeled_memory_update_metadata_type_0
|
|
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
|
|
@@ -0,0 +1,195 @@
|
|
|
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 enum import Enum
|
|
15
|
+
from typing import Any, List, Type, TypeVar, Union, cast
|
|
16
|
+
|
|
17
|
+
from attrs import define as _attrs_define
|
|
18
|
+
from attrs import field as _attrs_field
|
|
19
|
+
from dateutil.parser import isoparse
|
|
20
|
+
|
|
21
|
+
from ..models.pretrained_embedding_model_name import PretrainedEmbeddingModelName
|
|
22
|
+
from ..models.task_status import TaskStatus
|
|
23
|
+
|
|
24
|
+
T = TypeVar("T", bound="LabeledMemorysetMetadata")
|
|
25
|
+
|
|
26
|
+
|
|
27
|
+
@_attrs_define
|
|
28
|
+
class LabeledMemorysetMetadata:
|
|
29
|
+
"""
|
|
30
|
+
Attributes:
|
|
31
|
+
id (str):
|
|
32
|
+
org_id (str):
|
|
33
|
+
collection_name (str):
|
|
34
|
+
name (str):
|
|
35
|
+
length (int):
|
|
36
|
+
pretrained_embedding_model_name (Union[None, PretrainedEmbeddingModelName]):
|
|
37
|
+
finetuned_embedding_model_id (Union[None, str]):
|
|
38
|
+
label_names (List[str]):
|
|
39
|
+
created_at (datetime.datetime):
|
|
40
|
+
updated_at (datetime.datetime):
|
|
41
|
+
insertion_task_id (str):
|
|
42
|
+
insertion_status (TaskStatus): Status of task in the task queue
|
|
43
|
+
"""
|
|
44
|
+
|
|
45
|
+
id: str
|
|
46
|
+
org_id: str
|
|
47
|
+
collection_name: str
|
|
48
|
+
name: str
|
|
49
|
+
length: int
|
|
50
|
+
pretrained_embedding_model_name: Union[None, PretrainedEmbeddingModelName]
|
|
51
|
+
finetuned_embedding_model_id: Union[None, str]
|
|
52
|
+
label_names: List[str]
|
|
53
|
+
created_at: datetime.datetime
|
|
54
|
+
updated_at: datetime.datetime
|
|
55
|
+
insertion_task_id: str
|
|
56
|
+
insertion_status: TaskStatus
|
|
57
|
+
additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
|
|
58
|
+
|
|
59
|
+
def to_dict(self) -> dict[str, Any]:
|
|
60
|
+
id = self.id
|
|
61
|
+
|
|
62
|
+
org_id = self.org_id
|
|
63
|
+
|
|
64
|
+
collection_name = self.collection_name
|
|
65
|
+
|
|
66
|
+
name = self.name
|
|
67
|
+
|
|
68
|
+
length = self.length
|
|
69
|
+
|
|
70
|
+
pretrained_embedding_model_name: Union[None, str]
|
|
71
|
+
if isinstance(self.pretrained_embedding_model_name, PretrainedEmbeddingModelName):
|
|
72
|
+
pretrained_embedding_model_name = (
|
|
73
|
+
self.pretrained_embedding_model_name.value
|
|
74
|
+
if isinstance(self.pretrained_embedding_model_name, Enum)
|
|
75
|
+
else self.pretrained_embedding_model_name
|
|
76
|
+
)
|
|
77
|
+
else:
|
|
78
|
+
pretrained_embedding_model_name = self.pretrained_embedding_model_name
|
|
79
|
+
|
|
80
|
+
finetuned_embedding_model_id: Union[None, str]
|
|
81
|
+
finetuned_embedding_model_id = self.finetuned_embedding_model_id
|
|
82
|
+
|
|
83
|
+
label_names = self.label_names
|
|
84
|
+
|
|
85
|
+
created_at = self.created_at.isoformat()
|
|
86
|
+
|
|
87
|
+
updated_at = self.updated_at.isoformat()
|
|
88
|
+
|
|
89
|
+
insertion_task_id = self.insertion_task_id
|
|
90
|
+
|
|
91
|
+
insertion_status = (
|
|
92
|
+
self.insertion_status.value if isinstance(self.insertion_status, Enum) else self.insertion_status
|
|
93
|
+
)
|
|
94
|
+
|
|
95
|
+
field_dict: dict[str, Any] = {}
|
|
96
|
+
field_dict.update(self.additional_properties)
|
|
97
|
+
field_dict.update(
|
|
98
|
+
{
|
|
99
|
+
"id": id,
|
|
100
|
+
"org_id": org_id,
|
|
101
|
+
"collection_name": collection_name,
|
|
102
|
+
"name": name,
|
|
103
|
+
"length": length,
|
|
104
|
+
"pretrained_embedding_model_name": pretrained_embedding_model_name,
|
|
105
|
+
"finetuned_embedding_model_id": finetuned_embedding_model_id,
|
|
106
|
+
"label_names": label_names,
|
|
107
|
+
"created_at": created_at,
|
|
108
|
+
"updated_at": updated_at,
|
|
109
|
+
"insertion_task_id": insertion_task_id,
|
|
110
|
+
"insertion_status": insertion_status,
|
|
111
|
+
}
|
|
112
|
+
)
|
|
113
|
+
|
|
114
|
+
return field_dict
|
|
115
|
+
|
|
116
|
+
@classmethod
|
|
117
|
+
def from_dict(cls: Type[T], src_dict: dict[str, Any]) -> T:
|
|
118
|
+
d = src_dict.copy()
|
|
119
|
+
id = d.pop("id")
|
|
120
|
+
|
|
121
|
+
org_id = d.pop("org_id")
|
|
122
|
+
|
|
123
|
+
collection_name = d.pop("collection_name")
|
|
124
|
+
|
|
125
|
+
name = d.pop("name")
|
|
126
|
+
|
|
127
|
+
length = d.pop("length")
|
|
128
|
+
|
|
129
|
+
def _parse_pretrained_embedding_model_name(data: object) -> Union[None, PretrainedEmbeddingModelName]:
|
|
130
|
+
if data is None:
|
|
131
|
+
return data
|
|
132
|
+
try:
|
|
133
|
+
if not isinstance(data, str):
|
|
134
|
+
raise TypeError()
|
|
135
|
+
pretrained_embedding_model_name_type_0 = PretrainedEmbeddingModelName(data)
|
|
136
|
+
|
|
137
|
+
return pretrained_embedding_model_name_type_0
|
|
138
|
+
except: # noqa: E722
|
|
139
|
+
pass
|
|
140
|
+
return cast(Union[None, PretrainedEmbeddingModelName], data)
|
|
141
|
+
|
|
142
|
+
pretrained_embedding_model_name = _parse_pretrained_embedding_model_name(
|
|
143
|
+
d.pop("pretrained_embedding_model_name")
|
|
144
|
+
)
|
|
145
|
+
|
|
146
|
+
def _parse_finetuned_embedding_model_id(data: object) -> Union[None, str]:
|
|
147
|
+
if data is None:
|
|
148
|
+
return data
|
|
149
|
+
return cast(Union[None, str], data)
|
|
150
|
+
|
|
151
|
+
finetuned_embedding_model_id = _parse_finetuned_embedding_model_id(d.pop("finetuned_embedding_model_id"))
|
|
152
|
+
|
|
153
|
+
label_names = cast(List[str], d.pop("label_names"))
|
|
154
|
+
|
|
155
|
+
created_at = isoparse(d.pop("created_at"))
|
|
156
|
+
|
|
157
|
+
updated_at = isoparse(d.pop("updated_at"))
|
|
158
|
+
|
|
159
|
+
insertion_task_id = d.pop("insertion_task_id")
|
|
160
|
+
|
|
161
|
+
insertion_status = TaskStatus(d.pop("insertion_status"))
|
|
162
|
+
|
|
163
|
+
labeled_memoryset_metadata = cls(
|
|
164
|
+
id=id,
|
|
165
|
+
org_id=org_id,
|
|
166
|
+
collection_name=collection_name,
|
|
167
|
+
name=name,
|
|
168
|
+
length=length,
|
|
169
|
+
pretrained_embedding_model_name=pretrained_embedding_model_name,
|
|
170
|
+
finetuned_embedding_model_id=finetuned_embedding_model_id,
|
|
171
|
+
label_names=label_names,
|
|
172
|
+
created_at=created_at,
|
|
173
|
+
updated_at=updated_at,
|
|
174
|
+
insertion_task_id=insertion_task_id,
|
|
175
|
+
insertion_status=insertion_status,
|
|
176
|
+
)
|
|
177
|
+
|
|
178
|
+
labeled_memoryset_metadata.additional_properties = d
|
|
179
|
+
return labeled_memoryset_metadata
|
|
180
|
+
|
|
181
|
+
@property
|
|
182
|
+
def additional_keys(self) -> list[str]:
|
|
183
|
+
return list(self.additional_properties.keys())
|
|
184
|
+
|
|
185
|
+
def __getitem__(self, key: str) -> Any:
|
|
186
|
+
return self.additional_properties[key]
|
|
187
|
+
|
|
188
|
+
def __setitem__(self, key: str, value: Any) -> None:
|
|
189
|
+
self.additional_properties[key] = value
|
|
190
|
+
|
|
191
|
+
def __delitem__(self, key: str) -> None:
|
|
192
|
+
del self.additional_properties[key]
|
|
193
|
+
|
|
194
|
+
def __contains__(self, key: str) -> bool:
|
|
195
|
+
return key in self.additional_properties
|
orca_sdk/_generated_api_client/models/list_analyses_memoryset_name_or_id_analysis_get_type_type_0.py
ADDED
|
@@ -0,0 +1,9 @@
|
|
|
1
|
+
from enum import Enum
|
|
2
|
+
|
|
3
|
+
|
|
4
|
+
class ListAnalysesMemorysetNameOrIdAnalysisGetTypeType0(str, Enum):
|
|
5
|
+
ANALYZE_DUPLICATE_MEMORIES = "ANALYZE_DUPLICATE_MEMORIES"
|
|
6
|
+
ANALYZE_MEMORY_NEIGHBOR_LABELS = "ANALYZE_MEMORY_NEIGHBOR_LABELS"
|
|
7
|
+
|
|
8
|
+
def __str__(self) -> str:
|
|
9
|
+
return str(self.value)
|