orca-sdk 0.0.95__py3-none-any.whl → 0.0.97__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 (89) hide show
  1. orca_sdk/__init__.py +1 -5
  2. orca_sdk/_generated_api_client/api/__init__.py +22 -2
  3. orca_sdk/_generated_api_client/api/{datasource/create_datasource_datasource_post.py → auth/create_org_plan_auth_org_plan_post.py} +32 -31
  4. orca_sdk/_generated_api_client/api/auth/get_org_plan_auth_org_plan_get.py +122 -0
  5. orca_sdk/_generated_api_client/api/auth/update_org_plan_auth_org_plan_put.py +168 -0
  6. orca_sdk/_generated_api_client/api/datasource/create_datasource_from_content_datasource_post.py +224 -0
  7. orca_sdk/_generated_api_client/api/datasource/create_datasource_from_files_datasource_upload_post.py +229 -0
  8. orca_sdk/_generated_api_client/api/task/list_tasks_task_get.py +21 -26
  9. orca_sdk/_generated_api_client/api/telemetry/generate_memory_suggestions_telemetry_prediction_prediction_id_memory_suggestions_post.py +239 -0
  10. orca_sdk/_generated_api_client/api/telemetry/get_action_recommendation_telemetry_prediction_prediction_id_action_get.py +192 -0
  11. orca_sdk/_generated_api_client/models/__init__.py +54 -4
  12. orca_sdk/_generated_api_client/models/action_recommendation.py +82 -0
  13. orca_sdk/_generated_api_client/models/action_recommendation_action.py +11 -0
  14. orca_sdk/_generated_api_client/models/add_memory_recommendations.py +85 -0
  15. orca_sdk/_generated_api_client/models/add_memory_suggestion.py +79 -0
  16. orca_sdk/_generated_api_client/models/body_create_datasource_from_files_datasource_upload_post.py +145 -0
  17. orca_sdk/_generated_api_client/models/class_representatives.py +92 -0
  18. orca_sdk/_generated_api_client/models/classification_model_metadata.py +14 -0
  19. orca_sdk/_generated_api_client/models/clone_memoryset_request.py +40 -0
  20. orca_sdk/_generated_api_client/models/constraint_violation_error_response.py +8 -7
  21. orca_sdk/_generated_api_client/models/constraint_violation_error_response_status_code.py +8 -0
  22. orca_sdk/_generated_api_client/models/create_classification_model_request.py +40 -0
  23. orca_sdk/_generated_api_client/models/create_datasource_from_content_request.py +101 -0
  24. orca_sdk/_generated_api_client/models/create_memoryset_request.py +40 -0
  25. orca_sdk/_generated_api_client/models/create_org_plan_request.py +73 -0
  26. orca_sdk/_generated_api_client/models/create_org_plan_request_tier.py +11 -0
  27. orca_sdk/_generated_api_client/models/create_regression_model_request.py +20 -0
  28. orca_sdk/_generated_api_client/models/embed_request.py +20 -0
  29. orca_sdk/_generated_api_client/models/embedding_evaluation_payload.py +28 -10
  30. orca_sdk/_generated_api_client/models/embedding_evaluation_request.py +28 -10
  31. orca_sdk/_generated_api_client/models/embedding_model_result.py +9 -0
  32. orca_sdk/_generated_api_client/models/filter_item.py +31 -23
  33. orca_sdk/_generated_api_client/models/filter_item_field_type_1_item_type_0.py +8 -0
  34. orca_sdk/_generated_api_client/models/filter_item_field_type_2_item_type_0.py +8 -0
  35. orca_sdk/_generated_api_client/models/filter_item_field_type_2_item_type_1.py +2 -0
  36. orca_sdk/_generated_api_client/models/internal_server_error_response.py +8 -7
  37. orca_sdk/_generated_api_client/models/internal_server_error_response_status_code.py +8 -0
  38. orca_sdk/_generated_api_client/models/labeled_memory.py +5 -5
  39. orca_sdk/_generated_api_client/models/labeled_memory_update.py +16 -16
  40. orca_sdk/_generated_api_client/models/labeled_memory_with_feedback_metrics.py +5 -5
  41. orca_sdk/_generated_api_client/models/lookup_request.py +20 -0
  42. orca_sdk/_generated_api_client/models/memory_metrics.py +98 -0
  43. orca_sdk/_generated_api_client/models/memoryset_analysis_configs.py +33 -0
  44. orca_sdk/_generated_api_client/models/memoryset_class_patterns_analysis_config.py +79 -0
  45. orca_sdk/_generated_api_client/models/memoryset_class_patterns_metrics.py +138 -0
  46. orca_sdk/_generated_api_client/models/memoryset_metadata.py +42 -0
  47. orca_sdk/_generated_api_client/models/memoryset_metrics.py +33 -0
  48. orca_sdk/_generated_api_client/models/memoryset_update.py +20 -0
  49. orca_sdk/_generated_api_client/models/not_found_error_response.py +6 -7
  50. orca_sdk/_generated_api_client/models/not_found_error_response_resource_type_0.py +1 -0
  51. orca_sdk/_generated_api_client/models/not_found_error_response_status_code.py +8 -0
  52. orca_sdk/_generated_api_client/models/org_plan.py +99 -0
  53. orca_sdk/_generated_api_client/models/org_plan_tier.py +11 -0
  54. orca_sdk/_generated_api_client/models/paginated_task.py +108 -0
  55. orca_sdk/_generated_api_client/models/predictive_model_update.py +20 -0
  56. orca_sdk/_generated_api_client/models/pretrained_embedding_model_metadata.py +8 -0
  57. orca_sdk/_generated_api_client/models/regression_model_metadata.py +14 -0
  58. orca_sdk/_generated_api_client/models/scored_memory_update.py +9 -9
  59. orca_sdk/_generated_api_client/models/service_unavailable_error_response.py +8 -7
  60. orca_sdk/_generated_api_client/models/service_unavailable_error_response_status_code.py +8 -0
  61. orca_sdk/_generated_api_client/models/telemetry_field_type_0_item_type_0.py +8 -0
  62. orca_sdk/_generated_api_client/models/telemetry_field_type_1_item_type_0.py +8 -0
  63. orca_sdk/_generated_api_client/models/telemetry_field_type_1_item_type_1.py +8 -0
  64. orca_sdk/_generated_api_client/models/telemetry_filter_item.py +42 -30
  65. orca_sdk/_generated_api_client/models/telemetry_sort_options.py +42 -30
  66. orca_sdk/_generated_api_client/models/unauthenticated_error_response.py +8 -7
  67. orca_sdk/_generated_api_client/models/unauthenticated_error_response_status_code.py +8 -0
  68. orca_sdk/_generated_api_client/models/unauthorized_error_response.py +8 -7
  69. orca_sdk/_generated_api_client/models/unauthorized_error_response_status_code.py +8 -0
  70. orca_sdk/_generated_api_client/models/update_org_plan_request.py +73 -0
  71. orca_sdk/_generated_api_client/models/update_org_plan_request_tier.py +11 -0
  72. orca_sdk/_shared/metrics.py +1 -1
  73. orca_sdk/classification_model.py +4 -1
  74. orca_sdk/classification_model_test.py +53 -0
  75. orca_sdk/credentials.py +15 -1
  76. orca_sdk/datasource.py +180 -41
  77. orca_sdk/datasource_test.py +194 -0
  78. orca_sdk/embedding_model.py +51 -13
  79. orca_sdk/embedding_model_test.py +27 -0
  80. orca_sdk/job.py +15 -14
  81. orca_sdk/job_test.py +34 -0
  82. orca_sdk/memoryset.py +47 -7
  83. orca_sdk/regression_model_test.py +0 -1
  84. orca_sdk/telemetry.py +94 -3
  85. {orca_sdk-0.0.95.dist-info → orca_sdk-0.0.97.dist-info}/METADATA +18 -1
  86. {orca_sdk-0.0.95.dist-info → orca_sdk-0.0.97.dist-info}/RECORD +87 -56
  87. orca_sdk/_generated_api_client/models/body_create_datasource_datasource_post.py +0 -207
  88. orca_sdk/_generated_api_client/models/labeled_memory_metrics.py +0 -246
  89. {orca_sdk-0.0.95.dist-info → orca_sdk-0.0.97.dist-info}/WHEEL +0 -0
@@ -1,246 +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, 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_potential_duplicates (Union[Unset, bool]):
31
- potential_duplicate_memory_ids (Union[List[str], None, Unset]):
32
- cluster (Union[Unset, int]):
33
- embedding_2d (Union[Unset, List[float]]):
34
- anomaly_score (Union[Unset, float]):
35
- neighbor_label_logits (Union[Unset, List[float]]):
36
- neighbor_predicted_label (Union[Unset, int]):
37
- neighbor_predicted_label_ambiguity (Union[Unset, float]):
38
- neighbor_predicted_label_confidence (Union[Unset, float]):
39
- current_label_neighbor_confidence (Union[Unset, float]):
40
- normalized_neighbor_label_entropy (Union[Unset, float]):
41
- neighbor_predicted_label_matches_current_label (Union[None, Unset, bool]):
42
- """
43
-
44
- is_duplicate: Union[Unset, bool] = UNSET
45
- duplicate_memory_ids: Union[Unset, List[str]] = UNSET
46
- has_potential_duplicates: Union[Unset, bool] = UNSET
47
- potential_duplicate_memory_ids: Union[List[str], None, Unset] = UNSET
48
- cluster: Union[Unset, int] = UNSET
49
- embedding_2d: Union[Unset, List[float]] = UNSET
50
- anomaly_score: Union[Unset, float] = UNSET
51
- neighbor_label_logits: Union[Unset, List[float]] = UNSET
52
- neighbor_predicted_label: Union[Unset, int] = UNSET
53
- neighbor_predicted_label_ambiguity: Union[Unset, float] = UNSET
54
- neighbor_predicted_label_confidence: Union[Unset, float] = UNSET
55
- current_label_neighbor_confidence: Union[Unset, float] = UNSET
56
- normalized_neighbor_label_entropy: Union[Unset, float] = UNSET
57
- neighbor_predicted_label_matches_current_label: Union[None, Unset, bool] = UNSET
58
- additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
59
-
60
- def to_dict(self) -> dict[str, Any]:
61
- is_duplicate = self.is_duplicate
62
-
63
- duplicate_memory_ids: Union[Unset, List[str]] = UNSET
64
- if not isinstance(self.duplicate_memory_ids, Unset):
65
- duplicate_memory_ids = self.duplicate_memory_ids
66
-
67
- has_potential_duplicates = self.has_potential_duplicates
68
-
69
- potential_duplicate_memory_ids: Union[List[str], None, Unset]
70
- if isinstance(self.potential_duplicate_memory_ids, Unset):
71
- potential_duplicate_memory_ids = UNSET
72
- elif isinstance(self.potential_duplicate_memory_ids, list):
73
- potential_duplicate_memory_ids = self.potential_duplicate_memory_ids
74
-
75
- else:
76
- potential_duplicate_memory_ids = self.potential_duplicate_memory_ids
77
-
78
- cluster = self.cluster
79
-
80
- embedding_2d: Union[Unset, List[float]] = UNSET
81
- if not isinstance(self.embedding_2d, Unset):
82
- embedding_2d = []
83
- for embedding_2d_item_data in self.embedding_2d:
84
- embedding_2d_item: float
85
- embedding_2d_item = embedding_2d_item_data
86
- embedding_2d.append(embedding_2d_item)
87
-
88
- anomaly_score = self.anomaly_score
89
-
90
- neighbor_label_logits: Union[Unset, List[float]] = UNSET
91
- if not isinstance(self.neighbor_label_logits, Unset):
92
- neighbor_label_logits = self.neighbor_label_logits
93
-
94
- neighbor_predicted_label = self.neighbor_predicted_label
95
-
96
- neighbor_predicted_label_ambiguity = self.neighbor_predicted_label_ambiguity
97
-
98
- neighbor_predicted_label_confidence = self.neighbor_predicted_label_confidence
99
-
100
- current_label_neighbor_confidence = self.current_label_neighbor_confidence
101
-
102
- normalized_neighbor_label_entropy = self.normalized_neighbor_label_entropy
103
-
104
- neighbor_predicted_label_matches_current_label: Union[None, Unset, bool]
105
- if isinstance(self.neighbor_predicted_label_matches_current_label, Unset):
106
- neighbor_predicted_label_matches_current_label = UNSET
107
- else:
108
- neighbor_predicted_label_matches_current_label = self.neighbor_predicted_label_matches_current_label
109
-
110
- field_dict: dict[str, Any] = {}
111
- field_dict.update(self.additional_properties)
112
- field_dict.update({})
113
- if is_duplicate is not UNSET:
114
- field_dict["is_duplicate"] = is_duplicate
115
- if duplicate_memory_ids is not UNSET:
116
- field_dict["duplicate_memory_ids"] = duplicate_memory_ids
117
- if has_potential_duplicates is not UNSET:
118
- field_dict["has_potential_duplicates"] = has_potential_duplicates
119
- if potential_duplicate_memory_ids is not UNSET:
120
- field_dict["potential_duplicate_memory_ids"] = potential_duplicate_memory_ids
121
- if cluster is not UNSET:
122
- field_dict["cluster"] = cluster
123
- if embedding_2d is not UNSET:
124
- field_dict["embedding_2d"] = embedding_2d
125
- if anomaly_score is not UNSET:
126
- field_dict["anomaly_score"] = anomaly_score
127
- if neighbor_label_logits is not UNSET:
128
- field_dict["neighbor_label_logits"] = neighbor_label_logits
129
- if neighbor_predicted_label is not UNSET:
130
- field_dict["neighbor_predicted_label"] = neighbor_predicted_label
131
- if neighbor_predicted_label_ambiguity is not UNSET:
132
- field_dict["neighbor_predicted_label_ambiguity"] = neighbor_predicted_label_ambiguity
133
- if neighbor_predicted_label_confidence is not UNSET:
134
- field_dict["neighbor_predicted_label_confidence"] = neighbor_predicted_label_confidence
135
- if current_label_neighbor_confidence is not UNSET:
136
- field_dict["current_label_neighbor_confidence"] = current_label_neighbor_confidence
137
- if normalized_neighbor_label_entropy is not UNSET:
138
- field_dict["normalized_neighbor_label_entropy"] = normalized_neighbor_label_entropy
139
- if neighbor_predicted_label_matches_current_label is not UNSET:
140
- field_dict["neighbor_predicted_label_matches_current_label"] = (
141
- neighbor_predicted_label_matches_current_label
142
- )
143
-
144
- return field_dict
145
-
146
- @classmethod
147
- def from_dict(cls: Type[T], src_dict: dict[str, Any]) -> T:
148
- d = src_dict.copy()
149
- is_duplicate = d.pop("is_duplicate", UNSET)
150
-
151
- duplicate_memory_ids = cast(List[str], d.pop("duplicate_memory_ids", UNSET))
152
-
153
- has_potential_duplicates = d.pop("has_potential_duplicates", UNSET)
154
-
155
- def _parse_potential_duplicate_memory_ids(data: object) -> Union[List[str], None, Unset]:
156
- if data is None:
157
- return data
158
- if isinstance(data, Unset):
159
- return data
160
- try:
161
- if not isinstance(data, list):
162
- raise TypeError()
163
- potential_duplicate_memory_ids_type_0 = cast(List[str], data)
164
-
165
- return potential_duplicate_memory_ids_type_0
166
- except: # noqa: E722
167
- pass
168
- return cast(Union[List[str], None, Unset], data)
169
-
170
- potential_duplicate_memory_ids = _parse_potential_duplicate_memory_ids(
171
- d.pop("potential_duplicate_memory_ids", UNSET)
172
- )
173
-
174
- cluster = d.pop("cluster", UNSET)
175
-
176
- embedding_2d = []
177
- _embedding_2d = d.pop("embedding_2d", UNSET)
178
- for embedding_2d_item_data in _embedding_2d or []:
179
-
180
- def _parse_embedding_2d_item(data: object) -> float:
181
- return cast(float, data)
182
-
183
- embedding_2d_item = _parse_embedding_2d_item(embedding_2d_item_data)
184
-
185
- embedding_2d.append(embedding_2d_item)
186
-
187
- anomaly_score = d.pop("anomaly_score", UNSET)
188
-
189
- neighbor_label_logits = cast(List[float], d.pop("neighbor_label_logits", UNSET))
190
-
191
- neighbor_predicted_label = d.pop("neighbor_predicted_label", UNSET)
192
-
193
- neighbor_predicted_label_ambiguity = d.pop("neighbor_predicted_label_ambiguity", UNSET)
194
-
195
- neighbor_predicted_label_confidence = d.pop("neighbor_predicted_label_confidence", UNSET)
196
-
197
- current_label_neighbor_confidence = d.pop("current_label_neighbor_confidence", UNSET)
198
-
199
- normalized_neighbor_label_entropy = d.pop("normalized_neighbor_label_entropy", UNSET)
200
-
201
- def _parse_neighbor_predicted_label_matches_current_label(data: object) -> Union[None, Unset, bool]:
202
- if data is None:
203
- return data
204
- if isinstance(data, Unset):
205
- return data
206
- return cast(Union[None, Unset, bool], data)
207
-
208
- neighbor_predicted_label_matches_current_label = _parse_neighbor_predicted_label_matches_current_label(
209
- d.pop("neighbor_predicted_label_matches_current_label", UNSET)
210
- )
211
-
212
- labeled_memory_metrics = cls(
213
- is_duplicate=is_duplicate,
214
- duplicate_memory_ids=duplicate_memory_ids,
215
- has_potential_duplicates=has_potential_duplicates,
216
- potential_duplicate_memory_ids=potential_duplicate_memory_ids,
217
- cluster=cluster,
218
- embedding_2d=embedding_2d,
219
- anomaly_score=anomaly_score,
220
- neighbor_label_logits=neighbor_label_logits,
221
- neighbor_predicted_label=neighbor_predicted_label,
222
- neighbor_predicted_label_ambiguity=neighbor_predicted_label_ambiguity,
223
- neighbor_predicted_label_confidence=neighbor_predicted_label_confidence,
224
- current_label_neighbor_confidence=current_label_neighbor_confidence,
225
- normalized_neighbor_label_entropy=normalized_neighbor_label_entropy,
226
- neighbor_predicted_label_matches_current_label=neighbor_predicted_label_matches_current_label,
227
- )
228
-
229
- labeled_memory_metrics.additional_properties = d
230
- return labeled_memory_metrics
231
-
232
- @property
233
- def additional_keys(self) -> list[str]:
234
- return list(self.additional_properties.keys())
235
-
236
- def __getitem__(self, key: str) -> Any:
237
- return self.additional_properties[key]
238
-
239
- def __setitem__(self, key: str, value: Any) -> None:
240
- self.additional_properties[key] = value
241
-
242
- def __delitem__(self, key: str) -> None:
243
- del self.additional_properties[key]
244
-
245
- def __contains__(self, key: str) -> bool:
246
- return key in self.additional_properties