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,319 +0,0 @@
1
- from http import HTTPStatus
2
- from typing import Any
3
-
4
- import httpx
5
-
6
- from ... import errors
7
- from ...client import AuthenticatedClient, Client
8
- from ...models.concatenate_request import ConcatenateRequest
9
- from ...models.create_group_request import CreateGroupRequest
10
- from ...models.create_request import CreateRequest
11
- from ...models.dedupe_request_params import DedupeRequestParams
12
- from ...models.deep_merge_request import DeepMergeRequest
13
- from ...models.deep_rank_request import DeepRankRequest
14
- from ...models.deep_screen_request import DeepScreenRequest
15
- from ...models.derive_request import DeriveRequest
16
- from ...models.drop_columns_request import DropColumnsRequest
17
- from ...models.filter_request import FilterRequest
18
- from ...models.flatten_request import FlattenRequest
19
- from ...models.group_by_request import GroupByRequest
20
- from ...models.http_validation_error import HTTPValidationError
21
- from ...models.join_request import JoinRequest
22
- from ...models.map_agent_request_params import MapAgentRequestParams
23
- from ...models.map_multi_agent_request_params import MapMultiAgentRequestParams
24
- from ...models.reduce_agent_request_params import ReduceAgentRequestParams
25
- from ...models.reduce_multi_agent_request_params import ReduceMultiAgentRequestParams
26
- from ...models.resource_estimation_response import ResourceEstimationResponse
27
- from ...models.upload_csv_payload import UploadCsvPayload
28
- from ...types import Response
29
-
30
-
31
- def _get_kwargs(
32
- *,
33
- body: ConcatenateRequest
34
- | CreateGroupRequest
35
- | CreateRequest
36
- | DedupeRequestParams
37
- | DeepMergeRequest
38
- | DeepRankRequest
39
- | DeepScreenRequest
40
- | DeriveRequest
41
- | DropColumnsRequest
42
- | FilterRequest
43
- | FlattenRequest
44
- | GroupByRequest
45
- | JoinRequest
46
- | MapAgentRequestParams
47
- | MapMultiAgentRequestParams
48
- | ReduceAgentRequestParams
49
- | ReduceMultiAgentRequestParams
50
- | UploadCsvPayload,
51
- ) -> dict[str, Any]:
52
- headers: dict[str, Any] = {}
53
-
54
- _kwargs: dict[str, Any] = {
55
- "method": "post",
56
- "url": "/task-resource-estimation",
57
- }
58
-
59
- if isinstance(body, MapAgentRequestParams):
60
- _kwargs["json"] = body.to_dict()
61
- elif isinstance(body, ReduceAgentRequestParams):
62
- _kwargs["json"] = body.to_dict()
63
- elif isinstance(body, FilterRequest):
64
- _kwargs["json"] = body.to_dict()
65
- elif isinstance(body, DeriveRequest):
66
- _kwargs["json"] = body.to_dict()
67
- elif isinstance(body, JoinRequest):
68
- _kwargs["json"] = body.to_dict()
69
- elif isinstance(body, ConcatenateRequest):
70
- _kwargs["json"] = body.to_dict()
71
- elif isinstance(body, DropColumnsRequest):
72
- _kwargs["json"] = body.to_dict()
73
- elif isinstance(body, DedupeRequestParams):
74
- _kwargs["json"] = body.to_dict()
75
- elif isinstance(body, UploadCsvPayload):
76
- _kwargs["json"] = body.to_dict()
77
- elif isinstance(body, MapMultiAgentRequestParams):
78
- _kwargs["json"] = body.to_dict()
79
- elif isinstance(body, ReduceMultiAgentRequestParams):
80
- _kwargs["json"] = body.to_dict()
81
- elif isinstance(body, CreateRequest):
82
- _kwargs["json"] = body.to_dict()
83
- elif isinstance(body, CreateGroupRequest):
84
- _kwargs["json"] = body.to_dict()
85
- elif isinstance(body, FlattenRequest):
86
- _kwargs["json"] = body.to_dict()
87
- elif isinstance(body, GroupByRequest):
88
- _kwargs["json"] = body.to_dict()
89
- elif isinstance(body, DeepRankRequest):
90
- _kwargs["json"] = body.to_dict()
91
- elif isinstance(body, DeepMergeRequest):
92
- _kwargs["json"] = body.to_dict()
93
- else:
94
- _kwargs["json"] = body.to_dict()
95
-
96
- headers["Content-Type"] = "application/json"
97
-
98
- _kwargs["headers"] = headers
99
- return _kwargs
100
-
101
-
102
- def _parse_response(
103
- *, client: AuthenticatedClient | Client, response: httpx.Response
104
- ) -> HTTPValidationError | ResourceEstimationResponse | None:
105
- if response.status_code == 200:
106
- response_200 = ResourceEstimationResponse.from_dict(response.json())
107
-
108
- return response_200
109
-
110
- if response.status_code == 422:
111
- response_422 = HTTPValidationError.from_dict(response.json())
112
-
113
- return response_422
114
-
115
- if client.raise_on_unexpected_status:
116
- raise errors.UnexpectedStatus(response.status_code, response.content)
117
- else:
118
- return None
119
-
120
-
121
- def _build_response(
122
- *, client: AuthenticatedClient | Client, response: httpx.Response
123
- ) -> Response[HTTPValidationError | ResourceEstimationResponse]:
124
- return Response(
125
- status_code=HTTPStatus(response.status_code),
126
- content=response.content,
127
- headers=response.headers,
128
- parsed=_parse_response(client=client, response=response),
129
- )
130
-
131
-
132
- def sync_detailed(
133
- *,
134
- client: AuthenticatedClient,
135
- body: ConcatenateRequest
136
- | CreateGroupRequest
137
- | CreateRequest
138
- | DedupeRequestParams
139
- | DeepMergeRequest
140
- | DeepRankRequest
141
- | DeepScreenRequest
142
- | DeriveRequest
143
- | DropColumnsRequest
144
- | FilterRequest
145
- | FlattenRequest
146
- | GroupByRequest
147
- | JoinRequest
148
- | MapAgentRequestParams
149
- | MapMultiAgentRequestParams
150
- | ReduceAgentRequestParams
151
- | ReduceMultiAgentRequestParams
152
- | UploadCsvPayload,
153
- ) -> Response[HTTPValidationError | ResourceEstimationResponse]:
154
- """Task Resource Estimation
155
-
156
- Args:
157
- body (ConcatenateRequest | CreateGroupRequest | CreateRequest | DedupeRequestParams |
158
- DeepMergeRequest | DeepRankRequest | DeepScreenRequest | DeriveRequest |
159
- DropColumnsRequest | FilterRequest | FlattenRequest | GroupByRequest | JoinRequest |
160
- MapAgentRequestParams | MapMultiAgentRequestParams | ReduceAgentRequestParams |
161
- ReduceMultiAgentRequestParams | UploadCsvPayload):
162
-
163
- Raises:
164
- errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
165
- httpx.TimeoutException: If the request takes longer than Client.timeout.
166
-
167
- Returns:
168
- Response[HTTPValidationError | ResourceEstimationResponse]
169
- """
170
-
171
- kwargs = _get_kwargs(
172
- body=body,
173
- )
174
-
175
- response = client.get_httpx_client().request(
176
- **kwargs,
177
- )
178
-
179
- return _build_response(client=client, response=response)
180
-
181
-
182
- def sync(
183
- *,
184
- client: AuthenticatedClient,
185
- body: ConcatenateRequest
186
- | CreateGroupRequest
187
- | CreateRequest
188
- | DedupeRequestParams
189
- | DeepMergeRequest
190
- | DeepRankRequest
191
- | DeepScreenRequest
192
- | DeriveRequest
193
- | DropColumnsRequest
194
- | FilterRequest
195
- | FlattenRequest
196
- | GroupByRequest
197
- | JoinRequest
198
- | MapAgentRequestParams
199
- | MapMultiAgentRequestParams
200
- | ReduceAgentRequestParams
201
- | ReduceMultiAgentRequestParams
202
- | UploadCsvPayload,
203
- ) -> HTTPValidationError | ResourceEstimationResponse | None:
204
- """Task Resource Estimation
205
-
206
- Args:
207
- body (ConcatenateRequest | CreateGroupRequest | CreateRequest | DedupeRequestParams |
208
- DeepMergeRequest | DeepRankRequest | DeepScreenRequest | DeriveRequest |
209
- DropColumnsRequest | FilterRequest | FlattenRequest | GroupByRequest | JoinRequest |
210
- MapAgentRequestParams | MapMultiAgentRequestParams | ReduceAgentRequestParams |
211
- ReduceMultiAgentRequestParams | UploadCsvPayload):
212
-
213
- Raises:
214
- errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
215
- httpx.TimeoutException: If the request takes longer than Client.timeout.
216
-
217
- Returns:
218
- HTTPValidationError | ResourceEstimationResponse
219
- """
220
-
221
- return sync_detailed(
222
- client=client,
223
- body=body,
224
- ).parsed
225
-
226
-
227
- async def asyncio_detailed(
228
- *,
229
- client: AuthenticatedClient,
230
- body: ConcatenateRequest
231
- | CreateGroupRequest
232
- | CreateRequest
233
- | DedupeRequestParams
234
- | DeepMergeRequest
235
- | DeepRankRequest
236
- | DeepScreenRequest
237
- | DeriveRequest
238
- | DropColumnsRequest
239
- | FilterRequest
240
- | FlattenRequest
241
- | GroupByRequest
242
- | JoinRequest
243
- | MapAgentRequestParams
244
- | MapMultiAgentRequestParams
245
- | ReduceAgentRequestParams
246
- | ReduceMultiAgentRequestParams
247
- | UploadCsvPayload,
248
- ) -> Response[HTTPValidationError | ResourceEstimationResponse]:
249
- """Task Resource Estimation
250
-
251
- Args:
252
- body (ConcatenateRequest | CreateGroupRequest | CreateRequest | DedupeRequestParams |
253
- DeepMergeRequest | DeepRankRequest | DeepScreenRequest | DeriveRequest |
254
- DropColumnsRequest | FilterRequest | FlattenRequest | GroupByRequest | JoinRequest |
255
- MapAgentRequestParams | MapMultiAgentRequestParams | ReduceAgentRequestParams |
256
- ReduceMultiAgentRequestParams | UploadCsvPayload):
257
-
258
- Raises:
259
- errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
260
- httpx.TimeoutException: If the request takes longer than Client.timeout.
261
-
262
- Returns:
263
- Response[HTTPValidationError | ResourceEstimationResponse]
264
- """
265
-
266
- kwargs = _get_kwargs(
267
- body=body,
268
- )
269
-
270
- response = await client.get_async_httpx_client().request(**kwargs)
271
-
272
- return _build_response(client=client, response=response)
273
-
274
-
275
- async def asyncio(
276
- *,
277
- client: AuthenticatedClient,
278
- body: ConcatenateRequest
279
- | CreateGroupRequest
280
- | CreateRequest
281
- | DedupeRequestParams
282
- | DeepMergeRequest
283
- | DeepRankRequest
284
- | DeepScreenRequest
285
- | DeriveRequest
286
- | DropColumnsRequest
287
- | FilterRequest
288
- | FlattenRequest
289
- | GroupByRequest
290
- | JoinRequest
291
- | MapAgentRequestParams
292
- | MapMultiAgentRequestParams
293
- | ReduceAgentRequestParams
294
- | ReduceMultiAgentRequestParams
295
- | UploadCsvPayload,
296
- ) -> HTTPValidationError | ResourceEstimationResponse | None:
297
- """Task Resource Estimation
298
-
299
- Args:
300
- body (ConcatenateRequest | CreateGroupRequest | CreateRequest | DedupeRequestParams |
301
- DeepMergeRequest | DeepRankRequest | DeepScreenRequest | DeriveRequest |
302
- DropColumnsRequest | FilterRequest | FlattenRequest | GroupByRequest | JoinRequest |
303
- MapAgentRequestParams | MapMultiAgentRequestParams | ReduceAgentRequestParams |
304
- ReduceMultiAgentRequestParams | UploadCsvPayload):
305
-
306
- Raises:
307
- errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
308
- httpx.TimeoutException: If the request takes longer than Client.timeout.
309
-
310
- Returns:
311
- HTTPValidationError | ResourceEstimationResponse
312
- """
313
-
314
- return (
315
- await asyncio_detailed(
316
- client=client,
317
- body=body,
318
- )
319
- ).parsed
@@ -1,166 +0,0 @@
1
- from http import HTTPStatus
2
- from typing import Any
3
-
4
- import httpx
5
-
6
- from ... import errors
7
- from ...client import AuthenticatedClient, Client
8
- from ...models.http_validation_error import HTTPValidationError
9
- from ...models.trigger_workflow_execution_request import TriggerWorkflowExecutionRequest
10
- from ...models.trigger_workflow_execution_response import TriggerWorkflowExecutionResponse
11
- from ...types import Response
12
-
13
-
14
- def _get_kwargs(
15
- *,
16
- body: TriggerWorkflowExecutionRequest,
17
- ) -> dict[str, Any]:
18
- headers: dict[str, Any] = {}
19
-
20
- _kwargs: dict[str, Any] = {
21
- "method": "post",
22
- "url": "/workflows/trigger",
23
- }
24
-
25
- _kwargs["json"] = body.to_dict()
26
-
27
- headers["Content-Type"] = "application/json"
28
-
29
- _kwargs["headers"] = headers
30
- return _kwargs
31
-
32
-
33
- def _parse_response(
34
- *, client: AuthenticatedClient | Client, response: httpx.Response
35
- ) -> HTTPValidationError | TriggerWorkflowExecutionResponse | None:
36
- if response.status_code == 200:
37
- response_200 = TriggerWorkflowExecutionResponse.from_dict(response.json())
38
-
39
- return response_200
40
-
41
- if response.status_code == 422:
42
- response_422 = HTTPValidationError.from_dict(response.json())
43
-
44
- return response_422
45
-
46
- if client.raise_on_unexpected_status:
47
- raise errors.UnexpectedStatus(response.status_code, response.content)
48
- else:
49
- return None
50
-
51
-
52
- def _build_response(
53
- *, client: AuthenticatedClient | Client, response: httpx.Response
54
- ) -> Response[HTTPValidationError | TriggerWorkflowExecutionResponse]:
55
- return Response(
56
- status_code=HTTPStatus(response.status_code),
57
- content=response.content,
58
- headers=response.headers,
59
- parsed=_parse_response(client=client, response=response),
60
- )
61
-
62
-
63
- def sync_detailed(
64
- *,
65
- client: AuthenticatedClient,
66
- body: TriggerWorkflowExecutionRequest,
67
- ) -> Response[HTTPValidationError | TriggerWorkflowExecutionResponse]:
68
- """Trigger Workflow Execution Endpoint
69
-
70
- Args:
71
- body (TriggerWorkflowExecutionRequest):
72
-
73
- Raises:
74
- errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
75
- httpx.TimeoutException: If the request takes longer than Client.timeout.
76
-
77
- Returns:
78
- Response[HTTPValidationError | TriggerWorkflowExecutionResponse]
79
- """
80
-
81
- kwargs = _get_kwargs(
82
- body=body,
83
- )
84
-
85
- response = client.get_httpx_client().request(
86
- **kwargs,
87
- )
88
-
89
- return _build_response(client=client, response=response)
90
-
91
-
92
- def sync(
93
- *,
94
- client: AuthenticatedClient,
95
- body: TriggerWorkflowExecutionRequest,
96
- ) -> HTTPValidationError | TriggerWorkflowExecutionResponse | None:
97
- """Trigger Workflow Execution Endpoint
98
-
99
- Args:
100
- body (TriggerWorkflowExecutionRequest):
101
-
102
- Raises:
103
- errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
104
- httpx.TimeoutException: If the request takes longer than Client.timeout.
105
-
106
- Returns:
107
- HTTPValidationError | TriggerWorkflowExecutionResponse
108
- """
109
-
110
- return sync_detailed(
111
- client=client,
112
- body=body,
113
- ).parsed
114
-
115
-
116
- async def asyncio_detailed(
117
- *,
118
- client: AuthenticatedClient,
119
- body: TriggerWorkflowExecutionRequest,
120
- ) -> Response[HTTPValidationError | TriggerWorkflowExecutionResponse]:
121
- """Trigger Workflow Execution Endpoint
122
-
123
- Args:
124
- body (TriggerWorkflowExecutionRequest):
125
-
126
- Raises:
127
- errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
128
- httpx.TimeoutException: If the request takes longer than Client.timeout.
129
-
130
- Returns:
131
- Response[HTTPValidationError | TriggerWorkflowExecutionResponse]
132
- """
133
-
134
- kwargs = _get_kwargs(
135
- body=body,
136
- )
137
-
138
- response = await client.get_async_httpx_client().request(**kwargs)
139
-
140
- return _build_response(client=client, response=response)
141
-
142
-
143
- async def asyncio(
144
- *,
145
- client: AuthenticatedClient,
146
- body: TriggerWorkflowExecutionRequest,
147
- ) -> HTTPValidationError | TriggerWorkflowExecutionResponse | None:
148
- """Trigger Workflow Execution Endpoint
149
-
150
- Args:
151
- body (TriggerWorkflowExecutionRequest):
152
-
153
- Raises:
154
- errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
155
- httpx.TimeoutException: If the request takes longer than Client.timeout.
156
-
157
- Returns:
158
- HTTPValidationError | TriggerWorkflowExecutionResponse
159
- """
160
-
161
- return (
162
- await asyncio_detailed(
163
- client=client,
164
- body=body,
165
- )
166
- ).parsed
@@ -1,127 +0,0 @@
1
- from http import HTTPStatus
2
- from typing import Any
3
-
4
- import httpx
5
-
6
- from ... import errors
7
- from ...client import AuthenticatedClient, Client
8
- from ...models.whoami_whoami_get_response_whoami_whoami_get import WhoamiWhoamiGetResponseWhoamiWhoamiGet
9
- from ...types import Response
10
-
11
-
12
- def _get_kwargs() -> dict[str, Any]:
13
- _kwargs: dict[str, Any] = {
14
- "method": "get",
15
- "url": "/whoami",
16
- }
17
-
18
- return _kwargs
19
-
20
-
21
- def _parse_response(
22
- *, client: AuthenticatedClient | Client, response: httpx.Response
23
- ) -> WhoamiWhoamiGetResponseWhoamiWhoamiGet | None:
24
- if response.status_code == 200:
25
- response_200 = WhoamiWhoamiGetResponseWhoamiWhoamiGet.from_dict(response.json())
26
-
27
- return response_200
28
-
29
- if client.raise_on_unexpected_status:
30
- raise errors.UnexpectedStatus(response.status_code, response.content)
31
- else:
32
- return None
33
-
34
-
35
- def _build_response(
36
- *, client: AuthenticatedClient | Client, response: httpx.Response
37
- ) -> Response[WhoamiWhoamiGetResponseWhoamiWhoamiGet]:
38
- return Response(
39
- status_code=HTTPStatus(response.status_code),
40
- content=response.content,
41
- headers=response.headers,
42
- parsed=_parse_response(client=client, response=response),
43
- )
44
-
45
-
46
- def sync_detailed(
47
- *,
48
- client: AuthenticatedClient,
49
- ) -> Response[WhoamiWhoamiGetResponseWhoamiWhoamiGet]:
50
- """Whoami
51
-
52
- Raises:
53
- errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
54
- httpx.TimeoutException: If the request takes longer than Client.timeout.
55
-
56
- Returns:
57
- Response[WhoamiWhoamiGetResponseWhoamiWhoamiGet]
58
- """
59
-
60
- kwargs = _get_kwargs()
61
-
62
- response = client.get_httpx_client().request(
63
- **kwargs,
64
- )
65
-
66
- return _build_response(client=client, response=response)
67
-
68
-
69
- def sync(
70
- *,
71
- client: AuthenticatedClient,
72
- ) -> WhoamiWhoamiGetResponseWhoamiWhoamiGet | None:
73
- """Whoami
74
-
75
- Raises:
76
- errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
77
- httpx.TimeoutException: If the request takes longer than Client.timeout.
78
-
79
- Returns:
80
- WhoamiWhoamiGetResponseWhoamiWhoamiGet
81
- """
82
-
83
- return sync_detailed(
84
- client=client,
85
- ).parsed
86
-
87
-
88
- async def asyncio_detailed(
89
- *,
90
- client: AuthenticatedClient,
91
- ) -> Response[WhoamiWhoamiGetResponseWhoamiWhoamiGet]:
92
- """Whoami
93
-
94
- Raises:
95
- errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
96
- httpx.TimeoutException: If the request takes longer than Client.timeout.
97
-
98
- Returns:
99
- Response[WhoamiWhoamiGetResponseWhoamiWhoamiGet]
100
- """
101
-
102
- kwargs = _get_kwargs()
103
-
104
- response = await client.get_async_httpx_client().request(**kwargs)
105
-
106
- return _build_response(client=client, response=response)
107
-
108
-
109
- async def asyncio(
110
- *,
111
- client: AuthenticatedClient,
112
- ) -> WhoamiWhoamiGetResponseWhoamiWhoamiGet | None:
113
- """Whoami
114
-
115
- Raises:
116
- errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
117
- httpx.TimeoutException: If the request takes longer than Client.timeout.
118
-
119
- Returns:
120
- WhoamiWhoamiGetResponseWhoamiWhoamiGet
121
- """
122
-
123
- return (
124
- await asyncio_detailed(
125
- client=client,
126
- )
127
- ).parsed
@@ -1,69 +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="AgentImprovementInstruction")
10
-
11
-
12
- @_attrs_define
13
- class AgentImprovementInstruction:
14
- """
15
- Attributes:
16
- agent_id (str): Target agent ID
17
- feedback (str): Additional work to increase the quality of the agent's solution
18
- """
19
-
20
- agent_id: str
21
- feedback: str
22
- additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
23
-
24
- def to_dict(self) -> dict[str, Any]:
25
- agent_id = self.agent_id
26
-
27
- feedback = self.feedback
28
-
29
- field_dict: dict[str, Any] = {}
30
- field_dict.update(self.additional_properties)
31
- field_dict.update(
32
- {
33
- "agent_id": agent_id,
34
- "feedback": feedback,
35
- }
36
- )
37
-
38
- return field_dict
39
-
40
- @classmethod
41
- def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
42
- d = dict(src_dict)
43
- agent_id = d.pop("agent_id")
44
-
45
- feedback = d.pop("feedback")
46
-
47
- agent_improvement_instruction = cls(
48
- agent_id=agent_id,
49
- feedback=feedback,
50
- )
51
-
52
- agent_improvement_instruction.additional_properties = d
53
- return agent_improvement_instruction
54
-
55
- @property
56
- def additional_keys(self) -> list[str]:
57
- return list(self.additional_properties.keys())
58
-
59
- def __getitem__(self, key: str) -> Any:
60
- return self.additional_properties[key]
61
-
62
- def __setitem__(self, key: str, value: Any) -> None:
63
- self.additional_properties[key] = value
64
-
65
- def __delitem__(self, key: str) -> None:
66
- del self.additional_properties[key]
67
-
68
- def __contains__(self, key: str) -> bool:
69
- return key in self.additional_properties