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,313 +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.multi_agent_query_params import MultiAgentQueryParams
14
-
15
-
16
- T = TypeVar("T", bound="MapMultiAgentRequestParams")
17
-
18
-
19
- @_attrs_define
20
- class MapMultiAgentRequestParams:
21
- """
22
- Attributes:
23
- query (MultiAgentQueryParams):
24
- input_artifacts (list[UUID] | None | Unset):
25
- context_artifacts (list[UUID] | None | Unset):
26
- label (None | str | Unset): Short task label for use in the UI
27
- description (None | str | Unset): Task description for use in the UI
28
- task_id (None | Unset | UUID):
29
- replaces_task_id (None | Unset | UUID): The ID of the task that this task replaces. Used e.g. by the full
30
- version of a task that replaces a preview version.
31
- twin_artifact_id (None | Unset | UUID): The ID of a reference artifact, e.g. the right table in Deep Merge
32
- operation.
33
- task_type (Literal['multiagent'] | Unset): Default: 'multiagent'.
34
- processing_mode (Literal['map'] | Unset): Default: 'map'.
35
- join_with_input (bool | Unset): Default: False.
36
- """
37
-
38
- query: MultiAgentQueryParams
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["multiagent"] | Unset = "multiagent"
47
- processing_mode: Literal["map"] | Unset = "map"
48
- join_with_input: bool | Unset = False
49
- additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
50
-
51
- def to_dict(self) -> dict[str, Any]:
52
- query = self.query.to_dict()
53
-
54
- input_artifacts: list[str] | None | Unset
55
- if isinstance(self.input_artifacts, Unset):
56
- input_artifacts = UNSET
57
- elif isinstance(self.input_artifacts, list):
58
- input_artifacts = []
59
- for componentsschemas_input_artifacts_ids_item_data in self.input_artifacts:
60
- componentsschemas_input_artifacts_ids_item = str(componentsschemas_input_artifacts_ids_item_data)
61
- input_artifacts.append(componentsschemas_input_artifacts_ids_item)
62
-
63
- else:
64
- input_artifacts = self.input_artifacts
65
-
66
- context_artifacts: list[str] | None | Unset
67
- if isinstance(self.context_artifacts, Unset):
68
- context_artifacts = UNSET
69
- elif isinstance(self.context_artifacts, list):
70
- context_artifacts = []
71
- for componentsschemas_context_artifacts_ids_item_data in self.context_artifacts:
72
- componentsschemas_context_artifacts_ids_item = str(componentsschemas_context_artifacts_ids_item_data)
73
- context_artifacts.append(componentsschemas_context_artifacts_ids_item)
74
-
75
- else:
76
- context_artifacts = self.context_artifacts
77
-
78
- label: None | str | Unset
79
- if isinstance(self.label, Unset):
80
- label = UNSET
81
- else:
82
- label = self.label
83
-
84
- description: None | str | Unset
85
- if isinstance(self.description, Unset):
86
- description = UNSET
87
- else:
88
- description = self.description
89
-
90
- task_id: None | str | Unset
91
- if isinstance(self.task_id, Unset):
92
- task_id = UNSET
93
- elif isinstance(self.task_id, UUID):
94
- task_id = str(self.task_id)
95
- else:
96
- task_id = self.task_id
97
-
98
- replaces_task_id: None | str | Unset
99
- if isinstance(self.replaces_task_id, Unset):
100
- replaces_task_id = UNSET
101
- elif isinstance(self.replaces_task_id, UUID):
102
- replaces_task_id = str(self.replaces_task_id)
103
- else:
104
- replaces_task_id = self.replaces_task_id
105
-
106
- twin_artifact_id: None | str | Unset
107
- if isinstance(self.twin_artifact_id, Unset):
108
- twin_artifact_id = UNSET
109
- elif isinstance(self.twin_artifact_id, UUID):
110
- twin_artifact_id = str(self.twin_artifact_id)
111
- else:
112
- twin_artifact_id = self.twin_artifact_id
113
-
114
- task_type = self.task_type
115
-
116
- processing_mode = self.processing_mode
117
-
118
- join_with_input = self.join_with_input
119
-
120
- field_dict: dict[str, Any] = {}
121
- field_dict.update(self.additional_properties)
122
- field_dict.update(
123
- {
124
- "query": query,
125
- }
126
- )
127
- if input_artifacts is not UNSET:
128
- field_dict["input_artifacts"] = input_artifacts
129
- if context_artifacts is not UNSET:
130
- field_dict["context_artifacts"] = context_artifacts
131
- if label is not UNSET:
132
- field_dict["label"] = label
133
- if description is not UNSET:
134
- field_dict["description"] = description
135
- if task_id is not UNSET:
136
- field_dict["task_id"] = task_id
137
- if replaces_task_id is not UNSET:
138
- field_dict["replaces_task_id"] = replaces_task_id
139
- if twin_artifact_id is not UNSET:
140
- field_dict["twin_artifact_id"] = twin_artifact_id
141
- if task_type is not UNSET:
142
- field_dict["task_type"] = task_type
143
- if processing_mode is not UNSET:
144
- field_dict["processing_mode"] = processing_mode
145
- if join_with_input is not UNSET:
146
- field_dict["join_with_input"] = join_with_input
147
-
148
- return field_dict
149
-
150
- @classmethod
151
- def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
152
- from ..models.multi_agent_query_params import MultiAgentQueryParams
153
-
154
- d = dict(src_dict)
155
- query = MultiAgentQueryParams.from_dict(d.pop("query"))
156
-
157
- def _parse_input_artifacts(data: object) -> list[UUID] | None | Unset:
158
- if data is None:
159
- return data
160
- if isinstance(data, Unset):
161
- return data
162
- try:
163
- if not isinstance(data, list):
164
- raise TypeError()
165
- input_artifacts_type_0 = []
166
- _input_artifacts_type_0 = data
167
- for componentsschemas_input_artifacts_ids_item_data in _input_artifacts_type_0:
168
- componentsschemas_input_artifacts_ids_item = UUID(componentsschemas_input_artifacts_ids_item_data)
169
-
170
- input_artifacts_type_0.append(componentsschemas_input_artifacts_ids_item)
171
-
172
- return input_artifacts_type_0
173
- except (TypeError, ValueError, AttributeError, KeyError):
174
- pass
175
- return cast(list[UUID] | None | Unset, data)
176
-
177
- input_artifacts = _parse_input_artifacts(d.pop("input_artifacts", UNSET))
178
-
179
- def _parse_context_artifacts(data: object) -> list[UUID] | None | Unset:
180
- if data is None:
181
- return data
182
- if isinstance(data, Unset):
183
- return data
184
- try:
185
- if not isinstance(data, list):
186
- raise TypeError()
187
- context_artifacts_type_0 = []
188
- _context_artifacts_type_0 = data
189
- for componentsschemas_context_artifacts_ids_item_data in _context_artifacts_type_0:
190
- componentsschemas_context_artifacts_ids_item = UUID(
191
- componentsschemas_context_artifacts_ids_item_data
192
- )
193
-
194
- context_artifacts_type_0.append(componentsschemas_context_artifacts_ids_item)
195
-
196
- return context_artifacts_type_0
197
- except (TypeError, ValueError, AttributeError, KeyError):
198
- pass
199
- return cast(list[UUID] | None | Unset, data)
200
-
201
- context_artifacts = _parse_context_artifacts(d.pop("context_artifacts", UNSET))
202
-
203
- def _parse_label(data: object) -> None | str | Unset:
204
- if data is None:
205
- return data
206
- if isinstance(data, Unset):
207
- return data
208
- return cast(None | str | Unset, data)
209
-
210
- label = _parse_label(d.pop("label", UNSET))
211
-
212
- def _parse_description(data: object) -> None | str | Unset:
213
- if data is None:
214
- return data
215
- if isinstance(data, Unset):
216
- return data
217
- return cast(None | str | Unset, data)
218
-
219
- description = _parse_description(d.pop("description", UNSET))
220
-
221
- def _parse_task_id(data: object) -> None | Unset | UUID:
222
- if data is None:
223
- return data
224
- if isinstance(data, Unset):
225
- return data
226
- try:
227
- if not isinstance(data, str):
228
- raise TypeError()
229
- task_id_type_0 = UUID(data)
230
-
231
- return task_id_type_0
232
- except (TypeError, ValueError, AttributeError, KeyError):
233
- pass
234
- return cast(None | Unset | UUID, data)
235
-
236
- task_id = _parse_task_id(d.pop("task_id", UNSET))
237
-
238
- def _parse_replaces_task_id(data: object) -> None | Unset | UUID:
239
- if data is None:
240
- return data
241
- if isinstance(data, Unset):
242
- return data
243
- try:
244
- if not isinstance(data, str):
245
- raise TypeError()
246
- replaces_task_id_type_0 = UUID(data)
247
-
248
- return replaces_task_id_type_0
249
- except (TypeError, ValueError, AttributeError, KeyError):
250
- pass
251
- return cast(None | Unset | UUID, data)
252
-
253
- replaces_task_id = _parse_replaces_task_id(d.pop("replaces_task_id", UNSET))
254
-
255
- def _parse_twin_artifact_id(data: object) -> None | Unset | UUID:
256
- if data is None:
257
- return data
258
- if isinstance(data, Unset):
259
- return data
260
- try:
261
- if not isinstance(data, str):
262
- raise TypeError()
263
- twin_artifact_id_type_0 = UUID(data)
264
-
265
- return twin_artifact_id_type_0
266
- except (TypeError, ValueError, AttributeError, KeyError):
267
- pass
268
- return cast(None | Unset | UUID, data)
269
-
270
- twin_artifact_id = _parse_twin_artifact_id(d.pop("twin_artifact_id", UNSET))
271
-
272
- task_type = cast(Literal["multiagent"] | Unset, d.pop("task_type", UNSET))
273
- if task_type != "multiagent" and not isinstance(task_type, Unset):
274
- raise ValueError(f"task_type must match const 'multiagent', got '{task_type}'")
275
-
276
- processing_mode = cast(Literal["map"] | Unset, d.pop("processing_mode", UNSET))
277
- if processing_mode != "map" and not isinstance(processing_mode, Unset):
278
- raise ValueError(f"processing_mode must match const 'map', got '{processing_mode}'")
279
-
280
- join_with_input = d.pop("join_with_input", UNSET)
281
-
282
- map_multi_agent_request_params = cls(
283
- query=query,
284
- input_artifacts=input_artifacts,
285
- context_artifacts=context_artifacts,
286
- label=label,
287
- description=description,
288
- task_id=task_id,
289
- replaces_task_id=replaces_task_id,
290
- twin_artifact_id=twin_artifact_id,
291
- task_type=task_type,
292
- processing_mode=processing_mode,
293
- join_with_input=join_with_input,
294
- )
295
-
296
- map_multi_agent_request_params.additional_properties = d
297
- return map_multi_agent_request_params
298
-
299
- @property
300
- def additional_keys(self) -> list[str]:
301
- return list(self.additional_properties.keys())
302
-
303
- def __getitem__(self, key: str) -> Any:
304
- return self.additional_properties[key]
305
-
306
- def __setitem__(self, key: str, value: Any) -> None:
307
- self.additional_properties[key] = value
308
-
309
- def __delitem__(self, key: str) -> None:
310
- del self.additional_properties[key]
311
-
312
- def __contains__(self, key: str) -> bool:
313
- return key in self.additional_properties
@@ -1,98 +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
- from ..models.event_type import EventType
11
- from ..types import UNSET, Unset
12
-
13
- T = TypeVar("T", bound="MessageCreatedPayload")
14
-
15
-
16
- @_attrs_define
17
- class MessageCreatedPayload:
18
- """
19
- Attributes:
20
- conversation_ids (list[UUID]):
21
- event_type (EventType | Unset):
22
- hide_on_frontend (bool | Unset): Default: False.
23
- """
24
-
25
- conversation_ids: list[UUID]
26
- event_type: EventType | Unset = UNSET
27
- hide_on_frontend: bool | Unset = False
28
- additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
29
-
30
- def to_dict(self) -> dict[str, Any]:
31
- conversation_ids = []
32
- for conversation_ids_item_data in self.conversation_ids:
33
- conversation_ids_item = str(conversation_ids_item_data)
34
- conversation_ids.append(conversation_ids_item)
35
-
36
- event_type: str | Unset = UNSET
37
- if not isinstance(self.event_type, Unset):
38
- event_type = self.event_type.value
39
-
40
- hide_on_frontend = self.hide_on_frontend
41
-
42
- field_dict: dict[str, Any] = {}
43
- field_dict.update(self.additional_properties)
44
- field_dict.update(
45
- {
46
- "conversation_ids": conversation_ids,
47
- }
48
- )
49
- if event_type is not UNSET:
50
- field_dict["event_type"] = event_type
51
- if hide_on_frontend is not UNSET:
52
- field_dict["hide_on_frontend"] = hide_on_frontend
53
-
54
- return field_dict
55
-
56
- @classmethod
57
- def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
58
- d = dict(src_dict)
59
- conversation_ids = []
60
- _conversation_ids = d.pop("conversation_ids")
61
- for conversation_ids_item_data in _conversation_ids:
62
- conversation_ids_item = UUID(conversation_ids_item_data)
63
-
64
- conversation_ids.append(conversation_ids_item)
65
-
66
- _event_type = d.pop("event_type", UNSET)
67
- event_type: EventType | Unset
68
- if isinstance(_event_type, Unset):
69
- event_type = UNSET
70
- else:
71
- event_type = EventType(_event_type)
72
-
73
- hide_on_frontend = d.pop("hide_on_frontend", UNSET)
74
-
75
- message_created_payload = cls(
76
- conversation_ids=conversation_ids,
77
- event_type=event_type,
78
- hide_on_frontend=hide_on_frontend,
79
- )
80
-
81
- message_created_payload.additional_properties = d
82
- return message_created_payload
83
-
84
- @property
85
- def additional_keys(self) -> list[str]:
86
- return list(self.additional_properties.keys())
87
-
88
- def __getitem__(self, key: str) -> Any:
89
- return self.additional_properties[key]
90
-
91
- def __setitem__(self, key: str, value: Any) -> None:
92
- self.additional_properties[key] = value
93
-
94
- def __delitem__(self, key: str) -> None:
95
- del self.additional_properties[key]
96
-
97
- def __contains__(self, key: str) -> bool:
98
- return key in self.additional_properties
@@ -1,264 +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
- from ..models.llm_enum import LLMEnum
10
- from ..models.multi_agent_effort_level import MultiAgentEffortLevel
11
- from ..models.response_schema_type import ResponseSchemaType
12
- from ..types import UNSET, Unset
13
-
14
- T = TypeVar("T", bound="MultiAgentQueryParams")
15
-
16
-
17
- @_attrs_define
18
- class MultiAgentQueryParams:
19
- """
20
- Attributes:
21
- task (str): The task assigned to the multi-agent
22
- llm (LLMEnum | Unset):
23
- agenda (str | Unset): Research agenda, if any Default: ''.
24
- selected_input_fields (list[str] | None | Unset): Fields to select from the input artifacts. If None, select all
25
- fields.
26
- apply_at_depth (int | Unset): Default: -1.
27
- preview (bool | Unset): When true, process only the first 5 inputs in map/map-expand operations Default: False.
28
- is_expand (bool | Unset): When true, treat outputs as lists (expand behavior) Default: False.
29
- flatten_map_expand (bool | Unset): When true with MAP mode and is_expand, flatten expanded results into single
30
- group instead of nested groups Default: True.
31
- response_schema (Any | Unset): Default: {'_model_name': 'Root', 'answer': {'description': 'The response
32
- answer', 'optional': False, 'type': 'str'}}.
33
- response_schema_type (ResponseSchemaType | Unset): Type of response schema format.
34
- effort_level (MultiAgentEffortLevel | Unset): Effort level for the multi-agent task.
35
- agent_model (LLMEnum | Unset):
36
- driver_model (LLMEnum | Unset):
37
- agent_step_timeout_seconds (int | Unset): Agent step timeout in seconds (8-1400) Default: 300.
38
- timeout_seconds (int | Unset): Full task timeout in seconds (20-360000) Default: 600.
39
- retries (int | Unset): Number of retries (0-5) Default: 2.
40
- """
41
-
42
- task: str
43
- llm: LLMEnum | Unset = UNSET
44
- agenda: str | Unset = ""
45
- selected_input_fields: list[str] | None | Unset = UNSET
46
- apply_at_depth: int | Unset = -1
47
- preview: bool | Unset = False
48
- is_expand: bool | Unset = False
49
- flatten_map_expand: bool | Unset = True
50
- response_schema: Any | Unset = {
51
- "_model_name": "Root",
52
- "answer": {"description": "The response answer", "optional": False, "type": "str"},
53
- }
54
- response_schema_type: ResponseSchemaType | Unset = UNSET
55
- effort_level: MultiAgentEffortLevel | Unset = UNSET
56
- agent_model: LLMEnum | Unset = UNSET
57
- driver_model: LLMEnum | Unset = UNSET
58
- agent_step_timeout_seconds: int | Unset = 300
59
- timeout_seconds: int | Unset = 600
60
- retries: int | Unset = 2
61
- additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
62
-
63
- def to_dict(self) -> dict[str, Any]:
64
- task = self.task
65
-
66
- llm: str | Unset = UNSET
67
- if not isinstance(self.llm, Unset):
68
- llm = self.llm.value
69
-
70
- agenda = self.agenda
71
-
72
- selected_input_fields: list[str] | None | Unset
73
- if isinstance(self.selected_input_fields, Unset):
74
- selected_input_fields = UNSET
75
- elif isinstance(self.selected_input_fields, list):
76
- selected_input_fields = self.selected_input_fields
77
-
78
- else:
79
- selected_input_fields = self.selected_input_fields
80
-
81
- apply_at_depth = self.apply_at_depth
82
-
83
- preview = self.preview
84
-
85
- is_expand = self.is_expand
86
-
87
- flatten_map_expand = self.flatten_map_expand
88
-
89
- response_schema = self.response_schema
90
-
91
- response_schema_type: str | Unset = UNSET
92
- if not isinstance(self.response_schema_type, Unset):
93
- response_schema_type = self.response_schema_type.value
94
-
95
- effort_level: str | Unset = UNSET
96
- if not isinstance(self.effort_level, Unset):
97
- effort_level = self.effort_level.value
98
-
99
- agent_model: str | Unset = UNSET
100
- if not isinstance(self.agent_model, Unset):
101
- agent_model = self.agent_model.value
102
-
103
- driver_model: str | Unset = UNSET
104
- if not isinstance(self.driver_model, Unset):
105
- driver_model = self.driver_model.value
106
-
107
- agent_step_timeout_seconds = self.agent_step_timeout_seconds
108
-
109
- timeout_seconds = self.timeout_seconds
110
-
111
- retries = self.retries
112
-
113
- field_dict: dict[str, Any] = {}
114
- field_dict.update(self.additional_properties)
115
- field_dict.update(
116
- {
117
- "task": task,
118
- }
119
- )
120
- if llm is not UNSET:
121
- field_dict["llm"] = llm
122
- if agenda is not UNSET:
123
- field_dict["agenda"] = agenda
124
- if selected_input_fields is not UNSET:
125
- field_dict["selected_input_fields"] = selected_input_fields
126
- if apply_at_depth is not UNSET:
127
- field_dict["apply_at_depth"] = apply_at_depth
128
- if preview is not UNSET:
129
- field_dict["preview"] = preview
130
- if is_expand is not UNSET:
131
- field_dict["is_expand"] = is_expand
132
- if flatten_map_expand is not UNSET:
133
- field_dict["flatten_map_expand"] = flatten_map_expand
134
- if response_schema is not UNSET:
135
- field_dict["response_schema"] = response_schema
136
- if response_schema_type is not UNSET:
137
- field_dict["response_schema_type"] = response_schema_type
138
- if effort_level is not UNSET:
139
- field_dict["effort_level"] = effort_level
140
- if agent_model is not UNSET:
141
- field_dict["agent_model"] = agent_model
142
- if driver_model is not UNSET:
143
- field_dict["driver_model"] = driver_model
144
- if agent_step_timeout_seconds is not UNSET:
145
- field_dict["agent_step_timeout_seconds"] = agent_step_timeout_seconds
146
- if timeout_seconds is not UNSET:
147
- field_dict["timeout_seconds"] = timeout_seconds
148
- if retries is not UNSET:
149
- field_dict["retries"] = retries
150
-
151
- return field_dict
152
-
153
- @classmethod
154
- def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
155
- d = dict(src_dict)
156
- task = d.pop("task")
157
-
158
- _llm = d.pop("llm", UNSET)
159
- llm: LLMEnum | Unset
160
- if isinstance(_llm, Unset):
161
- llm = UNSET
162
- else:
163
- llm = LLMEnum(_llm)
164
-
165
- agenda = d.pop("agenda", UNSET)
166
-
167
- def _parse_selected_input_fields(data: object) -> list[str] | None | Unset:
168
- if data is None:
169
- return data
170
- if isinstance(data, Unset):
171
- return data
172
- try:
173
- if not isinstance(data, list):
174
- raise TypeError()
175
- selected_input_fields_type_0 = cast(list[str], data)
176
-
177
- return selected_input_fields_type_0
178
- except (TypeError, ValueError, AttributeError, KeyError):
179
- pass
180
- return cast(list[str] | None | Unset, data)
181
-
182
- selected_input_fields = _parse_selected_input_fields(d.pop("selected_input_fields", UNSET))
183
-
184
- apply_at_depth = d.pop("apply_at_depth", UNSET)
185
-
186
- preview = d.pop("preview", UNSET)
187
-
188
- is_expand = d.pop("is_expand", UNSET)
189
-
190
- flatten_map_expand = d.pop("flatten_map_expand", UNSET)
191
-
192
- response_schema = d.pop("response_schema", UNSET)
193
-
194
- _response_schema_type = d.pop("response_schema_type", UNSET)
195
- response_schema_type: ResponseSchemaType | Unset
196
- if isinstance(_response_schema_type, Unset):
197
- response_schema_type = UNSET
198
- else:
199
- response_schema_type = ResponseSchemaType(_response_schema_type)
200
-
201
- _effort_level = d.pop("effort_level", UNSET)
202
- effort_level: MultiAgentEffortLevel | Unset
203
- if isinstance(_effort_level, Unset):
204
- effort_level = UNSET
205
- else:
206
- effort_level = MultiAgentEffortLevel(_effort_level)
207
-
208
- _agent_model = d.pop("agent_model", UNSET)
209
- agent_model: LLMEnum | Unset
210
- if isinstance(_agent_model, Unset):
211
- agent_model = UNSET
212
- else:
213
- agent_model = LLMEnum(_agent_model)
214
-
215
- _driver_model = d.pop("driver_model", UNSET)
216
- driver_model: LLMEnum | Unset
217
- if isinstance(_driver_model, Unset):
218
- driver_model = UNSET
219
- else:
220
- driver_model = LLMEnum(_driver_model)
221
-
222
- agent_step_timeout_seconds = d.pop("agent_step_timeout_seconds", UNSET)
223
-
224
- timeout_seconds = d.pop("timeout_seconds", UNSET)
225
-
226
- retries = d.pop("retries", UNSET)
227
-
228
- multi_agent_query_params = cls(
229
- task=task,
230
- llm=llm,
231
- agenda=agenda,
232
- selected_input_fields=selected_input_fields,
233
- apply_at_depth=apply_at_depth,
234
- preview=preview,
235
- is_expand=is_expand,
236
- flatten_map_expand=flatten_map_expand,
237
- response_schema=response_schema,
238
- response_schema_type=response_schema_type,
239
- effort_level=effort_level,
240
- agent_model=agent_model,
241
- driver_model=driver_model,
242
- agent_step_timeout_seconds=agent_step_timeout_seconds,
243
- timeout_seconds=timeout_seconds,
244
- retries=retries,
245
- )
246
-
247
- multi_agent_query_params.additional_properties = d
248
- return multi_agent_query_params
249
-
250
- @property
251
- def additional_keys(self) -> list[str]:
252
- return list(self.additional_properties.keys())
253
-
254
- def __getitem__(self, key: str) -> Any:
255
- return self.additional_properties[key]
256
-
257
- def __setitem__(self, key: str, value: Any) -> None:
258
- self.additional_properties[key] = value
259
-
260
- def __delitem__(self, key: str) -> None:
261
- del self.additional_properties[key]
262
-
263
- def __contains__(self, key: str) -> bool:
264
- return key in self.additional_properties