# 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 .task_message import TaskMessage
from .task_message_content import TaskMessageContent
from .task_message_delta import TaskMessageDelta


class TaskMessageUpdate_Delta(UniversalBaseModel):
    type: typing.Literal["delta"] = "delta"
    delta: typing.Optional[TaskMessageDelta] = None
    index: typing.Optional[int] = None
    parent_task_message: typing.Optional[TaskMessage] = 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 TaskMessageUpdate_Done(UniversalBaseModel):
    type: typing.Literal["done"] = "done"
    index: typing.Optional[int] = None
    parent_task_message: typing.Optional[TaskMessage] = 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 TaskMessageUpdate_Full(UniversalBaseModel):
    type: typing.Literal["full"] = "full"
    content: TaskMessageContent
    index: typing.Optional[int] = None
    parent_task_message: typing.Optional[TaskMessage] = 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 TaskMessageUpdate_Start(UniversalBaseModel):
    type: typing.Literal["start"] = "start"
    content: TaskMessageContent
    index: typing.Optional[int] = None
    parent_task_message: typing.Optional[TaskMessage] = 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


TaskMessageUpdate = typing_extensions.Annotated[
    typing.Union[TaskMessageUpdate_Delta, TaskMessageUpdate_Done, TaskMessageUpdate_Full, TaskMessageUpdate_Start],
    pydantic.Field(discriminator="type"),
]
