everyrow 0.1.10__py3-none-any.whl → 0.2.0__py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (207) hide show
  1. everyrow/__init__.py +8 -1
  2. everyrow/api_utils.py +5 -1
  3. everyrow/billing.py +29 -0
  4. everyrow/constants.py +1 -1
  5. everyrow/generated/__init__.py +1 -1
  6. everyrow/generated/api/{default/interrupt_chat_task_tasks_chat_interrupt_post.py → artifacts/create_artifact_artifacts_post.py} +38 -32
  7. everyrow/generated/api/billing/__init__.py +1 -0
  8. everyrow/generated/api/{default/get_queues_stats_jobs_queues_get.py → billing/get_billing_balance_billing_get.py} +25 -17
  9. everyrow/generated/api/operations/__init__.py +1 -0
  10. everyrow/generated/api/{default/re_execute_task_endpoint_tasks_re_execute_post.py → operations/agent_map_operations_agent_map_post.py} +53 -41
  11. everyrow/generated/api/{default/create_workflow_from_artifact_workflows_from_artifact_post.py → operations/dedupe_operations_dedupe_post.py} +39 -33
  12. everyrow/generated/api/{default/submit_task_tasks_post.py → operations/merge_operations_merge_post.py} +37 -29
  13. everyrow/generated/api/{default/copy_workflow_endpoint_workflows_copy_post.py → operations/rank_operations_rank_post.py} +43 -29
  14. everyrow/generated/api/{default/export_to_google_sheets_export_post.py → operations/screen_operations_screen_post.py} +43 -31
  15. everyrow/generated/api/operations/single_agent_operations_single_agent_post.py +236 -0
  16. everyrow/generated/api/sessions/__init__.py +1 -0
  17. everyrow/generated/api/{default/copy_artifacts_artifacts_copy_post.py → sessions/create_session_endpoint_sessions_post.py} +35 -27
  18. everyrow/generated/api/tasks/__init__.py +1 -0
  19. everyrow/generated/api/{default/get_job_progress_for_task_jobs_progress_get.py → tasks/get_task_result_tasks_task_id_result_get.py} +45 -33
  20. everyrow/generated/api/{default/get_task_status_endpoint_tasks_task_id_status_get.py → tasks/get_task_status_tasks_task_id_status_get.py} +24 -42
  21. everyrow/generated/models/__init__.py +82 -266
  22. everyrow/generated/models/agent_map_operation.py +315 -0
  23. everyrow/generated/models/{artifact_group_record_metadata_type_0.py → agent_map_operation_input_type_1_item.py} +5 -5
  24. everyrow/generated/models/agent_map_operation_input_type_2.py +46 -0
  25. everyrow/generated/models/{standalone_artifact_record_analysis_type_0.py → agent_map_operation_response_schema_type_0.py} +5 -5
  26. everyrow/generated/models/{create_query_params.py → billing_response.py} +13 -12
  27. everyrow/generated/models/{continue_task_request.py → create_artifact_request.py} +43 -43
  28. everyrow/generated/models/create_artifact_request_data_type_0_item.py +46 -0
  29. everyrow/generated/models/{task_metadata_cols_to_rename_type_0.py → create_artifact_request_data_type_1.py} +5 -5
  30. everyrow/generated/models/{copy_artifacts_response.py → create_artifact_response.py} +12 -12
  31. everyrow/generated/models/{create_session_request.py → create_session.py} +6 -7
  32. everyrow/generated/models/dedupe_operation.py +151 -0
  33. everyrow/generated/models/dedupe_operation_input_type_1_item.py +46 -0
  34. everyrow/generated/models/{import_request_token_data.py → dedupe_operation_input_type_2.py} +5 -5
  35. everyrow/generated/models/error_response.py +109 -0
  36. everyrow/generated/models/{task_insert_query_params.py → error_response_details_type_0.py} +5 -5
  37. everyrow/generated/models/insufficient_balance_error.py +8 -0
  38. everyrow/generated/models/{llm_enum.py → llm_enum_public.py} +1 -20
  39. everyrow/generated/models/merge_operation.py +278 -0
  40. everyrow/generated/models/merge_operation_left_input_type_1_item.py +46 -0
  41. everyrow/generated/models/{chat_completion_message_tool_call.py → merge_operation_left_input_type_2.py} +5 -5
  42. everyrow/generated/models/merge_operation_right_input_type_1_item.py +46 -0
  43. everyrow/generated/models/merge_operation_right_input_type_2.py +46 -0
  44. everyrow/generated/models/merge_operation_use_web_search_type_0.py +10 -0
  45. everyrow/generated/models/operation_response.py +131 -0
  46. everyrow/generated/models/{multi_agent_effort_level.py → public_effort_level.py} +1 -1
  47. everyrow/generated/models/public_task_type.py +12 -0
  48. everyrow/generated/models/rank_operation.py +203 -0
  49. everyrow/generated/models/rank_operation_input_type_1_item.py +46 -0
  50. everyrow/generated/models/{export_request_token_data.py → rank_operation_input_type_2.py} +5 -5
  51. everyrow/generated/models/{artifact_group_record_analysis_type_0.py → rank_operation_response_schema_type_0.py} +5 -5
  52. everyrow/generated/models/screen_operation.py +186 -0
  53. everyrow/generated/models/screen_operation_input_type_1_item.py +46 -0
  54. everyrow/generated/models/screen_operation_input_type_2.py +46 -0
  55. everyrow/generated/models/screen_operation_response_schema_type_0.py +46 -0
  56. everyrow/generated/models/{create_session_response.py → session_response.py} +7 -8
  57. everyrow/generated/models/single_agent_operation.py +304 -0
  58. everyrow/generated/models/single_agent_operation_input_type_1_item.py +46 -0
  59. everyrow/generated/models/single_agent_operation_input_type_2.py +46 -0
  60. everyrow/generated/models/single_agent_operation_response_schema_type_0.py +46 -0
  61. everyrow/generated/models/task_result_response.py +185 -0
  62. everyrow/generated/models/task_result_response_data_type_0_item.py +46 -0
  63. everyrow/generated/models/task_result_response_data_type_1.py +46 -0
  64. everyrow/generated/models/task_status_response.py +99 -19
  65. everyrow/ops.py +360 -434
  66. everyrow/session.py +5 -7
  67. everyrow/task.py +68 -115
  68. {everyrow-0.1.10.dist-info → everyrow-0.2.0.dist-info}/METADATA +22 -8
  69. everyrow-0.2.0.dist-info/RECORD +81 -0
  70. everyrow/citations.py +0 -50
  71. everyrow/generated/api/default/continue_task_endpoint_tasks_continue_post.py +0 -208
  72. everyrow/generated/api/default/create_api_key_endpoint_api_keys_create_post.py +0 -186
  73. everyrow/generated/api/default/create_session_endpoint_sessions_create_post.py +0 -198
  74. everyrow/generated/api/default/generate_feedback_endpoint_tasks_generate_feedback_post.py +0 -186
  75. everyrow/generated/api/default/get_artifacts_artifacts_get.py +0 -260
  76. everyrow/generated/api/default/get_default_timeout_seconds_models_default_timeout_seconds_get.py +0 -165
  77. everyrow/generated/api/default/get_metrics_metrics_get.py +0 -80
  78. everyrow/generated/api/default/get_user_usage_usage_get.py +0 -123
  79. everyrow/generated/api/default/healthz_healthz_get.py +0 -127
  80. everyrow/generated/api/default/import_from_google_sheets_import_post.py +0 -170
  81. everyrow/generated/api/default/list_api_keys_endpoint_api_keys_get.py +0 -186
  82. everyrow/generated/api/default/revoke_api_key_endpoint_api_keys_key_id_revoke_post.py +0 -181
  83. everyrow/generated/api/default/revoke_jobs_for_task_jobs_revoke_post.py +0 -164
  84. everyrow/generated/api/default/rollback_to_message_endpoint_tasks_chat_rollback_post.py +0 -186
  85. everyrow/generated/api/default/submit_chat_task_tasks_chat_post.py +0 -164
  86. everyrow/generated/api/default/task_resource_estimation_task_resource_estimation_post.py +0 -319
  87. everyrow/generated/api/default/trigger_workflow_execution_endpoint_workflows_trigger_post.py +0 -166
  88. everyrow/generated/api/default/whoami_whoami_get.py +0 -127
  89. everyrow/generated/models/agent_improvement_instruction.py +0 -69
  90. everyrow/generated/models/agent_query_params.py +0 -383
  91. everyrow/generated/models/agent_query_params_system_prompt_kind_type_0.py +0 -10
  92. everyrow/generated/models/agent_task_args.py +0 -163
  93. everyrow/generated/models/agent_task_args_processing_mode.py +0 -9
  94. everyrow/generated/models/allowed_suggestions.py +0 -9
  95. everyrow/generated/models/api_key_info.py +0 -163
  96. everyrow/generated/models/artifact_changed_payload.py +0 -89
  97. everyrow/generated/models/artifact_group_record.py +0 -363
  98. everyrow/generated/models/artifact_group_record_trace_mapping_type_0.py +0 -46
  99. everyrow/generated/models/artifact_status.py +0 -14
  100. everyrow/generated/models/auto_cohort_conversation_message.py +0 -533
  101. everyrow/generated/models/aux_data.py +0 -128
  102. everyrow/generated/models/aux_data_source_bank.py +0 -59
  103. everyrow/generated/models/chat_message_metadata.py +0 -193
  104. everyrow/generated/models/concatenate_query_params.py +0 -46
  105. everyrow/generated/models/concatenate_request.py +0 -306
  106. everyrow/generated/models/continue_reason.py +0 -9
  107. everyrow/generated/models/controller_improvement_round.py +0 -79
  108. everyrow/generated/models/conversation_changed_payload.py +0 -89
  109. everyrow/generated/models/copy_artifacts_request.py +0 -70
  110. everyrow/generated/models/copy_workflow_request.py +0 -62
  111. everyrow/generated/models/copy_workflow_response.py +0 -70
  112. everyrow/generated/models/create_api_key_request.py +0 -95
  113. everyrow/generated/models/create_api_key_response.py +0 -96
  114. everyrow/generated/models/create_group_query_params.py +0 -61
  115. everyrow/generated/models/create_group_request.py +0 -305
  116. everyrow/generated/models/create_request.py +0 -305
  117. everyrow/generated/models/create_workflow_from_artifact_request.py +0 -92
  118. everyrow/generated/models/create_workflow_from_artifact_response.py +0 -70
  119. everyrow/generated/models/data_frame_method.py +0 -18
  120. everyrow/generated/models/date_cutoffs.py +0 -145
  121. everyrow/generated/models/dedupe_public_params.py +0 -64
  122. everyrow/generated/models/dedupe_request_params.py +0 -311
  123. everyrow/generated/models/deep_merge_public_params.py +0 -143
  124. everyrow/generated/models/deep_merge_request.py +0 -313
  125. everyrow/generated/models/deep_rank_public_params.py +0 -109
  126. everyrow/generated/models/deep_rank_request.py +0 -313
  127. everyrow/generated/models/deep_screen_public_params.py +0 -132
  128. everyrow/generated/models/deep_screen_request.py +0 -313
  129. everyrow/generated/models/derive_expression.py +0 -69
  130. everyrow/generated/models/derive_query_params.py +0 -75
  131. everyrow/generated/models/derive_request.py +0 -307
  132. everyrow/generated/models/document_query_tool.py +0 -12
  133. everyrow/generated/models/drop_columns_query_params.py +0 -61
  134. everyrow/generated/models/drop_columns_request.py +0 -305
  135. everyrow/generated/models/event_type.py +0 -14
  136. everyrow/generated/models/execution_metadata.py +0 -146
  137. everyrow/generated/models/export_request.py +0 -75
  138. everyrow/generated/models/export_to_google_sheets_export_post_response_export_to_google_sheets_export_post.py +0 -46
  139. everyrow/generated/models/filter_query_params.py +0 -91
  140. everyrow/generated/models/filter_request.py +0 -305
  141. everyrow/generated/models/flatten_query_params.py +0 -46
  142. everyrow/generated/models/flatten_request.py +0 -305
  143. everyrow/generated/models/generate_feedback_request.py +0 -62
  144. everyrow/generated/models/group_by_query_params.py +0 -62
  145. everyrow/generated/models/group_by_request.py +0 -305
  146. everyrow/generated/models/healthz_healthz_get_response_healthz_healthz_get.py +0 -46
  147. everyrow/generated/models/image_chat_content_part.py +0 -80
  148. everyrow/generated/models/image_chat_content_part_image_url.py +0 -46
  149. everyrow/generated/models/import_from_google_sheets_import_post_response_import_from_google_sheets_import_post.py +0 -46
  150. everyrow/generated/models/import_request.py +0 -83
  151. everyrow/generated/models/join_query_params.py +0 -73
  152. everyrow/generated/models/join_request.py +0 -305
  153. everyrow/generated/models/map_agent_request_params.py +0 -313
  154. everyrow/generated/models/map_multi_agent_request_params.py +0 -313
  155. everyrow/generated/models/message_created_payload.py +0 -98
  156. everyrow/generated/models/multi_agent_query_params.py +0 -264
  157. everyrow/generated/models/multi_modal_chat_message.py +0 -160
  158. everyrow/generated/models/multi_modal_chat_message_role.py +0 -10
  159. everyrow/generated/models/preview_metadata.py +0 -144
  160. everyrow/generated/models/processing_mode.py +0 -10
  161. everyrow/generated/models/progress_status.py +0 -83
  162. everyrow/generated/models/queue_stats.py +0 -77
  163. everyrow/generated/models/reduce_agent_request_params.py +0 -305
  164. everyrow/generated/models/reduce_multi_agent_request_params.py +0 -305
  165. everyrow/generated/models/resource_estimation_response.py +0 -85
  166. everyrow/generated/models/response_schema_type.py +0 -9
  167. everyrow/generated/models/revoke_api_key_response.py +0 -61
  168. everyrow/generated/models/rollback_to_message_request.py +0 -62
  169. everyrow/generated/models/rollback_to_message_response.py +0 -77
  170. everyrow/generated/models/session_changed_payload.py +0 -69
  171. everyrow/generated/models/simple_chat_message.py +0 -121
  172. everyrow/generated/models/simple_chat_message_role.py +0 -10
  173. everyrow/generated/models/simple_chat_message_with_tool_calls.py +0 -156
  174. everyrow/generated/models/source_database_entry.py +0 -92
  175. everyrow/generated/models/standalone_artifact_record.py +0 -311
  176. everyrow/generated/models/standalone_artifact_record_metadata_type_0.py +0 -46
  177. everyrow/generated/models/standalone_artifact_record_trace_mapping_type_0.py +0 -46
  178. everyrow/generated/models/status_count.py +0 -71
  179. everyrow/generated/models/status_count_status.py +0 -13
  180. everyrow/generated/models/submit_chat_task_body.py +0 -497
  181. everyrow/generated/models/submit_chat_task_body_selected_task_type_type_0.py +0 -11
  182. everyrow/generated/models/submit_task_body.py +0 -745
  183. everyrow/generated/models/task_changed_payload.py +0 -105
  184. everyrow/generated/models/task_effort.py +0 -10
  185. everyrow/generated/models/task_id_request.py +0 -62
  186. everyrow/generated/models/task_insert.py +0 -725
  187. everyrow/generated/models/task_metadata.py +0 -323
  188. everyrow/generated/models/task_response.py +0 -62
  189. everyrow/generated/models/task_type.py +0 -31
  190. everyrow/generated/models/text_chat_content_part.py +0 -74
  191. everyrow/generated/models/tool_response_message.py +0 -127
  192. everyrow/generated/models/toolkit_constants.py +0 -80
  193. everyrow/generated/models/trace_changed_payload.py +0 -94
  194. everyrow/generated/models/trace_info.py +0 -78
  195. everyrow/generated/models/trigger_workflow_execution_request.py +0 -112
  196. everyrow/generated/models/trigger_workflow_execution_request_task_params.py +0 -65
  197. everyrow/generated/models/trigger_workflow_execution_request_task_params_additional_property.py +0 -46
  198. everyrow/generated/models/trigger_workflow_execution_response.py +0 -69
  199. everyrow/generated/models/upload_csv_payload.py +0 -310
  200. everyrow/generated/models/upload_csv_query_params.py +0 -114
  201. everyrow/generated/models/usage_response.py +0 -77
  202. everyrow/generated/models/whoami_whoami_get_response_whoami_whoami_get.py +0 -46
  203. everyrow/generated/models/workflow_leaf_node_input.py +0 -70
  204. everyrow-0.1.10.dist-info/RECORD +0 -183
  205. /everyrow/generated/api/{default → artifacts}/__init__.py +0 -0
  206. {everyrow-0.1.10.dist-info → everyrow-0.2.0.dist-info}/WHEEL +0 -0
  207. {everyrow-0.1.10.dist-info → everyrow-0.2.0.dist-info}/licenses/LICENSE.txt +0 -0
@@ -1,533 +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 ..types import UNSET, Unset
11
-
12
- if TYPE_CHECKING:
13
- from ..models.chat_message_metadata import ChatMessageMetadata
14
- from ..models.concatenate_request import ConcatenateRequest
15
- from ..models.create_group_request import CreateGroupRequest
16
- from ..models.create_request import CreateRequest
17
- from ..models.dedupe_request_params import DedupeRequestParams
18
- from ..models.deep_merge_request import DeepMergeRequest
19
- from ..models.deep_rank_request import DeepRankRequest
20
- from ..models.deep_screen_request import DeepScreenRequest
21
- from ..models.derive_request import DeriveRequest
22
- from ..models.drop_columns_request import DropColumnsRequest
23
- from ..models.filter_request import FilterRequest
24
- from ..models.flatten_request import FlattenRequest
25
- from ..models.group_by_request import GroupByRequest
26
- from ..models.join_request import JoinRequest
27
- from ..models.map_agent_request_params import MapAgentRequestParams
28
- from ..models.map_multi_agent_request_params import MapMultiAgentRequestParams
29
- from ..models.multi_modal_chat_message import MultiModalChatMessage
30
- from ..models.reduce_agent_request_params import ReduceAgentRequestParams
31
- from ..models.reduce_multi_agent_request_params import ReduceMultiAgentRequestParams
32
- from ..models.simple_chat_message import SimpleChatMessage
33
- from ..models.simple_chat_message_with_tool_calls import SimpleChatMessageWithToolCalls
34
- from ..models.tool_response_message import ToolResponseMessage
35
- from ..models.upload_csv_payload import UploadCsvPayload
36
-
37
-
38
- T = TypeVar("T", bound="AutoCohortConversationMessage")
39
-
40
-
41
- @_attrs_define
42
- class AutoCohortConversationMessage:
43
- """
44
- Attributes:
45
- message (MultiModalChatMessage | SimpleChatMessage | SimpleChatMessageWithToolCalls | ToolResponseMessage):
46
- metadata (ChatMessageMetadata | Unset):
47
- parent_message_id (None | Unset | UUID):
48
- payload_is_suggestion (bool | None | Unset):
49
- validated_payload (ConcatenateRequest | CreateGroupRequest | CreateRequest | DedupeRequestParams |
50
- DeepMergeRequest | DeepRankRequest | DeepScreenRequest | DeriveRequest | DropColumnsRequest | FilterRequest |
51
- FlattenRequest | GroupByRequest | JoinRequest | MapAgentRequestParams | MapMultiAgentRequestParams | None |
52
- ReduceAgentRequestParams | ReduceMultiAgentRequestParams | Unset | UploadCsvPayload):
53
- hide_on_frontend (bool | None | Unset):
54
- """
55
-
56
- message: MultiModalChatMessage | SimpleChatMessage | SimpleChatMessageWithToolCalls | ToolResponseMessage
57
- metadata: ChatMessageMetadata | Unset = UNSET
58
- parent_message_id: None | Unset | UUID = UNSET
59
- payload_is_suggestion: bool | None | Unset = UNSET
60
- validated_payload: (
61
- ConcatenateRequest
62
- | CreateGroupRequest
63
- | CreateRequest
64
- | DedupeRequestParams
65
- | DeepMergeRequest
66
- | DeepRankRequest
67
- | DeepScreenRequest
68
- | DeriveRequest
69
- | DropColumnsRequest
70
- | FilterRequest
71
- | FlattenRequest
72
- | GroupByRequest
73
- | JoinRequest
74
- | MapAgentRequestParams
75
- | MapMultiAgentRequestParams
76
- | None
77
- | ReduceAgentRequestParams
78
- | ReduceMultiAgentRequestParams
79
- | Unset
80
- | UploadCsvPayload
81
- ) = UNSET
82
- hide_on_frontend: bool | None | Unset = UNSET
83
- additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
84
-
85
- def to_dict(self) -> dict[str, Any]:
86
- from ..models.concatenate_request import ConcatenateRequest
87
- from ..models.create_group_request import CreateGroupRequest
88
- from ..models.create_request import CreateRequest
89
- from ..models.dedupe_request_params import DedupeRequestParams
90
- from ..models.deep_merge_request import DeepMergeRequest
91
- from ..models.deep_rank_request import DeepRankRequest
92
- from ..models.deep_screen_request import DeepScreenRequest
93
- from ..models.derive_request import DeriveRequest
94
- from ..models.drop_columns_request import DropColumnsRequest
95
- from ..models.filter_request import FilterRequest
96
- from ..models.flatten_request import FlattenRequest
97
- from ..models.group_by_request import GroupByRequest
98
- from ..models.join_request import JoinRequest
99
- from ..models.map_agent_request_params import MapAgentRequestParams
100
- from ..models.map_multi_agent_request_params import MapMultiAgentRequestParams
101
- from ..models.multi_modal_chat_message import MultiModalChatMessage
102
- from ..models.reduce_agent_request_params import ReduceAgentRequestParams
103
- from ..models.reduce_multi_agent_request_params import ReduceMultiAgentRequestParams
104
- from ..models.simple_chat_message import SimpleChatMessage
105
- from ..models.simple_chat_message_with_tool_calls import SimpleChatMessageWithToolCalls
106
- from ..models.upload_csv_payload import UploadCsvPayload
107
-
108
- message: dict[str, Any]
109
- if isinstance(self.message, SimpleChatMessage):
110
- message = self.message.to_dict()
111
- elif isinstance(self.message, MultiModalChatMessage):
112
- message = self.message.to_dict()
113
- elif isinstance(self.message, SimpleChatMessageWithToolCalls):
114
- message = self.message.to_dict()
115
- else:
116
- message = self.message.to_dict()
117
-
118
- metadata: dict[str, Any] | Unset = UNSET
119
- if not isinstance(self.metadata, Unset):
120
- metadata = self.metadata.to_dict()
121
-
122
- parent_message_id: None | str | Unset
123
- if isinstance(self.parent_message_id, Unset):
124
- parent_message_id = UNSET
125
- elif isinstance(self.parent_message_id, UUID):
126
- parent_message_id = str(self.parent_message_id)
127
- else:
128
- parent_message_id = self.parent_message_id
129
-
130
- payload_is_suggestion: bool | None | Unset
131
- if isinstance(self.payload_is_suggestion, Unset):
132
- payload_is_suggestion = UNSET
133
- else:
134
- payload_is_suggestion = self.payload_is_suggestion
135
-
136
- validated_payload: dict[str, Any] | None | Unset
137
- if isinstance(self.validated_payload, Unset):
138
- validated_payload = UNSET
139
- elif isinstance(self.validated_payload, MapAgentRequestParams):
140
- validated_payload = self.validated_payload.to_dict()
141
- elif isinstance(self.validated_payload, ReduceAgentRequestParams):
142
- validated_payload = self.validated_payload.to_dict()
143
- elif isinstance(self.validated_payload, FilterRequest):
144
- validated_payload = self.validated_payload.to_dict()
145
- elif isinstance(self.validated_payload, DeriveRequest):
146
- validated_payload = self.validated_payload.to_dict()
147
- elif isinstance(self.validated_payload, JoinRequest):
148
- validated_payload = self.validated_payload.to_dict()
149
- elif isinstance(self.validated_payload, ConcatenateRequest):
150
- validated_payload = self.validated_payload.to_dict()
151
- elif isinstance(self.validated_payload, DropColumnsRequest):
152
- validated_payload = self.validated_payload.to_dict()
153
- elif isinstance(self.validated_payload, DedupeRequestParams):
154
- validated_payload = self.validated_payload.to_dict()
155
- elif isinstance(self.validated_payload, UploadCsvPayload):
156
- validated_payload = self.validated_payload.to_dict()
157
- elif isinstance(self.validated_payload, MapMultiAgentRequestParams):
158
- validated_payload = self.validated_payload.to_dict()
159
- elif isinstance(self.validated_payload, ReduceMultiAgentRequestParams):
160
- validated_payload = self.validated_payload.to_dict()
161
- elif isinstance(self.validated_payload, CreateRequest):
162
- validated_payload = self.validated_payload.to_dict()
163
- elif isinstance(self.validated_payload, CreateGroupRequest):
164
- validated_payload = self.validated_payload.to_dict()
165
- elif isinstance(self.validated_payload, FlattenRequest):
166
- validated_payload = self.validated_payload.to_dict()
167
- elif isinstance(self.validated_payload, GroupByRequest):
168
- validated_payload = self.validated_payload.to_dict()
169
- elif isinstance(self.validated_payload, DeepRankRequest):
170
- validated_payload = self.validated_payload.to_dict()
171
- elif isinstance(self.validated_payload, DeepMergeRequest):
172
- validated_payload = self.validated_payload.to_dict()
173
- elif isinstance(self.validated_payload, DeepScreenRequest):
174
- validated_payload = self.validated_payload.to_dict()
175
- else:
176
- validated_payload = self.validated_payload
177
-
178
- hide_on_frontend: bool | None | Unset
179
- if isinstance(self.hide_on_frontend, Unset):
180
- hide_on_frontend = UNSET
181
- else:
182
- hide_on_frontend = self.hide_on_frontend
183
-
184
- field_dict: dict[str, Any] = {}
185
- field_dict.update(self.additional_properties)
186
- field_dict.update(
187
- {
188
- "message": message,
189
- }
190
- )
191
- if metadata is not UNSET:
192
- field_dict["metadata"] = metadata
193
- if parent_message_id is not UNSET:
194
- field_dict["parent_message_id"] = parent_message_id
195
- if payload_is_suggestion is not UNSET:
196
- field_dict["payload_is_suggestion"] = payload_is_suggestion
197
- if validated_payload is not UNSET:
198
- field_dict["validated_payload"] = validated_payload
199
- if hide_on_frontend is not UNSET:
200
- field_dict["hide_on_frontend"] = hide_on_frontend
201
-
202
- return field_dict
203
-
204
- @classmethod
205
- def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
206
- from ..models.chat_message_metadata import ChatMessageMetadata
207
- from ..models.concatenate_request import ConcatenateRequest
208
- from ..models.create_group_request import CreateGroupRequest
209
- from ..models.create_request import CreateRequest
210
- from ..models.dedupe_request_params import DedupeRequestParams
211
- from ..models.deep_merge_request import DeepMergeRequest
212
- from ..models.deep_rank_request import DeepRankRequest
213
- from ..models.deep_screen_request import DeepScreenRequest
214
- from ..models.derive_request import DeriveRequest
215
- from ..models.drop_columns_request import DropColumnsRequest
216
- from ..models.filter_request import FilterRequest
217
- from ..models.flatten_request import FlattenRequest
218
- from ..models.group_by_request import GroupByRequest
219
- from ..models.join_request import JoinRequest
220
- from ..models.map_agent_request_params import MapAgentRequestParams
221
- from ..models.map_multi_agent_request_params import MapMultiAgentRequestParams
222
- from ..models.multi_modal_chat_message import MultiModalChatMessage
223
- from ..models.reduce_agent_request_params import ReduceAgentRequestParams
224
- from ..models.reduce_multi_agent_request_params import ReduceMultiAgentRequestParams
225
- from ..models.simple_chat_message import SimpleChatMessage
226
- from ..models.simple_chat_message_with_tool_calls import SimpleChatMessageWithToolCalls
227
- from ..models.tool_response_message import ToolResponseMessage
228
- from ..models.upload_csv_payload import UploadCsvPayload
229
-
230
- d = dict(src_dict)
231
-
232
- def _parse_message(
233
- data: object,
234
- ) -> MultiModalChatMessage | SimpleChatMessage | SimpleChatMessageWithToolCalls | ToolResponseMessage:
235
- try:
236
- if not isinstance(data, dict):
237
- raise TypeError()
238
- message_type_0 = SimpleChatMessage.from_dict(data)
239
-
240
- return message_type_0
241
- except (TypeError, ValueError, AttributeError, KeyError):
242
- pass
243
- try:
244
- if not isinstance(data, dict):
245
- raise TypeError()
246
- message_type_1 = MultiModalChatMessage.from_dict(data)
247
-
248
- return message_type_1
249
- except (TypeError, ValueError, AttributeError, KeyError):
250
- pass
251
- try:
252
- if not isinstance(data, dict):
253
- raise TypeError()
254
- message_type_2 = SimpleChatMessageWithToolCalls.from_dict(data)
255
-
256
- return message_type_2
257
- except (TypeError, ValueError, AttributeError, KeyError):
258
- pass
259
- if not isinstance(data, dict):
260
- raise TypeError()
261
- message_type_3 = ToolResponseMessage.from_dict(data)
262
-
263
- return message_type_3
264
-
265
- message = _parse_message(d.pop("message"))
266
-
267
- _metadata = d.pop("metadata", UNSET)
268
- metadata: ChatMessageMetadata | Unset
269
- if isinstance(_metadata, Unset):
270
- metadata = UNSET
271
- else:
272
- metadata = ChatMessageMetadata.from_dict(_metadata)
273
-
274
- def _parse_parent_message_id(data: object) -> None | Unset | UUID:
275
- if data is None:
276
- return data
277
- if isinstance(data, Unset):
278
- return data
279
- try:
280
- if not isinstance(data, str):
281
- raise TypeError()
282
- parent_message_id_type_0 = UUID(data)
283
-
284
- return parent_message_id_type_0
285
- except (TypeError, ValueError, AttributeError, KeyError):
286
- pass
287
- return cast(None | Unset | UUID, data)
288
-
289
- parent_message_id = _parse_parent_message_id(d.pop("parent_message_id", UNSET))
290
-
291
- def _parse_payload_is_suggestion(data: object) -> bool | None | Unset:
292
- if data is None:
293
- return data
294
- if isinstance(data, Unset):
295
- return data
296
- return cast(bool | None | Unset, data)
297
-
298
- payload_is_suggestion = _parse_payload_is_suggestion(d.pop("payload_is_suggestion", UNSET))
299
-
300
- def _parse_validated_payload(
301
- data: object,
302
- ) -> (
303
- ConcatenateRequest
304
- | CreateGroupRequest
305
- | CreateRequest
306
- | DedupeRequestParams
307
- | DeepMergeRequest
308
- | DeepRankRequest
309
- | DeepScreenRequest
310
- | DeriveRequest
311
- | DropColumnsRequest
312
- | FilterRequest
313
- | FlattenRequest
314
- | GroupByRequest
315
- | JoinRequest
316
- | MapAgentRequestParams
317
- | MapMultiAgentRequestParams
318
- | None
319
- | ReduceAgentRequestParams
320
- | ReduceMultiAgentRequestParams
321
- | Unset
322
- | UploadCsvPayload
323
- ):
324
- if data is None:
325
- return data
326
- if isinstance(data, Unset):
327
- return data
328
- try:
329
- if not isinstance(data, dict):
330
- raise TypeError()
331
- componentsschemas_agent_request_payload_type_0 = MapAgentRequestParams.from_dict(data)
332
-
333
- return componentsschemas_agent_request_payload_type_0
334
- except (TypeError, ValueError, AttributeError, KeyError):
335
- pass
336
- try:
337
- if not isinstance(data, dict):
338
- raise TypeError()
339
- componentsschemas_agent_request_payload_type_1 = ReduceAgentRequestParams.from_dict(data)
340
-
341
- return componentsschemas_agent_request_payload_type_1
342
- except (TypeError, ValueError, AttributeError, KeyError):
343
- pass
344
- try:
345
- if not isinstance(data, dict):
346
- raise TypeError()
347
- validated_payload_type_0_type_1 = FilterRequest.from_dict(data)
348
-
349
- return validated_payload_type_0_type_1
350
- except (TypeError, ValueError, AttributeError, KeyError):
351
- pass
352
- try:
353
- if not isinstance(data, dict):
354
- raise TypeError()
355
- validated_payload_type_0_type_2 = DeriveRequest.from_dict(data)
356
-
357
- return validated_payload_type_0_type_2
358
- except (TypeError, ValueError, AttributeError, KeyError):
359
- pass
360
- try:
361
- if not isinstance(data, dict):
362
- raise TypeError()
363
- validated_payload_type_0_type_3 = JoinRequest.from_dict(data)
364
-
365
- return validated_payload_type_0_type_3
366
- except (TypeError, ValueError, AttributeError, KeyError):
367
- pass
368
- try:
369
- if not isinstance(data, dict):
370
- raise TypeError()
371
- validated_payload_type_0_type_4 = ConcatenateRequest.from_dict(data)
372
-
373
- return validated_payload_type_0_type_4
374
- except (TypeError, ValueError, AttributeError, KeyError):
375
- pass
376
- try:
377
- if not isinstance(data, dict):
378
- raise TypeError()
379
- validated_payload_type_0_type_5 = DropColumnsRequest.from_dict(data)
380
-
381
- return validated_payload_type_0_type_5
382
- except (TypeError, ValueError, AttributeError, KeyError):
383
- pass
384
- try:
385
- if not isinstance(data, dict):
386
- raise TypeError()
387
- validated_payload_type_0_type_6 = DedupeRequestParams.from_dict(data)
388
-
389
- return validated_payload_type_0_type_6
390
- except (TypeError, ValueError, AttributeError, KeyError):
391
- pass
392
- try:
393
- if not isinstance(data, dict):
394
- raise TypeError()
395
- validated_payload_type_0_type_7 = UploadCsvPayload.from_dict(data)
396
-
397
- return validated_payload_type_0_type_7
398
- except (TypeError, ValueError, AttributeError, KeyError):
399
- pass
400
- try:
401
- if not isinstance(data, dict):
402
- raise TypeError()
403
- componentsschemas_multi_agent_request_payload_type_0 = MapMultiAgentRequestParams.from_dict(data)
404
-
405
- return componentsschemas_multi_agent_request_payload_type_0
406
- except (TypeError, ValueError, AttributeError, KeyError):
407
- pass
408
- try:
409
- if not isinstance(data, dict):
410
- raise TypeError()
411
- componentsschemas_multi_agent_request_payload_type_1 = ReduceMultiAgentRequestParams.from_dict(data)
412
-
413
- return componentsschemas_multi_agent_request_payload_type_1
414
- except (TypeError, ValueError, AttributeError, KeyError):
415
- pass
416
- try:
417
- if not isinstance(data, dict):
418
- raise TypeError()
419
- validated_payload_type_0_type_9 = CreateRequest.from_dict(data)
420
-
421
- return validated_payload_type_0_type_9
422
- except (TypeError, ValueError, AttributeError, KeyError):
423
- pass
424
- try:
425
- if not isinstance(data, dict):
426
- raise TypeError()
427
- validated_payload_type_0_type_10 = CreateGroupRequest.from_dict(data)
428
-
429
- return validated_payload_type_0_type_10
430
- except (TypeError, ValueError, AttributeError, KeyError):
431
- pass
432
- try:
433
- if not isinstance(data, dict):
434
- raise TypeError()
435
- validated_payload_type_0_type_11 = FlattenRequest.from_dict(data)
436
-
437
- return validated_payload_type_0_type_11
438
- except (TypeError, ValueError, AttributeError, KeyError):
439
- pass
440
- try:
441
- if not isinstance(data, dict):
442
- raise TypeError()
443
- validated_payload_type_0_type_12 = GroupByRequest.from_dict(data)
444
-
445
- return validated_payload_type_0_type_12
446
- except (TypeError, ValueError, AttributeError, KeyError):
447
- pass
448
- try:
449
- if not isinstance(data, dict):
450
- raise TypeError()
451
- validated_payload_type_0_type_13 = DeepRankRequest.from_dict(data)
452
-
453
- return validated_payload_type_0_type_13
454
- except (TypeError, ValueError, AttributeError, KeyError):
455
- pass
456
- try:
457
- if not isinstance(data, dict):
458
- raise TypeError()
459
- validated_payload_type_0_type_14 = DeepMergeRequest.from_dict(data)
460
-
461
- return validated_payload_type_0_type_14
462
- except (TypeError, ValueError, AttributeError, KeyError):
463
- pass
464
- try:
465
- if not isinstance(data, dict):
466
- raise TypeError()
467
- validated_payload_type_0_type_15 = DeepScreenRequest.from_dict(data)
468
-
469
- return validated_payload_type_0_type_15
470
- except (TypeError, ValueError, AttributeError, KeyError):
471
- pass
472
- return cast(
473
- ConcatenateRequest
474
- | CreateGroupRequest
475
- | CreateRequest
476
- | DedupeRequestParams
477
- | DeepMergeRequest
478
- | DeepRankRequest
479
- | DeepScreenRequest
480
- | DeriveRequest
481
- | DropColumnsRequest
482
- | FilterRequest
483
- | FlattenRequest
484
- | GroupByRequest
485
- | JoinRequest
486
- | MapAgentRequestParams
487
- | MapMultiAgentRequestParams
488
- | None
489
- | ReduceAgentRequestParams
490
- | ReduceMultiAgentRequestParams
491
- | Unset
492
- | UploadCsvPayload,
493
- data,
494
- )
495
-
496
- validated_payload = _parse_validated_payload(d.pop("validated_payload", UNSET))
497
-
498
- def _parse_hide_on_frontend(data: object) -> bool | None | Unset:
499
- if data is None:
500
- return data
501
- if isinstance(data, Unset):
502
- return data
503
- return cast(bool | None | Unset, data)
504
-
505
- hide_on_frontend = _parse_hide_on_frontend(d.pop("hide_on_frontend", UNSET))
506
-
507
- auto_cohort_conversation_message = cls(
508
- message=message,
509
- metadata=metadata,
510
- parent_message_id=parent_message_id,
511
- payload_is_suggestion=payload_is_suggestion,
512
- validated_payload=validated_payload,
513
- hide_on_frontend=hide_on_frontend,
514
- )
515
-
516
- auto_cohort_conversation_message.additional_properties = d
517
- return auto_cohort_conversation_message
518
-
519
- @property
520
- def additional_keys(self) -> list[str]:
521
- return list(self.additional_properties.keys())
522
-
523
- def __getitem__(self, key: str) -> Any:
524
- return self.additional_properties[key]
525
-
526
- def __setitem__(self, key: str, value: Any) -> None:
527
- self.additional_properties[key] = value
528
-
529
- def __delitem__(self, key: str) -> None:
530
- del self.additional_properties[key]
531
-
532
- def __contains__(self, key: str) -> bool:
533
- return key in self.additional_properties
@@ -1,128 +0,0 @@
1
- from __future__ import annotations
2
-
3
- from collections.abc import Mapping
4
- from typing import TYPE_CHECKING, Any, TypeVar, cast
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.aux_data_source_bank import AuxDataSourceBank
13
- from ..models.date_cutoffs import DateCutoffs
14
-
15
-
16
- T = TypeVar("T", bound="AuxData")
17
-
18
-
19
- @_attrs_define
20
- class AuxData:
21
- """
22
- Attributes:
23
- source_bank (AuxDataSourceBank | Unset):
24
- date_cutoffs (DateCutoffs | None | Unset):
25
- justification (None | str | Unset):
26
- """
27
-
28
- source_bank: AuxDataSourceBank | Unset = UNSET
29
- date_cutoffs: DateCutoffs | None | Unset = UNSET
30
- justification: None | str | Unset = UNSET
31
- additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
32
-
33
- def to_dict(self) -> dict[str, Any]:
34
- from ..models.date_cutoffs import DateCutoffs
35
-
36
- source_bank: dict[str, Any] | Unset = UNSET
37
- if not isinstance(self.source_bank, Unset):
38
- source_bank = self.source_bank.to_dict()
39
-
40
- date_cutoffs: dict[str, Any] | None | Unset
41
- if isinstance(self.date_cutoffs, Unset):
42
- date_cutoffs = UNSET
43
- elif isinstance(self.date_cutoffs, DateCutoffs):
44
- date_cutoffs = self.date_cutoffs.to_dict()
45
- else:
46
- date_cutoffs = self.date_cutoffs
47
-
48
- justification: None | str | Unset
49
- if isinstance(self.justification, Unset):
50
- justification = UNSET
51
- else:
52
- justification = self.justification
53
-
54
- field_dict: dict[str, Any] = {}
55
- field_dict.update(self.additional_properties)
56
- field_dict.update({})
57
- if source_bank is not UNSET:
58
- field_dict["source_bank"] = source_bank
59
- if date_cutoffs is not UNSET:
60
- field_dict["date_cutoffs"] = date_cutoffs
61
- if justification is not UNSET:
62
- field_dict["justification"] = justification
63
-
64
- return field_dict
65
-
66
- @classmethod
67
- def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
68
- from ..models.aux_data_source_bank import AuxDataSourceBank
69
- from ..models.date_cutoffs import DateCutoffs
70
-
71
- d = dict(src_dict)
72
- _source_bank = d.pop("source_bank", UNSET)
73
- source_bank: AuxDataSourceBank | Unset
74
- if isinstance(_source_bank, Unset):
75
- source_bank = UNSET
76
- else:
77
- source_bank = AuxDataSourceBank.from_dict(_source_bank)
78
-
79
- def _parse_date_cutoffs(data: object) -> DateCutoffs | None | Unset:
80
- if data is None:
81
- return data
82
- if isinstance(data, Unset):
83
- return data
84
- try:
85
- if not isinstance(data, dict):
86
- raise TypeError()
87
- date_cutoffs_type_0 = DateCutoffs.from_dict(data)
88
-
89
- return date_cutoffs_type_0
90
- except (TypeError, ValueError, AttributeError, KeyError):
91
- pass
92
- return cast(DateCutoffs | None | Unset, data)
93
-
94
- date_cutoffs = _parse_date_cutoffs(d.pop("date_cutoffs", UNSET))
95
-
96
- def _parse_justification(data: object) -> None | str | Unset:
97
- if data is None:
98
- return data
99
- if isinstance(data, Unset):
100
- return data
101
- return cast(None | str | Unset, data)
102
-
103
- justification = _parse_justification(d.pop("justification", UNSET))
104
-
105
- aux_data = cls(
106
- source_bank=source_bank,
107
- date_cutoffs=date_cutoffs,
108
- justification=justification,
109
- )
110
-
111
- aux_data.additional_properties = d
112
- return aux_data
113
-
114
- @property
115
- def additional_keys(self) -> list[str]:
116
- return list(self.additional_properties.keys())
117
-
118
- def __getitem__(self, key: str) -> Any:
119
- return self.additional_properties[key]
120
-
121
- def __setitem__(self, key: str, value: Any) -> None:
122
- self.additional_properties[key] = value
123
-
124
- def __delitem__(self, key: str) -> None:
125
- del self.additional_properties[key]
126
-
127
- def __contains__(self, key: str) -> bool:
128
- return key in self.additional_properties