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