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.
Files changed (188) hide show
  1. orca_sdk/__init__.py +24 -0
  2. orca_sdk/_generated_api_client/__init__.py +3 -0
  3. orca_sdk/_generated_api_client/api/__init__.py +205 -0
  4. orca_sdk/_generated_api_client/api/auth/__init__.py +0 -0
  5. orca_sdk/_generated_api_client/api/auth/check_authentication_auth_get.py +130 -0
  6. orca_sdk/_generated_api_client/api/auth/create_api_key_auth_api_key_post.py +172 -0
  7. orca_sdk/_generated_api_client/api/auth/delete_api_key_auth_api_key_name_or_id_delete.py +158 -0
  8. orca_sdk/_generated_api_client/api/auth/delete_org_auth_org_delete.py +132 -0
  9. orca_sdk/_generated_api_client/api/auth/list_api_keys_auth_api_key_get.py +129 -0
  10. orca_sdk/_generated_api_client/api/classification_model/__init__.py +0 -0
  11. orca_sdk/_generated_api_client/api/classification_model/create_evaluation_classification_model_model_name_or_id_evaluation_post.py +185 -0
  12. orca_sdk/_generated_api_client/api/classification_model/create_model_classification_model_post.py +172 -0
  13. orca_sdk/_generated_api_client/api/classification_model/delete_evaluation_classification_model_model_name_or_id_evaluation_task_id_delete.py +170 -0
  14. orca_sdk/_generated_api_client/api/classification_model/delete_model_classification_model_name_or_id_delete.py +156 -0
  15. orca_sdk/_generated_api_client/api/classification_model/get_evaluation_classification_model_model_name_or_id_evaluation_task_id_get.py +172 -0
  16. orca_sdk/_generated_api_client/api/classification_model/get_model_classification_model_name_or_id_get.py +158 -0
  17. orca_sdk/_generated_api_client/api/classification_model/list_evaluations_classification_model_model_name_or_id_evaluation_get.py +163 -0
  18. orca_sdk/_generated_api_client/api/classification_model/list_models_classification_model_get.py +129 -0
  19. orca_sdk/_generated_api_client/api/classification_model/predict_gpu_classification_model_name_or_id_prediction_post.py +192 -0
  20. orca_sdk/_generated_api_client/api/datasource/__init__.py +0 -0
  21. orca_sdk/_generated_api_client/api/datasource/create_datasource_datasource_post.py +169 -0
  22. orca_sdk/_generated_api_client/api/datasource/create_embedding_evaluation_datasource_name_or_id_embedding_evaluation_post.py +185 -0
  23. orca_sdk/_generated_api_client/api/datasource/delete_datasource_datasource_name_or_id_delete.py +158 -0
  24. orca_sdk/_generated_api_client/api/datasource/get_datasource_datasource_name_or_id_get.py +158 -0
  25. orca_sdk/_generated_api_client/api/datasource/get_embedding_evaluation_datasource_name_or_id_embedding_evaluation_task_id_get.py +171 -0
  26. orca_sdk/_generated_api_client/api/datasource/list_datasources_datasource_get.py +129 -0
  27. orca_sdk/_generated_api_client/api/datasource/list_embedding_evaluations_datasource_name_or_id_embedding_evaluation_get.py +237 -0
  28. orca_sdk/_generated_api_client/api/default/__init__.py +0 -0
  29. orca_sdk/_generated_api_client/api/default/healthcheck_get.py +120 -0
  30. orca_sdk/_generated_api_client/api/default/healthcheck_gpu_get.py +120 -0
  31. orca_sdk/_generated_api_client/api/finetuned_embedding_model/__init__.py +0 -0
  32. orca_sdk/_generated_api_client/api/finetuned_embedding_model/create_finetuned_embedding_model_finetuned_embedding_model_post.py +170 -0
  33. orca_sdk/_generated_api_client/api/finetuned_embedding_model/delete_finetuned_embedding_model_finetuned_embedding_model_name_or_id_delete.py +158 -0
  34. 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
  35. orca_sdk/_generated_api_client/api/finetuned_embedding_model/get_finetuned_embedding_model_finetuned_embedding_model_name_or_id_get.py +158 -0
  36. orca_sdk/_generated_api_client/api/finetuned_embedding_model/list_finetuned_embedding_models_finetuned_embedding_model_get.py +129 -0
  37. orca_sdk/_generated_api_client/api/memoryset/__init__.py +0 -0
  38. orca_sdk/_generated_api_client/api/memoryset/clone_memoryset_memoryset_name_or_id_clone_post.py +183 -0
  39. orca_sdk/_generated_api_client/api/memoryset/create_analysis_memoryset_name_or_id_analysis_post.py +185 -0
  40. orca_sdk/_generated_api_client/api/memoryset/create_memoryset_memoryset_post.py +170 -0
  41. orca_sdk/_generated_api_client/api/memoryset/delete_memories_memoryset_name_or_id_memories_delete_post.py +183 -0
  42. orca_sdk/_generated_api_client/api/memoryset/delete_memory_memoryset_name_or_id_memory_memory_id_delete.py +169 -0
  43. orca_sdk/_generated_api_client/api/memoryset/delete_memoryset_memoryset_name_or_id_delete.py +158 -0
  44. orca_sdk/_generated_api_client/api/memoryset/get_analysis_memoryset_name_or_id_analysis_analysis_task_id_get.py +171 -0
  45. orca_sdk/_generated_api_client/api/memoryset/get_memories_memoryset_name_or_id_memories_get_post.py +190 -0
  46. orca_sdk/_generated_api_client/api/memoryset/get_memory_memoryset_name_or_id_memory_memory_id_get.py +171 -0
  47. orca_sdk/_generated_api_client/api/memoryset/get_memoryset_memoryset_name_or_id_get.py +158 -0
  48. orca_sdk/_generated_api_client/api/memoryset/insert_memories_gpu_memoryset_name_or_id_memory_post.py +186 -0
  49. orca_sdk/_generated_api_client/api/memoryset/list_analyses_memoryset_name_or_id_analysis_get.py +262 -0
  50. orca_sdk/_generated_api_client/api/memoryset/list_memorysets_memoryset_get.py +129 -0
  51. orca_sdk/_generated_api_client/api/memoryset/memoryset_lookup_gpu_memoryset_name_or_id_lookup_post.py +195 -0
  52. orca_sdk/_generated_api_client/api/memoryset/query_memoryset_memoryset_name_or_id_memories_post.py +190 -0
  53. orca_sdk/_generated_api_client/api/memoryset/update_memories_gpu_memoryset_name_or_id_memories_patch.py +193 -0
  54. orca_sdk/_generated_api_client/api/memoryset/update_memory_gpu_memoryset_name_or_id_memory_patch.py +189 -0
  55. orca_sdk/_generated_api_client/api/pretrained_embedding_model/__init__.py +0 -0
  56. orca_sdk/_generated_api_client/api/pretrained_embedding_model/embed_with_pretrained_model_gpu_pretrained_embedding_model_model_name_embedding_post.py +194 -0
  57. orca_sdk/_generated_api_client/api/pretrained_embedding_model/get_pretrained_embedding_model_pretrained_embedding_model_model_name_get.py +163 -0
  58. orca_sdk/_generated_api_client/api/pretrained_embedding_model/list_pretrained_embedding_models_pretrained_embedding_model_get.py +129 -0
  59. orca_sdk/_generated_api_client/api/task/__init__.py +0 -0
  60. orca_sdk/_generated_api_client/api/task/abort_task_task_task_id_abort_delete.py +156 -0
  61. orca_sdk/_generated_api_client/api/task/get_task_status_task_task_id_status_get.py +158 -0
  62. orca_sdk/_generated_api_client/api/task/list_tasks_task_get.py +245 -0
  63. orca_sdk/_generated_api_client/api/telemetry/__init__.py +0 -0
  64. orca_sdk/_generated_api_client/api/telemetry/drop_feedback_category_with_data_telemetry_feedback_category_name_or_id_delete.py +164 -0
  65. orca_sdk/_generated_api_client/api/telemetry/get_feedback_category_telemetry_feedback_category_name_or_id_get.py +158 -0
  66. orca_sdk/_generated_api_client/api/telemetry/get_prediction_telemetry_prediction_prediction_id_get.py +159 -0
  67. orca_sdk/_generated_api_client/api/telemetry/list_feedback_categories_telemetry_feedback_category_get.py +129 -0
  68. orca_sdk/_generated_api_client/api/telemetry/list_predictions_telemetry_prediction_post.py +177 -0
  69. orca_sdk/_generated_api_client/api/telemetry/record_prediction_feedback_telemetry_prediction_feedback_put.py +173 -0
  70. orca_sdk/_generated_api_client/api/telemetry/update_prediction_telemetry_prediction_prediction_id_patch.py +183 -0
  71. orca_sdk/_generated_api_client/client.py +216 -0
  72. orca_sdk/_generated_api_client/errors.py +38 -0
  73. orca_sdk/_generated_api_client/models/__init__.py +179 -0
  74. orca_sdk/_generated_api_client/models/analyze_neighbor_labels_result.py +116 -0
  75. orca_sdk/_generated_api_client/models/api_key_metadata.py +137 -0
  76. orca_sdk/_generated_api_client/models/api_key_metadata_scope_item.py +9 -0
  77. orca_sdk/_generated_api_client/models/base_model.py +55 -0
  78. orca_sdk/_generated_api_client/models/body_create_datasource_datasource_post.py +176 -0
  79. orca_sdk/_generated_api_client/models/classification_evaluation_result.py +147 -0
  80. orca_sdk/_generated_api_client/models/clone_labeled_memoryset_request.py +150 -0
  81. orca_sdk/_generated_api_client/models/column_info.py +114 -0
  82. orca_sdk/_generated_api_client/models/column_type.py +14 -0
  83. orca_sdk/_generated_api_client/models/conflict_error_response.py +80 -0
  84. orca_sdk/_generated_api_client/models/create_api_key_request.py +120 -0
  85. orca_sdk/_generated_api_client/models/create_api_key_request_scope_item.py +9 -0
  86. orca_sdk/_generated_api_client/models/create_api_key_response.py +145 -0
  87. orca_sdk/_generated_api_client/models/create_api_key_response_scope_item.py +9 -0
  88. orca_sdk/_generated_api_client/models/create_labeled_memoryset_request.py +279 -0
  89. orca_sdk/_generated_api_client/models/create_rac_model_request.py +209 -0
  90. orca_sdk/_generated_api_client/models/datasource_metadata.py +142 -0
  91. orca_sdk/_generated_api_client/models/delete_memories_request.py +70 -0
  92. orca_sdk/_generated_api_client/models/embed_request.py +127 -0
  93. orca_sdk/_generated_api_client/models/embedding_evaluation_request.py +179 -0
  94. orca_sdk/_generated_api_client/models/embedding_evaluation_response.py +148 -0
  95. orca_sdk/_generated_api_client/models/embedding_evaluation_result.py +86 -0
  96. orca_sdk/_generated_api_client/models/embedding_finetuning_method.py +9 -0
  97. orca_sdk/_generated_api_client/models/embedding_model_result.py +114 -0
  98. orca_sdk/_generated_api_client/models/evaluation_request.py +180 -0
  99. orca_sdk/_generated_api_client/models/evaluation_response.py +140 -0
  100. orca_sdk/_generated_api_client/models/feedback_type.py +9 -0
  101. orca_sdk/_generated_api_client/models/field_validation_error.py +103 -0
  102. orca_sdk/_generated_api_client/models/filter_item.py +231 -0
  103. orca_sdk/_generated_api_client/models/filter_item_field_type_0_item.py +15 -0
  104. orca_sdk/_generated_api_client/models/filter_item_field_type_2_item_type_1.py +20 -0
  105. orca_sdk/_generated_api_client/models/filter_item_op.py +16 -0
  106. orca_sdk/_generated_api_client/models/find_duplicates_analysis_result.py +70 -0
  107. orca_sdk/_generated_api_client/models/finetune_embedding_model_request.py +259 -0
  108. orca_sdk/_generated_api_client/models/finetune_embedding_model_request_training_args.py +66 -0
  109. orca_sdk/_generated_api_client/models/finetuned_embedding_model_metadata.py +166 -0
  110. orca_sdk/_generated_api_client/models/get_memories_request.py +70 -0
  111. orca_sdk/_generated_api_client/models/internal_server_error_response.py +80 -0
  112. orca_sdk/_generated_api_client/models/label_class_metrics.py +108 -0
  113. orca_sdk/_generated_api_client/models/label_prediction_memory_lookup.py +274 -0
  114. orca_sdk/_generated_api_client/models/label_prediction_memory_lookup_metadata.py +68 -0
  115. orca_sdk/_generated_api_client/models/label_prediction_result.py +115 -0
  116. orca_sdk/_generated_api_client/models/label_prediction_with_memories_and_feedback.py +246 -0
  117. orca_sdk/_generated_api_client/models/labeled_memory.py +197 -0
  118. orca_sdk/_generated_api_client/models/labeled_memory_insert.py +128 -0
  119. orca_sdk/_generated_api_client/models/labeled_memory_insert_metadata.py +68 -0
  120. orca_sdk/_generated_api_client/models/labeled_memory_lookup.py +258 -0
  121. orca_sdk/_generated_api_client/models/labeled_memory_lookup_metadata.py +68 -0
  122. orca_sdk/_generated_api_client/models/labeled_memory_metadata.py +68 -0
  123. orca_sdk/_generated_api_client/models/labeled_memory_metrics.py +237 -0
  124. orca_sdk/_generated_api_client/models/labeled_memory_update.py +171 -0
  125. orca_sdk/_generated_api_client/models/labeled_memory_update_metadata_type_0.py +68 -0
  126. orca_sdk/_generated_api_client/models/labeled_memoryset_metadata.py +195 -0
  127. orca_sdk/_generated_api_client/models/list_analyses_memoryset_name_or_id_analysis_get_type_type_0.py +9 -0
  128. orca_sdk/_generated_api_client/models/list_memories_request.py +104 -0
  129. orca_sdk/_generated_api_client/models/list_predictions_request.py +257 -0
  130. orca_sdk/_generated_api_client/models/lookup_request.py +81 -0
  131. orca_sdk/_generated_api_client/models/memory_metrics.py +156 -0
  132. orca_sdk/_generated_api_client/models/memoryset_analysis_request.py +83 -0
  133. orca_sdk/_generated_api_client/models/memoryset_analysis_request_type.py +9 -0
  134. orca_sdk/_generated_api_client/models/memoryset_analysis_response.py +180 -0
  135. orca_sdk/_generated_api_client/models/memoryset_analysis_response_config.py +66 -0
  136. orca_sdk/_generated_api_client/models/memoryset_analysis_response_type.py +9 -0
  137. orca_sdk/_generated_api_client/models/not_found_error_response.py +100 -0
  138. orca_sdk/_generated_api_client/models/not_found_error_response_resource_type_0.py +21 -0
  139. orca_sdk/_generated_api_client/models/precision_recall_curve.py +94 -0
  140. orca_sdk/_generated_api_client/models/prediction_feedback.py +157 -0
  141. orca_sdk/_generated_api_client/models/prediction_feedback_category.py +115 -0
  142. orca_sdk/_generated_api_client/models/prediction_feedback_request.py +122 -0
  143. orca_sdk/_generated_api_client/models/prediction_feedback_result.py +102 -0
  144. orca_sdk/_generated_api_client/models/prediction_request.py +169 -0
  145. orca_sdk/_generated_api_client/models/prediction_sort_item_item_type_0.py +10 -0
  146. orca_sdk/_generated_api_client/models/prediction_sort_item_item_type_1.py +9 -0
  147. orca_sdk/_generated_api_client/models/pretrained_embedding_model_metadata.py +97 -0
  148. orca_sdk/_generated_api_client/models/pretrained_embedding_model_name.py +12 -0
  149. orca_sdk/_generated_api_client/models/rac_head_type.py +11 -0
  150. orca_sdk/_generated_api_client/models/rac_model_metadata.py +191 -0
  151. orca_sdk/_generated_api_client/models/roc_curve.py +94 -0
  152. orca_sdk/_generated_api_client/models/service_unavailable_error_response.py +80 -0
  153. orca_sdk/_generated_api_client/models/task.py +198 -0
  154. orca_sdk/_generated_api_client/models/task_status.py +14 -0
  155. orca_sdk/_generated_api_client/models/task_status_info.py +133 -0
  156. orca_sdk/_generated_api_client/models/unauthenticated_error_response.py +72 -0
  157. orca_sdk/_generated_api_client/models/unauthorized_error_response.py +80 -0
  158. orca_sdk/_generated_api_client/models/unprocessable_input_error_response.py +94 -0
  159. orca_sdk/_generated_api_client/models/update_prediction_request.py +93 -0
  160. orca_sdk/_generated_api_client/py.typed +1 -0
  161. orca_sdk/_generated_api_client/types.py +56 -0
  162. orca_sdk/_utils/__init__.py +0 -0
  163. orca_sdk/_utils/analysis_ui.py +192 -0
  164. orca_sdk/_utils/analysis_ui_style.css +54 -0
  165. orca_sdk/_utils/auth.py +68 -0
  166. orca_sdk/_utils/auth_test.py +31 -0
  167. orca_sdk/_utils/common.py +37 -0
  168. orca_sdk/_utils/data_parsing.py +99 -0
  169. orca_sdk/_utils/data_parsing_test.py +244 -0
  170. orca_sdk/_utils/prediction_result_ui.css +18 -0
  171. orca_sdk/_utils/prediction_result_ui.py +64 -0
  172. orca_sdk/_utils/task.py +73 -0
  173. orca_sdk/classification_model.py +508 -0
  174. orca_sdk/classification_model_test.py +272 -0
  175. orca_sdk/conftest.py +116 -0
  176. orca_sdk/credentials.py +126 -0
  177. orca_sdk/credentials_test.py +37 -0
  178. orca_sdk/datasource.py +333 -0
  179. orca_sdk/datasource_test.py +96 -0
  180. orca_sdk/embedding_model.py +347 -0
  181. orca_sdk/embedding_model_test.py +176 -0
  182. orca_sdk/memoryset.py +1209 -0
  183. orca_sdk/memoryset_test.py +287 -0
  184. orca_sdk/telemetry.py +398 -0
  185. orca_sdk/telemetry_test.py +109 -0
  186. orca_sdk-0.0.78.dist-info/METADATA +79 -0
  187. orca_sdk-0.0.78.dist-info/RECORD +188 -0
  188. 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
@@ -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)