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,191 +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 enum import Enum
15
- from typing import Any, Type, TypeVar, Union, cast
16
-
17
- from attrs import define as _attrs_define
18
- from attrs import field as _attrs_field
19
- from dateutil.parser import isoparse
20
-
21
- from ..models.rac_head_type import RACHeadType
22
-
23
- T = TypeVar("T", bound="RACModelMetadata")
24
-
25
-
26
- @_attrs_define
27
- class RACModelMetadata:
28
- """
29
- Attributes:
30
- id (str):
31
- org_id (str):
32
- name (str):
33
- version (int):
34
- num_classes (int):
35
- head_type (RACHeadType):
36
- memoryset_id (str):
37
- memory_lookup_count (int):
38
- weigh_memories (Union[None, bool]):
39
- min_memory_weight (Union[None, float]):
40
- storage_path (str):
41
- memoryset_collection_name (str):
42
- created_at (datetime.datetime):
43
- updated_at (datetime.datetime):
44
- """
45
-
46
- id: str
47
- org_id: str
48
- name: str
49
- version: int
50
- num_classes: int
51
- head_type: RACHeadType
52
- memoryset_id: str
53
- memory_lookup_count: int
54
- weigh_memories: Union[None, bool]
55
- min_memory_weight: Union[None, float]
56
- storage_path: str
57
- memoryset_collection_name: str
58
- created_at: datetime.datetime
59
- updated_at: datetime.datetime
60
- additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
61
-
62
- def to_dict(self) -> dict[str, Any]:
63
- id = self.id
64
-
65
- org_id = self.org_id
66
-
67
- name = self.name
68
-
69
- version = self.version
70
-
71
- num_classes = self.num_classes
72
-
73
- head_type = self.head_type.value if isinstance(self.head_type, Enum) else self.head_type
74
-
75
- memoryset_id = self.memoryset_id
76
-
77
- memory_lookup_count = self.memory_lookup_count
78
-
79
- weigh_memories: Union[None, bool]
80
- weigh_memories = self.weigh_memories
81
-
82
- min_memory_weight: Union[None, float]
83
- min_memory_weight = self.min_memory_weight
84
-
85
- storage_path = self.storage_path
86
-
87
- memoryset_collection_name = self.memoryset_collection_name
88
-
89
- created_at = self.created_at.isoformat()
90
-
91
- updated_at = self.updated_at.isoformat()
92
-
93
- field_dict: dict[str, Any] = {}
94
- field_dict.update(self.additional_properties)
95
- field_dict.update(
96
- {
97
- "id": id,
98
- "org_id": org_id,
99
- "name": name,
100
- "version": version,
101
- "num_classes": num_classes,
102
- "head_type": head_type,
103
- "memoryset_id": memoryset_id,
104
- "memory_lookup_count": memory_lookup_count,
105
- "weigh_memories": weigh_memories,
106
- "min_memory_weight": min_memory_weight,
107
- "storage_path": storage_path,
108
- "memoryset_collection_name": memoryset_collection_name,
109
- "created_at": created_at,
110
- "updated_at": updated_at,
111
- }
112
- )
113
-
114
- return field_dict
115
-
116
- @classmethod
117
- def from_dict(cls: Type[T], src_dict: dict[str, Any]) -> T:
118
- d = src_dict.copy()
119
- id = d.pop("id")
120
-
121
- org_id = d.pop("org_id")
122
-
123
- name = d.pop("name")
124
-
125
- version = d.pop("version")
126
-
127
- num_classes = d.pop("num_classes")
128
-
129
- head_type = RACHeadType(d.pop("head_type"))
130
-
131
- memoryset_id = d.pop("memoryset_id")
132
-
133
- memory_lookup_count = d.pop("memory_lookup_count")
134
-
135
- def _parse_weigh_memories(data: object) -> Union[None, bool]:
136
- if data is None:
137
- return data
138
- return cast(Union[None, bool], data)
139
-
140
- weigh_memories = _parse_weigh_memories(d.pop("weigh_memories"))
141
-
142
- def _parse_min_memory_weight(data: object) -> Union[None, float]:
143
- if data is None:
144
- return data
145
- return cast(Union[None, float], data)
146
-
147
- min_memory_weight = _parse_min_memory_weight(d.pop("min_memory_weight"))
148
-
149
- storage_path = d.pop("storage_path")
150
-
151
- memoryset_collection_name = d.pop("memoryset_collection_name")
152
-
153
- created_at = isoparse(d.pop("created_at"))
154
-
155
- updated_at = isoparse(d.pop("updated_at"))
156
-
157
- rac_model_metadata = cls(
158
- id=id,
159
- org_id=org_id,
160
- name=name,
161
- version=version,
162
- num_classes=num_classes,
163
- head_type=head_type,
164
- memoryset_id=memoryset_id,
165
- memory_lookup_count=memory_lookup_count,
166
- weigh_memories=weigh_memories,
167
- min_memory_weight=min_memory_weight,
168
- storage_path=storage_path,
169
- memoryset_collection_name=memoryset_collection_name,
170
- created_at=created_at,
171
- updated_at=updated_at,
172
- )
173
-
174
- rac_model_metadata.additional_properties = d
175
- return rac_model_metadata
176
-
177
- @property
178
- def additional_keys(self) -> list[str]:
179
- return list(self.additional_properties.keys())
180
-
181
- def __getitem__(self, key: str) -> Any:
182
- return self.additional_properties[key]
183
-
184
- def __setitem__(self, key: str, value: Any) -> None:
185
- self.additional_properties[key] = value
186
-
187
- def __delitem__(self, key: str) -> None:
188
- del self.additional_properties[key]
189
-
190
- def __contains__(self, key: str) -> bool:
191
- return key in self.additional_properties
@@ -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="ServiceUnavailableErrorResponse")
19
-
20
-
21
- @_attrs_define
22
- class ServiceUnavailableErrorResponse:
23
- """
24
- Attributes:
25
- status_code (Literal[503]):
26
- service (str):
27
- """
28
-
29
- status_code: Literal[503]
30
- service: 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
- service = self.service
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
- "service": service,
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[503], d.pop("status_code"))
53
- if status_code != 503:
54
- raise ValueError(f"status_code must match const 503, got '{status_code}'")
55
-
56
- service = d.pop("service")
57
-
58
- service_unavailable_error_response = cls(
59
- status_code=status_code,
60
- service=service,
61
- )
62
-
63
- service_unavailable_error_response.additional_properties = d
64
- return service_unavailable_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,198 +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 enum import Enum
15
- from typing import TYPE_CHECKING, Any, Dict, Type, TypeVar, Union, cast
16
-
17
- from attrs import define as _attrs_define
18
- from attrs import field as _attrs_field
19
- from dateutil.parser import isoparse
20
-
21
- from ..models.task_status import TaskStatus
22
-
23
- if TYPE_CHECKING:
24
- from ..models.base_model import BaseModel
25
-
26
-
27
- T = TypeVar("T", bound="Task")
28
-
29
-
30
- @_attrs_define
31
- class Task:
32
- """
33
- Attributes:
34
- status (TaskStatus): Status of task in the task queue
35
- steps_total (Union[None, int]):
36
- steps_completed (Union[None, int]):
37
- exception (Union[None, str]):
38
- updated_at (datetime.datetime):
39
- created_at (datetime.datetime):
40
- id (str):
41
- org_id (str):
42
- type (str):
43
- payload (BaseModel):
44
- result (Union['BaseModel', None]):
45
- """
46
-
47
- status: TaskStatus
48
- steps_total: Union[None, int]
49
- steps_completed: Union[None, int]
50
- exception: Union[None, str]
51
- updated_at: datetime.datetime
52
- created_at: datetime.datetime
53
- id: str
54
- org_id: str
55
- type: str
56
- payload: "BaseModel"
57
- result: Union["BaseModel", None]
58
- additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
59
-
60
- def to_dict(self) -> dict[str, Any]:
61
- from ..models.base_model import BaseModel
62
-
63
- status = self.status.value if isinstance(self.status, Enum) else self.status
64
-
65
- steps_total: Union[None, int]
66
- steps_total = self.steps_total
67
-
68
- steps_completed: Union[None, int]
69
- steps_completed = self.steps_completed
70
-
71
- exception: Union[None, str]
72
- exception = self.exception
73
-
74
- updated_at = self.updated_at.isoformat()
75
-
76
- created_at = self.created_at.isoformat()
77
-
78
- id = self.id
79
-
80
- org_id = self.org_id
81
-
82
- type = self.type
83
-
84
- payload = self.payload.to_dict()
85
-
86
- result: Union[Dict[str, Any], None]
87
- if isinstance(self.result, BaseModel):
88
- result = self.result.to_dict()
89
- else:
90
- result = self.result
91
-
92
- field_dict: dict[str, Any] = {}
93
- field_dict.update(self.additional_properties)
94
- field_dict.update(
95
- {
96
- "status": status,
97
- "steps_total": steps_total,
98
- "steps_completed": steps_completed,
99
- "exception": exception,
100
- "updated_at": updated_at,
101
- "created_at": created_at,
102
- "id": id,
103
- "org_id": org_id,
104
- "type": type,
105
- "payload": payload,
106
- "result": result,
107
- }
108
- )
109
-
110
- return field_dict
111
-
112
- @classmethod
113
- def from_dict(cls: Type[T], src_dict: dict[str, Any]) -> T:
114
- from ..models.base_model import BaseModel
115
-
116
- d = src_dict.copy()
117
- status = TaskStatus(d.pop("status"))
118
-
119
- def _parse_steps_total(data: object) -> Union[None, int]:
120
- if data is None:
121
- return data
122
- return cast(Union[None, int], data)
123
-
124
- steps_total = _parse_steps_total(d.pop("steps_total"))
125
-
126
- def _parse_steps_completed(data: object) -> Union[None, int]:
127
- if data is None:
128
- return data
129
- return cast(Union[None, int], data)
130
-
131
- steps_completed = _parse_steps_completed(d.pop("steps_completed"))
132
-
133
- def _parse_exception(data: object) -> Union[None, str]:
134
- if data is None:
135
- return data
136
- return cast(Union[None, str], data)
137
-
138
- exception = _parse_exception(d.pop("exception"))
139
-
140
- updated_at = isoparse(d.pop("updated_at"))
141
-
142
- created_at = isoparse(d.pop("created_at"))
143
-
144
- id = d.pop("id")
145
-
146
- org_id = d.pop("org_id")
147
-
148
- type = d.pop("type")
149
-
150
- payload = BaseModel.from_dict(d.pop("payload"))
151
-
152
- def _parse_result(data: object) -> Union["BaseModel", None]:
153
- if data is None:
154
- return data
155
- try:
156
- if not isinstance(data, dict):
157
- raise TypeError()
158
- result_type_0 = BaseModel.from_dict(data)
159
-
160
- return result_type_0
161
- except: # noqa: E722
162
- pass
163
- return cast(Union["BaseModel", None], data)
164
-
165
- result = _parse_result(d.pop("result"))
166
-
167
- task = cls(
168
- status=status,
169
- steps_total=steps_total,
170
- steps_completed=steps_completed,
171
- exception=exception,
172
- updated_at=updated_at,
173
- created_at=created_at,
174
- id=id,
175
- org_id=org_id,
176
- type=type,
177
- payload=payload,
178
- result=result,
179
- )
180
-
181
- task.additional_properties = d
182
- return task
183
-
184
- @property
185
- def additional_keys(self) -> list[str]:
186
- return list(self.additional_properties.keys())
187
-
188
- def __getitem__(self, key: str) -> Any:
189
- return self.additional_properties[key]
190
-
191
- def __setitem__(self, key: str, value: Any) -> None:
192
- self.additional_properties[key] = value
193
-
194
- def __delitem__(self, key: str) -> None:
195
- del self.additional_properties[key]
196
-
197
- def __contains__(self, key: str) -> bool:
198
- return key in self.additional_properties
@@ -1,14 +0,0 @@
1
- from enum import Enum
2
-
3
-
4
- class TaskStatus(str, Enum):
5
- ABORTED = "ABORTED"
6
- ABORTING = "ABORTING"
7
- COMPLETED = "COMPLETED"
8
- DISPATCHED = "DISPATCHED"
9
- FAILED = "FAILED"
10
- INITIALIZED = "INITIALIZED"
11
- PROCESSING = "PROCESSING"
12
-
13
- def __str__(self) -> str:
14
- return str(self.value)
@@ -1,133 +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 enum import Enum
15
- from typing import Any, Type, TypeVar, Union, cast
16
-
17
- from attrs import define as _attrs_define
18
- from attrs import field as _attrs_field
19
- from dateutil.parser import isoparse
20
-
21
- from ..models.task_status import TaskStatus
22
-
23
- T = TypeVar("T", bound="TaskStatusInfo")
24
-
25
-
26
- @_attrs_define
27
- class TaskStatusInfo:
28
- """
29
- Attributes:
30
- status (TaskStatus): Status of task in the task queue
31
- steps_total (Union[None, int]):
32
- steps_completed (Union[None, int]):
33
- exception (Union[None, str]):
34
- updated_at (datetime.datetime):
35
- created_at (datetime.datetime):
36
- """
37
-
38
- status: TaskStatus
39
- steps_total: Union[None, int]
40
- steps_completed: Union[None, int]
41
- exception: Union[None, str]
42
- updated_at: datetime.datetime
43
- created_at: datetime.datetime
44
- additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
45
-
46
- def to_dict(self) -> dict[str, Any]:
47
- status = self.status.value if isinstance(self.status, Enum) else self.status
48
-
49
- steps_total: Union[None, int]
50
- steps_total = self.steps_total
51
-
52
- steps_completed: Union[None, int]
53
- steps_completed = self.steps_completed
54
-
55
- exception: Union[None, str]
56
- exception = self.exception
57
-
58
- updated_at = self.updated_at.isoformat()
59
-
60
- created_at = self.created_at.isoformat()
61
-
62
- field_dict: dict[str, Any] = {}
63
- field_dict.update(self.additional_properties)
64
- field_dict.update(
65
- {
66
- "status": status,
67
- "steps_total": steps_total,
68
- "steps_completed": steps_completed,
69
- "exception": exception,
70
- "updated_at": updated_at,
71
- "created_at": created_at,
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
- status = TaskStatus(d.pop("status"))
81
-
82
- def _parse_steps_total(data: object) -> Union[None, int]:
83
- if data is None:
84
- return data
85
- return cast(Union[None, int], data)
86
-
87
- steps_total = _parse_steps_total(d.pop("steps_total"))
88
-
89
- def _parse_steps_completed(data: object) -> Union[None, int]:
90
- if data is None:
91
- return data
92
- return cast(Union[None, int], data)
93
-
94
- steps_completed = _parse_steps_completed(d.pop("steps_completed"))
95
-
96
- def _parse_exception(data: object) -> Union[None, str]:
97
- if data is None:
98
- return data
99
- return cast(Union[None, str], data)
100
-
101
- exception = _parse_exception(d.pop("exception"))
102
-
103
- updated_at = isoparse(d.pop("updated_at"))
104
-
105
- created_at = isoparse(d.pop("created_at"))
106
-
107
- task_status_info = cls(
108
- status=status,
109
- steps_total=steps_total,
110
- steps_completed=steps_completed,
111
- exception=exception,
112
- updated_at=updated_at,
113
- created_at=created_at,
114
- )
115
-
116
- task_status_info.additional_properties = d
117
- return task_status_info
118
-
119
- @property
120
- def additional_keys(self) -> list[str]:
121
- return list(self.additional_properties.keys())
122
-
123
- def __getitem__(self, key: str) -> Any:
124
- return self.additional_properties[key]
125
-
126
- def __setitem__(self, key: str, value: Any) -> None:
127
- self.additional_properties[key] = value
128
-
129
- def __delitem__(self, key: str) -> None:
130
- del self.additional_properties[key]
131
-
132
- def __contains__(self, key: str) -> bool:
133
- return key in self.additional_properties
@@ -1,72 +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="UnauthenticatedErrorResponse")
19
-
20
-
21
- @_attrs_define
22
- class UnauthenticatedErrorResponse:
23
- """
24
- Attributes:
25
- status_code (Literal[401]):
26
- """
27
-
28
- status_code: Literal[401]
29
- additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
30
-
31
- def to_dict(self) -> dict[str, Any]:
32
- status_code = self.status_code
33
-
34
- field_dict: dict[str, Any] = {}
35
- field_dict.update(self.additional_properties)
36
- field_dict.update(
37
- {
38
- "status_code": status_code,
39
- }
40
- )
41
-
42
- return field_dict
43
-
44
- @classmethod
45
- def from_dict(cls: Type[T], src_dict: dict[str, Any]) -> T:
46
- d = src_dict.copy()
47
- status_code = cast(Literal[401], d.pop("status_code"))
48
- if status_code != 401:
49
- raise ValueError(f"status_code must match const 401, got '{status_code}'")
50
-
51
- unauthenticated_error_response = cls(
52
- status_code=status_code,
53
- )
54
-
55
- unauthenticated_error_response.additional_properties = d
56
- return unauthenticated_error_response
57
-
58
- @property
59
- def additional_keys(self) -> list[str]:
60
- return list(self.additional_properties.keys())
61
-
62
- def __getitem__(self, key: str) -> Any:
63
- return self.additional_properties[key]
64
-
65
- def __setitem__(self, key: str, value: Any) -> None:
66
- self.additional_properties[key] = value
67
-
68
- def __delitem__(self, key: str) -> None:
69
- del self.additional_properties[key]
70
-
71
- def __contains__(self, key: str) -> bool:
72
- return key in self.additional_properties