# 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 .conversation_history_transcript_tool_call_common_model_input import (
    ConversationHistoryTranscriptToolCallCommonModelInput,
)


class WorkflowToolResponseModelInputStepsItem_Edge(UncheckedBaseModel):
    type: typing.Literal["edge"] = "edge"
    step_latency_secs: float
    edge_id: str
    target_node_id: str

    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 WorkflowToolResponseModelInputStepsItem_MaxIterationsExceeded(UncheckedBaseModel):
    type: typing.Literal["max_iterations_exceeded"] = "max_iterations_exceeded"
    step_latency_secs: float
    max_iterations: int

    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 WorkflowToolResponseModelInputStepsItem_NestedTools(UncheckedBaseModel):
    type: typing.Literal["nested_tools"] = "nested_tools"
    step_latency_secs: float
    node_id: str
    requests: typing.List[ConversationHistoryTranscriptToolCallCommonModelInput]
    results: typing.List["WorkflowToolNestedToolsStepModelInputResultsItem"]
    is_successful: bool

    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


WorkflowToolResponseModelInputStepsItem = typing_extensions.Annotated[
    typing.Union[
        WorkflowToolResponseModelInputStepsItem_Edge,
        WorkflowToolResponseModelInputStepsItem_MaxIterationsExceeded,
        WorkflowToolResponseModelInputStepsItem_NestedTools,
    ],
    UnionMetadata(discriminant="type"),
]
from .conversation_history_transcript_workflow_tools_result_common_model_input import (
    ConversationHistoryTranscriptWorkflowToolsResultCommonModelInput,
)  # noqa: E402, I001
from .workflow_tool_nested_tools_step_model_input_results_item import WorkflowToolNestedToolsStepModelInputResultsItem  # noqa: E402, I001
from .workflow_tool_response_model_input import WorkflowToolResponseModelInput  # noqa: E402, I001

update_forward_refs(
    WorkflowToolResponseModelInputStepsItem_NestedTools,
    ConversationHistoryTranscriptWorkflowToolsResultCommonModelInput=ConversationHistoryTranscriptWorkflowToolsResultCommonModelInput,
    WorkflowToolNestedToolsStepModelInputResultsItem=WorkflowToolNestedToolsStepModelInputResultsItem,
    WorkflowToolResponseModelInput=WorkflowToolResponseModelInput,
    WorkflowToolResponseModelInputStepsItem=WorkflowToolResponseModelInputStepsItem,
)
