2025-04-02 09:01:55 +02:00

34 lines
1.0 KiB
Python

"""Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
from __future__ import annotations
from .documenturlchunk import DocumentURLChunk, DocumentURLChunkTypedDict
from .imageurlchunk import ImageURLChunk, ImageURLChunkTypedDict
from .referencechunk import ReferenceChunk, ReferenceChunkTypedDict
from .textchunk import TextChunk, TextChunkTypedDict
from mistralai.utils import get_discriminator
from pydantic import Discriminator, Tag
from typing import Union
from typing_extensions import Annotated, TypeAliasType
ContentChunkTypedDict = TypeAliasType(
"ContentChunkTypedDict",
Union[
TextChunkTypedDict,
ImageURLChunkTypedDict,
ReferenceChunkTypedDict,
DocumentURLChunkTypedDict,
],
)
ContentChunk = Annotated[
Union[
Annotated[ImageURLChunk, Tag("image_url")],
Annotated[DocumentURLChunk, Tag("document_url")],
Annotated[TextChunk, Tag("text")],
Annotated[ReferenceChunk, Tag("reference")],
],
Discriminator(lambda m: get_discriminator(m, "type", "type")),
]