# This file was auto-generated by Fern from our API Definition.

from __future__ import annotations

import typing

import pydantic
import typing_extensions
from ..core.pydantic_utilities import IS_PYDANTIC_V2, update_forward_refs
from ..core.unchecked_base_model import UncheckedBaseModel, UnionMetadata
from .conversational_config_api_model_workflow_override_input import ConversationalConfigApiModelWorkflowOverrideInput
from .entry_behavior import EntryBehavior
from .knowledge_base_locator import KnowledgeBaseLocator
from .position_input import PositionInput
from .transfer_type_enum import TransferTypeEnum
from .uui_transfer_config import UuiTransferConfig
from .workflow_phone_number_node_model_input_custom_sip_headers_item import (
    WorkflowPhoneNumberNodeModelInputCustomSipHeadersItem,
)
from .workflow_phone_number_node_model_input_post_dial_digits import WorkflowPhoneNumberNodeModelInputPostDialDigits
from .workflow_phone_number_node_model_input_transfer_destination import (
    WorkflowPhoneNumberNodeModelInputTransferDestination,
)
from .workflow_tool_locator import WorkflowToolLocator


class AgentWorkflowRequestModelNodesValue_End(UncheckedBaseModel):
    type: typing.Literal["end"] = "end"
    position: typing.Optional[PositionInput] = None
    edge_order: typing.Optional[typing.List[str]] = None

    if IS_PYDANTIC_V2:
        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
    else:

        class Config:
            frozen = True
            smart_union = True
            extra = pydantic.Extra.allow


class AgentWorkflowRequestModelNodesValue_OverrideAgent(UncheckedBaseModel):
    type: typing.Literal["override_agent"] = "override_agent"
    position: typing.Optional[PositionInput] = None
    edge_order: typing.Optional[typing.List[str]] = None
    conversation_config: typing.Optional[ConversationalConfigApiModelWorkflowOverrideInput] = None
    additional_prompt: typing.Optional[str] = None
    additional_knowledge_base: typing.Optional[typing.List[KnowledgeBaseLocator]] = None
    additional_tool_ids: typing.Optional[typing.List[str]] = None
    label: str
    entry_behavior: typing.Optional[EntryBehavior] = None

    if IS_PYDANTIC_V2:
        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
    else:

        class Config:
            frozen = True
            smart_union = True
            extra = pydantic.Extra.allow


class AgentWorkflowRequestModelNodesValue_PhoneNumber(UncheckedBaseModel):
    type: typing.Literal["phone_number"] = "phone_number"
    custom_sip_headers: typing.Optional[typing.List[WorkflowPhoneNumberNodeModelInputCustomSipHeadersItem]] = None
    transfer_destination: WorkflowPhoneNumberNodeModelInputTransferDestination
    transfer_type: typing.Optional[TransferTypeEnum] = None
    uui: typing.Optional[UuiTransferConfig] = None
    post_dial_digits: typing.Optional[WorkflowPhoneNumberNodeModelInputPostDialDigits] = None
    position: typing.Optional[PositionInput] = None
    edge_order: typing.Optional[typing.List[str]] = None

    if IS_PYDANTIC_V2:
        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
    else:

        class Config:
            frozen = True
            smart_union = True
            extra = pydantic.Extra.allow


class AgentWorkflowRequestModelNodesValue_StandaloneAgent(UncheckedBaseModel):
    type: typing.Literal["standalone_agent"] = "standalone_agent"
    position: typing.Optional[PositionInput] = None
    edge_order: typing.Optional[typing.List[str]] = None
    agent_id: typing.Optional[str] = None
    node_id: typing.Optional[str] = None
    delay_ms: typing.Optional[int] = None
    transfer_message: typing.Optional[str] = None
    enable_transferred_agent_first_message: typing.Optional[bool] = None
    preserve_client_tts_overrides: typing.Optional[bool] = None

    if IS_PYDANTIC_V2:
        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
    else:

        class Config:
            frozen = True
            smart_union = True
            extra = pydantic.Extra.allow


class AgentWorkflowRequestModelNodesValue_Start(UncheckedBaseModel):
    type: typing.Literal["start"] = "start"
    position: typing.Optional[PositionInput] = None
    edge_order: typing.Optional[typing.List[str]] = None

    if IS_PYDANTIC_V2:
        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
    else:

        class Config:
            frozen = True
            smart_union = True
            extra = pydantic.Extra.allow


class AgentWorkflowRequestModelNodesValue_Tool(UncheckedBaseModel):
    type: typing.Literal["tool"] = "tool"
    position: typing.Optional[PositionInput] = None
    edge_order: typing.Optional[typing.List[str]] = None
    tools: typing.Optional[typing.List[WorkflowToolLocator]] = None

    if IS_PYDANTIC_V2:
        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
    else:

        class Config:
            frozen = True
            smart_union = True
            extra = pydantic.Extra.allow


AgentWorkflowRequestModelNodesValue = typing_extensions.Annotated[
    typing.Union[
        AgentWorkflowRequestModelNodesValue_End,
        AgentWorkflowRequestModelNodesValue_OverrideAgent,
        AgentWorkflowRequestModelNodesValue_PhoneNumber,
        AgentWorkflowRequestModelNodesValue_StandaloneAgent,
        AgentWorkflowRequestModelNodesValue_Start,
        AgentWorkflowRequestModelNodesValue_Tool,
    ],
    UnionMetadata(discriminant="type"),
]
update_forward_refs(AgentWorkflowRequestModelNodesValue_OverrideAgent)
