orca-sdk 0.1.1__py3-none-any.whl → 0.1.3__py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (186) hide show
  1. orca_sdk/__init__.py +10 -4
  2. orca_sdk/_shared/__init__.py +10 -0
  3. orca_sdk/_shared/metrics.py +393 -0
  4. orca_sdk/_shared/metrics_test.py +273 -0
  5. orca_sdk/_utils/analysis_ui.py +12 -10
  6. orca_sdk/_utils/analysis_ui_style.css +0 -3
  7. orca_sdk/_utils/auth.py +31 -29
  8. orca_sdk/_utils/data_parsing.py +28 -2
  9. orca_sdk/_utils/data_parsing_test.py +15 -15
  10. orca_sdk/_utils/pagination.py +126 -0
  11. orca_sdk/_utils/pagination_test.py +132 -0
  12. orca_sdk/_utils/prediction_result_ui.py +67 -21
  13. orca_sdk/_utils/tqdm_file_reader.py +12 -0
  14. orca_sdk/_utils/value_parser.py +45 -0
  15. orca_sdk/_utils/value_parser_test.py +39 -0
  16. orca_sdk/async_client.py +3795 -0
  17. orca_sdk/classification_model.py +601 -129
  18. orca_sdk/classification_model_test.py +415 -117
  19. orca_sdk/client.py +3787 -0
  20. orca_sdk/conftest.py +184 -38
  21. orca_sdk/credentials.py +162 -20
  22. orca_sdk/credentials_test.py +100 -16
  23. orca_sdk/datasource.py +268 -68
  24. orca_sdk/datasource_test.py +266 -18
  25. orca_sdk/embedding_model.py +434 -82
  26. orca_sdk/embedding_model_test.py +66 -33
  27. orca_sdk/job.py +343 -0
  28. orca_sdk/job_test.py +108 -0
  29. orca_sdk/memoryset.py +1690 -324
  30. orca_sdk/memoryset_test.py +456 -119
  31. orca_sdk/regression_model.py +694 -0
  32. orca_sdk/regression_model_test.py +378 -0
  33. orca_sdk/telemetry.py +460 -143
  34. orca_sdk/telemetry_test.py +43 -24
  35. {orca_sdk-0.1.1.dist-info → orca_sdk-0.1.3.dist-info}/METADATA +34 -16
  36. orca_sdk-0.1.3.dist-info/RECORD +41 -0
  37. {orca_sdk-0.1.1.dist-info → orca_sdk-0.1.3.dist-info}/WHEEL +1 -1
  38. orca_sdk/_generated_api_client/__init__.py +0 -3
  39. orca_sdk/_generated_api_client/api/__init__.py +0 -193
  40. orca_sdk/_generated_api_client/api/auth/__init__.py +0 -0
  41. orca_sdk/_generated_api_client/api/auth/check_authentication_auth_get.py +0 -128
  42. orca_sdk/_generated_api_client/api/auth/create_api_key_auth_api_key_post.py +0 -170
  43. orca_sdk/_generated_api_client/api/auth/delete_api_key_auth_api_key_name_or_id_delete.py +0 -156
  44. orca_sdk/_generated_api_client/api/auth/delete_org_auth_org_delete.py +0 -130
  45. orca_sdk/_generated_api_client/api/auth/list_api_keys_auth_api_key_get.py +0 -127
  46. orca_sdk/_generated_api_client/api/classification_model/__init__.py +0 -0
  47. orca_sdk/_generated_api_client/api/classification_model/create_evaluation_classification_model_model_name_or_id_evaluation_post.py +0 -183
  48. orca_sdk/_generated_api_client/api/classification_model/create_model_classification_model_post.py +0 -170
  49. orca_sdk/_generated_api_client/api/classification_model/delete_evaluation_classification_model_model_name_or_id_evaluation_task_id_delete.py +0 -168
  50. orca_sdk/_generated_api_client/api/classification_model/delete_model_classification_model_name_or_id_delete.py +0 -154
  51. orca_sdk/_generated_api_client/api/classification_model/get_evaluation_classification_model_model_name_or_id_evaluation_task_id_get.py +0 -170
  52. orca_sdk/_generated_api_client/api/classification_model/get_model_classification_model_name_or_id_get.py +0 -156
  53. orca_sdk/_generated_api_client/api/classification_model/list_evaluations_classification_model_model_name_or_id_evaluation_get.py +0 -161
  54. orca_sdk/_generated_api_client/api/classification_model/list_models_classification_model_get.py +0 -127
  55. orca_sdk/_generated_api_client/api/classification_model/predict_gpu_classification_model_name_or_id_prediction_post.py +0 -190
  56. orca_sdk/_generated_api_client/api/datasource/__init__.py +0 -0
  57. orca_sdk/_generated_api_client/api/datasource/create_datasource_datasource_post.py +0 -167
  58. orca_sdk/_generated_api_client/api/datasource/delete_datasource_datasource_name_or_id_delete.py +0 -156
  59. orca_sdk/_generated_api_client/api/datasource/get_datasource_datasource_name_or_id_get.py +0 -156
  60. orca_sdk/_generated_api_client/api/datasource/list_datasources_datasource_get.py +0 -127
  61. orca_sdk/_generated_api_client/api/default/__init__.py +0 -0
  62. orca_sdk/_generated_api_client/api/default/healthcheck_get.py +0 -118
  63. orca_sdk/_generated_api_client/api/default/healthcheck_gpu_get.py +0 -118
  64. orca_sdk/_generated_api_client/api/finetuned_embedding_model/__init__.py +0 -0
  65. orca_sdk/_generated_api_client/api/finetuned_embedding_model/create_finetuned_embedding_model_finetuned_embedding_model_post.py +0 -168
  66. orca_sdk/_generated_api_client/api/finetuned_embedding_model/delete_finetuned_embedding_model_finetuned_embedding_model_name_or_id_delete.py +0 -156
  67. orca_sdk/_generated_api_client/api/finetuned_embedding_model/embed_with_finetuned_model_gpu_finetuned_embedding_model_name_or_id_embedding_post.py +0 -189
  68. orca_sdk/_generated_api_client/api/finetuned_embedding_model/get_finetuned_embedding_model_finetuned_embedding_model_name_or_id_get.py +0 -156
  69. orca_sdk/_generated_api_client/api/finetuned_embedding_model/list_finetuned_embedding_models_finetuned_embedding_model_get.py +0 -127
  70. orca_sdk/_generated_api_client/api/memoryset/__init__.py +0 -0
  71. orca_sdk/_generated_api_client/api/memoryset/clone_memoryset_memoryset_name_or_id_clone_post.py +0 -181
  72. orca_sdk/_generated_api_client/api/memoryset/create_analysis_memoryset_name_or_id_analysis_post.py +0 -183
  73. orca_sdk/_generated_api_client/api/memoryset/create_memoryset_memoryset_post.py +0 -168
  74. orca_sdk/_generated_api_client/api/memoryset/delete_memories_memoryset_name_or_id_memories_delete_post.py +0 -181
  75. orca_sdk/_generated_api_client/api/memoryset/delete_memory_memoryset_name_or_id_memory_memory_id_delete.py +0 -167
  76. orca_sdk/_generated_api_client/api/memoryset/delete_memoryset_memoryset_name_or_id_delete.py +0 -156
  77. orca_sdk/_generated_api_client/api/memoryset/get_analysis_memoryset_name_or_id_analysis_analysis_task_id_get.py +0 -169
  78. orca_sdk/_generated_api_client/api/memoryset/get_memories_memoryset_name_or_id_memories_get_post.py +0 -188
  79. orca_sdk/_generated_api_client/api/memoryset/get_memory_memoryset_name_or_id_memory_memory_id_get.py +0 -169
  80. orca_sdk/_generated_api_client/api/memoryset/get_memoryset_memoryset_name_or_id_get.py +0 -156
  81. orca_sdk/_generated_api_client/api/memoryset/insert_memories_gpu_memoryset_name_or_id_memory_post.py +0 -184
  82. orca_sdk/_generated_api_client/api/memoryset/list_analyses_memoryset_name_or_id_analysis_get.py +0 -260
  83. orca_sdk/_generated_api_client/api/memoryset/list_memorysets_memoryset_get.py +0 -127
  84. orca_sdk/_generated_api_client/api/memoryset/memoryset_lookup_gpu_memoryset_name_or_id_lookup_post.py +0 -193
  85. orca_sdk/_generated_api_client/api/memoryset/query_memoryset_memoryset_name_or_id_memories_post.py +0 -188
  86. orca_sdk/_generated_api_client/api/memoryset/update_memories_gpu_memoryset_name_or_id_memories_patch.py +0 -191
  87. orca_sdk/_generated_api_client/api/memoryset/update_memory_gpu_memoryset_name_or_id_memory_patch.py +0 -187
  88. orca_sdk/_generated_api_client/api/pretrained_embedding_model/__init__.py +0 -0
  89. orca_sdk/_generated_api_client/api/pretrained_embedding_model/embed_with_pretrained_model_gpu_pretrained_embedding_model_model_name_embedding_post.py +0 -188
  90. orca_sdk/_generated_api_client/api/pretrained_embedding_model/get_pretrained_embedding_model_pretrained_embedding_model_model_name_get.py +0 -157
  91. orca_sdk/_generated_api_client/api/pretrained_embedding_model/list_pretrained_embedding_models_pretrained_embedding_model_get.py +0 -127
  92. orca_sdk/_generated_api_client/api/task/__init__.py +0 -0
  93. orca_sdk/_generated_api_client/api/task/abort_task_task_task_id_abort_delete.py +0 -154
  94. orca_sdk/_generated_api_client/api/task/get_task_status_task_task_id_status_get.py +0 -156
  95. orca_sdk/_generated_api_client/api/task/list_tasks_task_get.py +0 -243
  96. orca_sdk/_generated_api_client/api/telemetry/__init__.py +0 -0
  97. orca_sdk/_generated_api_client/api/telemetry/drop_feedback_category_with_data_telemetry_feedback_category_name_or_id_delete.py +0 -162
  98. orca_sdk/_generated_api_client/api/telemetry/get_feedback_category_telemetry_feedback_category_name_or_id_get.py +0 -156
  99. orca_sdk/_generated_api_client/api/telemetry/get_prediction_telemetry_prediction_prediction_id_get.py +0 -157
  100. orca_sdk/_generated_api_client/api/telemetry/list_feedback_categories_telemetry_feedback_category_get.py +0 -127
  101. orca_sdk/_generated_api_client/api/telemetry/list_predictions_telemetry_prediction_post.py +0 -175
  102. orca_sdk/_generated_api_client/api/telemetry/record_prediction_feedback_telemetry_prediction_feedback_put.py +0 -171
  103. orca_sdk/_generated_api_client/api/telemetry/update_prediction_telemetry_prediction_prediction_id_patch.py +0 -181
  104. orca_sdk/_generated_api_client/client.py +0 -216
  105. orca_sdk/_generated_api_client/errors.py +0 -38
  106. orca_sdk/_generated_api_client/models/__init__.py +0 -159
  107. orca_sdk/_generated_api_client/models/analyze_neighbor_labels_result.py +0 -84
  108. orca_sdk/_generated_api_client/models/api_key_metadata.py +0 -118
  109. orca_sdk/_generated_api_client/models/base_model.py +0 -55
  110. orca_sdk/_generated_api_client/models/body_create_datasource_datasource_post.py +0 -176
  111. orca_sdk/_generated_api_client/models/classification_evaluation_result.py +0 -114
  112. orca_sdk/_generated_api_client/models/clone_labeled_memoryset_request.py +0 -150
  113. orca_sdk/_generated_api_client/models/column_info.py +0 -114
  114. orca_sdk/_generated_api_client/models/column_type.py +0 -14
  115. orca_sdk/_generated_api_client/models/conflict_error_response.py +0 -80
  116. orca_sdk/_generated_api_client/models/create_api_key_request.py +0 -99
  117. orca_sdk/_generated_api_client/models/create_api_key_response.py +0 -126
  118. orca_sdk/_generated_api_client/models/create_labeled_memoryset_request.py +0 -259
  119. orca_sdk/_generated_api_client/models/create_rac_model_request.py +0 -209
  120. orca_sdk/_generated_api_client/models/datasource_metadata.py +0 -142
  121. orca_sdk/_generated_api_client/models/delete_memories_request.py +0 -70
  122. orca_sdk/_generated_api_client/models/embed_request.py +0 -127
  123. orca_sdk/_generated_api_client/models/embedding_finetuning_method.py +0 -9
  124. orca_sdk/_generated_api_client/models/evaluation_request.py +0 -180
  125. orca_sdk/_generated_api_client/models/evaluation_response.py +0 -140
  126. orca_sdk/_generated_api_client/models/feedback_type.py +0 -9
  127. orca_sdk/_generated_api_client/models/field_validation_error.py +0 -103
  128. orca_sdk/_generated_api_client/models/filter_item.py +0 -231
  129. orca_sdk/_generated_api_client/models/filter_item_field_type_0_item.py +0 -15
  130. orca_sdk/_generated_api_client/models/filter_item_field_type_2_item_type_1.py +0 -16
  131. orca_sdk/_generated_api_client/models/filter_item_op.py +0 -16
  132. orca_sdk/_generated_api_client/models/find_duplicates_analysis_result.py +0 -70
  133. orca_sdk/_generated_api_client/models/finetune_embedding_model_request.py +0 -259
  134. orca_sdk/_generated_api_client/models/finetune_embedding_model_request_training_args.py +0 -66
  135. orca_sdk/_generated_api_client/models/finetuned_embedding_model_metadata.py +0 -166
  136. orca_sdk/_generated_api_client/models/get_memories_request.py +0 -70
  137. orca_sdk/_generated_api_client/models/internal_server_error_response.py +0 -80
  138. orca_sdk/_generated_api_client/models/label_class_metrics.py +0 -108
  139. orca_sdk/_generated_api_client/models/label_prediction_memory_lookup.py +0 -274
  140. orca_sdk/_generated_api_client/models/label_prediction_memory_lookup_metadata.py +0 -68
  141. orca_sdk/_generated_api_client/models/label_prediction_result.py +0 -101
  142. orca_sdk/_generated_api_client/models/label_prediction_with_memories_and_feedback.py +0 -232
  143. orca_sdk/_generated_api_client/models/labeled_memory.py +0 -197
  144. orca_sdk/_generated_api_client/models/labeled_memory_insert.py +0 -108
  145. orca_sdk/_generated_api_client/models/labeled_memory_insert_metadata.py +0 -68
  146. orca_sdk/_generated_api_client/models/labeled_memory_lookup.py +0 -258
  147. orca_sdk/_generated_api_client/models/labeled_memory_lookup_metadata.py +0 -68
  148. orca_sdk/_generated_api_client/models/labeled_memory_metadata.py +0 -68
  149. orca_sdk/_generated_api_client/models/labeled_memory_metrics.py +0 -277
  150. orca_sdk/_generated_api_client/models/labeled_memory_update.py +0 -171
  151. orca_sdk/_generated_api_client/models/labeled_memory_update_metadata_type_0.py +0 -68
  152. orca_sdk/_generated_api_client/models/labeled_memoryset_metadata.py +0 -195
  153. orca_sdk/_generated_api_client/models/list_analyses_memoryset_name_or_id_analysis_get_type_type_0.py +0 -9
  154. orca_sdk/_generated_api_client/models/list_memories_request.py +0 -104
  155. orca_sdk/_generated_api_client/models/list_predictions_request.py +0 -234
  156. orca_sdk/_generated_api_client/models/list_predictions_request_sort_item_item_type_0.py +0 -9
  157. orca_sdk/_generated_api_client/models/list_predictions_request_sort_item_item_type_1.py +0 -9
  158. orca_sdk/_generated_api_client/models/lookup_request.py +0 -81
  159. orca_sdk/_generated_api_client/models/memoryset_analysis_request.py +0 -83
  160. orca_sdk/_generated_api_client/models/memoryset_analysis_request_type.py +0 -9
  161. orca_sdk/_generated_api_client/models/memoryset_analysis_response.py +0 -180
  162. orca_sdk/_generated_api_client/models/memoryset_analysis_response_config.py +0 -66
  163. orca_sdk/_generated_api_client/models/memoryset_analysis_response_type.py +0 -9
  164. orca_sdk/_generated_api_client/models/not_found_error_response.py +0 -100
  165. orca_sdk/_generated_api_client/models/not_found_error_response_resource_type_0.py +0 -20
  166. orca_sdk/_generated_api_client/models/prediction_feedback.py +0 -157
  167. orca_sdk/_generated_api_client/models/prediction_feedback_category.py +0 -115
  168. orca_sdk/_generated_api_client/models/prediction_feedback_request.py +0 -122
  169. orca_sdk/_generated_api_client/models/prediction_feedback_result.py +0 -102
  170. orca_sdk/_generated_api_client/models/prediction_request.py +0 -169
  171. orca_sdk/_generated_api_client/models/pretrained_embedding_model_metadata.py +0 -97
  172. orca_sdk/_generated_api_client/models/pretrained_embedding_model_name.py +0 -11
  173. orca_sdk/_generated_api_client/models/rac_head_type.py +0 -11
  174. orca_sdk/_generated_api_client/models/rac_model_metadata.py +0 -191
  175. orca_sdk/_generated_api_client/models/service_unavailable_error_response.py +0 -80
  176. orca_sdk/_generated_api_client/models/task.py +0 -198
  177. orca_sdk/_generated_api_client/models/task_status.py +0 -14
  178. orca_sdk/_generated_api_client/models/task_status_info.py +0 -133
  179. orca_sdk/_generated_api_client/models/unauthenticated_error_response.py +0 -72
  180. orca_sdk/_generated_api_client/models/unauthorized_error_response.py +0 -80
  181. orca_sdk/_generated_api_client/models/unprocessable_input_error_response.py +0 -94
  182. orca_sdk/_generated_api_client/models/update_prediction_request.py +0 -93
  183. orca_sdk/_generated_api_client/py.typed +0 -1
  184. orca_sdk/_generated_api_client/types.py +0 -56
  185. orca_sdk/_utils/task.py +0 -73
  186. orca_sdk-0.1.1.dist-info/RECORD +0 -175
@@ -1,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