everyrow 0.1.9__py3-none-any.whl → 0.2.0__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 (207) hide show
  1. everyrow/__init__.py +8 -1
  2. everyrow/api_utils.py +5 -1
  3. everyrow/billing.py +29 -0
  4. everyrow/constants.py +1 -1
  5. everyrow/generated/__init__.py +1 -1
  6. everyrow/generated/api/{default/interrupt_chat_task_tasks_chat_interrupt_post.py → artifacts/create_artifact_artifacts_post.py} +38 -32
  7. everyrow/generated/api/billing/__init__.py +1 -0
  8. everyrow/generated/api/{default/get_queues_stats_jobs_queues_get.py → billing/get_billing_balance_billing_get.py} +25 -17
  9. everyrow/generated/api/operations/__init__.py +1 -0
  10. everyrow/generated/api/{default/re_execute_task_endpoint_tasks_re_execute_post.py → operations/agent_map_operations_agent_map_post.py} +53 -41
  11. everyrow/generated/api/{default/create_workflow_from_artifact_workflows_from_artifact_post.py → operations/dedupe_operations_dedupe_post.py} +39 -33
  12. everyrow/generated/api/{default/submit_task_tasks_post.py → operations/merge_operations_merge_post.py} +37 -29
  13. everyrow/generated/api/{default/copy_workflow_endpoint_workflows_copy_post.py → operations/rank_operations_rank_post.py} +43 -29
  14. everyrow/generated/api/{default/export_to_google_sheets_export_post.py → operations/screen_operations_screen_post.py} +43 -31
  15. everyrow/generated/api/operations/single_agent_operations_single_agent_post.py +236 -0
  16. everyrow/generated/api/sessions/__init__.py +1 -0
  17. everyrow/generated/api/{default/copy_artifacts_artifacts_copy_post.py → sessions/create_session_endpoint_sessions_post.py} +35 -27
  18. everyrow/generated/api/tasks/__init__.py +1 -0
  19. everyrow/generated/api/{default/get_job_progress_for_task_jobs_progress_get.py → tasks/get_task_result_tasks_task_id_result_get.py} +45 -33
  20. everyrow/generated/api/{default/get_task_status_endpoint_tasks_task_id_status_get.py → tasks/get_task_status_tasks_task_id_status_get.py} +24 -42
  21. everyrow/generated/models/__init__.py +82 -266
  22. everyrow/generated/models/agent_map_operation.py +315 -0
  23. everyrow/generated/models/{artifact_group_record_metadata_type_0.py → agent_map_operation_input_type_1_item.py} +5 -5
  24. everyrow/generated/models/agent_map_operation_input_type_2.py +46 -0
  25. everyrow/generated/models/{standalone_artifact_record_analysis_type_0.py → agent_map_operation_response_schema_type_0.py} +5 -5
  26. everyrow/generated/models/{create_query_params.py → billing_response.py} +13 -12
  27. everyrow/generated/models/{continue_task_request.py → create_artifact_request.py} +43 -43
  28. everyrow/generated/models/create_artifact_request_data_type_0_item.py +46 -0
  29. everyrow/generated/models/{task_metadata_cols_to_rename_type_0.py → create_artifact_request_data_type_1.py} +5 -5
  30. everyrow/generated/models/{copy_artifacts_response.py → create_artifact_response.py} +12 -12
  31. everyrow/generated/models/{create_session_request.py → create_session.py} +6 -7
  32. everyrow/generated/models/dedupe_operation.py +151 -0
  33. everyrow/generated/models/dedupe_operation_input_type_1_item.py +46 -0
  34. everyrow/generated/models/{import_request_token_data.py → dedupe_operation_input_type_2.py} +5 -5
  35. everyrow/generated/models/error_response.py +109 -0
  36. everyrow/generated/models/{task_insert_query_params.py → error_response_details_type_0.py} +5 -5
  37. everyrow/generated/models/insufficient_balance_error.py +8 -0
  38. everyrow/generated/models/{llm_enum.py → llm_enum_public.py} +1 -20
  39. everyrow/generated/models/merge_operation.py +278 -0
  40. everyrow/generated/models/merge_operation_left_input_type_1_item.py +46 -0
  41. everyrow/generated/models/{chat_completion_message_tool_call.py → merge_operation_left_input_type_2.py} +5 -5
  42. everyrow/generated/models/merge_operation_right_input_type_1_item.py +46 -0
  43. everyrow/generated/models/merge_operation_right_input_type_2.py +46 -0
  44. everyrow/generated/models/merge_operation_use_web_search_type_0.py +10 -0
  45. everyrow/generated/models/operation_response.py +131 -0
  46. everyrow/generated/models/{multi_agent_effort_level.py → public_effort_level.py} +1 -1
  47. everyrow/generated/models/public_task_type.py +12 -0
  48. everyrow/generated/models/rank_operation.py +203 -0
  49. everyrow/generated/models/rank_operation_input_type_1_item.py +46 -0
  50. everyrow/generated/models/{export_request_token_data.py → rank_operation_input_type_2.py} +5 -5
  51. everyrow/generated/models/{artifact_group_record_analysis_type_0.py → rank_operation_response_schema_type_0.py} +5 -5
  52. everyrow/generated/models/screen_operation.py +186 -0
  53. everyrow/generated/models/screen_operation_input_type_1_item.py +46 -0
  54. everyrow/generated/models/screen_operation_input_type_2.py +46 -0
  55. everyrow/generated/models/screen_operation_response_schema_type_0.py +46 -0
  56. everyrow/generated/models/{create_session_response.py → session_response.py} +7 -8
  57. everyrow/generated/models/single_agent_operation.py +304 -0
  58. everyrow/generated/models/single_agent_operation_input_type_1_item.py +46 -0
  59. everyrow/generated/models/single_agent_operation_input_type_2.py +46 -0
  60. everyrow/generated/models/single_agent_operation_response_schema_type_0.py +46 -0
  61. everyrow/generated/models/task_result_response.py +185 -0
  62. everyrow/generated/models/task_result_response_data_type_0_item.py +46 -0
  63. everyrow/generated/models/task_result_response_data_type_1.py +46 -0
  64. everyrow/generated/models/task_status_response.py +99 -19
  65. everyrow/ops.py +360 -434
  66. everyrow/session.py +5 -7
  67. everyrow/task.py +68 -115
  68. {everyrow-0.1.9.dist-info → everyrow-0.2.0.dist-info}/METADATA +22 -8
  69. everyrow-0.2.0.dist-info/RECORD +81 -0
  70. everyrow/citations.py +0 -50
  71. everyrow/generated/api/default/continue_task_endpoint_tasks_continue_post.py +0 -208
  72. everyrow/generated/api/default/create_api_key_endpoint_api_keys_create_post.py +0 -186
  73. everyrow/generated/api/default/create_session_endpoint_sessions_create_post.py +0 -198
  74. everyrow/generated/api/default/generate_feedback_endpoint_tasks_generate_feedback_post.py +0 -186
  75. everyrow/generated/api/default/get_artifacts_artifacts_get.py +0 -260
  76. everyrow/generated/api/default/get_default_timeout_seconds_models_default_timeout_seconds_get.py +0 -165
  77. everyrow/generated/api/default/get_metrics_metrics_get.py +0 -80
  78. everyrow/generated/api/default/get_user_usage_usage_get.py +0 -123
  79. everyrow/generated/api/default/healthz_healthz_get.py +0 -127
  80. everyrow/generated/api/default/import_from_google_sheets_import_post.py +0 -170
  81. everyrow/generated/api/default/list_api_keys_endpoint_api_keys_get.py +0 -186
  82. everyrow/generated/api/default/revoke_api_key_endpoint_api_keys_key_id_revoke_post.py +0 -181
  83. everyrow/generated/api/default/revoke_jobs_for_task_jobs_revoke_post.py +0 -164
  84. everyrow/generated/api/default/rollback_to_message_endpoint_tasks_chat_rollback_post.py +0 -186
  85. everyrow/generated/api/default/submit_chat_task_tasks_chat_post.py +0 -164
  86. everyrow/generated/api/default/task_resource_estimation_task_resource_estimation_post.py +0 -319
  87. everyrow/generated/api/default/trigger_workflow_execution_endpoint_workflows_trigger_post.py +0 -166
  88. everyrow/generated/api/default/whoami_whoami_get.py +0 -127
  89. everyrow/generated/models/agent_improvement_instruction.py +0 -69
  90. everyrow/generated/models/agent_query_params.py +0 -383
  91. everyrow/generated/models/agent_query_params_system_prompt_kind_type_0.py +0 -10
  92. everyrow/generated/models/agent_task_args.py +0 -163
  93. everyrow/generated/models/agent_task_args_processing_mode.py +0 -9
  94. everyrow/generated/models/allowed_suggestions.py +0 -9
  95. everyrow/generated/models/api_key_info.py +0 -163
  96. everyrow/generated/models/artifact_changed_payload.py +0 -89
  97. everyrow/generated/models/artifact_group_record.py +0 -363
  98. everyrow/generated/models/artifact_group_record_trace_mapping_type_0.py +0 -46
  99. everyrow/generated/models/artifact_status.py +0 -14
  100. everyrow/generated/models/auto_cohort_conversation_message.py +0 -533
  101. everyrow/generated/models/aux_data.py +0 -128
  102. everyrow/generated/models/aux_data_source_bank.py +0 -59
  103. everyrow/generated/models/chat_message_metadata.py +0 -193
  104. everyrow/generated/models/concatenate_query_params.py +0 -46
  105. everyrow/generated/models/concatenate_request.py +0 -306
  106. everyrow/generated/models/continue_reason.py +0 -9
  107. everyrow/generated/models/controller_improvement_round.py +0 -79
  108. everyrow/generated/models/conversation_changed_payload.py +0 -89
  109. everyrow/generated/models/copy_artifacts_request.py +0 -70
  110. everyrow/generated/models/copy_workflow_request.py +0 -62
  111. everyrow/generated/models/copy_workflow_response.py +0 -70
  112. everyrow/generated/models/create_api_key_request.py +0 -95
  113. everyrow/generated/models/create_api_key_response.py +0 -96
  114. everyrow/generated/models/create_group_query_params.py +0 -61
  115. everyrow/generated/models/create_group_request.py +0 -305
  116. everyrow/generated/models/create_request.py +0 -305
  117. everyrow/generated/models/create_workflow_from_artifact_request.py +0 -92
  118. everyrow/generated/models/create_workflow_from_artifact_response.py +0 -70
  119. everyrow/generated/models/data_frame_method.py +0 -18
  120. everyrow/generated/models/date_cutoffs.py +0 -145
  121. everyrow/generated/models/dedupe_public_params.py +0 -64
  122. everyrow/generated/models/dedupe_request_params.py +0 -311
  123. everyrow/generated/models/deep_merge_public_params.py +0 -143
  124. everyrow/generated/models/deep_merge_request.py +0 -313
  125. everyrow/generated/models/deep_rank_public_params.py +0 -109
  126. everyrow/generated/models/deep_rank_request.py +0 -313
  127. everyrow/generated/models/deep_screen_public_params.py +0 -132
  128. everyrow/generated/models/deep_screen_request.py +0 -313
  129. everyrow/generated/models/derive_expression.py +0 -69
  130. everyrow/generated/models/derive_query_params.py +0 -75
  131. everyrow/generated/models/derive_request.py +0 -307
  132. everyrow/generated/models/document_query_tool.py +0 -12
  133. everyrow/generated/models/drop_columns_query_params.py +0 -61
  134. everyrow/generated/models/drop_columns_request.py +0 -305
  135. everyrow/generated/models/event_type.py +0 -14
  136. everyrow/generated/models/execution_metadata.py +0 -146
  137. everyrow/generated/models/export_request.py +0 -75
  138. everyrow/generated/models/export_to_google_sheets_export_post_response_export_to_google_sheets_export_post.py +0 -46
  139. everyrow/generated/models/filter_query_params.py +0 -91
  140. everyrow/generated/models/filter_request.py +0 -305
  141. everyrow/generated/models/flatten_query_params.py +0 -46
  142. everyrow/generated/models/flatten_request.py +0 -305
  143. everyrow/generated/models/generate_feedback_request.py +0 -62
  144. everyrow/generated/models/group_by_query_params.py +0 -62
  145. everyrow/generated/models/group_by_request.py +0 -305
  146. everyrow/generated/models/healthz_healthz_get_response_healthz_healthz_get.py +0 -46
  147. everyrow/generated/models/image_chat_content_part.py +0 -80
  148. everyrow/generated/models/image_chat_content_part_image_url.py +0 -46
  149. everyrow/generated/models/import_from_google_sheets_import_post_response_import_from_google_sheets_import_post.py +0 -46
  150. everyrow/generated/models/import_request.py +0 -83
  151. everyrow/generated/models/join_query_params.py +0 -73
  152. everyrow/generated/models/join_request.py +0 -305
  153. everyrow/generated/models/map_agent_request_params.py +0 -313
  154. everyrow/generated/models/map_multi_agent_request_params.py +0 -313
  155. everyrow/generated/models/message_created_payload.py +0 -98
  156. everyrow/generated/models/multi_agent_query_params.py +0 -264
  157. everyrow/generated/models/multi_modal_chat_message.py +0 -160
  158. everyrow/generated/models/multi_modal_chat_message_role.py +0 -10
  159. everyrow/generated/models/preview_metadata.py +0 -144
  160. everyrow/generated/models/processing_mode.py +0 -10
  161. everyrow/generated/models/progress_status.py +0 -83
  162. everyrow/generated/models/queue_stats.py +0 -77
  163. everyrow/generated/models/reduce_agent_request_params.py +0 -305
  164. everyrow/generated/models/reduce_multi_agent_request_params.py +0 -305
  165. everyrow/generated/models/resource_estimation_response.py +0 -85
  166. everyrow/generated/models/response_schema_type.py +0 -9
  167. everyrow/generated/models/revoke_api_key_response.py +0 -61
  168. everyrow/generated/models/rollback_to_message_request.py +0 -62
  169. everyrow/generated/models/rollback_to_message_response.py +0 -77
  170. everyrow/generated/models/session_changed_payload.py +0 -69
  171. everyrow/generated/models/simple_chat_message.py +0 -121
  172. everyrow/generated/models/simple_chat_message_role.py +0 -10
  173. everyrow/generated/models/simple_chat_message_with_tool_calls.py +0 -156
  174. everyrow/generated/models/source_database_entry.py +0 -92
  175. everyrow/generated/models/standalone_artifact_record.py +0 -311
  176. everyrow/generated/models/standalone_artifact_record_metadata_type_0.py +0 -46
  177. everyrow/generated/models/standalone_artifact_record_trace_mapping_type_0.py +0 -46
  178. everyrow/generated/models/status_count.py +0 -71
  179. everyrow/generated/models/status_count_status.py +0 -13
  180. everyrow/generated/models/submit_chat_task_body.py +0 -497
  181. everyrow/generated/models/submit_chat_task_body_selected_task_type_type_0.py +0 -11
  182. everyrow/generated/models/submit_task_body.py +0 -745
  183. everyrow/generated/models/task_changed_payload.py +0 -105
  184. everyrow/generated/models/task_effort.py +0 -10
  185. everyrow/generated/models/task_id_request.py +0 -62
  186. everyrow/generated/models/task_insert.py +0 -725
  187. everyrow/generated/models/task_metadata.py +0 -323
  188. everyrow/generated/models/task_response.py +0 -62
  189. everyrow/generated/models/task_type.py +0 -31
  190. everyrow/generated/models/text_chat_content_part.py +0 -74
  191. everyrow/generated/models/tool_response_message.py +0 -127
  192. everyrow/generated/models/toolkit_constants.py +0 -80
  193. everyrow/generated/models/trace_changed_payload.py +0 -94
  194. everyrow/generated/models/trace_info.py +0 -78
  195. everyrow/generated/models/trigger_workflow_execution_request.py +0 -112
  196. everyrow/generated/models/trigger_workflow_execution_request_task_params.py +0 -65
  197. everyrow/generated/models/trigger_workflow_execution_request_task_params_additional_property.py +0 -46
  198. everyrow/generated/models/trigger_workflow_execution_response.py +0 -69
  199. everyrow/generated/models/upload_csv_payload.py +0 -310
  200. everyrow/generated/models/upload_csv_query_params.py +0 -114
  201. everyrow/generated/models/usage_response.py +0 -77
  202. everyrow/generated/models/whoami_whoami_get_response_whoami_whoami_get.py +0 -46
  203. everyrow/generated/models/workflow_leaf_node_input.py +0 -70
  204. everyrow-0.1.9.dist-info/RECORD +0 -183
  205. /everyrow/generated/api/{default → artifacts}/__init__.py +0 -0
  206. {everyrow-0.1.9.dist-info → everyrow-0.2.0.dist-info}/WHEEL +0 -0
  207. {everyrow-0.1.9.dist-info → everyrow-0.2.0.dist-info}/licenses/LICENSE.txt +0 -0
@@ -1,311 +0,0 @@
1
- from __future__ import annotations
2
-
3
- from collections.abc import Mapping
4
- from typing import TYPE_CHECKING, Any, Literal, TypeVar, cast
5
- from uuid import UUID
6
-
7
- from attrs import define as _attrs_define
8
- from attrs import field as _attrs_field
9
-
10
- from ..models.artifact_status import ArtifactStatus
11
- from ..types import UNSET, Unset
12
-
13
- if TYPE_CHECKING:
14
- from ..models.aux_data import AuxData
15
- from ..models.standalone_artifact_record_analysis_type_0 import StandaloneArtifactRecordAnalysisType0
16
- from ..models.standalone_artifact_record_metadata_type_0 import StandaloneArtifactRecordMetadataType0
17
- from ..models.standalone_artifact_record_trace_mapping_type_0 import StandaloneArtifactRecordTraceMappingType0
18
-
19
-
20
- T = TypeVar("T", bound="StandaloneArtifactRecord")
21
-
22
-
23
- @_attrs_define
24
- class StandaloneArtifactRecord:
25
- """
26
- Attributes:
27
- uid (UUID):
28
- type_ (Literal['standalone']):
29
- data (Any):
30
- metadata (None | StandaloneArtifactRecordMetadataType0 | Unset):
31
- label (None | str | Unset):
32
- aux_data (AuxData | Unset):
33
- status (ArtifactStatus | Unset):
34
- trace_mapping (None | StandaloneArtifactRecordTraceMappingType0 | Unset):
35
- original_id (None | Unset | UUID):
36
- index_in_group (int | None | Unset):
37
- current_iteration (int | None | Unset):
38
- analysis (None | StandaloneArtifactRecordAnalysisType0 | Unset):
39
- """
40
-
41
- uid: UUID
42
- type_: Literal["standalone"]
43
- data: Any
44
- metadata: None | StandaloneArtifactRecordMetadataType0 | Unset = UNSET
45
- label: None | str | Unset = UNSET
46
- aux_data: AuxData | Unset = UNSET
47
- status: ArtifactStatus | Unset = UNSET
48
- trace_mapping: None | StandaloneArtifactRecordTraceMappingType0 | Unset = UNSET
49
- original_id: None | Unset | UUID = UNSET
50
- index_in_group: int | None | Unset = UNSET
51
- current_iteration: int | None | Unset = UNSET
52
- analysis: None | StandaloneArtifactRecordAnalysisType0 | Unset = UNSET
53
- additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
54
-
55
- def to_dict(self) -> dict[str, Any]:
56
- from ..models.standalone_artifact_record_analysis_type_0 import StandaloneArtifactRecordAnalysisType0
57
- from ..models.standalone_artifact_record_metadata_type_0 import StandaloneArtifactRecordMetadataType0
58
- from ..models.standalone_artifact_record_trace_mapping_type_0 import StandaloneArtifactRecordTraceMappingType0
59
-
60
- uid = str(self.uid)
61
-
62
- type_ = self.type_
63
-
64
- data = self.data
65
-
66
- metadata: dict[str, Any] | None | Unset
67
- if isinstance(self.metadata, Unset):
68
- metadata = UNSET
69
- elif isinstance(self.metadata, StandaloneArtifactRecordMetadataType0):
70
- metadata = self.metadata.to_dict()
71
- else:
72
- metadata = self.metadata
73
-
74
- label: None | str | Unset
75
- if isinstance(self.label, Unset):
76
- label = UNSET
77
- else:
78
- label = self.label
79
-
80
- aux_data: dict[str, Any] | Unset = UNSET
81
- if not isinstance(self.aux_data, Unset):
82
- aux_data = self.aux_data.to_dict()
83
-
84
- status: str | Unset = UNSET
85
- if not isinstance(self.status, Unset):
86
- status = self.status.value
87
-
88
- trace_mapping: dict[str, Any] | None | Unset
89
- if isinstance(self.trace_mapping, Unset):
90
- trace_mapping = UNSET
91
- elif isinstance(self.trace_mapping, StandaloneArtifactRecordTraceMappingType0):
92
- trace_mapping = self.trace_mapping.to_dict()
93
- else:
94
- trace_mapping = self.trace_mapping
95
-
96
- original_id: None | str | Unset
97
- if isinstance(self.original_id, Unset):
98
- original_id = UNSET
99
- elif isinstance(self.original_id, UUID):
100
- original_id = str(self.original_id)
101
- else:
102
- original_id = self.original_id
103
-
104
- index_in_group: int | None | Unset
105
- if isinstance(self.index_in_group, Unset):
106
- index_in_group = UNSET
107
- else:
108
- index_in_group = self.index_in_group
109
-
110
- current_iteration: int | None | Unset
111
- if isinstance(self.current_iteration, Unset):
112
- current_iteration = UNSET
113
- else:
114
- current_iteration = self.current_iteration
115
-
116
- analysis: dict[str, Any] | None | Unset
117
- if isinstance(self.analysis, Unset):
118
- analysis = UNSET
119
- elif isinstance(self.analysis, StandaloneArtifactRecordAnalysisType0):
120
- analysis = self.analysis.to_dict()
121
- else:
122
- analysis = self.analysis
123
-
124
- field_dict: dict[str, Any] = {}
125
- field_dict.update(self.additional_properties)
126
- field_dict.update(
127
- {
128
- "uid": uid,
129
- "type": type_,
130
- "data": data,
131
- }
132
- )
133
- if metadata is not UNSET:
134
- field_dict["metadata"] = metadata
135
- if label is not UNSET:
136
- field_dict["label"] = label
137
- if aux_data is not UNSET:
138
- field_dict["aux_data"] = aux_data
139
- if status is not UNSET:
140
- field_dict["status"] = status
141
- if trace_mapping is not UNSET:
142
- field_dict["trace_mapping"] = trace_mapping
143
- if original_id is not UNSET:
144
- field_dict["original_id"] = original_id
145
- if index_in_group is not UNSET:
146
- field_dict["index_in_group"] = index_in_group
147
- if current_iteration is not UNSET:
148
- field_dict["current_iteration"] = current_iteration
149
- if analysis is not UNSET:
150
- field_dict["analysis"] = analysis
151
-
152
- return field_dict
153
-
154
- @classmethod
155
- def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
156
- from ..models.aux_data import AuxData
157
- from ..models.standalone_artifact_record_analysis_type_0 import StandaloneArtifactRecordAnalysisType0
158
- from ..models.standalone_artifact_record_metadata_type_0 import StandaloneArtifactRecordMetadataType0
159
- from ..models.standalone_artifact_record_trace_mapping_type_0 import StandaloneArtifactRecordTraceMappingType0
160
-
161
- d = dict(src_dict)
162
- uid = UUID(d.pop("uid"))
163
-
164
- type_ = cast(Literal["standalone"], d.pop("type"))
165
- if type_ != "standalone":
166
- raise ValueError(f"type must match const 'standalone', got '{type_}'")
167
-
168
- data = d.pop("data")
169
-
170
- def _parse_metadata(data: object) -> None | StandaloneArtifactRecordMetadataType0 | Unset:
171
- if data is None:
172
- return data
173
- if isinstance(data, Unset):
174
- return data
175
- try:
176
- if not isinstance(data, dict):
177
- raise TypeError()
178
- metadata_type_0 = StandaloneArtifactRecordMetadataType0.from_dict(data)
179
-
180
- return metadata_type_0
181
- except (TypeError, ValueError, AttributeError, KeyError):
182
- pass
183
- return cast(None | StandaloneArtifactRecordMetadataType0 | Unset, data)
184
-
185
- metadata = _parse_metadata(d.pop("metadata", UNSET))
186
-
187
- def _parse_label(data: object) -> None | str | Unset:
188
- if data is None:
189
- return data
190
- if isinstance(data, Unset):
191
- return data
192
- return cast(None | str | Unset, data)
193
-
194
- label = _parse_label(d.pop("label", UNSET))
195
-
196
- _aux_data = d.pop("aux_data", UNSET)
197
- aux_data: AuxData | Unset
198
- if isinstance(_aux_data, Unset):
199
- aux_data = UNSET
200
- else:
201
- aux_data = AuxData.from_dict(_aux_data)
202
-
203
- _status = d.pop("status", UNSET)
204
- status: ArtifactStatus | Unset
205
- if isinstance(_status, Unset):
206
- status = UNSET
207
- else:
208
- status = ArtifactStatus(_status)
209
-
210
- def _parse_trace_mapping(data: object) -> None | StandaloneArtifactRecordTraceMappingType0 | Unset:
211
- if data is None:
212
- return data
213
- if isinstance(data, Unset):
214
- return data
215
- try:
216
- if not isinstance(data, dict):
217
- raise TypeError()
218
- trace_mapping_type_0 = StandaloneArtifactRecordTraceMappingType0.from_dict(data)
219
-
220
- return trace_mapping_type_0
221
- except (TypeError, ValueError, AttributeError, KeyError):
222
- pass
223
- return cast(None | StandaloneArtifactRecordTraceMappingType0 | Unset, data)
224
-
225
- trace_mapping = _parse_trace_mapping(d.pop("trace_mapping", UNSET))
226
-
227
- def _parse_original_id(data: object) -> None | Unset | UUID:
228
- if data is None:
229
- return data
230
- if isinstance(data, Unset):
231
- return data
232
- try:
233
- if not isinstance(data, str):
234
- raise TypeError()
235
- original_id_type_0 = UUID(data)
236
-
237
- return original_id_type_0
238
- except (TypeError, ValueError, AttributeError, KeyError):
239
- pass
240
- return cast(None | Unset | UUID, data)
241
-
242
- original_id = _parse_original_id(d.pop("original_id", UNSET))
243
-
244
- def _parse_index_in_group(data: object) -> int | None | Unset:
245
- if data is None:
246
- return data
247
- if isinstance(data, Unset):
248
- return data
249
- return cast(int | None | Unset, data)
250
-
251
- index_in_group = _parse_index_in_group(d.pop("index_in_group", UNSET))
252
-
253
- def _parse_current_iteration(data: object) -> int | None | Unset:
254
- if data is None:
255
- return data
256
- if isinstance(data, Unset):
257
- return data
258
- return cast(int | None | Unset, data)
259
-
260
- current_iteration = _parse_current_iteration(d.pop("current_iteration", UNSET))
261
-
262
- def _parse_analysis(data: object) -> None | StandaloneArtifactRecordAnalysisType0 | Unset:
263
- if data is None:
264
- return data
265
- if isinstance(data, Unset):
266
- return data
267
- try:
268
- if not isinstance(data, dict):
269
- raise TypeError()
270
- analysis_type_0 = StandaloneArtifactRecordAnalysisType0.from_dict(data)
271
-
272
- return analysis_type_0
273
- except (TypeError, ValueError, AttributeError, KeyError):
274
- pass
275
- return cast(None | StandaloneArtifactRecordAnalysisType0 | Unset, data)
276
-
277
- analysis = _parse_analysis(d.pop("analysis", UNSET))
278
-
279
- standalone_artifact_record = cls(
280
- uid=uid,
281
- type_=type_,
282
- data=data,
283
- metadata=metadata,
284
- label=label,
285
- aux_data=aux_data,
286
- status=status,
287
- trace_mapping=trace_mapping,
288
- original_id=original_id,
289
- index_in_group=index_in_group,
290
- current_iteration=current_iteration,
291
- analysis=analysis,
292
- )
293
-
294
- standalone_artifact_record.additional_properties = d
295
- return standalone_artifact_record
296
-
297
- @property
298
- def additional_keys(self) -> list[str]:
299
- return list(self.additional_properties.keys())
300
-
301
- def __getitem__(self, key: str) -> Any:
302
- return self.additional_properties[key]
303
-
304
- def __setitem__(self, key: str, value: Any) -> None:
305
- self.additional_properties[key] = value
306
-
307
- def __delitem__(self, key: str) -> None:
308
- del self.additional_properties[key]
309
-
310
- def __contains__(self, key: str) -> bool:
311
- return key in self.additional_properties
@@ -1,46 +0,0 @@
1
- from __future__ import annotations
2
-
3
- from collections.abc import Mapping
4
- from typing import Any, TypeVar
5
-
6
- from attrs import define as _attrs_define
7
- from attrs import field as _attrs_field
8
-
9
- T = TypeVar("T", bound="StandaloneArtifactRecordMetadataType0")
10
-
11
-
12
- @_attrs_define
13
- class StandaloneArtifactRecordMetadataType0:
14
- """ """
15
-
16
- additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
17
-
18
- def to_dict(self) -> dict[str, Any]:
19
- field_dict: dict[str, Any] = {}
20
- field_dict.update(self.additional_properties)
21
-
22
- return field_dict
23
-
24
- @classmethod
25
- def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
26
- d = dict(src_dict)
27
- standalone_artifact_record_metadata_type_0 = cls()
28
-
29
- standalone_artifact_record_metadata_type_0.additional_properties = d
30
- return standalone_artifact_record_metadata_type_0
31
-
32
- @property
33
- def additional_keys(self) -> list[str]:
34
- return list(self.additional_properties.keys())
35
-
36
- def __getitem__(self, key: str) -> Any:
37
- return self.additional_properties[key]
38
-
39
- def __setitem__(self, key: str, value: Any) -> None:
40
- self.additional_properties[key] = value
41
-
42
- def __delitem__(self, key: str) -> None:
43
- del self.additional_properties[key]
44
-
45
- def __contains__(self, key: str) -> bool:
46
- return key in self.additional_properties
@@ -1,46 +0,0 @@
1
- from __future__ import annotations
2
-
3
- from collections.abc import Mapping
4
- from typing import Any, TypeVar
5
-
6
- from attrs import define as _attrs_define
7
- from attrs import field as _attrs_field
8
-
9
- T = TypeVar("T", bound="StandaloneArtifactRecordTraceMappingType0")
10
-
11
-
12
- @_attrs_define
13
- class StandaloneArtifactRecordTraceMappingType0:
14
- """ """
15
-
16
- additional_properties: dict[str, str] = _attrs_field(init=False, factory=dict)
17
-
18
- def to_dict(self) -> dict[str, Any]:
19
- field_dict: dict[str, Any] = {}
20
- field_dict.update(self.additional_properties)
21
-
22
- return field_dict
23
-
24
- @classmethod
25
- def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
26
- d = dict(src_dict)
27
- standalone_artifact_record_trace_mapping_type_0 = cls()
28
-
29
- standalone_artifact_record_trace_mapping_type_0.additional_properties = d
30
- return standalone_artifact_record_trace_mapping_type_0
31
-
32
- @property
33
- def additional_keys(self) -> list[str]:
34
- return list(self.additional_properties.keys())
35
-
36
- def __getitem__(self, key: str) -> str:
37
- return self.additional_properties[key]
38
-
39
- def __setitem__(self, key: str, value: str) -> None:
40
- self.additional_properties[key] = value
41
-
42
- def __delitem__(self, key: str) -> None:
43
- del self.additional_properties[key]
44
-
45
- def __contains__(self, key: str) -> bool:
46
- return key in self.additional_properties
@@ -1,71 +0,0 @@
1
- from __future__ import annotations
2
-
3
- from collections.abc import Mapping
4
- from typing import Any, TypeVar
5
-
6
- from attrs import define as _attrs_define
7
- from attrs import field as _attrs_field
8
-
9
- from ..models.status_count_status import StatusCountStatus
10
-
11
- T = TypeVar("T", bound="StatusCount")
12
-
13
-
14
- @_attrs_define
15
- class StatusCount:
16
- """
17
- Attributes:
18
- status (StatusCountStatus):
19
- count (int):
20
- """
21
-
22
- status: StatusCountStatus
23
- count: int
24
- additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
25
-
26
- def to_dict(self) -> dict[str, Any]:
27
- status = self.status.value
28
-
29
- count = self.count
30
-
31
- field_dict: dict[str, Any] = {}
32
- field_dict.update(self.additional_properties)
33
- field_dict.update(
34
- {
35
- "status": status,
36
- "count": count,
37
- }
38
- )
39
-
40
- return field_dict
41
-
42
- @classmethod
43
- def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
44
- d = dict(src_dict)
45
- status = StatusCountStatus(d.pop("status"))
46
-
47
- count = d.pop("count")
48
-
49
- status_count = cls(
50
- status=status,
51
- count=count,
52
- )
53
-
54
- status_count.additional_properties = d
55
- return status_count
56
-
57
- @property
58
- def additional_keys(self) -> list[str]:
59
- return list(self.additional_properties.keys())
60
-
61
- def __getitem__(self, key: str) -> Any:
62
- return self.additional_properties[key]
63
-
64
- def __setitem__(self, key: str, value: Any) -> None:
65
- self.additional_properties[key] = value
66
-
67
- def __delitem__(self, key: str) -> None:
68
- del self.additional_properties[key]
69
-
70
- def __contains__(self, key: str) -> bool:
71
- return key in self.additional_properties
@@ -1,13 +0,0 @@
1
- from enum import Enum
2
-
3
-
4
- class StatusCountStatus(str, Enum):
5
- FAILURE = "FAILURE"
6
- PENDING = "PENDING"
7
- RETRY = "RETRY"
8
- REVOKED = "REVOKED"
9
- STARTED = "STARTED"
10
- SUCCESS = "SUCCESS"
11
-
12
- def __str__(self) -> str:
13
- return str(self.value)