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

windmill_api/models/ai_agent.py

@@ -0,0 +1,103 @@
+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.ai_agent_type import AiAgentType
+from ..types import UNSET, Unset
+
+if TYPE_CHECKING:
+    from ..models.ai_agent_input_transforms import AiAgentInputTransforms
+    from ..models.ai_agent_tools_item import AiAgentToolsItem
+
+
+T = TypeVar("T", bound="AiAgent")
+
+
+@_attrs_define
+class AiAgent:
+    """
+    Attributes:
+        input_transforms (AiAgentInputTransforms):
+        tools (List['AiAgentToolsItem']):
+        type (AiAgentType):
+        parallel (Union[Unset, bool]):
+    """
+
+    input_transforms: "AiAgentInputTransforms"
+    tools: List["AiAgentToolsItem"]
+    type: AiAgentType
+    parallel: Union[Unset, bool] = UNSET
+    additional_properties: Dict[str, Any] = _attrs_field(init=False, factory=dict)
+
+    def to_dict(self) -> Dict[str, Any]:
+        input_transforms = self.input_transforms.to_dict()
+
+        tools = []
+        for tools_item_data in self.tools:
+            tools_item = tools_item_data.to_dict()
+
+            tools.append(tools_item)
+
+        type = self.type.value
+
+        parallel = self.parallel
+
+        field_dict: Dict[str, Any] = {}
+        field_dict.update(self.additional_properties)
+        field_dict.update(
+            {
+                "input_transforms": input_transforms,
+                "tools": tools,
+                "type": type,
+            }
+        )
+        if parallel is not UNSET:
+            field_dict["parallel"] = parallel
+
+        return field_dict
+
+    @classmethod
+    def from_dict(cls: Type[T], src_dict: Dict[str, Any]) -> T:
+        from ..models.ai_agent_input_transforms import AiAgentInputTransforms
+        from ..models.ai_agent_tools_item import AiAgentToolsItem
+
+        d = src_dict.copy()
+        input_transforms = AiAgentInputTransforms.from_dict(d.pop("input_transforms"))
+
+        tools = []
+        _tools = d.pop("tools")
+        for tools_item_data in _tools:
+            tools_item = AiAgentToolsItem.from_dict(tools_item_data)
+
+            tools.append(tools_item)
+
+        type = AiAgentType(d.pop("type"))
+
+        parallel = d.pop("parallel", UNSET)
+
+        ai_agent = cls(
+            input_transforms=input_transforms,
+            tools=tools,
+            type=type,
+            parallel=parallel,
+        )
+
+        ai_agent.additional_properties = d
+        return ai_agent
+
+    @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/ai_agent_input_transforms.py

@@ -0,0 +1,104 @@
+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
+
+if TYPE_CHECKING:
+    from ..models.ai_agent_input_transforms_additional_property_type_0 import (
+        AiAgentInputTransformsAdditionalPropertyType0,
+    )
+    from ..models.ai_agent_input_transforms_additional_property_type_1 import (
+        AiAgentInputTransformsAdditionalPropertyType1,
+    )
+
+
+T = TypeVar("T", bound="AiAgentInputTransforms")
+
+
+@_attrs_define
+class AiAgentInputTransforms:
+    """ """
+
+    additional_properties: Dict[
+        str, Union["AiAgentInputTransformsAdditionalPropertyType0", "AiAgentInputTransformsAdditionalPropertyType1"]
+    ] = _attrs_field(init=False, factory=dict)
+
+    def to_dict(self) -> Dict[str, Any]:
+        from ..models.ai_agent_input_transforms_additional_property_type_0 import (
+            AiAgentInputTransformsAdditionalPropertyType0,
+        )
+
+        field_dict: Dict[str, Any] = {}
+        for prop_name, prop in self.additional_properties.items():
+            if isinstance(prop, AiAgentInputTransformsAdditionalPropertyType0):
+                field_dict[prop_name] = prop.to_dict()
+
+            else:
+                field_dict[prop_name] = prop.to_dict()
+
+        field_dict.update({})
+
+        return field_dict
+
+    @classmethod
+    def from_dict(cls: Type[T], src_dict: Dict[str, Any]) -> T:
+        from ..models.ai_agent_input_transforms_additional_property_type_0 import (
+            AiAgentInputTransformsAdditionalPropertyType0,
+        )
+        from ..models.ai_agent_input_transforms_additional_property_type_1 import (
+            AiAgentInputTransformsAdditionalPropertyType1,
+        )
+
+        d = src_dict.copy()
+        ai_agent_input_transforms = cls()
+
+        additional_properties = {}
+        for prop_name, prop_dict in d.items():
+
+            def _parse_additional_property(
+                data: object,
+            ) -> Union[
+                "AiAgentInputTransformsAdditionalPropertyType0", "AiAgentInputTransformsAdditionalPropertyType1"
+            ]:
+                try:
+                    if not isinstance(data, dict):
+                        raise TypeError()
+                    additional_property_type_0 = AiAgentInputTransformsAdditionalPropertyType0.from_dict(data)
+
+                    return additional_property_type_0
+                except:  # noqa: E722
+                    pass
+                if not isinstance(data, dict):
+                    raise TypeError()
+                additional_property_type_1 = AiAgentInputTransformsAdditionalPropertyType1.from_dict(data)
+
+                return additional_property_type_1
+
+            additional_property = _parse_additional_property(prop_dict)
+
+            additional_properties[prop_name] = additional_property
+
+        ai_agent_input_transforms.additional_properties = additional_properties
+        return ai_agent_input_transforms
+
+    @property
+    def additional_keys(self) -> List[str]:
+        return list(self.additional_properties.keys())
+
+    def __getitem__(
+        self, key: str
+    ) -> Union["AiAgentInputTransformsAdditionalPropertyType0", "AiAgentInputTransformsAdditionalPropertyType1"]:
+        return self.additional_properties[key]
+
+    def __setitem__(
+        self,
+        key: str,
+        value: Union["AiAgentInputTransformsAdditionalPropertyType0", "AiAgentInputTransformsAdditionalPropertyType1"],
+    ) -> 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/ai_agent_input_transforms_additional_property_type_0.py

@@ -0,0 +1,69 @@
+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_input_transforms_additional_property_type_0_type import (
+    AiAgentInputTransformsAdditionalPropertyType0Type,
+)
+
+T = TypeVar("T", bound="AiAgentInputTransformsAdditionalPropertyType0")
+
+
+@_attrs_define
+class AiAgentInputTransformsAdditionalPropertyType0:
+    """
+    Attributes:
+        value (Any):
+        type (AiAgentInputTransformsAdditionalPropertyType0Type):
+    """
+
+    value: Any
+    type: AiAgentInputTransformsAdditionalPropertyType0Type
+    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,
+            }
+        )
+
+        return field_dict
+
+    @classmethod
+    def from_dict(cls: Type[T], src_dict: Dict[str, Any]) -> T:
+        d = src_dict.copy()
+        value = d.pop("value")
+
+        type = AiAgentInputTransformsAdditionalPropertyType0Type(d.pop("type"))
+
+        ai_agent_input_transforms_additional_property_type_0 = cls(
+            value=value,
+            type=type,
+        )
+
+        ai_agent_input_transforms_additional_property_type_0.additional_properties = d
+        return ai_agent_input_transforms_additional_property_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/ai_agent_input_transforms_additional_property_type_0_type.py

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

windmill_api/models/ai_agent_input_transforms_additional_property_type_1.py

@@ -0,0 +1,69 @@
+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_input_transforms_additional_property_type_1_type import (
+    AiAgentInputTransformsAdditionalPropertyType1Type,
+)
+
+T = TypeVar("T", bound="AiAgentInputTransformsAdditionalPropertyType1")
+
+
+@_attrs_define
+class AiAgentInputTransformsAdditionalPropertyType1:
+    """
+    Attributes:
+        expr (str):
+        type (AiAgentInputTransformsAdditionalPropertyType1Type):
+    """
+
+    expr: str
+    type: AiAgentInputTransformsAdditionalPropertyType1Type
+    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,
+            }
+        )
+
+        return field_dict
+
+    @classmethod
+    def from_dict(cls: Type[T], src_dict: Dict[str, Any]) -> T:
+        d = src_dict.copy()
+        expr = d.pop("expr")
+
+        type = AiAgentInputTransformsAdditionalPropertyType1Type(d.pop("type"))
+
+        ai_agent_input_transforms_additional_property_type_1 = cls(
+            expr=expr,
+            type=type,
+        )
+
+        ai_agent_input_transforms_additional_property_type_1.additional_properties = d
+        return ai_agent_input_transforms_additional_property_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/ai_agent_input_transforms_additional_property_type_1_type.py

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

windmill_api/models/ai_agent_tools_item.py

@@ -0,0 +1,280 @@
+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.ai_agent_tools_item_mock import AiAgentToolsItemMock
+    from ..models.ai_agent_tools_item_retry import AiAgentToolsItemRetry
+    from ..models.ai_agent_tools_item_skip_if import AiAgentToolsItemSkipIf
+    from ..models.ai_agent_tools_item_sleep_type_0 import AiAgentToolsItemSleepType0
+    from ..models.ai_agent_tools_item_sleep_type_1 import AiAgentToolsItemSleepType1
+    from ..models.ai_agent_tools_item_stop_after_all_iters_if import AiAgentToolsItemStopAfterAllItersIf
+    from ..models.ai_agent_tools_item_stop_after_if import AiAgentToolsItemStopAfterIf
+    from ..models.ai_agent_tools_item_suspend import AiAgentToolsItemSuspend
+
+
+T = TypeVar("T", bound="AiAgentToolsItem")
+
+
+@_attrs_define
+class AiAgentToolsItem:
+    """
+    Attributes:
+        id (str):
+        value (Any):
+        stop_after_if (Union[Unset, AiAgentToolsItemStopAfterIf]):
+        stop_after_all_iters_if (Union[Unset, AiAgentToolsItemStopAfterAllItersIf]):
+        skip_if (Union[Unset, AiAgentToolsItemSkipIf]):
+        sleep (Union['AiAgentToolsItemSleepType0', 'AiAgentToolsItemSleepType1', Unset]):
+        cache_ttl (Union[Unset, float]):
+        timeout (Union[Unset, float]):
+        delete_after_use (Union[Unset, bool]):
+        summary (Union[Unset, str]):
+        mock (Union[Unset, AiAgentToolsItemMock]):
+        suspend (Union[Unset, AiAgentToolsItemSuspend]):
+        priority (Union[Unset, float]):
+        continue_on_error (Union[Unset, bool]):
+        retry (Union[Unset, AiAgentToolsItemRetry]):
+    """
+
+    id: str
+    value: Any
+    stop_after_if: Union[Unset, "AiAgentToolsItemStopAfterIf"] = UNSET
+    stop_after_all_iters_if: Union[Unset, "AiAgentToolsItemStopAfterAllItersIf"] = UNSET
+    skip_if: Union[Unset, "AiAgentToolsItemSkipIf"] = UNSET
+    sleep: Union["AiAgentToolsItemSleepType0", "AiAgentToolsItemSleepType1", Unset] = UNSET
+    cache_ttl: Union[Unset, float] = UNSET
+    timeout: Union[Unset, float] = UNSET
+    delete_after_use: Union[Unset, bool] = UNSET
+    summary: Union[Unset, str] = UNSET
+    mock: Union[Unset, "AiAgentToolsItemMock"] = UNSET
+    suspend: Union[Unset, "AiAgentToolsItemSuspend"] = UNSET
+    priority: Union[Unset, float] = UNSET
+    continue_on_error: Union[Unset, bool] = UNSET
+    retry: Union[Unset, "AiAgentToolsItemRetry"] = UNSET
+    additional_properties: Dict[str, Any] = _attrs_field(init=False, factory=dict)
+
+    def to_dict(self) -> Dict[str, Any]:
+        from ..models.ai_agent_tools_item_sleep_type_0 import AiAgentToolsItemSleepType0
+
+        id = self.id
+        value = self.value
+        stop_after_if: Union[Unset, Dict[str, Any]] = UNSET
+        if not isinstance(self.stop_after_if, Unset):
+            stop_after_if = self.stop_after_if.to_dict()
+
+        stop_after_all_iters_if: Union[Unset, Dict[str, Any]] = UNSET
+        if not isinstance(self.stop_after_all_iters_if, Unset):
+            stop_after_all_iters_if = self.stop_after_all_iters_if.to_dict()
+
+        skip_if: Union[Unset, Dict[str, Any]] = UNSET
+        if not isinstance(self.skip_if, Unset):
+            skip_if = self.skip_if.to_dict()
+
+        sleep: Union[Dict[str, Any], Unset]
+        if isinstance(self.sleep, Unset):
+            sleep = UNSET
+
+        elif isinstance(self.sleep, AiAgentToolsItemSleepType0):
+            sleep = UNSET
+            if not isinstance(self.sleep, Unset):
+                sleep = self.sleep.to_dict()
+
+        else:
+            sleep = UNSET
+            if not isinstance(self.sleep, Unset):
+                sleep = self.sleep.to_dict()
+
+        cache_ttl = self.cache_ttl
+        timeout = self.timeout
+        delete_after_use = self.delete_after_use
+        summary = self.summary
+        mock: Union[Unset, Dict[str, Any]] = UNSET
+        if not isinstance(self.mock, Unset):
+            mock = self.mock.to_dict()
+
+        suspend: Union[Unset, Dict[str, Any]] = UNSET
+        if not isinstance(self.suspend, Unset):
+            suspend = self.suspend.to_dict()
+
+        priority = self.priority
+        continue_on_error = self.continue_on_error
+        retry: Union[Unset, Dict[str, Any]] = UNSET
+        if not isinstance(self.retry, Unset):
+            retry = self.retry.to_dict()
+
+        field_dict: Dict[str, Any] = {}
+        field_dict.update(self.additional_properties)
+        field_dict.update(
+            {
+                "id": id,
+                "value": value,
+            }
+        )
+        if stop_after_if is not UNSET:
+            field_dict["stop_after_if"] = stop_after_if
+        if stop_after_all_iters_if is not UNSET:
+            field_dict["stop_after_all_iters_if"] = stop_after_all_iters_if
+        if skip_if is not UNSET:
+            field_dict["skip_if"] = skip_if
+        if sleep is not UNSET:
+            field_dict["sleep"] = sleep
+        if cache_ttl is not UNSET:
+            field_dict["cache_ttl"] = cache_ttl
+        if timeout is not UNSET:
+            field_dict["timeout"] = timeout
+        if delete_after_use is not UNSET:
+            field_dict["delete_after_use"] = delete_after_use
+        if summary is not UNSET:
+            field_dict["summary"] = summary
+        if mock is not UNSET:
+            field_dict["mock"] = mock
+        if suspend is not UNSET:
+            field_dict["suspend"] = suspend
+        if priority is not UNSET:
+            field_dict["priority"] = priority
+        if continue_on_error is not UNSET:
+            field_dict["continue_on_error"] = continue_on_error
+        if retry is not UNSET:
+            field_dict["retry"] = retry
+
+        return field_dict
+
+    @classmethod
+    def from_dict(cls: Type[T], src_dict: Dict[str, Any]) -> T:
+        from ..models.ai_agent_tools_item_mock import AiAgentToolsItemMock
+        from ..models.ai_agent_tools_item_retry import AiAgentToolsItemRetry
+        from ..models.ai_agent_tools_item_skip_if import AiAgentToolsItemSkipIf
+        from ..models.ai_agent_tools_item_sleep_type_0 import AiAgentToolsItemSleepType0
+        from ..models.ai_agent_tools_item_sleep_type_1 import AiAgentToolsItemSleepType1
+        from ..models.ai_agent_tools_item_stop_after_all_iters_if import AiAgentToolsItemStopAfterAllItersIf
+        from ..models.ai_agent_tools_item_stop_after_if import AiAgentToolsItemStopAfterIf
+        from ..models.ai_agent_tools_item_suspend import AiAgentToolsItemSuspend
+
+        d = src_dict.copy()
+        id = d.pop("id")
+
+        value = d.pop("value")
+
+        _stop_after_if = d.pop("stop_after_if", UNSET)
+        stop_after_if: Union[Unset, AiAgentToolsItemStopAfterIf]
+        if isinstance(_stop_after_if, Unset):
+            stop_after_if = UNSET
+        else:
+            stop_after_if = AiAgentToolsItemStopAfterIf.from_dict(_stop_after_if)
+
+        _stop_after_all_iters_if = d.pop("stop_after_all_iters_if", UNSET)
+        stop_after_all_iters_if: Union[Unset, AiAgentToolsItemStopAfterAllItersIf]
+        if isinstance(_stop_after_all_iters_if, Unset):
+            stop_after_all_iters_if = UNSET
+        else:
+            stop_after_all_iters_if = AiAgentToolsItemStopAfterAllItersIf.from_dict(_stop_after_all_iters_if)
+
+        _skip_if = d.pop("skip_if", UNSET)
+        skip_if: Union[Unset, AiAgentToolsItemSkipIf]
+        if isinstance(_skip_if, Unset):
+            skip_if = UNSET
+        else:
+            skip_if = AiAgentToolsItemSkipIf.from_dict(_skip_if)
+
+        def _parse_sleep(data: object) -> Union["AiAgentToolsItemSleepType0", "AiAgentToolsItemSleepType1", Unset]:
+            if isinstance(data, Unset):
+                return data
+            try:
+                if not isinstance(data, dict):
+                    raise TypeError()
+                _sleep_type_0 = data
+                sleep_type_0: Union[Unset, AiAgentToolsItemSleepType0]
+                if isinstance(_sleep_type_0, Unset):
+                    sleep_type_0 = UNSET
+                else:
+                    sleep_type_0 = AiAgentToolsItemSleepType0.from_dict(_sleep_type_0)
+
+                return sleep_type_0
+            except:  # noqa: E722
+                pass
+            if not isinstance(data, dict):
+                raise TypeError()
+            _sleep_type_1 = data
+            sleep_type_1: Union[Unset, AiAgentToolsItemSleepType1]
+            if isinstance(_sleep_type_1, Unset):
+                sleep_type_1 = UNSET
+            else:
+                sleep_type_1 = AiAgentToolsItemSleepType1.from_dict(_sleep_type_1)
+
+            return sleep_type_1
+
+        sleep = _parse_sleep(d.pop("sleep", UNSET))
+
+        cache_ttl = d.pop("cache_ttl", UNSET)
+
+        timeout = d.pop("timeout", UNSET)
+
+        delete_after_use = d.pop("delete_after_use", UNSET)
+
+        summary = d.pop("summary", UNSET)
+
+        _mock = d.pop("mock", UNSET)
+        mock: Union[Unset, AiAgentToolsItemMock]
+        if isinstance(_mock, Unset):
+            mock = UNSET
+        else:
+            mock = AiAgentToolsItemMock.from_dict(_mock)
+
+        _suspend = d.pop("suspend", UNSET)
+        suspend: Union[Unset, AiAgentToolsItemSuspend]
+        if isinstance(_suspend, Unset):
+            suspend = UNSET
+        else:
+            suspend = AiAgentToolsItemSuspend.from_dict(_suspend)
+
+        priority = d.pop("priority", UNSET)
+
+        continue_on_error = d.pop("continue_on_error", UNSET)
+
+        _retry = d.pop("retry", UNSET)
+        retry: Union[Unset, AiAgentToolsItemRetry]
+        if isinstance(_retry, Unset):
+            retry = UNSET
+        else:
+            retry = AiAgentToolsItemRetry.from_dict(_retry)
+
+        ai_agent_tools_item = cls(
+            id=id,
+            value=value,
+            stop_after_if=stop_after_if,
+            stop_after_all_iters_if=stop_after_all_iters_if,
+            skip_if=skip_if,
+            sleep=sleep,
+            cache_ttl=cache_ttl,
+            timeout=timeout,
+            delete_after_use=delete_after_use,
+            summary=summary,
+            mock=mock,
+            suspend=suspend,
+            priority=priority,
+            continue_on_error=continue_on_error,
+            retry=retry,
+        )
+
+        ai_agent_tools_item.additional_properties = d
+        return ai_agent_tools_item
+
+    @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