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

from __future__ import annotations

import typing

import typing_extensions

from .fulfilled_execute_prompt_streaming_response import FulfilledExecutePromptStreamingResponse
from .initiated_execute_prompt_streaming_response import InitiatedExecutePromptStreamingResponse
from .rejected_execute_prompt_streaming_response import RejectedExecutePromptStreamingResponse
from .streaming_execute_prompt_streaming_response import StreamingExecutePromptStreamingResponse


class ExecutePromptStreamingResponse_Initiated(InitiatedExecutePromptStreamingResponse):
    state: typing_extensions.Literal["INITIATED"]

    class Config:
        frozen = True
        smart_union = True
        allow_population_by_field_name = True


class ExecutePromptStreamingResponse_Streaming(StreamingExecutePromptStreamingResponse):
    state: typing_extensions.Literal["STREAMING"]

    class Config:
        frozen = True
        smart_union = True
        allow_population_by_field_name = True


class ExecutePromptStreamingResponse_Fulfilled(FulfilledExecutePromptStreamingResponse):
    state: typing_extensions.Literal["FULFILLED"]

    class Config:
        frozen = True
        smart_union = True
        allow_population_by_field_name = True


class ExecutePromptStreamingResponse_Rejected(RejectedExecutePromptStreamingResponse):
    state: typing_extensions.Literal["REJECTED"]

    class Config:
        frozen = True
        smart_union = True
        allow_population_by_field_name = True


ExecutePromptStreamingResponse = typing.Union[
    ExecutePromptStreamingResponse_Initiated,
    ExecutePromptStreamingResponse_Streaming,
    ExecutePromptStreamingResponse_Fulfilled,
    ExecutePromptStreamingResponse_Rejected,
]
