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

windmill_api/models/queued_job_flow_status_preprocessor_module.py

@@ -7,6 +7,12 @@ from ..models.queued_job_flow_status_preprocessor_module_type import QueuedJobFl
 from ..types import UNSET, Unset
 
 if TYPE_CHECKING:
+    from ..models.queued_job_flow_status_preprocessor_module_agent_actions_item_type_0 import (
+        QueuedJobFlowStatusPreprocessorModuleAgentActionsItemType0,
+    )
+    from ..models.queued_job_flow_status_preprocessor_module_agent_actions_item_type_1 import (
+        QueuedJobFlowStatusPreprocessorModuleAgentActionsItemType1,
+    )
     from ..models.queued_job_flow_status_preprocessor_module_approvers_item import (
         QueuedJobFlowStatusPreprocessorModuleApproversItem,
     )
@@ -41,6 +47,9 @@ class QueuedJobFlowStatusPreprocessorModule:
         approvers (Union[Unset, List['QueuedJobFlowStatusPreprocessorModuleApproversItem']]):
         failed_retries (Union[Unset, List[str]]):
         skipped (Union[Unset, bool]):
+        agent_actions (Union[Unset, List[Union['QueuedJobFlowStatusPreprocessorModuleAgentActionsItemType0',
+            'QueuedJobFlowStatusPreprocessorModuleAgentActionsItemType1']]]):
+        agent_actions_success (Union[Unset, List[bool]]):
     """
 
     type: QueuedJobFlowStatusPreprocessorModuleType
@@ -56,9 +65,23 @@ class QueuedJobFlowStatusPreprocessorModule:
     approvers: Union[Unset, List["QueuedJobFlowStatusPreprocessorModuleApproversItem"]] = UNSET
     failed_retries: Union[Unset, List[str]] = UNSET
     skipped: Union[Unset, bool] = UNSET
+    agent_actions: Union[
+        Unset,
+        List[
+            Union[
+                "QueuedJobFlowStatusPreprocessorModuleAgentActionsItemType0",
+                "QueuedJobFlowStatusPreprocessorModuleAgentActionsItemType1",
+            ]
+        ],
+    ] = 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_preprocessor_module_agent_actions_item_type_0 import (
+            QueuedJobFlowStatusPreprocessorModuleAgentActionsItemType0,
+        )
+
         type = self.type.value
 
         id = self.id
@@ -98,6 +121,23 @@ class QueuedJobFlowStatusPreprocessorModule:
             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, QueuedJobFlowStatusPreprocessorModuleAgentActionsItemType0):
+                    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 QueuedJobFlowStatusPreprocessorModule:
             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_preprocessor_module_agent_actions_item_type_0 import (
+            QueuedJobFlowStatusPreprocessorModuleAgentActionsItemType0,
+        )
+        from ..models.queued_job_flow_status_preprocessor_module_agent_actions_item_type_1 import (
+            QueuedJobFlowStatusPreprocessorModuleAgentActionsItemType1,
+        )
         from ..models.queued_job_flow_status_preprocessor_module_approvers_item import (
             QueuedJobFlowStatusPreprocessorModuleApproversItem,
         )
@@ -195,6 +245,38 @@ class QueuedJobFlowStatusPreprocessorModule:
 
         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[
+                "QueuedJobFlowStatusPreprocessorModuleAgentActionsItemType0",
+                "QueuedJobFlowStatusPreprocessorModuleAgentActionsItemType1",
+            ]:
+                try:
+                    if not isinstance(data, dict):
+                        raise TypeError()
+                    agent_actions_item_type_0 = QueuedJobFlowStatusPreprocessorModuleAgentActionsItemType0.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 = QueuedJobFlowStatusPreprocessorModuleAgentActionsItemType1.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_preprocessor_module = cls(
             type=type,
             id=id,
@@ -209,6 +291,8 @@ class QueuedJobFlowStatusPreprocessorModule:
             approvers=approvers,
             failed_retries=failed_retries,
             skipped=skipped,
+            agent_actions=agent_actions,
+            agent_actions_success=agent_actions_success,
         )
 
         queued_job_flow_status_preprocessor_module.additional_properties = d

windmill_api/models/queued_job_flow_status_preprocessor_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_preprocessor_module_agent_actions_item_type_0_type import (
+    QueuedJobFlowStatusPreprocessorModuleAgentActionsItemType0Type,
+)
+
+T = TypeVar("T", bound="QueuedJobFlowStatusPreprocessorModuleAgentActionsItemType0")
+
+
+@_attrs_define
+class QueuedJobFlowStatusPreprocessorModuleAgentActionsItemType0:
+    """
+    Attributes:
+        job_id (str):
+        function_name (str):
+        type (QueuedJobFlowStatusPreprocessorModuleAgentActionsItemType0Type):
+        module_id (str):
+    """
+
+    job_id: str
+    function_name: str
+    type: QueuedJobFlowStatusPreprocessorModuleAgentActionsItemType0Type
+    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 = QueuedJobFlowStatusPreprocessorModuleAgentActionsItemType0Type(d.pop("type"))
+
+        module_id = d.pop("module_id")
+
+        queued_job_flow_status_preprocessor_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_preprocessor_module_agent_actions_item_type_0.additional_properties = d
+        return queued_job_flow_status_preprocessor_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_preprocessor_module_agent_actions_item_type_0_type.py

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

windmill_api/models/queued_job_flow_status_preprocessor_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_preprocessor_module_agent_actions_item_type_1_type import (
+    QueuedJobFlowStatusPreprocessorModuleAgentActionsItemType1Type,
+)
+
+T = TypeVar("T", bound="QueuedJobFlowStatusPreprocessorModuleAgentActionsItemType1")
+
+
+@_attrs_define
+class QueuedJobFlowStatusPreprocessorModuleAgentActionsItemType1:
+    """
+    Attributes:
+        type (QueuedJobFlowStatusPreprocessorModuleAgentActionsItemType1Type):
+    """
+
+    type: QueuedJobFlowStatusPreprocessorModuleAgentActionsItemType1Type
+    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 = QueuedJobFlowStatusPreprocessorModuleAgentActionsItemType1Type(d.pop("type"))
+
+        queued_job_flow_status_preprocessor_module_agent_actions_item_type_1 = cls(
+            type=type,
+        )
+
+        queued_job_flow_status_preprocessor_module_agent_actions_item_type_1.additional_properties = d
+        return queued_job_flow_status_preprocessor_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_preprocessor_module_agent_actions_item_type_1_type.py

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

windmill_api/models/queued_job_job_kind.py

@@ -2,6 +2,7 @@ from enum import Enum
 
 
 class QueuedJobJobKind(str, Enum):
+    AIAGENT = "aiagent"
     APPDEPENDENCIES = "appdependencies"
     APPSCRIPT = "appscript"
     DEPENDENCIES = "dependencies"

windmill_api/models/run_flow_preview_and_wait_result_json_body.py

@@ -0,0 +1,114 @@
+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 ..types import UNSET, Unset
+
+if TYPE_CHECKING:
+    from ..models.run_flow_preview_and_wait_result_json_body_args import RunFlowPreviewAndWaitResultJsonBodyArgs
+    from ..models.run_flow_preview_and_wait_result_json_body_restarted_from import (
+        RunFlowPreviewAndWaitResultJsonBodyRestartedFrom,
+    )
+    from ..models.run_flow_preview_and_wait_result_json_body_value import RunFlowPreviewAndWaitResultJsonBodyValue
+
+
+T = TypeVar("T", bound="RunFlowPreviewAndWaitResultJsonBody")
+
+
+@_attrs_define
+class RunFlowPreviewAndWaitResultJsonBody:
+    """
+    Attributes:
+        value (RunFlowPreviewAndWaitResultJsonBodyValue):
+        args (RunFlowPreviewAndWaitResultJsonBodyArgs): The arguments to pass to the script or flow
+        path (Union[Unset, str]):
+        tag (Union[Unset, str]):
+        restarted_from (Union[Unset, RunFlowPreviewAndWaitResultJsonBodyRestartedFrom]):
+    """
+
+    value: "RunFlowPreviewAndWaitResultJsonBodyValue"
+    args: "RunFlowPreviewAndWaitResultJsonBodyArgs"
+    path: Union[Unset, str] = UNSET
+    tag: Union[Unset, str] = UNSET
+    restarted_from: Union[Unset, "RunFlowPreviewAndWaitResultJsonBodyRestartedFrom"] = UNSET
+    additional_properties: Dict[str, Any] = _attrs_field(init=False, factory=dict)
+
+    def to_dict(self) -> Dict[str, Any]:
+        value = self.value.to_dict()
+
+        args = self.args.to_dict()
+
+        path = self.path
+        tag = self.tag
+        restarted_from: Union[Unset, Dict[str, Any]] = UNSET
+        if not isinstance(self.restarted_from, Unset):
+            restarted_from = self.restarted_from.to_dict()
+
+        field_dict: Dict[str, Any] = {}
+        field_dict.update(self.additional_properties)
+        field_dict.update(
+            {
+                "value": value,
+                "args": args,
+            }
+        )
+        if path is not UNSET:
+            field_dict["path"] = path
+        if tag is not UNSET:
+            field_dict["tag"] = tag
+        if restarted_from is not UNSET:
+            field_dict["restarted_from"] = restarted_from
+
+        return field_dict
+
+    @classmethod
+    def from_dict(cls: Type[T], src_dict: Dict[str, Any]) -> T:
+        from ..models.run_flow_preview_and_wait_result_json_body_args import RunFlowPreviewAndWaitResultJsonBodyArgs
+        from ..models.run_flow_preview_and_wait_result_json_body_restarted_from import (
+            RunFlowPreviewAndWaitResultJsonBodyRestartedFrom,
+        )
+        from ..models.run_flow_preview_and_wait_result_json_body_value import RunFlowPreviewAndWaitResultJsonBodyValue
+
+        d = src_dict.copy()
+        value = RunFlowPreviewAndWaitResultJsonBodyValue.from_dict(d.pop("value"))
+
+        args = RunFlowPreviewAndWaitResultJsonBodyArgs.from_dict(d.pop("args"))
+
+        path = d.pop("path", UNSET)
+
+        tag = d.pop("tag", UNSET)
+
+        _restarted_from = d.pop("restarted_from", UNSET)
+        restarted_from: Union[Unset, RunFlowPreviewAndWaitResultJsonBodyRestartedFrom]
+        if isinstance(_restarted_from, Unset):
+            restarted_from = UNSET
+        else:
+            restarted_from = RunFlowPreviewAndWaitResultJsonBodyRestartedFrom.from_dict(_restarted_from)
+
+        run_flow_preview_and_wait_result_json_body = cls(
+            value=value,
+            args=args,
+            path=path,
+            tag=tag,
+            restarted_from=restarted_from,
+        )
+
+        run_flow_preview_and_wait_result_json_body.additional_properties = d
+        return run_flow_preview_and_wait_result_json_body
+
+    @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/run_flow_preview_and_wait_result_json_body_args.py

@@ -0,0 +1,44 @@
+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="RunFlowPreviewAndWaitResultJsonBodyArgs")
+
+
+@_attrs_define
+class RunFlowPreviewAndWaitResultJsonBodyArgs:
+    """The arguments to pass to the script or flow"""
+
+    additional_properties: Dict[str, Any] = _attrs_field(init=False, factory=dict)
+
+    def to_dict(self) -> Dict[str, Any]:
+        field_dict: Dict[str, Any] = {}
+        field_dict.update(self.additional_properties)
+        field_dict.update({})
+
+        return field_dict
+
+    @classmethod
+    def from_dict(cls: Type[T], src_dict: Dict[str, Any]) -> T:
+        d = src_dict.copy()
+        run_flow_preview_and_wait_result_json_body_args = cls()
+
+        run_flow_preview_and_wait_result_json_body_args.additional_properties = d
+        return run_flow_preview_and_wait_result_json_body_args
+
+    @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/run_flow_preview_and_wait_result_json_body_restarted_from.py

@@ -0,0 +1,74 @@
+from typing import Any, Dict, List, Type, TypeVar, Union
+
+from attrs import define as _attrs_define
+from attrs import field as _attrs_field
+
+from ..types import UNSET, Unset
+
+T = TypeVar("T", bound="RunFlowPreviewAndWaitResultJsonBodyRestartedFrom")
+
+
+@_attrs_define
+class RunFlowPreviewAndWaitResultJsonBodyRestartedFrom:
+    """
+    Attributes:
+        flow_job_id (Union[Unset, str]):
+        step_id (Union[Unset, str]):
+        branch_or_iteration_n (Union[Unset, int]):
+    """
+
+    flow_job_id: Union[Unset, str] = UNSET
+    step_id: Union[Unset, str] = UNSET
+    branch_or_iteration_n: Union[Unset, int] = UNSET
+    additional_properties: Dict[str, Any] = _attrs_field(init=False, factory=dict)
+
+    def to_dict(self) -> Dict[str, Any]:
+        flow_job_id = self.flow_job_id
+        step_id = self.step_id
+        branch_or_iteration_n = self.branch_or_iteration_n
+
+        field_dict: Dict[str, Any] = {}
+        field_dict.update(self.additional_properties)
+        field_dict.update({})
+        if flow_job_id is not UNSET:
+            field_dict["flow_job_id"] = flow_job_id
+        if step_id is not UNSET:
+            field_dict["step_id"] = step_id
+        if branch_or_iteration_n is not UNSET:
+            field_dict["branch_or_iteration_n"] = branch_or_iteration_n
+
+        return field_dict
+
+    @classmethod
+    def from_dict(cls: Type[T], src_dict: Dict[str, Any]) -> T:
+        d = src_dict.copy()
+        flow_job_id = d.pop("flow_job_id", UNSET)
+
+        step_id = d.pop("step_id", UNSET)
+
+        branch_or_iteration_n = d.pop("branch_or_iteration_n", UNSET)
+
+        run_flow_preview_and_wait_result_json_body_restarted_from = cls(
+            flow_job_id=flow_job_id,
+            step_id=step_id,
+            branch_or_iteration_n=branch_or_iteration_n,
+        )
+
+        run_flow_preview_and_wait_result_json_body_restarted_from.additional_properties = d
+        return run_flow_preview_and_wait_result_json_body_restarted_from
+
+    @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