vellum-ai 0.3.23__py3-none-any.whl → 0.5.0__py3-none-any.whl

vellum/resources/registered_prompts/client.py

@@ -1,327 +0,0 @@
-# This file was auto-generated by Fern from our API Definition.
-
-import typing
-import urllib.parse
-from json.decoder import JSONDecodeError
-
-from ...core.api_error import ApiError
-from ...core.client_wrapper import AsyncClientWrapper, SyncClientWrapper
-from ...core.jsonable_encoder import jsonable_encoder
-from ...core.remove_none_from_dict import remove_none_from_dict
-from ...core.request_options import RequestOptions
-from ...errors.bad_request_error import BadRequestError
-from ...errors.conflict_error import ConflictError
-from ...errors.not_found_error import NotFoundError
-from ...types.provider_enum import ProviderEnum
-from ...types.register_prompt_error_response import RegisterPromptErrorResponse
-from ...types.register_prompt_model_parameters_request import RegisterPromptModelParametersRequest
-from ...types.register_prompt_prompt_info_request import RegisterPromptPromptInfoRequest
-from ...types.register_prompt_response import RegisterPromptResponse
-
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
-
-# this is used as the default value for optional parameters
-OMIT = typing.cast(typing.Any, ...)
-
-
-class RegisteredPromptsClient:
-    def __init__(self, *, client_wrapper: SyncClientWrapper):
-        self._client_wrapper = client_wrapper
-
-    def register_prompt(
-        self,
-        *,
-        label: str,
-        name: str,
-        prompt: RegisterPromptPromptInfoRequest,
-        provider: typing.Optional[ProviderEnum] = OMIT,
-        model: str,
-        parameters: RegisterPromptModelParametersRequest,
-        meta: typing.Optional[typing.Dict[str, typing.Any]] = OMIT,
-        request_options: typing.Optional[RequestOptions] = None,
-    ) -> RegisterPromptResponse:
-        """
-        Registers a prompt within Vellum and creates associated Vellum entities. Intended to be used by integration
-        partners, not directly by Vellum users.
-
-        Under the hood, this endpoint creates a new sandbox, a new model version, and a new deployment.
-
-        Parameters:
-            - label: str. A human-friendly label for corresponding entities created in Vellum.
-
-            - name: str. A uniquely-identifying name for corresponding entities created in Vellum.
-
-            - prompt: RegisterPromptPromptInfoRequest. Information about how to execute the prompt template.
-
-            - provider: typing.Optional[ProviderEnum]. The initial LLM provider to use for this prompt
-
-                                                       * `ANTHROPIC` - Anthropic
-                                                       * `AWS_BEDROCK` - AWS Bedrock
-                                                       * `AZURE_OPENAI` - Azure OpenAI
-                                                       * `COHERE` - Cohere
-                                                       * `GOOGLE` - Google
-                                                       * `HOSTED` - Hosted
-                                                       * `MOSAICML` - MosaicML
-                                                       * `OPENAI` - OpenAI
-                                                       * `FIREWORKS_AI` - Fireworks AI
-                                                       * `HUGGINGFACE` - HuggingFace
-                                                       * `MYSTIC` - Mystic
-                                                       * `PYQ` - Pyq
-                                                       * `REPLICATE` - Replicate
-            - model: str. The initial model to use for this prompt
-
-            - parameters: RegisterPromptModelParametersRequest. The initial model parameters to use for  this prompt
-
-            - meta: typing.Optional[typing.Dict[str, typing.Any]]. Optionally include additional metadata to store along with the prompt.
-
-            - request_options: typing.Optional[RequestOptions]. Request-specific configuration.
-        ---
-        from vellum import (
-            BlockTypeEnum,
-            PromptTemplateBlockDataRequest,
-            PromptTemplateBlockPropertiesRequest,
-            PromptTemplateBlockRequest,
-            RegisteredPromptInputVariableRequest,
-            RegisterPromptModelParametersRequest,
-            RegisterPromptPromptInfoRequest,
-        )
-        from vellum.client import Vellum
-
-        client = Vellum(
-            api_key="YOUR_API_KEY",
-        )
-        client.registered_prompts.register_prompt(
-            label="label",
-            name="name",
-            prompt=RegisterPromptPromptInfoRequest(
-                prompt_block_data=PromptTemplateBlockDataRequest(
-                    version=1,
-                    blocks=[
-                        PromptTemplateBlockRequest(
-                            id="id",
-                            block_type=BlockTypeEnum.CHAT_MESSAGE,
-                            properties=PromptTemplateBlockPropertiesRequest(),
-                        )
-                    ],
-                ),
-                input_variables=[
-                    RegisteredPromptInputVariableRequest(
-                        key="key",
-                    )
-                ],
-            ),
-            model="model",
-            parameters=RegisterPromptModelParametersRequest(
-                temperature=1.1,
-                max_tokens=1,
-                top_p=1.1,
-                frequency_penalty=1.1,
-                presence_penalty=1.1,
-            ),
-        )
-        """
-        _request: typing.Dict[str, typing.Any] = {
-            "label": label,
-            "name": name,
-            "prompt": prompt,
-            "model": model,
-            "parameters": parameters,
-        }
-        if provider is not OMIT:
-            _request["provider"] = provider
-        if meta is not OMIT:
-            _request["meta"] = meta
-        _response = self._client_wrapper.httpx_client.request(
-            "POST",
-            urllib.parse.urljoin(
-                f"{self._client_wrapper.get_environment().default}/", "v1/registered-prompts/register"
-            ),
-            params=jsonable_encoder(
-                request_options.get("additional_query_parameters") if request_options is not None else None
-            ),
-            json=jsonable_encoder(_request)
-            if request_options is None or request_options.get("additional_body_parameters") is None
-            else {
-                **jsonable_encoder(_request),
-                **(jsonable_encoder(remove_none_from_dict(request_options.get("additional_body_parameters", {})))),
-            },
-            headers=jsonable_encoder(
-                remove_none_from_dict(
-                    {
-                        **self._client_wrapper.get_headers(),
-                        **(request_options.get("additional_headers", {}) if request_options is not None else {}),
-                    }
-                )
-            ),
-            timeout=request_options.get("timeout_in_seconds")
-            if request_options is not None and request_options.get("timeout_in_seconds") is not None
-            else self._client_wrapper.get_timeout(),
-            retries=0,
-            max_retries=request_options.get("max_retries") if request_options is not None else 0,  # type: ignore
-        )
-        if 200 <= _response.status_code < 300:
-            return pydantic.parse_obj_as(RegisterPromptResponse, _response.json())  # type: ignore
-        if _response.status_code == 400:
-            raise BadRequestError(pydantic.parse_obj_as(typing.Any, _response.json()))  # type: ignore
-        if _response.status_code == 404:
-            raise NotFoundError(pydantic.parse_obj_as(typing.Any, _response.json()))  # type: ignore
-        if _response.status_code == 409:
-            raise ConflictError(pydantic.parse_obj_as(RegisterPromptErrorResponse, _response.json()))  # type: ignore
-        try:
-            _response_json = _response.json()
-        except JSONDecodeError:
-            raise ApiError(status_code=_response.status_code, body=_response.text)
-        raise ApiError(status_code=_response.status_code, body=_response_json)
-
-
-class AsyncRegisteredPromptsClient:
-    def __init__(self, *, client_wrapper: AsyncClientWrapper):
-        self._client_wrapper = client_wrapper
-
-    async def register_prompt(
-        self,
-        *,
-        label: str,
-        name: str,
-        prompt: RegisterPromptPromptInfoRequest,
-        provider: typing.Optional[ProviderEnum] = OMIT,
-        model: str,
-        parameters: RegisterPromptModelParametersRequest,
-        meta: typing.Optional[typing.Dict[str, typing.Any]] = OMIT,
-        request_options: typing.Optional[RequestOptions] = None,
-    ) -> RegisterPromptResponse:
-        """
-        Registers a prompt within Vellum and creates associated Vellum entities. Intended to be used by integration
-        partners, not directly by Vellum users.
-
-        Under the hood, this endpoint creates a new sandbox, a new model version, and a new deployment.
-
-        Parameters:
-            - label: str. A human-friendly label for corresponding entities created in Vellum.
-
-            - name: str. A uniquely-identifying name for corresponding entities created in Vellum.
-
-            - prompt: RegisterPromptPromptInfoRequest. Information about how to execute the prompt template.
-
-            - provider: typing.Optional[ProviderEnum]. The initial LLM provider to use for this prompt
-
-                                                       * `ANTHROPIC` - Anthropic
-                                                       * `AWS_BEDROCK` - AWS Bedrock
-                                                       * `AZURE_OPENAI` - Azure OpenAI
-                                                       * `COHERE` - Cohere
-                                                       * `GOOGLE` - Google
-                                                       * `HOSTED` - Hosted
-                                                       * `MOSAICML` - MosaicML
-                                                       * `OPENAI` - OpenAI
-                                                       * `FIREWORKS_AI` - Fireworks AI
-                                                       * `HUGGINGFACE` - HuggingFace
-                                                       * `MYSTIC` - Mystic
-                                                       * `PYQ` - Pyq
-                                                       * `REPLICATE` - Replicate
-            - model: str. The initial model to use for this prompt
-
-            - parameters: RegisterPromptModelParametersRequest. The initial model parameters to use for  this prompt
-
-            - meta: typing.Optional[typing.Dict[str, typing.Any]]. Optionally include additional metadata to store along with the prompt.
-
-            - request_options: typing.Optional[RequestOptions]. Request-specific configuration.
-        ---
-        from vellum import (
-            BlockTypeEnum,
-            PromptTemplateBlockDataRequest,
-            PromptTemplateBlockPropertiesRequest,
-            PromptTemplateBlockRequest,
-            RegisteredPromptInputVariableRequest,
-            RegisterPromptModelParametersRequest,
-            RegisterPromptPromptInfoRequest,
-        )
-        from vellum.client import AsyncVellum
-
-        client = AsyncVellum(
-            api_key="YOUR_API_KEY",
-        )
-        await client.registered_prompts.register_prompt(
-            label="label",
-            name="name",
-            prompt=RegisterPromptPromptInfoRequest(
-                prompt_block_data=PromptTemplateBlockDataRequest(
-                    version=1,
-                    blocks=[
-                        PromptTemplateBlockRequest(
-                            id="id",
-                            block_type=BlockTypeEnum.CHAT_MESSAGE,
-                            properties=PromptTemplateBlockPropertiesRequest(),
-                        )
-                    ],
-                ),
-                input_variables=[
-                    RegisteredPromptInputVariableRequest(
-                        key="key",
-                    )
-                ],
-            ),
-            model="model",
-            parameters=RegisterPromptModelParametersRequest(
-                temperature=1.1,
-                max_tokens=1,
-                top_p=1.1,
-                frequency_penalty=1.1,
-                presence_penalty=1.1,
-            ),
-        )
-        """
-        _request: typing.Dict[str, typing.Any] = {
-            "label": label,
-            "name": name,
-            "prompt": prompt,
-            "model": model,
-            "parameters": parameters,
-        }
-        if provider is not OMIT:
-            _request["provider"] = provider
-        if meta is not OMIT:
-            _request["meta"] = meta
-        _response = await self._client_wrapper.httpx_client.request(
-            "POST",
-            urllib.parse.urljoin(
-                f"{self._client_wrapper.get_environment().default}/", "v1/registered-prompts/register"
-            ),
-            params=jsonable_encoder(
-                request_options.get("additional_query_parameters") if request_options is not None else None
-            ),
-            json=jsonable_encoder(_request)
-            if request_options is None or request_options.get("additional_body_parameters") is None
-            else {
-                **jsonable_encoder(_request),
-                **(jsonable_encoder(remove_none_from_dict(request_options.get("additional_body_parameters", {})))),
-            },
-            headers=jsonable_encoder(
-                remove_none_from_dict(
-                    {
-                        **self._client_wrapper.get_headers(),
-                        **(request_options.get("additional_headers", {}) if request_options is not None else {}),
-                    }
-                )
-            ),
-            timeout=request_options.get("timeout_in_seconds")
-            if request_options is not None and request_options.get("timeout_in_seconds") is not None
-            else self._client_wrapper.get_timeout(),
-            retries=0,
-            max_retries=request_options.get("max_retries") if request_options is not None else 0,  # type: ignore
-        )
-        if 200 <= _response.status_code < 300:
-            return pydantic.parse_obj_as(RegisterPromptResponse, _response.json())  # type: ignore
-        if _response.status_code == 400:
-            raise BadRequestError(pydantic.parse_obj_as(typing.Any, _response.json()))  # type: ignore
-        if _response.status_code == 404:
-            raise NotFoundError(pydantic.parse_obj_as(typing.Any, _response.json()))  # type: ignore
-        if _response.status_code == 409:
-            raise ConflictError(pydantic.parse_obj_as(RegisterPromptErrorResponse, _response.json()))  # type: ignore
-        try:
-            _response_json = _response.json()
-        except JSONDecodeError:
-            raise ApiError(status_code=_response.status_code, body=_response.text)
-        raise ApiError(status_code=_response.status_code, body=_response_json)

vellum/types/block_type_enum.py

@@ -1,36 +0,0 @@
-# This file was auto-generated by Fern from our API Definition.
-
-import enum
-import typing
-
-T_Result = typing.TypeVar("T_Result")
-
-
-class BlockTypeEnum(str, enum.Enum):
-    """
-    - `CHAT_MESSAGE` - CHAT_MESSAGE
-    - `CHAT_HISTORY` - CHAT_HISTORY
-    - `JINJA` - JINJA
-    - `FUNCTION_DEFINITION` - FUNCTION_DEFINITION
-    """
-
-    CHAT_MESSAGE = "CHAT_MESSAGE"
-    CHAT_HISTORY = "CHAT_HISTORY"
-    JINJA = "JINJA"
-    FUNCTION_DEFINITION = "FUNCTION_DEFINITION"
-
-    def visit(
-        self,
-        chat_message: typing.Callable[[], T_Result],
-        chat_history: typing.Callable[[], T_Result],
-        jinja: typing.Callable[[], T_Result],
-        function_definition: typing.Callable[[], T_Result],
-    ) -> T_Result:
-        if self is BlockTypeEnum.CHAT_MESSAGE:
-            return chat_message()
-        if self is BlockTypeEnum.CHAT_HISTORY:
-            return chat_history()
-        if self is BlockTypeEnum.JINJA:
-            return jinja()
-        if self is BlockTypeEnum.FUNCTION_DEFINITION:
-            return function_definition()

vellum/types/model_version_build_config.py

@@ -1,40 +0,0 @@
-# This file was auto-generated by Fern from our API Definition.
-
-import datetime as dt
-import typing
-
-from ..core.datetime_utils import serialize_datetime
-from .model_version_sandbox_snapshot import ModelVersionSandboxSnapshot
-
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
-
-
-class ModelVersionBuildConfig(pydantic.BaseModel):
-    base_model: str = pydantic.Field()
-    """
-    The name of the base model used to create this model version, as identified by the LLM provider.
-    """
-
-    sandbox_snapshot: typing.Optional[ModelVersionSandboxSnapshot] = pydantic.Field(default=None)
-    """
-    Information about the sandbox snapshot that was used to create this model version, if applicable.
-    """
-
-    prompt_version_id: typing.Optional[str] = None
-
-    def json(self, **kwargs: typing.Any) -> str:
-        kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
-        return super().json(**kwargs_with_defaults)
-
-    def dict(self, **kwargs: typing.Any) -> typing.Dict[str, typing.Any]:
-        kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
-        return super().dict(**kwargs_with_defaults)
-
-    class Config:
-        frozen = True
-        smart_union = True
-        extra = pydantic.Extra.allow
-        json_encoders = {dt.datetime: serialize_datetime}

vellum/types/model_version_exec_config.py

@@ -1,48 +0,0 @@
-# This file was auto-generated by Fern from our API Definition.
-
-import datetime as dt
-import typing
-
-from ..core.datetime_utils import serialize_datetime
-from .model_version_exec_config_parameters import ModelVersionExecConfigParameters
-from .prompt_template_block_data import PromptTemplateBlockData
-from .vellum_variable import VellumVariable
-
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
-
-
-class ModelVersionExecConfig(pydantic.BaseModel):
-    parameters: ModelVersionExecConfigParameters = pydantic.Field()
-    """
-    The generation parameters that are passed to the LLM provider at runtime.
-    """
-
-    input_variables: typing.List[VellumVariable] = pydantic.Field()
-    """
-    Input variables specified in the prompt template.
-    """
-
-    prompt_template: typing.Optional[str] = pydantic.Field(default=None)
-    """
-    The template used to generate prompts for this model version.
-    """
-
-    prompt_block_data: typing.Optional[PromptTemplateBlockData] = None
-    prompt_syntax_version: typing.Optional[int] = None
-
-    def json(self, **kwargs: typing.Any) -> str:
-        kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
-        return super().json(**kwargs_with_defaults)
-
-    def dict(self, **kwargs: typing.Any) -> typing.Dict[str, typing.Any]:
-        kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
-        return super().dict(**kwargs_with_defaults)
-
-    class Config:
-        frozen = True
-        smart_union = True
-        extra = pydantic.Extra.allow
-        json_encoders = {dt.datetime: serialize_datetime}

vellum/types/model_version_exec_config_parameters.py

@@ -1,37 +0,0 @@
-# This file was auto-generated by Fern from our API Definition.
-
-import datetime as dt
-import typing
-
-from ..core.datetime_utils import serialize_datetime
-
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
-
-
-class ModelVersionExecConfigParameters(pydantic.BaseModel):
-    temperature: typing.Optional[float] = None
-    max_tokens: typing.Optional[int] = None
-    top_p: float
-    frequency_penalty: float
-    presence_penalty: float
-    logit_bias: typing.Optional[typing.Dict[str, typing.Optional[float]]] = None
-    stop: typing.Optional[typing.List[str]] = None
-    top_k: typing.Optional[float] = None
-    custom_parameters: typing.Optional[typing.Dict[str, typing.Any]] = None
-
-    def json(self, **kwargs: typing.Any) -> str:
-        kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
-        return super().json(**kwargs_with_defaults)
-
-    def dict(self, **kwargs: typing.Any) -> typing.Dict[str, typing.Any]:
-        kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
-        return super().dict(**kwargs_with_defaults)
-
-    class Config:
-        frozen = True
-        smart_union = True
-        extra = pydantic.Extra.allow
-        json_encoders = {dt.datetime: serialize_datetime}

vellum/types/model_version_read.py

@@ -1,82 +0,0 @@
-# This file was auto-generated by Fern from our API Definition.
-
-import datetime as dt
-import typing
-
-from ..core.datetime_utils import serialize_datetime
-from .model_version_build_config import ModelVersionBuildConfig
-from .model_version_exec_config import ModelVersionExecConfig
-from .model_version_read_status_enum import ModelVersionReadStatusEnum
-from .provider_enum import ProviderEnum
-
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
-
-
-class ModelVersionRead(pydantic.BaseModel):
-    id: str = pydantic.Field()
-    """
-    Vellum-generated ID that uniquely identifies this model version.
-    """
-
-    created: dt.datetime = pydantic.Field()
-    """
-    Timestamp of when this model version was created.
-    """
-
-    label: str = pydantic.Field()
-    """
-    Human-friendly name for this model version.
-    """
-
-    provider: ProviderEnum = pydantic.Field()
-    """
-    Which LLM provider this model version is associated with.
-    
-    - `ANTHROPIC` - Anthropic
-    - `AWS_BEDROCK` - AWS Bedrock
-    - `AZURE_OPENAI` - Azure OpenAI
-    - `COHERE` - Cohere
-    - `GOOGLE` - Google
-    - `HOSTED` - Hosted
-    - `MOSAICML` - MosaicML
-    - `OPENAI` - OpenAI
-    - `FIREWORKS_AI` - Fireworks AI
-    - `HUGGINGFACE` - HuggingFace
-    - `MYSTIC` - Mystic
-    - `PYQ` - Pyq
-    - `REPLICATE` - Replicate
-    """
-
-    external_id: str = pydantic.Field()
-    """
-    The unique id of this model version as it exists in the above provider's system.
-    """
-
-    build_config: ModelVersionBuildConfig = pydantic.Field()
-    """
-    Configuration used to build this model version.
-    """
-
-    exec_config: ModelVersionExecConfig = pydantic.Field()
-    """
-    Configuration used to execute this model version.
-    """
-
-    status: typing.Optional[ModelVersionReadStatusEnum] = None
-
-    def json(self, **kwargs: typing.Any) -> str:
-        kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
-        return super().json(**kwargs_with_defaults)
-
-    def dict(self, **kwargs: typing.Any) -> typing.Dict[str, typing.Any]:
-        kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
-        return super().dict(**kwargs_with_defaults)
-
-    class Config:
-        frozen = True
-        smart_union = True
-        extra = pydantic.Extra.allow
-        json_encoders = {dt.datetime: serialize_datetime}

vellum/types/model_version_read_status_enum.py

@@ -1,36 +0,0 @@
-# This file was auto-generated by Fern from our API Definition.
-
-import enum
-import typing
-
-T_Result = typing.TypeVar("T_Result")
-
-
-class ModelVersionReadStatusEnum(str, enum.Enum):
-    """
-    - `CREATING` - Creating
-    - `READY` - Ready
-    - `CREATION_FAILED` - Creation Failed
-    - `DISABLED` - Disabled
-    """
-
-    CREATING = "CREATING"
-    READY = "READY"
-    CREATION_FAILED = "CREATION_FAILED"
-    DISABLED = "DISABLED"
-
-    def visit(
-        self,
-        creating: typing.Callable[[], T_Result],
-        ready: typing.Callable[[], T_Result],
-        creation_failed: typing.Callable[[], T_Result],
-        disabled: typing.Callable[[], T_Result],
-    ) -> T_Result:
-        if self is ModelVersionReadStatusEnum.CREATING:
-            return creating()
-        if self is ModelVersionReadStatusEnum.READY:
-            return ready()
-        if self is ModelVersionReadStatusEnum.CREATION_FAILED:
-            return creation_failed()
-        if self is ModelVersionReadStatusEnum.DISABLED:
-            return disabled()

vellum/types/model_version_sandbox_snapshot.py

@@ -1,44 +0,0 @@
-# This file was auto-generated by Fern from our API Definition.
-
-import datetime as dt
-import typing
-
-from ..core.datetime_utils import serialize_datetime
-
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
-
-
-class ModelVersionSandboxSnapshot(pydantic.BaseModel):
-    id: str = pydantic.Field()
-    """
-    The ID of the sandbox snapshot.
-    """
-
-    prompt_index: typing.Optional[int] = pydantic.Field(default=None)
-    """
-    The index of the prompt in the sandbox snapshot.
-    """
-
-    prompt_id: typing.Optional[str] = pydantic.Field(default=None)
-    """
-    The id of the prompt in the sandbox snapshot.
-    """
-
-    sandbox_id: typing.Optional[str] = None
-
-    def json(self, **kwargs: typing.Any) -> str:
-        kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
-        return super().json(**kwargs_with_defaults)
-
-    def dict(self, **kwargs: typing.Any) -> typing.Dict[str, typing.Any]:
-        kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
-        return super().dict(**kwargs_with_defaults)
-
-    class Config:
-        frozen = True
-        smart_union = True
-        extra = pydantic.Extra.allow
-        json_encoders = {dt.datetime: serialize_datetime}