# 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, UniversalBaseModel
from ...core.serialization import FieldMetadata


class SendMessageAgentsRequestConversation_New(UniversalBaseModel):
    type: typing.Literal["new"] = "new"
    data_input_variable_selections: typing_extensions.Annotated[
        typing.Optional[typing.Dict[str, typing.Optional[typing.List[str]]]],
        FieldMetadata(alias="dataInputVariableSelections"),
        pydantic.Field(alias="dataInputVariableSelections"),
    ] = 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 SendMessageAgentsRequestConversation_Existing(UniversalBaseModel):
    type: typing.Literal["existing"] = "existing"
    conversation_id: typing_extensions.Annotated[
        str, FieldMetadata(alias="conversationId"), pydantic.Field(alias="conversationId")
    ]

    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


SendMessageAgentsRequestConversation = typing_extensions.Annotated[
    typing.Union[SendMessageAgentsRequestConversation_New, SendMessageAgentsRequestConversation_Existing],
    pydantic.Field(discriminator="type"),
]
