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

from __future__ import annotations
from mistralai_gcp.types import BaseModel
from mistralai_gcp.utils import validate_const
import pydantic
from pydantic.functional_validators import AfterValidator
from typing import List, Literal, Optional
from typing_extensions import Annotated, TypedDict


ReferenceChunkType = Literal["reference"]


class ReferenceChunkTypedDict(TypedDict):
    reference_ids: List[int]
    type: ReferenceChunkType


class ReferenceChunk(BaseModel):
    reference_ids: List[int]

    TYPE: Annotated[
        Annotated[
            Optional[ReferenceChunkType], AfterValidator(validate_const("reference"))
        ],
        pydantic.Field(alias="type"),
    ] = "reference"
