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,181 +0,0 @@
1
- from http import HTTPStatus
2
- from typing import Any
3
- from urllib.parse import quote
4
- from uuid import UUID
5
-
6
- import httpx
7
-
8
- from ... import errors
9
- from ...client import AuthenticatedClient, Client
10
- from ...models.http_validation_error import HTTPValidationError
11
- from ...models.revoke_api_key_response import RevokeAPIKeyResponse
12
- from ...types import Response
13
-
14
-
15
- def _get_kwargs(
16
- key_id: UUID,
17
- ) -> dict[str, Any]:
18
- _kwargs: dict[str, Any] = {
19
- "method": "post",
20
- "url": "/api-keys/{key_id}/revoke".format(
21
- key_id=quote(str(key_id), safe=""),
22
- ),
23
- }
24
-
25
- return _kwargs
26
-
27
-
28
- def _parse_response(
29
- *, client: AuthenticatedClient | Client, response: httpx.Response
30
- ) -> HTTPValidationError | RevokeAPIKeyResponse | None:
31
- if response.status_code == 200:
32
- response_200 = RevokeAPIKeyResponse.from_dict(response.json())
33
-
34
- return response_200
35
-
36
- if response.status_code == 422:
37
- response_422 = HTTPValidationError.from_dict(response.json())
38
-
39
- return response_422
40
-
41
- if client.raise_on_unexpected_status:
42
- raise errors.UnexpectedStatus(response.status_code, response.content)
43
- else:
44
- return None
45
-
46
-
47
- def _build_response(
48
- *, client: AuthenticatedClient | Client, response: httpx.Response
49
- ) -> Response[HTTPValidationError | RevokeAPIKeyResponse]:
50
- return Response(
51
- status_code=HTTPStatus(response.status_code),
52
- content=response.content,
53
- headers=response.headers,
54
- parsed=_parse_response(client=client, response=response),
55
- )
56
-
57
-
58
- def sync_detailed(
59
- key_id: UUID,
60
- *,
61
- client: AuthenticatedClient,
62
- ) -> Response[HTTPValidationError | RevokeAPIKeyResponse]:
63
- """Revoke Api Key Endpoint
64
-
65
- Revoke an API key (soft delete).
66
-
67
- The key will no longer be usable for authentication, but the record
68
- remains in the database for audit purposes.
69
-
70
- Args:
71
- key_id (UUID):
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 | RevokeAPIKeyResponse]
79
- """
80
-
81
- kwargs = _get_kwargs(
82
- key_id=key_id,
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
- key_id: UUID,
94
- *,
95
- client: AuthenticatedClient,
96
- ) -> HTTPValidationError | RevokeAPIKeyResponse | None:
97
- """Revoke Api Key Endpoint
98
-
99
- Revoke an API key (soft delete).
100
-
101
- The key will no longer be usable for authentication, but the record
102
- remains in the database for audit purposes.
103
-
104
- Args:
105
- key_id (UUID):
106
-
107
- Raises:
108
- errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
109
- httpx.TimeoutException: If the request takes longer than Client.timeout.
110
-
111
- Returns:
112
- HTTPValidationError | RevokeAPIKeyResponse
113
- """
114
-
115
- return sync_detailed(
116
- key_id=key_id,
117
- client=client,
118
- ).parsed
119
-
120
-
121
- async def asyncio_detailed(
122
- key_id: UUID,
123
- *,
124
- client: AuthenticatedClient,
125
- ) -> Response[HTTPValidationError | RevokeAPIKeyResponse]:
126
- """Revoke Api Key Endpoint
127
-
128
- Revoke an API key (soft delete).
129
-
130
- The key will no longer be usable for authentication, but the record
131
- remains in the database for audit purposes.
132
-
133
- Args:
134
- key_id (UUID):
135
-
136
- Raises:
137
- errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
138
- httpx.TimeoutException: If the request takes longer than Client.timeout.
139
-
140
- Returns:
141
- Response[HTTPValidationError | RevokeAPIKeyResponse]
142
- """
143
-
144
- kwargs = _get_kwargs(
145
- key_id=key_id,
146
- )
147
-
148
- response = await client.get_async_httpx_client().request(**kwargs)
149
-
150
- return _build_response(client=client, response=response)
151
-
152
-
153
- async def asyncio(
154
- key_id: UUID,
155
- *,
156
- client: AuthenticatedClient,
157
- ) -> HTTPValidationError | RevokeAPIKeyResponse | None:
158
- """Revoke Api Key Endpoint
159
-
160
- Revoke an API key (soft delete).
161
-
162
- The key will no longer be usable for authentication, but the record
163
- remains in the database for audit purposes.
164
-
165
- Args:
166
- key_id (UUID):
167
-
168
- Raises:
169
- errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
170
- httpx.TimeoutException: If the request takes longer than Client.timeout.
171
-
172
- Returns:
173
- HTTPValidationError | RevokeAPIKeyResponse
174
- """
175
-
176
- return (
177
- await asyncio_detailed(
178
- key_id=key_id,
179
- client=client,
180
- )
181
- ).parsed
@@ -1,164 +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.task_id_request import TaskIdRequest
10
- from ...types import Response
11
-
12
-
13
- def _get_kwargs(
14
- *,
15
- body: TaskIdRequest,
16
- ) -> dict[str, Any]:
17
- headers: dict[str, Any] = {}
18
-
19
- _kwargs: dict[str, Any] = {
20
- "method": "post",
21
- "url": "/jobs/revoke",
22
- }
23
-
24
- _kwargs["json"] = body.to_dict()
25
-
26
- headers["Content-Type"] = "application/json"
27
-
28
- _kwargs["headers"] = headers
29
- return _kwargs
30
-
31
-
32
- def _parse_response(
33
- *, client: AuthenticatedClient | Client, response: httpx.Response
34
- ) -> Any | HTTPValidationError | None:
35
- if response.status_code == 200:
36
- response_200 = response.json()
37
- return response_200
38
-
39
- if response.status_code == 422:
40
- response_422 = HTTPValidationError.from_dict(response.json())
41
-
42
- return response_422
43
-
44
- if client.raise_on_unexpected_status:
45
- raise errors.UnexpectedStatus(response.status_code, response.content)
46
- else:
47
- return None
48
-
49
-
50
- def _build_response(
51
- *, client: AuthenticatedClient | Client, response: httpx.Response
52
- ) -> Response[Any | HTTPValidationError]:
53
- return Response(
54
- status_code=HTTPStatus(response.status_code),
55
- content=response.content,
56
- headers=response.headers,
57
- parsed=_parse_response(client=client, response=response),
58
- )
59
-
60
-
61
- def sync_detailed(
62
- *,
63
- client: AuthenticatedClient,
64
- body: TaskIdRequest,
65
- ) -> Response[Any | HTTPValidationError]:
66
- """Revoke Jobs For Task
67
-
68
- Args:
69
- body (TaskIdRequest):
70
-
71
- Raises:
72
- errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
73
- httpx.TimeoutException: If the request takes longer than Client.timeout.
74
-
75
- Returns:
76
- Response[Any | HTTPValidationError]
77
- """
78
-
79
- kwargs = _get_kwargs(
80
- body=body,
81
- )
82
-
83
- response = client.get_httpx_client().request(
84
- **kwargs,
85
- )
86
-
87
- return _build_response(client=client, response=response)
88
-
89
-
90
- def sync(
91
- *,
92
- client: AuthenticatedClient,
93
- body: TaskIdRequest,
94
- ) -> Any | HTTPValidationError | None:
95
- """Revoke Jobs For Task
96
-
97
- Args:
98
- body (TaskIdRequest):
99
-
100
- Raises:
101
- errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
102
- httpx.TimeoutException: If the request takes longer than Client.timeout.
103
-
104
- Returns:
105
- Any | HTTPValidationError
106
- """
107
-
108
- return sync_detailed(
109
- client=client,
110
- body=body,
111
- ).parsed
112
-
113
-
114
- async def asyncio_detailed(
115
- *,
116
- client: AuthenticatedClient,
117
- body: TaskIdRequest,
118
- ) -> Response[Any | HTTPValidationError]:
119
- """Revoke Jobs For Task
120
-
121
- Args:
122
- body (TaskIdRequest):
123
-
124
- Raises:
125
- errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
126
- httpx.TimeoutException: If the request takes longer than Client.timeout.
127
-
128
- Returns:
129
- Response[Any | HTTPValidationError]
130
- """
131
-
132
- kwargs = _get_kwargs(
133
- body=body,
134
- )
135
-
136
- response = await client.get_async_httpx_client().request(**kwargs)
137
-
138
- return _build_response(client=client, response=response)
139
-
140
-
141
- async def asyncio(
142
- *,
143
- client: AuthenticatedClient,
144
- body: TaskIdRequest,
145
- ) -> Any | HTTPValidationError | None:
146
- """Revoke Jobs For Task
147
-
148
- Args:
149
- body (TaskIdRequest):
150
-
151
- Raises:
152
- errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
153
- httpx.TimeoutException: If the request takes longer than Client.timeout.
154
-
155
- Returns:
156
- Any | HTTPValidationError
157
- """
158
-
159
- return (
160
- await asyncio_detailed(
161
- client=client,
162
- body=body,
163
- )
164
- ).parsed
@@ -1,186 +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.rollback_to_message_request import RollbackToMessageRequest
10
- from ...models.rollback_to_message_response import RollbackToMessageResponse
11
- from ...types import Response
12
-
13
-
14
- def _get_kwargs(
15
- *,
16
- body: RollbackToMessageRequest,
17
- ) -> dict[str, Any]:
18
- headers: dict[str, Any] = {}
19
-
20
- _kwargs: dict[str, Any] = {
21
- "method": "post",
22
- "url": "/tasks/chat/rollback",
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 | RollbackToMessageResponse | None:
36
- if response.status_code == 200:
37
- response_200 = RollbackToMessageResponse.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 | RollbackToMessageResponse]:
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: RollbackToMessageRequest,
67
- ) -> Response[HTTPValidationError | RollbackToMessageResponse]:
68
- """Rollback To Message Endpoint
69
-
70
- Delete a given message and all chat messages, tasks, and artifacts created at or after that message.
71
-
72
- ⚠️ DANGEROUS: This endpoint permanently deletes data and requires
73
- ENGINE_ENABLE_DANGEROUS_DEBUG_FEATURES=true.
74
-
75
- Args:
76
- body (RollbackToMessageRequest):
77
-
78
- Raises:
79
- errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
80
- httpx.TimeoutException: If the request takes longer than Client.timeout.
81
-
82
- Returns:
83
- Response[HTTPValidationError | RollbackToMessageResponse]
84
- """
85
-
86
- kwargs = _get_kwargs(
87
- body=body,
88
- )
89
-
90
- response = client.get_httpx_client().request(
91
- **kwargs,
92
- )
93
-
94
- return _build_response(client=client, response=response)
95
-
96
-
97
- def sync(
98
- *,
99
- client: AuthenticatedClient,
100
- body: RollbackToMessageRequest,
101
- ) -> HTTPValidationError | RollbackToMessageResponse | None:
102
- """Rollback To Message Endpoint
103
-
104
- Delete a given message and all chat messages, tasks, and artifacts created at or after that message.
105
-
106
- ⚠️ DANGEROUS: This endpoint permanently deletes data and requires
107
- ENGINE_ENABLE_DANGEROUS_DEBUG_FEATURES=true.
108
-
109
- Args:
110
- body (RollbackToMessageRequest):
111
-
112
- Raises:
113
- errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
114
- httpx.TimeoutException: If the request takes longer than Client.timeout.
115
-
116
- Returns:
117
- HTTPValidationError | RollbackToMessageResponse
118
- """
119
-
120
- return sync_detailed(
121
- client=client,
122
- body=body,
123
- ).parsed
124
-
125
-
126
- async def asyncio_detailed(
127
- *,
128
- client: AuthenticatedClient,
129
- body: RollbackToMessageRequest,
130
- ) -> Response[HTTPValidationError | RollbackToMessageResponse]:
131
- """Rollback To Message Endpoint
132
-
133
- Delete a given message and all chat messages, tasks, and artifacts created at or after that message.
134
-
135
- ⚠️ DANGEROUS: This endpoint permanently deletes data and requires
136
- ENGINE_ENABLE_DANGEROUS_DEBUG_FEATURES=true.
137
-
138
- Args:
139
- body (RollbackToMessageRequest):
140
-
141
- Raises:
142
- errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
143
- httpx.TimeoutException: If the request takes longer than Client.timeout.
144
-
145
- Returns:
146
- Response[HTTPValidationError | RollbackToMessageResponse]
147
- """
148
-
149
- kwargs = _get_kwargs(
150
- body=body,
151
- )
152
-
153
- response = await client.get_async_httpx_client().request(**kwargs)
154
-
155
- return _build_response(client=client, response=response)
156
-
157
-
158
- async def asyncio(
159
- *,
160
- client: AuthenticatedClient,
161
- body: RollbackToMessageRequest,
162
- ) -> HTTPValidationError | RollbackToMessageResponse | None:
163
- """Rollback To Message Endpoint
164
-
165
- Delete a given message and all chat messages, tasks, and artifacts created at or after that message.
166
-
167
- ⚠️ DANGEROUS: This endpoint permanently deletes data and requires
168
- ENGINE_ENABLE_DANGEROUS_DEBUG_FEATURES=true.
169
-
170
- Args:
171
- body (RollbackToMessageRequest):
172
-
173
- Raises:
174
- errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
175
- httpx.TimeoutException: If the request takes longer than Client.timeout.
176
-
177
- Returns:
178
- HTTPValidationError | RollbackToMessageResponse
179
- """
180
-
181
- return (
182
- await asyncio_detailed(
183
- client=client,
184
- body=body,
185
- )
186
- ).parsed
@@ -1,164 +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.submit_chat_task_body import SubmitChatTaskBody
10
- from ...types import Response
11
-
12
-
13
- def _get_kwargs(
14
- *,
15
- body: SubmitChatTaskBody,
16
- ) -> dict[str, Any]:
17
- headers: dict[str, Any] = {}
18
-
19
- _kwargs: dict[str, Any] = {
20
- "method": "post",
21
- "url": "/tasks/chat",
22
- }
23
-
24
- _kwargs["json"] = body.to_dict()
25
-
26
- headers["Content-Type"] = "application/json"
27
-
28
- _kwargs["headers"] = headers
29
- return _kwargs
30
-
31
-
32
- def _parse_response(
33
- *, client: AuthenticatedClient | Client, response: httpx.Response
34
- ) -> Any | HTTPValidationError | None:
35
- if response.status_code == 200:
36
- response_200 = response.json()
37
- return response_200
38
-
39
- if response.status_code == 422:
40
- response_422 = HTTPValidationError.from_dict(response.json())
41
-
42
- return response_422
43
-
44
- if client.raise_on_unexpected_status:
45
- raise errors.UnexpectedStatus(response.status_code, response.content)
46
- else:
47
- return None
48
-
49
-
50
- def _build_response(
51
- *, client: AuthenticatedClient | Client, response: httpx.Response
52
- ) -> Response[Any | HTTPValidationError]:
53
- return Response(
54
- status_code=HTTPStatus(response.status_code),
55
- content=response.content,
56
- headers=response.headers,
57
- parsed=_parse_response(client=client, response=response),
58
- )
59
-
60
-
61
- def sync_detailed(
62
- *,
63
- client: AuthenticatedClient,
64
- body: SubmitChatTaskBody,
65
- ) -> Response[Any | HTTPValidationError]:
66
- """Submit Chat Task
67
-
68
- Args:
69
- body (SubmitChatTaskBody):
70
-
71
- Raises:
72
- errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
73
- httpx.TimeoutException: If the request takes longer than Client.timeout.
74
-
75
- Returns:
76
- Response[Any | HTTPValidationError]
77
- """
78
-
79
- kwargs = _get_kwargs(
80
- body=body,
81
- )
82
-
83
- response = client.get_httpx_client().request(
84
- **kwargs,
85
- )
86
-
87
- return _build_response(client=client, response=response)
88
-
89
-
90
- def sync(
91
- *,
92
- client: AuthenticatedClient,
93
- body: SubmitChatTaskBody,
94
- ) -> Any | HTTPValidationError | None:
95
- """Submit Chat Task
96
-
97
- Args:
98
- body (SubmitChatTaskBody):
99
-
100
- Raises:
101
- errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
102
- httpx.TimeoutException: If the request takes longer than Client.timeout.
103
-
104
- Returns:
105
- Any | HTTPValidationError
106
- """
107
-
108
- return sync_detailed(
109
- client=client,
110
- body=body,
111
- ).parsed
112
-
113
-
114
- async def asyncio_detailed(
115
- *,
116
- client: AuthenticatedClient,
117
- body: SubmitChatTaskBody,
118
- ) -> Response[Any | HTTPValidationError]:
119
- """Submit Chat Task
120
-
121
- Args:
122
- body (SubmitChatTaskBody):
123
-
124
- Raises:
125
- errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
126
- httpx.TimeoutException: If the request takes longer than Client.timeout.
127
-
128
- Returns:
129
- Response[Any | HTTPValidationError]
130
- """
131
-
132
- kwargs = _get_kwargs(
133
- body=body,
134
- )
135
-
136
- response = await client.get_async_httpx_client().request(**kwargs)
137
-
138
- return _build_response(client=client, response=response)
139
-
140
-
141
- async def asyncio(
142
- *,
143
- client: AuthenticatedClient,
144
- body: SubmitChatTaskBody,
145
- ) -> Any | HTTPValidationError | None:
146
- """Submit Chat Task
147
-
148
- Args:
149
- body (SubmitChatTaskBody):
150
-
151
- Raises:
152
- errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
153
- httpx.TimeoutException: If the request takes longer than Client.timeout.
154
-
155
- Returns:
156
- Any | HTTPValidationError
157
- """
158
-
159
- return (
160
- await asyncio_detailed(
161
- client=client,
162
- body=body,
163
- )
164
- ).parsed