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,310 +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.processing_mode import ProcessingMode
11
- from ..types import UNSET, Unset
12
-
13
- if TYPE_CHECKING:
14
- from ..models.upload_csv_query_params import UploadCsvQueryParams
15
-
16
-
17
- T = TypeVar("T", bound="UploadCsvPayload")
18
-
19
-
20
- @_attrs_define
21
- class UploadCsvPayload:
22
- """
23
- Attributes:
24
- query (UploadCsvQueryParams):
25
- input_artifacts (list[UUID] | None | Unset):
26
- context_artifacts (list[UUID] | None | Unset):
27
- label (None | str | Unset): Short task label for use in the UI
28
- description (None | str | Unset): Task description for use in the UI
29
- task_id (None | Unset | UUID):
30
- replaces_task_id (None | Unset | UUID): The ID of the task that this task replaces. Used e.g. by the full
31
- version of a task that replaces a preview version.
32
- twin_artifact_id (None | Unset | UUID): The ID of a reference artifact, e.g. the right table in Deep Merge
33
- operation.
34
- processing_mode (ProcessingMode | Unset):
35
- task_type (Literal['upload_csv'] | Unset): Default: 'upload_csv'.
36
- """
37
-
38
- query: UploadCsvQueryParams
39
- input_artifacts: list[UUID] | None | Unset = UNSET
40
- context_artifacts: list[UUID] | None | Unset = UNSET
41
- label: None | str | Unset = UNSET
42
- description: None | str | Unset = UNSET
43
- task_id: None | Unset | UUID = UNSET
44
- replaces_task_id: None | Unset | UUID = UNSET
45
- twin_artifact_id: None | Unset | UUID = UNSET
46
- processing_mode: ProcessingMode | Unset = UNSET
47
- task_type: Literal["upload_csv"] | Unset = "upload_csv"
48
- additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
49
-
50
- def to_dict(self) -> dict[str, Any]:
51
- query = self.query.to_dict()
52
-
53
- input_artifacts: list[str] | None | Unset
54
- if isinstance(self.input_artifacts, Unset):
55
- input_artifacts = UNSET
56
- elif isinstance(self.input_artifacts, list):
57
- input_artifacts = []
58
- for componentsschemas_input_artifacts_ids_item_data in self.input_artifacts:
59
- componentsschemas_input_artifacts_ids_item = str(componentsschemas_input_artifacts_ids_item_data)
60
- input_artifacts.append(componentsschemas_input_artifacts_ids_item)
61
-
62
- else:
63
- input_artifacts = self.input_artifacts
64
-
65
- context_artifacts: list[str] | None | Unset
66
- if isinstance(self.context_artifacts, Unset):
67
- context_artifacts = UNSET
68
- elif isinstance(self.context_artifacts, list):
69
- context_artifacts = []
70
- for componentsschemas_context_artifacts_ids_item_data in self.context_artifacts:
71
- componentsschemas_context_artifacts_ids_item = str(componentsschemas_context_artifacts_ids_item_data)
72
- context_artifacts.append(componentsschemas_context_artifacts_ids_item)
73
-
74
- else:
75
- context_artifacts = self.context_artifacts
76
-
77
- label: None | str | Unset
78
- if isinstance(self.label, Unset):
79
- label = UNSET
80
- else:
81
- label = self.label
82
-
83
- description: None | str | Unset
84
- if isinstance(self.description, Unset):
85
- description = UNSET
86
- else:
87
- description = self.description
88
-
89
- task_id: None | str | Unset
90
- if isinstance(self.task_id, Unset):
91
- task_id = UNSET
92
- elif isinstance(self.task_id, UUID):
93
- task_id = str(self.task_id)
94
- else:
95
- task_id = self.task_id
96
-
97
- replaces_task_id: None | str | Unset
98
- if isinstance(self.replaces_task_id, Unset):
99
- replaces_task_id = UNSET
100
- elif isinstance(self.replaces_task_id, UUID):
101
- replaces_task_id = str(self.replaces_task_id)
102
- else:
103
- replaces_task_id = self.replaces_task_id
104
-
105
- twin_artifact_id: None | str | Unset
106
- if isinstance(self.twin_artifact_id, Unset):
107
- twin_artifact_id = UNSET
108
- elif isinstance(self.twin_artifact_id, UUID):
109
- twin_artifact_id = str(self.twin_artifact_id)
110
- else:
111
- twin_artifact_id = self.twin_artifact_id
112
-
113
- processing_mode: str | Unset = UNSET
114
- if not isinstance(self.processing_mode, Unset):
115
- processing_mode = self.processing_mode.value
116
-
117
- task_type = self.task_type
118
-
119
- field_dict: dict[str, Any] = {}
120
- field_dict.update(self.additional_properties)
121
- field_dict.update(
122
- {
123
- "query": query,
124
- }
125
- )
126
- if input_artifacts is not UNSET:
127
- field_dict["input_artifacts"] = input_artifacts
128
- if context_artifacts is not UNSET:
129
- field_dict["context_artifacts"] = context_artifacts
130
- if label is not UNSET:
131
- field_dict["label"] = label
132
- if description is not UNSET:
133
- field_dict["description"] = description
134
- if task_id is not UNSET:
135
- field_dict["task_id"] = task_id
136
- if replaces_task_id is not UNSET:
137
- field_dict["replaces_task_id"] = replaces_task_id
138
- if twin_artifact_id is not UNSET:
139
- field_dict["twin_artifact_id"] = twin_artifact_id
140
- if processing_mode is not UNSET:
141
- field_dict["processing_mode"] = processing_mode
142
- if task_type is not UNSET:
143
- field_dict["task_type"] = task_type
144
-
145
- return field_dict
146
-
147
- @classmethod
148
- def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
149
- from ..models.upload_csv_query_params import UploadCsvQueryParams
150
-
151
- d = dict(src_dict)
152
- query = UploadCsvQueryParams.from_dict(d.pop("query"))
153
-
154
- def _parse_input_artifacts(data: object) -> list[UUID] | None | Unset:
155
- if data is None:
156
- return data
157
- if isinstance(data, Unset):
158
- return data
159
- try:
160
- if not isinstance(data, list):
161
- raise TypeError()
162
- input_artifacts_type_0 = []
163
- _input_artifacts_type_0 = data
164
- for componentsschemas_input_artifacts_ids_item_data in _input_artifacts_type_0:
165
- componentsschemas_input_artifacts_ids_item = UUID(componentsschemas_input_artifacts_ids_item_data)
166
-
167
- input_artifacts_type_0.append(componentsschemas_input_artifacts_ids_item)
168
-
169
- return input_artifacts_type_0
170
- except (TypeError, ValueError, AttributeError, KeyError):
171
- pass
172
- return cast(list[UUID] | None | Unset, data)
173
-
174
- input_artifacts = _parse_input_artifacts(d.pop("input_artifacts", UNSET))
175
-
176
- def _parse_context_artifacts(data: object) -> list[UUID] | None | Unset:
177
- if data is None:
178
- return data
179
- if isinstance(data, Unset):
180
- return data
181
- try:
182
- if not isinstance(data, list):
183
- raise TypeError()
184
- context_artifacts_type_0 = []
185
- _context_artifacts_type_0 = data
186
- for componentsschemas_context_artifacts_ids_item_data in _context_artifacts_type_0:
187
- componentsschemas_context_artifacts_ids_item = UUID(
188
- componentsschemas_context_artifacts_ids_item_data
189
- )
190
-
191
- context_artifacts_type_0.append(componentsschemas_context_artifacts_ids_item)
192
-
193
- return context_artifacts_type_0
194
- except (TypeError, ValueError, AttributeError, KeyError):
195
- pass
196
- return cast(list[UUID] | None | Unset, data)
197
-
198
- context_artifacts = _parse_context_artifacts(d.pop("context_artifacts", UNSET))
199
-
200
- def _parse_label(data: object) -> None | str | Unset:
201
- if data is None:
202
- return data
203
- if isinstance(data, Unset):
204
- return data
205
- return cast(None | str | Unset, data)
206
-
207
- label = _parse_label(d.pop("label", UNSET))
208
-
209
- def _parse_description(data: object) -> None | str | Unset:
210
- if data is None:
211
- return data
212
- if isinstance(data, Unset):
213
- return data
214
- return cast(None | str | Unset, data)
215
-
216
- description = _parse_description(d.pop("description", UNSET))
217
-
218
- def _parse_task_id(data: object) -> None | Unset | UUID:
219
- if data is None:
220
- return data
221
- if isinstance(data, Unset):
222
- return data
223
- try:
224
- if not isinstance(data, str):
225
- raise TypeError()
226
- task_id_type_0 = UUID(data)
227
-
228
- return task_id_type_0
229
- except (TypeError, ValueError, AttributeError, KeyError):
230
- pass
231
- return cast(None | Unset | UUID, data)
232
-
233
- task_id = _parse_task_id(d.pop("task_id", UNSET))
234
-
235
- def _parse_replaces_task_id(data: object) -> None | Unset | UUID:
236
- if data is None:
237
- return data
238
- if isinstance(data, Unset):
239
- return data
240
- try:
241
- if not isinstance(data, str):
242
- raise TypeError()
243
- replaces_task_id_type_0 = UUID(data)
244
-
245
- return replaces_task_id_type_0
246
- except (TypeError, ValueError, AttributeError, KeyError):
247
- pass
248
- return cast(None | Unset | UUID, data)
249
-
250
- replaces_task_id = _parse_replaces_task_id(d.pop("replaces_task_id", UNSET))
251
-
252
- def _parse_twin_artifact_id(data: object) -> None | Unset | UUID:
253
- if data is None:
254
- return data
255
- if isinstance(data, Unset):
256
- return data
257
- try:
258
- if not isinstance(data, str):
259
- raise TypeError()
260
- twin_artifact_id_type_0 = UUID(data)
261
-
262
- return twin_artifact_id_type_0
263
- except (TypeError, ValueError, AttributeError, KeyError):
264
- pass
265
- return cast(None | Unset | UUID, data)
266
-
267
- twin_artifact_id = _parse_twin_artifact_id(d.pop("twin_artifact_id", UNSET))
268
-
269
- _processing_mode = d.pop("processing_mode", UNSET)
270
- processing_mode: ProcessingMode | Unset
271
- if isinstance(_processing_mode, Unset):
272
- processing_mode = UNSET
273
- else:
274
- processing_mode = ProcessingMode(_processing_mode)
275
-
276
- task_type = cast(Literal["upload_csv"] | Unset, d.pop("task_type", UNSET))
277
- if task_type != "upload_csv" and not isinstance(task_type, Unset):
278
- raise ValueError(f"task_type must match const 'upload_csv', got '{task_type}'")
279
-
280
- upload_csv_payload = cls(
281
- query=query,
282
- input_artifacts=input_artifacts,
283
- context_artifacts=context_artifacts,
284
- label=label,
285
- description=description,
286
- task_id=task_id,
287
- replaces_task_id=replaces_task_id,
288
- twin_artifact_id=twin_artifact_id,
289
- processing_mode=processing_mode,
290
- task_type=task_type,
291
- )
292
-
293
- upload_csv_payload.additional_properties = d
294
- return upload_csv_payload
295
-
296
- @property
297
- def additional_keys(self) -> list[str]:
298
- return list(self.additional_properties.keys())
299
-
300
- def __getitem__(self, key: str) -> Any:
301
- return self.additional_properties[key]
302
-
303
- def __setitem__(self, key: str, value: Any) -> None:
304
- self.additional_properties[key] = value
305
-
306
- def __delitem__(self, key: str) -> None:
307
- del self.additional_properties[key]
308
-
309
- def __contains__(self, key: str) -> bool:
310
- return key in self.additional_properties
@@ -1,114 +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="UploadCsvQueryParams")
13
-
14
-
15
- @_attrs_define
16
- class UploadCsvQueryParams:
17
- """
18
- Attributes:
19
- file (str): Base64 encoded CSV file
20
- filename (None | str | Unset): Original filename of the uploaded CSV
21
- group_artifact_id (None | Unset | UUID): Optional group artifact ID to use for the uploaded data
22
- """
23
-
24
- file: str
25
- filename: None | str | Unset = UNSET
26
- group_artifact_id: None | Unset | UUID = UNSET
27
- additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
28
-
29
- def to_dict(self) -> dict[str, Any]:
30
- file = self.file
31
-
32
- filename: None | str | Unset
33
- if isinstance(self.filename, Unset):
34
- filename = UNSET
35
- else:
36
- filename = self.filename
37
-
38
- group_artifact_id: None | str | Unset
39
- if isinstance(self.group_artifact_id, Unset):
40
- group_artifact_id = UNSET
41
- elif isinstance(self.group_artifact_id, UUID):
42
- group_artifact_id = str(self.group_artifact_id)
43
- else:
44
- group_artifact_id = self.group_artifact_id
45
-
46
- field_dict: dict[str, Any] = {}
47
- field_dict.update(self.additional_properties)
48
- field_dict.update(
49
- {
50
- "file": file,
51
- }
52
- )
53
- if filename is not UNSET:
54
- field_dict["filename"] = filename
55
- if group_artifact_id is not UNSET:
56
- field_dict["group_artifact_id"] = group_artifact_id
57
-
58
- return field_dict
59
-
60
- @classmethod
61
- def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
62
- d = dict(src_dict)
63
- file = d.pop("file")
64
-
65
- def _parse_filename(data: object) -> None | str | Unset:
66
- if data is None:
67
- return data
68
- if isinstance(data, Unset):
69
- return data
70
- return cast(None | str | Unset, data)
71
-
72
- filename = _parse_filename(d.pop("filename", UNSET))
73
-
74
- def _parse_group_artifact_id(data: object) -> None | Unset | UUID:
75
- if data is None:
76
- return data
77
- if isinstance(data, Unset):
78
- return data
79
- try:
80
- if not isinstance(data, str):
81
- raise TypeError()
82
- group_artifact_id_type_0 = UUID(data)
83
-
84
- return group_artifact_id_type_0
85
- except (TypeError, ValueError, AttributeError, KeyError):
86
- pass
87
- return cast(None | Unset | UUID, data)
88
-
89
- group_artifact_id = _parse_group_artifact_id(d.pop("group_artifact_id", UNSET))
90
-
91
- upload_csv_query_params = cls(
92
- file=file,
93
- filename=filename,
94
- group_artifact_id=group_artifact_id,
95
- )
96
-
97
- upload_csv_query_params.additional_properties = d
98
- return upload_csv_query_params
99
-
100
- @property
101
- def additional_keys(self) -> list[str]:
102
- return list(self.additional_properties.keys())
103
-
104
- def __getitem__(self, key: str) -> Any:
105
- return self.additional_properties[key]
106
-
107
- def __setitem__(self, key: str, value: Any) -> None:
108
- self.additional_properties[key] = value
109
-
110
- def __delitem__(self, key: str) -> None:
111
- del self.additional_properties[key]
112
-
113
- def __contains__(self, key: str) -> bool:
114
- return key in self.additional_properties
@@ -1,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="UsageResponse")
10
-
11
-
12
- @_attrs_define
13
- class UsageResponse:
14
- """
15
- Attributes:
16
- usage (int):
17
- capacity (int):
18
- distinct_tasks (int):
19
- """
20
-
21
- usage: int
22
- capacity: int
23
- distinct_tasks: int
24
- additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
25
-
26
- def to_dict(self) -> dict[str, Any]:
27
- usage = self.usage
28
-
29
- capacity = self.capacity
30
-
31
- distinct_tasks = self.distinct_tasks
32
-
33
- field_dict: dict[str, Any] = {}
34
- field_dict.update(self.additional_properties)
35
- field_dict.update(
36
- {
37
- "usage": usage,
38
- "capacity": capacity,
39
- "distinct_tasks": distinct_tasks,
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
- usage = d.pop("usage")
49
-
50
- capacity = d.pop("capacity")
51
-
52
- distinct_tasks = d.pop("distinct_tasks")
53
-
54
- usage_response = cls(
55
- usage=usage,
56
- capacity=capacity,
57
- distinct_tasks=distinct_tasks,
58
- )
59
-
60
- usage_response.additional_properties = d
61
- return usage_response
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
@@ -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="WhoamiWhoamiGetResponseWhoamiWhoamiGet")
10
-
11
-
12
- @_attrs_define
13
- class WhoamiWhoamiGetResponseWhoamiWhoamiGet:
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
- whoami_whoami_get_response_whoami_whoami_get = cls()
28
-
29
- whoami_whoami_get_response_whoami_whoami_get.additional_properties = d
30
- return whoami_whoami_get_response_whoami_whoami_get
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,70 +0,0 @@
1
- from __future__ import annotations
2
-
3
- from collections.abc import Mapping
4
- from typing import Any, TypeVar
5
- from uuid import UUID
6
-
7
- from attrs import define as _attrs_define
8
- from attrs import field as _attrs_field
9
-
10
- T = TypeVar("T", bound="WorkflowLeafNodeInput")
11
-
12
-
13
- @_attrs_define
14
- class WorkflowLeafNodeInput:
15
- """
16
- Attributes:
17
- workflow_task_id (UUID):
18
- input_artifact_id (UUID):
19
- """
20
-
21
- workflow_task_id: UUID
22
- input_artifact_id: UUID
23
- additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
24
-
25
- def to_dict(self) -> dict[str, Any]:
26
- workflow_task_id = str(self.workflow_task_id)
27
-
28
- input_artifact_id = str(self.input_artifact_id)
29
-
30
- field_dict: dict[str, Any] = {}
31
- field_dict.update(self.additional_properties)
32
- field_dict.update(
33
- {
34
- "workflow_task_id": workflow_task_id,
35
- "input_artifact_id": input_artifact_id,
36
- }
37
- )
38
-
39
- return field_dict
40
-
41
- @classmethod
42
- def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
43
- d = dict(src_dict)
44
- workflow_task_id = UUID(d.pop("workflow_task_id"))
45
-
46
- input_artifact_id = UUID(d.pop("input_artifact_id"))
47
-
48
- workflow_leaf_node_input = cls(
49
- workflow_task_id=workflow_task_id,
50
- input_artifact_id=input_artifact_id,
51
- )
52
-
53
- workflow_leaf_node_input.additional_properties = d
54
- return workflow_leaf_node_input
55
-
56
- @property
57
- def additional_keys(self) -> list[str]:
58
- return list(self.additional_properties.keys())
59
-
60
- def __getitem__(self, key: str) -> Any:
61
- return self.additional_properties[key]
62
-
63
- def __setitem__(self, key: str, value: Any) -> None:
64
- self.additional_properties[key] = value
65
-
66
- def __delitem__(self, key: str) -> None:
67
- del self.additional_properties[key]
68
-
69
- def __contains__(self, key: str) -> bool:
70
- return key in self.additional_properties