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

vellum/types/node_output_compiled_string_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 NodeOutputCompiledStringValue(pydantic.BaseModel):
+class NodeOutputCompiledStringValue(pydantic_v1.BaseModel):
     node_output_id: str
     value: typing.Optional[str] = None
 
@@ -26,5 +22,5 @@ class NodeOutputCompiledStringValue(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/normalized_log_probs.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 .normalized_token_log_probs import NormalizedTokenLogProbs
 
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
 
-
-class NormalizedLogProbs(pydantic.BaseModel):
+class NormalizedLogProbs(pydantic_v1.BaseModel):
     tokens: typing.List[NormalizedTokenLogProbs]
     likelihood: typing.Optional[float] = None
 
@@ -27,5 +23,5 @@ class NormalizedLogProbs(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/normalized_token_log_probs.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 NormalizedTokenLogProbs(pydantic.BaseModel):
+class NormalizedTokenLogProbs(pydantic_v1.BaseModel):
     token: str
     logprob: typing.Optional[float] = None
     top_logprobs: typing.Optional[typing.Dict[str, typing.Optional[float]]] = None
@@ -28,5 +24,5 @@ class NormalizedTokenLogProbs(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/number_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 NumberVariableValue(pydantic.BaseModel):
+class NumberVariableValue(pydantic_v1.BaseModel):
     value: typing.Optional[float] = None
 
     def json(self, **kwargs: typing.Any) -> str:
@@ -25,5 +21,5 @@ class NumberVariableValue(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/paginated_document_index_read_list.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 .document_index_read import DocumentIndexRead
 
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
 
-
-class PaginatedDocumentIndexReadList(pydantic.BaseModel):
+class PaginatedDocumentIndexReadList(pydantic_v1.BaseModel):
     count: typing.Optional[int] = None
     next: typing.Optional[str] = None
     previous: typing.Optional[str] = None
@@ -29,5 +25,5 @@ class PaginatedDocumentIndexReadList(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/paginated_slim_deployment_read_list.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 .slim_deployment_read import SlimDeploymentRead
 
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
 
-
-class PaginatedSlimDeploymentReadList(pydantic.BaseModel):
+class PaginatedSlimDeploymentReadList(pydantic_v1.BaseModel):
     count: typing.Optional[int] = None
     next: typing.Optional[str] = None
     previous: typing.Optional[str] = None
@@ -29,5 +25,5 @@ class PaginatedSlimDeploymentReadList(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/paginated_slim_document_list.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 .slim_document import SlimDocument
 
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
 
-
-class PaginatedSlimDocumentList(pydantic.BaseModel):
+class PaginatedSlimDocumentList(pydantic_v1.BaseModel):
     count: typing.Optional[int] = None
     next: typing.Optional[str] = None
     previous: typing.Optional[str] = None
@@ -29,5 +25,5 @@ class PaginatedSlimDocumentList(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/paginated_slim_workflow_deployment_list.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 .slim_workflow_deployment import SlimWorkflowDeployment
 
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
 
-
-class PaginatedSlimWorkflowDeploymentList(pydantic.BaseModel):
+class PaginatedSlimWorkflowDeploymentList(pydantic_v1.BaseModel):
     count: typing.Optional[int] = None
     next: typing.Optional[str] = None
     previous: typing.Optional[str] = None
@@ -29,5 +25,5 @@ class PaginatedSlimWorkflowDeploymentList(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/paginated_test_suite_run_execution_list.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 .test_suite_run_execution import TestSuiteRunExecution
 
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
 
-
-class PaginatedTestSuiteRunExecutionList(pydantic.BaseModel):
+class PaginatedTestSuiteRunExecutionList(pydantic_v1.BaseModel):
     count: int
     next: typing.Optional[str] = None
     previous: typing.Optional[str] = None
@@ -29,5 +25,5 @@ class PaginatedTestSuiteRunExecutionList(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/paginated_test_suite_test_case_list.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 .test_suite_test_case import TestSuiteTestCase
 
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
 
-
-class PaginatedTestSuiteTestCaseList(pydantic.BaseModel):
+class PaginatedTestSuiteTestCaseList(pydantic_v1.BaseModel):
     count: int
     next: typing.Optional[str] = None
     previous: typing.Optional[str] = None
@@ -29,5 +25,5 @@ class PaginatedTestSuiteTestCaseList(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/processing_failure_reason_enum.py

@@ -1,24 +1,5 @@
 # This file was auto-generated by Fern from our API Definition.
 
-import enum
 import typing
 
-T_Result = typing.TypeVar("T_Result")
-
-
-class ProcessingFailureReasonEnum(str, enum.Enum):
-    """
-    - `EXCEEDED_CHARACTER_LIMIT` - Exceeded Character Limit
-    - `INVALID_FILE` - Invalid File
-    """
-
-    EXCEEDED_CHARACTER_LIMIT = "EXCEEDED_CHARACTER_LIMIT"
-    INVALID_FILE = "INVALID_FILE"
-
-    def visit(
-        self, exceeded_character_limit: typing.Callable[[], T_Result], invalid_file: typing.Callable[[], T_Result]
-    ) -> T_Result:
-        if self is ProcessingFailureReasonEnum.EXCEEDED_CHARACTER_LIMIT:
-            return exceeded_character_limit()
-        if self is ProcessingFailureReasonEnum.INVALID_FILE:
-            return invalid_file()
+ProcessingFailureReasonEnum = typing.Union[typing.Literal["EXCEEDED_CHARACTER_LIMIT", "INVALID_FILE"], typing.Any]

vellum/types/processing_state_enum.py

@@ -1,36 +1,5 @@
 # This file was auto-generated by Fern from our API Definition.
 
-import enum
 import typing
 
-T_Result = typing.TypeVar("T_Result")
-
-
-class ProcessingStateEnum(str, enum.Enum):
-    """
-    - `QUEUED` - Queued
-    - `PROCESSING` - Processing
-    - `PROCESSED` - Processed
-    - `FAILED` - Failed
-    """
-
-    QUEUED = "QUEUED"
-    PROCESSING = "PROCESSING"
-    PROCESSED = "PROCESSED"
-    FAILED = "FAILED"
-
-    def visit(
-        self,
-        queued: typing.Callable[[], T_Result],
-        processing: typing.Callable[[], T_Result],
-        processed: typing.Callable[[], T_Result],
-        failed: typing.Callable[[], T_Result],
-    ) -> T_Result:
-        if self is ProcessingStateEnum.QUEUED:
-            return queued()
-        if self is ProcessingStateEnum.PROCESSING:
-            return processing()
-        if self is ProcessingStateEnum.PROCESSED:
-            return processed()
-        if self is ProcessingStateEnum.FAILED:
-            return failed()
+ProcessingStateEnum = typing.Union[typing.Literal["QUEUED", "PROCESSING", "PROCESSED", "FAILED"], typing.Any]

vellum/types/prompt_deployment_expand_meta_request_request.py

@@ -4,40 +4,36 @@ 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 PromptDeploymentExpandMetaRequestRequest(pydantic.BaseModel):
-    model_name: typing.Optional[bool] = pydantic.Field(default=None)
+class PromptDeploymentExpandMetaRequestRequest(pydantic_v1.BaseModel):
+    model_name: typing.Optional[bool] = pydantic_v1.Field(default=None)
     """
     If enabled, the response will include the model identifier representing the ML Model invoked by the Prompt Deployment.
     """
 
-    latency: typing.Optional[bool] = pydantic.Field(default=None)
+    latency: typing.Optional[bool] = pydantic_v1.Field(default=None)
     """
     If enabled, the response will include the time in nanoseconds it took to execute the Prompt Deployment.
     """
 
-    deployment_release_tag: typing.Optional[bool] = pydantic.Field(default=None)
+    deployment_release_tag: typing.Optional[bool] = pydantic_v1.Field(default=None)
     """
     If enabled, the response will include the release tag of the Prompt Deployment.
     """
 
-    prompt_version_id: typing.Optional[bool] = pydantic.Field(default=None)
+    prompt_version_id: typing.Optional[bool] = pydantic_v1.Field(default=None)
     """
     If enabled, the response will include the ID of the Prompt Version backing the deployment.
     """
 
-    finish_reason: typing.Optional[bool] = pydantic.Field(default=None)
+    finish_reason: typing.Optional[bool] = pydantic_v1.Field(default=None)
     """
     If enabled, the response will include the reason provided by the model for why the execution finished.
     """
 
-    usage: typing.Optional[bool] = pydantic.Field(default=None)
+    usage: typing.Optional[bool] = pydantic_v1.Field(default=None)
     """
     If enabled, the response will include model host usage tracking. This may increase latency for some model hosts.
     """
@@ -53,5 +49,5 @@ class PromptDeploymentExpandMetaRequestRequest(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/prompt_execution_meta.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 .finish_reason_enum import FinishReasonEnum
 from .ml_model_usage import MlModelUsage
 
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
 
-
-class PromptExecutionMeta(pydantic.BaseModel):
+class PromptExecutionMeta(pydantic_v1.BaseModel):
     """
     The subset of the metadata tracked by Vellum during prompt execution that the request opted into with `expand_meta`.
     """
@@ -36,5 +32,5 @@ class PromptExecutionMeta(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/prompt_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 .prompt_node_result_data import PromptNodeResultData
 
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
 
-
-class PromptNodeResult(pydantic.BaseModel):
+class PromptNodeResult(pydantic_v1.BaseModel):
     """
     A Node Result Event emitted from a Prompt Node.
     """
@@ -30,5 +26,5 @@ class PromptNodeResult(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/prompt_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 PromptNodeResultData(pydantic.BaseModel):
+class PromptNodeResultData(pydantic_v1.BaseModel):
     output_id: str
     array_output_id: typing.Optional[str] = None
     text: typing.Optional[str] = None
@@ -28,5 +24,5 @@ class PromptNodeResultData(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/raw_prompt_execution_overrides_request.py

@@ -4,21 +4,17 @@ 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 RawPromptExecutionOverridesRequest(pydantic.BaseModel):
+class RawPromptExecutionOverridesRequest(pydantic_v1.BaseModel):
     body: typing.Optional[typing.Dict[str, typing.Any]] = None
-    headers: typing.Optional[typing.Dict[str, typing.Optional[str]]] = pydantic.Field(default=None)
+    headers: typing.Optional[typing.Dict[str, typing.Optional[str]]] = pydantic_v1.Field(default=None)
     """
     The raw headers to send to the model host.
     """
 
-    url: typing.Optional[str] = pydantic.Field(default=None)
+    url: typing.Optional[str] = pydantic_v1.Field(default=None)
     """
     The raw URL to send to the model host.
     """
@@ -34,5 +30,5 @@ class RawPromptExecutionOverridesRequest(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/rejected_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 .rejected_prompt_execution_meta import RejectedPromptExecutionMeta
 from .vellum_error import VellumError
 
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
 
-
-class RejectedExecutePromptEvent(pydantic.BaseModel):
+class RejectedExecutePromptEvent(pydantic_v1.BaseModel):
     """
     The final data returned indicating an error occurred during the stream.
     """
@@ -33,5 +29,5 @@ class RejectedExecutePromptEvent(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/rejected_execute_prompt_response.py

@@ -4,27 +4,23 @@ import datetime as dt
 import typing
 
 from ..core.datetime_utils import serialize_datetime
+from ..core.pydantic_utilities import pydantic_v1
 from .prompt_execution_meta import PromptExecutionMeta
 from .vellum_error import VellumError
 
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
 
-
-class RejectedExecutePromptResponse(pydantic.BaseModel):
+class RejectedExecutePromptResponse(pydantic_v1.BaseModel):
     """
     The unsuccessful response from the model containing an error of what went wrong.
     """
 
     meta: typing.Optional[PromptExecutionMeta] = 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`.
     """
 
-    execution_id: str = pydantic.Field()
+    execution_id: str = pydantic_v1.Field()
     """
     The ID of the execution.
     """
@@ -42,5 +38,5 @@ class RejectedExecutePromptResponse(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/rejected_execute_workflow_workflow_result_event.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 .workflow_event_error import WorkflowEventError
 
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
 
-
-class RejectedExecuteWorkflowWorkflowResultEvent(pydantic.BaseModel):
+class RejectedExecuteWorkflowWorkflowResultEvent(pydantic_v1.BaseModel):
     """
     The unsuccessful response from the Workflow execution containing an error specifying what went wrong.
     """
@@ -32,5 +28,5 @@ class RejectedExecuteWorkflowWorkflowResultEvent(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/rejected_function_call.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 RejectedFunctionCall(pydantic.BaseModel):
+class RejectedFunctionCall(pydantic_v1.BaseModel):
     """
     Returned if the function call failed to parse for some reason.
     """
@@ -32,5 +28,5 @@ class RejectedFunctionCall(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/rejected_prompt_execution_meta.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 .finish_reason_enum import FinishReasonEnum
 
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
 
-
-class RejectedPromptExecutionMeta(pydantic.BaseModel):
+class RejectedPromptExecutionMeta(pydantic_v1.BaseModel):
     """
     The subset of the metadata tracked by Vellum during prompt execution that the request opted into with `expand_meta`.
     """
@@ -31,5 +27,5 @@ class RejectedPromptExecutionMeta(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/rejected_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 .workflow_event_error import WorkflowEventError
 from .workflow_node_result_data import WorkflowNodeResultData
 
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
 
-
-class RejectedWorkflowNodeResultEvent(pydantic.BaseModel):
+class RejectedWorkflowNodeResultEvent(pydantic_v1.BaseModel):
     """
     An event that indicates that the node has rejected its execution.
     """
@@ -37,5 +33,5 @@ class RejectedWorkflowNodeResultEvent(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
-        extra = pydantic.Extra.allow
+        extra = pydantic_v1.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}