vellum-ai 0.0.42__py3-none-any.whl → 0.0.44__py3-none-any.whl

vellum/__init__.py

@@ -38,7 +38,6 @@ from .types import (
     MetadataFilterConfigRequest,
     MetadataFilterRuleCombinator,
     MetadataFilterRuleRequest,
-    ModelTypeDeprecated,
     ModelVersionBuildConfig,
     ModelVersionCompilePromptResponse,
     ModelVersionCompiledPrompt,
@@ -48,12 +47,14 @@ from .types import (
     ModelVersionReadStatusEnum,
     ModelVersionSandboxSnapshot,
     NodeInputCompiledChatHistoryValue,
+    NodeInputCompiledErrorValue,
     NodeInputCompiledJsonValue,
     NodeInputCompiledNumberValue,
     NodeInputCompiledSearchResultsValue,
     NodeInputCompiledStringValue,
     NodeInputVariableCompiledValue,
     NodeInputVariableCompiledValue_ChatHistory,
+    NodeInputVariableCompiledValue_Error,
     NodeInputVariableCompiledValue_Json,
     NodeInputVariableCompiledValue_Number,
     NodeInputVariableCompiledValue_SearchResults,
@@ -106,12 +107,14 @@ from .types import (
     SubmitWorkflowExecutionActualRequest_Json,
     SubmitWorkflowExecutionActualRequest_String,
     TemplatingNodeChatHistoryResult,
+    TemplatingNodeErrorResult,
     TemplatingNodeJsonResult,
     TemplatingNodeNumberResult,
     TemplatingNodeResult,
     TemplatingNodeResultData,
     TemplatingNodeResultOutput,
     TemplatingNodeResultOutput_ChatHistory,
+    TemplatingNodeResultOutput_Error,
     TemplatingNodeResultOutput_Json,
     TemplatingNodeResultOutput_Number,
     TemplatingNodeResultOutput_SearchResults,
@@ -119,12 +122,14 @@ from .types import (
     TemplatingNodeSearchResultsResult,
     TemplatingNodeStringResult,
     TerminalNodeChatHistoryResult,
+    TerminalNodeErrorResult,
     TerminalNodeJsonResult,
     TerminalNodeNumberResult,
     TerminalNodeResult,
     TerminalNodeResultData,
     TerminalNodeResultOutput,
     TerminalNodeResultOutput_ChatHistory,
+    TerminalNodeResultOutput_Error,
     TerminalNodeResultOutput_Json,
     TerminalNodeResultOutput_Number,
     TerminalNodeResultOutput_SearchResults,
@@ -134,6 +139,8 @@ from .types import (
     TestSuiteTestCase,
     UploadDocumentErrorResponse,
     UploadDocumentResponse,
+    VellumError,
+    VellumErrorCodeEnum,
     VellumVariable,
     VellumVariableType,
     WorkflowEventError,
@@ -163,11 +170,13 @@ from .types import (
     WorkflowResultEvent,
     WorkflowResultEventOutputData,
     WorkflowResultEventOutputDataChatHistory,
+    WorkflowResultEventOutputDataError,
     WorkflowResultEventOutputDataJson,
     WorkflowResultEventOutputDataNumber,
     WorkflowResultEventOutputDataSearchResults,
     WorkflowResultEventOutputDataString,
     WorkflowResultEventOutputData_ChatHistory,
+    WorkflowResultEventOutputData_Error,
     WorkflowResultEventOutputData_Json,
     WorkflowResultEventOutputData_Number,
     WorkflowResultEventOutputData_SearchResults,
@@ -230,7 +239,6 @@ __all__ = [
     "MetadataFilterConfigRequest",
     "MetadataFilterRuleCombinator",
     "MetadataFilterRuleRequest",
-    "ModelTypeDeprecated",
     "ModelVersionBuildConfig",
     "ModelVersionCompilePromptResponse",
     "ModelVersionCompiledPrompt",
@@ -240,12 +248,14 @@ __all__ = [
     "ModelVersionReadStatusEnum",
     "ModelVersionSandboxSnapshot",
     "NodeInputCompiledChatHistoryValue",
+    "NodeInputCompiledErrorValue",
     "NodeInputCompiledJsonValue",
     "NodeInputCompiledNumberValue",
     "NodeInputCompiledSearchResultsValue",
     "NodeInputCompiledStringValue",
     "NodeInputVariableCompiledValue",
     "NodeInputVariableCompiledValue_ChatHistory",
+    "NodeInputVariableCompiledValue_Error",
     "NodeInputVariableCompiledValue_Json",
     "NodeInputVariableCompiledValue_Number",
     "NodeInputVariableCompiledValue_SearchResults",
@@ -299,12 +309,14 @@ __all__ = [
     "SubmitWorkflowExecutionActualRequest_Json",
     "SubmitWorkflowExecutionActualRequest_String",
     "TemplatingNodeChatHistoryResult",
+    "TemplatingNodeErrorResult",
     "TemplatingNodeJsonResult",
     "TemplatingNodeNumberResult",
     "TemplatingNodeResult",
     "TemplatingNodeResultData",
     "TemplatingNodeResultOutput",
     "TemplatingNodeResultOutput_ChatHistory",
+    "TemplatingNodeResultOutput_Error",
     "TemplatingNodeResultOutput_Json",
     "TemplatingNodeResultOutput_Number",
     "TemplatingNodeResultOutput_SearchResults",
@@ -312,12 +324,14 @@ __all__ = [
     "TemplatingNodeSearchResultsResult",
     "TemplatingNodeStringResult",
     "TerminalNodeChatHistoryResult",
+    "TerminalNodeErrorResult",
     "TerminalNodeJsonResult",
     "TerminalNodeNumberResult",
     "TerminalNodeResult",
     "TerminalNodeResultData",
     "TerminalNodeResultOutput",
     "TerminalNodeResultOutput_ChatHistory",
+    "TerminalNodeResultOutput_Error",
     "TerminalNodeResultOutput_Json",
     "TerminalNodeResultOutput_Number",
     "TerminalNodeResultOutput_SearchResults",
@@ -328,6 +342,8 @@ __all__ = [
     "UploadDocumentErrorResponse",
     "UploadDocumentResponse",
     "VellumEnvironment",
+    "VellumError",
+    "VellumErrorCodeEnum",
     "VellumVariable",
     "VellumVariableType",
     "WorkflowEventError",
@@ -357,11 +373,13 @@ __all__ = [
     "WorkflowResultEvent",
     "WorkflowResultEventOutputData",
     "WorkflowResultEventOutputDataChatHistory",
+    "WorkflowResultEventOutputDataError",
     "WorkflowResultEventOutputDataJson",
     "WorkflowResultEventOutputDataNumber",
     "WorkflowResultEventOutputDataSearchResults",
     "WorkflowResultEventOutputDataString",
     "WorkflowResultEventOutputData_ChatHistory",
+    "WorkflowResultEventOutputData_Error",
     "WorkflowResultEventOutputData_Json",
     "WorkflowResultEventOutputData_Number",
     "WorkflowResultEventOutputData_SearchResults",

vellum/client.py

@@ -6,7 +6,6 @@ import urllib.parse
 from json.decoder import JSONDecodeError
 
 import httpx
-import pydantic
 
 from .core.api_error import ApiError
 from .core.client_wrapper import AsyncClientWrapper, SyncClientWrapper
@@ -36,6 +35,11 @@ from .types.workflow_execution_event_type import WorkflowExecutionEventType
 from .types.workflow_request_input_request import WorkflowRequestInputRequest
 from .types.workflow_stream_event import WorkflowStreamEvent
 
+try:
+    import pydantic.v1 as pydantic  # type: ignore
+except ImportError:
+    import pydantic  # type: ignore
+
 # this is used as the default value for optional parameters
 OMIT = typing.cast(typing.Any, ...)
 

vellum/core/client_wrapper.py

@@ -16,7 +16,7 @@ class BaseClientWrapper:
         headers: typing.Dict[str, str] = {
             "X-Fern-Language": "Python",
             "X-Fern-SDK-Name": "vellum-ai",
-            "X-Fern-SDK-Version": "v0.0.42",
+            "X-Fern-SDK-Version": "v0.0.44",
         }
         headers["X_API_KEY"] = self.api_key
         return headers

vellum/core/jsonable_encoder.py

@@ -16,8 +16,10 @@ from pathlib import PurePath
 from types import GeneratorType
 from typing import Any, Callable, Dict, List, Optional, Set, Tuple, Union
 
-from pydantic import BaseModel
-from pydantic.json import ENCODERS_BY_TYPE
+try:
+    import pydantic.v1 as pydantic  # type: ignore
+except ImportError:
+    import pydantic  # type: ignore
 
 from .datetime_utils import serialize_datetime
 
@@ -34,7 +36,7 @@ def generate_encoders_by_class_tuples(
     return encoders_by_class_tuples
 
 
-encoders_by_class_tuples = generate_encoders_by_class_tuples(ENCODERS_BY_TYPE)
+encoders_by_class_tuples = generate_encoders_by_class_tuples(pydantic.json.ENCODERS_BY_TYPE)
 
 
 def jsonable_encoder(obj: Any, custom_encoder: Optional[Dict[Any, Callable[[Any], Any]]] = None) -> Any:
@@ -46,7 +48,7 @@ def jsonable_encoder(obj: Any, custom_encoder: Optional[Dict[Any, Callable[[Any]
             for encoder_type, encoder_instance in custom_encoder.items():
                 if isinstance(obj, encoder_type):
                     return encoder_instance(obj)
-    if isinstance(obj, BaseModel):
+    if isinstance(obj, pydantic.BaseModel):
         encoder = getattr(obj.__config__, "json_encoders", {})
         if custom_encoder:
             encoder.update(custom_encoder)
@@ -82,8 +84,8 @@ def jsonable_encoder(obj: Any, custom_encoder: Optional[Dict[Any, Callable[[Any]
             encoded_list.append(jsonable_encoder(item, custom_encoder=custom_encoder))
         return encoded_list
 
-    if type(obj) in ENCODERS_BY_TYPE:
-        return ENCODERS_BY_TYPE[type(obj)](obj)
+    if type(obj) in pydantic.json.ENCODERS_BY_TYPE:
+        return pydantic.json.ENCODERS_BY_TYPE[type(obj)](obj)
     for encoder, classes_tuple in encoders_by_class_tuples.items():
         if isinstance(obj, classes_tuple):
             return encoder(obj)

vellum/resources/deployments/client.py

@@ -3,12 +3,15 @@
 import urllib.parse
 from json.decoder import JSONDecodeError
 
-import pydantic
-
 from ...core.api_error import ApiError
 from ...core.client_wrapper import AsyncClientWrapper, SyncClientWrapper
 from ...types.deployment_read import DeploymentRead
 
+try:
+    import pydantic.v1 as pydantic  # type: ignore
+except ImportError:
+    import pydantic  # type: ignore
+
 
 class DeploymentsClient:
     def __init__(self, *, client_wrapper: SyncClientWrapper):

vellum/resources/document_indexes/client.py

@@ -4,8 +4,6 @@ import typing
 import urllib.parse
 from json.decoder import JSONDecodeError
 
-import pydantic
-
 from ...core.api_error import ApiError
 from ...core.client_wrapper import AsyncClientWrapper, SyncClientWrapper
 from ...core.jsonable_encoder import jsonable_encoder
@@ -13,6 +11,11 @@ from ...types.document_index_read import DocumentIndexRead
 from ...types.document_index_status import DocumentIndexStatus
 from ...types.environment_enum import EnvironmentEnum
 
+try:
+    import pydantic.v1 as pydantic  # type: ignore
+except ImportError:
+    import pydantic  # type: ignore
+
 # this is used as the default value for optional parameters
 OMIT = typing.cast(typing.Any, ...)
 

vellum/resources/documents/client.py

@@ -4,8 +4,6 @@ import typing
 import urllib.parse
 from json.decoder import JSONDecodeError
 
-import pydantic
-
 from ...core.api_error import ApiError
 from ...core.client_wrapper import AsyncClientWrapper, SyncClientWrapper
 from ...core.jsonable_encoder import jsonable_encoder
@@ -18,6 +16,11 @@ from ...types.document_status import DocumentStatus
 from ...types.paginated_slim_document_list import PaginatedSlimDocumentList
 from ...types.upload_document_response import UploadDocumentResponse
 
+try:
+    import pydantic.v1 as pydantic  # type: ignore
+except ImportError:
+    import pydantic  # type: ignore
+
 # this is used as the default value for optional parameters
 OMIT = typing.cast(typing.Any, ...)
 

vellum/resources/model_versions/client.py

@@ -4,14 +4,17 @@ import typing
 import urllib.parse
 from json.decoder import JSONDecodeError
 
-import pydantic
-
 from ...core.api_error import ApiError
 from ...core.client_wrapper import AsyncClientWrapper, SyncClientWrapper
 from ...core.jsonable_encoder import jsonable_encoder
 from ...types.model_version_compile_prompt_response import ModelVersionCompilePromptResponse
 from ...types.model_version_read import ModelVersionRead
 
+try:
+    import pydantic.v1 as pydantic  # type: ignore
+except ImportError:
+    import pydantic  # type: ignore
+
 # this is used as the default value for optional parameters
 OMIT = typing.cast(typing.Any, ...)
 

vellum/resources/registered_prompts/client.py

@@ -4,18 +4,23 @@ import typing
 import urllib.parse
 from json.decoder import JSONDecodeError
 
-import pydantic
-
 from ...core.api_error import ApiError
 from ...core.client_wrapper import AsyncClientWrapper, SyncClientWrapper
 from ...core.jsonable_encoder import jsonable_encoder
+from ...errors.bad_request_error import BadRequestError
 from ...errors.conflict_error import ConflictError
+from ...errors.not_found_error import NotFoundError
 from ...types.provider_enum import ProviderEnum
 from ...types.register_prompt_error_response import RegisterPromptErrorResponse
 from ...types.register_prompt_model_parameters_request import RegisterPromptModelParametersRequest
 from ...types.register_prompt_prompt_info_request import RegisterPromptPromptInfoRequest
 from ...types.register_prompt_response import RegisterPromptResponse
 
+try:
+    import pydantic.v1 as pydantic  # type: ignore
+except ImportError:
+    import pydantic  # type: ignore
+
 # this is used as the default value for optional parameters
 OMIT = typing.cast(typing.Any, ...)
 
@@ -55,8 +60,9 @@ class RegisteredPromptsClient:
                                       * `GOOGLE` - Google
                                       * `HOSTED` - Hosted
                                       * `MOSAICML` - MosaicML
-                                      * `MYSTIC` - Mystic
                                       * `OPENAI` - OpenAI
+                                      * `HUGGINGFACE` - HuggingFace
+                                      * `MYSTIC` - Mystic
                                       * `PYQ` - Pyq
             - model: str. The initial model to use for this prompt
 
@@ -85,6 +91,10 @@ class RegisteredPromptsClient:
         )
         if 200 <= _response.status_code < 300:
             return pydantic.parse_obj_as(RegisterPromptResponse, _response.json())  # type: ignore
+        if _response.status_code == 400:
+            raise BadRequestError(pydantic.parse_obj_as(typing.Any, _response.json()))  # type: ignore
+        if _response.status_code == 404:
+            raise NotFoundError(pydantic.parse_obj_as(typing.Any, _response.json()))  # type: ignore
         if _response.status_code == 409:
             raise ConflictError(pydantic.parse_obj_as(RegisterPromptErrorResponse, _response.json()))  # type: ignore
         try:
@@ -129,8 +139,9 @@ class AsyncRegisteredPromptsClient:
                                       * `GOOGLE` - Google
                                       * `HOSTED` - Hosted
                                       * `MOSAICML` - MosaicML
-                                      * `MYSTIC` - Mystic
                                       * `OPENAI` - OpenAI
+                                      * `HUGGINGFACE` - HuggingFace
+                                      * `MYSTIC` - Mystic
                                       * `PYQ` - Pyq
             - model: str. The initial model to use for this prompt
 
@@ -159,6 +170,10 @@ class AsyncRegisteredPromptsClient:
         )
         if 200 <= _response.status_code < 300:
             return pydantic.parse_obj_as(RegisterPromptResponse, _response.json())  # type: ignore
+        if _response.status_code == 400:
+            raise BadRequestError(pydantic.parse_obj_as(typing.Any, _response.json()))  # type: ignore
+        if _response.status_code == 404:
+            raise NotFoundError(pydantic.parse_obj_as(typing.Any, _response.json()))  # type: ignore
         if _response.status_code == 409:
             raise ConflictError(pydantic.parse_obj_as(RegisterPromptErrorResponse, _response.json()))  # type: ignore
         try:

vellum/resources/sandboxes/client.py

@@ -4,8 +4,6 @@ import typing
 import urllib.parse
 from json.decoder import JSONDecodeError
 
-import pydantic
-
 from ...core.api_error import ApiError
 from ...core.client_wrapper import AsyncClientWrapper, SyncClientWrapper
 from ...core.jsonable_encoder import jsonable_encoder
@@ -13,6 +11,11 @@ from ...types.sandbox_metric_input_params_request import SandboxMetricInputParam
 from ...types.sandbox_scenario import SandboxScenario
 from ...types.scenario_input_request import ScenarioInputRequest
 
+try:
+    import pydantic.v1 as pydantic  # type: ignore
+except ImportError:
+    import pydantic  # type: ignore
+
 # this is used as the default value for optional parameters
 OMIT = typing.cast(typing.Any, ...)
 

vellum/resources/test_suites/client.py

@@ -4,14 +4,17 @@ import typing
 import urllib.parse
 from json.decoder import JSONDecodeError
 
-import pydantic
-
 from ...core.api_error import ApiError
 from ...core.client_wrapper import AsyncClientWrapper, SyncClientWrapper
 from ...core.jsonable_encoder import jsonable_encoder
 from ...types.evaluation_params_request import EvaluationParamsRequest
 from ...types.test_suite_test_case import TestSuiteTestCase
 
+try:
+    import pydantic.v1 as pydantic  # type: ignore
+except ImportError:
+    import pydantic  # type: ignore
+
 # this is used as the default value for optional parameters
 OMIT = typing.cast(typing.Any, ...)
 

vellum/types/__init__.py

@@ -37,7 +37,6 @@ from .logprobs_enum import LogprobsEnum
 from .metadata_filter_config_request import MetadataFilterConfigRequest
 from .metadata_filter_rule_combinator import MetadataFilterRuleCombinator
 from .metadata_filter_rule_request import MetadataFilterRuleRequest
-from .model_type_deprecated import ModelTypeDeprecated
 from .model_version_build_config import ModelVersionBuildConfig
 from .model_version_compile_prompt_response import ModelVersionCompilePromptResponse
 from .model_version_compiled_prompt import ModelVersionCompiledPrompt
@@ -47,6 +46,7 @@ from .model_version_read import ModelVersionRead
 from .model_version_read_status_enum import ModelVersionReadStatusEnum
 from .model_version_sandbox_snapshot import ModelVersionSandboxSnapshot
 from .node_input_compiled_chat_history_value import NodeInputCompiledChatHistoryValue
+from .node_input_compiled_error_value import NodeInputCompiledErrorValue
 from .node_input_compiled_json_value import NodeInputCompiledJsonValue
 from .node_input_compiled_number_value import NodeInputCompiledNumberValue
 from .node_input_compiled_search_results_value import NodeInputCompiledSearchResultsValue
@@ -54,6 +54,7 @@ from .node_input_compiled_string_value import NodeInputCompiledStringValue
 from .node_input_variable_compiled_value import (
     NodeInputVariableCompiledValue,
     NodeInputVariableCompiledValue_ChatHistory,
+    NodeInputVariableCompiledValue_Error,
     NodeInputVariableCompiledValue_Json,
     NodeInputVariableCompiledValue_Number,
     NodeInputVariableCompiledValue_SearchResults,
@@ -109,6 +110,7 @@ from .submit_workflow_execution_actual_request import (
     SubmitWorkflowExecutionActualRequest_String,
 )
 from .templating_node_chat_history_result import TemplatingNodeChatHistoryResult
+from .templating_node_error_result import TemplatingNodeErrorResult
 from .templating_node_json_result import TemplatingNodeJsonResult
 from .templating_node_number_result import TemplatingNodeNumberResult
 from .templating_node_result import TemplatingNodeResult
@@ -116,6 +118,7 @@ from .templating_node_result_data import TemplatingNodeResultData
 from .templating_node_result_output import (
     TemplatingNodeResultOutput,
     TemplatingNodeResultOutput_ChatHistory,
+    TemplatingNodeResultOutput_Error,
     TemplatingNodeResultOutput_Json,
     TemplatingNodeResultOutput_Number,
     TemplatingNodeResultOutput_SearchResults,
@@ -124,6 +127,7 @@ from .templating_node_result_output import (
 from .templating_node_search_results_result import TemplatingNodeSearchResultsResult
 from .templating_node_string_result import TemplatingNodeStringResult
 from .terminal_node_chat_history_result import TerminalNodeChatHistoryResult
+from .terminal_node_error_result import TerminalNodeErrorResult
 from .terminal_node_json_result import TerminalNodeJsonResult
 from .terminal_node_number_result import TerminalNodeNumberResult
 from .terminal_node_result import TerminalNodeResult
@@ -131,6 +135,7 @@ from .terminal_node_result_data import TerminalNodeResultData
 from .terminal_node_result_output import (
     TerminalNodeResultOutput,
     TerminalNodeResultOutput_ChatHistory,
+    TerminalNodeResultOutput_Error,
     TerminalNodeResultOutput_Json,
     TerminalNodeResultOutput_Number,
     TerminalNodeResultOutput_SearchResults,
@@ -141,6 +146,8 @@ from .terminal_node_string_result import TerminalNodeStringResult
 from .test_suite_test_case import TestSuiteTestCase
 from .upload_document_error_response import UploadDocumentErrorResponse
 from .upload_document_response import UploadDocumentResponse
+from .vellum_error import VellumError
+from .vellum_error_code_enum import VellumErrorCodeEnum
 from .vellum_variable import VellumVariable
 from .vellum_variable_type import VellumVariableType
 from .workflow_event_error import WorkflowEventError
@@ -175,12 +182,14 @@ from .workflow_result_event import WorkflowResultEvent
 from .workflow_result_event_output_data import (
     WorkflowResultEventOutputData,
     WorkflowResultEventOutputData_ChatHistory,
+    WorkflowResultEventOutputData_Error,
     WorkflowResultEventOutputData_Json,
     WorkflowResultEventOutputData_Number,
     WorkflowResultEventOutputData_SearchResults,
     WorkflowResultEventOutputData_String,
 )
 from .workflow_result_event_output_data_chat_history import WorkflowResultEventOutputDataChatHistory
+from .workflow_result_event_output_data_error import WorkflowResultEventOutputDataError
 from .workflow_result_event_output_data_json import WorkflowResultEventOutputDataJson
 from .workflow_result_event_output_data_number import WorkflowResultEventOutputDataNumber
 from .workflow_result_event_output_data_search_results import WorkflowResultEventOutputDataSearchResults
@@ -225,7 +234,6 @@ __all__ = [
     "MetadataFilterConfigRequest",
     "MetadataFilterRuleCombinator",
     "MetadataFilterRuleRequest",
-    "ModelTypeDeprecated",
     "ModelVersionBuildConfig",
     "ModelVersionCompilePromptResponse",
     "ModelVersionCompiledPrompt",
@@ -235,12 +243,14 @@ __all__ = [
     "ModelVersionReadStatusEnum",
     "ModelVersionSandboxSnapshot",
     "NodeInputCompiledChatHistoryValue",
+    "NodeInputCompiledErrorValue",
     "NodeInputCompiledJsonValue",
     "NodeInputCompiledNumberValue",
     "NodeInputCompiledSearchResultsValue",
     "NodeInputCompiledStringValue",
     "NodeInputVariableCompiledValue",
     "NodeInputVariableCompiledValue_ChatHistory",
+    "NodeInputVariableCompiledValue_Error",
     "NodeInputVariableCompiledValue_Json",
     "NodeInputVariableCompiledValue_Number",
     "NodeInputVariableCompiledValue_SearchResults",
@@ -293,12 +303,14 @@ __all__ = [
     "SubmitWorkflowExecutionActualRequest_Json",
     "SubmitWorkflowExecutionActualRequest_String",
     "TemplatingNodeChatHistoryResult",
+    "TemplatingNodeErrorResult",
     "TemplatingNodeJsonResult",
     "TemplatingNodeNumberResult",
     "TemplatingNodeResult",
     "TemplatingNodeResultData",
     "TemplatingNodeResultOutput",
     "TemplatingNodeResultOutput_ChatHistory",
+    "TemplatingNodeResultOutput_Error",
     "TemplatingNodeResultOutput_Json",
     "TemplatingNodeResultOutput_Number",
     "TemplatingNodeResultOutput_SearchResults",
@@ -306,12 +318,14 @@ __all__ = [
     "TemplatingNodeSearchResultsResult",
     "TemplatingNodeStringResult",
     "TerminalNodeChatHistoryResult",
+    "TerminalNodeErrorResult",
     "TerminalNodeJsonResult",
     "TerminalNodeNumberResult",
     "TerminalNodeResult",
     "TerminalNodeResultData",
     "TerminalNodeResultOutput",
     "TerminalNodeResultOutput_ChatHistory",
+    "TerminalNodeResultOutput_Error",
     "TerminalNodeResultOutput_Json",
     "TerminalNodeResultOutput_Number",
     "TerminalNodeResultOutput_SearchResults",
@@ -321,6 +335,8 @@ __all__ = [
     "TestSuiteTestCase",
     "UploadDocumentErrorResponse",
     "UploadDocumentResponse",
+    "VellumError",
+    "VellumErrorCodeEnum",
     "VellumVariable",
     "VellumVariableType",
     "WorkflowEventError",
@@ -350,11 +366,13 @@ __all__ = [
     "WorkflowResultEvent",
     "WorkflowResultEventOutputData",
     "WorkflowResultEventOutputDataChatHistory",
+    "WorkflowResultEventOutputDataError",
     "WorkflowResultEventOutputDataJson",
     "WorkflowResultEventOutputDataNumber",
     "WorkflowResultEventOutputDataSearchResults",
     "WorkflowResultEventOutputDataString",
     "WorkflowResultEventOutputData_ChatHistory",
+    "WorkflowResultEventOutputData_Error",
     "WorkflowResultEventOutputData_Json",
     "WorkflowResultEventOutputData_Number",
     "WorkflowResultEventOutputData_SearchResults",

vellum/types/api_node_result.py

@@ -3,11 +3,14 @@
 import datetime as dt
 import typing
 
-import pydantic
-
 from ..core.datetime_utils import serialize_datetime
 from .api_node_result_data import ApiNodeResultData
 
+try:
+    import pydantic.v1 as pydantic  # type: ignore
+except ImportError:
+    import pydantic  # type: ignore
+
 
 class ApiNodeResult(pydantic.BaseModel):
     data: ApiNodeResultData

vellum/types/api_node_result_data.py

@@ -3,10 +3,13 @@
 import datetime as dt
 import typing
 
-import pydantic
-
 from ..core.datetime_utils import serialize_datetime
 
+try:
+    import pydantic.v1 as pydantic  # type: ignore
+except ImportError:
+    import pydantic  # type: ignore
+
 
 class ApiNodeResultData(pydantic.BaseModel):
     text_output_id: str

vellum/types/chat_message.py

@@ -3,11 +3,14 @@
 import datetime as dt
 import typing
 
-import pydantic
-
 from ..core.datetime_utils import serialize_datetime
 from .chat_message_role import ChatMessageRole
 
+try:
+    import pydantic.v1 as pydantic  # type: ignore
+except ImportError:
+    import pydantic  # type: ignore
+
 
 class ChatMessage(pydantic.BaseModel):
     text: str

vellum/types/chat_message_request.py

@@ -3,11 +3,14 @@
 import datetime as dt
 import typing
 
-import pydantic
-
 from ..core.datetime_utils import serialize_datetime
 from .chat_message_role import ChatMessageRole
 
+try:
+    import pydantic.v1 as pydantic  # type: ignore
+except ImportError:
+    import pydantic  # type: ignore
+
 
 class ChatMessageRequest(pydantic.BaseModel):
     text: str

vellum/types/conditional_node_result.py

@@ -3,11 +3,14 @@
 import datetime as dt
 import typing
 
-import pydantic
-
 from ..core.datetime_utils import serialize_datetime
 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):
     data: ConditionalNodeResultData

vellum/types/conditional_node_result_data.py

@@ -3,10 +3,13 @@
 import datetime as dt
 import typing
 
-import pydantic
-
 from ..core.datetime_utils import serialize_datetime
 
+try:
+    import pydantic.v1 as pydantic  # type: ignore
+except ImportError:
+    import pydantic  # type: ignore
+
 
 class ConditionalNodeResultData(pydantic.BaseModel):
     source_handle_id: typing.Optional[str]