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,143 +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 ..types import UNSET, Unset
11
-
12
- T = TypeVar("T", bound="DeepMergePublicParams")
13
-
14
-
15
- @_attrs_define
16
- class DeepMergePublicParams:
17
- """
18
- Attributes:
19
- task (str): The task for each agent to perform.
20
- merge_on_left (None | str | Unset): Column name for merge table
21
- merge_on_right (None | str | Unset): Column name for merge table
22
- merge_model (LLMEnum | None | Unset): LLM model for merge operation Default: LLMEnum.GEMINI_3_FLASH_MINIMAL.
23
- preview (bool | Unset): When true, process only the first few inputs Default: False.
24
- """
25
-
26
- task: str
27
- merge_on_left: None | str | Unset = UNSET
28
- merge_on_right: None | str | Unset = UNSET
29
- merge_model: LLMEnum | None | Unset = LLMEnum.GEMINI_3_FLASH_MINIMAL
30
- preview: bool | Unset = False
31
- additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
32
-
33
- def to_dict(self) -> dict[str, Any]:
34
- task = self.task
35
-
36
- merge_on_left: None | str | Unset
37
- if isinstance(self.merge_on_left, Unset):
38
- merge_on_left = UNSET
39
- else:
40
- merge_on_left = self.merge_on_left
41
-
42
- merge_on_right: None | str | Unset
43
- if isinstance(self.merge_on_right, Unset):
44
- merge_on_right = UNSET
45
- else:
46
- merge_on_right = self.merge_on_right
47
-
48
- merge_model: None | str | Unset
49
- if isinstance(self.merge_model, Unset):
50
- merge_model = UNSET
51
- elif isinstance(self.merge_model, LLMEnum):
52
- merge_model = self.merge_model.value
53
- else:
54
- merge_model = self.merge_model
55
-
56
- preview = self.preview
57
-
58
- field_dict: dict[str, Any] = {}
59
- field_dict.update(self.additional_properties)
60
- field_dict.update(
61
- {
62
- "task": task,
63
- }
64
- )
65
- if merge_on_left is not UNSET:
66
- field_dict["merge_on_left"] = merge_on_left
67
- if merge_on_right is not UNSET:
68
- field_dict["merge_on_right"] = merge_on_right
69
- if merge_model is not UNSET:
70
- field_dict["merge_model"] = merge_model
71
- if preview is not UNSET:
72
- field_dict["preview"] = preview
73
-
74
- return field_dict
75
-
76
- @classmethod
77
- def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
78
- d = dict(src_dict)
79
- task = d.pop("task")
80
-
81
- def _parse_merge_on_left(data: object) -> None | str | Unset:
82
- if data is None:
83
- return data
84
- if isinstance(data, Unset):
85
- return data
86
- return cast(None | str | Unset, data)
87
-
88
- merge_on_left = _parse_merge_on_left(d.pop("merge_on_left", UNSET))
89
-
90
- def _parse_merge_on_right(data: object) -> None | str | Unset:
91
- if data is None:
92
- return data
93
- if isinstance(data, Unset):
94
- return data
95
- return cast(None | str | Unset, data)
96
-
97
- merge_on_right = _parse_merge_on_right(d.pop("merge_on_right", UNSET))
98
-
99
- def _parse_merge_model(data: object) -> LLMEnum | None | Unset:
100
- if data is None:
101
- return data
102
- if isinstance(data, Unset):
103
- return data
104
- try:
105
- if not isinstance(data, str):
106
- raise TypeError()
107
- merge_model_type_0 = LLMEnum(data)
108
-
109
- return merge_model_type_0
110
- except (TypeError, ValueError, AttributeError, KeyError):
111
- pass
112
- return cast(LLMEnum | None | Unset, data)
113
-
114
- merge_model = _parse_merge_model(d.pop("merge_model", UNSET))
115
-
116
- preview = d.pop("preview", UNSET)
117
-
118
- deep_merge_public_params = cls(
119
- task=task,
120
- merge_on_left=merge_on_left,
121
- merge_on_right=merge_on_right,
122
- merge_model=merge_model,
123
- preview=preview,
124
- )
125
-
126
- deep_merge_public_params.additional_properties = d
127
- return deep_merge_public_params
128
-
129
- @property
130
- def additional_keys(self) -> list[str]:
131
- return list(self.additional_properties.keys())
132
-
133
- def __getitem__(self, key: str) -> Any:
134
- return self.additional_properties[key]
135
-
136
- def __setitem__(self, key: str, value: Any) -> None:
137
- self.additional_properties[key] = value
138
-
139
- def __delitem__(self, key: str) -> None:
140
- del self.additional_properties[key]
141
-
142
- def __contains__(self, key: str) -> bool:
143
- return key in self.additional_properties
@@ -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.deep_merge_public_params import DeepMergePublicParams
14
-
15
-
16
- T = TypeVar("T", bound="DeepMergeRequest")
17
-
18
-
19
- @_attrs_define
20
- class DeepMergeRequest:
21
- """
22
- Attributes:
23
- query (DeepMergePublicParams):
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['deep_merge'] | Unset): Default: 'deep_merge'.
34
- processing_mode (Literal['map'] | Unset): Default: 'map'.
35
- join_with_input (bool | Unset): Default: True.
36
- """
37
-
38
- query: DeepMergePublicParams
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["deep_merge"] | Unset = "deep_merge"
47
- processing_mode: Literal["map"] | Unset = "map"
48
- join_with_input: bool | Unset = True
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.deep_merge_public_params import DeepMergePublicParams
153
-
154
- d = dict(src_dict)
155
- query = DeepMergePublicParams.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["deep_merge"] | Unset, d.pop("task_type", UNSET))
273
- if task_type != "deep_merge" and not isinstance(task_type, Unset):
274
- raise ValueError(f"task_type must match const 'deep_merge', 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
- deep_merge_request = 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
- deep_merge_request.additional_properties = d
297
- return deep_merge_request
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,109 +0,0 @@
1
- from __future__ import annotations
2
-
3
- from collections.abc import Mapping
4
- from typing import Any, TypeVar
5
-
6
- from attrs import define as _attrs_define
7
- from attrs import field as _attrs_field
8
-
9
- from ..types import UNSET, Unset
10
-
11
- T = TypeVar("T", bound="DeepRankPublicParams")
12
-
13
-
14
- @_attrs_define
15
- class DeepRankPublicParams:
16
- """
17
- Attributes:
18
- task (str): The task for each agent to perform.
19
- response_schema (Any): The schema for each agent's response.
20
- field_to_sort_by (str): The field to use when sorting the output artifacts. Must be a top-level field in the
21
- root model of response_schema. Typically has a numeric type but not necessarily.
22
- ascending_order (bool | Unset): If true, sort the output artifacts in ascending order according to
23
- field_to_sort_by Default: True.
24
- include_provenance_and_notes (bool | Unset): Whether to include an additional provenance and notes field in the
25
- output and prompt Default: True.
26
- preview (bool | Unset): When true, process only the first few inputs Default: False.
27
- """
28
-
29
- task: str
30
- response_schema: Any
31
- field_to_sort_by: str
32
- ascending_order: bool | Unset = True
33
- include_provenance_and_notes: bool | Unset = True
34
- preview: bool | Unset = False
35
- additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
36
-
37
- def to_dict(self) -> dict[str, Any]:
38
- task = self.task
39
-
40
- response_schema = self.response_schema
41
-
42
- field_to_sort_by = self.field_to_sort_by
43
-
44
- ascending_order = self.ascending_order
45
-
46
- include_provenance_and_notes = self.include_provenance_and_notes
47
-
48
- preview = self.preview
49
-
50
- field_dict: dict[str, Any] = {}
51
- field_dict.update(self.additional_properties)
52
- field_dict.update(
53
- {
54
- "task": task,
55
- "response_schema": response_schema,
56
- "field_to_sort_by": field_to_sort_by,
57
- }
58
- )
59
- if ascending_order is not UNSET:
60
- field_dict["ascending_order"] = ascending_order
61
- if include_provenance_and_notes is not UNSET:
62
- field_dict["include_provenance_and_notes"] = include_provenance_and_notes
63
- if preview is not UNSET:
64
- field_dict["preview"] = preview
65
-
66
- return field_dict
67
-
68
- @classmethod
69
- def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
70
- d = dict(src_dict)
71
- task = d.pop("task")
72
-
73
- response_schema = d.pop("response_schema")
74
-
75
- field_to_sort_by = d.pop("field_to_sort_by")
76
-
77
- ascending_order = d.pop("ascending_order", UNSET)
78
-
79
- include_provenance_and_notes = d.pop("include_provenance_and_notes", UNSET)
80
-
81
- preview = d.pop("preview", UNSET)
82
-
83
- deep_rank_public_params = cls(
84
- task=task,
85
- response_schema=response_schema,
86
- field_to_sort_by=field_to_sort_by,
87
- ascending_order=ascending_order,
88
- include_provenance_and_notes=include_provenance_and_notes,
89
- preview=preview,
90
- )
91
-
92
- deep_rank_public_params.additional_properties = d
93
- return deep_rank_public_params
94
-
95
- @property
96
- def additional_keys(self) -> list[str]:
97
- return list(self.additional_properties.keys())
98
-
99
- def __getitem__(self, key: str) -> Any:
100
- return self.additional_properties[key]
101
-
102
- def __setitem__(self, key: str, value: Any) -> None:
103
- self.additional_properties[key] = value
104
-
105
- def __delitem__(self, key: str) -> None:
106
- del self.additional_properties[key]
107
-
108
- def __contains__(self, key: str) -> bool:
109
- return key in self.additional_properties