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

vellum/types/vellum_variable_type.py

@@ -1,61 +1,10 @@
 # This file was auto-generated by Fern from our API Definition.
 
-import enum
 import typing
 
-T_Result = typing.TypeVar("T_Result")
-
-
-class VellumVariableType(str, enum.Enum):
-    """
-    - `STRING` - STRING
-    - `NUMBER` - NUMBER
-    - `JSON` - JSON
-    - `CHAT_HISTORY` - CHAT_HISTORY
-    - `SEARCH_RESULTS` - SEARCH_RESULTS
-    - `ERROR` - ERROR
-    - `ARRAY` - ARRAY
-    - `FUNCTION_CALL` - FUNCTION_CALL
-    - `IMAGE` - IMAGE
-    """
-
-    STRING = "STRING"
-    NUMBER = "NUMBER"
-    JSON = "JSON"
-    CHAT_HISTORY = "CHAT_HISTORY"
-    SEARCH_RESULTS = "SEARCH_RESULTS"
-    ERROR = "ERROR"
-    ARRAY = "ARRAY"
-    FUNCTION_CALL = "FUNCTION_CALL"
-    IMAGE = "IMAGE"
-
-    def visit(
-        self,
-        string: typing.Callable[[], T_Result],
-        number: typing.Callable[[], T_Result],
-        json: typing.Callable[[], T_Result],
-        chat_history: typing.Callable[[], T_Result],
-        search_results: typing.Callable[[], T_Result],
-        error: typing.Callable[[], T_Result],
-        array: typing.Callable[[], T_Result],
-        function_call: typing.Callable[[], T_Result],
-        image: typing.Callable[[], T_Result],
-    ) -> T_Result:
-        if self is VellumVariableType.STRING:
-            return string()
-        if self is VellumVariableType.NUMBER:
-            return number()
-        if self is VellumVariableType.JSON:
-            return json()
-        if self is VellumVariableType.CHAT_HISTORY:
-            return chat_history()
-        if self is VellumVariableType.SEARCH_RESULTS:
-            return search_results()
-        if self is VellumVariableType.ERROR:
-            return error()
-        if self is VellumVariableType.ARRAY:
-            return array()
-        if self is VellumVariableType.FUNCTION_CALL:
-            return function_call()
-        if self is VellumVariableType.IMAGE:
-            return image()
+VellumVariableType = typing.Union[
+    typing.Literal[
+        "STRING", "NUMBER", "JSON", "CHAT_HISTORY", "SEARCH_RESULTS", "ERROR", "ARRAY", "FUNCTION_CALL", "IMAGE", "NULL"
+    ],
+    typing.Any,
+]

vellum/types/workflow_deployment_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
 from .vellum_variable import VellumVariable
 
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
 
-
-class WorkflowDeploymentRead(pydantic.BaseModel):
+class WorkflowDeploymentRead(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 WorkflowDeploymentRead(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 WorkflowDeploymentRead(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 produces values for when it's executed.
     """
@@ -66,5 +62,5 @@ class WorkflowDeploymentRead(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/workflow_event_error.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_execution_event_error_code import WorkflowExecutionEventErrorCode
 
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
 
-
-class WorkflowEventError(pydantic.BaseModel):
+class WorkflowEventError(pydantic_v1.BaseModel):
     message: str
     code: WorkflowExecutionEventErrorCode
 
@@ -27,5 +23,5 @@ class WorkflowEventError(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/workflow_execution_actual_chat_history_request.py

@@ -4,36 +4,32 @@ import datetime as dt
 import typing
 
 from ..core.datetime_utils import serialize_datetime
+from ..core.pydantic_utilities import pydantic_v1
 from .chat_message_request import ChatMessageRequest
 
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
 
-
-class WorkflowExecutionActualChatHistoryRequest(pydantic.BaseModel):
-    output_id: typing.Optional[str] = pydantic.Field(default=None)
+class WorkflowExecutionActualChatHistoryRequest(pydantic_v1.BaseModel):
+    output_id: typing.Optional[str] = pydantic_v1.Field(default=None)
     """
     The Vellum-generated ID of a workflow output. Must provide either this or output_key. output_key is typically preferred.
     """
 
-    output_key: typing.Optional[str] = pydantic.Field(default=None)
+    output_key: typing.Optional[str] = pydantic_v1.Field(default=None)
     """
     The user-defined name of a workflow output. Must provide either this or output_id. Should correspond to the `Name` specified in a Final Output Node. Generally preferred over output_id.
     """
 
-    quality: typing.Optional[float] = pydantic.Field(default=None)
+    quality: typing.Optional[float] = pydantic_v1.Field(default=None)
     """
     Optionally provide a decimal number between 0.0 and 1.0 (inclusive) representing the quality of the output. 0 is the worst, 1 is the best.
     """
 
-    timestamp: typing.Optional[float] = pydantic.Field(default=None)
+    timestamp: typing.Optional[float] = pydantic_v1.Field(default=None)
     """
     Optionally provide the timestamp representing when this feedback was collected. Used for reporting purposes.
     """
 
-    desired_output_value: typing.Optional[typing.List[ChatMessageRequest]] = pydantic.Field(default=None)
+    desired_output_value: typing.Optional[typing.List[ChatMessageRequest]] = pydantic_v1.Field(default=None)
     """
     Optionally provide the value that the output ideally should have been.
     """
@@ -49,5 +45,5 @@ class WorkflowExecutionActualChatHistoryRequest(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/workflow_execution_actual_json_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 WorkflowExecutionActualJsonRequest(pydantic.BaseModel):
-    output_id: typing.Optional[str] = pydantic.Field(default=None)
+class WorkflowExecutionActualJsonRequest(pydantic_v1.BaseModel):
+    output_id: typing.Optional[str] = pydantic_v1.Field(default=None)
     """
     The Vellum-generated ID of a workflow output. Must provide either this or output_key. output_key is typically preferred.
     """
 
-    output_key: typing.Optional[str] = pydantic.Field(default=None)
+    output_key: typing.Optional[str] = pydantic_v1.Field(default=None)
     """
     The user-defined name of a workflow output. Must provide either this or output_id. Should correspond to the `Name` specified in a Final Output Node. Generally preferred over output_id.
     """
 
-    quality: typing.Optional[float] = pydantic.Field(default=None)
+    quality: typing.Optional[float] = pydantic_v1.Field(default=None)
     """
     Optionally provide a decimal number between 0.0 and 1.0 (inclusive) representing the quality of the output. 0 is the worst, 1 is the best.
     """
 
-    timestamp: typing.Optional[float] = pydantic.Field(default=None)
+    timestamp: typing.Optional[float] = pydantic_v1.Field(default=None)
     """
     Optionally provide the timestamp representing when this feedback was collected. Used for reporting purposes.
     """
 
-    desired_output_value: typing.Optional[typing.Dict[str, typing.Any]] = pydantic.Field(default=None)
+    desired_output_value: typing.Optional[typing.Dict[str, typing.Any]] = pydantic_v1.Field(default=None)
     """
     Optionally provide the value that the output ideally should have been.
     """
@@ -48,5 +44,5 @@ class WorkflowExecutionActualJsonRequest(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/workflow_execution_actual_string_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 WorkflowExecutionActualStringRequest(pydantic.BaseModel):
-    output_id: typing.Optional[str] = pydantic.Field(default=None)
+class WorkflowExecutionActualStringRequest(pydantic_v1.BaseModel):
+    output_id: typing.Optional[str] = pydantic_v1.Field(default=None)
     """
     The Vellum-generated ID of a workflow output. Must provide either this or output_key. output_key is typically preferred.
     """
 
-    output_key: typing.Optional[str] = pydantic.Field(default=None)
+    output_key: typing.Optional[str] = pydantic_v1.Field(default=None)
     """
     The user-defined name of a workflow output. Must provide either this or output_id. Should correspond to the `Name` specified in a Final Output Node. Generally preferred over output_id.
     """
 
-    quality: typing.Optional[float] = pydantic.Field(default=None)
+    quality: typing.Optional[float] = pydantic_v1.Field(default=None)
     """
     Optionally provide a decimal number between 0.0 and 1.0 (inclusive) representing the quality of the output. 0 is the worst, 1 is the best.
     """
 
-    timestamp: typing.Optional[float] = pydantic.Field(default=None)
+    timestamp: typing.Optional[float] = pydantic_v1.Field(default=None)
     """
     Optionally provide the timestamp representing when this feedback was collected. Used for reporting purposes.
     """
 
-    desired_output_value: typing.Optional[str] = pydantic.Field(default=None)
+    desired_output_value: typing.Optional[str] = pydantic_v1.Field(default=None)
     """
     Optionally provide the value that the output ideally should have been.
     """
@@ -48,5 +44,5 @@ class WorkflowExecutionActualStringRequest(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/workflow_execution_event_error_code.py

@@ -1,56 +1,17 @@
 # This file was auto-generated by Fern from our API Definition.
 
-import enum
 import typing
 
-T_Result = typing.TypeVar("T_Result")
-
-
-class WorkflowExecutionEventErrorCode(str, enum.Enum):
-    """
-    - `WORKFLOW_INITIALIZATION` - WORKFLOW_INITIALIZATION
-    - `WORKFLOW_CANCELLED` - WORKFLOW_CANCELLED
-    - `NODE_EXECUTION_COUNT_LIMIT_REACHED` - NODE_EXECUTION_COUNT_LIMIT_REACHED
-    - `INTERNAL_SERVER_ERROR` - INTERNAL_SERVER_ERROR
-    - `NODE_EXECUTION` - NODE_EXECUTION
-    - `LLM_PROVIDER` - LLM_PROVIDER
-    - `INVALID_TEMPLATE` - INVALID_TEMPLATE
-    - `USER_DEFINED_ERROR` - USER_DEFINED_ERROR
-    """
-
-    WORKFLOW_INITIALIZATION = "WORKFLOW_INITIALIZATION"
-    WORKFLOW_CANCELLED = "WORKFLOW_CANCELLED"
-    NODE_EXECUTION_COUNT_LIMIT_REACHED = "NODE_EXECUTION_COUNT_LIMIT_REACHED"
-    INTERNAL_SERVER_ERROR = "INTERNAL_SERVER_ERROR"
-    NODE_EXECUTION = "NODE_EXECUTION"
-    LLM_PROVIDER = "LLM_PROVIDER"
-    INVALID_TEMPLATE = "INVALID_TEMPLATE"
-    USER_DEFINED_ERROR = "USER_DEFINED_ERROR"
-
-    def visit(
-        self,
-        workflow_initialization: typing.Callable[[], T_Result],
-        workflow_cancelled: typing.Callable[[], T_Result],
-        node_execution_count_limit_reached: typing.Callable[[], T_Result],
-        internal_server_error: typing.Callable[[], T_Result],
-        node_execution: typing.Callable[[], T_Result],
-        llm_provider: typing.Callable[[], T_Result],
-        invalid_template: typing.Callable[[], T_Result],
-        user_defined_error: typing.Callable[[], T_Result],
-    ) -> T_Result:
-        if self is WorkflowExecutionEventErrorCode.WORKFLOW_INITIALIZATION:
-            return workflow_initialization()
-        if self is WorkflowExecutionEventErrorCode.WORKFLOW_CANCELLED:
-            return workflow_cancelled()
-        if self is WorkflowExecutionEventErrorCode.NODE_EXECUTION_COUNT_LIMIT_REACHED:
-            return node_execution_count_limit_reached()
-        if self is WorkflowExecutionEventErrorCode.INTERNAL_SERVER_ERROR:
-            return internal_server_error()
-        if self is WorkflowExecutionEventErrorCode.NODE_EXECUTION:
-            return node_execution()
-        if self is WorkflowExecutionEventErrorCode.LLM_PROVIDER:
-            return llm_provider()
-        if self is WorkflowExecutionEventErrorCode.INVALID_TEMPLATE:
-            return invalid_template()
-        if self is WorkflowExecutionEventErrorCode.USER_DEFINED_ERROR:
-            return user_defined_error()
+WorkflowExecutionEventErrorCode = typing.Union[
+    typing.Literal[
+        "WORKFLOW_INITIALIZATION",
+        "WORKFLOW_CANCELLED",
+        "NODE_EXECUTION_COUNT_LIMIT_REACHED",
+        "INTERNAL_SERVER_ERROR",
+        "NODE_EXECUTION",
+        "LLM_PROVIDER",
+        "INVALID_TEMPLATE",
+        "USER_DEFINED_ERROR",
+    ],
+    typing.Any,
+]

vellum/types/workflow_execution_event_type.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 WorkflowExecutionEventType(str, enum.Enum):
-    """
-    - `NODE` - NODE
-    - `WORKFLOW` - WORKFLOW
-    """
-
-    NODE = "NODE"
-    WORKFLOW = "WORKFLOW"
-
-    def visit(self, node: typing.Callable[[], T_Result], workflow: typing.Callable[[], T_Result]) -> T_Result:
-        if self is WorkflowExecutionEventType.NODE:
-            return node()
-        if self is WorkflowExecutionEventType.WORKFLOW:
-            return workflow()
+WorkflowExecutionEventType = typing.Union[typing.Literal["NODE", "WORKFLOW"], typing.Any]

vellum/types/workflow_execution_node_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_node_result_event import WorkflowNodeResultEvent
 
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
 
-
-class WorkflowExecutionNodeResultEvent(pydantic.BaseModel):
+class WorkflowExecutionNodeResultEvent(pydantic_v1.BaseModel):
     """
     A NODE-level event emitted from the workflow's execution.
     """
@@ -33,5 +29,5 @@ class WorkflowExecutionNodeResultEvent(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/workflow_execution_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_result_event import WorkflowResultEvent
 
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
 
-
-class WorkflowExecutionWorkflowResultEvent(pydantic.BaseModel):
+class WorkflowExecutionWorkflowResultEvent(pydantic_v1.BaseModel):
     """
     A WORKFLOW-level event emitted from the workflow's execution.
     """
@@ -33,5 +29,5 @@ class WorkflowExecutionWorkflowResultEvent(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/workflow_node_result_event_state.py

@@ -1,36 +1,7 @@
 # This file was auto-generated by Fern from our API Definition.
 
-import enum
 import typing
 
-T_Result = typing.TypeVar("T_Result")
-
-
-class WorkflowNodeResultEventState(str, enum.Enum):
-    """
-    - `INITIATED` - INITIATED
-    - `STREAMING` - STREAMING
-    - `FULFILLED` - FULFILLED
-    - `REJECTED` - REJECTED
-    """
-
-    INITIATED = "INITIATED"
-    STREAMING = "STREAMING"
-    FULFILLED = "FULFILLED"
-    REJECTED = "REJECTED"
-
-    def visit(
-        self,
-        initiated: typing.Callable[[], T_Result],
-        streaming: typing.Callable[[], T_Result],
-        fulfilled: typing.Callable[[], T_Result],
-        rejected: typing.Callable[[], T_Result],
-    ) -> T_Result:
-        if self is WorkflowNodeResultEventState.INITIATED:
-            return initiated()
-        if self is WorkflowNodeResultEventState.STREAMING:
-            return streaming()
-        if self is WorkflowNodeResultEventState.FULFILLED:
-            return fulfilled()
-        if self is WorkflowNodeResultEventState.REJECTED:
-            return rejected()
+WorkflowNodeResultEventState = typing.Union[
+    typing.Literal["INITIATED", "STREAMING", "FULFILLED", "REJECTED"], typing.Any
+]

vellum/types/workflow_output_array.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
 from .array_variable_value_item import ArrayVariableValueItem
 
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
 
-
-class WorkflowOutputArray(pydantic.BaseModel):
+class WorkflowOutputArray(pydantic_v1.BaseModel):
     """
     An array output from a Workflow execution.
     """
 
     id: str
-    name: str = pydantic.Field()
+    name: str = pydantic_v1.Field()
     """
     The output's name, as defined in the workflow
     """
@@ -36,5 +32,5 @@ class WorkflowOutputArray(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/workflow_output_chat_history.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
 from .chat_message import ChatMessage
 
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
 
-
-class WorkflowOutputChatHistory(pydantic.BaseModel):
+class WorkflowOutputChatHistory(pydantic_v1.BaseModel):
     """
     A chat history output from a Workflow execution.
     """
 
     id: str
-    name: str = pydantic.Field()
+    name: str = pydantic_v1.Field()
     """
     The output's name, as defined in the workflow
     """
@@ -36,5 +32,5 @@ class WorkflowOutputChatHistory(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/workflow_output_error.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
 from .vellum_error import VellumError
 
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
 
-
-class WorkflowOutputError(pydantic.BaseModel):
+class WorkflowOutputError(pydantic_v1.BaseModel):
     """
     An error output from a Workflow execution.
     """
 
     id: str
-    name: str = pydantic.Field()
+    name: str = pydantic_v1.Field()
     """
     The output's name, as defined in the workflow
     """
@@ -36,5 +32,5 @@ class WorkflowOutputError(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/workflow_output_function_call.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
 from .function_call import FunctionCall
 
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
 
-
-class WorkflowOutputFunctionCall(pydantic.BaseModel):
+class WorkflowOutputFunctionCall(pydantic_v1.BaseModel):
     """
     A function call output from a Workflow execution.
     """
 
     id: str
-    name: str = pydantic.Field()
+    name: str = pydantic_v1.Field()
     """
     The output's name, as defined in the workflow
     """
@@ -36,5 +32,5 @@ class WorkflowOutputFunctionCall(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/workflow_output_image.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
 from .vellum_image import VellumImage
 
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
 
-
-class WorkflowOutputImage(pydantic.BaseModel):
+class WorkflowOutputImage(pydantic_v1.BaseModel):
     """
     An image output from a Workflow execution.
     """
 
     id: str
-    name: str = pydantic.Field()
+    name: str = pydantic_v1.Field()
     """
     The output's name, as defined in the workflow
     """
@@ -36,5 +32,5 @@ class WorkflowOutputImage(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
-        extra = pydantic.Extra.allow
+        extra = pydantic_v1.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}