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,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.controller_improvement_round import ControllerImprovementRound
9
- from ...models.generate_feedback_request import GenerateFeedbackRequest
10
- from ...models.http_validation_error import HTTPValidationError
11
- from ...types import Response
12
-
13
-
14
- def _get_kwargs(
15
- *,
16
- body: GenerateFeedbackRequest,
17
- ) -> dict[str, Any]:
18
- headers: dict[str, Any] = {}
19
-
20
- _kwargs: dict[str, Any] = {
21
- "method": "post",
22
- "url": "/tasks/generate_feedback",
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
- ) -> ControllerImprovementRound | HTTPValidationError | None:
36
- if response.status_code == 200:
37
- response_200 = ControllerImprovementRound.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[ControllerImprovementRound | HTTPValidationError]:
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: GenerateFeedbackRequest,
67
- ) -> Response[ControllerImprovementRound | HTTPValidationError]:
68
- """Generate Feedback Endpoint
69
-
70
- Generate controller feedback for a completed MAP task.
71
-
72
- This endpoint invokes a Celery task that calls generate_feedback_for_map_task
73
- with the provided task_id and default values for other parameters.
74
-
75
- Args:
76
- body (GenerateFeedbackRequest):
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[ControllerImprovementRound | HTTPValidationError]
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: GenerateFeedbackRequest,
101
- ) -> ControllerImprovementRound | HTTPValidationError | None:
102
- """Generate Feedback Endpoint
103
-
104
- Generate controller feedback for a completed MAP task.
105
-
106
- This endpoint invokes a Celery task that calls generate_feedback_for_map_task
107
- with the provided task_id and default values for other parameters.
108
-
109
- Args:
110
- body (GenerateFeedbackRequest):
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
- ControllerImprovementRound | HTTPValidationError
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: GenerateFeedbackRequest,
130
- ) -> Response[ControllerImprovementRound | HTTPValidationError]:
131
- """Generate Feedback Endpoint
132
-
133
- Generate controller feedback for a completed MAP task.
134
-
135
- This endpoint invokes a Celery task that calls generate_feedback_for_map_task
136
- with the provided task_id and default values for other parameters.
137
-
138
- Args:
139
- body (GenerateFeedbackRequest):
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[ControllerImprovementRound | HTTPValidationError]
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: GenerateFeedbackRequest,
162
- ) -> ControllerImprovementRound | HTTPValidationError | None:
163
- """Generate Feedback Endpoint
164
-
165
- Generate controller feedback for a completed MAP task.
166
-
167
- This endpoint invokes a Celery task that calls generate_feedback_for_map_task
168
- with the provided task_id and default values for other parameters.
169
-
170
- Args:
171
- body (GenerateFeedbackRequest):
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
- ControllerImprovementRound | HTTPValidationError
179
- """
180
-
181
- return (
182
- await asyncio_detailed(
183
- client=client,
184
- body=body,
185
- )
186
- ).parsed
@@ -1,260 +0,0 @@
1
- from http import HTTPStatus
2
- from typing import Any
3
- from uuid import UUID
4
-
5
- import httpx
6
-
7
- from ... import errors
8
- from ...client import AuthenticatedClient, Client
9
- from ...models.artifact_group_record import ArtifactGroupRecord
10
- from ...models.http_validation_error import HTTPValidationError
11
- from ...models.standalone_artifact_record import StandaloneArtifactRecord
12
- from ...types import UNSET, Response, Unset
13
-
14
-
15
- def _get_kwargs(
16
- *,
17
- artifact_ids: list[UUID],
18
- include_failed: bool | Unset = False,
19
- ) -> dict[str, Any]:
20
- params: dict[str, Any] = {}
21
-
22
- json_artifact_ids = []
23
- for artifact_ids_item_data in artifact_ids:
24
- artifact_ids_item = str(artifact_ids_item_data)
25
- json_artifact_ids.append(artifact_ids_item)
26
-
27
- params["artifact_ids"] = json_artifact_ids
28
-
29
- params["include_failed"] = include_failed
30
-
31
- params = {k: v for k, v in params.items() if v is not UNSET and v is not None}
32
-
33
- _kwargs: dict[str, Any] = {
34
- "method": "get",
35
- "url": "/artifacts",
36
- "params": params,
37
- }
38
-
39
- return _kwargs
40
-
41
-
42
- def _parse_response(
43
- *, client: AuthenticatedClient | Client, response: httpx.Response
44
- ) -> HTTPValidationError | list[ArtifactGroupRecord | StandaloneArtifactRecord] | None:
45
- if response.status_code == 200:
46
- response_200 = []
47
- _response_200 = response.json()
48
- for response_200_item_data in _response_200:
49
-
50
- def _parse_response_200_item(data: object) -> ArtifactGroupRecord | StandaloneArtifactRecord:
51
- try:
52
- if not isinstance(data, dict):
53
- raise TypeError()
54
- response_200_item_type_0 = StandaloneArtifactRecord.from_dict(data)
55
-
56
- return response_200_item_type_0
57
- except (TypeError, ValueError, AttributeError, KeyError):
58
- pass
59
- if not isinstance(data, dict):
60
- raise TypeError()
61
- response_200_item_type_1 = ArtifactGroupRecord.from_dict(data)
62
-
63
- return response_200_item_type_1
64
-
65
- response_200_item = _parse_response_200_item(response_200_item_data)
66
-
67
- response_200.append(response_200_item)
68
-
69
- return response_200
70
-
71
- if response.status_code == 422:
72
- response_422 = HTTPValidationError.from_dict(response.json())
73
-
74
- return response_422
75
-
76
- if client.raise_on_unexpected_status:
77
- raise errors.UnexpectedStatus(response.status_code, response.content)
78
- else:
79
- return None
80
-
81
-
82
- def _build_response(
83
- *, client: AuthenticatedClient | Client, response: httpx.Response
84
- ) -> Response[HTTPValidationError | list[ArtifactGroupRecord | StandaloneArtifactRecord]]:
85
- return Response(
86
- status_code=HTTPStatus(response.status_code),
87
- content=response.content,
88
- headers=response.headers,
89
- parsed=_parse_response(client=client, response=response),
90
- )
91
-
92
-
93
- def sync_detailed(
94
- *,
95
- client: AuthenticatedClient | Client,
96
- artifact_ids: list[UUID],
97
- include_failed: bool | Unset = False,
98
- ) -> Response[HTTPValidationError | list[ArtifactGroupRecord | StandaloneArtifactRecord]]:
99
- """Get Artifacts
100
-
101
- Get artifacts by artifact ID.
102
-
103
- Caution: This is not protected by RLS, so it will return the artifacts
104
- as long as the IDs are valid. The reason for it is that we would need to handle
105
- case for both authenticated and non-authenticated users similarly to how we handle
106
- stuff on the frontend, and that doesn't seem to be worth it at the moment.
107
-
108
- Alternative would be ditching this whole endpoint and replicating fetching the
109
- same artifacts in the frontend from the DB directly. It would be probably much
110
- faster and simpler anyway, with only slight duplicity of the code between python
111
- and typescript (as we need to be able to fetch artifacts here for the various
112
- workflow purposes).
113
-
114
- Args:
115
- artifact_ids (list[UUID]):
116
- include_failed (bool | Unset): Default: False.
117
-
118
- Raises:
119
- errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
120
- httpx.TimeoutException: If the request takes longer than Client.timeout.
121
-
122
- Returns:
123
- Response[HTTPValidationError | list[ArtifactGroupRecord | StandaloneArtifactRecord]]
124
- """
125
-
126
- kwargs = _get_kwargs(
127
- artifact_ids=artifact_ids,
128
- include_failed=include_failed,
129
- )
130
-
131
- response = client.get_httpx_client().request(
132
- **kwargs,
133
- )
134
-
135
- return _build_response(client=client, response=response)
136
-
137
-
138
- def sync(
139
- *,
140
- client: AuthenticatedClient | Client,
141
- artifact_ids: list[UUID],
142
- include_failed: bool | Unset = False,
143
- ) -> HTTPValidationError | list[ArtifactGroupRecord | StandaloneArtifactRecord] | None:
144
- """Get Artifacts
145
-
146
- Get artifacts by artifact ID.
147
-
148
- Caution: This is not protected by RLS, so it will return the artifacts
149
- as long as the IDs are valid. The reason for it is that we would need to handle
150
- case for both authenticated and non-authenticated users similarly to how we handle
151
- stuff on the frontend, and that doesn't seem to be worth it at the moment.
152
-
153
- Alternative would be ditching this whole endpoint and replicating fetching the
154
- same artifacts in the frontend from the DB directly. It would be probably much
155
- faster and simpler anyway, with only slight duplicity of the code between python
156
- and typescript (as we need to be able to fetch artifacts here for the various
157
- workflow purposes).
158
-
159
- Args:
160
- artifact_ids (list[UUID]):
161
- include_failed (bool | Unset): Default: False.
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
- HTTPValidationError | list[ArtifactGroupRecord | StandaloneArtifactRecord]
169
- """
170
-
171
- return sync_detailed(
172
- client=client,
173
- artifact_ids=artifact_ids,
174
- include_failed=include_failed,
175
- ).parsed
176
-
177
-
178
- async def asyncio_detailed(
179
- *,
180
- client: AuthenticatedClient | Client,
181
- artifact_ids: list[UUID],
182
- include_failed: bool | Unset = False,
183
- ) -> Response[HTTPValidationError | list[ArtifactGroupRecord | StandaloneArtifactRecord]]:
184
- """Get Artifacts
185
-
186
- Get artifacts by artifact ID.
187
-
188
- Caution: This is not protected by RLS, so it will return the artifacts
189
- as long as the IDs are valid. The reason for it is that we would need to handle
190
- case for both authenticated and non-authenticated users similarly to how we handle
191
- stuff on the frontend, and that doesn't seem to be worth it at the moment.
192
-
193
- Alternative would be ditching this whole endpoint and replicating fetching the
194
- same artifacts in the frontend from the DB directly. It would be probably much
195
- faster and simpler anyway, with only slight duplicity of the code between python
196
- and typescript (as we need to be able to fetch artifacts here for the various
197
- workflow purposes).
198
-
199
- Args:
200
- artifact_ids (list[UUID]):
201
- include_failed (bool | Unset): Default: False.
202
-
203
- Raises:
204
- errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
205
- httpx.TimeoutException: If the request takes longer than Client.timeout.
206
-
207
- Returns:
208
- Response[HTTPValidationError | list[ArtifactGroupRecord | StandaloneArtifactRecord]]
209
- """
210
-
211
- kwargs = _get_kwargs(
212
- artifact_ids=artifact_ids,
213
- include_failed=include_failed,
214
- )
215
-
216
- response = await client.get_async_httpx_client().request(**kwargs)
217
-
218
- return _build_response(client=client, response=response)
219
-
220
-
221
- async def asyncio(
222
- *,
223
- client: AuthenticatedClient | Client,
224
- artifact_ids: list[UUID],
225
- include_failed: bool | Unset = False,
226
- ) -> HTTPValidationError | list[ArtifactGroupRecord | StandaloneArtifactRecord] | None:
227
- """Get Artifacts
228
-
229
- Get artifacts by artifact ID.
230
-
231
- Caution: This is not protected by RLS, so it will return the artifacts
232
- as long as the IDs are valid. The reason for it is that we would need to handle
233
- case for both authenticated and non-authenticated users similarly to how we handle
234
- stuff on the frontend, and that doesn't seem to be worth it at the moment.
235
-
236
- Alternative would be ditching this whole endpoint and replicating fetching the
237
- same artifacts in the frontend from the DB directly. It would be probably much
238
- faster and simpler anyway, with only slight duplicity of the code between python
239
- and typescript (as we need to be able to fetch artifacts here for the various
240
- workflow purposes).
241
-
242
- Args:
243
- artifact_ids (list[UUID]):
244
- include_failed (bool | Unset): Default: False.
245
-
246
- Raises:
247
- errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
248
- httpx.TimeoutException: If the request takes longer than Client.timeout.
249
-
250
- Returns:
251
- HTTPValidationError | list[ArtifactGroupRecord | StandaloneArtifactRecord]
252
- """
253
-
254
- return (
255
- await asyncio_detailed(
256
- client=client,
257
- artifact_ids=artifact_ids,
258
- include_failed=include_failed,
259
- )
260
- ).parsed
@@ -1,165 +0,0 @@
1
- from http import HTTPStatus
2
- from typing import Any, cast
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.llm_enum import LLMEnum
10
- from ...types import UNSET, Response
11
-
12
-
13
- def _get_kwargs(
14
- *,
15
- model: LLMEnum,
16
- ) -> dict[str, Any]:
17
- params: dict[str, Any] = {}
18
-
19
- json_model = model.value
20
- params["model"] = json_model
21
-
22
- params = {k: v for k, v in params.items() if v is not UNSET and v is not None}
23
-
24
- _kwargs: dict[str, Any] = {
25
- "method": "get",
26
- "url": "/models/default_timeout_seconds",
27
- "params": params,
28
- }
29
-
30
- return _kwargs
31
-
32
-
33
- def _parse_response(
34
- *, client: AuthenticatedClient | Client, response: httpx.Response
35
- ) -> HTTPValidationError | int | None:
36
- if response.status_code == 200:
37
- response_200 = cast(int, response.json())
38
- return response_200
39
-
40
- if response.status_code == 422:
41
- response_422 = HTTPValidationError.from_dict(response.json())
42
-
43
- return response_422
44
-
45
- if client.raise_on_unexpected_status:
46
- raise errors.UnexpectedStatus(response.status_code, response.content)
47
- else:
48
- return None
49
-
50
-
51
- def _build_response(
52
- *, client: AuthenticatedClient | Client, response: httpx.Response
53
- ) -> Response[HTTPValidationError | int]:
54
- return Response(
55
- status_code=HTTPStatus(response.status_code),
56
- content=response.content,
57
- headers=response.headers,
58
- parsed=_parse_response(client=client, response=response),
59
- )
60
-
61
-
62
- def sync_detailed(
63
- *,
64
- client: AuthenticatedClient | Client,
65
- model: LLMEnum,
66
- ) -> Response[HTTPValidationError | int]:
67
- """Get Default Timeout Seconds
68
-
69
- Args:
70
- model (LLMEnum):
71
-
72
- Raises:
73
- errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
74
- httpx.TimeoutException: If the request takes longer than Client.timeout.
75
-
76
- Returns:
77
- Response[HTTPValidationError | int]
78
- """
79
-
80
- kwargs = _get_kwargs(
81
- model=model,
82
- )
83
-
84
- response = client.get_httpx_client().request(
85
- **kwargs,
86
- )
87
-
88
- return _build_response(client=client, response=response)
89
-
90
-
91
- def sync(
92
- *,
93
- client: AuthenticatedClient | Client,
94
- model: LLMEnum,
95
- ) -> HTTPValidationError | int | None:
96
- """Get Default Timeout Seconds
97
-
98
- Args:
99
- model (LLMEnum):
100
-
101
- Raises:
102
- errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
103
- httpx.TimeoutException: If the request takes longer than Client.timeout.
104
-
105
- Returns:
106
- HTTPValidationError | int
107
- """
108
-
109
- return sync_detailed(
110
- client=client,
111
- model=model,
112
- ).parsed
113
-
114
-
115
- async def asyncio_detailed(
116
- *,
117
- client: AuthenticatedClient | Client,
118
- model: LLMEnum,
119
- ) -> Response[HTTPValidationError | int]:
120
- """Get Default Timeout Seconds
121
-
122
- Args:
123
- model (LLMEnum):
124
-
125
- Raises:
126
- errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
127
- httpx.TimeoutException: If the request takes longer than Client.timeout.
128
-
129
- Returns:
130
- Response[HTTPValidationError | int]
131
- """
132
-
133
- kwargs = _get_kwargs(
134
- model=model,
135
- )
136
-
137
- response = await client.get_async_httpx_client().request(**kwargs)
138
-
139
- return _build_response(client=client, response=response)
140
-
141
-
142
- async def asyncio(
143
- *,
144
- client: AuthenticatedClient | Client,
145
- model: LLMEnum,
146
- ) -> HTTPValidationError | int | None:
147
- """Get Default Timeout Seconds
148
-
149
- Args:
150
- model (LLMEnum):
151
-
152
- Raises:
153
- errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
154
- httpx.TimeoutException: If the request takes longer than Client.timeout.
155
-
156
- Returns:
157
- HTTPValidationError | int
158
- """
159
-
160
- return (
161
- await asyncio_detailed(
162
- client=client,
163
- model=model,
164
- )
165
- ).parsed
@@ -1,80 +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 ...types import Response
9
-
10
-
11
- def _get_kwargs() -> dict[str, Any]:
12
- _kwargs: dict[str, Any] = {
13
- "method": "get",
14
- "url": "/metrics",
15
- }
16
-
17
- return _kwargs
18
-
19
-
20
- def _parse_response(*, client: AuthenticatedClient | Client, response: httpx.Response) -> Any | None:
21
- if response.status_code == 200:
22
- return None
23
-
24
- if client.raise_on_unexpected_status:
25
- raise errors.UnexpectedStatus(response.status_code, response.content)
26
- else:
27
- return None
28
-
29
-
30
- def _build_response(*, client: AuthenticatedClient | Client, response: httpx.Response) -> Response[Any]:
31
- return Response(
32
- status_code=HTTPStatus(response.status_code),
33
- content=response.content,
34
- headers=response.headers,
35
- parsed=_parse_response(client=client, response=response),
36
- )
37
-
38
-
39
- def sync_detailed(
40
- *,
41
- client: AuthenticatedClient | Client,
42
- ) -> Response[Any]:
43
- """Get Metrics
44
-
45
- Raises:
46
- errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
47
- httpx.TimeoutException: If the request takes longer than Client.timeout.
48
-
49
- Returns:
50
- Response[Any]
51
- """
52
-
53
- kwargs = _get_kwargs()
54
-
55
- response = client.get_httpx_client().request(
56
- **kwargs,
57
- )
58
-
59
- return _build_response(client=client, response=response)
60
-
61
-
62
- async def asyncio_detailed(
63
- *,
64
- client: AuthenticatedClient | Client,
65
- ) -> Response[Any]:
66
- """Get Metrics
67
-
68
- Raises:
69
- errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
70
- httpx.TimeoutException: If the request takes longer than Client.timeout.
71
-
72
- Returns:
73
- Response[Any]
74
- """
75
-
76
- kwargs = _get_kwargs()
77
-
78
- response = await client.get_async_httpx_client().request(**kwargs)
79
-
80
- return _build_response(client=client, response=response)