orca-sdk 0.1.1__py3-none-any.whl → 0.1.3__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 (186) hide show
  1. orca_sdk/__init__.py +10 -4
  2. orca_sdk/_shared/__init__.py +10 -0
  3. orca_sdk/_shared/metrics.py +393 -0
  4. orca_sdk/_shared/metrics_test.py +273 -0
  5. orca_sdk/_utils/analysis_ui.py +12 -10
  6. orca_sdk/_utils/analysis_ui_style.css +0 -3
  7. orca_sdk/_utils/auth.py +31 -29
  8. orca_sdk/_utils/data_parsing.py +28 -2
  9. orca_sdk/_utils/data_parsing_test.py +15 -15
  10. orca_sdk/_utils/pagination.py +126 -0
  11. orca_sdk/_utils/pagination_test.py +132 -0
  12. orca_sdk/_utils/prediction_result_ui.py +67 -21
  13. orca_sdk/_utils/tqdm_file_reader.py +12 -0
  14. orca_sdk/_utils/value_parser.py +45 -0
  15. orca_sdk/_utils/value_parser_test.py +39 -0
  16. orca_sdk/async_client.py +3795 -0
  17. orca_sdk/classification_model.py +601 -129
  18. orca_sdk/classification_model_test.py +415 -117
  19. orca_sdk/client.py +3787 -0
  20. orca_sdk/conftest.py +184 -38
  21. orca_sdk/credentials.py +162 -20
  22. orca_sdk/credentials_test.py +100 -16
  23. orca_sdk/datasource.py +268 -68
  24. orca_sdk/datasource_test.py +266 -18
  25. orca_sdk/embedding_model.py +434 -82
  26. orca_sdk/embedding_model_test.py +66 -33
  27. orca_sdk/job.py +343 -0
  28. orca_sdk/job_test.py +108 -0
  29. orca_sdk/memoryset.py +1690 -324
  30. orca_sdk/memoryset_test.py +456 -119
  31. orca_sdk/regression_model.py +694 -0
  32. orca_sdk/regression_model_test.py +378 -0
  33. orca_sdk/telemetry.py +460 -143
  34. orca_sdk/telemetry_test.py +43 -24
  35. {orca_sdk-0.1.1.dist-info → orca_sdk-0.1.3.dist-info}/METADATA +34 -16
  36. orca_sdk-0.1.3.dist-info/RECORD +41 -0
  37. {orca_sdk-0.1.1.dist-info → orca_sdk-0.1.3.dist-info}/WHEEL +1 -1
  38. orca_sdk/_generated_api_client/__init__.py +0 -3
  39. orca_sdk/_generated_api_client/api/__init__.py +0 -193
  40. orca_sdk/_generated_api_client/api/auth/__init__.py +0 -0
  41. orca_sdk/_generated_api_client/api/auth/check_authentication_auth_get.py +0 -128
  42. orca_sdk/_generated_api_client/api/auth/create_api_key_auth_api_key_post.py +0 -170
  43. orca_sdk/_generated_api_client/api/auth/delete_api_key_auth_api_key_name_or_id_delete.py +0 -156
  44. orca_sdk/_generated_api_client/api/auth/delete_org_auth_org_delete.py +0 -130
  45. orca_sdk/_generated_api_client/api/auth/list_api_keys_auth_api_key_get.py +0 -127
  46. orca_sdk/_generated_api_client/api/classification_model/__init__.py +0 -0
  47. orca_sdk/_generated_api_client/api/classification_model/create_evaluation_classification_model_model_name_or_id_evaluation_post.py +0 -183
  48. orca_sdk/_generated_api_client/api/classification_model/create_model_classification_model_post.py +0 -170
  49. orca_sdk/_generated_api_client/api/classification_model/delete_evaluation_classification_model_model_name_or_id_evaluation_task_id_delete.py +0 -168
  50. orca_sdk/_generated_api_client/api/classification_model/delete_model_classification_model_name_or_id_delete.py +0 -154
  51. orca_sdk/_generated_api_client/api/classification_model/get_evaluation_classification_model_model_name_or_id_evaluation_task_id_get.py +0 -170
  52. orca_sdk/_generated_api_client/api/classification_model/get_model_classification_model_name_or_id_get.py +0 -156
  53. orca_sdk/_generated_api_client/api/classification_model/list_evaluations_classification_model_model_name_or_id_evaluation_get.py +0 -161
  54. orca_sdk/_generated_api_client/api/classification_model/list_models_classification_model_get.py +0 -127
  55. orca_sdk/_generated_api_client/api/classification_model/predict_gpu_classification_model_name_or_id_prediction_post.py +0 -190
  56. orca_sdk/_generated_api_client/api/datasource/__init__.py +0 -0
  57. orca_sdk/_generated_api_client/api/datasource/create_datasource_datasource_post.py +0 -167
  58. orca_sdk/_generated_api_client/api/datasource/delete_datasource_datasource_name_or_id_delete.py +0 -156
  59. orca_sdk/_generated_api_client/api/datasource/get_datasource_datasource_name_or_id_get.py +0 -156
  60. orca_sdk/_generated_api_client/api/datasource/list_datasources_datasource_get.py +0 -127
  61. orca_sdk/_generated_api_client/api/default/__init__.py +0 -0
  62. orca_sdk/_generated_api_client/api/default/healthcheck_get.py +0 -118
  63. orca_sdk/_generated_api_client/api/default/healthcheck_gpu_get.py +0 -118
  64. orca_sdk/_generated_api_client/api/finetuned_embedding_model/__init__.py +0 -0
  65. orca_sdk/_generated_api_client/api/finetuned_embedding_model/create_finetuned_embedding_model_finetuned_embedding_model_post.py +0 -168
  66. orca_sdk/_generated_api_client/api/finetuned_embedding_model/delete_finetuned_embedding_model_finetuned_embedding_model_name_or_id_delete.py +0 -156
  67. orca_sdk/_generated_api_client/api/finetuned_embedding_model/embed_with_finetuned_model_gpu_finetuned_embedding_model_name_or_id_embedding_post.py +0 -189
  68. orca_sdk/_generated_api_client/api/finetuned_embedding_model/get_finetuned_embedding_model_finetuned_embedding_model_name_or_id_get.py +0 -156
  69. orca_sdk/_generated_api_client/api/finetuned_embedding_model/list_finetuned_embedding_models_finetuned_embedding_model_get.py +0 -127
  70. orca_sdk/_generated_api_client/api/memoryset/__init__.py +0 -0
  71. orca_sdk/_generated_api_client/api/memoryset/clone_memoryset_memoryset_name_or_id_clone_post.py +0 -181
  72. orca_sdk/_generated_api_client/api/memoryset/create_analysis_memoryset_name_or_id_analysis_post.py +0 -183
  73. orca_sdk/_generated_api_client/api/memoryset/create_memoryset_memoryset_post.py +0 -168
  74. orca_sdk/_generated_api_client/api/memoryset/delete_memories_memoryset_name_or_id_memories_delete_post.py +0 -181
  75. orca_sdk/_generated_api_client/api/memoryset/delete_memory_memoryset_name_or_id_memory_memory_id_delete.py +0 -167
  76. orca_sdk/_generated_api_client/api/memoryset/delete_memoryset_memoryset_name_or_id_delete.py +0 -156
  77. orca_sdk/_generated_api_client/api/memoryset/get_analysis_memoryset_name_or_id_analysis_analysis_task_id_get.py +0 -169
  78. orca_sdk/_generated_api_client/api/memoryset/get_memories_memoryset_name_or_id_memories_get_post.py +0 -188
  79. orca_sdk/_generated_api_client/api/memoryset/get_memory_memoryset_name_or_id_memory_memory_id_get.py +0 -169
  80. orca_sdk/_generated_api_client/api/memoryset/get_memoryset_memoryset_name_or_id_get.py +0 -156
  81. orca_sdk/_generated_api_client/api/memoryset/insert_memories_gpu_memoryset_name_or_id_memory_post.py +0 -184
  82. orca_sdk/_generated_api_client/api/memoryset/list_analyses_memoryset_name_or_id_analysis_get.py +0 -260
  83. orca_sdk/_generated_api_client/api/memoryset/list_memorysets_memoryset_get.py +0 -127
  84. orca_sdk/_generated_api_client/api/memoryset/memoryset_lookup_gpu_memoryset_name_or_id_lookup_post.py +0 -193
  85. orca_sdk/_generated_api_client/api/memoryset/query_memoryset_memoryset_name_or_id_memories_post.py +0 -188
  86. orca_sdk/_generated_api_client/api/memoryset/update_memories_gpu_memoryset_name_or_id_memories_patch.py +0 -191
  87. orca_sdk/_generated_api_client/api/memoryset/update_memory_gpu_memoryset_name_or_id_memory_patch.py +0 -187
  88. orca_sdk/_generated_api_client/api/pretrained_embedding_model/__init__.py +0 -0
  89. orca_sdk/_generated_api_client/api/pretrained_embedding_model/embed_with_pretrained_model_gpu_pretrained_embedding_model_model_name_embedding_post.py +0 -188
  90. orca_sdk/_generated_api_client/api/pretrained_embedding_model/get_pretrained_embedding_model_pretrained_embedding_model_model_name_get.py +0 -157
  91. orca_sdk/_generated_api_client/api/pretrained_embedding_model/list_pretrained_embedding_models_pretrained_embedding_model_get.py +0 -127
  92. orca_sdk/_generated_api_client/api/task/__init__.py +0 -0
  93. orca_sdk/_generated_api_client/api/task/abort_task_task_task_id_abort_delete.py +0 -154
  94. orca_sdk/_generated_api_client/api/task/get_task_status_task_task_id_status_get.py +0 -156
  95. orca_sdk/_generated_api_client/api/task/list_tasks_task_get.py +0 -243
  96. orca_sdk/_generated_api_client/api/telemetry/__init__.py +0 -0
  97. orca_sdk/_generated_api_client/api/telemetry/drop_feedback_category_with_data_telemetry_feedback_category_name_or_id_delete.py +0 -162
  98. orca_sdk/_generated_api_client/api/telemetry/get_feedback_category_telemetry_feedback_category_name_or_id_get.py +0 -156
  99. orca_sdk/_generated_api_client/api/telemetry/get_prediction_telemetry_prediction_prediction_id_get.py +0 -157
  100. orca_sdk/_generated_api_client/api/telemetry/list_feedback_categories_telemetry_feedback_category_get.py +0 -127
  101. orca_sdk/_generated_api_client/api/telemetry/list_predictions_telemetry_prediction_post.py +0 -175
  102. orca_sdk/_generated_api_client/api/telemetry/record_prediction_feedback_telemetry_prediction_feedback_put.py +0 -171
  103. orca_sdk/_generated_api_client/api/telemetry/update_prediction_telemetry_prediction_prediction_id_patch.py +0 -181
  104. orca_sdk/_generated_api_client/client.py +0 -216
  105. orca_sdk/_generated_api_client/errors.py +0 -38
  106. orca_sdk/_generated_api_client/models/__init__.py +0 -159
  107. orca_sdk/_generated_api_client/models/analyze_neighbor_labels_result.py +0 -84
  108. orca_sdk/_generated_api_client/models/api_key_metadata.py +0 -118
  109. orca_sdk/_generated_api_client/models/base_model.py +0 -55
  110. orca_sdk/_generated_api_client/models/body_create_datasource_datasource_post.py +0 -176
  111. orca_sdk/_generated_api_client/models/classification_evaluation_result.py +0 -114
  112. orca_sdk/_generated_api_client/models/clone_labeled_memoryset_request.py +0 -150
  113. orca_sdk/_generated_api_client/models/column_info.py +0 -114
  114. orca_sdk/_generated_api_client/models/column_type.py +0 -14
  115. orca_sdk/_generated_api_client/models/conflict_error_response.py +0 -80
  116. orca_sdk/_generated_api_client/models/create_api_key_request.py +0 -99
  117. orca_sdk/_generated_api_client/models/create_api_key_response.py +0 -126
  118. orca_sdk/_generated_api_client/models/create_labeled_memoryset_request.py +0 -259
  119. orca_sdk/_generated_api_client/models/create_rac_model_request.py +0 -209
  120. orca_sdk/_generated_api_client/models/datasource_metadata.py +0 -142
  121. orca_sdk/_generated_api_client/models/delete_memories_request.py +0 -70
  122. orca_sdk/_generated_api_client/models/embed_request.py +0 -127
  123. orca_sdk/_generated_api_client/models/embedding_finetuning_method.py +0 -9
  124. orca_sdk/_generated_api_client/models/evaluation_request.py +0 -180
  125. orca_sdk/_generated_api_client/models/evaluation_response.py +0 -140
  126. orca_sdk/_generated_api_client/models/feedback_type.py +0 -9
  127. orca_sdk/_generated_api_client/models/field_validation_error.py +0 -103
  128. orca_sdk/_generated_api_client/models/filter_item.py +0 -231
  129. orca_sdk/_generated_api_client/models/filter_item_field_type_0_item.py +0 -15
  130. orca_sdk/_generated_api_client/models/filter_item_field_type_2_item_type_1.py +0 -16
  131. orca_sdk/_generated_api_client/models/filter_item_op.py +0 -16
  132. orca_sdk/_generated_api_client/models/find_duplicates_analysis_result.py +0 -70
  133. orca_sdk/_generated_api_client/models/finetune_embedding_model_request.py +0 -259
  134. orca_sdk/_generated_api_client/models/finetune_embedding_model_request_training_args.py +0 -66
  135. orca_sdk/_generated_api_client/models/finetuned_embedding_model_metadata.py +0 -166
  136. orca_sdk/_generated_api_client/models/get_memories_request.py +0 -70
  137. orca_sdk/_generated_api_client/models/internal_server_error_response.py +0 -80
  138. orca_sdk/_generated_api_client/models/label_class_metrics.py +0 -108
  139. orca_sdk/_generated_api_client/models/label_prediction_memory_lookup.py +0 -274
  140. orca_sdk/_generated_api_client/models/label_prediction_memory_lookup_metadata.py +0 -68
  141. orca_sdk/_generated_api_client/models/label_prediction_result.py +0 -101
  142. orca_sdk/_generated_api_client/models/label_prediction_with_memories_and_feedback.py +0 -232
  143. orca_sdk/_generated_api_client/models/labeled_memory.py +0 -197
  144. orca_sdk/_generated_api_client/models/labeled_memory_insert.py +0 -108
  145. orca_sdk/_generated_api_client/models/labeled_memory_insert_metadata.py +0 -68
  146. orca_sdk/_generated_api_client/models/labeled_memory_lookup.py +0 -258
  147. orca_sdk/_generated_api_client/models/labeled_memory_lookup_metadata.py +0 -68
  148. orca_sdk/_generated_api_client/models/labeled_memory_metadata.py +0 -68
  149. orca_sdk/_generated_api_client/models/labeled_memory_metrics.py +0 -277
  150. orca_sdk/_generated_api_client/models/labeled_memory_update.py +0 -171
  151. orca_sdk/_generated_api_client/models/labeled_memory_update_metadata_type_0.py +0 -68
  152. orca_sdk/_generated_api_client/models/labeled_memoryset_metadata.py +0 -195
  153. orca_sdk/_generated_api_client/models/list_analyses_memoryset_name_or_id_analysis_get_type_type_0.py +0 -9
  154. orca_sdk/_generated_api_client/models/list_memories_request.py +0 -104
  155. orca_sdk/_generated_api_client/models/list_predictions_request.py +0 -234
  156. orca_sdk/_generated_api_client/models/list_predictions_request_sort_item_item_type_0.py +0 -9
  157. orca_sdk/_generated_api_client/models/list_predictions_request_sort_item_item_type_1.py +0 -9
  158. orca_sdk/_generated_api_client/models/lookup_request.py +0 -81
  159. orca_sdk/_generated_api_client/models/memoryset_analysis_request.py +0 -83
  160. orca_sdk/_generated_api_client/models/memoryset_analysis_request_type.py +0 -9
  161. orca_sdk/_generated_api_client/models/memoryset_analysis_response.py +0 -180
  162. orca_sdk/_generated_api_client/models/memoryset_analysis_response_config.py +0 -66
  163. orca_sdk/_generated_api_client/models/memoryset_analysis_response_type.py +0 -9
  164. orca_sdk/_generated_api_client/models/not_found_error_response.py +0 -100
  165. orca_sdk/_generated_api_client/models/not_found_error_response_resource_type_0.py +0 -20
  166. orca_sdk/_generated_api_client/models/prediction_feedback.py +0 -157
  167. orca_sdk/_generated_api_client/models/prediction_feedback_category.py +0 -115
  168. orca_sdk/_generated_api_client/models/prediction_feedback_request.py +0 -122
  169. orca_sdk/_generated_api_client/models/prediction_feedback_result.py +0 -102
  170. orca_sdk/_generated_api_client/models/prediction_request.py +0 -169
  171. orca_sdk/_generated_api_client/models/pretrained_embedding_model_metadata.py +0 -97
  172. orca_sdk/_generated_api_client/models/pretrained_embedding_model_name.py +0 -11
  173. orca_sdk/_generated_api_client/models/rac_head_type.py +0 -11
  174. orca_sdk/_generated_api_client/models/rac_model_metadata.py +0 -191
  175. orca_sdk/_generated_api_client/models/service_unavailable_error_response.py +0 -80
  176. orca_sdk/_generated_api_client/models/task.py +0 -198
  177. orca_sdk/_generated_api_client/models/task_status.py +0 -14
  178. orca_sdk/_generated_api_client/models/task_status_info.py +0 -133
  179. orca_sdk/_generated_api_client/models/unauthenticated_error_response.py +0 -72
  180. orca_sdk/_generated_api_client/models/unauthorized_error_response.py +0 -80
  181. orca_sdk/_generated_api_client/models/unprocessable_input_error_response.py +0 -94
  182. orca_sdk/_generated_api_client/models/update_prediction_request.py +0 -93
  183. orca_sdk/_generated_api_client/py.typed +0 -1
  184. orca_sdk/_generated_api_client/types.py +0 -56
  185. orca_sdk/_utils/task.py +0 -73
  186. orca_sdk-0.1.1.dist-info/RECORD +0 -175
@@ -1,114 +0,0 @@
1
- """
2
- This file is generated by the openapi-python-client tool via the generate_api_client.py script
3
-
4
- It is a customized template from the openapi-python-client tool's default template:
5
- https://github.com/openapi-generators/openapi-python-client/blob/861ef5622f10fc96d240dc9becb0edf94e61446c/openapi_python_client/templates/model.py.jinja
6
-
7
- The main change is:
8
- - Fix typing issues
9
- """
10
-
11
- # flake8: noqa: C901
12
-
13
- from typing import Any, Type, TypeVar, Union, cast
14
-
15
- from attrs import define as _attrs_define
16
- from attrs import field as _attrs_field
17
-
18
- T = TypeVar("T", bound="ClassificationEvaluationResult")
19
-
20
-
21
- @_attrs_define
22
- class ClassificationEvaluationResult:
23
- """
24
- Attributes:
25
- f1_score (float):
26
- roc_auc (Union[None, float]):
27
- pr_auc (Union[None, float]):
28
- accuracy (float):
29
- loss (float):
30
- """
31
-
32
- f1_score: float
33
- roc_auc: Union[None, float]
34
- pr_auc: Union[None, float]
35
- accuracy: float
36
- loss: float
37
- additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
38
-
39
- def to_dict(self) -> dict[str, Any]:
40
- f1_score = self.f1_score
41
-
42
- roc_auc: Union[None, float]
43
- roc_auc = self.roc_auc
44
-
45
- pr_auc: Union[None, float]
46
- pr_auc = self.pr_auc
47
-
48
- accuracy = self.accuracy
49
-
50
- loss = self.loss
51
-
52
- field_dict: dict[str, Any] = {}
53
- field_dict.update(self.additional_properties)
54
- field_dict.update(
55
- {
56
- "f1_score": f1_score,
57
- "roc_auc": roc_auc,
58
- "pr_auc": pr_auc,
59
- "accuracy": accuracy,
60
- "loss": loss,
61
- }
62
- )
63
-
64
- return field_dict
65
-
66
- @classmethod
67
- def from_dict(cls: Type[T], src_dict: dict[str, Any]) -> T:
68
- d = src_dict.copy()
69
- f1_score = d.pop("f1_score")
70
-
71
- def _parse_roc_auc(data: object) -> Union[None, float]:
72
- if data is None:
73
- return data
74
- return cast(Union[None, float], data)
75
-
76
- roc_auc = _parse_roc_auc(d.pop("roc_auc"))
77
-
78
- def _parse_pr_auc(data: object) -> Union[None, float]:
79
- if data is None:
80
- return data
81
- return cast(Union[None, float], data)
82
-
83
- pr_auc = _parse_pr_auc(d.pop("pr_auc"))
84
-
85
- accuracy = d.pop("accuracy")
86
-
87
- loss = d.pop("loss")
88
-
89
- classification_evaluation_result = cls(
90
- f1_score=f1_score,
91
- roc_auc=roc_auc,
92
- pr_auc=pr_auc,
93
- accuracy=accuracy,
94
- loss=loss,
95
- )
96
-
97
- classification_evaluation_result.additional_properties = d
98
- return classification_evaluation_result
99
-
100
- @property
101
- def additional_keys(self) -> list[str]:
102
- return list(self.additional_properties.keys())
103
-
104
- def __getitem__(self, key: str) -> Any:
105
- return self.additional_properties[key]
106
-
107
- def __setitem__(self, key: str, value: Any) -> None:
108
- self.additional_properties[key] = value
109
-
110
- def __delitem__(self, key: str) -> None:
111
- del self.additional_properties[key]
112
-
113
- def __contains__(self, key: str) -> bool:
114
- return key in self.additional_properties
@@ -1,150 +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 enum import Enum
14
- from typing import Any, Type, TypeVar, Union, cast
15
-
16
- from attrs import define as _attrs_define
17
- from attrs import field as _attrs_field
18
-
19
- from ..models.pretrained_embedding_model_name import PretrainedEmbeddingModelName
20
- from ..types import UNSET, Unset
21
-
22
- T = TypeVar("T", bound="CloneLabeledMemorysetRequest")
23
-
24
-
25
- @_attrs_define
26
- class CloneLabeledMemorysetRequest:
27
- """
28
- Attributes:
29
- name (str):
30
- pretrained_embedding_model_name (Union[None, PretrainedEmbeddingModelName, Unset]):
31
- finetuned_embedding_model_id (Union[None, Unset, str]):
32
- max_seq_length_override (Union[None, Unset, int]):
33
- """
34
-
35
- name: str
36
- pretrained_embedding_model_name: Union[None, PretrainedEmbeddingModelName, Unset] = UNSET
37
- finetuned_embedding_model_id: Union[None, Unset, str] = UNSET
38
- max_seq_length_override: Union[None, Unset, int] = UNSET
39
- additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
40
-
41
- def to_dict(self) -> dict[str, Any]:
42
- name = self.name
43
-
44
- pretrained_embedding_model_name: Union[None, Unset, str]
45
- if isinstance(self.pretrained_embedding_model_name, Unset):
46
- pretrained_embedding_model_name = UNSET
47
- elif isinstance(self.pretrained_embedding_model_name, PretrainedEmbeddingModelName):
48
- pretrained_embedding_model_name = (
49
- self.pretrained_embedding_model_name.value
50
- if isinstance(self.pretrained_embedding_model_name, Enum)
51
- else self.pretrained_embedding_model_name
52
- )
53
- else:
54
- pretrained_embedding_model_name = self.pretrained_embedding_model_name
55
-
56
- finetuned_embedding_model_id: Union[None, Unset, str]
57
- if isinstance(self.finetuned_embedding_model_id, Unset):
58
- finetuned_embedding_model_id = UNSET
59
- else:
60
- finetuned_embedding_model_id = self.finetuned_embedding_model_id
61
-
62
- max_seq_length_override: Union[None, Unset, int]
63
- if isinstance(self.max_seq_length_override, Unset):
64
- max_seq_length_override = UNSET
65
- else:
66
- max_seq_length_override = self.max_seq_length_override
67
-
68
- field_dict: dict[str, Any] = {}
69
- field_dict.update(self.additional_properties)
70
- field_dict.update(
71
- {
72
- "name": name,
73
- }
74
- )
75
- if pretrained_embedding_model_name is not UNSET:
76
- field_dict["pretrained_embedding_model_name"] = pretrained_embedding_model_name
77
- if finetuned_embedding_model_id is not UNSET:
78
- field_dict["finetuned_embedding_model_id"] = finetuned_embedding_model_id
79
- if max_seq_length_override is not UNSET:
80
- field_dict["max_seq_length_override"] = max_seq_length_override
81
-
82
- return field_dict
83
-
84
- @classmethod
85
- def from_dict(cls: Type[T], src_dict: dict[str, Any]) -> T:
86
- d = src_dict.copy()
87
- name = d.pop("name")
88
-
89
- def _parse_pretrained_embedding_model_name(data: object) -> Union[None, PretrainedEmbeddingModelName, Unset]:
90
- if data is None:
91
- return data
92
- if isinstance(data, Unset):
93
- return data
94
- try:
95
- if not isinstance(data, str):
96
- raise TypeError()
97
- pretrained_embedding_model_name_type_0 = PretrainedEmbeddingModelName(data)
98
-
99
- return pretrained_embedding_model_name_type_0
100
- except: # noqa: E722
101
- pass
102
- return cast(Union[None, PretrainedEmbeddingModelName, Unset], data)
103
-
104
- pretrained_embedding_model_name = _parse_pretrained_embedding_model_name(
105
- d.pop("pretrained_embedding_model_name", UNSET)
106
- )
107
-
108
- def _parse_finetuned_embedding_model_id(data: object) -> Union[None, Unset, str]:
109
- if data is None:
110
- return data
111
- if isinstance(data, Unset):
112
- return data
113
- return cast(Union[None, Unset, str], data)
114
-
115
- finetuned_embedding_model_id = _parse_finetuned_embedding_model_id(d.pop("finetuned_embedding_model_id", UNSET))
116
-
117
- def _parse_max_seq_length_override(data: object) -> Union[None, Unset, int]:
118
- if data is None:
119
- return data
120
- if isinstance(data, Unset):
121
- return data
122
- return cast(Union[None, Unset, int], data)
123
-
124
- max_seq_length_override = _parse_max_seq_length_override(d.pop("max_seq_length_override", UNSET))
125
-
126
- clone_labeled_memoryset_request = cls(
127
- name=name,
128
- pretrained_embedding_model_name=pretrained_embedding_model_name,
129
- finetuned_embedding_model_id=finetuned_embedding_model_id,
130
- max_seq_length_override=max_seq_length_override,
131
- )
132
-
133
- clone_labeled_memoryset_request.additional_properties = d
134
- return clone_labeled_memoryset_request
135
-
136
- @property
137
- def additional_keys(self) -> list[str]:
138
- return list(self.additional_properties.keys())
139
-
140
- def __getitem__(self, key: str) -> Any:
141
- return self.additional_properties[key]
142
-
143
- def __setitem__(self, key: str, value: Any) -> None:
144
- self.additional_properties[key] = value
145
-
146
- def __delitem__(self, key: str) -> None:
147
- del self.additional_properties[key]
148
-
149
- def __contains__(self, key: str) -> bool:
150
- return key in self.additional_properties
@@ -1,114 +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 enum import Enum
14
- from typing import Any, List, Type, TypeVar, Union, cast
15
-
16
- from attrs import define as _attrs_define
17
- from attrs import field as _attrs_field
18
-
19
- from ..models.column_type import ColumnType
20
- from ..types import UNSET, Unset
21
-
22
- T = TypeVar("T", bound="ColumnInfo")
23
-
24
-
25
- @_attrs_define
26
- class ColumnInfo:
27
- """Information about a column in a datasource
28
-
29
- Attributes:
30
- name (str):
31
- type (ColumnType): The type of a column in a datasource
32
- enum_options (Union[List[str], None, Unset]):
33
- """
34
-
35
- name: str
36
- type: ColumnType
37
- enum_options: Union[List[str], None, Unset] = UNSET
38
- additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
39
-
40
- def to_dict(self) -> dict[str, Any]:
41
- name = self.name
42
-
43
- type = self.type.value if isinstance(self.type, Enum) else self.type
44
-
45
- enum_options: Union[List[str], None, Unset]
46
- if isinstance(self.enum_options, Unset):
47
- enum_options = UNSET
48
- elif isinstance(self.enum_options, list):
49
- enum_options = self.enum_options
50
-
51
- else:
52
- enum_options = self.enum_options
53
-
54
- field_dict: dict[str, Any] = {}
55
- field_dict.update(self.additional_properties)
56
- field_dict.update(
57
- {
58
- "name": name,
59
- "type": type,
60
- }
61
- )
62
- if enum_options is not UNSET:
63
- field_dict["enum_options"] = enum_options
64
-
65
- return field_dict
66
-
67
- @classmethod
68
- def from_dict(cls: Type[T], src_dict: dict[str, Any]) -> T:
69
- d = src_dict.copy()
70
- name = d.pop("name")
71
-
72
- type = ColumnType(d.pop("type"))
73
-
74
- def _parse_enum_options(data: object) -> Union[List[str], None, Unset]:
75
- if data is None:
76
- return data
77
- if isinstance(data, Unset):
78
- return data
79
- try:
80
- if not isinstance(data, list):
81
- raise TypeError()
82
- enum_options_type_0 = cast(List[str], data)
83
-
84
- return enum_options_type_0
85
- except: # noqa: E722
86
- pass
87
- return cast(Union[List[str], None, Unset], data)
88
-
89
- enum_options = _parse_enum_options(d.pop("enum_options", UNSET))
90
-
91
- column_info = cls(
92
- name=name,
93
- type=type,
94
- enum_options=enum_options,
95
- )
96
-
97
- column_info.additional_properties = d
98
- return column_info
99
-
100
- @property
101
- def additional_keys(self) -> list[str]:
102
- return list(self.additional_properties.keys())
103
-
104
- def __getitem__(self, key: str) -> Any:
105
- return self.additional_properties[key]
106
-
107
- def __setitem__(self, key: str, value: Any) -> None:
108
- self.additional_properties[key] = value
109
-
110
- def __delitem__(self, key: str) -> None:
111
- del self.additional_properties[key]
112
-
113
- def __contains__(self, key: str) -> bool:
114
- return key in self.additional_properties
@@ -1,14 +0,0 @@
1
- from enum import Enum
2
-
3
-
4
- class ColumnType(str, Enum):
5
- BOOL = "BOOL"
6
- ENUM = "ENUM"
7
- FLOAT = "FLOAT"
8
- IMAGE = "IMAGE"
9
- INT = "INT"
10
- OTHER = "OTHER"
11
- STRING = "STRING"
12
-
13
- def __str__(self) -> str:
14
- return str(self.value)
@@ -1,80 +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, Literal, Type, TypeVar, cast
14
-
15
- from attrs import define as _attrs_define
16
- from attrs import field as _attrs_field
17
-
18
- T = TypeVar("T", bound="ConflictErrorResponse")
19
-
20
-
21
- @_attrs_define
22
- class ConflictErrorResponse:
23
- """
24
- Attributes:
25
- status_code (Literal[409]):
26
- constraint (str):
27
- """
28
-
29
- status_code: Literal[409]
30
- constraint: str
31
- additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
32
-
33
- def to_dict(self) -> dict[str, Any]:
34
- status_code = self.status_code
35
-
36
- constraint = self.constraint
37
-
38
- field_dict: dict[str, Any] = {}
39
- field_dict.update(self.additional_properties)
40
- field_dict.update(
41
- {
42
- "status_code": status_code,
43
- "constraint": constraint,
44
- }
45
- )
46
-
47
- return field_dict
48
-
49
- @classmethod
50
- def from_dict(cls: Type[T], src_dict: dict[str, Any]) -> T:
51
- d = src_dict.copy()
52
- status_code = cast(Literal[409], d.pop("status_code"))
53
- if status_code != 409:
54
- raise ValueError(f"status_code must match const 409, got '{status_code}'")
55
-
56
- constraint = d.pop("constraint")
57
-
58
- conflict_error_response = cls(
59
- status_code=status_code,
60
- constraint=constraint,
61
- )
62
-
63
- conflict_error_response.additional_properties = d
64
- return conflict_error_response
65
-
66
- @property
67
- def additional_keys(self) -> list[str]:
68
- return list(self.additional_properties.keys())
69
-
70
- def __getitem__(self, key: str) -> Any:
71
- return self.additional_properties[key]
72
-
73
- def __setitem__(self, key: str, value: Any) -> None:
74
- self.additional_properties[key] = value
75
-
76
- def __delitem__(self, key: str) -> None:
77
- del self.additional_properties[key]
78
-
79
- def __contains__(self, key: str) -> bool:
80
- return key in self.additional_properties
@@ -1,99 +0,0 @@
1
- """
2
- This file is generated by the openapi-python-client tool via the generate_api_client.py script
3
-
4
- It is a customized template from the openapi-python-client tool's default template:
5
- https://github.com/openapi-generators/openapi-python-client/blob/861ef5622f10fc96d240dc9becb0edf94e61446c/openapi_python_client/templates/model.py.jinja
6
-
7
- The main change is:
8
- - Fix typing issues
9
- """
10
-
11
- # flake8: noqa: C901
12
-
13
- from typing import Any, Type, TypeVar, Union, cast
14
-
15
- from attrs import define as _attrs_define
16
- from attrs import field as _attrs_field
17
-
18
- from ..types import UNSET, Unset
19
-
20
- T = TypeVar("T", bound="CreateApiKeyRequest")
21
-
22
-
23
- @_attrs_define
24
- class CreateApiKeyRequest:
25
- """
26
- Attributes:
27
- id (Union[Unset, str]):
28
- name (Union[Unset, str]): Default: 'default'.
29
- created_by (Union[None, Unset, str]):
30
- """
31
-
32
- id: Union[Unset, str] = UNSET
33
- name: Union[Unset, str] = "default"
34
- created_by: Union[None, Unset, str] = UNSET
35
- additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
36
-
37
- def to_dict(self) -> dict[str, Any]:
38
- id = self.id
39
-
40
- name = self.name
41
-
42
- created_by: Union[None, Unset, str]
43
- if isinstance(self.created_by, Unset):
44
- created_by = UNSET
45
- else:
46
- created_by = self.created_by
47
-
48
- field_dict: dict[str, Any] = {}
49
- field_dict.update(self.additional_properties)
50
- field_dict.update({})
51
- if id is not UNSET:
52
- field_dict["id"] = id
53
- if name is not UNSET:
54
- field_dict["name"] = name
55
- if created_by is not UNSET:
56
- field_dict["created_by"] = created_by
57
-
58
- return field_dict
59
-
60
- @classmethod
61
- def from_dict(cls: Type[T], src_dict: dict[str, Any]) -> T:
62
- d = src_dict.copy()
63
- id = d.pop("id", UNSET)
64
-
65
- name = d.pop("name", UNSET)
66
-
67
- def _parse_created_by(data: object) -> Union[None, Unset, str]:
68
- if data is None:
69
- return data
70
- if isinstance(data, Unset):
71
- return data
72
- return cast(Union[None, Unset, str], data)
73
-
74
- created_by = _parse_created_by(d.pop("created_by", UNSET))
75
-
76
- create_api_key_request = cls(
77
- id=id,
78
- name=name,
79
- created_by=created_by,
80
- )
81
-
82
- create_api_key_request.additional_properties = d
83
- return create_api_key_request
84
-
85
- @property
86
- def additional_keys(self) -> list[str]:
87
- return list(self.additional_properties.keys())
88
-
89
- def __getitem__(self, key: str) -> Any:
90
- return self.additional_properties[key]
91
-
92
- def __setitem__(self, key: str, value: Any) -> None:
93
- self.additional_properties[key] = value
94
-
95
- def __delitem__(self, key: str) -> None:
96
- del self.additional_properties[key]
97
-
98
- def __contains__(self, key: str) -> bool:
99
- return key in self.additional_properties
@@ -1,126 +0,0 @@
1
- """
2
- This file is generated by the openapi-python-client tool via the generate_api_client.py script
3
-
4
- It is a customized template from the openapi-python-client tool's default template:
5
- https://github.com/openapi-generators/openapi-python-client/blob/861ef5622f10fc96d240dc9becb0edf94e61446c/openapi_python_client/templates/model.py.jinja
6
-
7
- The main change is:
8
- - Fix typing issues
9
- """
10
-
11
- # flake8: noqa: C901
12
-
13
- import datetime
14
- from typing import Any, Type, TypeVar, Union, cast
15
-
16
- from attrs import define as _attrs_define
17
- from attrs import field as _attrs_field
18
- from dateutil.parser import isoparse
19
-
20
- T = TypeVar("T", bound="CreateApiKeyResponse")
21
-
22
-
23
- @_attrs_define
24
- class CreateApiKeyResponse:
25
- """
26
- Attributes:
27
- id (str):
28
- org_id (str):
29
- name (str):
30
- created_by (Union[None, str]):
31
- created_at (datetime.datetime):
32
- updated_at (datetime.datetime):
33
- api_key (str):
34
- """
35
-
36
- id: str
37
- org_id: str
38
- name: str
39
- created_by: Union[None, str]
40
- created_at: datetime.datetime
41
- updated_at: datetime.datetime
42
- api_key: str
43
- additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
44
-
45
- def to_dict(self) -> dict[str, Any]:
46
- id = self.id
47
-
48
- org_id = self.org_id
49
-
50
- name = self.name
51
-
52
- created_by: Union[None, str]
53
- created_by = self.created_by
54
-
55
- created_at = self.created_at.isoformat()
56
-
57
- updated_at = self.updated_at.isoformat()
58
-
59
- api_key = self.api_key
60
-
61
- field_dict: dict[str, Any] = {}
62
- field_dict.update(self.additional_properties)
63
- field_dict.update(
64
- {
65
- "id": id,
66
- "org_id": org_id,
67
- "name": name,
68
- "created_by": created_by,
69
- "created_at": created_at,
70
- "updated_at": updated_at,
71
- "api_key": api_key,
72
- }
73
- )
74
-
75
- return field_dict
76
-
77
- @classmethod
78
- def from_dict(cls: Type[T], src_dict: dict[str, Any]) -> T:
79
- d = src_dict.copy()
80
- id = d.pop("id")
81
-
82
- org_id = d.pop("org_id")
83
-
84
- name = d.pop("name")
85
-
86
- def _parse_created_by(data: object) -> Union[None, str]:
87
- if data is None:
88
- return data
89
- return cast(Union[None, str], data)
90
-
91
- created_by = _parse_created_by(d.pop("created_by"))
92
-
93
- created_at = isoparse(d.pop("created_at"))
94
-
95
- updated_at = isoparse(d.pop("updated_at"))
96
-
97
- api_key = d.pop("api_key")
98
-
99
- create_api_key_response = cls(
100
- id=id,
101
- org_id=org_id,
102
- name=name,
103
- created_by=created_by,
104
- created_at=created_at,
105
- updated_at=updated_at,
106
- api_key=api_key,
107
- )
108
-
109
- create_api_key_response.additional_properties = d
110
- return create_api_key_response
111
-
112
- @property
113
- def additional_keys(self) -> list[str]:
114
- return list(self.additional_properties.keys())
115
-
116
- def __getitem__(self, key: str) -> Any:
117
- return self.additional_properties[key]
118
-
119
- def __setitem__(self, key: str, value: Any) -> None:
120
- self.additional_properties[key] = value
121
-
122
- def __delitem__(self, key: str) -> None:
123
- del self.additional_properties[key]
124
-
125
- def __contains__(self, key: str) -> bool:
126
- return key in self.additional_properties