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

windmill_api/models/queued_job_flow_status_failure_module.py

@@ -7,6 +7,12 @@ from ..models.queued_job_flow_status_failure_module_type import QueuedJobFlowSta
 from ..types import UNSET, Unset
 
 if TYPE_CHECKING:
+    from ..models.queued_job_flow_status_failure_module_agent_actions_item_type_0 import (
+        QueuedJobFlowStatusFailureModuleAgentActionsItemType0,
+    )
+    from ..models.queued_job_flow_status_failure_module_agent_actions_item_type_1 import (
+        QueuedJobFlowStatusFailureModuleAgentActionsItemType1,
+    )
     from ..models.queued_job_flow_status_failure_module_approvers_item import (
         QueuedJobFlowStatusFailureModuleApproversItem,
     )
@@ -37,6 +43,9 @@ class QueuedJobFlowStatusFailureModule:
         approvers (Union[Unset, List['QueuedJobFlowStatusFailureModuleApproversItem']]):
         failed_retries (Union[Unset, List[str]]):
         skipped (Union[Unset, bool]):
+        agent_actions (Union[Unset, List[Union['QueuedJobFlowStatusFailureModuleAgentActionsItemType0',
+            'QueuedJobFlowStatusFailureModuleAgentActionsItemType1']]]):
+        agent_actions_success (Union[Unset, List[bool]]):
         parent_module (Union[Unset, str]):
     """
 
@@ -53,10 +62,24 @@ class QueuedJobFlowStatusFailureModule:
     approvers: Union[Unset, List["QueuedJobFlowStatusFailureModuleApproversItem"]] = UNSET
     failed_retries: Union[Unset, List[str]] = UNSET
     skipped: Union[Unset, bool] = UNSET
+    agent_actions: Union[
+        Unset,
+        List[
+            Union[
+                "QueuedJobFlowStatusFailureModuleAgentActionsItemType0",
+                "QueuedJobFlowStatusFailureModuleAgentActionsItemType1",
+            ]
+        ],
+    ] = UNSET
+    agent_actions_success: Union[Unset, List[bool]] = UNSET
     parent_module: Union[Unset, str] = UNSET
     additional_properties: Dict[str, Any] = _attrs_field(init=False, factory=dict)
 
     def to_dict(self) -> Dict[str, Any]:
+        from ..models.queued_job_flow_status_failure_module_agent_actions_item_type_0 import (
+            QueuedJobFlowStatusFailureModuleAgentActionsItemType0,
+        )
+
         type = self.type.value
 
         id = self.id
@@ -96,6 +119,24 @@ class QueuedJobFlowStatusFailureModule:
             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, QueuedJobFlowStatusFailureModuleAgentActionsItemType0):
+                    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
+
         parent_module = self.parent_module
 
         field_dict: Dict[str, Any] = {}
@@ -129,6 +170,10 @@ class QueuedJobFlowStatusFailureModule:
             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
         if parent_module is not UNSET:
             field_dict["parent_module"] = parent_module
 
@@ -136,6 +181,12 @@ class QueuedJobFlowStatusFailureModule:
 
     @classmethod
     def from_dict(cls: Type[T], src_dict: Dict[str, Any]) -> T:
+        from ..models.queued_job_flow_status_failure_module_agent_actions_item_type_0 import (
+            QueuedJobFlowStatusFailureModuleAgentActionsItemType0,
+        )
+        from ..models.queued_job_flow_status_failure_module_agent_actions_item_type_1 import (
+            QueuedJobFlowStatusFailureModuleAgentActionsItemType1,
+        )
         from ..models.queued_job_flow_status_failure_module_approvers_item import (
             QueuedJobFlowStatusFailureModuleApproversItem,
         )
@@ -192,6 +243,36 @@ class QueuedJobFlowStatusFailureModule:
 
         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[
+                "QueuedJobFlowStatusFailureModuleAgentActionsItemType0",
+                "QueuedJobFlowStatusFailureModuleAgentActionsItemType1",
+            ]:
+                try:
+                    if not isinstance(data, dict):
+                        raise TypeError()
+                    agent_actions_item_type_0 = QueuedJobFlowStatusFailureModuleAgentActionsItemType0.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 = QueuedJobFlowStatusFailureModuleAgentActionsItemType1.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))
+
         parent_module = d.pop("parent_module", UNSET)
 
         queued_job_flow_status_failure_module = cls(
@@ -208,6 +289,8 @@ class QueuedJobFlowStatusFailureModule:
             approvers=approvers,
             failed_retries=failed_retries,
             skipped=skipped,
+            agent_actions=agent_actions,
+            agent_actions_success=agent_actions_success,
             parent_module=parent_module,
         )
 

windmill_api/models/queued_job_flow_status_failure_module_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.queued_job_flow_status_failure_module_agent_actions_item_type_0_type import (
+    QueuedJobFlowStatusFailureModuleAgentActionsItemType0Type,
+)
+
+T = TypeVar("T", bound="QueuedJobFlowStatusFailureModuleAgentActionsItemType0")
+
+
+@_attrs_define
+class QueuedJobFlowStatusFailureModuleAgentActionsItemType0:
+    """
+    Attributes:
+        job_id (str):
+        function_name (str):
+        type (QueuedJobFlowStatusFailureModuleAgentActionsItemType0Type):
+        module_id (str):
+    """
+
+    job_id: str
+    function_name: str
+    type: QueuedJobFlowStatusFailureModuleAgentActionsItemType0Type
+    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 = QueuedJobFlowStatusFailureModuleAgentActionsItemType0Type(d.pop("type"))
+
+        module_id = d.pop("module_id")
+
+        queued_job_flow_status_failure_module_agent_actions_item_type_0 = cls(
+            job_id=job_id,
+            function_name=function_name,
+            type=type,
+            module_id=module_id,
+        )
+
+        queued_job_flow_status_failure_module_agent_actions_item_type_0.additional_properties = d
+        return queued_job_flow_status_failure_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/queued_job_flow_status_failure_module_agent_actions_item_type_0_type.py

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

windmill_api/models/queued_job_flow_status_failure_module_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.queued_job_flow_status_failure_module_agent_actions_item_type_1_type import (
+    QueuedJobFlowStatusFailureModuleAgentActionsItemType1Type,
+)
+
+T = TypeVar("T", bound="QueuedJobFlowStatusFailureModuleAgentActionsItemType1")
+
+
+@_attrs_define
+class QueuedJobFlowStatusFailureModuleAgentActionsItemType1:
+    """
+    Attributes:
+        type (QueuedJobFlowStatusFailureModuleAgentActionsItemType1Type):
+    """
+
+    type: QueuedJobFlowStatusFailureModuleAgentActionsItemType1Type
+    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 = QueuedJobFlowStatusFailureModuleAgentActionsItemType1Type(d.pop("type"))
+
+        queued_job_flow_status_failure_module_agent_actions_item_type_1 = cls(
+            type=type,
+        )
+
+        queued_job_flow_status_failure_module_agent_actions_item_type_1.additional_properties = d
+        return queued_job_flow_status_failure_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/queued_job_flow_status_failure_module_agent_actions_item_type_1_type.py

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

windmill_api/models/queued_job_flow_status_modules_item.py

@@ -7,6 +7,12 @@ from ..models.queued_job_flow_status_modules_item_type import QueuedJobFlowStatu
 from ..types import UNSET, Unset
 
 if TYPE_CHECKING:
+    from ..models.queued_job_flow_status_modules_item_agent_actions_item_type_0 import (
+        QueuedJobFlowStatusModulesItemAgentActionsItemType0,
+    )
+    from ..models.queued_job_flow_status_modules_item_agent_actions_item_type_1 import (
+        QueuedJobFlowStatusModulesItemAgentActionsItemType1,
+    )
     from ..models.queued_job_flow_status_modules_item_approvers_item import QueuedJobFlowStatusModulesItemApproversItem
     from ..models.queued_job_flow_status_modules_item_branch_chosen import QueuedJobFlowStatusModulesItemBranchChosen
     from ..models.queued_job_flow_status_modules_item_branchall import QueuedJobFlowStatusModulesItemBranchall
@@ -33,6 +39,9 @@ class QueuedJobFlowStatusModulesItem:
         approvers (Union[Unset, List['QueuedJobFlowStatusModulesItemApproversItem']]):
         failed_retries (Union[Unset, List[str]]):
         skipped (Union[Unset, bool]):
+        agent_actions (Union[Unset, List[Union['QueuedJobFlowStatusModulesItemAgentActionsItemType0',
+            'QueuedJobFlowStatusModulesItemAgentActionsItemType1']]]):
+        agent_actions_success (Union[Unset, List[bool]]):
     """
 
     type: QueuedJobFlowStatusModulesItemType
@@ -48,9 +57,23 @@ class QueuedJobFlowStatusModulesItem:
     approvers: Union[Unset, List["QueuedJobFlowStatusModulesItemApproversItem"]] = UNSET
     failed_retries: Union[Unset, List[str]] = UNSET
     skipped: Union[Unset, bool] = UNSET
+    agent_actions: Union[
+        Unset,
+        List[
+            Union[
+                "QueuedJobFlowStatusModulesItemAgentActionsItemType0",
+                "QueuedJobFlowStatusModulesItemAgentActionsItemType1",
+            ]
+        ],
+    ] = 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.queued_job_flow_status_modules_item_agent_actions_item_type_0 import (
+            QueuedJobFlowStatusModulesItemAgentActionsItemType0,
+        )
+
         type = self.type.value
 
         id = self.id
@@ -90,6 +113,23 @@ class QueuedJobFlowStatusModulesItem:
             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, QueuedJobFlowStatusModulesItemAgentActionsItemType0):
+                    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 +162,21 @@ class QueuedJobFlowStatusModulesItem:
             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.queued_job_flow_status_modules_item_agent_actions_item_type_0 import (
+            QueuedJobFlowStatusModulesItemAgentActionsItemType0,
+        )
+        from ..models.queued_job_flow_status_modules_item_agent_actions_item_type_1 import (
+            QueuedJobFlowStatusModulesItemAgentActionsItemType1,
+        )
         from ..models.queued_job_flow_status_modules_item_approvers_item import (
             QueuedJobFlowStatusModulesItemApproversItem,
         )
@@ -183,6 +233,36 @@ class QueuedJobFlowStatusModulesItem:
 
         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[
+                "QueuedJobFlowStatusModulesItemAgentActionsItemType0",
+                "QueuedJobFlowStatusModulesItemAgentActionsItemType1",
+            ]:
+                try:
+                    if not isinstance(data, dict):
+                        raise TypeError()
+                    agent_actions_item_type_0 = QueuedJobFlowStatusModulesItemAgentActionsItemType0.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 = QueuedJobFlowStatusModulesItemAgentActionsItemType1.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))
+
         queued_job_flow_status_modules_item = cls(
             type=type,
             id=id,
@@ -197,6 +277,8 @@ class QueuedJobFlowStatusModulesItem:
             approvers=approvers,
             failed_retries=failed_retries,
             skipped=skipped,
+            agent_actions=agent_actions,
+            agent_actions_success=agent_actions_success,
         )
 
         queued_job_flow_status_modules_item.additional_properties = d

windmill_api/models/queued_job_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.queued_job_flow_status_modules_item_agent_actions_item_type_0_type import (
+    QueuedJobFlowStatusModulesItemAgentActionsItemType0Type,
+)
+
+T = TypeVar("T", bound="QueuedJobFlowStatusModulesItemAgentActionsItemType0")
+
+
+@_attrs_define
+class QueuedJobFlowStatusModulesItemAgentActionsItemType0:
+    """
+    Attributes:
+        job_id (str):
+        function_name (str):
+        type (QueuedJobFlowStatusModulesItemAgentActionsItemType0Type):
+        module_id (str):
+    """
+
+    job_id: str
+    function_name: str
+    type: QueuedJobFlowStatusModulesItemAgentActionsItemType0Type
+    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 = QueuedJobFlowStatusModulesItemAgentActionsItemType0Type(d.pop("type"))
+
+        module_id = d.pop("module_id")
+
+        queued_job_flow_status_modules_item_agent_actions_item_type_0 = cls(
+            job_id=job_id,
+            function_name=function_name,
+            type=type,
+            module_id=module_id,
+        )
+
+        queued_job_flow_status_modules_item_agent_actions_item_type_0.additional_properties = d
+        return queued_job_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/queued_job_flow_status_modules_item_agent_actions_item_type_0_type.py

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

windmill_api/models/queued_job_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.queued_job_flow_status_modules_item_agent_actions_item_type_1_type import (
+    QueuedJobFlowStatusModulesItemAgentActionsItemType1Type,
+)
+
+T = TypeVar("T", bound="QueuedJobFlowStatusModulesItemAgentActionsItemType1")
+
+
+@_attrs_define
+class QueuedJobFlowStatusModulesItemAgentActionsItemType1:
+    """
+    Attributes:
+        type (QueuedJobFlowStatusModulesItemAgentActionsItemType1Type):
+    """
+
+    type: QueuedJobFlowStatusModulesItemAgentActionsItemType1Type
+    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 = QueuedJobFlowStatusModulesItemAgentActionsItemType1Type(d.pop("type"))
+
+        queued_job_flow_status_modules_item_agent_actions_item_type_1 = cls(
+            type=type,
+        )
+
+        queued_job_flow_status_modules_item_agent_actions_item_type_1.additional_properties = d
+        return queued_job_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/queued_job_flow_status_modules_item_agent_actions_item_type_1_type.py

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