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

import typing

import pydantic
from ...core.pydantic_utilities import IS_PYDANTIC_V2
from ...core.unchecked_base_model import UncheckedBaseModel
from .templates_create_agents_from_template_response_agents_item_tool_rules_item_child_output_mapping_default_child import (
    TemplatesCreateAgentsFromTemplateResponseAgentsItemToolRulesItemChildOutputMappingDefaultChild,
)
from .templates_create_agents_from_template_response_agents_item_tool_rules_item_child_output_mapping_prompt_template import (
    TemplatesCreateAgentsFromTemplateResponseAgentsItemToolRulesItemChildOutputMappingPromptTemplate,
)
from .templates_create_agents_from_template_response_agents_item_tool_rules_item_child_output_mapping_require_output_mapping import (
    TemplatesCreateAgentsFromTemplateResponseAgentsItemToolRulesItemChildOutputMappingRequireOutputMapping,
)
from .templates_create_agents_from_template_response_agents_item_tool_rules_item_child_output_mapping_type import (
    TemplatesCreateAgentsFromTemplateResponseAgentsItemToolRulesItemChildOutputMappingType,
)


class TemplatesCreateAgentsFromTemplateResponseAgentsItemToolRulesItemChildOutputMapping(UncheckedBaseModel):
    tool_name: str
    type: typing.Optional[TemplatesCreateAgentsFromTemplateResponseAgentsItemToolRulesItemChildOutputMappingType] = None
    prompt_template: typing.Optional[
        TemplatesCreateAgentsFromTemplateResponseAgentsItemToolRulesItemChildOutputMappingPromptTemplate
    ] = None
    default_child: typing.Optional[
        TemplatesCreateAgentsFromTemplateResponseAgentsItemToolRulesItemChildOutputMappingDefaultChild
    ] = None
    child_output_mapping: typing.Optional[typing.Optional[typing.Any]] = None
    require_output_mapping: typing.Optional[
        TemplatesCreateAgentsFromTemplateResponseAgentsItemToolRulesItemChildOutputMappingRequireOutputMapping
    ] = 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
