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

vellum/types/code_execution_node_result.py

@@ -4,15 +4,11 @@ import datetime as dt
 import typing
 
 from ..core.datetime_utils import serialize_datetime
+from ..core.pydantic_utilities import pydantic_v1
 from .code_execution_node_result_data import CodeExecutionNodeResultData
 
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
 
-
-class CodeExecutionNodeResult(pydantic.BaseModel):
+class CodeExecutionNodeResult(pydantic_v1.BaseModel):
     """
     A Node Result Event emitted from a Code Execution Node.
     """
@@ -30,5 +26,5 @@ class CodeExecutionNodeResult(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
-        extra = pydantic.Extra.allow
+        extra = pydantic_v1.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}

vellum/types/code_execution_node_result_data.py

@@ -4,15 +4,11 @@ import datetime as dt
 import typing
 
 from ..core.datetime_utils import serialize_datetime
+from ..core.pydantic_utilities import pydantic_v1
 from .code_execution_node_result_output import CodeExecutionNodeResultOutput
 
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
 
-
-class CodeExecutionNodeResultData(pydantic.BaseModel):
+class CodeExecutionNodeResultData(pydantic_v1.BaseModel):
     output: CodeExecutionNodeResultOutput
     log_output_id: typing.Optional[str] = None
 
@@ -27,5 +23,5 @@ class CodeExecutionNodeResultData(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
-        extra = pydantic.Extra.allow
+        extra = pydantic_v1.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}

vellum/types/code_execution_node_search_results_result.py

@@ -4,15 +4,11 @@ import datetime as dt
 import typing
 
 from ..core.datetime_utils import serialize_datetime
+from ..core.pydantic_utilities import pydantic_v1
 from .search_result import SearchResult
 
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
 
-
-class CodeExecutionNodeSearchResultsResult(pydantic.BaseModel):
+class CodeExecutionNodeSearchResultsResult(pydantic_v1.BaseModel):
     id: str
     value: typing.Optional[typing.List[SearchResult]] = None
 
@@ -27,5 +23,5 @@ class CodeExecutionNodeSearchResultsResult(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
-        extra = pydantic.Extra.allow
+        extra = pydantic_v1.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}

vellum/types/code_execution_node_string_result.py

@@ -4,14 +4,10 @@ import datetime as dt
 import typing
 
 from ..core.datetime_utils import serialize_datetime
+from ..core.pydantic_utilities import pydantic_v1
 
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
 
-
-class CodeExecutionNodeStringResult(pydantic.BaseModel):
+class CodeExecutionNodeStringResult(pydantic_v1.BaseModel):
     id: str
     value: typing.Optional[str] = None
 
@@ -26,5 +22,5 @@ class CodeExecutionNodeStringResult(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
-        extra = pydantic.Extra.allow
+        extra = pydantic_v1.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}

vellum/types/conditional_node_result.py

@@ -4,15 +4,11 @@ import datetime as dt
 import typing
 
 from ..core.datetime_utils import serialize_datetime
+from ..core.pydantic_utilities import pydantic_v1
 from .conditional_node_result_data import ConditionalNodeResultData
 
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
 
-
-class ConditionalNodeResult(pydantic.BaseModel):
+class ConditionalNodeResult(pydantic_v1.BaseModel):
     """
     A Node Result Event emitted from a Conditional Node.
     """
@@ -30,5 +26,5 @@ class ConditionalNodeResult(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
-        extra = pydantic.Extra.allow
+        extra = pydantic_v1.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}

vellum/types/conditional_node_result_data.py

@@ -4,14 +4,10 @@ import datetime as dt
 import typing
 
 from ..core.datetime_utils import serialize_datetime
+from ..core.pydantic_utilities import pydantic_v1
 
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
 
-
-class ConditionalNodeResultData(pydantic.BaseModel):
+class ConditionalNodeResultData(pydantic_v1.BaseModel):
     source_handle_id: typing.Optional[str] = None
 
     def json(self, **kwargs: typing.Any) -> str:
@@ -25,5 +21,5 @@ class ConditionalNodeResultData(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
-        extra = pydantic.Extra.allow
+        extra = pydantic_v1.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}

vellum/types/deployment_provider_payload_response.py

@@ -4,14 +4,10 @@ import datetime as dt
 import typing
 
 from ..core.datetime_utils import serialize_datetime
+from ..core.pydantic_utilities import pydantic_v1
 
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
 
-
-class DeploymentProviderPayloadResponse(pydantic.BaseModel):
+class DeploymentProviderPayloadResponse(pydantic_v1.BaseModel):
     payload: typing.Dict[str, typing.Any]
 
     def json(self, **kwargs: typing.Any) -> str:
@@ -25,5 +21,5 @@ class DeploymentProviderPayloadResponse(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
-        extra = pydantic.Extra.allow
+        extra = pydantic_v1.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}

vellum/types/deployment_read.py

@@ -4,30 +4,26 @@ import datetime as dt
 import typing
 
 from ..core.datetime_utils import serialize_datetime
+from ..core.pydantic_utilities import pydantic_v1
 from .entity_status import EntityStatus
 from .environment_enum import EnvironmentEnum
 from .vellum_variable import VellumVariable
 
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
 
-
-class DeploymentRead(pydantic.BaseModel):
+class DeploymentRead(pydantic_v1.BaseModel):
     id: str
     created: dt.datetime
-    label: str = pydantic.Field()
+    label: str = pydantic_v1.Field()
     """
     A human-readable label for the deployment
     """
 
-    name: str = pydantic.Field()
+    name: str = pydantic_v1.Field()
     """
     A name that uniquely identifies this deployment within its workspace
     """
 
-    status: typing.Optional[EntityStatus] = pydantic.Field(default=None)
+    status: typing.Optional[EntityStatus] = pydantic_v1.Field(default=None)
     """
     The current status of the deployment
     
@@ -35,7 +31,7 @@ class DeploymentRead(pydantic.BaseModel):
     - `ARCHIVED` - Archived
     """
 
-    environment: typing.Optional[EnvironmentEnum] = pydantic.Field(default=None)
+    environment: typing.Optional[EnvironmentEnum] = pydantic_v1.Field(default=None)
     """
     The environment this deployment is used in
     
@@ -46,7 +42,7 @@ class DeploymentRead(pydantic.BaseModel):
 
     last_deployed_on: dt.datetime
     input_variables: typing.List[VellumVariable]
-    active_model_version_ids: typing.List[str] = pydantic.Field()
+    active_model_version_ids: typing.List[str] = pydantic_v1.Field()
     """
     Deprecated. The Prompt execution endpoints return a `prompt_version_id` that could be used instead.
     """
@@ -62,5 +58,5 @@ class DeploymentRead(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
-        extra = pydantic.Extra.allow
+        extra = pydantic_v1.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}

vellum/types/document_document_to_document_index.py

@@ -4,26 +4,22 @@ import datetime as dt
 import typing
 
 from ..core.datetime_utils import serialize_datetime
+from ..core.pydantic_utilities import pydantic_v1
 from .indexing_state_enum import IndexingStateEnum
 
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
 
-
-class DocumentDocumentToDocumentIndex(pydantic.BaseModel):
-    id: str = pydantic.Field()
+class DocumentDocumentToDocumentIndex(pydantic_v1.BaseModel):
+    id: str = pydantic_v1.Field()
     """
     Vellum-generated ID that uniquely identifies this link.
     """
 
-    document_index_id: str = pydantic.Field()
+    document_index_id: str = pydantic_v1.Field()
     """
     Vellum-generated ID that uniquely identifies the index this document is included in.
     """
 
-    indexing_state: typing.Optional[IndexingStateEnum] = pydantic.Field(default=None)
+    indexing_state: typing.Optional[IndexingStateEnum] = pydantic_v1.Field(default=None)
     """
     An enum value representing where this document is along its indexing lifecycle for this index.
     
@@ -45,5 +41,5 @@ class DocumentDocumentToDocumentIndex(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
-        extra = pydantic.Extra.allow
+        extra = pydantic_v1.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}

vellum/types/document_index_read.py

@@ -4,29 +4,25 @@ import datetime as dt
 import typing
 
 from ..core.datetime_utils import serialize_datetime
+from ..core.pydantic_utilities import pydantic_v1
 from .entity_status import EntityStatus
 from .environment_enum import EnvironmentEnum
 
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
 
-
-class DocumentIndexRead(pydantic.BaseModel):
+class DocumentIndexRead(pydantic_v1.BaseModel):
     id: str
     created: dt.datetime
-    label: str = pydantic.Field()
+    label: str = pydantic_v1.Field()
     """
     A human-readable label for the document index
     """
 
-    name: str = pydantic.Field()
+    name: str = pydantic_v1.Field()
     """
     A name that uniquely identifies this index within its workspace
     """
 
-    status: typing.Optional[EntityStatus] = pydantic.Field(default=None)
+    status: typing.Optional[EntityStatus] = pydantic_v1.Field(default=None)
     """
     The current status of the document index
     
@@ -34,7 +30,7 @@ class DocumentIndexRead(pydantic.BaseModel):
     - `ARCHIVED` - Archived
     """
 
-    environment: typing.Optional[EnvironmentEnum] = pydantic.Field(default=None)
+    environment: typing.Optional[EnvironmentEnum] = pydantic_v1.Field(default=None)
     """
     The environment this document index is used in
     
@@ -43,7 +39,7 @@ class DocumentIndexRead(pydantic.BaseModel):
     - `PRODUCTION` - Production
     """
 
-    indexing_config: typing.Dict[str, typing.Any] = pydantic.Field()
+    indexing_config: typing.Dict[str, typing.Any] = pydantic_v1.Field()
     """
     Configuration representing how documents should be indexed
     """
@@ -59,5 +55,5 @@ class DocumentIndexRead(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
-        extra = pydantic.Extra.allow
+        extra = pydantic_v1.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}

vellum/types/document_read.py

@@ -4,30 +4,26 @@ import datetime as dt
 import typing
 
 from ..core.datetime_utils import serialize_datetime
+from ..core.pydantic_utilities import pydantic_v1
 from .document_document_to_document_index import DocumentDocumentToDocumentIndex
 from .document_status import DocumentStatus
 from .processing_state_enum import ProcessingStateEnum
 
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
 
-
-class DocumentRead(pydantic.BaseModel):
+class DocumentRead(pydantic_v1.BaseModel):
     id: str
-    external_id: typing.Optional[str] = pydantic.Field(default=None)
+    external_id: typing.Optional[str] = pydantic_v1.Field(default=None)
     """
     The unique id of this document as it exists in the user's system.
     """
 
     last_uploaded_at: dt.datetime
-    label: str = pydantic.Field()
+    label: str = pydantic_v1.Field()
     """
     A human-readable label for the document. Defaults to the originally uploaded file's file name.
     """
 
-    processing_state: typing.Optional[ProcessingStateEnum] = pydantic.Field(default=None)
+    processing_state: typing.Optional[ProcessingStateEnum] = pydantic_v1.Field(default=None)
     """
     The current processing state of the document
     
@@ -37,7 +33,7 @@ class DocumentRead(pydantic.BaseModel):
     - `FAILED` - Failed
     """
 
-    status: typing.Optional[DocumentStatus] = pydantic.Field(default=None)
+    status: typing.Optional[DocumentStatus] = pydantic_v1.Field(default=None)
     """
     The current status of the document
     
@@ -47,7 +43,7 @@ class DocumentRead(pydantic.BaseModel):
     original_file_url: typing.Optional[str] = None
     processed_file_url: typing.Optional[str] = None
     document_to_document_indexes: typing.List[DocumentDocumentToDocumentIndex]
-    metadata: typing.Optional[typing.Dict[str, typing.Any]] = pydantic.Field(default=None)
+    metadata: typing.Optional[typing.Dict[str, typing.Any]] = pydantic_v1.Field(default=None)
     """
     A previously supplied JSON object containing metadata that can be filtered on when searching.
     """
@@ -63,5 +59,5 @@ class DocumentRead(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
-        extra = pydantic.Extra.allow
+        extra = pydantic_v1.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}

vellum/types/enriched_normalized_completion.py

@@ -4,33 +4,29 @@ import datetime as dt
 import typing
 
 from ..core.datetime_utils import serialize_datetime
+from ..core.pydantic_utilities import pydantic_v1
 from .finish_reason_enum import FinishReasonEnum
 from .normalized_log_probs import NormalizedLogProbs
 from .vellum_variable_type import VellumVariableType
 
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
 
-
-class EnrichedNormalizedCompletion(pydantic.BaseModel):
-    id: str = pydantic.Field()
+class EnrichedNormalizedCompletion(pydantic_v1.BaseModel):
+    id: str = pydantic_v1.Field()
     """
     The Vellum-generated ID of the completion.
     """
 
-    external_id: typing.Optional[str] = pydantic.Field(default=None)
+    external_id: typing.Optional[str] = pydantic_v1.Field(default=None)
     """
     The external ID that was originally provided along with the generation request, which uniquely identifies this generation in an external system.
     """
 
-    text: str = pydantic.Field()
+    text: str = pydantic_v1.Field()
     """
     The text generated by the LLM.
     """
 
-    finish_reason: typing.Optional[FinishReasonEnum] = pydantic.Field(default=None)
+    finish_reason: typing.Optional[FinishReasonEnum] = pydantic_v1.Field(default=None)
     """
     The reason the generation finished.
     
@@ -39,12 +35,12 @@ class EnrichedNormalizedCompletion(pydantic.BaseModel):
     - `UNKNOWN` - UNKNOWN
     """
 
-    logprobs: typing.Optional[NormalizedLogProbs] = pydantic.Field(default=None)
+    logprobs: typing.Optional[NormalizedLogProbs] = pydantic_v1.Field(default=None)
     """
     The logprobs of the completion. Only present if specified in the original request options.
     """
 
-    model_version_id: str = pydantic.Field()
+    model_version_id: str = pydantic_v1.Field()
     """
     The ID of the model version used to generate this completion.
     """
@@ -65,5 +61,5 @@ class EnrichedNormalizedCompletion(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
-        extra = pydantic.Extra.allow
+        extra = pydantic_v1.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}

vellum/types/entity_status.py

@@ -1,22 +1,5 @@
 # This file was auto-generated by Fern from our API Definition.
 
-import enum
 import typing
 
-T_Result = typing.TypeVar("T_Result")
-
-
-class EntityStatus(str, enum.Enum):
-    """
-    - `ACTIVE` - Active
-    - `ARCHIVED` - Archived
-    """
-
-    ACTIVE = "ACTIVE"
-    ARCHIVED = "ARCHIVED"
-
-    def visit(self, active: typing.Callable[[], T_Result], archived: typing.Callable[[], T_Result]) -> T_Result:
-        if self is EntityStatus.ACTIVE:
-            return active()
-        if self is EntityStatus.ARCHIVED:
-            return archived()
+EntityStatus = typing.Union[typing.Literal["ACTIVE", "ARCHIVED"], typing.Any]

vellum/types/environment_enum.py

@@ -1,31 +1,5 @@
 # This file was auto-generated by Fern from our API Definition.
 
-import enum
 import typing
 
-T_Result = typing.TypeVar("T_Result")
-
-
-class EnvironmentEnum(str, enum.Enum):
-    """
-    - `DEVELOPMENT` - Development
-    - `STAGING` - Staging
-    - `PRODUCTION` - Production
-    """
-
-    DEVELOPMENT = "DEVELOPMENT"
-    STAGING = "STAGING"
-    PRODUCTION = "PRODUCTION"
-
-    def visit(
-        self,
-        development: typing.Callable[[], T_Result],
-        staging: typing.Callable[[], T_Result],
-        production: typing.Callable[[], T_Result],
-    ) -> T_Result:
-        if self is EnvironmentEnum.DEVELOPMENT:
-            return development()
-        if self is EnvironmentEnum.STAGING:
-            return staging()
-        if self is EnvironmentEnum.PRODUCTION:
-            return production()
+EnvironmentEnum = typing.Union[typing.Literal["DEVELOPMENT", "STAGING", "PRODUCTION"], typing.Any]

vellum/types/error_variable_value.py

@@ -4,15 +4,11 @@ import datetime as dt
 import typing
 
 from ..core.datetime_utils import serialize_datetime
+from ..core.pydantic_utilities import pydantic_v1
 from .vellum_error import VellumError
 
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
 
-
-class ErrorVariableValue(pydantic.BaseModel):
+class ErrorVariableValue(pydantic_v1.BaseModel):
     value: typing.Optional[VellumError] = None
 
     def json(self, **kwargs: typing.Any) -> str:
@@ -26,5 +22,5 @@ class ErrorVariableValue(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
-        extra = pydantic.Extra.allow
+        extra = pydantic_v1.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}

vellum/types/execute_prompt_api_error_response.py

@@ -4,15 +4,11 @@ import datetime as dt
 import typing
 
 from ..core.datetime_utils import serialize_datetime
+from ..core.pydantic_utilities import pydantic_v1
 
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
 
-
-class ExecutePromptApiErrorResponse(pydantic.BaseModel):
-    detail: str = pydantic.Field()
+class ExecutePromptApiErrorResponse(pydantic_v1.BaseModel):
+    detail: str = pydantic_v1.Field()
     """
     Details about why the request failed.
     """
@@ -28,5 +24,5 @@ class ExecutePromptApiErrorResponse(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
-        extra = pydantic.Extra.allow
+        extra = pydantic_v1.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}

vellum/types/execute_workflow_error_response.py

@@ -4,15 +4,11 @@ import datetime as dt
 import typing
 
 from ..core.datetime_utils import serialize_datetime
+from ..core.pydantic_utilities import pydantic_v1
 
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
 
-
-class ExecuteWorkflowErrorResponse(pydantic.BaseModel):
-    detail: str = pydantic.Field()
+class ExecuteWorkflowErrorResponse(pydantic_v1.BaseModel):
+    detail: str = pydantic_v1.Field()
     """
     Details about why the request failed.
     """
@@ -28,5 +24,5 @@ class ExecuteWorkflowErrorResponse(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
-        extra = pydantic.Extra.allow
+        extra = pydantic_v1.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}

vellum/types/execute_workflow_response.py

@@ -4,15 +4,11 @@ import datetime as dt
 import typing
 
 from ..core.datetime_utils import serialize_datetime
+from ..core.pydantic_utilities import pydantic_v1
 from .execute_workflow_workflow_result_event import ExecuteWorkflowWorkflowResultEvent
 
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
 
-
-class ExecuteWorkflowResponse(pydantic.BaseModel):
+class ExecuteWorkflowResponse(pydantic_v1.BaseModel):
     execution_id: str
     run_id: typing.Optional[str] = None
     external_id: typing.Optional[str] = None
@@ -29,5 +25,5 @@ class ExecuteWorkflowResponse(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
-        extra = pydantic.Extra.allow
+        extra = pydantic_v1.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}

vellum/types/execute_workflow_stream_error_response.py

@@ -4,15 +4,11 @@ import datetime as dt
 import typing
 
 from ..core.datetime_utils import serialize_datetime
+from ..core.pydantic_utilities import pydantic_v1
 
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
 
-
-class ExecuteWorkflowStreamErrorResponse(pydantic.BaseModel):
-    detail: str = pydantic.Field()
+class ExecuteWorkflowStreamErrorResponse(pydantic_v1.BaseModel):
+    detail: str = pydantic_v1.Field()
     """
     Details about why the request failed.
     """
@@ -28,5 +24,5 @@ class ExecuteWorkflowStreamErrorResponse(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
-        extra = pydantic.Extra.allow
+        extra = pydantic_v1.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}

vellum/types/execution_array_vellum_value.py

@@ -4,16 +4,12 @@ import datetime as dt
 import typing
 
 from ..core.datetime_utils import serialize_datetime
+from ..core.pydantic_utilities import pydantic_v1
 from .array_variable_value_item import ArrayVariableValueItem
 
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
 
-
-class ExecutionArrayVellumValue(pydantic.BaseModel):
-    id: str = pydantic.Field()
+class ExecutionArrayVellumValue(pydantic_v1.BaseModel):
+    id: str = pydantic_v1.Field()
     """
     The variable's uniquely identifying internal id.
     """
@@ -32,5 +28,5 @@ class ExecutionArrayVellumValue(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
-        extra = pydantic.Extra.allow
+        extra = pydantic_v1.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}