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
@@ -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, Literal, Type, TypeVar, Union, cast
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 (Literal[404]):
29
+ status_code (NotFoundErrorResponseStatusCode):
29
30
  resource (Union[None, NotFoundErrorResponseResourceType0]):
30
31
  """
31
32
 
32
- status_code: Literal[404]
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 = cast(Literal[404], d.pop("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"
@@ -0,0 +1,8 @@
1
+ from enum import IntEnum
2
+
3
+
4
+ class NotFoundErrorResponseStatusCode(IntEnum):
5
+ VALUE_404 = 404
6
+
7
+ def __str__(self) -> str:
8
+ return str(self.value)