everyrow 0.1.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 (183) hide show
  1. everyrow/__init__.py +4 -0
  2. everyrow/api_utils.py +45 -0
  3. everyrow/citations.py +46 -0
  4. everyrow/constants.py +4 -0
  5. everyrow/generated/__init__.py +8 -0
  6. everyrow/generated/api/__init__.py +1 -0
  7. everyrow/generated/api/default/__init__.py +1 -0
  8. everyrow/generated/api/default/continue_task_endpoint_tasks_continue_post.py +208 -0
  9. everyrow/generated/api/default/copy_artifacts_artifacts_copy_post.py +166 -0
  10. everyrow/generated/api/default/copy_workflow_endpoint_workflows_copy_post.py +166 -0
  11. everyrow/generated/api/default/create_api_key_endpoint_api_keys_create_post.py +186 -0
  12. everyrow/generated/api/default/create_session_endpoint_sessions_create_post.py +198 -0
  13. everyrow/generated/api/default/create_workflow_from_artifact_workflows_from_artifact_post.py +174 -0
  14. everyrow/generated/api/default/export_to_google_sheets_export_post.py +168 -0
  15. everyrow/generated/api/default/generate_feedback_endpoint_tasks_generate_feedback_post.py +186 -0
  16. everyrow/generated/api/default/get_artifacts_artifacts_get.py +260 -0
  17. everyrow/generated/api/default/get_default_timeout_seconds_models_default_timeout_seconds_get.py +165 -0
  18. everyrow/generated/api/default/get_job_progress_for_task_jobs_progress_get.py +167 -0
  19. everyrow/generated/api/default/get_metrics_metrics_get.py +80 -0
  20. everyrow/generated/api/default/get_queues_stats_jobs_queues_get.py +123 -0
  21. everyrow/generated/api/default/get_task_status_endpoint_tasks_task_id_status_get.py +193 -0
  22. everyrow/generated/api/default/get_user_usage_usage_get.py +123 -0
  23. everyrow/generated/api/default/healthz_healthz_get.py +127 -0
  24. everyrow/generated/api/default/import_from_google_sheets_import_post.py +170 -0
  25. everyrow/generated/api/default/interrupt_chat_task_tasks_chat_interrupt_post.py +172 -0
  26. everyrow/generated/api/default/list_api_keys_endpoint_api_keys_get.py +186 -0
  27. everyrow/generated/api/default/re_execute_task_endpoint_tasks_re_execute_post.py +192 -0
  28. everyrow/generated/api/default/revoke_api_key_endpoint_api_keys_key_id_revoke_post.py +181 -0
  29. everyrow/generated/api/default/revoke_jobs_for_task_jobs_revoke_post.py +164 -0
  30. everyrow/generated/api/default/rollback_to_message_endpoint_tasks_chat_rollback_post.py +186 -0
  31. everyrow/generated/api/default/submit_chat_task_tasks_chat_post.py +164 -0
  32. everyrow/generated/api/default/submit_task_tasks_post.py +172 -0
  33. everyrow/generated/api/default/task_resource_estimation_task_resource_estimation_post.py +319 -0
  34. everyrow/generated/api/default/trigger_workflow_execution_endpoint_workflows_trigger_post.py +166 -0
  35. everyrow/generated/api/default/whoami_whoami_get.py +127 -0
  36. everyrow/generated/client.py +268 -0
  37. everyrow/generated/errors.py +16 -0
  38. everyrow/generated/models/__init__.py +281 -0
  39. everyrow/generated/models/agent_improvement_instruction.py +69 -0
  40. everyrow/generated/models/agent_query_params.py +362 -0
  41. everyrow/generated/models/agent_query_params_system_prompt_kind_type_0.py +10 -0
  42. everyrow/generated/models/agent_task_args.py +163 -0
  43. everyrow/generated/models/agent_task_args_processing_mode.py +9 -0
  44. everyrow/generated/models/allowed_suggestions.py +8 -0
  45. everyrow/generated/models/api_key_info.py +163 -0
  46. everyrow/generated/models/artifact_changed_payload.py +89 -0
  47. everyrow/generated/models/artifact_group_record.py +330 -0
  48. everyrow/generated/models/artifact_group_record_metadata_type_0.py +46 -0
  49. everyrow/generated/models/artifact_group_record_trace_mapping_type_0.py +46 -0
  50. everyrow/generated/models/artifact_status.py +14 -0
  51. everyrow/generated/models/auto_cohort_conversation_message.py +533 -0
  52. everyrow/generated/models/aux_data.py +128 -0
  53. everyrow/generated/models/aux_data_source_bank.py +59 -0
  54. everyrow/generated/models/chat_completion_message_tool_call.py +46 -0
  55. everyrow/generated/models/chat_message_metadata.py +193 -0
  56. everyrow/generated/models/concatenate_query_params.py +46 -0
  57. everyrow/generated/models/concatenate_request.py +306 -0
  58. everyrow/generated/models/continue_reason.py +9 -0
  59. everyrow/generated/models/continue_task_request.py +133 -0
  60. everyrow/generated/models/controller_improvement_round.py +79 -0
  61. everyrow/generated/models/conversation_changed_payload.py +89 -0
  62. everyrow/generated/models/copy_artifacts_request.py +70 -0
  63. everyrow/generated/models/copy_artifacts_response.py +70 -0
  64. everyrow/generated/models/copy_workflow_request.py +62 -0
  65. everyrow/generated/models/copy_workflow_response.py +70 -0
  66. everyrow/generated/models/create_api_key_request.py +95 -0
  67. everyrow/generated/models/create_api_key_response.py +96 -0
  68. everyrow/generated/models/create_group_query_params.py +61 -0
  69. everyrow/generated/models/create_group_request.py +305 -0
  70. everyrow/generated/models/create_query_params.py +61 -0
  71. everyrow/generated/models/create_request.py +305 -0
  72. everyrow/generated/models/create_session_request.py +62 -0
  73. everyrow/generated/models/create_session_response.py +63 -0
  74. everyrow/generated/models/create_workflow_from_artifact_request.py +92 -0
  75. everyrow/generated/models/create_workflow_from_artifact_response.py +70 -0
  76. everyrow/generated/models/data_frame_method.py +18 -0
  77. everyrow/generated/models/date_cutoffs.py +145 -0
  78. everyrow/generated/models/dedupe_mode.py +9 -0
  79. everyrow/generated/models/dedupe_query_params.py +174 -0
  80. everyrow/generated/models/dedupe_request_params.py +311 -0
  81. everyrow/generated/models/deep_merge_public_params.py +143 -0
  82. everyrow/generated/models/deep_merge_request.py +313 -0
  83. everyrow/generated/models/deep_rank_public_params.py +99 -0
  84. everyrow/generated/models/deep_rank_request.py +313 -0
  85. everyrow/generated/models/deep_screen_public_params.py +122 -0
  86. everyrow/generated/models/deep_screen_request.py +313 -0
  87. everyrow/generated/models/derive_expression.py +69 -0
  88. everyrow/generated/models/derive_query_params.py +75 -0
  89. everyrow/generated/models/derive_request.py +307 -0
  90. everyrow/generated/models/document_query_tool.py +12 -0
  91. everyrow/generated/models/drop_columns_query_params.py +61 -0
  92. everyrow/generated/models/drop_columns_request.py +305 -0
  93. everyrow/generated/models/embedding_models.py +9 -0
  94. everyrow/generated/models/event_type.py +14 -0
  95. everyrow/generated/models/execution_metadata.py +146 -0
  96. everyrow/generated/models/export_request.py +75 -0
  97. everyrow/generated/models/export_request_token_data.py +46 -0
  98. everyrow/generated/models/export_to_google_sheets_export_post_response_export_to_google_sheets_export_post.py +46 -0
  99. everyrow/generated/models/filter_query_params.py +91 -0
  100. everyrow/generated/models/filter_request.py +305 -0
  101. everyrow/generated/models/flatten_query_params.py +46 -0
  102. everyrow/generated/models/flatten_request.py +305 -0
  103. everyrow/generated/models/generate_feedback_request.py +62 -0
  104. everyrow/generated/models/group_by_query_params.py +62 -0
  105. everyrow/generated/models/group_by_request.py +305 -0
  106. everyrow/generated/models/healthz_healthz_get_response_healthz_healthz_get.py +46 -0
  107. everyrow/generated/models/http_validation_error.py +79 -0
  108. everyrow/generated/models/image_chat_content_part.py +80 -0
  109. everyrow/generated/models/image_chat_content_part_image_url.py +46 -0
  110. everyrow/generated/models/import_from_google_sheets_import_post_response_import_from_google_sheets_import_post.py +46 -0
  111. everyrow/generated/models/import_request.py +83 -0
  112. everyrow/generated/models/import_request_token_data.py +46 -0
  113. everyrow/generated/models/insufficient_balance_error.py +81 -0
  114. everyrow/generated/models/join_query_params.py +73 -0
  115. everyrow/generated/models/join_request.py +305 -0
  116. everyrow/generated/models/llm_enum.py +54 -0
  117. everyrow/generated/models/map_agent_request_params.py +313 -0
  118. everyrow/generated/models/map_multi_agent_request_params.py +313 -0
  119. everyrow/generated/models/message_created_payload.py +98 -0
  120. everyrow/generated/models/multi_agent_effort_level.py +10 -0
  121. everyrow/generated/models/multi_agent_query_params.py +264 -0
  122. everyrow/generated/models/multi_modal_chat_message.py +160 -0
  123. everyrow/generated/models/multi_modal_chat_message_role.py +10 -0
  124. everyrow/generated/models/preview_metadata.py +144 -0
  125. everyrow/generated/models/processing_mode.py +10 -0
  126. everyrow/generated/models/progress_status.py +83 -0
  127. everyrow/generated/models/queue_stats.py +77 -0
  128. everyrow/generated/models/reduce_agent_request_params.py +305 -0
  129. everyrow/generated/models/reduce_multi_agent_request_params.py +305 -0
  130. everyrow/generated/models/resource_estimation_response.py +85 -0
  131. everyrow/generated/models/response_schema_type.py +9 -0
  132. everyrow/generated/models/revoke_api_key_response.py +61 -0
  133. everyrow/generated/models/rollback_to_message_request.py +62 -0
  134. everyrow/generated/models/rollback_to_message_response.py +77 -0
  135. everyrow/generated/models/session_changed_payload.py +69 -0
  136. everyrow/generated/models/simple_chat_message.py +121 -0
  137. everyrow/generated/models/simple_chat_message_role.py +10 -0
  138. everyrow/generated/models/simple_chat_message_with_tool_calls.py +156 -0
  139. everyrow/generated/models/source_database_entry.py +92 -0
  140. everyrow/generated/models/standalone_artifact_record.py +278 -0
  141. everyrow/generated/models/standalone_artifact_record_metadata_type_0.py +46 -0
  142. everyrow/generated/models/standalone_artifact_record_trace_mapping_type_0.py +46 -0
  143. everyrow/generated/models/status_count.py +71 -0
  144. everyrow/generated/models/status_count_status.py +13 -0
  145. everyrow/generated/models/submit_chat_task_body.py +497 -0
  146. everyrow/generated/models/submit_chat_task_body_selected_task_type_type_0.py +11 -0
  147. everyrow/generated/models/submit_task_body.py +745 -0
  148. everyrow/generated/models/task_changed_payload.py +105 -0
  149. everyrow/generated/models/task_effort.py +10 -0
  150. everyrow/generated/models/task_id_request.py +62 -0
  151. everyrow/generated/models/task_insert.py +725 -0
  152. everyrow/generated/models/task_insert_query_params.py +46 -0
  153. everyrow/generated/models/task_metadata.py +323 -0
  154. everyrow/generated/models/task_metadata_cols_to_rename_type_0.py +46 -0
  155. everyrow/generated/models/task_response.py +62 -0
  156. everyrow/generated/models/task_status.py +12 -0
  157. everyrow/generated/models/task_status_response.py +112 -0
  158. everyrow/generated/models/task_type.py +31 -0
  159. everyrow/generated/models/text_chat_content_part.py +74 -0
  160. everyrow/generated/models/tool_response_message.py +127 -0
  161. everyrow/generated/models/toolkit_constants.py +80 -0
  162. everyrow/generated/models/trace_changed_payload.py +94 -0
  163. everyrow/generated/models/trace_info.py +78 -0
  164. everyrow/generated/models/trigger_workflow_execution_request.py +112 -0
  165. everyrow/generated/models/trigger_workflow_execution_request_task_params.py +65 -0
  166. everyrow/generated/models/trigger_workflow_execution_request_task_params_additional_property.py +46 -0
  167. everyrow/generated/models/trigger_workflow_execution_response.py +69 -0
  168. everyrow/generated/models/upload_csv_payload.py +310 -0
  169. everyrow/generated/models/upload_csv_query_params.py +114 -0
  170. everyrow/generated/models/usage_response.py +77 -0
  171. everyrow/generated/models/validation_error.py +90 -0
  172. everyrow/generated/models/whoami_whoami_get_response_whoami_whoami_get.py +46 -0
  173. everyrow/generated/models/workflow_leaf_node_input.py +70 -0
  174. everyrow/generated/py.typed +1 -0
  175. everyrow/generated/types.py +54 -0
  176. everyrow/ops.py +672 -0
  177. everyrow/result.py +25 -0
  178. everyrow/session.py +53 -0
  179. everyrow/task.py +143 -0
  180. everyrow-0.1.0.dist-info/METADATA +238 -0
  181. everyrow-0.1.0.dist-info/RECORD +183 -0
  182. everyrow-0.1.0.dist-info/WHEEL +4 -0
  183. everyrow-0.1.0.dist-info/licenses/LICENSE.txt +21 -0
@@ -0,0 +1,74 @@
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 ..types import UNSET, Unset
10
+
11
+ T = TypeVar("T", bound="TextChatContentPart")
12
+
13
+
14
+ @_attrs_define
15
+ class TextChatContentPart:
16
+ """
17
+ Attributes:
18
+ text (str):
19
+ type_ (Literal['text'] | Unset): Default: 'text'.
20
+ """
21
+
22
+ text: str
23
+ type_: Literal["text"] | Unset = "text"
24
+ additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
25
+
26
+ def to_dict(self) -> dict[str, Any]:
27
+ text = self.text
28
+
29
+ type_ = self.type_
30
+
31
+ field_dict: dict[str, Any] = {}
32
+ field_dict.update(self.additional_properties)
33
+ field_dict.update(
34
+ {
35
+ "text": text,
36
+ }
37
+ )
38
+ if type_ is not UNSET:
39
+ field_dict["type"] = type_
40
+
41
+ return field_dict
42
+
43
+ @classmethod
44
+ def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
45
+ d = dict(src_dict)
46
+ text = d.pop("text")
47
+
48
+ type_ = cast(Literal["text"] | Unset, d.pop("type", UNSET))
49
+ if type_ != "text" and not isinstance(type_, Unset):
50
+ raise ValueError(f"type must match const 'text', got '{type_}'")
51
+
52
+ text_chat_content_part = cls(
53
+ text=text,
54
+ type_=type_,
55
+ )
56
+
57
+ text_chat_content_part.additional_properties = d
58
+ return text_chat_content_part
59
+
60
+ @property
61
+ def additional_keys(self) -> list[str]:
62
+ return list(self.additional_properties.keys())
63
+
64
+ def __getitem__(self, key: str) -> Any:
65
+ return self.additional_properties[key]
66
+
67
+ def __setitem__(self, key: str, value: Any) -> None:
68
+ self.additional_properties[key] = value
69
+
70
+ def __delitem__(self, key: str) -> None:
71
+ del self.additional_properties[key]
72
+
73
+ def __contains__(self, key: str) -> bool:
74
+ return key in self.additional_properties
@@ -0,0 +1,127 @@
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 ..types import UNSET, Unset
10
+
11
+ T = TypeVar("T", bound="ToolResponseMessage")
12
+
13
+
14
+ @_attrs_define
15
+ class ToolResponseMessage:
16
+ """
17
+ Attributes:
18
+ tool_call_id (str):
19
+ content (str):
20
+ description (str):
21
+ tool_name (str):
22
+ message_type (Literal['tool_response'] | Unset): Default: 'tool_response'.
23
+ role (Literal['tool'] | Unset): Default: 'tool'.
24
+ cacheable (bool | Unset): Default: False.
25
+ explicit_cache_breakpoint (bool | Unset): Default: False.
26
+ """
27
+
28
+ tool_call_id: str
29
+ content: str
30
+ description: str
31
+ tool_name: str
32
+ message_type: Literal["tool_response"] | Unset = "tool_response"
33
+ role: Literal["tool"] | Unset = "tool"
34
+ cacheable: bool | Unset = False
35
+ explicit_cache_breakpoint: bool | Unset = False
36
+ additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
37
+
38
+ def to_dict(self) -> dict[str, Any]:
39
+ tool_call_id = self.tool_call_id
40
+
41
+ content = self.content
42
+
43
+ description = self.description
44
+
45
+ tool_name = self.tool_name
46
+
47
+ message_type = self.message_type
48
+
49
+ role = self.role
50
+
51
+ cacheable = self.cacheable
52
+
53
+ explicit_cache_breakpoint = self.explicit_cache_breakpoint
54
+
55
+ field_dict: dict[str, Any] = {}
56
+ field_dict.update(self.additional_properties)
57
+ field_dict.update(
58
+ {
59
+ "tool_call_id": tool_call_id,
60
+ "content": content,
61
+ "description": description,
62
+ "tool_name": tool_name,
63
+ }
64
+ )
65
+ if message_type is not UNSET:
66
+ field_dict["message_type"] = message_type
67
+ if role is not UNSET:
68
+ field_dict["role"] = role
69
+ if cacheable is not UNSET:
70
+ field_dict["cacheable"] = cacheable
71
+ if explicit_cache_breakpoint is not UNSET:
72
+ field_dict["explicit_cache_breakpoint"] = explicit_cache_breakpoint
73
+
74
+ return field_dict
75
+
76
+ @classmethod
77
+ def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
78
+ d = dict(src_dict)
79
+ tool_call_id = d.pop("tool_call_id")
80
+
81
+ content = d.pop("content")
82
+
83
+ description = d.pop("description")
84
+
85
+ tool_name = d.pop("tool_name")
86
+
87
+ message_type = cast(Literal["tool_response"] | Unset, d.pop("message_type", UNSET))
88
+ if message_type != "tool_response" and not isinstance(message_type, Unset):
89
+ raise ValueError(f"message_type must match const 'tool_response', got '{message_type}'")
90
+
91
+ role = cast(Literal["tool"] | Unset, d.pop("role", UNSET))
92
+ if role != "tool" and not isinstance(role, Unset):
93
+ raise ValueError(f"role must match const 'tool', got '{role}'")
94
+
95
+ cacheable = d.pop("cacheable", UNSET)
96
+
97
+ explicit_cache_breakpoint = d.pop("explicit_cache_breakpoint", UNSET)
98
+
99
+ tool_response_message = cls(
100
+ tool_call_id=tool_call_id,
101
+ content=content,
102
+ description=description,
103
+ tool_name=tool_name,
104
+ message_type=message_type,
105
+ role=role,
106
+ cacheable=cacheable,
107
+ explicit_cache_breakpoint=explicit_cache_breakpoint,
108
+ )
109
+
110
+ tool_response_message.additional_properties = d
111
+ return tool_response_message
112
+
113
+ @property
114
+ def additional_keys(self) -> list[str]:
115
+ return list(self.additional_properties.keys())
116
+
117
+ def __getitem__(self, key: str) -> Any:
118
+ return self.additional_properties[key]
119
+
120
+ def __setitem__(self, key: str, value: Any) -> None:
121
+ self.additional_properties[key] = value
122
+
123
+ def __delitem__(self, key: str) -> None:
124
+ del self.additional_properties[key]
125
+
126
+ def __contains__(self, key: str) -> bool:
127
+ return key in self.additional_properties
@@ -0,0 +1,80 @@
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 ..types import UNSET, Unset
10
+
11
+ T = TypeVar("T", bound="ToolkitConstants")
12
+
13
+
14
+ @_attrs_define
15
+ class ToolkitConstants:
16
+ """
17
+ Attributes:
18
+ low_effort_agent_iteration_budget (Literal[5] | Unset): Default: 5.
19
+ high_effort_agent_iteration_budget (Literal[10] | Unset): Default: 10.
20
+ """
21
+
22
+ low_effort_agent_iteration_budget: Literal[5] | Unset = 5
23
+ high_effort_agent_iteration_budget: Literal[10] | Unset = 10
24
+ additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
25
+
26
+ def to_dict(self) -> dict[str, Any]:
27
+ low_effort_agent_iteration_budget = self.low_effort_agent_iteration_budget
28
+
29
+ high_effort_agent_iteration_budget = self.high_effort_agent_iteration_budget
30
+
31
+ field_dict: dict[str, Any] = {}
32
+ field_dict.update(self.additional_properties)
33
+ field_dict.update({})
34
+ if low_effort_agent_iteration_budget is not UNSET:
35
+ field_dict["low_effort_agent_iteration_budget"] = low_effort_agent_iteration_budget
36
+ if high_effort_agent_iteration_budget is not UNSET:
37
+ field_dict["high_effort_agent_iteration_budget"] = high_effort_agent_iteration_budget
38
+
39
+ return field_dict
40
+
41
+ @classmethod
42
+ def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
43
+ d = dict(src_dict)
44
+ low_effort_agent_iteration_budget = cast(Literal[5] | Unset, d.pop("low_effort_agent_iteration_budget", UNSET))
45
+ if low_effort_agent_iteration_budget != 5 and not isinstance(low_effort_agent_iteration_budget, Unset):
46
+ raise ValueError(
47
+ f"low_effort_agent_iteration_budget must match const 5, got '{low_effort_agent_iteration_budget}'"
48
+ )
49
+
50
+ high_effort_agent_iteration_budget = cast(
51
+ Literal[10] | Unset, d.pop("high_effort_agent_iteration_budget", UNSET)
52
+ )
53
+ if high_effort_agent_iteration_budget != 10 and not isinstance(high_effort_agent_iteration_budget, Unset):
54
+ raise ValueError(
55
+ f"high_effort_agent_iteration_budget must match const 10, got '{high_effort_agent_iteration_budget}'"
56
+ )
57
+
58
+ toolkit_constants = cls(
59
+ low_effort_agent_iteration_budget=low_effort_agent_iteration_budget,
60
+ high_effort_agent_iteration_budget=high_effort_agent_iteration_budget,
61
+ )
62
+
63
+ toolkit_constants.additional_properties = d
64
+ return toolkit_constants
65
+
66
+ @property
67
+ def additional_keys(self) -> list[str]:
68
+ return list(self.additional_properties.keys())
69
+
70
+ def __getitem__(self, key: str) -> Any:
71
+ return self.additional_properties[key]
72
+
73
+ def __setitem__(self, key: str, value: Any) -> None:
74
+ self.additional_properties[key] = value
75
+
76
+ def __delitem__(self, key: str) -> None:
77
+ del self.additional_properties[key]
78
+
79
+ def __contains__(self, key: str) -> bool:
80
+ return key in self.additional_properties
@@ -0,0 +1,94 @@
1
+ from __future__ import annotations
2
+
3
+ from collections.abc import Mapping
4
+ from typing import TYPE_CHECKING, 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
+ if TYPE_CHECKING:
13
+ from ..models.trace_info import TraceInfo
14
+
15
+
16
+ T = TypeVar("T", bound="TraceChangedPayload")
17
+
18
+
19
+ @_attrs_define
20
+ class TraceChangedPayload:
21
+ """
22
+ Attributes:
23
+ traces (list[TraceInfo]):
24
+ event_type (EventType | Unset):
25
+ """
26
+
27
+ traces: list[TraceInfo]
28
+ event_type: EventType | Unset = UNSET
29
+ additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
30
+
31
+ def to_dict(self) -> dict[str, Any]:
32
+ traces = []
33
+ for traces_item_data in self.traces:
34
+ traces_item = traces_item_data.to_dict()
35
+ traces.append(traces_item)
36
+
37
+ event_type: str | Unset = UNSET
38
+ if not isinstance(self.event_type, Unset):
39
+ event_type = self.event_type.value
40
+
41
+ field_dict: dict[str, Any] = {}
42
+ field_dict.update(self.additional_properties)
43
+ field_dict.update(
44
+ {
45
+ "traces": traces,
46
+ }
47
+ )
48
+ if event_type is not UNSET:
49
+ field_dict["event_type"] = event_type
50
+
51
+ return field_dict
52
+
53
+ @classmethod
54
+ def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
55
+ from ..models.trace_info import TraceInfo
56
+
57
+ d = dict(src_dict)
58
+ traces = []
59
+ _traces = d.pop("traces")
60
+ for traces_item_data in _traces:
61
+ traces_item = TraceInfo.from_dict(traces_item_data)
62
+
63
+ traces.append(traces_item)
64
+
65
+ _event_type = d.pop("event_type", UNSET)
66
+ event_type: EventType | Unset
67
+ if isinstance(_event_type, Unset):
68
+ event_type = UNSET
69
+ else:
70
+ event_type = EventType(_event_type)
71
+
72
+ trace_changed_payload = cls(
73
+ traces=traces,
74
+ event_type=event_type,
75
+ )
76
+
77
+ trace_changed_payload.additional_properties = d
78
+ return trace_changed_payload
79
+
80
+ @property
81
+ def additional_keys(self) -> list[str]:
82
+ return list(self.additional_properties.keys())
83
+
84
+ def __getitem__(self, key: str) -> Any:
85
+ return self.additional_properties[key]
86
+
87
+ def __setitem__(self, key: str, value: Any) -> None:
88
+ self.additional_properties[key] = value
89
+
90
+ def __delitem__(self, key: str) -> None:
91
+ del self.additional_properties[key]
92
+
93
+ def __contains__(self, key: str) -> bool:
94
+ return key in self.additional_properties
@@ -0,0 +1,78 @@
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="TraceInfo")
11
+
12
+
13
+ @_attrs_define
14
+ class TraceInfo:
15
+ """
16
+ Attributes:
17
+ task_id (UUID):
18
+ trace_id (UUID):
19
+ artifact_id (UUID):
20
+ """
21
+
22
+ task_id: UUID
23
+ trace_id: UUID
24
+ artifact_id: UUID
25
+ additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
26
+
27
+ def to_dict(self) -> dict[str, Any]:
28
+ task_id = str(self.task_id)
29
+
30
+ trace_id = str(self.trace_id)
31
+
32
+ artifact_id = str(self.artifact_id)
33
+
34
+ field_dict: dict[str, Any] = {}
35
+ field_dict.update(self.additional_properties)
36
+ field_dict.update(
37
+ {
38
+ "task_id": task_id,
39
+ "trace_id": trace_id,
40
+ "artifact_id": artifact_id,
41
+ }
42
+ )
43
+
44
+ return field_dict
45
+
46
+ @classmethod
47
+ def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
48
+ d = dict(src_dict)
49
+ task_id = UUID(d.pop("task_id"))
50
+
51
+ trace_id = UUID(d.pop("trace_id"))
52
+
53
+ artifact_id = UUID(d.pop("artifact_id"))
54
+
55
+ trace_info = cls(
56
+ task_id=task_id,
57
+ trace_id=trace_id,
58
+ artifact_id=artifact_id,
59
+ )
60
+
61
+ trace_info.additional_properties = d
62
+ return trace_info
63
+
64
+ @property
65
+ def additional_keys(self) -> list[str]:
66
+ return list(self.additional_properties.keys())
67
+
68
+ def __getitem__(self, key: str) -> Any:
69
+ return self.additional_properties[key]
70
+
71
+ def __setitem__(self, key: str, value: Any) -> None:
72
+ self.additional_properties[key] = value
73
+
74
+ def __delitem__(self, key: str) -> None:
75
+ del self.additional_properties[key]
76
+
77
+ def __contains__(self, key: str) -> bool:
78
+ return key in self.additional_properties
@@ -0,0 +1,112 @@
1
+ from __future__ import annotations
2
+
3
+ from collections.abc import Mapping
4
+ from typing import TYPE_CHECKING, 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
+ from ..types import UNSET, Unset
11
+
12
+ if TYPE_CHECKING:
13
+ from ..models.trigger_workflow_execution_request_task_params import TriggerWorkflowExecutionRequestTaskParams
14
+ from ..models.workflow_leaf_node_input import WorkflowLeafNodeInput
15
+
16
+
17
+ T = TypeVar("T", bound="TriggerWorkflowExecutionRequest")
18
+
19
+
20
+ @_attrs_define
21
+ class TriggerWorkflowExecutionRequest:
22
+ """
23
+ Attributes:
24
+ workflow_id (UUID):
25
+ inputs (list[WorkflowLeafNodeInput]):
26
+ session_id (UUID):
27
+ task_params (TriggerWorkflowExecutionRequestTaskParams | Unset):
28
+ """
29
+
30
+ workflow_id: UUID
31
+ inputs: list[WorkflowLeafNodeInput]
32
+ session_id: UUID
33
+ task_params: TriggerWorkflowExecutionRequestTaskParams | Unset = UNSET
34
+ additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
35
+
36
+ def to_dict(self) -> dict[str, Any]:
37
+ workflow_id = str(self.workflow_id)
38
+
39
+ inputs = []
40
+ for inputs_item_data in self.inputs:
41
+ inputs_item = inputs_item_data.to_dict()
42
+ inputs.append(inputs_item)
43
+
44
+ session_id = str(self.session_id)
45
+
46
+ task_params: dict[str, Any] | Unset = UNSET
47
+ if not isinstance(self.task_params, Unset):
48
+ task_params = self.task_params.to_dict()
49
+
50
+ field_dict: dict[str, Any] = {}
51
+ field_dict.update(self.additional_properties)
52
+ field_dict.update(
53
+ {
54
+ "workflow_id": workflow_id,
55
+ "inputs": inputs,
56
+ "session_id": session_id,
57
+ }
58
+ )
59
+ if task_params is not UNSET:
60
+ field_dict["task_params"] = task_params
61
+
62
+ return field_dict
63
+
64
+ @classmethod
65
+ def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
66
+ from ..models.trigger_workflow_execution_request_task_params import TriggerWorkflowExecutionRequestTaskParams
67
+ from ..models.workflow_leaf_node_input import WorkflowLeafNodeInput
68
+
69
+ d = dict(src_dict)
70
+ workflow_id = UUID(d.pop("workflow_id"))
71
+
72
+ inputs = []
73
+ _inputs = d.pop("inputs")
74
+ for inputs_item_data in _inputs:
75
+ inputs_item = WorkflowLeafNodeInput.from_dict(inputs_item_data)
76
+
77
+ inputs.append(inputs_item)
78
+
79
+ session_id = UUID(d.pop("session_id"))
80
+
81
+ _task_params = d.pop("task_params", UNSET)
82
+ task_params: TriggerWorkflowExecutionRequestTaskParams | Unset
83
+ if isinstance(_task_params, Unset):
84
+ task_params = UNSET
85
+ else:
86
+ task_params = TriggerWorkflowExecutionRequestTaskParams.from_dict(_task_params)
87
+
88
+ trigger_workflow_execution_request = cls(
89
+ workflow_id=workflow_id,
90
+ inputs=inputs,
91
+ session_id=session_id,
92
+ task_params=task_params,
93
+ )
94
+
95
+ trigger_workflow_execution_request.additional_properties = d
96
+ return trigger_workflow_execution_request
97
+
98
+ @property
99
+ def additional_keys(self) -> list[str]:
100
+ return list(self.additional_properties.keys())
101
+
102
+ def __getitem__(self, key: str) -> Any:
103
+ return self.additional_properties[key]
104
+
105
+ def __setitem__(self, key: str, value: Any) -> None:
106
+ self.additional_properties[key] = value
107
+
108
+ def __delitem__(self, key: str) -> None:
109
+ del self.additional_properties[key]
110
+
111
+ def __contains__(self, key: str) -> bool:
112
+ return key in self.additional_properties
@@ -0,0 +1,65 @@
1
+ from __future__ import annotations
2
+
3
+ from collections.abc import Mapping
4
+ from typing import TYPE_CHECKING, Any, TypeVar
5
+
6
+ from attrs import define as _attrs_define
7
+ from attrs import field as _attrs_field
8
+
9
+ if TYPE_CHECKING:
10
+ from ..models.trigger_workflow_execution_request_task_params_additional_property import (
11
+ TriggerWorkflowExecutionRequestTaskParamsAdditionalProperty,
12
+ )
13
+
14
+
15
+ T = TypeVar("T", bound="TriggerWorkflowExecutionRequestTaskParams")
16
+
17
+
18
+ @_attrs_define
19
+ class TriggerWorkflowExecutionRequestTaskParams:
20
+ """ """
21
+
22
+ additional_properties: dict[str, TriggerWorkflowExecutionRequestTaskParamsAdditionalProperty] = _attrs_field(
23
+ init=False, factory=dict
24
+ )
25
+
26
+ def to_dict(self) -> dict[str, Any]:
27
+ field_dict: dict[str, Any] = {}
28
+ for prop_name, prop in self.additional_properties.items():
29
+ field_dict[prop_name] = prop.to_dict()
30
+
31
+ return field_dict
32
+
33
+ @classmethod
34
+ def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
35
+ from ..models.trigger_workflow_execution_request_task_params_additional_property import (
36
+ TriggerWorkflowExecutionRequestTaskParamsAdditionalProperty,
37
+ )
38
+
39
+ d = dict(src_dict)
40
+ trigger_workflow_execution_request_task_params = cls()
41
+
42
+ additional_properties = {}
43
+ for prop_name, prop_dict in d.items():
44
+ additional_property = TriggerWorkflowExecutionRequestTaskParamsAdditionalProperty.from_dict(prop_dict)
45
+
46
+ additional_properties[prop_name] = additional_property
47
+
48
+ trigger_workflow_execution_request_task_params.additional_properties = additional_properties
49
+ return trigger_workflow_execution_request_task_params
50
+
51
+ @property
52
+ def additional_keys(self) -> list[str]:
53
+ return list(self.additional_properties.keys())
54
+
55
+ def __getitem__(self, key: str) -> TriggerWorkflowExecutionRequestTaskParamsAdditionalProperty:
56
+ return self.additional_properties[key]
57
+
58
+ def __setitem__(self, key: str, value: TriggerWorkflowExecutionRequestTaskParamsAdditionalProperty) -> None:
59
+ self.additional_properties[key] = value
60
+
61
+ def __delitem__(self, key: str) -> None:
62
+ del self.additional_properties[key]
63
+
64
+ def __contains__(self, key: str) -> bool:
65
+ return key in self.additional_properties
@@ -0,0 +1,46 @@
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="TriggerWorkflowExecutionRequestTaskParamsAdditionalProperty")
10
+
11
+
12
+ @_attrs_define
13
+ class TriggerWorkflowExecutionRequestTaskParamsAdditionalProperty:
14
+ """ """
15
+
16
+ additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
17
+
18
+ def to_dict(self) -> dict[str, Any]:
19
+ field_dict: dict[str, Any] = {}
20
+ field_dict.update(self.additional_properties)
21
+
22
+ return field_dict
23
+
24
+ @classmethod
25
+ def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
26
+ d = dict(src_dict)
27
+ trigger_workflow_execution_request_task_params_additional_property = cls()
28
+
29
+ trigger_workflow_execution_request_task_params_additional_property.additional_properties = d
30
+ return trigger_workflow_execution_request_task_params_additional_property
31
+
32
+ @property
33
+ def additional_keys(self) -> list[str]:
34
+ return list(self.additional_properties.keys())
35
+
36
+ def __getitem__(self, key: str) -> Any:
37
+ return self.additional_properties[key]
38
+
39
+ def __setitem__(self, key: str, value: Any) -> None:
40
+ self.additional_properties[key] = value
41
+
42
+ def __delitem__(self, key: str) -> None:
43
+ del self.additional_properties[key]
44
+
45
+ def __contains__(self, key: str) -> bool:
46
+ return key in self.additional_properties