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

vellum/types/slim_document.py

@@ -4,39 +4,35 @@ 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_failure_reason_enum import ProcessingFailureReasonEnum
 from .processing_state_enum import ProcessingStateEnum
 
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
 
-
-class SlimDocument(pydantic.BaseModel):
-    id: str = pydantic.Field()
+class SlimDocument(pydantic_v1.BaseModel):
+    id: str = pydantic_v1.Field()
     """
     Vellum-generated ID that uniquely identifies this document.
     """
 
-    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 when uploading the document.
     """
 
-    last_uploaded_at: dt.datetime = pydantic.Field()
+    last_uploaded_at: dt.datetime = pydantic_v1.Field()
     """
     A timestamp representing when this document was most recently uploaded.
     """
 
-    label: str = pydantic.Field()
+    label: str = pydantic_v1.Field()
     """
     Human-friendly name for this document.
     """
 
-    processing_state: typing.Optional[ProcessingStateEnum] = pydantic.Field(default=None)
+    processing_state: typing.Optional[ProcessingStateEnum] = pydantic_v1.Field(default=None)
     """
     An enum value representing where this document is along its processing lifecycle. Note that this is different than its indexing lifecycle.
     
@@ -46,7 +42,7 @@ class SlimDocument(pydantic.BaseModel):
     - `FAILED` - Failed
     """
 
-    processing_failure_reason: typing.Optional[ProcessingFailureReasonEnum] = pydantic.Field(default=None)
+    processing_failure_reason: typing.Optional[ProcessingFailureReasonEnum] = pydantic_v1.Field(default=None)
     """
     An enum value representing why the document could not be processed. Is null unless processing_state is FAILED.
     
@@ -54,19 +50,19 @@ class SlimDocument(pydantic.BaseModel):
     - `INVALID_FILE` - Invalid File
     """
 
-    status: typing.Optional[DocumentStatus] = pydantic.Field(default=None)
+    status: typing.Optional[DocumentStatus] = pydantic_v1.Field(default=None)
     """
     The document's current status.
     
     - `ACTIVE` - Active
     """
 
-    keywords: typing.Optional[typing.List[str]] = pydantic.Field(default=None)
+    keywords: typing.Optional[typing.List[str]] = pydantic_v1.Field(default=None)
     """
     A list of keywords associated with this document. Originally provided when uploading the document.
     """
 
-    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.
     """
@@ -84,5 +80,5 @@ class SlimDocument(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/slim_workflow_deployment.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
 from .vellum_variable import VellumVariable
 
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
 
-
-class SlimWorkflowDeployment(pydantic.BaseModel):
+class SlimWorkflowDeployment(pydantic_v1.BaseModel):
     id: str
-    name: str = pydantic.Field()
+    name: str = pydantic_v1.Field()
     """
     A name that uniquely identifies this workflow deployment within its workspace
     """
 
-    label: str = pydantic.Field()
+    label: str = pydantic_v1.Field()
     """
     A human-readable label for the workflow deployment
     """
 
-    status: typing.Optional[EntityStatus] = pydantic.Field(default=None)
+    status: typing.Optional[EntityStatus] = pydantic_v1.Field(default=None)
     """
     The current status of the workflow deployment
     
@@ -34,7 +30,7 @@ class SlimWorkflowDeployment(pydantic.BaseModel):
     - `ARCHIVED` - Archived
     """
 
-    environment: typing.Optional[EnvironmentEnum] = pydantic.Field(default=None)
+    environment: typing.Optional[EnvironmentEnum] = pydantic_v1.Field(default=None)
     """
     The environment this workflow deployment is used in
     
@@ -45,12 +41,12 @@ class SlimWorkflowDeployment(pydantic.BaseModel):
 
     created: dt.datetime
     last_deployed_on: dt.datetime
-    input_variables: typing.List[VellumVariable] = pydantic.Field()
+    input_variables: typing.List[VellumVariable] = pydantic_v1.Field()
     """
     The input variables this Workflow Deployment expects to receive values for when it is executed.
     """
 
-    output_variables: typing.List[VellumVariable] = pydantic.Field()
+    output_variables: typing.List[VellumVariable] = pydantic_v1.Field()
     """
     The output variables this Workflow Deployment will produce when it is executed.
     """
@@ -66,5 +62,5 @@ class SlimWorkflowDeployment(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/streaming_execute_prompt_event.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 .prompt_output import PromptOutput
 from .streaming_prompt_execution_meta import StreamingPromptExecutionMeta
 
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
 
-
-class StreamingExecutePromptEvent(pydantic.BaseModel):
+class StreamingExecutePromptEvent(pydantic_v1.BaseModel):
     """
     The data returned for each delta during the prompt execution stream.
     """
@@ -22,7 +18,7 @@ class StreamingExecutePromptEvent(pydantic.BaseModel):
     output_index: int
     execution_id: str
     meta: typing.Optional[StreamingPromptExecutionMeta] = None
-    raw: typing.Optional[typing.Dict[str, typing.Any]] = pydantic.Field(default=None)
+    raw: typing.Optional[typing.Dict[str, typing.Any]] = pydantic_v1.Field(default=None)
     """
     The subset of the raw response from the model that the request opted into with `expand_raw`.
     """
@@ -38,5 +34,5 @@ class StreamingExecutePromptEvent(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/streaming_prompt_execution_meta.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 StreamingPromptExecutionMeta(pydantic.BaseModel):
+class StreamingPromptExecutionMeta(pydantic_v1.BaseModel):
     """
     The subset of the metadata tracked by Vellum during prompt execution that the request opted into with `expand_meta`.
     """
@@ -29,5 +25,5 @@ class StreamingPromptExecutionMeta(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/streaming_workflow_node_result_event.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 .node_output_compiled_value import NodeOutputCompiledValue
 from .workflow_node_result_data import WorkflowNodeResultData
 
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
 
-
-class StreamingWorkflowNodeResultEvent(pydantic.BaseModel):
+class StreamingWorkflowNodeResultEvent(pydantic_v1.BaseModel):
     """
     An event that indicates that the node has execution is in progress.
     """
@@ -38,5 +34,5 @@ class StreamingWorkflowNodeResultEvent(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/string_chat_message_content.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 StringChatMessageContent(pydantic.BaseModel):
+class StringChatMessageContent(pydantic_v1.BaseModel):
     """
     A string value that is used in a chat message.
     """
@@ -29,5 +25,5 @@ class StringChatMessageContent(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/string_chat_message_content_request.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 StringChatMessageContentRequest(pydantic.BaseModel):
+class StringChatMessageContentRequest(pydantic_v1.BaseModel):
     """
     A string value that is used in a chat message.
     """
@@ -29,5 +25,5 @@ class StringChatMessageContentRequest(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/string_input_request.py

@@ -4,19 +4,15 @@ 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 StringInputRequest(pydantic.BaseModel):
+class StringInputRequest(pydantic_v1.BaseModel):
     """
     A user input representing a string value
     """
 
-    name: str = pydantic.Field()
+    name: str = pydantic_v1.Field()
     """
     The variable's name, as defined in the deployment.
     """
@@ -34,5 +30,5 @@ class StringInputRequest(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/string_variable_value.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 StringVariableValue(pydantic.BaseModel):
+class StringVariableValue(pydantic_v1.BaseModel):
     value: typing.Optional[str] = None
 
     def json(self, **kwargs: typing.Any) -> str:
@@ -25,5 +21,5 @@ class StringVariableValue(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/submit_completion_actual_request.py

@@ -4,35 +4,31 @@ 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 SubmitCompletionActualRequest(pydantic.BaseModel):
-    id: typing.Optional[str] = pydantic.Field(default=None)
+class SubmitCompletionActualRequest(pydantic_v1.BaseModel):
+    id: typing.Optional[str] = pydantic_v1.Field(default=None)
     """
     The Vellum-generated ID of a previously generated completion. Must provide either this or external_id.
     """
 
-    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 when generating the completion that you'd now like to submit actuals for. Must provide either this or id.
     """
 
-    text: typing.Optional[str] = pydantic.Field(default=None)
+    text: typing.Optional[str] = pydantic_v1.Field(default=None)
     """
     Text representing what the completion _should_ have been.
     """
 
-    quality: typing.Optional[float] = pydantic.Field(default=None)
+    quality: typing.Optional[float] = pydantic_v1.Field(default=None)
     """
     A number between 0 and 1 representing the quality of the completion. 0 is the worst, 1 is the best.
     """
 
-    timestamp: typing.Optional[dt.datetime] = pydantic.Field(default=None)
+    timestamp: typing.Optional[dt.datetime] = pydantic_v1.Field(default=None)
     """
     Optionally provide the timestamp representing when this feedback was collected. Used for reporting purposes.
     """
@@ -48,5 +44,5 @@ class SubmitCompletionActualRequest(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/submit_completion_actuals_error_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 SubmitCompletionActualsErrorResponse(pydantic.BaseModel):
+class SubmitCompletionActualsErrorResponse(pydantic_v1.BaseModel):
     detail: str
 
     def json(self, **kwargs: typing.Any) -> str:
@@ -25,5 +21,5 @@ class SubmitCompletionActualsErrorResponse(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/subworkflow_node_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 SubworkflowNodeResult(pydantic.BaseModel):
+class SubworkflowNodeResult(pydantic_v1.BaseModel):
     """
     A Node Result Event emitted from a Subworkflow Node.
     """
@@ -27,5 +23,5 @@ class SubworkflowNodeResult(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/templating_node_array_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 .array_variable_value_item import ArrayVariableValueItem
 
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
 
-
-class TemplatingNodeArrayResult(pydantic.BaseModel):
+class TemplatingNodeArrayResult(pydantic_v1.BaseModel):
     id: str
     value: typing.Optional[typing.List[ArrayVariableValueItem]] = None
 
@@ -27,5 +23,5 @@ class TemplatingNodeArrayResult(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/templating_node_chat_history_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 .chat_message import ChatMessage
 
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
 
-
-class TemplatingNodeChatHistoryResult(pydantic.BaseModel):
+class TemplatingNodeChatHistoryResult(pydantic_v1.BaseModel):
     id: str
     value: typing.Optional[typing.List[ChatMessage]] = None
 
@@ -27,5 +23,5 @@ class TemplatingNodeChatHistoryResult(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/templating_node_error_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 .vellum_error import VellumError
 
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
 
-
-class TemplatingNodeErrorResult(pydantic.BaseModel):
+class TemplatingNodeErrorResult(pydantic_v1.BaseModel):
     id: str
     value: typing.Optional[VellumError] = None
 
@@ -27,5 +23,5 @@ class TemplatingNodeErrorResult(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/templating_node_function_call_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 .function_call import FunctionCall
 
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
 
-
-class TemplatingNodeFunctionCallResult(pydantic.BaseModel):
+class TemplatingNodeFunctionCallResult(pydantic_v1.BaseModel):
     id: str
     value: typing.Optional[FunctionCall] = None
 
@@ -27,5 +23,5 @@ class TemplatingNodeFunctionCallResult(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/templating_node_json_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 TemplatingNodeJsonResult(pydantic.BaseModel):
+class TemplatingNodeJsonResult(pydantic_v1.BaseModel):
     id: str
     value: typing.Optional[typing.Dict[str, typing.Any]] = None
 
@@ -26,5 +22,5 @@ class TemplatingNodeJsonResult(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/templating_node_number_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 TemplatingNodeNumberResult(pydantic.BaseModel):
+class TemplatingNodeNumberResult(pydantic_v1.BaseModel):
     id: str
     value: typing.Optional[float] = None
 
@@ -26,5 +22,5 @@ class TemplatingNodeNumberResult(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/templating_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 .templating_node_result_data import TemplatingNodeResultData
 
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
 
-
-class TemplatingNodeResult(pydantic.BaseModel):
+class TemplatingNodeResult(pydantic_v1.BaseModel):
     """
     A Node Result Event emitted from a Templating Node.
     """
@@ -30,5 +26,5 @@ class TemplatingNodeResult(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/templating_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 .templating_node_result_output import TemplatingNodeResultOutput
 
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
 
-
-class TemplatingNodeResultData(pydantic.BaseModel):
+class TemplatingNodeResultData(pydantic_v1.BaseModel):
     output: TemplatingNodeResultOutput
 
     def json(self, **kwargs: typing.Any) -> str:
@@ -26,5 +22,5 @@ class TemplatingNodeResultData(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/templating_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 TemplatingNodeSearchResultsResult(pydantic.BaseModel):
+class TemplatingNodeSearchResultsResult(pydantic_v1.BaseModel):
     id: str
     value: typing.Optional[typing.List[SearchResult]] = None
 
@@ -27,5 +23,5 @@ class TemplatingNodeSearchResultsResult(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
-        extra = pydantic.Extra.allow
+        extra = pydantic_v1.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}