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,9 +0,0 @@
1
- from enum import Enum
2
-
3
-
4
- class ResponseSchemaType(str, Enum):
5
- CUSTOM = "CUSTOM"
6
- JSON = "JSON"
7
-
8
- def __str__(self) -> str:
9
- return str(self.value)
@@ -1,61 +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="RevokeAPIKeyResponse")
10
-
11
-
12
- @_attrs_define
13
- class RevokeAPIKeyResponse:
14
- """
15
- Attributes:
16
- message (str):
17
- """
18
-
19
- message: str
20
- additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
21
-
22
- def to_dict(self) -> dict[str, Any]:
23
- message = self.message
24
-
25
- field_dict: dict[str, Any] = {}
26
- field_dict.update(self.additional_properties)
27
- field_dict.update(
28
- {
29
- "message": message,
30
- }
31
- )
32
-
33
- return field_dict
34
-
35
- @classmethod
36
- def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
37
- d = dict(src_dict)
38
- message = d.pop("message")
39
-
40
- revoke_api_key_response = cls(
41
- message=message,
42
- )
43
-
44
- revoke_api_key_response.additional_properties = d
45
- return revoke_api_key_response
46
-
47
- @property
48
- def additional_keys(self) -> list[str]:
49
- return list(self.additional_properties.keys())
50
-
51
- def __getitem__(self, key: str) -> Any:
52
- return self.additional_properties[key]
53
-
54
- def __setitem__(self, key: str, value: Any) -> None:
55
- self.additional_properties[key] = value
56
-
57
- def __delitem__(self, key: str) -> None:
58
- del self.additional_properties[key]
59
-
60
- def __contains__(self, key: str) -> bool:
61
- return key in self.additional_properties
@@ -1,62 +0,0 @@
1
- from __future__ import annotations
2
-
3
- from collections.abc import Mapping
4
- from typing import Any, TypeVar
5
- from uuid import UUID
6
-
7
- from attrs import define as _attrs_define
8
- from attrs import field as _attrs_field
9
-
10
- T = TypeVar("T", bound="RollbackToMessageRequest")
11
-
12
-
13
- @_attrs_define
14
- class RollbackToMessageRequest:
15
- """
16
- Attributes:
17
- message_id (UUID):
18
- """
19
-
20
- message_id: UUID
21
- additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
22
-
23
- def to_dict(self) -> dict[str, Any]:
24
- message_id = str(self.message_id)
25
-
26
- field_dict: dict[str, Any] = {}
27
- field_dict.update(self.additional_properties)
28
- field_dict.update(
29
- {
30
- "message_id": message_id,
31
- }
32
- )
33
-
34
- return field_dict
35
-
36
- @classmethod
37
- def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
38
- d = dict(src_dict)
39
- message_id = UUID(d.pop("message_id"))
40
-
41
- rollback_to_message_request = cls(
42
- message_id=message_id,
43
- )
44
-
45
- rollback_to_message_request.additional_properties = d
46
- return rollback_to_message_request
47
-
48
- @property
49
- def additional_keys(self) -> list[str]:
50
- return list(self.additional_properties.keys())
51
-
52
- def __getitem__(self, key: str) -> Any:
53
- return self.additional_properties[key]
54
-
55
- def __setitem__(self, key: str, value: Any) -> None:
56
- self.additional_properties[key] = value
57
-
58
- def __delitem__(self, key: str) -> None:
59
- del self.additional_properties[key]
60
-
61
- def __contains__(self, key: str) -> bool:
62
- return key in self.additional_properties
@@ -1,77 +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="RollbackToMessageResponse")
10
-
11
-
12
- @_attrs_define
13
- class RollbackToMessageResponse:
14
- """
15
- Attributes:
16
- deleted_messages (int):
17
- deleted_tasks (int):
18
- message (str):
19
- """
20
-
21
- deleted_messages: int
22
- deleted_tasks: int
23
- message: str
24
- additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
25
-
26
- def to_dict(self) -> dict[str, Any]:
27
- deleted_messages = self.deleted_messages
28
-
29
- deleted_tasks = self.deleted_tasks
30
-
31
- message = self.message
32
-
33
- field_dict: dict[str, Any] = {}
34
- field_dict.update(self.additional_properties)
35
- field_dict.update(
36
- {
37
- "deleted_messages": deleted_messages,
38
- "deleted_tasks": deleted_tasks,
39
- "message": message,
40
- }
41
- )
42
-
43
- return field_dict
44
-
45
- @classmethod
46
- def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
47
- d = dict(src_dict)
48
- deleted_messages = d.pop("deleted_messages")
49
-
50
- deleted_tasks = d.pop("deleted_tasks")
51
-
52
- message = d.pop("message")
53
-
54
- rollback_to_message_response = cls(
55
- deleted_messages=deleted_messages,
56
- deleted_tasks=deleted_tasks,
57
- message=message,
58
- )
59
-
60
- rollback_to_message_response.additional_properties = d
61
- return rollback_to_message_response
62
-
63
- @property
64
- def additional_keys(self) -> list[str]:
65
- return list(self.additional_properties.keys())
66
-
67
- def __getitem__(self, key: str) -> Any:
68
- return self.additional_properties[key]
69
-
70
- def __setitem__(self, key: str, value: Any) -> None:
71
- self.additional_properties[key] = value
72
-
73
- def __delitem__(self, key: str) -> None:
74
- del self.additional_properties[key]
75
-
76
- def __contains__(self, key: str) -> bool:
77
- return key in self.additional_properties
@@ -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
- from ..models.event_type import EventType
10
- from ..types import UNSET, Unset
11
-
12
- T = TypeVar("T", bound="SessionChangedPayload")
13
-
14
-
15
- @_attrs_define
16
- class SessionChangedPayload:
17
- """
18
- Attributes:
19
- event_type (EventType | Unset):
20
- """
21
-
22
- event_type: EventType | Unset = UNSET
23
- additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
24
-
25
- def to_dict(self) -> dict[str, Any]:
26
- event_type: str | Unset = UNSET
27
- if not isinstance(self.event_type, Unset):
28
- event_type = self.event_type.value
29
-
30
- field_dict: dict[str, Any] = {}
31
- field_dict.update(self.additional_properties)
32
- field_dict.update({})
33
- if event_type is not UNSET:
34
- field_dict["event_type"] = event_type
35
-
36
- return field_dict
37
-
38
- @classmethod
39
- def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
40
- d = dict(src_dict)
41
- _event_type = d.pop("event_type", UNSET)
42
- event_type: EventType | Unset
43
- if isinstance(_event_type, Unset):
44
- event_type = UNSET
45
- else:
46
- event_type = EventType(_event_type)
47
-
48
- session_changed_payload = cls(
49
- event_type=event_type,
50
- )
51
-
52
- session_changed_payload.additional_properties = d
53
- return session_changed_payload
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
@@ -1,121 +0,0 @@
1
- from __future__ import annotations
2
-
3
- from collections.abc import Mapping
4
- from typing import Any, Literal, TypeVar, cast
5
-
6
- from attrs import define as _attrs_define
7
- from attrs import field as _attrs_field
8
-
9
- from ..models.simple_chat_message_role import SimpleChatMessageRole
10
- from ..types import UNSET, Unset
11
-
12
- T = TypeVar("T", bound="SimpleChatMessage")
13
-
14
-
15
- @_attrs_define
16
- class SimpleChatMessage:
17
- """
18
- Attributes:
19
- content (str):
20
- role (SimpleChatMessageRole):
21
- message_type (Literal['simple'] | Unset): Default: 'simple'.
22
- completion_tokens (int | None | Unset):
23
- cacheable (bool | Unset): Default: False.
24
- explicit_cache_breakpoint (bool | Unset): Default: False.
25
- """
26
-
27
- content: str
28
- role: SimpleChatMessageRole
29
- message_type: Literal["simple"] | Unset = "simple"
30
- completion_tokens: int | None | Unset = UNSET
31
- cacheable: bool | Unset = False
32
- explicit_cache_breakpoint: bool | Unset = False
33
- additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
34
-
35
- def to_dict(self) -> dict[str, Any]:
36
- content = self.content
37
-
38
- role = self.role.value
39
-
40
- message_type = self.message_type
41
-
42
- completion_tokens: int | None | Unset
43
- if isinstance(self.completion_tokens, Unset):
44
- completion_tokens = UNSET
45
- else:
46
- completion_tokens = self.completion_tokens
47
-
48
- cacheable = self.cacheable
49
-
50
- explicit_cache_breakpoint = self.explicit_cache_breakpoint
51
-
52
- field_dict: dict[str, Any] = {}
53
- field_dict.update(self.additional_properties)
54
- field_dict.update(
55
- {
56
- "content": content,
57
- "role": role,
58
- }
59
- )
60
- if message_type is not UNSET:
61
- field_dict["message_type"] = message_type
62
- if completion_tokens is not UNSET:
63
- field_dict["completion_tokens"] = completion_tokens
64
- if cacheable is not UNSET:
65
- field_dict["cacheable"] = cacheable
66
- if explicit_cache_breakpoint is not UNSET:
67
- field_dict["explicit_cache_breakpoint"] = explicit_cache_breakpoint
68
-
69
- return field_dict
70
-
71
- @classmethod
72
- def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
73
- d = dict(src_dict)
74
- content = d.pop("content")
75
-
76
- role = SimpleChatMessageRole(d.pop("role"))
77
-
78
- message_type = cast(Literal["simple"] | Unset, d.pop("message_type", UNSET))
79
- if message_type != "simple" and not isinstance(message_type, Unset):
80
- raise ValueError(f"message_type must match const 'simple', got '{message_type}'")
81
-
82
- def _parse_completion_tokens(data: object) -> int | None | Unset:
83
- if data is None:
84
- return data
85
- if isinstance(data, Unset):
86
- return data
87
- return cast(int | None | Unset, data)
88
-
89
- completion_tokens = _parse_completion_tokens(d.pop("completion_tokens", UNSET))
90
-
91
- cacheable = d.pop("cacheable", UNSET)
92
-
93
- explicit_cache_breakpoint = d.pop("explicit_cache_breakpoint", UNSET)
94
-
95
- simple_chat_message = cls(
96
- content=content,
97
- role=role,
98
- message_type=message_type,
99
- completion_tokens=completion_tokens,
100
- cacheable=cacheable,
101
- explicit_cache_breakpoint=explicit_cache_breakpoint,
102
- )
103
-
104
- simple_chat_message.additional_properties = d
105
- return simple_chat_message
106
-
107
- @property
108
- def additional_keys(self) -> list[str]:
109
- return list(self.additional_properties.keys())
110
-
111
- def __getitem__(self, key: str) -> Any:
112
- return self.additional_properties[key]
113
-
114
- def __setitem__(self, key: str, value: Any) -> None:
115
- self.additional_properties[key] = value
116
-
117
- def __delitem__(self, key: str) -> None:
118
- del self.additional_properties[key]
119
-
120
- def __contains__(self, key: str) -> bool:
121
- return key in self.additional_properties
@@ -1,10 +0,0 @@
1
- from enum import Enum
2
-
3
-
4
- class SimpleChatMessageRole(str, Enum):
5
- ASSISTANT = "assistant"
6
- SYSTEM = "system"
7
- USER = "user"
8
-
9
- def __str__(self) -> str:
10
- return str(self.value)
@@ -1,156 +0,0 @@
1
- from __future__ import annotations
2
-
3
- from collections.abc import Mapping
4
- from typing import TYPE_CHECKING, Any, Literal, TypeVar, cast
5
-
6
- from attrs import define as _attrs_define
7
- from attrs import field as _attrs_field
8
-
9
- from ..types import UNSET, Unset
10
-
11
- if TYPE_CHECKING:
12
- from ..models.chat_completion_message_tool_call import ChatCompletionMessageToolCall
13
-
14
-
15
- T = TypeVar("T", bound="SimpleChatMessageWithToolCalls")
16
-
17
-
18
- @_attrs_define
19
- class SimpleChatMessageWithToolCalls:
20
- """
21
- Attributes:
22
- role (Literal['assistant']):
23
- tool_calls (list[ChatCompletionMessageToolCall]):
24
- message_type (Literal['simple_with_tool_calls'] | Unset): Default: 'simple_with_tool_calls'.
25
- content (None | str | Unset):
26
- completion_tokens (int | None | Unset):
27
- cacheable (bool | Unset): Default: False.
28
- explicit_cache_breakpoint (bool | Unset): Default: False.
29
- """
30
-
31
- role: Literal["assistant"]
32
- tool_calls: list[ChatCompletionMessageToolCall]
33
- message_type: Literal["simple_with_tool_calls"] | Unset = "simple_with_tool_calls"
34
- content: None | str | Unset = UNSET
35
- completion_tokens: int | None | Unset = UNSET
36
- cacheable: bool | Unset = False
37
- explicit_cache_breakpoint: bool | Unset = False
38
- additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
39
-
40
- def to_dict(self) -> dict[str, Any]:
41
- role = self.role
42
-
43
- tool_calls = []
44
- for tool_calls_item_data in self.tool_calls:
45
- tool_calls_item = tool_calls_item_data.to_dict()
46
- tool_calls.append(tool_calls_item)
47
-
48
- message_type = self.message_type
49
-
50
- content: None | str | Unset
51
- if isinstance(self.content, Unset):
52
- content = UNSET
53
- else:
54
- content = self.content
55
-
56
- completion_tokens: int | None | Unset
57
- if isinstance(self.completion_tokens, Unset):
58
- completion_tokens = UNSET
59
- else:
60
- completion_tokens = self.completion_tokens
61
-
62
- cacheable = self.cacheable
63
-
64
- explicit_cache_breakpoint = self.explicit_cache_breakpoint
65
-
66
- field_dict: dict[str, Any] = {}
67
- field_dict.update(self.additional_properties)
68
- field_dict.update(
69
- {
70
- "role": role,
71
- "tool_calls": tool_calls,
72
- }
73
- )
74
- if message_type is not UNSET:
75
- field_dict["message_type"] = message_type
76
- if content is not UNSET:
77
- field_dict["content"] = content
78
- if completion_tokens is not UNSET:
79
- field_dict["completion_tokens"] = completion_tokens
80
- if cacheable is not UNSET:
81
- field_dict["cacheable"] = cacheable
82
- if explicit_cache_breakpoint is not UNSET:
83
- field_dict["explicit_cache_breakpoint"] = explicit_cache_breakpoint
84
-
85
- return field_dict
86
-
87
- @classmethod
88
- def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
89
- from ..models.chat_completion_message_tool_call import ChatCompletionMessageToolCall
90
-
91
- d = dict(src_dict)
92
- role = cast(Literal["assistant"], d.pop("role"))
93
- if role != "assistant":
94
- raise ValueError(f"role must match const 'assistant', got '{role}'")
95
-
96
- tool_calls = []
97
- _tool_calls = d.pop("tool_calls")
98
- for tool_calls_item_data in _tool_calls:
99
- tool_calls_item = ChatCompletionMessageToolCall.from_dict(tool_calls_item_data)
100
-
101
- tool_calls.append(tool_calls_item)
102
-
103
- message_type = cast(Literal["simple_with_tool_calls"] | Unset, d.pop("message_type", UNSET))
104
- if message_type != "simple_with_tool_calls" and not isinstance(message_type, Unset):
105
- raise ValueError(f"message_type must match const 'simple_with_tool_calls', got '{message_type}'")
106
-
107
- def _parse_content(data: object) -> None | str | Unset:
108
- if data is None:
109
- return data
110
- if isinstance(data, Unset):
111
- return data
112
- return cast(None | str | Unset, data)
113
-
114
- content = _parse_content(d.pop("content", UNSET))
115
-
116
- def _parse_completion_tokens(data: object) -> int | None | Unset:
117
- if data is None:
118
- return data
119
- if isinstance(data, Unset):
120
- return data
121
- return cast(int | None | Unset, data)
122
-
123
- completion_tokens = _parse_completion_tokens(d.pop("completion_tokens", UNSET))
124
-
125
- cacheable = d.pop("cacheable", UNSET)
126
-
127
- explicit_cache_breakpoint = d.pop("explicit_cache_breakpoint", UNSET)
128
-
129
- simple_chat_message_with_tool_calls = cls(
130
- role=role,
131
- tool_calls=tool_calls,
132
- message_type=message_type,
133
- content=content,
134
- completion_tokens=completion_tokens,
135
- cacheable=cacheable,
136
- explicit_cache_breakpoint=explicit_cache_breakpoint,
137
- )
138
-
139
- simple_chat_message_with_tool_calls.additional_properties = d
140
- return simple_chat_message_with_tool_calls
141
-
142
- @property
143
- def additional_keys(self) -> list[str]:
144
- return list(self.additional_properties.keys())
145
-
146
- def __getitem__(self, key: str) -> Any:
147
- return self.additional_properties[key]
148
-
149
- def __setitem__(self, key: str, value: Any) -> None:
150
- self.additional_properties[key] = value
151
-
152
- def __delitem__(self, key: str) -> None:
153
- del self.additional_properties[key]
154
-
155
- def __contains__(self, key: str) -> bool:
156
- return key in self.additional_properties
@@ -1,92 +0,0 @@
1
- from __future__ import annotations
2
-
3
- from collections.abc import Mapping
4
- from typing import Any, TypeVar, cast
5
-
6
- from attrs import define as _attrs_define
7
- from attrs import field as _attrs_field
8
-
9
- from ..types import UNSET, Unset
10
-
11
- T = TypeVar("T", bound="SourceDatabaseEntry")
12
-
13
-
14
- @_attrs_define
15
- class SourceDatabaseEntry:
16
- """Simplified version of toolkit.agents.source_bank.SourceBase for the database.
17
-
18
- Attributes:
19
- url (str):
20
- content_summary (str):
21
- title (None | str | Unset):
22
- """
23
-
24
- url: str
25
- content_summary: str
26
- title: None | str | Unset = UNSET
27
- additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
28
-
29
- def to_dict(self) -> dict[str, Any]:
30
- url = self.url
31
-
32
- content_summary = self.content_summary
33
-
34
- title: None | str | Unset
35
- if isinstance(self.title, Unset):
36
- title = UNSET
37
- else:
38
- title = self.title
39
-
40
- field_dict: dict[str, Any] = {}
41
- field_dict.update(self.additional_properties)
42
- field_dict.update(
43
- {
44
- "url": url,
45
- "content_summary": content_summary,
46
- }
47
- )
48
- if title is not UNSET:
49
- field_dict["title"] = title
50
-
51
- return field_dict
52
-
53
- @classmethod
54
- def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
55
- d = dict(src_dict)
56
- url = d.pop("url")
57
-
58
- content_summary = d.pop("content_summary")
59
-
60
- def _parse_title(data: object) -> None | str | Unset:
61
- if data is None:
62
- return data
63
- if isinstance(data, Unset):
64
- return data
65
- return cast(None | str | Unset, data)
66
-
67
- title = _parse_title(d.pop("title", UNSET))
68
-
69
- source_database_entry = cls(
70
- url=url,
71
- content_summary=content_summary,
72
- title=title,
73
- )
74
-
75
- source_database_entry.additional_properties = d
76
- return source_database_entry
77
-
78
- @property
79
- def additional_keys(self) -> list[str]:
80
- return list(self.additional_properties.keys())
81
-
82
- def __getitem__(self, key: str) -> Any:
83
- return self.additional_properties[key]
84
-
85
- def __setitem__(self, key: str, value: Any) -> None:
86
- self.additional_properties[key] = value
87
-
88
- def __delitem__(self, key: str) -> None:
89
- del self.additional_properties[key]
90
-
91
- def __contains__(self, key: str) -> bool:
92
- return key in self.additional_properties