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.
- orca_sdk/__init__.py +1 -5
- orca_sdk/_generated_api_client/api/__init__.py +22 -2
- orca_sdk/_generated_api_client/api/{datasource/create_datasource_datasource_post.py → auth/create_org_plan_auth_org_plan_post.py} +32 -31
- orca_sdk/_generated_api_client/api/auth/get_org_plan_auth_org_plan_get.py +122 -0
- orca_sdk/_generated_api_client/api/auth/update_org_plan_auth_org_plan_put.py +168 -0
- orca_sdk/_generated_api_client/api/datasource/create_datasource_from_content_datasource_post.py +224 -0
- orca_sdk/_generated_api_client/api/datasource/create_datasource_from_files_datasource_upload_post.py +229 -0
- orca_sdk/_generated_api_client/api/task/list_tasks_task_get.py +21 -26
- orca_sdk/_generated_api_client/api/telemetry/generate_memory_suggestions_telemetry_prediction_prediction_id_memory_suggestions_post.py +239 -0
- orca_sdk/_generated_api_client/api/telemetry/get_action_recommendation_telemetry_prediction_prediction_id_action_get.py +192 -0
- orca_sdk/_generated_api_client/models/__init__.py +54 -4
- orca_sdk/_generated_api_client/models/action_recommendation.py +82 -0
- orca_sdk/_generated_api_client/models/action_recommendation_action.py +11 -0
- orca_sdk/_generated_api_client/models/add_memory_recommendations.py +85 -0
- orca_sdk/_generated_api_client/models/add_memory_suggestion.py +79 -0
- orca_sdk/_generated_api_client/models/body_create_datasource_from_files_datasource_upload_post.py +145 -0
- orca_sdk/_generated_api_client/models/class_representatives.py +92 -0
- orca_sdk/_generated_api_client/models/classification_model_metadata.py +14 -0
- orca_sdk/_generated_api_client/models/clone_memoryset_request.py +40 -0
- orca_sdk/_generated_api_client/models/constraint_violation_error_response.py +8 -7
- orca_sdk/_generated_api_client/models/constraint_violation_error_response_status_code.py +8 -0
- orca_sdk/_generated_api_client/models/create_classification_model_request.py +40 -0
- orca_sdk/_generated_api_client/models/create_datasource_from_content_request.py +101 -0
- orca_sdk/_generated_api_client/models/create_memoryset_request.py +40 -0
- orca_sdk/_generated_api_client/models/create_org_plan_request.py +73 -0
- orca_sdk/_generated_api_client/models/create_org_plan_request_tier.py +11 -0
- orca_sdk/_generated_api_client/models/create_regression_model_request.py +20 -0
- orca_sdk/_generated_api_client/models/embed_request.py +20 -0
- orca_sdk/_generated_api_client/models/embedding_evaluation_payload.py +28 -10
- orca_sdk/_generated_api_client/models/embedding_evaluation_request.py +28 -10
- orca_sdk/_generated_api_client/models/embedding_model_result.py +9 -0
- orca_sdk/_generated_api_client/models/filter_item.py +31 -23
- orca_sdk/_generated_api_client/models/filter_item_field_type_1_item_type_0.py +8 -0
- orca_sdk/_generated_api_client/models/filter_item_field_type_2_item_type_0.py +8 -0
- orca_sdk/_generated_api_client/models/filter_item_field_type_2_item_type_1.py +2 -0
- orca_sdk/_generated_api_client/models/internal_server_error_response.py +8 -7
- orca_sdk/_generated_api_client/models/internal_server_error_response_status_code.py +8 -0
- orca_sdk/_generated_api_client/models/labeled_memory.py +5 -5
- orca_sdk/_generated_api_client/models/labeled_memory_update.py +16 -16
- orca_sdk/_generated_api_client/models/labeled_memory_with_feedback_metrics.py +5 -5
- orca_sdk/_generated_api_client/models/lookup_request.py +20 -0
- orca_sdk/_generated_api_client/models/memory_metrics.py +98 -0
- orca_sdk/_generated_api_client/models/memoryset_analysis_configs.py +33 -0
- orca_sdk/_generated_api_client/models/memoryset_class_patterns_analysis_config.py +79 -0
- orca_sdk/_generated_api_client/models/memoryset_class_patterns_metrics.py +138 -0
- orca_sdk/_generated_api_client/models/memoryset_metadata.py +42 -0
- orca_sdk/_generated_api_client/models/memoryset_metrics.py +33 -0
- orca_sdk/_generated_api_client/models/memoryset_update.py +20 -0
- orca_sdk/_generated_api_client/models/not_found_error_response.py +6 -7
- orca_sdk/_generated_api_client/models/not_found_error_response_resource_type_0.py +1 -0
- orca_sdk/_generated_api_client/models/not_found_error_response_status_code.py +8 -0
- orca_sdk/_generated_api_client/models/org_plan.py +99 -0
- orca_sdk/_generated_api_client/models/org_plan_tier.py +11 -0
- orca_sdk/_generated_api_client/models/paginated_task.py +108 -0
- orca_sdk/_generated_api_client/models/predictive_model_update.py +20 -0
- orca_sdk/_generated_api_client/models/pretrained_embedding_model_metadata.py +8 -0
- orca_sdk/_generated_api_client/models/regression_model_metadata.py +14 -0
- orca_sdk/_generated_api_client/models/scored_memory_update.py +9 -9
- orca_sdk/_generated_api_client/models/service_unavailable_error_response.py +8 -7
- orca_sdk/_generated_api_client/models/service_unavailable_error_response_status_code.py +8 -0
- orca_sdk/_generated_api_client/models/telemetry_field_type_0_item_type_0.py +8 -0
- orca_sdk/_generated_api_client/models/telemetry_field_type_1_item_type_0.py +8 -0
- orca_sdk/_generated_api_client/models/telemetry_field_type_1_item_type_1.py +8 -0
- orca_sdk/_generated_api_client/models/telemetry_filter_item.py +42 -30
- orca_sdk/_generated_api_client/models/telemetry_sort_options.py +42 -30
- orca_sdk/_generated_api_client/models/unauthenticated_error_response.py +8 -7
- orca_sdk/_generated_api_client/models/unauthenticated_error_response_status_code.py +8 -0
- orca_sdk/_generated_api_client/models/unauthorized_error_response.py +8 -7
- orca_sdk/_generated_api_client/models/unauthorized_error_response_status_code.py +8 -0
- orca_sdk/_generated_api_client/models/update_org_plan_request.py +73 -0
- orca_sdk/_generated_api_client/models/update_org_plan_request_tier.py +11 -0
- orca_sdk/_shared/metrics.py +1 -1
- orca_sdk/classification_model.py +4 -1
- orca_sdk/classification_model_test.py +53 -0
- orca_sdk/credentials.py +15 -1
- orca_sdk/datasource.py +180 -41
- orca_sdk/datasource_test.py +194 -0
- orca_sdk/embedding_model.py +51 -13
- orca_sdk/embedding_model_test.py +27 -0
- orca_sdk/job.py +15 -14
- orca_sdk/job_test.py +34 -0
- orca_sdk/memoryset.py +47 -7
- orca_sdk/regression_model_test.py +0 -1
- orca_sdk/telemetry.py +94 -3
- {orca_sdk-0.0.95.dist-info → orca_sdk-0.0.97.dist-info}/METADATA +18 -1
- {orca_sdk-0.0.95.dist-info → orca_sdk-0.0.97.dist-info}/RECORD +87 -56
- orca_sdk/_generated_api_client/models/body_create_datasource_datasource_post.py +0 -207
- orca_sdk/_generated_api_client/models/labeled_memory_metrics.py +0 -246
- {orca_sdk-0.0.95.dist-info → orca_sdk-0.0.97.dist-info}/WHEEL +0 -0
|
@@ -17,6 +17,7 @@ from attrs import define as _attrs_define
|
|
|
17
17
|
from ..types import UNSET, Unset
|
|
18
18
|
|
|
19
19
|
if TYPE_CHECKING:
|
|
20
|
+
from ..models.memoryset_class_patterns_analysis_config import MemorysetClassPatternsAnalysisConfig
|
|
20
21
|
from ..models.memoryset_cluster_analysis_config import MemorysetClusterAnalysisConfig
|
|
21
22
|
from ..models.memoryset_duplicate_analysis_config import MemorysetDuplicateAnalysisConfig
|
|
22
23
|
from ..models.memoryset_label_analysis_config import MemorysetLabelAnalysisConfig
|
|
@@ -36,6 +37,7 @@ class MemorysetAnalysisConfigs:
|
|
|
36
37
|
duplicate (Union['MemorysetDuplicateAnalysisConfig', None, Unset]):
|
|
37
38
|
projection (Union['MemorysetProjectionAnalysisConfig', None, Unset]):
|
|
38
39
|
cluster (Union['MemorysetClusterAnalysisConfig', None, Unset]):
|
|
40
|
+
class_patterns (Union['MemorysetClassPatternsAnalysisConfig', None, Unset]):
|
|
39
41
|
"""
|
|
40
42
|
|
|
41
43
|
neighbor: Union["MemorysetNeighborAnalysisConfig", None, Unset] = UNSET
|
|
@@ -43,8 +45,10 @@ class MemorysetAnalysisConfigs:
|
|
|
43
45
|
duplicate: Union["MemorysetDuplicateAnalysisConfig", None, Unset] = UNSET
|
|
44
46
|
projection: Union["MemorysetProjectionAnalysisConfig", None, Unset] = UNSET
|
|
45
47
|
cluster: Union["MemorysetClusterAnalysisConfig", None, Unset] = UNSET
|
|
48
|
+
class_patterns: Union["MemorysetClassPatternsAnalysisConfig", None, Unset] = UNSET
|
|
46
49
|
|
|
47
50
|
def to_dict(self) -> dict[str, Any]:
|
|
51
|
+
from ..models.memoryset_class_patterns_analysis_config import MemorysetClassPatternsAnalysisConfig
|
|
48
52
|
from ..models.memoryset_cluster_analysis_config import MemorysetClusterAnalysisConfig
|
|
49
53
|
from ..models.memoryset_duplicate_analysis_config import MemorysetDuplicateAnalysisConfig
|
|
50
54
|
from ..models.memoryset_label_analysis_config import MemorysetLabelAnalysisConfig
|
|
@@ -91,6 +95,14 @@ class MemorysetAnalysisConfigs:
|
|
|
91
95
|
else:
|
|
92
96
|
cluster = self.cluster
|
|
93
97
|
|
|
98
|
+
class_patterns: Union[Dict[str, Any], None, Unset]
|
|
99
|
+
if isinstance(self.class_patterns, Unset):
|
|
100
|
+
class_patterns = UNSET
|
|
101
|
+
elif isinstance(self.class_patterns, MemorysetClassPatternsAnalysisConfig):
|
|
102
|
+
class_patterns = self.class_patterns.to_dict()
|
|
103
|
+
else:
|
|
104
|
+
class_patterns = self.class_patterns
|
|
105
|
+
|
|
94
106
|
field_dict: dict[str, Any] = {}
|
|
95
107
|
field_dict.update({})
|
|
96
108
|
if neighbor is not UNSET:
|
|
@@ -103,11 +115,14 @@ class MemorysetAnalysisConfigs:
|
|
|
103
115
|
field_dict["projection"] = projection
|
|
104
116
|
if cluster is not UNSET:
|
|
105
117
|
field_dict["cluster"] = cluster
|
|
118
|
+
if class_patterns is not UNSET:
|
|
119
|
+
field_dict["class_patterns"] = class_patterns
|
|
106
120
|
|
|
107
121
|
return field_dict
|
|
108
122
|
|
|
109
123
|
@classmethod
|
|
110
124
|
def from_dict(cls: Type[T], src_dict: dict[str, Any]) -> T:
|
|
125
|
+
from ..models.memoryset_class_patterns_analysis_config import MemorysetClassPatternsAnalysisConfig
|
|
111
126
|
from ..models.memoryset_cluster_analysis_config import MemorysetClusterAnalysisConfig
|
|
112
127
|
from ..models.memoryset_duplicate_analysis_config import MemorysetDuplicateAnalysisConfig
|
|
113
128
|
from ..models.memoryset_label_analysis_config import MemorysetLabelAnalysisConfig
|
|
@@ -201,12 +216,30 @@ class MemorysetAnalysisConfigs:
|
|
|
201
216
|
|
|
202
217
|
cluster = _parse_cluster(d.pop("cluster", UNSET))
|
|
203
218
|
|
|
219
|
+
def _parse_class_patterns(data: object) -> Union["MemorysetClassPatternsAnalysisConfig", None, Unset]:
|
|
220
|
+
if data is None:
|
|
221
|
+
return data
|
|
222
|
+
if isinstance(data, Unset):
|
|
223
|
+
return data
|
|
224
|
+
try:
|
|
225
|
+
if not isinstance(data, dict):
|
|
226
|
+
raise TypeError()
|
|
227
|
+
class_patterns_type_0 = MemorysetClassPatternsAnalysisConfig.from_dict(data)
|
|
228
|
+
|
|
229
|
+
return class_patterns_type_0
|
|
230
|
+
except: # noqa: E722
|
|
231
|
+
pass
|
|
232
|
+
return cast(Union["MemorysetClassPatternsAnalysisConfig", None, Unset], data)
|
|
233
|
+
|
|
234
|
+
class_patterns = _parse_class_patterns(d.pop("class_patterns", UNSET))
|
|
235
|
+
|
|
204
236
|
memoryset_analysis_configs = cls(
|
|
205
237
|
neighbor=neighbor,
|
|
206
238
|
label=label,
|
|
207
239
|
duplicate=duplicate,
|
|
208
240
|
projection=projection,
|
|
209
241
|
cluster=cluster,
|
|
242
|
+
class_patterns=class_patterns,
|
|
210
243
|
)
|
|
211
244
|
|
|
212
245
|
return memoryset_analysis_configs
|
|
@@ -0,0 +1,79 @@
|
|
|
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
|
|
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="MemorysetClassPatternsAnalysisConfig")
|
|
21
|
+
|
|
22
|
+
|
|
23
|
+
@_attrs_define
|
|
24
|
+
class MemorysetClassPatternsAnalysisConfig:
|
|
25
|
+
"""
|
|
26
|
+
Attributes:
|
|
27
|
+
representatives_per_class (Union[Unset, int]): Default: 5.
|
|
28
|
+
enable_patterns_description (Union[Unset, bool]): Default: True.
|
|
29
|
+
"""
|
|
30
|
+
|
|
31
|
+
representatives_per_class: Union[Unset, int] = 5
|
|
32
|
+
enable_patterns_description: Union[Unset, bool] = True
|
|
33
|
+
additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
|
|
34
|
+
|
|
35
|
+
def to_dict(self) -> dict[str, Any]:
|
|
36
|
+
representatives_per_class = self.representatives_per_class
|
|
37
|
+
|
|
38
|
+
enable_patterns_description = self.enable_patterns_description
|
|
39
|
+
|
|
40
|
+
field_dict: dict[str, Any] = {}
|
|
41
|
+
field_dict.update(self.additional_properties)
|
|
42
|
+
field_dict.update({})
|
|
43
|
+
if representatives_per_class is not UNSET:
|
|
44
|
+
field_dict["representatives_per_class"] = representatives_per_class
|
|
45
|
+
if enable_patterns_description is not UNSET:
|
|
46
|
+
field_dict["enable_patterns_description"] = enable_patterns_description
|
|
47
|
+
|
|
48
|
+
return field_dict
|
|
49
|
+
|
|
50
|
+
@classmethod
|
|
51
|
+
def from_dict(cls: Type[T], src_dict: dict[str, Any]) -> T:
|
|
52
|
+
d = src_dict.copy()
|
|
53
|
+
representatives_per_class = d.pop("representatives_per_class", UNSET)
|
|
54
|
+
|
|
55
|
+
enable_patterns_description = d.pop("enable_patterns_description", UNSET)
|
|
56
|
+
|
|
57
|
+
memoryset_class_patterns_analysis_config = cls(
|
|
58
|
+
representatives_per_class=representatives_per_class,
|
|
59
|
+
enable_patterns_description=enable_patterns_description,
|
|
60
|
+
)
|
|
61
|
+
|
|
62
|
+
memoryset_class_patterns_analysis_config.additional_properties = d
|
|
63
|
+
return memoryset_class_patterns_analysis_config
|
|
64
|
+
|
|
65
|
+
@property
|
|
66
|
+
def additional_keys(self) -> list[str]:
|
|
67
|
+
return list(self.additional_properties.keys())
|
|
68
|
+
|
|
69
|
+
def __getitem__(self, key: str) -> Any:
|
|
70
|
+
return self.additional_properties[key]
|
|
71
|
+
|
|
72
|
+
def __setitem__(self, key: str, value: Any) -> None:
|
|
73
|
+
self.additional_properties[key] = value
|
|
74
|
+
|
|
75
|
+
def __delitem__(self, key: str) -> None:
|
|
76
|
+
del self.additional_properties[key]
|
|
77
|
+
|
|
78
|
+
def __contains__(self, key: str) -> bool:
|
|
79
|
+
return key in self.additional_properties
|
|
@@ -0,0 +1,138 @@
|
|
|
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, 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
|
+
if TYPE_CHECKING:
|
|
21
|
+
from ..models.class_representatives import ClassRepresentatives
|
|
22
|
+
|
|
23
|
+
|
|
24
|
+
T = TypeVar("T", bound="MemorysetClassPatternsMetrics")
|
|
25
|
+
|
|
26
|
+
|
|
27
|
+
@_attrs_define
|
|
28
|
+
class MemorysetClassPatternsMetrics:
|
|
29
|
+
"""
|
|
30
|
+
Attributes:
|
|
31
|
+
class_representatives (List['ClassRepresentatives']):
|
|
32
|
+
mean_spread (float):
|
|
33
|
+
variance_spread (float):
|
|
34
|
+
mean_uniformity (float):
|
|
35
|
+
variance_uniformity (float):
|
|
36
|
+
patterns_description (Union[None, Unset, str]):
|
|
37
|
+
"""
|
|
38
|
+
|
|
39
|
+
class_representatives: List["ClassRepresentatives"]
|
|
40
|
+
mean_spread: float
|
|
41
|
+
variance_spread: float
|
|
42
|
+
mean_uniformity: float
|
|
43
|
+
variance_uniformity: float
|
|
44
|
+
patterns_description: Union[None, Unset, str] = UNSET
|
|
45
|
+
additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
|
|
46
|
+
|
|
47
|
+
def to_dict(self) -> dict[str, Any]:
|
|
48
|
+
class_representatives = []
|
|
49
|
+
for class_representatives_item_data in self.class_representatives:
|
|
50
|
+
class_representatives_item = class_representatives_item_data.to_dict()
|
|
51
|
+
class_representatives.append(class_representatives_item)
|
|
52
|
+
|
|
53
|
+
mean_spread = self.mean_spread
|
|
54
|
+
|
|
55
|
+
variance_spread = self.variance_spread
|
|
56
|
+
|
|
57
|
+
mean_uniformity = self.mean_uniformity
|
|
58
|
+
|
|
59
|
+
variance_uniformity = self.variance_uniformity
|
|
60
|
+
|
|
61
|
+
patterns_description: Union[None, Unset, str]
|
|
62
|
+
if isinstance(self.patterns_description, Unset):
|
|
63
|
+
patterns_description = UNSET
|
|
64
|
+
else:
|
|
65
|
+
patterns_description = self.patterns_description
|
|
66
|
+
|
|
67
|
+
field_dict: dict[str, Any] = {}
|
|
68
|
+
field_dict.update(self.additional_properties)
|
|
69
|
+
field_dict.update(
|
|
70
|
+
{
|
|
71
|
+
"class_representatives": class_representatives,
|
|
72
|
+
"mean_spread": mean_spread,
|
|
73
|
+
"variance_spread": variance_spread,
|
|
74
|
+
"mean_uniformity": mean_uniformity,
|
|
75
|
+
"variance_uniformity": variance_uniformity,
|
|
76
|
+
}
|
|
77
|
+
)
|
|
78
|
+
if patterns_description is not UNSET:
|
|
79
|
+
field_dict["patterns_description"] = patterns_description
|
|
80
|
+
|
|
81
|
+
return field_dict
|
|
82
|
+
|
|
83
|
+
@classmethod
|
|
84
|
+
def from_dict(cls: Type[T], src_dict: dict[str, Any]) -> T:
|
|
85
|
+
from ..models.class_representatives import ClassRepresentatives
|
|
86
|
+
|
|
87
|
+
d = src_dict.copy()
|
|
88
|
+
class_representatives = []
|
|
89
|
+
_class_representatives = d.pop("class_representatives")
|
|
90
|
+
for class_representatives_item_data in _class_representatives:
|
|
91
|
+
class_representatives_item = ClassRepresentatives.from_dict(class_representatives_item_data)
|
|
92
|
+
|
|
93
|
+
class_representatives.append(class_representatives_item)
|
|
94
|
+
|
|
95
|
+
mean_spread = d.pop("mean_spread")
|
|
96
|
+
|
|
97
|
+
variance_spread = d.pop("variance_spread")
|
|
98
|
+
|
|
99
|
+
mean_uniformity = d.pop("mean_uniformity")
|
|
100
|
+
|
|
101
|
+
variance_uniformity = d.pop("variance_uniformity")
|
|
102
|
+
|
|
103
|
+
def _parse_patterns_description(data: object) -> Union[None, Unset, str]:
|
|
104
|
+
if data is None:
|
|
105
|
+
return data
|
|
106
|
+
if isinstance(data, Unset):
|
|
107
|
+
return data
|
|
108
|
+
return cast(Union[None, Unset, str], data)
|
|
109
|
+
|
|
110
|
+
patterns_description = _parse_patterns_description(d.pop("patterns_description", UNSET))
|
|
111
|
+
|
|
112
|
+
memoryset_class_patterns_metrics = cls(
|
|
113
|
+
class_representatives=class_representatives,
|
|
114
|
+
mean_spread=mean_spread,
|
|
115
|
+
variance_spread=variance_spread,
|
|
116
|
+
mean_uniformity=mean_uniformity,
|
|
117
|
+
variance_uniformity=variance_uniformity,
|
|
118
|
+
patterns_description=patterns_description,
|
|
119
|
+
)
|
|
120
|
+
|
|
121
|
+
memoryset_class_patterns_metrics.additional_properties = d
|
|
122
|
+
return memoryset_class_patterns_metrics
|
|
123
|
+
|
|
124
|
+
@property
|
|
125
|
+
def additional_keys(self) -> list[str]:
|
|
126
|
+
return list(self.additional_properties.keys())
|
|
127
|
+
|
|
128
|
+
def __getitem__(self, key: str) -> Any:
|
|
129
|
+
return self.additional_properties[key]
|
|
130
|
+
|
|
131
|
+
def __setitem__(self, key: str, value: Any) -> None:
|
|
132
|
+
self.additional_properties[key] = value
|
|
133
|
+
|
|
134
|
+
def __delitem__(self, key: str) -> None:
|
|
135
|
+
del self.additional_properties[key]
|
|
136
|
+
|
|
137
|
+
def __contains__(self, key: str) -> bool:
|
|
138
|
+
return key in self.additional_properties
|
|
@@ -40,6 +40,7 @@ class MemorysetMetadata:
|
|
|
40
40
|
collection_name (str):
|
|
41
41
|
name (str):
|
|
42
42
|
description (Union[None, str]):
|
|
43
|
+
notes (Union[None, str]):
|
|
43
44
|
length (int):
|
|
44
45
|
pretrained_embedding_model_name (Union[None, PretrainedEmbeddingModelName]):
|
|
45
46
|
finetuned_embedding_model_id (Union[None, str]):
|
|
@@ -54,6 +55,8 @@ class MemorysetMetadata:
|
|
|
54
55
|
index_type (MemorysetMetadataIndexType):
|
|
55
56
|
index_params (MemorysetMetadataIndexParams):
|
|
56
57
|
database_uri (Union[None, str]):
|
|
58
|
+
document_prompt_override (Union[None, str]):
|
|
59
|
+
query_prompt_override (Union[None, str]):
|
|
57
60
|
"""
|
|
58
61
|
|
|
59
62
|
id: str
|
|
@@ -61,6 +64,7 @@ class MemorysetMetadata:
|
|
|
61
64
|
collection_name: str
|
|
62
65
|
name: str
|
|
63
66
|
description: Union[None, str]
|
|
67
|
+
notes: Union[None, str]
|
|
64
68
|
length: int
|
|
65
69
|
pretrained_embedding_model_name: Union[None, PretrainedEmbeddingModelName]
|
|
66
70
|
finetuned_embedding_model_id: Union[None, str]
|
|
@@ -75,6 +79,8 @@ class MemorysetMetadata:
|
|
|
75
79
|
index_type: MemorysetMetadataIndexType
|
|
76
80
|
index_params: "MemorysetMetadataIndexParams"
|
|
77
81
|
database_uri: Union[None, str]
|
|
82
|
+
document_prompt_override: Union[None, str]
|
|
83
|
+
query_prompt_override: Union[None, str]
|
|
78
84
|
additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
|
|
79
85
|
|
|
80
86
|
def to_dict(self) -> dict[str, Any]:
|
|
@@ -89,6 +95,9 @@ class MemorysetMetadata:
|
|
|
89
95
|
description: Union[None, str]
|
|
90
96
|
description = self.description
|
|
91
97
|
|
|
98
|
+
notes: Union[None, str]
|
|
99
|
+
notes = self.notes
|
|
100
|
+
|
|
92
101
|
length = self.length
|
|
93
102
|
|
|
94
103
|
pretrained_embedding_model_name: Union[None, str]
|
|
@@ -134,6 +143,12 @@ class MemorysetMetadata:
|
|
|
134
143
|
database_uri: Union[None, str]
|
|
135
144
|
database_uri = self.database_uri
|
|
136
145
|
|
|
146
|
+
document_prompt_override: Union[None, str]
|
|
147
|
+
document_prompt_override = self.document_prompt_override
|
|
148
|
+
|
|
149
|
+
query_prompt_override: Union[None, str]
|
|
150
|
+
query_prompt_override = self.query_prompt_override
|
|
151
|
+
|
|
137
152
|
field_dict: dict[str, Any] = {}
|
|
138
153
|
field_dict.update(self.additional_properties)
|
|
139
154
|
field_dict.update(
|
|
@@ -143,6 +158,7 @@ class MemorysetMetadata:
|
|
|
143
158
|
"collection_name": collection_name,
|
|
144
159
|
"name": name,
|
|
145
160
|
"description": description,
|
|
161
|
+
"notes": notes,
|
|
146
162
|
"length": length,
|
|
147
163
|
"pretrained_embedding_model_name": pretrained_embedding_model_name,
|
|
148
164
|
"finetuned_embedding_model_id": finetuned_embedding_model_id,
|
|
@@ -157,6 +173,8 @@ class MemorysetMetadata:
|
|
|
157
173
|
"index_type": index_type,
|
|
158
174
|
"index_params": index_params,
|
|
159
175
|
"database_uri": database_uri,
|
|
176
|
+
"document_prompt_override": document_prompt_override,
|
|
177
|
+
"query_prompt_override": query_prompt_override,
|
|
160
178
|
}
|
|
161
179
|
)
|
|
162
180
|
|
|
@@ -183,6 +201,13 @@ class MemorysetMetadata:
|
|
|
183
201
|
|
|
184
202
|
description = _parse_description(d.pop("description"))
|
|
185
203
|
|
|
204
|
+
def _parse_notes(data: object) -> Union[None, str]:
|
|
205
|
+
if data is None:
|
|
206
|
+
return data
|
|
207
|
+
return cast(Union[None, str], data)
|
|
208
|
+
|
|
209
|
+
notes = _parse_notes(d.pop("notes"))
|
|
210
|
+
|
|
186
211
|
length = d.pop("length")
|
|
187
212
|
|
|
188
213
|
def _parse_pretrained_embedding_model_name(data: object) -> Union[None, PretrainedEmbeddingModelName]:
|
|
@@ -249,12 +274,27 @@ class MemorysetMetadata:
|
|
|
249
274
|
|
|
250
275
|
database_uri = _parse_database_uri(d.pop("database_uri"))
|
|
251
276
|
|
|
277
|
+
def _parse_document_prompt_override(data: object) -> Union[None, str]:
|
|
278
|
+
if data is None:
|
|
279
|
+
return data
|
|
280
|
+
return cast(Union[None, str], data)
|
|
281
|
+
|
|
282
|
+
document_prompt_override = _parse_document_prompt_override(d.pop("document_prompt_override"))
|
|
283
|
+
|
|
284
|
+
def _parse_query_prompt_override(data: object) -> Union[None, str]:
|
|
285
|
+
if data is None:
|
|
286
|
+
return data
|
|
287
|
+
return cast(Union[None, str], data)
|
|
288
|
+
|
|
289
|
+
query_prompt_override = _parse_query_prompt_override(d.pop("query_prompt_override"))
|
|
290
|
+
|
|
252
291
|
memoryset_metadata = cls(
|
|
253
292
|
id=id,
|
|
254
293
|
org_id=org_id,
|
|
255
294
|
collection_name=collection_name,
|
|
256
295
|
name=name,
|
|
257
296
|
description=description,
|
|
297
|
+
notes=notes,
|
|
258
298
|
length=length,
|
|
259
299
|
pretrained_embedding_model_name=pretrained_embedding_model_name,
|
|
260
300
|
finetuned_embedding_model_id=finetuned_embedding_model_id,
|
|
@@ -269,6 +309,8 @@ class MemorysetMetadata:
|
|
|
269
309
|
index_type=index_type,
|
|
270
310
|
index_params=index_params,
|
|
271
311
|
database_uri=database_uri,
|
|
312
|
+
document_prompt_override=document_prompt_override,
|
|
313
|
+
query_prompt_override=query_prompt_override,
|
|
272
314
|
)
|
|
273
315
|
|
|
274
316
|
memoryset_metadata.additional_properties = d
|
|
@@ -18,6 +18,7 @@ from attrs import field as _attrs_field
|
|
|
18
18
|
from ..types import UNSET, Unset
|
|
19
19
|
|
|
20
20
|
if TYPE_CHECKING:
|
|
21
|
+
from ..models.memoryset_class_patterns_metrics import MemorysetClassPatternsMetrics
|
|
21
22
|
from ..models.memoryset_cluster_metrics import MemorysetClusterMetrics
|
|
22
23
|
from ..models.memoryset_duplicate_metrics import MemorysetDuplicateMetrics
|
|
23
24
|
from ..models.memoryset_label_metrics import MemorysetLabelMetrics
|
|
@@ -37,6 +38,7 @@ class MemorysetMetrics:
|
|
|
37
38
|
duplicate (Union['MemorysetDuplicateMetrics', None, Unset]):
|
|
38
39
|
projection (Union['MemorysetProjectionMetrics', None, Unset]):
|
|
39
40
|
cluster (Union['MemorysetClusterMetrics', None, Unset]):
|
|
41
|
+
class_patterns (Union['MemorysetClassPatternsMetrics', None, Unset]):
|
|
40
42
|
"""
|
|
41
43
|
|
|
42
44
|
neighbor: Union["MemorysetNeighborMetrics", None, Unset] = UNSET
|
|
@@ -44,9 +46,11 @@ class MemorysetMetrics:
|
|
|
44
46
|
duplicate: Union["MemorysetDuplicateMetrics", None, Unset] = UNSET
|
|
45
47
|
projection: Union["MemorysetProjectionMetrics", None, Unset] = UNSET
|
|
46
48
|
cluster: Union["MemorysetClusterMetrics", None, Unset] = UNSET
|
|
49
|
+
class_patterns: Union["MemorysetClassPatternsMetrics", None, Unset] = UNSET
|
|
47
50
|
additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
|
|
48
51
|
|
|
49
52
|
def to_dict(self) -> dict[str, Any]:
|
|
53
|
+
from ..models.memoryset_class_patterns_metrics import MemorysetClassPatternsMetrics
|
|
50
54
|
from ..models.memoryset_cluster_metrics import MemorysetClusterMetrics
|
|
51
55
|
from ..models.memoryset_duplicate_metrics import MemorysetDuplicateMetrics
|
|
52
56
|
from ..models.memoryset_label_metrics import MemorysetLabelMetrics
|
|
@@ -93,6 +97,14 @@ class MemorysetMetrics:
|
|
|
93
97
|
else:
|
|
94
98
|
cluster = self.cluster
|
|
95
99
|
|
|
100
|
+
class_patterns: Union[Dict[str, Any], None, Unset]
|
|
101
|
+
if isinstance(self.class_patterns, Unset):
|
|
102
|
+
class_patterns = UNSET
|
|
103
|
+
elif isinstance(self.class_patterns, MemorysetClassPatternsMetrics):
|
|
104
|
+
class_patterns = self.class_patterns.to_dict()
|
|
105
|
+
else:
|
|
106
|
+
class_patterns = self.class_patterns
|
|
107
|
+
|
|
96
108
|
field_dict: dict[str, Any] = {}
|
|
97
109
|
field_dict.update(self.additional_properties)
|
|
98
110
|
field_dict.update({})
|
|
@@ -106,11 +118,14 @@ class MemorysetMetrics:
|
|
|
106
118
|
field_dict["projection"] = projection
|
|
107
119
|
if cluster is not UNSET:
|
|
108
120
|
field_dict["cluster"] = cluster
|
|
121
|
+
if class_patterns is not UNSET:
|
|
122
|
+
field_dict["class_patterns"] = class_patterns
|
|
109
123
|
|
|
110
124
|
return field_dict
|
|
111
125
|
|
|
112
126
|
@classmethod
|
|
113
127
|
def from_dict(cls: Type[T], src_dict: dict[str, Any]) -> T:
|
|
128
|
+
from ..models.memoryset_class_patterns_metrics import MemorysetClassPatternsMetrics
|
|
114
129
|
from ..models.memoryset_cluster_metrics import MemorysetClusterMetrics
|
|
115
130
|
from ..models.memoryset_duplicate_metrics import MemorysetDuplicateMetrics
|
|
116
131
|
from ..models.memoryset_label_metrics import MemorysetLabelMetrics
|
|
@@ -204,12 +219,30 @@ class MemorysetMetrics:
|
|
|
204
219
|
|
|
205
220
|
cluster = _parse_cluster(d.pop("cluster", UNSET))
|
|
206
221
|
|
|
222
|
+
def _parse_class_patterns(data: object) -> Union["MemorysetClassPatternsMetrics", None, Unset]:
|
|
223
|
+
if data is None:
|
|
224
|
+
return data
|
|
225
|
+
if isinstance(data, Unset):
|
|
226
|
+
return data
|
|
227
|
+
try:
|
|
228
|
+
if not isinstance(data, dict):
|
|
229
|
+
raise TypeError()
|
|
230
|
+
class_patterns_type_0 = MemorysetClassPatternsMetrics.from_dict(data)
|
|
231
|
+
|
|
232
|
+
return class_patterns_type_0
|
|
233
|
+
except: # noqa: E722
|
|
234
|
+
pass
|
|
235
|
+
return cast(Union["MemorysetClassPatternsMetrics", None, Unset], data)
|
|
236
|
+
|
|
237
|
+
class_patterns = _parse_class_patterns(d.pop("class_patterns", UNSET))
|
|
238
|
+
|
|
207
239
|
memoryset_metrics = cls(
|
|
208
240
|
neighbor=neighbor,
|
|
209
241
|
label=label,
|
|
210
242
|
duplicate=duplicate,
|
|
211
243
|
projection=projection,
|
|
212
244
|
cluster=cluster,
|
|
245
|
+
class_patterns=class_patterns,
|
|
213
246
|
)
|
|
214
247
|
|
|
215
248
|
memoryset_metrics.additional_properties = d
|
|
@@ -27,11 +27,13 @@ class MemorysetUpdate:
|
|
|
27
27
|
label_names (Union[Unset, List[str]]):
|
|
28
28
|
description (Union[None, Unset, str]):
|
|
29
29
|
name (Union[Unset, str]):
|
|
30
|
+
notes (Union[None, Unset, str]):
|
|
30
31
|
"""
|
|
31
32
|
|
|
32
33
|
label_names: Union[Unset, List[str]] = UNSET
|
|
33
34
|
description: Union[None, Unset, str] = UNSET
|
|
34
35
|
name: Union[Unset, str] = UNSET
|
|
36
|
+
notes: Union[None, Unset, str] = UNSET
|
|
35
37
|
additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
|
|
36
38
|
|
|
37
39
|
def to_dict(self) -> dict[str, Any]:
|
|
@@ -47,6 +49,12 @@ class MemorysetUpdate:
|
|
|
47
49
|
|
|
48
50
|
name = self.name
|
|
49
51
|
|
|
52
|
+
notes: Union[None, Unset, str]
|
|
53
|
+
if isinstance(self.notes, Unset):
|
|
54
|
+
notes = UNSET
|
|
55
|
+
else:
|
|
56
|
+
notes = self.notes
|
|
57
|
+
|
|
50
58
|
field_dict: dict[str, Any] = {}
|
|
51
59
|
field_dict.update(self.additional_properties)
|
|
52
60
|
field_dict.update({})
|
|
@@ -56,6 +64,8 @@ class MemorysetUpdate:
|
|
|
56
64
|
field_dict["description"] = description
|
|
57
65
|
if name is not UNSET:
|
|
58
66
|
field_dict["name"] = name
|
|
67
|
+
if notes is not UNSET:
|
|
68
|
+
field_dict["notes"] = notes
|
|
59
69
|
|
|
60
70
|
return field_dict
|
|
61
71
|
|
|
@@ -75,10 +85,20 @@ class MemorysetUpdate:
|
|
|
75
85
|
|
|
76
86
|
name = d.pop("name", UNSET)
|
|
77
87
|
|
|
88
|
+
def _parse_notes(data: object) -> Union[None, Unset, str]:
|
|
89
|
+
if data is None:
|
|
90
|
+
return data
|
|
91
|
+
if isinstance(data, Unset):
|
|
92
|
+
return data
|
|
93
|
+
return cast(Union[None, Unset, str], data)
|
|
94
|
+
|
|
95
|
+
notes = _parse_notes(d.pop("notes", UNSET))
|
|
96
|
+
|
|
78
97
|
memoryset_update = cls(
|
|
79
98
|
label_names=label_names,
|
|
80
99
|
description=description,
|
|
81
100
|
name=name,
|
|
101
|
+
notes=notes,
|
|
82
102
|
)
|
|
83
103
|
|
|
84
104
|
memoryset_update.additional_properties = d
|
|
@@ -11,12 +11,13 @@ The main change is:
|
|
|
11
11
|
# flake8: noqa: C901
|
|
12
12
|
|
|
13
13
|
from enum import Enum
|
|
14
|
-
from typing import Any,
|
|
14
|
+
from typing import Any, Type, TypeVar, Union, cast
|
|
15
15
|
|
|
16
16
|
from attrs import define as _attrs_define
|
|
17
17
|
from attrs import field as _attrs_field
|
|
18
18
|
|
|
19
19
|
from ..models.not_found_error_response_resource_type_0 import NotFoundErrorResponseResourceType0
|
|
20
|
+
from ..models.not_found_error_response_status_code import NotFoundErrorResponseStatusCode
|
|
20
21
|
|
|
21
22
|
T = TypeVar("T", bound="NotFoundErrorResponse")
|
|
22
23
|
|
|
@@ -25,16 +26,16 @@ T = TypeVar("T", bound="NotFoundErrorResponse")
|
|
|
25
26
|
class NotFoundErrorResponse:
|
|
26
27
|
"""
|
|
27
28
|
Attributes:
|
|
28
|
-
status_code (
|
|
29
|
+
status_code (NotFoundErrorResponseStatusCode):
|
|
29
30
|
resource (Union[None, NotFoundErrorResponseResourceType0]):
|
|
30
31
|
"""
|
|
31
32
|
|
|
32
|
-
status_code:
|
|
33
|
+
status_code: NotFoundErrorResponseStatusCode
|
|
33
34
|
resource: Union[None, NotFoundErrorResponseResourceType0]
|
|
34
35
|
additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
|
|
35
36
|
|
|
36
37
|
def to_dict(self) -> dict[str, Any]:
|
|
37
|
-
status_code = self.status_code
|
|
38
|
+
status_code = self.status_code.value if isinstance(self.status_code, Enum) else self.status_code
|
|
38
39
|
|
|
39
40
|
resource: Union[None, str]
|
|
40
41
|
if isinstance(self.resource, NotFoundErrorResponseResourceType0):
|
|
@@ -56,9 +57,7 @@ class NotFoundErrorResponse:
|
|
|
56
57
|
@classmethod
|
|
57
58
|
def from_dict(cls: Type[T], src_dict: dict[str, Any]) -> T:
|
|
58
59
|
d = src_dict.copy()
|
|
59
|
-
status_code =
|
|
60
|
-
if status_code != 404:
|
|
61
|
-
raise ValueError(f"status_code must match const 404, got '{status_code}'")
|
|
60
|
+
status_code = NotFoundErrorResponseStatusCode(d.pop("status_code"))
|
|
62
61
|
|
|
63
62
|
def _parse_resource(data: object) -> Union[None, NotFoundErrorResponseResourceType0]:
|
|
64
63
|
if data is None:
|
|
@@ -13,6 +13,7 @@ class NotFoundErrorResponseResourceType0(str, Enum):
|
|
|
13
13
|
MEMORY = "memory"
|
|
14
14
|
MEMORYSET = "memoryset"
|
|
15
15
|
ORG = "org"
|
|
16
|
+
ORG_PLAN = "org_plan"
|
|
16
17
|
PREDICTION = "prediction"
|
|
17
18
|
PRETRAINED_EMBEDDING_MODEL = "pretrained_embedding_model"
|
|
18
19
|
REGRESSION_MODEL = "regression_model"
|