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

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