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

vellum/types/__init__.py

@@ -28,7 +28,6 @@ from .array_variable_value_item import (
     ArrayVariableValueItem_SearchResults,
     ArrayVariableValueItem_String,
 )
-from .block_type_enum import BlockTypeEnum
 from .chat_history_enum import ChatHistoryEnum
 from .chat_history_input_request import ChatHistoryInputRequest
 from .chat_history_variable_value import ChatHistoryVariableValue
@@ -131,6 +130,7 @@ from .fulfilled_execute_prompt_event import FulfilledExecutePromptEvent
 from .fulfilled_execute_prompt_response import FulfilledExecutePromptResponse
 from .fulfilled_execute_workflow_workflow_result_event import FulfilledExecuteWorkflowWorkflowResultEvent
 from .fulfilled_function_call import FulfilledFunctionCall
+from .fulfilled_function_call_request import FulfilledFunctionCallRequest
 from .fulfilled_prompt_execution_meta import FulfilledPromptExecutionMeta
 from .fulfilled_workflow_node_result_event import FulfilledWorkflowNodeResultEvent
 from .function_call import FunctionCall, FunctionCall_Fulfilled, FunctionCall_Rejected
@@ -168,16 +168,19 @@ from .metadata_filter_config_request import MetadataFilterConfigRequest
 from .metadata_filter_rule_combinator import MetadataFilterRuleCombinator
 from .metadata_filter_rule_request import MetadataFilterRuleRequest
 from .ml_model_usage import MlModelUsage
-from .model_version_build_config import ModelVersionBuildConfig
-from .model_version_exec_config import ModelVersionExecConfig
-from .model_version_exec_config_parameters import ModelVersionExecConfigParameters
-from .model_version_read import ModelVersionRead
-from .model_version_read_status_enum import ModelVersionReadStatusEnum
-from .model_version_sandbox_snapshot import ModelVersionSandboxSnapshot
+from .named_scenario_input_chat_history_variable_value_request import NamedScenarioInputChatHistoryVariableValueRequest
+from .named_scenario_input_request import (
+    NamedScenarioInputRequest,
+    NamedScenarioInputRequest_ChatHistory,
+    NamedScenarioInputRequest_String,
+)
+from .named_scenario_input_string_variable_value_request import NamedScenarioInputStringVariableValueRequest
 from .named_test_case_chat_history_variable_value import NamedTestCaseChatHistoryVariableValue
 from .named_test_case_chat_history_variable_value_request import NamedTestCaseChatHistoryVariableValueRequest
 from .named_test_case_error_variable_value import NamedTestCaseErrorVariableValue
 from .named_test_case_error_variable_value_request import NamedTestCaseErrorVariableValueRequest
+from .named_test_case_function_call_variable_value import NamedTestCaseFunctionCallVariableValue
+from .named_test_case_function_call_variable_value_request import NamedTestCaseFunctionCallVariableValueRequest
 from .named_test_case_json_variable_value import NamedTestCaseJsonVariableValue
 from .named_test_case_json_variable_value_request import NamedTestCaseJsonVariableValueRequest
 from .named_test_case_number_variable_value import NamedTestCaseNumberVariableValue
@@ -190,6 +193,7 @@ from .named_test_case_variable_value import (
     NamedTestCaseVariableValue,
     NamedTestCaseVariableValue_ChatHistory,
     NamedTestCaseVariableValue_Error,
+    NamedTestCaseVariableValue_FunctionCall,
     NamedTestCaseVariableValue_Json,
     NamedTestCaseVariableValue_Number,
     NamedTestCaseVariableValue_SearchResults,
@@ -199,6 +203,7 @@ from .named_test_case_variable_value_request import (
     NamedTestCaseVariableValueRequest,
     NamedTestCaseVariableValueRequest_ChatHistory,
     NamedTestCaseVariableValueRequest_Error,
+    NamedTestCaseVariableValueRequest_FunctionCall,
     NamedTestCaseVariableValueRequest_Json,
     NamedTestCaseVariableValueRequest_Number,
     NamedTestCaseVariableValueRequest_SearchResults,
@@ -271,25 +276,7 @@ from .prompt_output import (
     PromptOutput_Json,
     PromptOutput_String,
 )
-from .prompt_template_block import PromptTemplateBlock
-from .prompt_template_block_data import PromptTemplateBlockData
-from .prompt_template_block_data_request import PromptTemplateBlockDataRequest
-from .prompt_template_block_properties import PromptTemplateBlockProperties
-from .prompt_template_block_properties_request import PromptTemplateBlockPropertiesRequest
-from .prompt_template_block_request import PromptTemplateBlockRequest
-from .prompt_template_block_state import PromptTemplateBlockState
-from .provider_enum import ProviderEnum
 from .raw_prompt_execution_overrides_request import RawPromptExecutionOverridesRequest
-from .register_prompt_error_response import RegisterPromptErrorResponse
-from .register_prompt_model_parameters_request import RegisterPromptModelParametersRequest
-from .register_prompt_prompt import RegisterPromptPrompt
-from .register_prompt_prompt_info_request import RegisterPromptPromptInfoRequest
-from .register_prompt_response import RegisterPromptResponse
-from .registered_prompt_deployment import RegisteredPromptDeployment
-from .registered_prompt_input_variable_request import RegisteredPromptInputVariableRequest
-from .registered_prompt_model_version import RegisteredPromptModelVersion
-from .registered_prompt_sandbox import RegisteredPromptSandbox
-from .registered_prompt_sandbox_snapshot import RegisteredPromptSandboxSnapshot
 from .rejected_enum import RejectedEnum
 from .rejected_execute_prompt_event import RejectedExecutePromptEvent
 from .rejected_execute_prompt_response import RejectedExecutePromptResponse
@@ -298,9 +285,9 @@ from .rejected_function_call import RejectedFunctionCall
 from .rejected_prompt_execution_meta import RejectedPromptExecutionMeta
 from .rejected_workflow_node_result_event import RejectedWorkflowNodeResultEvent
 from .sandbox_scenario import SandboxScenario
-from .scenario_input import ScenarioInput
-from .scenario_input_request import ScenarioInputRequest
-from .scenario_input_type_enum import ScenarioInputTypeEnum
+from .scenario_input import ScenarioInput, ScenarioInput_ChatHistory, ScenarioInput_String
+from .scenario_input_chat_history_variable_value import ScenarioInputChatHistoryVariableValue
+from .scenario_input_string_variable_value import ScenarioInputStringVariableValue
 from .search_error_response import SearchErrorResponse
 from .search_filters_request import SearchFiltersRequest
 from .search_node_result import SearchNodeResult
@@ -381,6 +368,7 @@ from .terminal_node_search_results_result import TerminalNodeSearchResultsResult
 from .terminal_node_string_result import TerminalNodeStringResult
 from .test_case_chat_history_variable_value import TestCaseChatHistoryVariableValue
 from .test_case_error_variable_value import TestCaseErrorVariableValue
+from .test_case_function_call_variable_value import TestCaseFunctionCallVariableValue
 from .test_case_json_variable_value import TestCaseJsonVariableValue
 from .test_case_number_variable_value import TestCaseNumberVariableValue
 from .test_case_search_results_variable_value import TestCaseSearchResultsVariableValue
@@ -389,6 +377,7 @@ from .test_case_variable_value import (
     TestCaseVariableValue,
     TestCaseVariableValue_ChatHistory,
     TestCaseVariableValue_Error,
+    TestCaseVariableValue_FunctionCall,
     TestCaseVariableValue_Json,
     TestCaseVariableValue_Number,
     TestCaseVariableValue_SearchResults,
@@ -418,6 +407,7 @@ from .test_suite_run_exec_config_request import (
 from .test_suite_run_execution import TestSuiteRunExecution
 from .test_suite_run_execution_chat_history_output import TestSuiteRunExecutionChatHistoryOutput
 from .test_suite_run_execution_error_output import TestSuiteRunExecutionErrorOutput
+from .test_suite_run_execution_function_call_output import TestSuiteRunExecutionFunctionCallOutput
 from .test_suite_run_execution_json_output import TestSuiteRunExecutionJsonOutput
 from .test_suite_run_execution_metric_definition import TestSuiteRunExecutionMetricDefinition
 from .test_suite_run_execution_metric_result import TestSuiteRunExecutionMetricResult
@@ -426,6 +416,7 @@ from .test_suite_run_execution_output import (
     TestSuiteRunExecutionOutput,
     TestSuiteRunExecutionOutput_ChatHistory,
     TestSuiteRunExecutionOutput_Error,
+    TestSuiteRunExecutionOutput_FunctionCall,
     TestSuiteRunExecutionOutput_Json,
     TestSuiteRunExecutionOutput_Number,
     TestSuiteRunExecutionOutput_SearchResults,
@@ -575,7 +566,6 @@ __all__ = [
     "ArrayVariableValueItem_Number",
     "ArrayVariableValueItem_SearchResults",
     "ArrayVariableValueItem_String",
-    "BlockTypeEnum",
     "ChatHistoryEnum",
     "ChatHistoryInputRequest",
     "ChatHistoryVariableValue",
@@ -664,6 +654,7 @@ __all__ = [
     "FulfilledExecutePromptResponse",
     "FulfilledExecuteWorkflowWorkflowResultEvent",
     "FulfilledFunctionCall",
+    "FulfilledFunctionCallRequest",
     "FulfilledPromptExecutionMeta",
     "FulfilledWorkflowNodeResultEvent",
     "FunctionCall",
@@ -703,16 +694,17 @@ __all__ = [
     "MetadataFilterRuleCombinator",
     "MetadataFilterRuleRequest",
     "MlModelUsage",
-    "ModelVersionBuildConfig",
-    "ModelVersionExecConfig",
-    "ModelVersionExecConfigParameters",
-    "ModelVersionRead",
-    "ModelVersionReadStatusEnum",
-    "ModelVersionSandboxSnapshot",
+    "NamedScenarioInputChatHistoryVariableValueRequest",
+    "NamedScenarioInputRequest",
+    "NamedScenarioInputRequest_ChatHistory",
+    "NamedScenarioInputRequest_String",
+    "NamedScenarioInputStringVariableValueRequest",
     "NamedTestCaseChatHistoryVariableValue",
     "NamedTestCaseChatHistoryVariableValueRequest",
     "NamedTestCaseErrorVariableValue",
     "NamedTestCaseErrorVariableValueRequest",
+    "NamedTestCaseFunctionCallVariableValue",
+    "NamedTestCaseFunctionCallVariableValueRequest",
     "NamedTestCaseJsonVariableValue",
     "NamedTestCaseJsonVariableValueRequest",
     "NamedTestCaseNumberVariableValue",
@@ -725,12 +717,14 @@ __all__ = [
     "NamedTestCaseVariableValueRequest",
     "NamedTestCaseVariableValueRequest_ChatHistory",
     "NamedTestCaseVariableValueRequest_Error",
+    "NamedTestCaseVariableValueRequest_FunctionCall",
     "NamedTestCaseVariableValueRequest_Json",
     "NamedTestCaseVariableValueRequest_Number",
     "NamedTestCaseVariableValueRequest_SearchResults",
     "NamedTestCaseVariableValueRequest_String",
     "NamedTestCaseVariableValue_ChatHistory",
     "NamedTestCaseVariableValue_Error",
+    "NamedTestCaseVariableValue_FunctionCall",
     "NamedTestCaseVariableValue_Json",
     "NamedTestCaseVariableValue_Number",
     "NamedTestCaseVariableValue_SearchResults",
@@ -794,25 +788,7 @@ __all__ = [
     "PromptOutput_FunctionCall",
     "PromptOutput_Json",
     "PromptOutput_String",
-    "PromptTemplateBlock",
-    "PromptTemplateBlockData",
-    "PromptTemplateBlockDataRequest",
-    "PromptTemplateBlockProperties",
-    "PromptTemplateBlockPropertiesRequest",
-    "PromptTemplateBlockRequest",
-    "PromptTemplateBlockState",
-    "ProviderEnum",
     "RawPromptExecutionOverridesRequest",
-    "RegisterPromptErrorResponse",
-    "RegisterPromptModelParametersRequest",
-    "RegisterPromptPrompt",
-    "RegisterPromptPromptInfoRequest",
-    "RegisterPromptResponse",
-    "RegisteredPromptDeployment",
-    "RegisteredPromptInputVariableRequest",
-    "RegisteredPromptModelVersion",
-    "RegisteredPromptSandbox",
-    "RegisteredPromptSandboxSnapshot",
     "RejectedEnum",
     "RejectedExecutePromptEvent",
     "RejectedExecutePromptResponse",
@@ -822,8 +798,10 @@ __all__ = [
     "RejectedWorkflowNodeResultEvent",
     "SandboxScenario",
     "ScenarioInput",
-    "ScenarioInputRequest",
-    "ScenarioInputTypeEnum",
+    "ScenarioInputChatHistoryVariableValue",
+    "ScenarioInputStringVariableValue",
+    "ScenarioInput_ChatHistory",
+    "ScenarioInput_String",
     "SearchErrorResponse",
     "SearchFiltersRequest",
     "SearchNodeResult",
@@ -898,6 +876,7 @@ __all__ = [
     "TerminalNodeStringResult",
     "TestCaseChatHistoryVariableValue",
     "TestCaseErrorVariableValue",
+    "TestCaseFunctionCallVariableValue",
     "TestCaseJsonVariableValue",
     "TestCaseNumberVariableValue",
     "TestCaseSearchResultsVariableValue",
@@ -905,6 +884,7 @@ __all__ = [
     "TestCaseVariableValue",
     "TestCaseVariableValue_ChatHistory",
     "TestCaseVariableValue_Error",
+    "TestCaseVariableValue_FunctionCall",
     "TestCaseVariableValue_Json",
     "TestCaseVariableValue_Number",
     "TestCaseVariableValue_SearchResults",
@@ -925,6 +905,7 @@ __all__ = [
     "TestSuiteRunExecution",
     "TestSuiteRunExecutionChatHistoryOutput",
     "TestSuiteRunExecutionErrorOutput",
+    "TestSuiteRunExecutionFunctionCallOutput",
     "TestSuiteRunExecutionJsonOutput",
     "TestSuiteRunExecutionMetricDefinition",
     "TestSuiteRunExecutionMetricResult",
@@ -932,6 +913,7 @@ __all__ = [
     "TestSuiteRunExecutionOutput",
     "TestSuiteRunExecutionOutput_ChatHistory",
     "TestSuiteRunExecutionOutput_Error",
+    "TestSuiteRunExecutionOutput_FunctionCall",
     "TestSuiteRunExecutionOutput_Json",
     "TestSuiteRunExecutionOutput_Number",
     "TestSuiteRunExecutionOutput_SearchResults",

vellum/types/api_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 .api_node_result_data import ApiNodeResultData
 
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
 
-
-class ApiNodeResult(pydantic.BaseModel):
+class ApiNodeResult(pydantic_v1.BaseModel):
     """
     A Node Result Event emitted from an API Node.
     """
@@ -30,5 +26,5 @@ class ApiNodeResult(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/api_node_result_data.py

@@ -4,18 +4,14 @@ 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 ApiNodeResultData(pydantic.BaseModel):
+class ApiNodeResultData(pydantic_v1.BaseModel):
     text_output_id: str
     text: typing.Optional[str] = None
     json_output_id: str
-    json_: typing.Optional[typing.Dict[str, typing.Any]] = pydantic.Field(alias="json", default=None)
+    json_: typing.Optional[typing.Dict[str, typing.Any]] = pydantic_v1.Field(alias="json", default=None)
     status_code_output_id: str
     status_code: int
 
@@ -32,5 +28,5 @@ class ApiNodeResultData(pydantic.BaseModel):
         smart_union = True
         allow_population_by_field_name = True
         populate_by_name = True
-        extra = pydantic.Extra.allow
+        extra = pydantic_v1.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}

vellum/types/array_chat_message_content.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_chat_message_content_item import ArrayChatMessageContentItem
 
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
 
-
-class ArrayChatMessageContent(pydantic.BaseModel):
+class ArrayChatMessageContent(pydantic_v1.BaseModel):
     """
     A list of chat message content items.
     """
@@ -30,5 +26,5 @@ class ArrayChatMessageContent(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/array_chat_message_content_request.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_chat_message_content_item_request import ArrayChatMessageContentItemRequest
 
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
 
-
-class ArrayChatMessageContentRequest(pydantic.BaseModel):
+class ArrayChatMessageContentRequest(pydantic_v1.BaseModel):
     """
     A list of chat message content items.
     """
@@ -30,5 +26,5 @@ class ArrayChatMessageContentRequest(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/chat_history_input_request.py

@@ -4,20 +4,16 @@ 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 ChatHistoryInputRequest(pydantic.BaseModel):
+class ChatHistoryInputRequest(pydantic_v1.BaseModel):
     """
     A user input representing a list of chat messages
     """
 
-    name: str = pydantic.Field()
+    name: str = pydantic_v1.Field()
     """
     The variable's name, as defined in the deployment.
     """
@@ -35,5 +31,5 @@ class ChatHistoryInputRequest(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/chat_history_variable_value.py

@@ -4,15 +4,11 @@ import datetime as dt
 import typing
 
 from ..core.datetime_utils import serialize_datetime
+from ..core.pydantic_utilities import pydantic_v1
 from .chat_message import ChatMessage
 
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
 
-
-class ChatHistoryVariableValue(pydantic.BaseModel):
+class ChatHistoryVariableValue(pydantic_v1.BaseModel):
     value: typing.Optional[typing.List[ChatMessage]] = None
 
     def json(self, **kwargs: typing.Any) -> str:
@@ -26,5 +22,5 @@ class ChatHistoryVariableValue(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/chat_message.py

@@ -4,20 +4,16 @@ import datetime as dt
 import typing
 
 from ..core.datetime_utils import serialize_datetime
+from ..core.pydantic_utilities import pydantic_v1
 from .chat_message_content import ChatMessageContent
 from .chat_message_role import ChatMessageRole
 
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
 
-
-class ChatMessage(pydantic.BaseModel):
+class ChatMessage(pydantic_v1.BaseModel):
     text: typing.Optional[str] = None
     role: ChatMessageRole
     content: typing.Optional[ChatMessageContent] = None
-    source: typing.Optional[str] = pydantic.Field(default=None)
+    source: typing.Optional[str] = pydantic_v1.Field(default=None)
     """
     An optional identifier representing who or what generated this message.
     """
@@ -33,5 +29,5 @@ class ChatMessage(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/chat_message_request.py

@@ -4,20 +4,16 @@ import datetime as dt
 import typing
 
 from ..core.datetime_utils import serialize_datetime
+from ..core.pydantic_utilities import pydantic_v1
 from .chat_message_content_request import ChatMessageContentRequest
 from .chat_message_role import ChatMessageRole
 
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
 
-
-class ChatMessageRequest(pydantic.BaseModel):
+class ChatMessageRequest(pydantic_v1.BaseModel):
     text: typing.Optional[str] = None
     role: ChatMessageRole
     content: typing.Optional[ChatMessageContentRequest] = None
-    source: typing.Optional[str] = pydantic.Field(default=None)
+    source: typing.Optional[str] = pydantic_v1.Field(default=None)
     """
     An optional identifier representing who or what generated this message.
     """
@@ -33,5 +29,5 @@ class ChatMessageRequest(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/chat_message_role.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 ChatMessageRole(str, enum.Enum):
-    """
-    - `SYSTEM` - System
-    - `ASSISTANT` - Assistant
-    - `USER` - User
-    - `FUNCTION` - Function
-    """
-
-    SYSTEM = "SYSTEM"
-    ASSISTANT = "ASSISTANT"
-    USER = "USER"
-    FUNCTION = "FUNCTION"
-
-    def visit(
-        self,
-        system: typing.Callable[[], T_Result],
-        assistant: typing.Callable[[], T_Result],
-        user: typing.Callable[[], T_Result],
-        function: typing.Callable[[], T_Result],
-    ) -> T_Result:
-        if self is ChatMessageRole.SYSTEM:
-            return system()
-        if self is ChatMessageRole.ASSISTANT:
-            return assistant()
-        if self is ChatMessageRole.USER:
-            return user()
-        if self is ChatMessageRole.FUNCTION:
-            return function()
+ChatMessageRole = typing.Union[typing.Literal["SYSTEM", "ASSISTANT", "USER", "FUNCTION"], typing.Any]

vellum/types/code_execution_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 CodeExecutionNodeArrayResult(pydantic.BaseModel):
+class CodeExecutionNodeArrayResult(pydantic_v1.BaseModel):
     id: str
     value: typing.Optional[typing.List[ArrayVariableValueItem]] = None
 
@@ -27,5 +23,5 @@ class CodeExecutionNodeArrayResult(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
-        extra = pydantic.Extra.allow
+        extra = pydantic_v1.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}

vellum/types/code_execution_node_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 CodeExecutionNodeChatHistoryResult(pydantic.BaseModel):
+class CodeExecutionNodeChatHistoryResult(pydantic_v1.BaseModel):
     id: str
     value: typing.Optional[typing.List[ChatMessage]] = None
 
@@ -27,5 +23,5 @@ class CodeExecutionNodeChatHistoryResult(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
-        extra = pydantic.Extra.allow
+        extra = pydantic_v1.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}

vellum/types/code_execution_node_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 CodeExecutionNodeErrorResult(pydantic.BaseModel):
+class CodeExecutionNodeErrorResult(pydantic_v1.BaseModel):
     id: str
     value: typing.Optional[VellumError] = None
 
@@ -27,5 +23,5 @@ class CodeExecutionNodeErrorResult(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
-        extra = pydantic.Extra.allow
+        extra = pydantic_v1.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}

vellum/types/code_execution_node_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 CodeExecutionNodeFunctionCallResult(pydantic.BaseModel):
+class CodeExecutionNodeFunctionCallResult(pydantic_v1.BaseModel):
     id: str
     value: typing.Optional[FunctionCall] = None
 
@@ -27,5 +23,5 @@ class CodeExecutionNodeFunctionCallResult(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
-        extra = pydantic.Extra.allow
+        extra = pydantic_v1.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}

vellum/types/code_execution_node_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 CodeExecutionNodeJsonResult(pydantic.BaseModel):
+class CodeExecutionNodeJsonResult(pydantic_v1.BaseModel):
     id: str
     value: typing.Optional[typing.Dict[str, typing.Any]] = None
 
@@ -26,5 +22,5 @@ class CodeExecutionNodeJsonResult(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
-        extra = pydantic.Extra.allow
+        extra = pydantic_v1.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}

vellum/types/code_execution_node_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 CodeExecutionNodeNumberResult(pydantic.BaseModel):
+class CodeExecutionNodeNumberResult(pydantic_v1.BaseModel):
     id: str
     value: typing.Optional[float] = None
 
@@ -26,5 +22,5 @@ class CodeExecutionNodeNumberResult(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
-        extra = pydantic.Extra.allow
+        extra = pydantic_v1.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}