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

windmill_api/models/flow_status_failure_module.py

@@ -13,6 +13,9 @@ if TYPE_CHECKING:
     from ..models.flow_status_failure_module_agent_actions_item_type_1 import (
         FlowStatusFailureModuleAgentActionsItemType1,
     )
+    from ..models.flow_status_failure_module_agent_actions_item_type_2 import (
+        FlowStatusFailureModuleAgentActionsItemType2,
+    )
     from ..models.flow_status_failure_module_approvers_item import FlowStatusFailureModuleApproversItem
     from ..models.flow_status_failure_module_branch_chosen import FlowStatusFailureModuleBranchChosen
     from ..models.flow_status_failure_module_branchall import FlowStatusFailureModuleBranchall
@@ -42,7 +45,7 @@ class FlowStatusFailureModule:
         failed_retries (Union[Unset, List[str]]):
         skipped (Union[Unset, bool]):
         agent_actions (Union[Unset, List[Union['FlowStatusFailureModuleAgentActionsItemType0',
-            'FlowStatusFailureModuleAgentActionsItemType1']]]):
+            'FlowStatusFailureModuleAgentActionsItemType1', 'FlowStatusFailureModuleAgentActionsItemType2']]]):
         agent_actions_success (Union[Unset, List[bool]]):
         parent_module (Union[Unset, str]):
     """
@@ -63,7 +66,13 @@ class FlowStatusFailureModule:
     skipped: Union[Unset, bool] = UNSET
     agent_actions: Union[
         Unset,
-        List[Union["FlowStatusFailureModuleAgentActionsItemType0", "FlowStatusFailureModuleAgentActionsItemType1"]],
+        List[
+            Union[
+                "FlowStatusFailureModuleAgentActionsItemType0",
+                "FlowStatusFailureModuleAgentActionsItemType1",
+                "FlowStatusFailureModuleAgentActionsItemType2",
+            ]
+        ],
     ] = UNSET
     agent_actions_success: Union[Unset, List[bool]] = UNSET
     parent_module: Union[Unset, str] = UNSET
@@ -73,6 +82,9 @@ class FlowStatusFailureModule:
         from ..models.flow_status_failure_module_agent_actions_item_type_0 import (
             FlowStatusFailureModuleAgentActionsItemType0,
         )
+        from ..models.flow_status_failure_module_agent_actions_item_type_1 import (
+            FlowStatusFailureModuleAgentActionsItemType1,
+        )
 
         type = self.type.value
 
@@ -126,6 +138,9 @@ class FlowStatusFailureModule:
                 if isinstance(agent_actions_item_data, FlowStatusFailureModuleAgentActionsItemType0):
                     agent_actions_item = agent_actions_item_data.to_dict()
 
+                elif isinstance(agent_actions_item_data, FlowStatusFailureModuleAgentActionsItemType1):
+                    agent_actions_item = agent_actions_item_data.to_dict()
+
                 else:
                     agent_actions_item = agent_actions_item_data.to_dict()
 
@@ -187,6 +202,9 @@ class FlowStatusFailureModule:
         from ..models.flow_status_failure_module_agent_actions_item_type_1 import (
             FlowStatusFailureModuleAgentActionsItemType1,
         )
+        from ..models.flow_status_failure_module_agent_actions_item_type_2 import (
+            FlowStatusFailureModuleAgentActionsItemType2,
+        )
         from ..models.flow_status_failure_module_approvers_item import FlowStatusFailureModuleApproversItem
         from ..models.flow_status_failure_module_branch_chosen import FlowStatusFailureModuleBranchChosen
         from ..models.flow_status_failure_module_branchall import FlowStatusFailureModuleBranchall
@@ -253,7 +271,11 @@ class FlowStatusFailureModule:
 
             def _parse_agent_actions_item(
                 data: object,
-            ) -> Union["FlowStatusFailureModuleAgentActionsItemType0", "FlowStatusFailureModuleAgentActionsItemType1"]:
+            ) -> Union[
+                "FlowStatusFailureModuleAgentActionsItemType0",
+                "FlowStatusFailureModuleAgentActionsItemType1",
+                "FlowStatusFailureModuleAgentActionsItemType2",
+            ]:
                 try:
                     if not isinstance(data, dict):
                         raise TypeError()
@@ -262,11 +284,19 @@ class FlowStatusFailureModule:
                     return agent_actions_item_type_0
                 except:  # noqa: E722
                     pass
+                try:
+                    if not isinstance(data, dict):
+                        raise TypeError()
+                    agent_actions_item_type_1 = FlowStatusFailureModuleAgentActionsItemType1.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 = FlowStatusFailureModuleAgentActionsItemType1.from_dict(data)
+                agent_actions_item_type_2 = FlowStatusFailureModuleAgentActionsItemType2.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/flow_status_failure_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.flow_status_failure_module_agent_actions_item_type_1_type import (
     FlowStatusFailureModuleAgentActionsItemType1Type,
 )
+from ..types import UNSET, Unset
+
+if TYPE_CHECKING:
+    from ..models.flow_status_failure_module_agent_actions_item_type_1_arguments import (
+        FlowStatusFailureModuleAgentActionsItemType1Arguments,
+    )
+
 
 T = TypeVar("T", bound="FlowStatusFailureModuleAgentActionsItemType1")
 
@@ -14,32 +21,73 @@ T = TypeVar("T", bound="FlowStatusFailureModuleAgentActionsItemType1")
 class FlowStatusFailureModuleAgentActionsItemType1:
     """
     Attributes:
+        call_id (str):
+        function_name (str):
+        resource_path (str):
         type (FlowStatusFailureModuleAgentActionsItemType1Type):
+        arguments (Union[Unset, FlowStatusFailureModuleAgentActionsItemType1Arguments]):
     """
 
+    call_id: str
+    function_name: str
+    resource_path: str
     type: FlowStatusFailureModuleAgentActionsItemType1Type
+    arguments: Union[Unset, "FlowStatusFailureModuleAgentActionsItemType1Arguments"] = 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.flow_status_failure_module_agent_actions_item_type_1_arguments import (
+            FlowStatusFailureModuleAgentActionsItemType1Arguments,
+        )
+
         d = src_dict.copy()
+        call_id = d.pop("call_id")
+
+        function_name = d.pop("function_name")
+
+        resource_path = d.pop("resource_path")
+
         type = FlowStatusFailureModuleAgentActionsItemType1Type(d.pop("type"))
 
+        _arguments = d.pop("arguments", UNSET)
+        arguments: Union[Unset, FlowStatusFailureModuleAgentActionsItemType1Arguments]
+        if isinstance(_arguments, Unset):
+            arguments = UNSET
+        else:
+            arguments = FlowStatusFailureModuleAgentActionsItemType1Arguments.from_dict(_arguments)
+
         flow_status_failure_module_agent_actions_item_type_1 = cls(
+            call_id=call_id,
+            function_name=function_name,
+            resource_path=resource_path,
             type=type,
+            arguments=arguments,
         )
 
         flow_status_failure_module_agent_actions_item_type_1.additional_properties = d

windmill_api/models/{flow_module_value_2_type_8_tools_item_suspend_resume_form_schema.py → flow_status_failure_module_agent_actions_item_type_1_arguments.py}

@@ -3,11 +3,11 @@ 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="FlowModuleValue2Type8ToolsItemSuspendResumeFormSchema")
+T = TypeVar("T", bound="FlowStatusFailureModuleAgentActionsItemType1Arguments")
 
 
 @_attrs_define
-class FlowModuleValue2Type8ToolsItemSuspendResumeFormSchema:
+class FlowStatusFailureModuleAgentActionsItemType1Arguments:
     """ """
 
     additional_properties: Dict[str, Any] = _attrs_field(init=False, factory=dict)
@@ -22,10 +22,10 @@ class FlowModuleValue2Type8ToolsItemSuspendResumeFormSchema:
     @classmethod
     def from_dict(cls: Type[T], src_dict: Dict[str, Any]) -> T:
         d = src_dict.copy()
-        flow_module_value_2_type_8_tools_item_suspend_resume_form_schema = cls()
+        flow_status_failure_module_agent_actions_item_type_1_arguments = cls()
 
-        flow_module_value_2_type_8_tools_item_suspend_resume_form_schema.additional_properties = d
-        return flow_module_value_2_type_8_tools_item_suspend_resume_form_schema
+        flow_status_failure_module_agent_actions_item_type_1_arguments.additional_properties = d
+        return flow_status_failure_module_agent_actions_item_type_1_arguments
 
     @property
     def additional_keys(self) -> List[str]:

windmill_api/models/flow_status_failure_module_agent_actions_item_type_1_type.py

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

windmill_api/models/{ai_agent_tools_item_timeout_type_0.py → flow_status_failure_module_agent_actions_item_type_2.py}

@@ -3,32 +3,30 @@ from typing import Any, Dict, List, Type, TypeVar
 from attrs import define as _attrs_define
 from attrs import field as _attrs_field
 
-from ..models.ai_agent_tools_item_timeout_type_0_type import AiAgentToolsItemTimeoutType0Type
+from ..models.flow_status_failure_module_agent_actions_item_type_2_type import (
+    FlowStatusFailureModuleAgentActionsItemType2Type,
+)
 
-T = TypeVar("T", bound="AiAgentToolsItemTimeoutType0")
+T = TypeVar("T", bound="FlowStatusFailureModuleAgentActionsItemType2")
 
 
 @_attrs_define
-class AiAgentToolsItemTimeoutType0:
+class FlowStatusFailureModuleAgentActionsItemType2:
     """
     Attributes:
-        value (Any):
-        type (AiAgentToolsItemTimeoutType0Type):
+        type (FlowStatusFailureModuleAgentActionsItemType2Type):
     """
 
-    value: Any
-    type: AiAgentToolsItemTimeoutType0Type
+    type: FlowStatusFailureModuleAgentActionsItemType2Type
     additional_properties: Dict[str, Any] = _attrs_field(init=False, factory=dict)
 
     def to_dict(self) -> Dict[str, Any]:
-        value = self.value
         type = self.type.value
 
         field_dict: Dict[str, Any] = {}
         field_dict.update(self.additional_properties)
         field_dict.update(
             {
-                "value": value,
                 "type": type,
             }
         )
@@ -38,17 +36,14 @@ class AiAgentToolsItemTimeoutType0:
     @classmethod
     def from_dict(cls: Type[T], src_dict: Dict[str, Any]) -> T:
         d = src_dict.copy()
-        value = d.pop("value")
+        type = FlowStatusFailureModuleAgentActionsItemType2Type(d.pop("type"))
 
-        type = AiAgentToolsItemTimeoutType0Type(d.pop("type"))
-
-        ai_agent_tools_item_timeout_type_0 = cls(
-            value=value,
+        flow_status_failure_module_agent_actions_item_type_2 = cls(
             type=type,
         )
 
-        ai_agent_tools_item_timeout_type_0.additional_properties = d
-        return ai_agent_tools_item_timeout_type_0
+        flow_status_failure_module_agent_actions_item_type_2.additional_properties = d
+        return flow_status_failure_module_agent_actions_item_type_2
 
     @property
     def additional_keys(self) -> List[str]:

windmill_api/models/flow_status_failure_module_agent_actions_item_type_2_type.py

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

windmill_api/models/flow_status_module.py

@@ -9,6 +9,7 @@ from ..types import UNSET, Unset
 if TYPE_CHECKING:
     from ..models.flow_status_module_agent_actions_item_type_0 import FlowStatusModuleAgentActionsItemType0
     from ..models.flow_status_module_agent_actions_item_type_1 import FlowStatusModuleAgentActionsItemType1
+    from ..models.flow_status_module_agent_actions_item_type_2 import FlowStatusModuleAgentActionsItemType2
     from ..models.flow_status_module_approvers_item import FlowStatusModuleApproversItem
     from ..models.flow_status_module_branch_chosen import FlowStatusModuleBranchChosen
     from ..models.flow_status_module_branchall import FlowStatusModuleBranchall
@@ -38,7 +39,7 @@ class FlowStatusModule:
         failed_retries (Union[Unset, List[str]]):
         skipped (Union[Unset, bool]):
         agent_actions (Union[Unset, List[Union['FlowStatusModuleAgentActionsItemType0',
-            'FlowStatusModuleAgentActionsItemType1']]]):
+            'FlowStatusModuleAgentActionsItemType1', 'FlowStatusModuleAgentActionsItemType2']]]):
         agent_actions_success (Union[Unset, List[bool]]):
     """
 
@@ -57,13 +58,21 @@ class FlowStatusModule:
     failed_retries: Union[Unset, List[str]] = UNSET
     skipped: Union[Unset, bool] = UNSET
     agent_actions: Union[
-        Unset, List[Union["FlowStatusModuleAgentActionsItemType0", "FlowStatusModuleAgentActionsItemType1"]]
+        Unset,
+        List[
+            Union[
+                "FlowStatusModuleAgentActionsItemType0",
+                "FlowStatusModuleAgentActionsItemType1",
+                "FlowStatusModuleAgentActionsItemType2",
+            ]
+        ],
     ] = UNSET
     agent_actions_success: Union[Unset, List[bool]] = UNSET
     additional_properties: Dict[str, Any] = _attrs_field(init=False, factory=dict)
 
     def to_dict(self) -> Dict[str, Any]:
         from ..models.flow_status_module_agent_actions_item_type_0 import FlowStatusModuleAgentActionsItemType0
+        from ..models.flow_status_module_agent_actions_item_type_1 import FlowStatusModuleAgentActionsItemType1
 
         type = self.type.value
 
@@ -117,6 +126,9 @@ class FlowStatusModule:
                 if isinstance(agent_actions_item_data, FlowStatusModuleAgentActionsItemType0):
                     agent_actions_item = agent_actions_item_data.to_dict()
 
+                elif isinstance(agent_actions_item_data, FlowStatusModuleAgentActionsItemType1):
+                    agent_actions_item = agent_actions_item_data.to_dict()
+
                 else:
                     agent_actions_item = agent_actions_item_data.to_dict()
 
@@ -170,6 +182,7 @@ class FlowStatusModule:
     def from_dict(cls: Type[T], src_dict: Dict[str, Any]) -> T:
         from ..models.flow_status_module_agent_actions_item_type_0 import FlowStatusModuleAgentActionsItemType0
         from ..models.flow_status_module_agent_actions_item_type_1 import FlowStatusModuleAgentActionsItemType1
+        from ..models.flow_status_module_agent_actions_item_type_2 import FlowStatusModuleAgentActionsItemType2
         from ..models.flow_status_module_approvers_item import FlowStatusModuleApproversItem
         from ..models.flow_status_module_branch_chosen import FlowStatusModuleBranchChosen
         from ..models.flow_status_module_branchall import FlowStatusModuleBranchall
@@ -236,7 +249,11 @@ class FlowStatusModule:
 
             def _parse_agent_actions_item(
                 data: object,
-            ) -> Union["FlowStatusModuleAgentActionsItemType0", "FlowStatusModuleAgentActionsItemType1"]:
+            ) -> Union[
+                "FlowStatusModuleAgentActionsItemType0",
+                "FlowStatusModuleAgentActionsItemType1",
+                "FlowStatusModuleAgentActionsItemType2",
+            ]:
                 try:
                     if not isinstance(data, dict):
                         raise TypeError()
@@ -245,11 +262,19 @@ class FlowStatusModule:
                     return agent_actions_item_type_0
                 except:  # noqa: E722
                     pass
+                try:
+                    if not isinstance(data, dict):
+                        raise TypeError()
+                    agent_actions_item_type_1 = FlowStatusModuleAgentActionsItemType1.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 = FlowStatusModuleAgentActionsItemType1.from_dict(data)
+                agent_actions_item_type_2 = FlowStatusModuleAgentActionsItemType2.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/flow_status_module_agent_actions_item_type_1.py

@@ -1,9 +1,16 @@
-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
 
 from ..models.flow_status_module_agent_actions_item_type_1_type import FlowStatusModuleAgentActionsItemType1Type
+from ..types import UNSET, Unset
+
+if TYPE_CHECKING:
+    from ..models.flow_status_module_agent_actions_item_type_1_arguments import (
+        FlowStatusModuleAgentActionsItemType1Arguments,
+    )
+
 
 T = TypeVar("T", bound="FlowStatusModuleAgentActionsItemType1")
 
@@ -12,32 +19,73 @@ T = TypeVar("T", bound="FlowStatusModuleAgentActionsItemType1")
 class FlowStatusModuleAgentActionsItemType1:
     """
     Attributes:
+        call_id (str):
+        function_name (str):
+        resource_path (str):
         type (FlowStatusModuleAgentActionsItemType1Type):
+        arguments (Union[Unset, FlowStatusModuleAgentActionsItemType1Arguments]):
     """
 
+    call_id: str
+    function_name: str
+    resource_path: str
     type: FlowStatusModuleAgentActionsItemType1Type
+    arguments: Union[Unset, "FlowStatusModuleAgentActionsItemType1Arguments"] = 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.flow_status_module_agent_actions_item_type_1_arguments import (
+            FlowStatusModuleAgentActionsItemType1Arguments,
+        )
+
         d = src_dict.copy()
+        call_id = d.pop("call_id")
+
+        function_name = d.pop("function_name")
+
+        resource_path = d.pop("resource_path")
+
         type = FlowStatusModuleAgentActionsItemType1Type(d.pop("type"))
 
+        _arguments = d.pop("arguments", UNSET)
+        arguments: Union[Unset, FlowStatusModuleAgentActionsItemType1Arguments]
+        if isinstance(_arguments, Unset):
+            arguments = UNSET
+        else:
+            arguments = FlowStatusModuleAgentActionsItemType1Arguments.from_dict(_arguments)
+
         flow_status_module_agent_actions_item_type_1 = cls(
+            call_id=call_id,
+            function_name=function_name,
+            resource_path=resource_path,
             type=type,
+            arguments=arguments,
         )
 
         flow_status_module_agent_actions_item_type_1.additional_properties = d

windmill_api/models/{ai_agent_tools_item_skip_if.py → flow_status_module_agent_actions_item_type_1_arguments.py}

@@ -3,43 +3,29 @@ 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="AiAgentToolsItemSkipIf")
+T = TypeVar("T", bound="FlowStatusModuleAgentActionsItemType1Arguments")
 
 
 @_attrs_define
-class AiAgentToolsItemSkipIf:
-    """
-    Attributes:
-        expr (str):
-    """
+class FlowStatusModuleAgentActionsItemType1Arguments:
+    """ """
 
-    expr: str
     additional_properties: Dict[str, Any] = _attrs_field(init=False, factory=dict)
 
     def to_dict(self) -> Dict[str, Any]:
-        expr = self.expr
-
         field_dict: Dict[str, Any] = {}
         field_dict.update(self.additional_properties)
-        field_dict.update(
-            {
-                "expr": expr,
-            }
-        )
+        field_dict.update({})
 
         return field_dict
 
     @classmethod
     def from_dict(cls: Type[T], src_dict: Dict[str, Any]) -> T:
         d = src_dict.copy()
-        expr = d.pop("expr")
-
-        ai_agent_tools_item_skip_if = cls(
-            expr=expr,
-        )
+        flow_status_module_agent_actions_item_type_1_arguments = cls()
 
-        ai_agent_tools_item_skip_if.additional_properties = d
-        return ai_agent_tools_item_skip_if
+        flow_status_module_agent_actions_item_type_1_arguments.additional_properties = d
+        return flow_status_module_agent_actions_item_type_1_arguments
 
     @property
     def additional_keys(self) -> List[str]:

windmill_api/models/flow_status_module_agent_actions_item_type_1_type.py

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

windmill_api/models/{ai_agent_tools_item_sleep_type_1.py → flow_status_module_agent_actions_item_type_2.py}

@@ -3,32 +3,28 @@ from typing import Any, Dict, List, Type, TypeVar
 from attrs import define as _attrs_define
 from attrs import field as _attrs_field
 
-from ..models.ai_agent_tools_item_sleep_type_1_type import AiAgentToolsItemSleepType1Type
+from ..models.flow_status_module_agent_actions_item_type_2_type import FlowStatusModuleAgentActionsItemType2Type
 
-T = TypeVar("T", bound="AiAgentToolsItemSleepType1")
+T = TypeVar("T", bound="FlowStatusModuleAgentActionsItemType2")
 
 
 @_attrs_define
-class AiAgentToolsItemSleepType1:
+class FlowStatusModuleAgentActionsItemType2:
     """
     Attributes:
-        expr (str):
-        type (AiAgentToolsItemSleepType1Type):
+        type (FlowStatusModuleAgentActionsItemType2Type):
     """
 
-    expr: str
-    type: AiAgentToolsItemSleepType1Type
+    type: FlowStatusModuleAgentActionsItemType2Type
     additional_properties: Dict[str, Any] = _attrs_field(init=False, factory=dict)
 
     def to_dict(self) -> Dict[str, Any]:
-        expr = self.expr
         type = self.type.value
 
         field_dict: Dict[str, Any] = {}
         field_dict.update(self.additional_properties)
         field_dict.update(
             {
-                "expr": expr,
                 "type": type,
             }
         )
@@ -38,17 +34,14 @@ class AiAgentToolsItemSleepType1:
     @classmethod
     def from_dict(cls: Type[T], src_dict: Dict[str, Any]) -> T:
         d = src_dict.copy()
-        expr = d.pop("expr")
+        type = FlowStatusModuleAgentActionsItemType2Type(d.pop("type"))
 
-        type = AiAgentToolsItemSleepType1Type(d.pop("type"))
-
-        ai_agent_tools_item_sleep_type_1 = cls(
-            expr=expr,
+        flow_status_module_agent_actions_item_type_2 = cls(
             type=type,
         )
 
-        ai_agent_tools_item_sleep_type_1.additional_properties = d
-        return ai_agent_tools_item_sleep_type_1
+        flow_status_module_agent_actions_item_type_2.additional_properties = d
+        return flow_status_module_agent_actions_item_type_2
 
     @property
     def additional_keys(self) -> List[str]:

windmill_api/models/flow_status_module_agent_actions_item_type_2_type.py

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