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

import typing

import pydantic
from ...core.pydantic_utilities import IS_PYDANTIC_V2
from ...core.unchecked_base_model import UncheckedBaseModel
from .templates_create_agents_from_template_response_agents_item_embedding_config_azure_deployment import (
    TemplatesCreateAgentsFromTemplateResponseAgentsItemEmbeddingConfigAzureDeployment,
)
from .templates_create_agents_from_template_response_agents_item_embedding_config_azure_endpoint import (
    TemplatesCreateAgentsFromTemplateResponseAgentsItemEmbeddingConfigAzureEndpoint,
)
from .templates_create_agents_from_template_response_agents_item_embedding_config_azure_version import (
    TemplatesCreateAgentsFromTemplateResponseAgentsItemEmbeddingConfigAzureVersion,
)
from .templates_create_agents_from_template_response_agents_item_embedding_config_batch_size import (
    TemplatesCreateAgentsFromTemplateResponseAgentsItemEmbeddingConfigBatchSize,
)
from .templates_create_agents_from_template_response_agents_item_embedding_config_embedding_chunk_size import (
    TemplatesCreateAgentsFromTemplateResponseAgentsItemEmbeddingConfigEmbeddingChunkSize,
)
from .templates_create_agents_from_template_response_agents_item_embedding_config_embedding_endpoint import (
    TemplatesCreateAgentsFromTemplateResponseAgentsItemEmbeddingConfigEmbeddingEndpoint,
)
from .templates_create_agents_from_template_response_agents_item_embedding_config_embedding_endpoint_type import (
    TemplatesCreateAgentsFromTemplateResponseAgentsItemEmbeddingConfigEmbeddingEndpointType,
)
from .templates_create_agents_from_template_response_agents_item_embedding_config_handle import (
    TemplatesCreateAgentsFromTemplateResponseAgentsItemEmbeddingConfigHandle,
)


class TemplatesCreateAgentsFromTemplateResponseAgentsItemEmbeddingConfig(UncheckedBaseModel):
    embedding_endpoint_type: TemplatesCreateAgentsFromTemplateResponseAgentsItemEmbeddingConfigEmbeddingEndpointType
    embedding_endpoint: typing.Optional[
        TemplatesCreateAgentsFromTemplateResponseAgentsItemEmbeddingConfigEmbeddingEndpoint
    ] = None
    embedding_model: str
    embedding_dim: float
    embedding_chunk_size: typing.Optional[
        TemplatesCreateAgentsFromTemplateResponseAgentsItemEmbeddingConfigEmbeddingChunkSize
    ] = None
    handle: typing.Optional[TemplatesCreateAgentsFromTemplateResponseAgentsItemEmbeddingConfigHandle] = None
    batch_size: typing.Optional[TemplatesCreateAgentsFromTemplateResponseAgentsItemEmbeddingConfigBatchSize] = None
    azure_endpoint: typing.Optional[TemplatesCreateAgentsFromTemplateResponseAgentsItemEmbeddingConfigAzureEndpoint] = (
        None
    )
    azure_version: typing.Optional[TemplatesCreateAgentsFromTemplateResponseAgentsItemEmbeddingConfigAzureVersion] = (
        None
    )
    azure_deployment: typing.Optional[
        TemplatesCreateAgentsFromTemplateResponseAgentsItemEmbeddingConfigAzureDeployment
    ] = 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
