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

from ...core.unchecked_base_model import UncheckedBaseModel
from .templates_create_agents_response_agents_item_sources_item_embedding_config_embedding_endpoint_type import (
    TemplatesCreateAgentsResponseAgentsItemSourcesItemEmbeddingConfigEmbeddingEndpointType,
)
import typing
from .templates_create_agents_response_agents_item_sources_item_embedding_config_embedding_endpoint import (
    TemplatesCreateAgentsResponseAgentsItemSourcesItemEmbeddingConfigEmbeddingEndpoint,
)
from .templates_create_agents_response_agents_item_sources_item_embedding_config_embedding_chunk_size import (
    TemplatesCreateAgentsResponseAgentsItemSourcesItemEmbeddingConfigEmbeddingChunkSize,
)
from .templates_create_agents_response_agents_item_sources_item_embedding_config_handle import (
    TemplatesCreateAgentsResponseAgentsItemSourcesItemEmbeddingConfigHandle,
)
from .templates_create_agents_response_agents_item_sources_item_embedding_config_azure_endpoint import (
    TemplatesCreateAgentsResponseAgentsItemSourcesItemEmbeddingConfigAzureEndpoint,
)
from .templates_create_agents_response_agents_item_sources_item_embedding_config_azure_version import (
    TemplatesCreateAgentsResponseAgentsItemSourcesItemEmbeddingConfigAzureVersion,
)
from .templates_create_agents_response_agents_item_sources_item_embedding_config_azure_deployment import (
    TemplatesCreateAgentsResponseAgentsItemSourcesItemEmbeddingConfigAzureDeployment,
)
from ...core.pydantic_utilities import IS_PYDANTIC_V2
import pydantic


class TemplatesCreateAgentsResponseAgentsItemSourcesItemEmbeddingConfig(UncheckedBaseModel):
    embedding_endpoint_type: TemplatesCreateAgentsResponseAgentsItemSourcesItemEmbeddingConfigEmbeddingEndpointType
    embedding_endpoint: typing.Optional[
        TemplatesCreateAgentsResponseAgentsItemSourcesItemEmbeddingConfigEmbeddingEndpoint
    ] = None
    embedding_model: str
    embedding_dim: float
    embedding_chunk_size: typing.Optional[
        TemplatesCreateAgentsResponseAgentsItemSourcesItemEmbeddingConfigEmbeddingChunkSize
    ] = None
    handle: typing.Optional[TemplatesCreateAgentsResponseAgentsItemSourcesItemEmbeddingConfigHandle] = None
    azure_endpoint: typing.Optional[TemplatesCreateAgentsResponseAgentsItemSourcesItemEmbeddingConfigAzureEndpoint] = (
        None
    )
    azure_version: typing.Optional[TemplatesCreateAgentsResponseAgentsItemSourcesItemEmbeddingConfigAzureVersion] = None
    azure_deployment: typing.Optional[
        TemplatesCreateAgentsResponseAgentsItemSourcesItemEmbeddingConfigAzureDeployment
    ] = 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
