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

vellum/types/{registered_prompt_sandbox_snapshot.py → named_test_case_number_variable_value.py}

@@ -4,19 +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 RegisteredPromptSandboxSnapshot(pydantic.BaseModel):
-    id: str = pydantic.Field()
+class NamedTestCaseNumberVariableValue(pydantic_v1.BaseModel):
     """
-    The ID of the generated sandbox snapshot.
+    Named Test Case value that is of type NUMBER
     """
 
+    value: typing.Optional[float] = None
+    name: str
+
     def json(self, **kwargs: typing.Any) -> str:
         kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
         return super().json(**kwargs_with_defaults)
@@ -28,5 +26,5 @@ class RegisteredPromptSandboxSnapshot(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/named_test_case_number_variable_value_request.py

@@ -4,14 +4,10 @@ import datetime as dt
 import typing
 
 from ..core.datetime_utils import serialize_datetime
+from ..core.pydantic_utilities import pydantic_v1
 
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
 
-
-class NamedTestCaseNumberVariableValueRequest(pydantic.BaseModel):
+class NamedTestCaseNumberVariableValueRequest(pydantic_v1.BaseModel):
     """
     Named Test Case value that is of type NUMBER
     """
@@ -30,5 +26,5 @@ class NamedTestCaseNumberVariableValueRequest(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/named_test_case_search_results_variable_value.py

@@ -0,0 +1,31 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import datetime as dt
+import typing
+
+from ..core.datetime_utils import serialize_datetime
+from ..core.pydantic_utilities import pydantic_v1
+from .search_result import SearchResult
+
+
+class NamedTestCaseSearchResultsVariableValue(pydantic_v1.BaseModel):
+    """
+    Named Test Case value that is of type SEARCH_RESULTS
+    """
+
+    value: typing.Optional[typing.List[SearchResult]] = None
+    name: str
+
+    def json(self, **kwargs: typing.Any) -> str:
+        kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
+        return super().json(**kwargs_with_defaults)
+
+    def dict(self, **kwargs: typing.Any) -> typing.Dict[str, typing.Any]:
+        kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
+        return super().dict(**kwargs_with_defaults)
+
+    class Config:
+        frozen = True
+        smart_union = True
+        extra = pydantic_v1.Extra.allow
+        json_encoders = {dt.datetime: serialize_datetime}

vellum/types/named_test_case_search_results_variable_value_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 .search_result_request import SearchResultRequest
 
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
 
-
-class NamedTestCaseSearchResultsVariableValueRequest(pydantic.BaseModel):
+class NamedTestCaseSearchResultsVariableValueRequest(pydantic_v1.BaseModel):
     """
     Named Test Case value that is of type SEARCH_RESULTS
     """
@@ -31,5 +27,5 @@ class NamedTestCaseSearchResultsVariableValueRequest(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/named_test_case_string_variable_value.py

@@ -0,0 +1,30 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import datetime as dt
+import typing
+
+from ..core.datetime_utils import serialize_datetime
+from ..core.pydantic_utilities import pydantic_v1
+
+
+class NamedTestCaseStringVariableValue(pydantic_v1.BaseModel):
+    """
+    Named Test Case value that is of type STRING
+    """
+
+    value: typing.Optional[str] = None
+    name: str
+
+    def json(self, **kwargs: typing.Any) -> str:
+        kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
+        return super().json(**kwargs_with_defaults)
+
+    def dict(self, **kwargs: typing.Any) -> typing.Dict[str, typing.Any]:
+        kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
+        return super().dict(**kwargs_with_defaults)
+
+    class Config:
+        frozen = True
+        smart_union = True
+        extra = pydantic_v1.Extra.allow
+        json_encoders = {dt.datetime: serialize_datetime}

vellum/types/named_test_case_string_variable_value_request.py

@@ -4,14 +4,10 @@ import datetime as dt
 import typing
 
 from ..core.datetime_utils import serialize_datetime
+from ..core.pydantic_utilities import pydantic_v1
 
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
 
-
-class NamedTestCaseStringVariableValueRequest(pydantic.BaseModel):
+class NamedTestCaseStringVariableValueRequest(pydantic_v1.BaseModel):
     """
     Named Test Case value that is of type STRING
     """
@@ -30,5 +26,5 @@ class NamedTestCaseStringVariableValueRequest(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/named_test_case_variable_value.py

@@ -0,0 +1,94 @@
+# This file was auto-generated by Fern from our API Definition.
+
+from __future__ import annotations
+
+import typing
+
+from .named_test_case_chat_history_variable_value import NamedTestCaseChatHistoryVariableValue
+from .named_test_case_error_variable_value import NamedTestCaseErrorVariableValue
+from .named_test_case_function_call_variable_value import NamedTestCaseFunctionCallVariableValue
+from .named_test_case_json_variable_value import NamedTestCaseJsonVariableValue
+from .named_test_case_number_variable_value import NamedTestCaseNumberVariableValue
+from .named_test_case_search_results_variable_value import NamedTestCaseSearchResultsVariableValue
+from .named_test_case_string_variable_value import NamedTestCaseStringVariableValue
+
+
+class NamedTestCaseVariableValue_String(NamedTestCaseStringVariableValue):
+    type: typing.Literal["STRING"] = "STRING"
+
+    class Config:
+        frozen = True
+        smart_union = True
+        allow_population_by_field_name = True
+        populate_by_name = True
+
+
+class NamedTestCaseVariableValue_Number(NamedTestCaseNumberVariableValue):
+    type: typing.Literal["NUMBER"] = "NUMBER"
+
+    class Config:
+        frozen = True
+        smart_union = True
+        allow_population_by_field_name = True
+        populate_by_name = True
+
+
+class NamedTestCaseVariableValue_Json(NamedTestCaseJsonVariableValue):
+    type: typing.Literal["JSON"] = "JSON"
+
+    class Config:
+        frozen = True
+        smart_union = True
+        allow_population_by_field_name = True
+        populate_by_name = True
+
+
+class NamedTestCaseVariableValue_ChatHistory(NamedTestCaseChatHistoryVariableValue):
+    type: typing.Literal["CHAT_HISTORY"] = "CHAT_HISTORY"
+
+    class Config:
+        frozen = True
+        smart_union = True
+        allow_population_by_field_name = True
+        populate_by_name = True
+
+
+class NamedTestCaseVariableValue_SearchResults(NamedTestCaseSearchResultsVariableValue):
+    type: typing.Literal["SEARCH_RESULTS"] = "SEARCH_RESULTS"
+
+    class Config:
+        frozen = True
+        smart_union = True
+        allow_population_by_field_name = True
+        populate_by_name = True
+
+
+class NamedTestCaseVariableValue_Error(NamedTestCaseErrorVariableValue):
+    type: typing.Literal["ERROR"] = "ERROR"
+
+    class Config:
+        frozen = True
+        smart_union = True
+        allow_population_by_field_name = True
+        populate_by_name = True
+
+
+class NamedTestCaseVariableValue_FunctionCall(NamedTestCaseFunctionCallVariableValue):
+    type: typing.Literal["FUNCTION_CALL"] = "FUNCTION_CALL"
+
+    class Config:
+        frozen = True
+        smart_union = True
+        allow_population_by_field_name = True
+        populate_by_name = True
+
+
+NamedTestCaseVariableValue = typing.Union[
+    NamedTestCaseVariableValue_String,
+    NamedTestCaseVariableValue_Number,
+    NamedTestCaseVariableValue_Json,
+    NamedTestCaseVariableValue_ChatHistory,
+    NamedTestCaseVariableValue_SearchResults,
+    NamedTestCaseVariableValue_Error,
+    NamedTestCaseVariableValue_FunctionCall,
+]

vellum/types/named_test_case_variable_value_request.py

@@ -6,6 +6,7 @@ import typing
 
 from .named_test_case_chat_history_variable_value_request import NamedTestCaseChatHistoryVariableValueRequest
 from .named_test_case_error_variable_value_request import NamedTestCaseErrorVariableValueRequest
+from .named_test_case_function_call_variable_value_request import NamedTestCaseFunctionCallVariableValueRequest
 from .named_test_case_json_variable_value_request import NamedTestCaseJsonVariableValueRequest
 from .named_test_case_number_variable_value_request import NamedTestCaseNumberVariableValueRequest
 from .named_test_case_search_results_variable_value_request import NamedTestCaseSearchResultsVariableValueRequest
@@ -72,6 +73,16 @@ class NamedTestCaseVariableValueRequest_Error(NamedTestCaseErrorVariableValueReq
         populate_by_name = True
 
 
+class NamedTestCaseVariableValueRequest_FunctionCall(NamedTestCaseFunctionCallVariableValueRequest):
+    type: typing.Literal["FUNCTION_CALL"] = "FUNCTION_CALL"
+
+    class Config:
+        frozen = True
+        smart_union = True
+        allow_population_by_field_name = True
+        populate_by_name = True
+
+
 NamedTestCaseVariableValueRequest = typing.Union[
     NamedTestCaseVariableValueRequest_String,
     NamedTestCaseVariableValueRequest_Number,
@@ -79,4 +90,5 @@ NamedTestCaseVariableValueRequest = typing.Union[
     NamedTestCaseVariableValueRequest_ChatHistory,
     NamedTestCaseVariableValueRequest_SearchResults,
     NamedTestCaseVariableValueRequest_Error,
+    NamedTestCaseVariableValueRequest_FunctionCall,
 ]

vellum/types/node_input_compiled_array_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 .array_variable_value_item import ArrayVariableValueItem
 
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
 
-
-class NodeInputCompiledArrayValue(pydantic.BaseModel):
+class NodeInputCompiledArrayValue(pydantic_v1.BaseModel):
     node_input_id: str
     key: str
     value: typing.Optional[typing.List[ArrayVariableValueItem]] = None
@@ -28,5 +24,5 @@ class NodeInputCompiledArrayValue(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/node_input_compiled_chat_history_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 NodeInputCompiledChatHistoryValue(pydantic.BaseModel):
+class NodeInputCompiledChatHistoryValue(pydantic_v1.BaseModel):
     node_input_id: str
     key: str
     value: typing.Optional[typing.List[ChatMessage]] = None
@@ -28,5 +24,5 @@ class NodeInputCompiledChatHistoryValue(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/node_input_compiled_error_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 .vellum_error import VellumError
 
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
 
-
-class NodeInputCompiledErrorValue(pydantic.BaseModel):
+class NodeInputCompiledErrorValue(pydantic_v1.BaseModel):
     node_input_id: str
     key: str
     value: typing.Optional[VellumError] = None
@@ -28,5 +24,5 @@ class NodeInputCompiledErrorValue(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/node_input_compiled_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 .function_call import FunctionCall
 
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
 
-
-class NodeInputCompiledFunctionCall(pydantic.BaseModel):
+class NodeInputCompiledFunctionCall(pydantic_v1.BaseModel):
     node_input_id: str
     key: str
     value: typing.Optional[FunctionCall] = None
@@ -28,5 +24,5 @@ class NodeInputCompiledFunctionCall(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/node_input_compiled_json_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 NodeInputCompiledJsonValue(pydantic.BaseModel):
+class NodeInputCompiledJsonValue(pydantic_v1.BaseModel):
     node_input_id: str
     key: str
     value: typing.Optional[typing.Dict[str, typing.Any]] = None
@@ -27,5 +23,5 @@ class NodeInputCompiledJsonValue(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/node_input_compiled_number_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 NodeInputCompiledNumberValue(pydantic.BaseModel):
+class NodeInputCompiledNumberValue(pydantic_v1.BaseModel):
     node_input_id: str
     key: str
     value: typing.Optional[float] = None
@@ -27,5 +23,5 @@ class NodeInputCompiledNumberValue(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/node_input_compiled_search_results_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 .search_result import SearchResult
 
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
 
-
-class NodeInputCompiledSearchResultsValue(pydantic.BaseModel):
+class NodeInputCompiledSearchResultsValue(pydantic_v1.BaseModel):
     node_input_id: str
     key: str
     value: typing.Optional[typing.List[SearchResult]] = None
@@ -28,5 +24,5 @@ class NodeInputCompiledSearchResultsValue(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/node_input_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 NodeInputCompiledStringValue(pydantic.BaseModel):
+class NodeInputCompiledStringValue(pydantic_v1.BaseModel):
     node_input_id: str
     key: str
     value: typing.Optional[str] = None
@@ -27,5 +23,5 @@ class NodeInputCompiledStringValue(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/node_output_compiled_array_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 .array_variable_value_item import ArrayVariableValueItem
 
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
 
-
-class NodeOutputCompiledArrayValue(pydantic.BaseModel):
+class NodeOutputCompiledArrayValue(pydantic_v1.BaseModel):
     node_output_id: str
     value: typing.Optional[typing.List[ArrayVariableValueItem]] = None
 
@@ -27,5 +23,5 @@ class NodeOutputCompiledArrayValue(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/node_output_compiled_chat_history_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 NodeOutputCompiledChatHistoryValue(pydantic.BaseModel):
+class NodeOutputCompiledChatHistoryValue(pydantic_v1.BaseModel):
     node_output_id: str
     value: typing.Optional[typing.List[ChatMessage]] = None
 
@@ -27,5 +23,5 @@ class NodeOutputCompiledChatHistoryValue(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/node_output_compiled_error_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 .vellum_error import VellumError
 
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
 
-
-class NodeOutputCompiledErrorValue(pydantic.BaseModel):
+class NodeOutputCompiledErrorValue(pydantic_v1.BaseModel):
     node_output_id: str
     value: typing.Optional[VellumError] = None
 
@@ -27,5 +23,5 @@ class NodeOutputCompiledErrorValue(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/node_output_compiled_function_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 .function_call import FunctionCall
 
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
 
-
-class NodeOutputCompiledFunctionValue(pydantic.BaseModel):
+class NodeOutputCompiledFunctionValue(pydantic_v1.BaseModel):
     node_output_id: str
     value: typing.Optional[FunctionCall] = None
 
@@ -27,5 +23,5 @@ class NodeOutputCompiledFunctionValue(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/node_output_compiled_json_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 NodeOutputCompiledJsonValue(pydantic.BaseModel):
+class NodeOutputCompiledJsonValue(pydantic_v1.BaseModel):
     node_output_id: str
     value: typing.Optional[typing.Dict[str, typing.Any]] = None
 
@@ -26,5 +22,5 @@ class NodeOutputCompiledJsonValue(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/node_output_compiled_number_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 NodeOutputCompiledNumberValue(pydantic.BaseModel):
+class NodeOutputCompiledNumberValue(pydantic_v1.BaseModel):
     node_output_id: str
     value: typing.Optional[float] = None
 
@@ -26,5 +22,5 @@ class NodeOutputCompiledNumberValue(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/node_output_compiled_search_results_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 .search_result import SearchResult
 
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
 
-
-class NodeOutputCompiledSearchResultsValue(pydantic.BaseModel):
+class NodeOutputCompiledSearchResultsValue(pydantic_v1.BaseModel):
     node_output_id: str
     value: typing.Optional[typing.List[SearchResult]] = None
 
@@ -27,5 +23,5 @@ class NodeOutputCompiledSearchResultsValue(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
-        extra = pydantic.Extra.allow
+        extra = pydantic_v1.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}