windmill-api 1.529.0__py3-none-any.whl → 1.531.0__py3-none-any.whl

windmill_api/models/flow_status_module.py

@@ -7,6 +7,8 @@ from ..models.flow_status_module_type import FlowStatusModuleType
 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_approvers_item import FlowStatusModuleApproversItem
     from ..models.flow_status_module_branch_chosen import FlowStatusModuleBranchChosen
     from ..models.flow_status_module_branchall import FlowStatusModuleBranchall
@@ -33,6 +35,9 @@ class FlowStatusModule:
         approvers (Union[Unset, List['FlowStatusModuleApproversItem']]):
         failed_retries (Union[Unset, List[str]]):
         skipped (Union[Unset, bool]):
+        agent_actions (Union[Unset, List[Union['FlowStatusModuleAgentActionsItemType0',
+            'FlowStatusModuleAgentActionsItemType1']]]):
+        agent_actions_success (Union[Unset, List[bool]]):
     """
 
     type: FlowStatusModuleType
@@ -48,9 +53,15 @@ class FlowStatusModule:
     approvers: Union[Unset, List["FlowStatusModuleApproversItem"]] = UNSET
     failed_retries: Union[Unset, List[str]] = UNSET
     skipped: Union[Unset, bool] = UNSET
+    agent_actions: Union[
+        Unset, List[Union["FlowStatusModuleAgentActionsItemType0", "FlowStatusModuleAgentActionsItemType1"]]
+    ] = 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
+
         type = self.type.value
 
         id = self.id
@@ -90,6 +101,23 @@ class FlowStatusModule:
             failed_retries = self.failed_retries
 
         skipped = self.skipped
+        agent_actions: Union[Unset, List[Dict[str, Any]]] = UNSET
+        if not isinstance(self.agent_actions, Unset):
+            agent_actions = []
+            for agent_actions_item_data in self.agent_actions:
+                agent_actions_item: Dict[str, Any]
+
+                if isinstance(agent_actions_item_data, FlowStatusModuleAgentActionsItemType0):
+                    agent_actions_item = agent_actions_item_data.to_dict()
+
+                else:
+                    agent_actions_item = agent_actions_item_data.to_dict()
+
+                agent_actions.append(agent_actions_item)
+
+        agent_actions_success: Union[Unset, List[bool]] = UNSET
+        if not isinstance(self.agent_actions_success, Unset):
+            agent_actions_success = self.agent_actions_success
 
         field_dict: Dict[str, Any] = {}
         field_dict.update(self.additional_properties)
@@ -122,11 +150,17 @@ class FlowStatusModule:
             field_dict["failed_retries"] = failed_retries
         if skipped is not UNSET:
             field_dict["skipped"] = skipped
+        if agent_actions is not UNSET:
+            field_dict["agent_actions"] = agent_actions
+        if agent_actions_success is not UNSET:
+            field_dict["agent_actions_success"] = agent_actions_success
 
         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_0 import FlowStatusModuleAgentActionsItemType0
+        from ..models.flow_status_module_agent_actions_item_type_1 import FlowStatusModuleAgentActionsItemType1
         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
@@ -179,6 +213,33 @@ class FlowStatusModule:
 
         skipped = d.pop("skipped", UNSET)
 
+        agent_actions = []
+        _agent_actions = d.pop("agent_actions", UNSET)
+        for agent_actions_item_data in _agent_actions or []:
+
+            def _parse_agent_actions_item(
+                data: object,
+            ) -> Union["FlowStatusModuleAgentActionsItemType0", "FlowStatusModuleAgentActionsItemType1"]:
+                try:
+                    if not isinstance(data, dict):
+                        raise TypeError()
+                    agent_actions_item_type_0 = FlowStatusModuleAgentActionsItemType0.from_dict(data)
+
+                    return agent_actions_item_type_0
+                except:  # noqa: E722
+                    pass
+                if not isinstance(data, dict):
+                    raise TypeError()
+                agent_actions_item_type_1 = FlowStatusModuleAgentActionsItemType1.from_dict(data)
+
+                return agent_actions_item_type_1
+
+            agent_actions_item = _parse_agent_actions_item(agent_actions_item_data)
+
+            agent_actions.append(agent_actions_item)
+
+        agent_actions_success = cast(List[bool], d.pop("agent_actions_success", UNSET))
+
         flow_status_module = cls(
             type=type,
             id=id,
@@ -193,6 +254,8 @@ class FlowStatusModule:
             approvers=approvers,
             failed_retries=failed_retries,
             skipped=skipped,
+            agent_actions=agent_actions,
+            agent_actions_success=agent_actions_success,
         )
 
         flow_status_module.additional_properties = d

windmill_api/models/flow_status_module_agent_actions_item_type_0.py

@@ -0,0 +1,82 @@
+from typing import Any, Dict, List, Type, TypeVar
+
+from attrs import define as _attrs_define
+from attrs import field as _attrs_field
+
+from ..models.flow_status_module_agent_actions_item_type_0_type import FlowStatusModuleAgentActionsItemType0Type
+
+T = TypeVar("T", bound="FlowStatusModuleAgentActionsItemType0")
+
+
+@_attrs_define
+class FlowStatusModuleAgentActionsItemType0:
+    """
+    Attributes:
+        job_id (str):
+        function_name (str):
+        type (FlowStatusModuleAgentActionsItemType0Type):
+        module_id (str):
+    """
+
+    job_id: str
+    function_name: str
+    type: FlowStatusModuleAgentActionsItemType0Type
+    module_id: str
+    additional_properties: Dict[str, Any] = _attrs_field(init=False, factory=dict)
+
+    def to_dict(self) -> Dict[str, Any]:
+        job_id = self.job_id
+        function_name = self.function_name
+        type = self.type.value
+
+        module_id = self.module_id
+
+        field_dict: Dict[str, Any] = {}
+        field_dict.update(self.additional_properties)
+        field_dict.update(
+            {
+                "job_id": job_id,
+                "function_name": function_name,
+                "type": type,
+                "module_id": module_id,
+            }
+        )
+
+        return field_dict
+
+    @classmethod
+    def from_dict(cls: Type[T], src_dict: Dict[str, Any]) -> T:
+        d = src_dict.copy()
+        job_id = d.pop("job_id")
+
+        function_name = d.pop("function_name")
+
+        type = FlowStatusModuleAgentActionsItemType0Type(d.pop("type"))
+
+        module_id = d.pop("module_id")
+
+        flow_status_module_agent_actions_item_type_0 = cls(
+            job_id=job_id,
+            function_name=function_name,
+            type=type,
+            module_id=module_id,
+        )
+
+        flow_status_module_agent_actions_item_type_0.additional_properties = d
+        return flow_status_module_agent_actions_item_type_0
+
+    @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/flow_status_module_agent_actions_item_type_0_type.py

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

windmill_api/models/flow_status_module_agent_actions_item_type_1.py

@@ -0,0 +1,60 @@
+from typing import Any, Dict, List, Type, TypeVar
+
+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
+
+T = TypeVar("T", bound="FlowStatusModuleAgentActionsItemType1")
+
+
+@_attrs_define
+class FlowStatusModuleAgentActionsItemType1:
+    """
+    Attributes:
+        type (FlowStatusModuleAgentActionsItemType1Type):
+    """
+
+    type: FlowStatusModuleAgentActionsItemType1Type
+    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 = FlowStatusModuleAgentActionsItemType1Type(d.pop("type"))
+
+        flow_status_module_agent_actions_item_type_1 = cls(
+            type=type,
+        )
+
+        flow_status_module_agent_actions_item_type_1.additional_properties = d
+        return flow_status_module_agent_actions_item_type_1
+
+    @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/flow_status_module_agent_actions_item_type_1_type.py

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

windmill_api/models/flow_status_modules_item.py

@@ -7,6 +7,8 @@ from ..models.flow_status_modules_item_type import FlowStatusModulesItemType
 from ..types import UNSET, Unset
 
 if TYPE_CHECKING:
+    from ..models.flow_status_modules_item_agent_actions_item_type_0 import FlowStatusModulesItemAgentActionsItemType0
+    from ..models.flow_status_modules_item_agent_actions_item_type_1 import FlowStatusModulesItemAgentActionsItemType1
     from ..models.flow_status_modules_item_approvers_item import FlowStatusModulesItemApproversItem
     from ..models.flow_status_modules_item_branch_chosen import FlowStatusModulesItemBranchChosen
     from ..models.flow_status_modules_item_branchall import FlowStatusModulesItemBranchall
@@ -33,6 +35,9 @@ class FlowStatusModulesItem:
         approvers (Union[Unset, List['FlowStatusModulesItemApproversItem']]):
         failed_retries (Union[Unset, List[str]]):
         skipped (Union[Unset, bool]):
+        agent_actions (Union[Unset, List[Union['FlowStatusModulesItemAgentActionsItemType0',
+            'FlowStatusModulesItemAgentActionsItemType1']]]):
+        agent_actions_success (Union[Unset, List[bool]]):
     """
 
     type: FlowStatusModulesItemType
@@ -48,9 +53,17 @@ class FlowStatusModulesItem:
     approvers: Union[Unset, List["FlowStatusModulesItemApproversItem"]] = UNSET
     failed_retries: Union[Unset, List[str]] = UNSET
     skipped: Union[Unset, bool] = UNSET
+    agent_actions: Union[
+        Unset, List[Union["FlowStatusModulesItemAgentActionsItemType0", "FlowStatusModulesItemAgentActionsItemType1"]]
+    ] = 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_modules_item_agent_actions_item_type_0 import (
+            FlowStatusModulesItemAgentActionsItemType0,
+        )
+
         type = self.type.value
 
         id = self.id
@@ -90,6 +103,23 @@ class FlowStatusModulesItem:
             failed_retries = self.failed_retries
 
         skipped = self.skipped
+        agent_actions: Union[Unset, List[Dict[str, Any]]] = UNSET
+        if not isinstance(self.agent_actions, Unset):
+            agent_actions = []
+            for agent_actions_item_data in self.agent_actions:
+                agent_actions_item: Dict[str, Any]
+
+                if isinstance(agent_actions_item_data, FlowStatusModulesItemAgentActionsItemType0):
+                    agent_actions_item = agent_actions_item_data.to_dict()
+
+                else:
+                    agent_actions_item = agent_actions_item_data.to_dict()
+
+                agent_actions.append(agent_actions_item)
+
+        agent_actions_success: Union[Unset, List[bool]] = UNSET
+        if not isinstance(self.agent_actions_success, Unset):
+            agent_actions_success = self.agent_actions_success
 
         field_dict: Dict[str, Any] = {}
         field_dict.update(self.additional_properties)
@@ -122,11 +152,21 @@ class FlowStatusModulesItem:
             field_dict["failed_retries"] = failed_retries
         if skipped is not UNSET:
             field_dict["skipped"] = skipped
+        if agent_actions is not UNSET:
+            field_dict["agent_actions"] = agent_actions
+        if agent_actions_success is not UNSET:
+            field_dict["agent_actions_success"] = agent_actions_success
 
         return field_dict
 
     @classmethod
     def from_dict(cls: Type[T], src_dict: Dict[str, Any]) -> T:
+        from ..models.flow_status_modules_item_agent_actions_item_type_0 import (
+            FlowStatusModulesItemAgentActionsItemType0,
+        )
+        from ..models.flow_status_modules_item_agent_actions_item_type_1 import (
+            FlowStatusModulesItemAgentActionsItemType1,
+        )
         from ..models.flow_status_modules_item_approvers_item import FlowStatusModulesItemApproversItem
         from ..models.flow_status_modules_item_branch_chosen import FlowStatusModulesItemBranchChosen
         from ..models.flow_status_modules_item_branchall import FlowStatusModulesItemBranchall
@@ -179,6 +219,33 @@ class FlowStatusModulesItem:
 
         skipped = d.pop("skipped", UNSET)
 
+        agent_actions = []
+        _agent_actions = d.pop("agent_actions", UNSET)
+        for agent_actions_item_data in _agent_actions or []:
+
+            def _parse_agent_actions_item(
+                data: object,
+            ) -> Union["FlowStatusModulesItemAgentActionsItemType0", "FlowStatusModulesItemAgentActionsItemType1"]:
+                try:
+                    if not isinstance(data, dict):
+                        raise TypeError()
+                    agent_actions_item_type_0 = FlowStatusModulesItemAgentActionsItemType0.from_dict(data)
+
+                    return agent_actions_item_type_0
+                except:  # noqa: E722
+                    pass
+                if not isinstance(data, dict):
+                    raise TypeError()
+                agent_actions_item_type_1 = FlowStatusModulesItemAgentActionsItemType1.from_dict(data)
+
+                return agent_actions_item_type_1
+
+            agent_actions_item = _parse_agent_actions_item(agent_actions_item_data)
+
+            agent_actions.append(agent_actions_item)
+
+        agent_actions_success = cast(List[bool], d.pop("agent_actions_success", UNSET))
+
         flow_status_modules_item = cls(
             type=type,
             id=id,
@@ -193,6 +260,8 @@ class FlowStatusModulesItem:
             approvers=approvers,
             failed_retries=failed_retries,
             skipped=skipped,
+            agent_actions=agent_actions,
+            agent_actions_success=agent_actions_success,
         )
 
         flow_status_modules_item.additional_properties = d

windmill_api/models/flow_status_modules_item_agent_actions_item_type_0.py

@@ -0,0 +1,84 @@
+from typing import Any, Dict, List, Type, TypeVar
+
+from attrs import define as _attrs_define
+from attrs import field as _attrs_field
+
+from ..models.flow_status_modules_item_agent_actions_item_type_0_type import (
+    FlowStatusModulesItemAgentActionsItemType0Type,
+)
+
+T = TypeVar("T", bound="FlowStatusModulesItemAgentActionsItemType0")
+
+
+@_attrs_define
+class FlowStatusModulesItemAgentActionsItemType0:
+    """
+    Attributes:
+        job_id (str):
+        function_name (str):
+        type (FlowStatusModulesItemAgentActionsItemType0Type):
+        module_id (str):
+    """
+
+    job_id: str
+    function_name: str
+    type: FlowStatusModulesItemAgentActionsItemType0Type
+    module_id: str
+    additional_properties: Dict[str, Any] = _attrs_field(init=False, factory=dict)
+
+    def to_dict(self) -> Dict[str, Any]:
+        job_id = self.job_id
+        function_name = self.function_name
+        type = self.type.value
+
+        module_id = self.module_id
+
+        field_dict: Dict[str, Any] = {}
+        field_dict.update(self.additional_properties)
+        field_dict.update(
+            {
+                "job_id": job_id,
+                "function_name": function_name,
+                "type": type,
+                "module_id": module_id,
+            }
+        )
+
+        return field_dict
+
+    @classmethod
+    def from_dict(cls: Type[T], src_dict: Dict[str, Any]) -> T:
+        d = src_dict.copy()
+        job_id = d.pop("job_id")
+
+        function_name = d.pop("function_name")
+
+        type = FlowStatusModulesItemAgentActionsItemType0Type(d.pop("type"))
+
+        module_id = d.pop("module_id")
+
+        flow_status_modules_item_agent_actions_item_type_0 = cls(
+            job_id=job_id,
+            function_name=function_name,
+            type=type,
+            module_id=module_id,
+        )
+
+        flow_status_modules_item_agent_actions_item_type_0.additional_properties = d
+        return flow_status_modules_item_agent_actions_item_type_0
+
+    @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/flow_status_modules_item_agent_actions_item_type_0_type.py

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

windmill_api/models/flow_status_modules_item_agent_actions_item_type_1.py

@@ -0,0 +1,62 @@
+from typing import Any, Dict, List, Type, TypeVar
+
+from attrs import define as _attrs_define
+from attrs import field as _attrs_field
+
+from ..models.flow_status_modules_item_agent_actions_item_type_1_type import (
+    FlowStatusModulesItemAgentActionsItemType1Type,
+)
+
+T = TypeVar("T", bound="FlowStatusModulesItemAgentActionsItemType1")
+
+
+@_attrs_define
+class FlowStatusModulesItemAgentActionsItemType1:
+    """
+    Attributes:
+        type (FlowStatusModulesItemAgentActionsItemType1Type):
+    """
+
+    type: FlowStatusModulesItemAgentActionsItemType1Type
+    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 = FlowStatusModulesItemAgentActionsItemType1Type(d.pop("type"))
+
+        flow_status_modules_item_agent_actions_item_type_1 = cls(
+            type=type,
+        )
+
+        flow_status_modules_item_agent_actions_item_type_1.additional_properties = d
+        return flow_status_modules_item_agent_actions_item_type_1
+
+    @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/flow_status_modules_item_agent_actions_item_type_1_type.py

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