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