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

vellum/types/metadata_filter_config_request.py

@@ -4,17 +4,13 @@ import datetime as dt
 import typing
 
 from ..core.datetime_utils import serialize_datetime
+from ..core.pydantic_utilities import pydantic_v1
 from .logical_operator import LogicalOperator
 from .metadata_filter_rule_combinator import MetadataFilterRuleCombinator
 from .metadata_filter_rule_request import MetadataFilterRuleRequest
 
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
 
-
-class MetadataFilterConfigRequest(pydantic.BaseModel):
+class MetadataFilterConfigRequest(pydantic_v1.BaseModel):
     combinator: typing.Optional[MetadataFilterRuleCombinator] = None
     negated: typing.Optional[bool] = None
     rules: typing.Optional[typing.List[MetadataFilterRuleRequest]] = None
@@ -33,5 +29,5 @@ class MetadataFilterConfigRequest(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/metadata_filter_rule_combinator.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 MetadataFilterRuleCombinator(str, enum.Enum):
-    """
-    - `and` - AND
-    - `or` - OR
-    """
-
-    AND = "and"
-    OR = "or"
-
-    def visit(self, and_: typing.Callable[[], T_Result], or_: typing.Callable[[], T_Result]) -> T_Result:
-        if self is MetadataFilterRuleCombinator.AND:
-            return and_()
-        if self is MetadataFilterRuleCombinator.OR:
-            return or_()
+MetadataFilterRuleCombinator = typing.Union[typing.Literal["and", "or"], typing.Any]

vellum/types/metadata_filter_rule_request.py

@@ -6,16 +6,12 @@ import datetime as dt
 import typing
 
 from ..core.datetime_utils import serialize_datetime
+from ..core.pydantic_utilities import pydantic_v1
 from .logical_operator import LogicalOperator
 from .metadata_filter_rule_combinator import MetadataFilterRuleCombinator
 
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
 
-
-class MetadataFilterRuleRequest(pydantic.BaseModel):
+class MetadataFilterRuleRequest(pydantic_v1.BaseModel):
     combinator: typing.Optional[MetadataFilterRuleCombinator] = None
     negated: typing.Optional[bool] = None
     rules: typing.Optional[typing.List[MetadataFilterRuleRequest]] = None
@@ -34,7 +30,7 @@ class MetadataFilterRuleRequest(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/ml_model_usage.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 MlModelUsage(pydantic.BaseModel):
+class MlModelUsage(pydantic_v1.BaseModel):
     output_token_count: typing.Optional[int] = None
     input_token_count: typing.Optional[int] = None
     input_char_count: typing.Optional[int] = None
@@ -29,5 +25,5 @@ class MlModelUsage(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_scenario_input_chat_history_variable_value_request.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 .chat_message_request import ChatMessageRequest
+
+
+class NamedScenarioInputChatHistoryVariableValueRequest(pydantic_v1.BaseModel):
+    """
+    Named Prompt Sandbox Scenario input value that is of type CHAT_HISTORY
+    """
+
+    value: typing.Optional[typing.List[ChatMessageRequest]] = 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_scenario_input_request.py

@@ -0,0 +1,31 @@
+# This file was auto-generated by Fern from our API Definition.
+
+from __future__ import annotations
+
+import typing
+
+from .named_scenario_input_chat_history_variable_value_request import NamedScenarioInputChatHistoryVariableValueRequest
+from .named_scenario_input_string_variable_value_request import NamedScenarioInputStringVariableValueRequest
+
+
+class NamedScenarioInputRequest_String(NamedScenarioInputStringVariableValueRequest):
+    type: typing.Literal["STRING"] = "STRING"
+
+    class Config:
+        frozen = True
+        smart_union = True
+        allow_population_by_field_name = True
+        populate_by_name = True
+
+
+class NamedScenarioInputRequest_ChatHistory(NamedScenarioInputChatHistoryVariableValueRequest):
+    type: typing.Literal["CHAT_HISTORY"] = "CHAT_HISTORY"
+
+    class Config:
+        frozen = True
+        smart_union = True
+        allow_population_by_field_name = True
+        populate_by_name = True
+
+
+NamedScenarioInputRequest = typing.Union[NamedScenarioInputRequest_String, NamedScenarioInputRequest_ChatHistory]

vellum/types/{registered_prompt_sandbox_snapshot.py → named_scenario_input_string_variable_value_request.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 NamedScenarioInputStringVariableValueRequest(pydantic_v1.BaseModel):
     """
-    The ID of the generated sandbox snapshot.
+    Named Prompt Sandbox Scenario input 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)
@@ -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_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 NamedTestCaseChatHistoryVariableValue(pydantic.BaseModel):
+class NamedTestCaseChatHistoryVariableValue(pydantic_v1.BaseModel):
     """
     Named Test Case value that is of type CHAT_HISTORY
     """
@@ -31,5 +27,5 @@ class NamedTestCaseChatHistoryVariableValue(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_chat_history_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 .chat_message_request import ChatMessageRequest
 
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
 
-
-class NamedTestCaseChatHistoryVariableValueRequest(pydantic.BaseModel):
+class NamedTestCaseChatHistoryVariableValueRequest(pydantic_v1.BaseModel):
     """
     Named Test Case value that is of type CHAT_HISTORY
     """
@@ -31,5 +27,5 @@ class NamedTestCaseChatHistoryVariableValueRequest(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_error_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 .vellum_error import VellumError
 
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
 
-
-class NamedTestCaseErrorVariableValue(pydantic.BaseModel):
+class NamedTestCaseErrorVariableValue(pydantic_v1.BaseModel):
     """
     Named Test Case value that is of type ERROR
     """
@@ -31,5 +27,5 @@ class NamedTestCaseErrorVariableValue(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_error_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 .vellum_error_request import VellumErrorRequest
 
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
 
-
-class NamedTestCaseErrorVariableValueRequest(pydantic.BaseModel):
+class NamedTestCaseErrorVariableValueRequest(pydantic_v1.BaseModel):
     """
     Named Test Case value that is of type ERROR
     """
@@ -31,5 +27,5 @@ class NamedTestCaseErrorVariableValueRequest(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/{register_prompt_prompt.py → named_test_case_function_call_variable_value.py}

@@ -4,23 +4,17 @@ import datetime as dt
 import typing
 
 from ..core.datetime_utils import serialize_datetime
+from ..core.pydantic_utilities import pydantic_v1
+from .fulfilled_function_call import FulfilledFunctionCall
 
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
 
-
-class RegisterPromptPrompt(pydantic.BaseModel):
-    id: str = pydantic.Field()
+class NamedTestCaseFunctionCallVariableValue(pydantic_v1.BaseModel):
     """
-    The ID of the generated prompt.
+    Named Test Case value that is of type FUNCTION_CALL
     """
 
-    label: str = pydantic.Field()
-    """
-    A human-friendly label for the generated prompt.
-    """
+    value: typing.Optional[FulfilledFunctionCall] = None
+    name: str
 
     def json(self, **kwargs: typing.Any) -> str:
         kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
@@ -33,5 +27,5 @@ class RegisterPromptPrompt(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_function_call_variable_value_request.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 .fulfilled_function_call_request import FulfilledFunctionCallRequest
+
+
+class NamedTestCaseFunctionCallVariableValueRequest(pydantic_v1.BaseModel):
+    """
+    Named Test Case value that is of type FUNCTION_CALL
+    """
+
+    value: typing.Optional[FulfilledFunctionCallRequest] = 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_json_variable_value.py

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

@@ -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 NamedTestCaseSearchResultsVariableValue(pydantic.BaseModel):
+class NamedTestCaseSearchResultsVariableValue(pydantic_v1.BaseModel):
     """
     Named Test Case value that is of type SEARCH_RESULTS
     """
@@ -31,5 +27,5 @@ class NamedTestCaseSearchResultsVariableValue(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_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

@@ -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 NamedTestCaseStringVariableValue(pydantic.BaseModel):
+class NamedTestCaseStringVariableValue(pydantic_v1.BaseModel):
     """
     Named Test Case value that is of type STRING
     """
@@ -30,5 +26,5 @@ class NamedTestCaseStringVariableValue(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_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

@@ -6,6 +6,7 @@ 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
@@ -72,6 +73,16 @@ class NamedTestCaseVariableValue_Error(NamedTestCaseErrorVariableValue):
         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,
@@ -79,4 +90,5 @@ NamedTestCaseVariableValue = typing.Union[
     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}