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,160 +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
-
6
- from attrs import define as _attrs_define
7
- from attrs import field as _attrs_field
8
-
9
- from ..models.multi_modal_chat_message_role import MultiModalChatMessageRole
10
- from ..types import UNSET, Unset
11
-
12
- if TYPE_CHECKING:
13
- from ..models.image_chat_content_part import ImageChatContentPart
14
- from ..models.text_chat_content_part import TextChatContentPart
15
-
16
-
17
- T = TypeVar("T", bound="MultiModalChatMessage")
18
-
19
-
20
- @_attrs_define
21
- class MultiModalChatMessage:
22
- """
23
- Attributes:
24
- role (MultiModalChatMessageRole):
25
- content (list[ImageChatContentPart | TextChatContentPart]):
26
- message_type (Literal['multi_modal'] | Unset): Default: 'multi_modal'.
27
- completion_tokens (int | None | Unset):
28
- cacheable (bool | Unset): Default: False.
29
- explicit_cache_breakpoint (bool | Unset): Default: False.
30
- """
31
-
32
- role: MultiModalChatMessageRole
33
- content: list[ImageChatContentPart | TextChatContentPart]
34
- message_type: Literal["multi_modal"] | Unset = "multi_modal"
35
- completion_tokens: int | None | Unset = UNSET
36
- cacheable: bool | Unset = False
37
- explicit_cache_breakpoint: bool | Unset = False
38
- additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
39
-
40
- def to_dict(self) -> dict[str, Any]:
41
- from ..models.text_chat_content_part import TextChatContentPart
42
-
43
- role = self.role.value
44
-
45
- content = []
46
- for content_item_data in self.content:
47
- content_item: dict[str, Any]
48
- if isinstance(content_item_data, TextChatContentPart):
49
- content_item = content_item_data.to_dict()
50
- else:
51
- content_item = content_item_data.to_dict()
52
-
53
- content.append(content_item)
54
-
55
- message_type = self.message_type
56
-
57
- completion_tokens: int | None | Unset
58
- if isinstance(self.completion_tokens, Unset):
59
- completion_tokens = UNSET
60
- else:
61
- completion_tokens = self.completion_tokens
62
-
63
- cacheable = self.cacheable
64
-
65
- explicit_cache_breakpoint = self.explicit_cache_breakpoint
66
-
67
- field_dict: dict[str, Any] = {}
68
- field_dict.update(self.additional_properties)
69
- field_dict.update(
70
- {
71
- "role": role,
72
- "content": content,
73
- }
74
- )
75
- if message_type is not UNSET:
76
- field_dict["message_type"] = message_type
77
- if completion_tokens is not UNSET:
78
- field_dict["completion_tokens"] = completion_tokens
79
- if cacheable is not UNSET:
80
- field_dict["cacheable"] = cacheable
81
- if explicit_cache_breakpoint is not UNSET:
82
- field_dict["explicit_cache_breakpoint"] = explicit_cache_breakpoint
83
-
84
- return field_dict
85
-
86
- @classmethod
87
- def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
88
- from ..models.image_chat_content_part import ImageChatContentPart
89
- from ..models.text_chat_content_part import TextChatContentPart
90
-
91
- d = dict(src_dict)
92
- role = MultiModalChatMessageRole(d.pop("role"))
93
-
94
- content = []
95
- _content = d.pop("content")
96
- for content_item_data in _content:
97
-
98
- def _parse_content_item(data: object) -> ImageChatContentPart | TextChatContentPart:
99
- try:
100
- if not isinstance(data, dict):
101
- raise TypeError()
102
- content_item_type_0 = TextChatContentPart.from_dict(data)
103
-
104
- return content_item_type_0
105
- except (TypeError, ValueError, AttributeError, KeyError):
106
- pass
107
- if not isinstance(data, dict):
108
- raise TypeError()
109
- content_item_type_1 = ImageChatContentPart.from_dict(data)
110
-
111
- return content_item_type_1
112
-
113
- content_item = _parse_content_item(content_item_data)
114
-
115
- content.append(content_item)
116
-
117
- message_type = cast(Literal["multi_modal"] | Unset, d.pop("message_type", UNSET))
118
- if message_type != "multi_modal" and not isinstance(message_type, Unset):
119
- raise ValueError(f"message_type must match const 'multi_modal', got '{message_type}'")
120
-
121
- def _parse_completion_tokens(data: object) -> int | None | Unset:
122
- if data is None:
123
- return data
124
- if isinstance(data, Unset):
125
- return data
126
- return cast(int | None | Unset, data)
127
-
128
- completion_tokens = _parse_completion_tokens(d.pop("completion_tokens", UNSET))
129
-
130
- cacheable = d.pop("cacheable", UNSET)
131
-
132
- explicit_cache_breakpoint = d.pop("explicit_cache_breakpoint", UNSET)
133
-
134
- multi_modal_chat_message = cls(
135
- role=role,
136
- content=content,
137
- message_type=message_type,
138
- completion_tokens=completion_tokens,
139
- cacheable=cacheable,
140
- explicit_cache_breakpoint=explicit_cache_breakpoint,
141
- )
142
-
143
- multi_modal_chat_message.additional_properties = d
144
- return multi_modal_chat_message
145
-
146
- @property
147
- def additional_keys(self) -> list[str]:
148
- return list(self.additional_properties.keys())
149
-
150
- def __getitem__(self, key: str) -> Any:
151
- return self.additional_properties[key]
152
-
153
- def __setitem__(self, key: str, value: Any) -> None:
154
- self.additional_properties[key] = value
155
-
156
- def __delitem__(self, key: str) -> None:
157
- del self.additional_properties[key]
158
-
159
- def __contains__(self, key: str) -> bool:
160
- return key in self.additional_properties
@@ -1,10 +0,0 @@
1
- from enum import Enum
2
-
3
-
4
- class MultiModalChatMessageRole(str, Enum):
5
- ASSISTANT = "assistant"
6
- SYSTEM = "system"
7
- USER = "user"
8
-
9
- def __str__(self) -> str:
10
- return str(self.value)
@@ -1,144 +0,0 @@
1
- from __future__ import annotations
2
-
3
- from collections.abc import Mapping
4
- from typing import Any, 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 ..types import UNSET, Unset
11
-
12
- T = TypeVar("T", bound="PreviewMetadata")
13
-
14
-
15
- @_attrs_define
16
- class PreviewMetadata:
17
- """
18
- Attributes:
19
- original_task_id (UUID):
20
- index (int | None | Unset):
21
- label (None | str | Unset):
22
- description (None | str | Unset):
23
- num_omitted_artifacts (int | None | Unset):
24
- """
25
-
26
- original_task_id: UUID
27
- index: int | None | Unset = UNSET
28
- label: None | str | Unset = UNSET
29
- description: None | str | Unset = UNSET
30
- num_omitted_artifacts: int | None | Unset = UNSET
31
- additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
32
-
33
- def to_dict(self) -> dict[str, Any]:
34
- original_task_id = str(self.original_task_id)
35
-
36
- index: int | None | Unset
37
- if isinstance(self.index, Unset):
38
- index = UNSET
39
- else:
40
- index = self.index
41
-
42
- label: None | str | Unset
43
- if isinstance(self.label, Unset):
44
- label = UNSET
45
- else:
46
- label = self.label
47
-
48
- description: None | str | Unset
49
- if isinstance(self.description, Unset):
50
- description = UNSET
51
- else:
52
- description = self.description
53
-
54
- num_omitted_artifacts: int | None | Unset
55
- if isinstance(self.num_omitted_artifacts, Unset):
56
- num_omitted_artifacts = UNSET
57
- else:
58
- num_omitted_artifacts = self.num_omitted_artifacts
59
-
60
- field_dict: dict[str, Any] = {}
61
- field_dict.update(self.additional_properties)
62
- field_dict.update(
63
- {
64
- "original_task_id": original_task_id,
65
- }
66
- )
67
- if index is not UNSET:
68
- field_dict["index"] = index
69
- if label is not UNSET:
70
- field_dict["label"] = label
71
- if description is not UNSET:
72
- field_dict["description"] = description
73
- if num_omitted_artifacts is not UNSET:
74
- field_dict["num_omitted_artifacts"] = num_omitted_artifacts
75
-
76
- return field_dict
77
-
78
- @classmethod
79
- def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
80
- d = dict(src_dict)
81
- original_task_id = UUID(d.pop("original_task_id"))
82
-
83
- def _parse_index(data: object) -> int | None | Unset:
84
- if data is None:
85
- return data
86
- if isinstance(data, Unset):
87
- return data
88
- return cast(int | None | Unset, data)
89
-
90
- index = _parse_index(d.pop("index", UNSET))
91
-
92
- def _parse_label(data: object) -> None | str | Unset:
93
- if data is None:
94
- return data
95
- if isinstance(data, Unset):
96
- return data
97
- return cast(None | str | Unset, data)
98
-
99
- label = _parse_label(d.pop("label", UNSET))
100
-
101
- def _parse_description(data: object) -> None | str | Unset:
102
- if data is None:
103
- return data
104
- if isinstance(data, Unset):
105
- return data
106
- return cast(None | str | Unset, data)
107
-
108
- description = _parse_description(d.pop("description", UNSET))
109
-
110
- def _parse_num_omitted_artifacts(data: object) -> int | None | Unset:
111
- if data is None:
112
- return data
113
- if isinstance(data, Unset):
114
- return data
115
- return cast(int | None | Unset, data)
116
-
117
- num_omitted_artifacts = _parse_num_omitted_artifacts(d.pop("num_omitted_artifacts", UNSET))
118
-
119
- preview_metadata = cls(
120
- original_task_id=original_task_id,
121
- index=index,
122
- label=label,
123
- description=description,
124
- num_omitted_artifacts=num_omitted_artifacts,
125
- )
126
-
127
- preview_metadata.additional_properties = d
128
- return preview_metadata
129
-
130
- @property
131
- def additional_keys(self) -> list[str]:
132
- return list(self.additional_properties.keys())
133
-
134
- def __getitem__(self, key: str) -> Any:
135
- return self.additional_properties[key]
136
-
137
- def __setitem__(self, key: str, value: Any) -> None:
138
- self.additional_properties[key] = value
139
-
140
- def __delitem__(self, key: str) -> None:
141
- del self.additional_properties[key]
142
-
143
- def __contains__(self, key: str) -> bool:
144
- return key in self.additional_properties
@@ -1,10 +0,0 @@
1
- from enum import Enum
2
-
3
-
4
- class ProcessingMode(str, Enum):
5
- MAP = "map"
6
- REDUCE = "reduce"
7
- TRANSFORM = "transform"
8
-
9
- def __str__(self) -> str:
10
- return str(self.value)
@@ -1,83 +0,0 @@
1
- from __future__ import annotations
2
-
3
- from collections.abc import Mapping
4
- from typing import TYPE_CHECKING, Any, TypeVar
5
-
6
- from attrs import define as _attrs_define
7
- from attrs import field as _attrs_field
8
-
9
- if TYPE_CHECKING:
10
- from ..models.status_count import StatusCount
11
-
12
-
13
- T = TypeVar("T", bound="ProgressStatus")
14
-
15
-
16
- @_attrs_define
17
- class ProgressStatus:
18
- """
19
- Attributes:
20
- by_status (list[StatusCount]):
21
- total (int):
22
- """
23
-
24
- by_status: list[StatusCount]
25
- total: int
26
- additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
27
-
28
- def to_dict(self) -> dict[str, Any]:
29
- by_status = []
30
- for by_status_item_data in self.by_status:
31
- by_status_item = by_status_item_data.to_dict()
32
- by_status.append(by_status_item)
33
-
34
- total = self.total
35
-
36
- field_dict: dict[str, Any] = {}
37
- field_dict.update(self.additional_properties)
38
- field_dict.update(
39
- {
40
- "by_status": by_status,
41
- "total": total,
42
- }
43
- )
44
-
45
- return field_dict
46
-
47
- @classmethod
48
- def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
49
- from ..models.status_count import StatusCount
50
-
51
- d = dict(src_dict)
52
- by_status = []
53
- _by_status = d.pop("by_status")
54
- for by_status_item_data in _by_status:
55
- by_status_item = StatusCount.from_dict(by_status_item_data)
56
-
57
- by_status.append(by_status_item)
58
-
59
- total = d.pop("total")
60
-
61
- progress_status = cls(
62
- by_status=by_status,
63
- total=total,
64
- )
65
-
66
- progress_status.additional_properties = d
67
- return progress_status
68
-
69
- @property
70
- def additional_keys(self) -> list[str]:
71
- return list(self.additional_properties.keys())
72
-
73
- def __getitem__(self, key: str) -> Any:
74
- return self.additional_properties[key]
75
-
76
- def __setitem__(self, key: str, value: Any) -> None:
77
- self.additional_properties[key] = value
78
-
79
- def __delitem__(self, key: str) -> None:
80
- del self.additional_properties[key]
81
-
82
- def __contains__(self, key: str) -> bool:
83
- return key in self.additional_properties
@@ -1,77 +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="QueueStats")
10
-
11
-
12
- @_attrs_define
13
- class QueueStats:
14
- """
15
- Attributes:
16
- low (int):
17
- default (int):
18
- high (int):
19
- """
20
-
21
- low: int
22
- default: int
23
- high: int
24
- additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
25
-
26
- def to_dict(self) -> dict[str, Any]:
27
- low = self.low
28
-
29
- default = self.default
30
-
31
- high = self.high
32
-
33
- field_dict: dict[str, Any] = {}
34
- field_dict.update(self.additional_properties)
35
- field_dict.update(
36
- {
37
- "low": low,
38
- "default": default,
39
- "high": high,
40
- }
41
- )
42
-
43
- return field_dict
44
-
45
- @classmethod
46
- def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
47
- d = dict(src_dict)
48
- low = d.pop("low")
49
-
50
- default = d.pop("default")
51
-
52
- high = d.pop("high")
53
-
54
- queue_stats = cls(
55
- low=low,
56
- default=default,
57
- high=high,
58
- )
59
-
60
- queue_stats.additional_properties = d
61
- return queue_stats
62
-
63
- @property
64
- def additional_keys(self) -> list[str]:
65
- return list(self.additional_properties.keys())
66
-
67
- def __getitem__(self, key: str) -> Any:
68
- return self.additional_properties[key]
69
-
70
- def __setitem__(self, key: str, value: Any) -> None:
71
- self.additional_properties[key] = value
72
-
73
- def __delitem__(self, key: str) -> None:
74
- del self.additional_properties[key]
75
-
76
- def __contains__(self, key: str) -> bool:
77
- return key in self.additional_properties