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

windmill_api/api/job/run_flow_preview_and_wait_result.py

@@ -0,0 +1,107 @@
+from http import HTTPStatus
+from typing import Any, Dict, Optional, Union
+
+import httpx
+
+from ... import errors
+from ...client import AuthenticatedClient, Client
+from ...models.run_flow_preview_and_wait_result_json_body import RunFlowPreviewAndWaitResultJsonBody
+from ...types import Response
+
+
+def _get_kwargs(
+    workspace: str,
+    *,
+    json_body: RunFlowPreviewAndWaitResultJsonBody,
+) -> Dict[str, Any]:
+    pass
+
+    json_json_body = json_body.to_dict()
+
+    return {
+        "method": "post",
+        "url": "/w/{workspace}/jobs/run_wait_result/preview_flow".format(
+            workspace=workspace,
+        ),
+        "json": json_json_body,
+    }
+
+
+def _parse_response(*, client: Union[AuthenticatedClient, Client], response: httpx.Response) -> Optional[Any]:
+    if response.status_code == HTTPStatus.OK:
+        return None
+    if client.raise_on_unexpected_status:
+        raise errors.UnexpectedStatus(response.status_code, response.content)
+    else:
+        return None
+
+
+def _build_response(*, client: Union[AuthenticatedClient, Client], response: httpx.Response) -> Response[Any]:
+    return Response(
+        status_code=HTTPStatus(response.status_code),
+        content=response.content,
+        headers=response.headers,
+        parsed=_parse_response(client=client, response=response),
+    )
+
+
+def sync_detailed(
+    workspace: str,
+    *,
+    client: Union[AuthenticatedClient, Client],
+    json_body: RunFlowPreviewAndWaitResultJsonBody,
+) -> Response[Any]:
+    """run flow preview and wait for result
+
+    Args:
+        workspace (str):
+        json_body (RunFlowPreviewAndWaitResultJsonBody):
+
+    Raises:
+        errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
+        httpx.TimeoutException: If the request takes longer than Client.timeout.
+
+    Returns:
+        Response[Any]
+    """
+
+    kwargs = _get_kwargs(
+        workspace=workspace,
+        json_body=json_body,
+    )
+
+    response = client.get_httpx_client().request(
+        **kwargs,
+    )
+
+    return _build_response(client=client, response=response)
+
+
+async def asyncio_detailed(
+    workspace: str,
+    *,
+    client: Union[AuthenticatedClient, Client],
+    json_body: RunFlowPreviewAndWaitResultJsonBody,
+) -> Response[Any]:
+    """run flow preview and wait for result
+
+    Args:
+        workspace (str):
+        json_body (RunFlowPreviewAndWaitResultJsonBody):
+
+    Raises:
+        errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
+        httpx.TimeoutException: If the request takes longer than Client.timeout.
+
+    Returns:
+        Response[Any]
+    """
+
+    kwargs = _get_kwargs(
+        workspace=workspace,
+        json_body=json_body,
+    )
+
+    response = await client.get_async_httpx_client().request(**kwargs)
+
+    return _build_response(client=client, response=response)

windmill_api/api/job/run_script_preview_and_wait_result.py

@@ -0,0 +1,107 @@
+from http import HTTPStatus
+from typing import Any, Dict, Optional, Union
+
+import httpx
+
+from ... import errors
+from ...client import AuthenticatedClient, Client
+from ...models.run_script_preview_and_wait_result_json_body import RunScriptPreviewAndWaitResultJsonBody
+from ...types import Response
+
+
+def _get_kwargs(
+    workspace: str,
+    *,
+    json_body: RunScriptPreviewAndWaitResultJsonBody,
+) -> Dict[str, Any]:
+    pass
+
+    json_json_body = json_body.to_dict()
+
+    return {
+        "method": "post",
+        "url": "/w/{workspace}/jobs/run_wait_result/preview".format(
+            workspace=workspace,
+        ),
+        "json": json_json_body,
+    }
+
+
+def _parse_response(*, client: Union[AuthenticatedClient, Client], response: httpx.Response) -> Optional[Any]:
+    if response.status_code == HTTPStatus.OK:
+        return None
+    if client.raise_on_unexpected_status:
+        raise errors.UnexpectedStatus(response.status_code, response.content)
+    else:
+        return None
+
+
+def _build_response(*, client: Union[AuthenticatedClient, Client], response: httpx.Response) -> Response[Any]:
+    return Response(
+        status_code=HTTPStatus(response.status_code),
+        content=response.content,
+        headers=response.headers,
+        parsed=_parse_response(client=client, response=response),
+    )
+
+
+def sync_detailed(
+    workspace: str,
+    *,
+    client: Union[AuthenticatedClient, Client],
+    json_body: RunScriptPreviewAndWaitResultJsonBody,
+) -> Response[Any]:
+    """run script preview and wait for result
+
+    Args:
+        workspace (str):
+        json_body (RunScriptPreviewAndWaitResultJsonBody):
+
+    Raises:
+        errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
+        httpx.TimeoutException: If the request takes longer than Client.timeout.
+
+    Returns:
+        Response[Any]
+    """
+
+    kwargs = _get_kwargs(
+        workspace=workspace,
+        json_body=json_body,
+    )
+
+    response = client.get_httpx_client().request(
+        **kwargs,
+    )
+
+    return _build_response(client=client, response=response)
+
+
+async def asyncio_detailed(
+    workspace: str,
+    *,
+    client: Union[AuthenticatedClient, Client],
+    json_body: RunScriptPreviewAndWaitResultJsonBody,
+) -> Response[Any]:
+    """run script preview and wait for result
+
+    Args:
+        workspace (str):
+        json_body (RunScriptPreviewAndWaitResultJsonBody):
+
+    Raises:
+        errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
+        httpx.TimeoutException: If the request takes longer than Client.timeout.
+
+    Returns:
+        Response[Any]
+    """
+
+    kwargs = _get_kwargs(
+        workspace=workspace,
+        json_body=json_body,
+    )
+
+    response = await client.get_async_httpx_client().request(**kwargs)
+
+    return _build_response(client=client, response=response)

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)