windmill-api 1.530.0__py3-none-any.whl → 1.532.0__py3-none-any.whl

windmill_api/models/completed_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.completed_job_flow_status_failure_module_agent_actions_item_type_0_type import (
+    CompletedJobFlowStatusFailureModuleAgentActionsItemType0Type,
+)
+
+T = TypeVar("T", bound="CompletedJobFlowStatusFailureModuleAgentActionsItemType0")
+
+
+@_attrs_define
+class CompletedJobFlowStatusFailureModuleAgentActionsItemType0:
+    """
+    Attributes:
+        job_id (str):
+        function_name (str):
+        type (CompletedJobFlowStatusFailureModuleAgentActionsItemType0Type):
+        module_id (str):
+    """
+
+    job_id: str
+    function_name: str
+    type: CompletedJobFlowStatusFailureModuleAgentActionsItemType0Type
+    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 = CompletedJobFlowStatusFailureModuleAgentActionsItemType0Type(d.pop("type"))
+
+        module_id = d.pop("module_id")
+
+        completed_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,
+        )
+
+        completed_job_flow_status_failure_module_agent_actions_item_type_0.additional_properties = d
+        return completed_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/completed_job_flow_status_failure_module_agent_actions_item_type_0_type.py

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

windmill_api/models/completed_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.completed_job_flow_status_failure_module_agent_actions_item_type_1_type import (
+    CompletedJobFlowStatusFailureModuleAgentActionsItemType1Type,
+)
+
+T = TypeVar("T", bound="CompletedJobFlowStatusFailureModuleAgentActionsItemType1")
+
+
+@_attrs_define
+class CompletedJobFlowStatusFailureModuleAgentActionsItemType1:
+    """
+    Attributes:
+        type (CompletedJobFlowStatusFailureModuleAgentActionsItemType1Type):
+    """
+
+    type: CompletedJobFlowStatusFailureModuleAgentActionsItemType1Type
+    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 = CompletedJobFlowStatusFailureModuleAgentActionsItemType1Type(d.pop("type"))
+
+        completed_job_flow_status_failure_module_agent_actions_item_type_1 = cls(
+            type=type,
+        )
+
+        completed_job_flow_status_failure_module_agent_actions_item_type_1.additional_properties = d
+        return completed_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/completed_job_flow_status_failure_module_agent_actions_item_type_1_type.py

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

windmill_api/models/completed_job_flow_status_modules_item.py

@@ -7,6 +7,12 @@ from ..models.completed_job_flow_status_modules_item_type import CompletedJobFlo
 from ..types import UNSET, Unset
 
 if TYPE_CHECKING:
+    from ..models.completed_job_flow_status_modules_item_agent_actions_item_type_0 import (
+        CompletedJobFlowStatusModulesItemAgentActionsItemType0,
+    )
+    from ..models.completed_job_flow_status_modules_item_agent_actions_item_type_1 import (
+        CompletedJobFlowStatusModulesItemAgentActionsItemType1,
+    )
     from ..models.completed_job_flow_status_modules_item_approvers_item import (
         CompletedJobFlowStatusModulesItemApproversItem,
     )
@@ -37,6 +43,9 @@ class CompletedJobFlowStatusModulesItem:
         approvers (Union[Unset, List['CompletedJobFlowStatusModulesItemApproversItem']]):
         failed_retries (Union[Unset, List[str]]):
         skipped (Union[Unset, bool]):
+        agent_actions (Union[Unset, List[Union['CompletedJobFlowStatusModulesItemAgentActionsItemType0',
+            'CompletedJobFlowStatusModulesItemAgentActionsItemType1']]]):
+        agent_actions_success (Union[Unset, List[bool]]):
     """
 
     type: CompletedJobFlowStatusModulesItemType
@@ -52,9 +61,23 @@ class CompletedJobFlowStatusModulesItem:
     approvers: Union[Unset, List["CompletedJobFlowStatusModulesItemApproversItem"]] = UNSET
     failed_retries: Union[Unset, List[str]] = UNSET
     skipped: Union[Unset, bool] = UNSET
+    agent_actions: Union[
+        Unset,
+        List[
+            Union[
+                "CompletedJobFlowStatusModulesItemAgentActionsItemType0",
+                "CompletedJobFlowStatusModulesItemAgentActionsItemType1",
+            ]
+        ],
+    ] = 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.completed_job_flow_status_modules_item_agent_actions_item_type_0 import (
+            CompletedJobFlowStatusModulesItemAgentActionsItemType0,
+        )
+
         type = self.type.value
 
         id = self.id
@@ -94,6 +117,23 @@ class CompletedJobFlowStatusModulesItem:
             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, CompletedJobFlowStatusModulesItemAgentActionsItemType0):
+                    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)
@@ -126,11 +166,21 @@ class CompletedJobFlowStatusModulesItem:
             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.completed_job_flow_status_modules_item_agent_actions_item_type_0 import (
+            CompletedJobFlowStatusModulesItemAgentActionsItemType0,
+        )
+        from ..models.completed_job_flow_status_modules_item_agent_actions_item_type_1 import (
+            CompletedJobFlowStatusModulesItemAgentActionsItemType1,
+        )
         from ..models.completed_job_flow_status_modules_item_approvers_item import (
             CompletedJobFlowStatusModulesItemApproversItem,
         )
@@ -187,6 +237,36 @@ class CompletedJobFlowStatusModulesItem:
 
         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[
+                "CompletedJobFlowStatusModulesItemAgentActionsItemType0",
+                "CompletedJobFlowStatusModulesItemAgentActionsItemType1",
+            ]:
+                try:
+                    if not isinstance(data, dict):
+                        raise TypeError()
+                    agent_actions_item_type_0 = CompletedJobFlowStatusModulesItemAgentActionsItemType0.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 = CompletedJobFlowStatusModulesItemAgentActionsItemType1.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))
+
         completed_job_flow_status_modules_item = cls(
             type=type,
             id=id,
@@ -201,6 +281,8 @@ class CompletedJobFlowStatusModulesItem:
             approvers=approvers,
             failed_retries=failed_retries,
             skipped=skipped,
+            agent_actions=agent_actions,
+            agent_actions_success=agent_actions_success,
         )
 
         completed_job_flow_status_modules_item.additional_properties = d

windmill_api/models/completed_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.completed_job_flow_status_modules_item_agent_actions_item_type_0_type import (
+    CompletedJobFlowStatusModulesItemAgentActionsItemType0Type,
+)
+
+T = TypeVar("T", bound="CompletedJobFlowStatusModulesItemAgentActionsItemType0")
+
+
+@_attrs_define
+class CompletedJobFlowStatusModulesItemAgentActionsItemType0:
+    """
+    Attributes:
+        job_id (str):
+        function_name (str):
+        type (CompletedJobFlowStatusModulesItemAgentActionsItemType0Type):
+        module_id (str):
+    """
+
+    job_id: str
+    function_name: str
+    type: CompletedJobFlowStatusModulesItemAgentActionsItemType0Type
+    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 = CompletedJobFlowStatusModulesItemAgentActionsItemType0Type(d.pop("type"))
+
+        module_id = d.pop("module_id")
+
+        completed_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,
+        )
+
+        completed_job_flow_status_modules_item_agent_actions_item_type_0.additional_properties = d
+        return completed_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/completed_job_flow_status_modules_item_agent_actions_item_type_0_type.py

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

windmill_api/models/completed_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.completed_job_flow_status_modules_item_agent_actions_item_type_1_type import (
+    CompletedJobFlowStatusModulesItemAgentActionsItemType1Type,
+)
+
+T = TypeVar("T", bound="CompletedJobFlowStatusModulesItemAgentActionsItemType1")
+
+
+@_attrs_define
+class CompletedJobFlowStatusModulesItemAgentActionsItemType1:
+    """
+    Attributes:
+        type (CompletedJobFlowStatusModulesItemAgentActionsItemType1Type):
+    """
+
+    type: CompletedJobFlowStatusModulesItemAgentActionsItemType1Type
+    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 = CompletedJobFlowStatusModulesItemAgentActionsItemType1Type(d.pop("type"))
+
+        completed_job_flow_status_modules_item_agent_actions_item_type_1 = cls(
+            type=type,
+        )
+
+        completed_job_flow_status_modules_item_agent_actions_item_type_1.additional_properties = d
+        return completed_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/completed_job_flow_status_modules_item_agent_actions_item_type_1_type.py

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

windmill_api/models/completed_job_flow_status_preprocessor_module.py

@@ -7,6 +7,12 @@ from ..models.completed_job_flow_status_preprocessor_module_type import Complete
 from ..types import UNSET, Unset
 
 if TYPE_CHECKING:
+    from ..models.completed_job_flow_status_preprocessor_module_agent_actions_item_type_0 import (
+        CompletedJobFlowStatusPreprocessorModuleAgentActionsItemType0,
+    )
+    from ..models.completed_job_flow_status_preprocessor_module_agent_actions_item_type_1 import (
+        CompletedJobFlowStatusPreprocessorModuleAgentActionsItemType1,
+    )
     from ..models.completed_job_flow_status_preprocessor_module_approvers_item import (
         CompletedJobFlowStatusPreprocessorModuleApproversItem,
     )
@@ -41,6 +47,9 @@ class CompletedJobFlowStatusPreprocessorModule:
         approvers (Union[Unset, List['CompletedJobFlowStatusPreprocessorModuleApproversItem']]):
         failed_retries (Union[Unset, List[str]]):
         skipped (Union[Unset, bool]):
+        agent_actions (Union[Unset, List[Union['CompletedJobFlowStatusPreprocessorModuleAgentActionsItemType0',
+            'CompletedJobFlowStatusPreprocessorModuleAgentActionsItemType1']]]):
+        agent_actions_success (Union[Unset, List[bool]]):
     """
 
     type: CompletedJobFlowStatusPreprocessorModuleType
@@ -56,9 +65,23 @@ class CompletedJobFlowStatusPreprocessorModule:
     approvers: Union[Unset, List["CompletedJobFlowStatusPreprocessorModuleApproversItem"]] = UNSET
     failed_retries: Union[Unset, List[str]] = UNSET
     skipped: Union[Unset, bool] = UNSET
+    agent_actions: Union[
+        Unset,
+        List[
+            Union[
+                "CompletedJobFlowStatusPreprocessorModuleAgentActionsItemType0",
+                "CompletedJobFlowStatusPreprocessorModuleAgentActionsItemType1",
+            ]
+        ],
+    ] = 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.completed_job_flow_status_preprocessor_module_agent_actions_item_type_0 import (
+            CompletedJobFlowStatusPreprocessorModuleAgentActionsItemType0,
+        )
+
         type = self.type.value
 
         id = self.id
@@ -98,6 +121,23 @@ class CompletedJobFlowStatusPreprocessorModule:
             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, CompletedJobFlowStatusPreprocessorModuleAgentActionsItemType0):
+                    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)
@@ -130,11 +170,21 @@ class CompletedJobFlowStatusPreprocessorModule:
             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.completed_job_flow_status_preprocessor_module_agent_actions_item_type_0 import (
+            CompletedJobFlowStatusPreprocessorModuleAgentActionsItemType0,
+        )
+        from ..models.completed_job_flow_status_preprocessor_module_agent_actions_item_type_1 import (
+            CompletedJobFlowStatusPreprocessorModuleAgentActionsItemType1,
+        )
         from ..models.completed_job_flow_status_preprocessor_module_approvers_item import (
             CompletedJobFlowStatusPreprocessorModuleApproversItem,
         )
@@ -195,6 +245,40 @@ class CompletedJobFlowStatusPreprocessorModule:
 
         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[
+                "CompletedJobFlowStatusPreprocessorModuleAgentActionsItemType0",
+                "CompletedJobFlowStatusPreprocessorModuleAgentActionsItemType1",
+            ]:
+                try:
+                    if not isinstance(data, dict):
+                        raise TypeError()
+                    agent_actions_item_type_0 = CompletedJobFlowStatusPreprocessorModuleAgentActionsItemType0.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 = CompletedJobFlowStatusPreprocessorModuleAgentActionsItemType1.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))
+
         completed_job_flow_status_preprocessor_module = cls(
             type=type,
             id=id,
@@ -209,6 +293,8 @@ class CompletedJobFlowStatusPreprocessorModule:
             approvers=approvers,
             failed_retries=failed_retries,
             skipped=skipped,
+            agent_actions=agent_actions,
+            agent_actions_success=agent_actions_success,
         )
 
         completed_job_flow_status_preprocessor_module.additional_properties = d