windmill-api 1.563.3__py3-none-any.whl → 1.564.0__py3-none-any.whl

windmill_api/models/list_completed_jobs_response_200_item_flow_status_modules_item_agent_actions_item_type_1.py

@@ -1,4 +1,4 @@
-from typing import Any, Dict, List, Type, TypeVar
+from typing import TYPE_CHECKING, Any, Dict, List, Type, TypeVar, Union
 
 from attrs import define as _attrs_define
 from attrs import field as _attrs_field
@@ -6,6 +6,13 @@ from attrs import field as _attrs_field
 from ..models.list_completed_jobs_response_200_item_flow_status_modules_item_agent_actions_item_type_1_type import (
     ListCompletedJobsResponse200ItemFlowStatusModulesItemAgentActionsItemType1Type,
 )
+from ..types import UNSET, Unset
+
+if TYPE_CHECKING:
+    from ..models.list_completed_jobs_response_200_item_flow_status_modules_item_agent_actions_item_type_1_arguments import (
+        ListCompletedJobsResponse200ItemFlowStatusModulesItemAgentActionsItemType1Arguments,
+    )
+
 
 T = TypeVar("T", bound="ListCompletedJobsResponse200ItemFlowStatusModulesItemAgentActionsItemType1")
 
@@ -14,32 +21,77 @@ T = TypeVar("T", bound="ListCompletedJobsResponse200ItemFlowStatusModulesItemAge
 class ListCompletedJobsResponse200ItemFlowStatusModulesItemAgentActionsItemType1:
     """
     Attributes:
+        call_id (str):
+        function_name (str):
+        resource_path (str):
         type (ListCompletedJobsResponse200ItemFlowStatusModulesItemAgentActionsItemType1Type):
+        arguments (Union[Unset, ListCompletedJobsResponse200ItemFlowStatusModulesItemAgentActionsItemType1Arguments]):
     """
 
+    call_id: str
+    function_name: str
+    resource_path: str
     type: ListCompletedJobsResponse200ItemFlowStatusModulesItemAgentActionsItemType1Type
+    arguments: Union[
+        Unset, "ListCompletedJobsResponse200ItemFlowStatusModulesItemAgentActionsItemType1Arguments"
+    ] = UNSET
     additional_properties: Dict[str, Any] = _attrs_field(init=False, factory=dict)
 
     def to_dict(self) -> Dict[str, Any]:
+        call_id = self.call_id
+        function_name = self.function_name
+        resource_path = self.resource_path
         type = self.type.value
 
+        arguments: Union[Unset, Dict[str, Any]] = UNSET
+        if not isinstance(self.arguments, Unset):
+            arguments = self.arguments.to_dict()
+
         field_dict: Dict[str, Any] = {}
         field_dict.update(self.additional_properties)
         field_dict.update(
             {
+                "call_id": call_id,
+                "function_name": function_name,
+                "resource_path": resource_path,
                 "type": type,
             }
         )
+        if arguments is not UNSET:
+            field_dict["arguments"] = arguments
 
         return field_dict
 
     @classmethod
     def from_dict(cls: Type[T], src_dict: Dict[str, Any]) -> T:
+        from ..models.list_completed_jobs_response_200_item_flow_status_modules_item_agent_actions_item_type_1_arguments import (
+            ListCompletedJobsResponse200ItemFlowStatusModulesItemAgentActionsItemType1Arguments,
+        )
+
         d = src_dict.copy()
+        call_id = d.pop("call_id")
+
+        function_name = d.pop("function_name")
+
+        resource_path = d.pop("resource_path")
+
         type = ListCompletedJobsResponse200ItemFlowStatusModulesItemAgentActionsItemType1Type(d.pop("type"))
 
+        _arguments = d.pop("arguments", UNSET)
+        arguments: Union[Unset, ListCompletedJobsResponse200ItemFlowStatusModulesItemAgentActionsItemType1Arguments]
+        if isinstance(_arguments, Unset):
+            arguments = UNSET
+        else:
+            arguments = ListCompletedJobsResponse200ItemFlowStatusModulesItemAgentActionsItemType1Arguments.from_dict(
+                _arguments
+            )
+
         list_completed_jobs_response_200_item_flow_status_modules_item_agent_actions_item_type_1 = cls(
+            call_id=call_id,
+            function_name=function_name,
+            resource_path=resource_path,
             type=type,
+            arguments=arguments,
         )
 
         list_completed_jobs_response_200_item_flow_status_modules_item_agent_actions_item_type_1.additional_properties = (

windmill_api/models/list_completed_jobs_response_200_item_flow_status_modules_item_agent_actions_item_type_1_arguments.py

@@ -0,0 +1,46 @@
+from typing import Any, Dict, List, Type, TypeVar
+
+from attrs import define as _attrs_define
+from attrs import field as _attrs_field
+
+T = TypeVar("T", bound="ListCompletedJobsResponse200ItemFlowStatusModulesItemAgentActionsItemType1Arguments")
+
+
+@_attrs_define
+class ListCompletedJobsResponse200ItemFlowStatusModulesItemAgentActionsItemType1Arguments:
+    """ """
+
+    additional_properties: Dict[str, Any] = _attrs_field(init=False, factory=dict)
+
+    def to_dict(self) -> Dict[str, Any]:
+        field_dict: Dict[str, Any] = {}
+        field_dict.update(self.additional_properties)
+        field_dict.update({})
+
+        return field_dict
+
+    @classmethod
+    def from_dict(cls: Type[T], src_dict: Dict[str, Any]) -> T:
+        d = src_dict.copy()
+        list_completed_jobs_response_200_item_flow_status_modules_item_agent_actions_item_type_1_arguments = cls()
+
+        list_completed_jobs_response_200_item_flow_status_modules_item_agent_actions_item_type_1_arguments.additional_properties = (
+            d
+        )
+        return list_completed_jobs_response_200_item_flow_status_modules_item_agent_actions_item_type_1_arguments
+
+    @property
+    def additional_keys(self) -> List[str]:
+        return list(self.additional_properties.keys())
+
+    def __getitem__(self, key: str) -> Any:
+        return self.additional_properties[key]
+
+    def __setitem__(self, key: str, value: Any) -> None:
+        self.additional_properties[key] = value
+
+    def __delitem__(self, key: str) -> None:
+        del self.additional_properties[key]
+
+    def __contains__(self, key: str) -> bool:
+        return key in self.additional_properties

windmill_api/models/list_completed_jobs_response_200_item_flow_status_modules_item_agent_actions_item_type_1_type.py

@@ -2,7 +2,7 @@ from enum import Enum
 
 
 class ListCompletedJobsResponse200ItemFlowStatusModulesItemAgentActionsItemType1Type(str, Enum):
-    MESSAGE = "message"
+    MCP_TOOL_CALL = "mcp_tool_call"
 
     def __str__(self) -> str:
         return str(self.value)

windmill_api/models/list_completed_jobs_response_200_item_flow_status_modules_item_agent_actions_item_type_2.py

@@ -0,0 +1,64 @@
+from typing import Any, Dict, List, Type, TypeVar
+
+from attrs import define as _attrs_define
+from attrs import field as _attrs_field
+
+from ..models.list_completed_jobs_response_200_item_flow_status_modules_item_agent_actions_item_type_2_type import (
+    ListCompletedJobsResponse200ItemFlowStatusModulesItemAgentActionsItemType2Type,
+)
+
+T = TypeVar("T", bound="ListCompletedJobsResponse200ItemFlowStatusModulesItemAgentActionsItemType2")
+
+
+@_attrs_define
+class ListCompletedJobsResponse200ItemFlowStatusModulesItemAgentActionsItemType2:
+    """
+    Attributes:
+        type (ListCompletedJobsResponse200ItemFlowStatusModulesItemAgentActionsItemType2Type):
+    """
+
+    type: ListCompletedJobsResponse200ItemFlowStatusModulesItemAgentActionsItemType2Type
+    additional_properties: Dict[str, Any] = _attrs_field(init=False, factory=dict)
+
+    def to_dict(self) -> Dict[str, Any]:
+        type = self.type.value
+
+        field_dict: Dict[str, Any] = {}
+        field_dict.update(self.additional_properties)
+        field_dict.update(
+            {
+                "type": type,
+            }
+        )
+
+        return field_dict
+
+    @classmethod
+    def from_dict(cls: Type[T], src_dict: Dict[str, Any]) -> T:
+        d = src_dict.copy()
+        type = ListCompletedJobsResponse200ItemFlowStatusModulesItemAgentActionsItemType2Type(d.pop("type"))
+
+        list_completed_jobs_response_200_item_flow_status_modules_item_agent_actions_item_type_2 = cls(
+            type=type,
+        )
+
+        list_completed_jobs_response_200_item_flow_status_modules_item_agent_actions_item_type_2.additional_properties = (
+            d
+        )
+        return list_completed_jobs_response_200_item_flow_status_modules_item_agent_actions_item_type_2
+
+    @property
+    def additional_keys(self) -> List[str]:
+        return list(self.additional_properties.keys())
+
+    def __getitem__(self, key: str) -> Any:
+        return self.additional_properties[key]
+
+    def __setitem__(self, key: str, value: Any) -> None:
+        self.additional_properties[key] = value
+
+    def __delitem__(self, key: str) -> None:
+        del self.additional_properties[key]
+
+    def __contains__(self, key: str) -> bool:
+        return key in self.additional_properties

windmill_api/models/list_completed_jobs_response_200_item_flow_status_modules_item_agent_actions_item_type_2_type.py

@@ -0,0 +1,8 @@
+from enum import Enum
+
+
+class ListCompletedJobsResponse200ItemFlowStatusModulesItemAgentActionsItemType2Type(str, Enum):
+    MESSAGE = "message"
+
+    def __str__(self) -> str:
+        return str(self.value)

windmill_api/models/list_completed_jobs_response_200_item_flow_status_preprocessor_module.py

@@ -15,6 +15,9 @@ if TYPE_CHECKING:
     from ..models.list_completed_jobs_response_200_item_flow_status_preprocessor_module_agent_actions_item_type_1 import (
         ListCompletedJobsResponse200ItemFlowStatusPreprocessorModuleAgentActionsItemType1,
     )
+    from ..models.list_completed_jobs_response_200_item_flow_status_preprocessor_module_agent_actions_item_type_2 import (
+        ListCompletedJobsResponse200ItemFlowStatusPreprocessorModuleAgentActionsItemType2,
+    )
     from ..models.list_completed_jobs_response_200_item_flow_status_preprocessor_module_approvers_item import (
         ListCompletedJobsResponse200ItemFlowStatusPreprocessorModuleApproversItem,
     )
@@ -55,7 +58,8 @@ class ListCompletedJobsResponse200ItemFlowStatusPreprocessorModule:
         skipped (Union[Unset, bool]):
         agent_actions (Union[Unset,
             List[Union['ListCompletedJobsResponse200ItemFlowStatusPreprocessorModuleAgentActionsItemType0',
-            'ListCompletedJobsResponse200ItemFlowStatusPreprocessorModuleAgentActionsItemType1']]]):
+            'ListCompletedJobsResponse200ItemFlowStatusPreprocessorModuleAgentActionsItemType1',
+            'ListCompletedJobsResponse200ItemFlowStatusPreprocessorModuleAgentActionsItemType2']]]):
         agent_actions_success (Union[Unset, List[bool]]):
     """
 
@@ -81,6 +85,7 @@ class ListCompletedJobsResponse200ItemFlowStatusPreprocessorModule:
             Union[
                 "ListCompletedJobsResponse200ItemFlowStatusPreprocessorModuleAgentActionsItemType0",
                 "ListCompletedJobsResponse200ItemFlowStatusPreprocessorModuleAgentActionsItemType1",
+                "ListCompletedJobsResponse200ItemFlowStatusPreprocessorModuleAgentActionsItemType2",
             ]
         ],
     ] = UNSET
@@ -91,6 +96,9 @@ class ListCompletedJobsResponse200ItemFlowStatusPreprocessorModule:
         from ..models.list_completed_jobs_response_200_item_flow_status_preprocessor_module_agent_actions_item_type_0 import (
             ListCompletedJobsResponse200ItemFlowStatusPreprocessorModuleAgentActionsItemType0,
         )
+        from ..models.list_completed_jobs_response_200_item_flow_status_preprocessor_module_agent_actions_item_type_1 import (
+            ListCompletedJobsResponse200ItemFlowStatusPreprocessorModuleAgentActionsItemType1,
+        )
 
         type = self.type.value
 
@@ -147,6 +155,12 @@ class ListCompletedJobsResponse200ItemFlowStatusPreprocessorModule:
                 ):
                     agent_actions_item = agent_actions_item_data.to_dict()
 
+                elif isinstance(
+                    agent_actions_item_data,
+                    ListCompletedJobsResponse200ItemFlowStatusPreprocessorModuleAgentActionsItemType1,
+                ):
+                    agent_actions_item = agent_actions_item_data.to_dict()
+
                 else:
                     agent_actions_item = agent_actions_item_data.to_dict()
 
@@ -204,6 +218,9 @@ class ListCompletedJobsResponse200ItemFlowStatusPreprocessorModule:
         from ..models.list_completed_jobs_response_200_item_flow_status_preprocessor_module_agent_actions_item_type_1 import (
             ListCompletedJobsResponse200ItemFlowStatusPreprocessorModuleAgentActionsItemType1,
         )
+        from ..models.list_completed_jobs_response_200_item_flow_status_preprocessor_module_agent_actions_item_type_2 import (
+            ListCompletedJobsResponse200ItemFlowStatusPreprocessorModuleAgentActionsItemType2,
+        )
         from ..models.list_completed_jobs_response_200_item_flow_status_preprocessor_module_approvers_item import (
             ListCompletedJobsResponse200ItemFlowStatusPreprocessorModuleApproversItem,
         )
@@ -289,6 +306,7 @@ class ListCompletedJobsResponse200ItemFlowStatusPreprocessorModule:
             ) -> Union[
                 "ListCompletedJobsResponse200ItemFlowStatusPreprocessorModuleAgentActionsItemType0",
                 "ListCompletedJobsResponse200ItemFlowStatusPreprocessorModuleAgentActionsItemType1",
+                "ListCompletedJobsResponse200ItemFlowStatusPreprocessorModuleAgentActionsItemType2",
             ]:
                 try:
                     if not isinstance(data, dict):
@@ -302,13 +320,25 @@ class ListCompletedJobsResponse200ItemFlowStatusPreprocessorModule:
                     return agent_actions_item_type_0
                 except:  # noqa: E722
                     pass
+                try:
+                    if not isinstance(data, dict):
+                        raise TypeError()
+                    agent_actions_item_type_1 = (
+                        ListCompletedJobsResponse200ItemFlowStatusPreprocessorModuleAgentActionsItemType1.from_dict(
+                            data
+                        )
+                    )
+
+                    return agent_actions_item_type_1
+                except:  # noqa: E722
+                    pass
                 if not isinstance(data, dict):
                     raise TypeError()
-                agent_actions_item_type_1 = (
-                    ListCompletedJobsResponse200ItemFlowStatusPreprocessorModuleAgentActionsItemType1.from_dict(data)
+                agent_actions_item_type_2 = (
+                    ListCompletedJobsResponse200ItemFlowStatusPreprocessorModuleAgentActionsItemType2.from_dict(data)
                 )
 
-                return agent_actions_item_type_1
+                return agent_actions_item_type_2
 
             agent_actions_item = _parse_agent_actions_item(agent_actions_item_data)
 

windmill_api/models/list_completed_jobs_response_200_item_flow_status_preprocessor_module_agent_actions_item_type_1.py

@@ -1,4 +1,4 @@
-from typing import Any, Dict, List, Type, TypeVar
+from typing import TYPE_CHECKING, Any, Dict, List, Type, TypeVar, Union
 
 from attrs import define as _attrs_define
 from attrs import field as _attrs_field
@@ -6,6 +6,13 @@ from attrs import field as _attrs_field
 from ..models.list_completed_jobs_response_200_item_flow_status_preprocessor_module_agent_actions_item_type_1_type import (
     ListCompletedJobsResponse200ItemFlowStatusPreprocessorModuleAgentActionsItemType1Type,
 )
+from ..types import UNSET, Unset
+
+if TYPE_CHECKING:
+    from ..models.list_completed_jobs_response_200_item_flow_status_preprocessor_module_agent_actions_item_type_1_arguments import (
+        ListCompletedJobsResponse200ItemFlowStatusPreprocessorModuleAgentActionsItemType1Arguments,
+    )
+
 
 T = TypeVar("T", bound="ListCompletedJobsResponse200ItemFlowStatusPreprocessorModuleAgentActionsItemType1")
 
@@ -14,32 +21,82 @@ T = TypeVar("T", bound="ListCompletedJobsResponse200ItemFlowStatusPreprocessorMo
 class ListCompletedJobsResponse200ItemFlowStatusPreprocessorModuleAgentActionsItemType1:
     """
     Attributes:
+        call_id (str):
+        function_name (str):
+        resource_path (str):
         type (ListCompletedJobsResponse200ItemFlowStatusPreprocessorModuleAgentActionsItemType1Type):
+        arguments (Union[Unset,
+            ListCompletedJobsResponse200ItemFlowStatusPreprocessorModuleAgentActionsItemType1Arguments]):
     """
 
+    call_id: str
+    function_name: str
+    resource_path: str
     type: ListCompletedJobsResponse200ItemFlowStatusPreprocessorModuleAgentActionsItemType1Type
+    arguments: Union[
+        Unset, "ListCompletedJobsResponse200ItemFlowStatusPreprocessorModuleAgentActionsItemType1Arguments"
+    ] = UNSET
     additional_properties: Dict[str, Any] = _attrs_field(init=False, factory=dict)
 
     def to_dict(self) -> Dict[str, Any]:
+        call_id = self.call_id
+        function_name = self.function_name
+        resource_path = self.resource_path
         type = self.type.value
 
+        arguments: Union[Unset, Dict[str, Any]] = UNSET
+        if not isinstance(self.arguments, Unset):
+            arguments = self.arguments.to_dict()
+
         field_dict: Dict[str, Any] = {}
         field_dict.update(self.additional_properties)
         field_dict.update(
             {
+                "call_id": call_id,
+                "function_name": function_name,
+                "resource_path": resource_path,
                 "type": type,
             }
         )
+        if arguments is not UNSET:
+            field_dict["arguments"] = arguments
 
         return field_dict
 
     @classmethod
     def from_dict(cls: Type[T], src_dict: Dict[str, Any]) -> T:
+        from ..models.list_completed_jobs_response_200_item_flow_status_preprocessor_module_agent_actions_item_type_1_arguments import (
+            ListCompletedJobsResponse200ItemFlowStatusPreprocessorModuleAgentActionsItemType1Arguments,
+        )
+
         d = src_dict.copy()
+        call_id = d.pop("call_id")
+
+        function_name = d.pop("function_name")
+
+        resource_path = d.pop("resource_path")
+
         type = ListCompletedJobsResponse200ItemFlowStatusPreprocessorModuleAgentActionsItemType1Type(d.pop("type"))
 
+        _arguments = d.pop("arguments", UNSET)
+        arguments: Union[
+            Unset, ListCompletedJobsResponse200ItemFlowStatusPreprocessorModuleAgentActionsItemType1Arguments
+        ]
+        if isinstance(_arguments, Unset):
+            arguments = UNSET
+        else:
+            arguments = (
+                ListCompletedJobsResponse200ItemFlowStatusPreprocessorModuleAgentActionsItemType1Arguments.from_dict(
+                    _arguments
+                )
+            )
+
         list_completed_jobs_response_200_item_flow_status_preprocessor_module_agent_actions_item_type_1 = cls(
+            call_id=call_id,
+            function_name=function_name,
+            resource_path=resource_path,
             type=type,
+            arguments=arguments,
         )
 
         list_completed_jobs_response_200_item_flow_status_preprocessor_module_agent_actions_item_type_1.additional_properties = (

windmill_api/models/list_completed_jobs_response_200_item_flow_status_preprocessor_module_agent_actions_item_type_1_arguments.py

@@ -0,0 +1,48 @@
+from typing import Any, Dict, List, Type, TypeVar
+
+from attrs import define as _attrs_define
+from attrs import field as _attrs_field
+
+T = TypeVar("T", bound="ListCompletedJobsResponse200ItemFlowStatusPreprocessorModuleAgentActionsItemType1Arguments")
+
+
+@_attrs_define
+class ListCompletedJobsResponse200ItemFlowStatusPreprocessorModuleAgentActionsItemType1Arguments:
+    """ """
+
+    additional_properties: Dict[str, Any] = _attrs_field(init=False, factory=dict)
+
+    def to_dict(self) -> Dict[str, Any]:
+        field_dict: Dict[str, Any] = {}
+        field_dict.update(self.additional_properties)
+        field_dict.update({})
+
+        return field_dict
+
+    @classmethod
+    def from_dict(cls: Type[T], src_dict: Dict[str, Any]) -> T:
+        d = src_dict.copy()
+        list_completed_jobs_response_200_item_flow_status_preprocessor_module_agent_actions_item_type_1_arguments = (
+            cls()
+        )
+
+        list_completed_jobs_response_200_item_flow_status_preprocessor_module_agent_actions_item_type_1_arguments.additional_properties = (
+            d
+        )
+        return list_completed_jobs_response_200_item_flow_status_preprocessor_module_agent_actions_item_type_1_arguments
+
+    @property
+    def additional_keys(self) -> List[str]:
+        return list(self.additional_properties.keys())
+
+    def __getitem__(self, key: str) -> Any:
+        return self.additional_properties[key]
+
+    def __setitem__(self, key: str, value: Any) -> None:
+        self.additional_properties[key] = value
+
+    def __delitem__(self, key: str) -> None:
+        del self.additional_properties[key]
+
+    def __contains__(self, key: str) -> bool:
+        return key in self.additional_properties

windmill_api/models/list_completed_jobs_response_200_item_flow_status_preprocessor_module_agent_actions_item_type_1_type.py

@@ -2,7 +2,7 @@ from enum import Enum
 
 
 class ListCompletedJobsResponse200ItemFlowStatusPreprocessorModuleAgentActionsItemType1Type(str, Enum):
-    MESSAGE = "message"
+    MCP_TOOL_CALL = "mcp_tool_call"
 
     def __str__(self) -> str:
         return str(self.value)

windmill_api/models/list_completed_jobs_response_200_item_flow_status_preprocessor_module_agent_actions_item_type_2.py

@@ -0,0 +1,64 @@
+from typing import Any, Dict, List, Type, TypeVar
+
+from attrs import define as _attrs_define
+from attrs import field as _attrs_field
+
+from ..models.list_completed_jobs_response_200_item_flow_status_preprocessor_module_agent_actions_item_type_2_type import (
+    ListCompletedJobsResponse200ItemFlowStatusPreprocessorModuleAgentActionsItemType2Type,
+)
+
+T = TypeVar("T", bound="ListCompletedJobsResponse200ItemFlowStatusPreprocessorModuleAgentActionsItemType2")
+
+
+@_attrs_define
+class ListCompletedJobsResponse200ItemFlowStatusPreprocessorModuleAgentActionsItemType2:
+    """
+    Attributes:
+        type (ListCompletedJobsResponse200ItemFlowStatusPreprocessorModuleAgentActionsItemType2Type):
+    """
+
+    type: ListCompletedJobsResponse200ItemFlowStatusPreprocessorModuleAgentActionsItemType2Type
+    additional_properties: Dict[str, Any] = _attrs_field(init=False, factory=dict)
+
+    def to_dict(self) -> Dict[str, Any]:
+        type = self.type.value
+
+        field_dict: Dict[str, Any] = {}
+        field_dict.update(self.additional_properties)
+        field_dict.update(
+            {
+                "type": type,
+            }
+        )
+
+        return field_dict
+
+    @classmethod
+    def from_dict(cls: Type[T], src_dict: Dict[str, Any]) -> T:
+        d = src_dict.copy()
+        type = ListCompletedJobsResponse200ItemFlowStatusPreprocessorModuleAgentActionsItemType2Type(d.pop("type"))
+
+        list_completed_jobs_response_200_item_flow_status_preprocessor_module_agent_actions_item_type_2 = cls(
+            type=type,
+        )
+
+        list_completed_jobs_response_200_item_flow_status_preprocessor_module_agent_actions_item_type_2.additional_properties = (
+            d
+        )
+        return list_completed_jobs_response_200_item_flow_status_preprocessor_module_agent_actions_item_type_2
+
+    @property
+    def additional_keys(self) -> List[str]:
+        return list(self.additional_properties.keys())
+
+    def __getitem__(self, key: str) -> Any:
+        return self.additional_properties[key]
+
+    def __setitem__(self, key: str, value: Any) -> None:
+        self.additional_properties[key] = value
+
+    def __delitem__(self, key: str) -> None:
+        del self.additional_properties[key]
+
+    def __contains__(self, key: str) -> bool:
+        return key in self.additional_properties

windmill_api/models/list_completed_jobs_response_200_item_flow_status_preprocessor_module_agent_actions_item_type_2_type.py

@@ -0,0 +1,8 @@
+from enum import Enum
+
+
+class ListCompletedJobsResponse200ItemFlowStatusPreprocessorModuleAgentActionsItemType2Type(str, Enum):
+    MESSAGE = "message"
+
+    def __str__(self) -> str:
+        return str(self.value)

windmill_api/models/list_extended_jobs_response_200_jobs_item_type_0_flow_status_failure_module.py

@@ -15,6 +15,9 @@ if TYPE_CHECKING:
     from ..models.list_extended_jobs_response_200_jobs_item_type_0_flow_status_failure_module_agent_actions_item_type_1 import (
         ListExtendedJobsResponse200JobsItemType0FlowStatusFailureModuleAgentActionsItemType1,
     )
+    from ..models.list_extended_jobs_response_200_jobs_item_type_0_flow_status_failure_module_agent_actions_item_type_2 import (
+        ListExtendedJobsResponse200JobsItemType0FlowStatusFailureModuleAgentActionsItemType2,
+    )
     from ..models.list_extended_jobs_response_200_jobs_item_type_0_flow_status_failure_module_approvers_item import (
         ListExtendedJobsResponse200JobsItemType0FlowStatusFailureModuleApproversItem,
     )
@@ -56,7 +59,8 @@ class ListExtendedJobsResponse200JobsItemType0FlowStatusFailureModule:
         skipped (Union[Unset, bool]):
         agent_actions (Union[Unset,
             List[Union['ListExtendedJobsResponse200JobsItemType0FlowStatusFailureModuleAgentActionsItemType0',
-            'ListExtendedJobsResponse200JobsItemType0FlowStatusFailureModuleAgentActionsItemType1']]]):
+            'ListExtendedJobsResponse200JobsItemType0FlowStatusFailureModuleAgentActionsItemType1',
+            'ListExtendedJobsResponse200JobsItemType0FlowStatusFailureModuleAgentActionsItemType2']]]):
         agent_actions_success (Union[Unset, List[bool]]):
         parent_module (Union[Unset, str]):
     """
@@ -85,6 +89,7 @@ class ListExtendedJobsResponse200JobsItemType0FlowStatusFailureModule:
             Union[
                 "ListExtendedJobsResponse200JobsItemType0FlowStatusFailureModuleAgentActionsItemType0",
                 "ListExtendedJobsResponse200JobsItemType0FlowStatusFailureModuleAgentActionsItemType1",
+                "ListExtendedJobsResponse200JobsItemType0FlowStatusFailureModuleAgentActionsItemType2",
             ]
         ],
     ] = UNSET
@@ -96,6 +101,9 @@ class ListExtendedJobsResponse200JobsItemType0FlowStatusFailureModule:
         from ..models.list_extended_jobs_response_200_jobs_item_type_0_flow_status_failure_module_agent_actions_item_type_0 import (
             ListExtendedJobsResponse200JobsItemType0FlowStatusFailureModuleAgentActionsItemType0,
         )
+        from ..models.list_extended_jobs_response_200_jobs_item_type_0_flow_status_failure_module_agent_actions_item_type_1 import (
+            ListExtendedJobsResponse200JobsItemType0FlowStatusFailureModuleAgentActionsItemType1,
+        )
 
         type = self.type.value
 
@@ -152,6 +160,12 @@ class ListExtendedJobsResponse200JobsItemType0FlowStatusFailureModule:
                 ):
                     agent_actions_item = agent_actions_item_data.to_dict()
 
+                elif isinstance(
+                    agent_actions_item_data,
+                    ListExtendedJobsResponse200JobsItemType0FlowStatusFailureModuleAgentActionsItemType1,
+                ):
+                    agent_actions_item = agent_actions_item_data.to_dict()
+
                 else:
                     agent_actions_item = agent_actions_item_data.to_dict()
 
@@ -213,6 +227,9 @@ class ListExtendedJobsResponse200JobsItemType0FlowStatusFailureModule:
         from ..models.list_extended_jobs_response_200_jobs_item_type_0_flow_status_failure_module_agent_actions_item_type_1 import (
             ListExtendedJobsResponse200JobsItemType0FlowStatusFailureModuleAgentActionsItemType1,
         )
+        from ..models.list_extended_jobs_response_200_jobs_item_type_0_flow_status_failure_module_agent_actions_item_type_2 import (
+            ListExtendedJobsResponse200JobsItemType0FlowStatusFailureModuleAgentActionsItemType2,
+        )
         from ..models.list_extended_jobs_response_200_jobs_item_type_0_flow_status_failure_module_approvers_item import (
             ListExtendedJobsResponse200JobsItemType0FlowStatusFailureModuleApproversItem,
         )
@@ -302,6 +319,7 @@ class ListExtendedJobsResponse200JobsItemType0FlowStatusFailureModule:
             ) -> Union[
                 "ListExtendedJobsResponse200JobsItemType0FlowStatusFailureModuleAgentActionsItemType0",
                 "ListExtendedJobsResponse200JobsItemType0FlowStatusFailureModuleAgentActionsItemType1",
+                "ListExtendedJobsResponse200JobsItemType0FlowStatusFailureModuleAgentActionsItemType2",
             ]:
                 try:
                     if not isinstance(data, dict):
@@ -315,13 +333,25 @@ class ListExtendedJobsResponse200JobsItemType0FlowStatusFailureModule:
                     return agent_actions_item_type_0
                 except:  # noqa: E722
                     pass
+                try:
+                    if not isinstance(data, dict):
+                        raise TypeError()
+                    agent_actions_item_type_1 = (
+                        ListExtendedJobsResponse200JobsItemType0FlowStatusFailureModuleAgentActionsItemType1.from_dict(
+                            data
+                        )
+                    )
+
+                    return agent_actions_item_type_1
+                except:  # noqa: E722
+                    pass
                 if not isinstance(data, dict):
                     raise TypeError()
-                agent_actions_item_type_1 = (
-                    ListExtendedJobsResponse200JobsItemType0FlowStatusFailureModuleAgentActionsItemType1.from_dict(data)
+                agent_actions_item_type_2 = (
+                    ListExtendedJobsResponse200JobsItemType0FlowStatusFailureModuleAgentActionsItemType2.from_dict(data)
                 )
 
-                return agent_actions_item_type_1
+                return agent_actions_item_type_2
 
             agent_actions_item = _parse_agent_actions_item(agent_actions_item_data)