"""Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""

from __future__ import annotations
from .toolcallstatus import ToolCallStatus
from openrouter.types import BaseModel
from openrouter.utils import validate_open_enum
import pydantic
from pydantic.functional_validators import PlainValidator
from typing import Literal, Optional
from typing_extensions import Annotated, NotRequired, TypedDict


OutputImageGenerationServerToolItemType = Literal["openrouter:image_generation",]


class OutputImageGenerationServerToolItemTypedDict(TypedDict):
    r"""An openrouter:image_generation server tool output item"""

    status: ToolCallStatus
    type: OutputImageGenerationServerToolItemType
    id: NotRequired[str]
    image_b64: NotRequired[str]
    image_url: NotRequired[str]
    revised_prompt: NotRequired[str]


class OutputImageGenerationServerToolItem(BaseModel):
    r"""An openrouter:image_generation server tool output item"""

    status: Annotated[ToolCallStatus, PlainValidator(validate_open_enum(False))]

    type: OutputImageGenerationServerToolItemType

    id: Optional[str] = None

    image_b64: Annotated[Optional[str], pydantic.Field(alias="imageB64")] = None

    image_url: Annotated[Optional[str], pydantic.Field(alias="imageUrl")] = None

    revised_prompt: Annotated[Optional[str], pydantic.Field(alias="revisedPrompt")] = (
        None
    )
