vellum-ai 1.2.3__py3-none-any.whl → 1.2.5__py3-none-any.whl

vellum/client/types/named_scenario_input_audio_variable_value_request.py

@@ -0,0 +1,22 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+from .vellum_audio_request import VellumAudioRequest
+
+
+class NamedScenarioInputAudioVariableValueRequest(UniversalBaseModel):
+    type: typing.Literal["AUDIO"] = "AUDIO"
+    value: VellumAudioRequest
+    name: str
+
+    if IS_PYDANTIC_V2:
+        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
+    else:
+
+        class Config:
+            frozen = True
+            smart_union = True
+            extra = pydantic.Extra.allow

vellum/client/types/named_scenario_input_document_variable_value_request.py

@@ -0,0 +1,22 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+from .vellum_document_request import VellumDocumentRequest
+
+
+class NamedScenarioInputDocumentVariableValueRequest(UniversalBaseModel):
+    type: typing.Literal["DOCUMENT"] = "DOCUMENT"
+    value: VellumDocumentRequest
+    name: str
+
+    if IS_PYDANTIC_V2:
+        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
+    else:
+
+        class Config:
+            frozen = True
+            smart_union = True
+            extra = pydantic.Extra.allow

vellum/client/types/named_scenario_input_image_variable_value_request.py

@@ -0,0 +1,22 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+from .vellum_image_request import VellumImageRequest
+
+
+class NamedScenarioInputImageVariableValueRequest(UniversalBaseModel):
+    type: typing.Literal["IMAGE"] = "IMAGE"
+    value: VellumImageRequest
+    name: str
+
+    if IS_PYDANTIC_V2:
+        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
+    else:
+
+        class Config:
+            frozen = True
+            smart_union = True
+            extra = pydantic.Extra.allow

vellum/client/types/named_scenario_input_request.py

@@ -2,12 +2,20 @@
 
 import typing
 
+from .named_scenario_input_audio_variable_value_request import NamedScenarioInputAudioVariableValueRequest
 from .named_scenario_input_chat_history_variable_value_request import NamedScenarioInputChatHistoryVariableValueRequest
+from .named_scenario_input_document_variable_value_request import NamedScenarioInputDocumentVariableValueRequest
+from .named_scenario_input_image_variable_value_request import NamedScenarioInputImageVariableValueRequest
 from .named_scenario_input_json_variable_value_request import NamedScenarioInputJsonVariableValueRequest
 from .named_scenario_input_string_variable_value_request import NamedScenarioInputStringVariableValueRequest
+from .named_scenario_input_video_variable_value_request import NamedScenarioInputVideoVariableValueRequest
 
 NamedScenarioInputRequest = typing.Union[
     NamedScenarioInputStringVariableValueRequest,
     NamedScenarioInputJsonVariableValueRequest,
     NamedScenarioInputChatHistoryVariableValueRequest,
+    NamedScenarioInputAudioVariableValueRequest,
+    NamedScenarioInputVideoVariableValueRequest,
+    NamedScenarioInputImageVariableValueRequest,
+    NamedScenarioInputDocumentVariableValueRequest,
 ]

vellum/client/types/named_scenario_input_video_variable_value_request.py

@@ -0,0 +1,22 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+from .vellum_video_request import VellumVideoRequest
+
+
+class NamedScenarioInputVideoVariableValueRequest(UniversalBaseModel):
+    type: typing.Literal["VIDEO"] = "VIDEO"
+    value: VellumVideoRequest
+    name: str
+
+    if IS_PYDANTIC_V2:
+        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
+    else:
+
+        class Config:
+            frozen = True
+            smart_union = True
+            extra = pydantic.Extra.allow

vellum/client/types/named_test_case_audio_variable_value.py

@@ -0,0 +1,26 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+from .vellum_audio import VellumAudio
+
+
+class NamedTestCaseAudioVariableValue(UniversalBaseModel):
+    """
+    Named Test Case value that is of type AUDIO
+    """
+
+    type: typing.Literal["AUDIO"] = "AUDIO"
+    value: VellumAudio
+    name: str
+
+    if IS_PYDANTIC_V2:
+        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
+    else:
+
+        class Config:
+            frozen = True
+            smart_union = True
+            extra = pydantic.Extra.allow

vellum/client/types/named_test_case_audio_variable_value_request.py

@@ -0,0 +1,26 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+from .vellum_audio_request import VellumAudioRequest
+
+
+class NamedTestCaseAudioVariableValueRequest(UniversalBaseModel):
+    """
+    Named Test Case value that is of type AUDIO
+    """
+
+    type: typing.Literal["AUDIO"] = "AUDIO"
+    value: VellumAudioRequest
+    name: str
+
+    if IS_PYDANTIC_V2:
+        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
+    else:
+
+        class Config:
+            frozen = True
+            smart_union = True
+            extra = pydantic.Extra.allow

vellum/client/types/named_test_case_document_variable_value.py

@@ -0,0 +1,22 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+from .vellum_document import VellumDocument
+
+
+class NamedTestCaseDocumentVariableValue(UniversalBaseModel):
+    type: typing.Literal["DOCUMENT"] = "DOCUMENT"
+    value: VellumDocument
+    name: str
+
+    if IS_PYDANTIC_V2:
+        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
+    else:
+
+        class Config:
+            frozen = True
+            smart_union = True
+            extra = pydantic.Extra.allow

vellum/client/types/named_test_case_document_variable_value_request.py

@@ -0,0 +1,22 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+from .vellum_document_request import VellumDocumentRequest
+
+
+class NamedTestCaseDocumentVariableValueRequest(UniversalBaseModel):
+    type: typing.Literal["DOCUMENT"] = "DOCUMENT"
+    value: VellumDocumentRequest
+    name: str
+
+    if IS_PYDANTIC_V2:
+        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
+    else:
+
+        class Config:
+            frozen = True
+            smart_union = True
+            extra = pydantic.Extra.allow

vellum/client/types/named_test_case_image_variable_value.py

@@ -0,0 +1,22 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+from .vellum_image import VellumImage
+
+
+class NamedTestCaseImageVariableValue(UniversalBaseModel):
+    type: typing.Literal["IMAGE"] = "IMAGE"
+    value: VellumImage
+    name: str
+
+    if IS_PYDANTIC_V2:
+        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
+    else:
+
+        class Config:
+            frozen = True
+            smart_union = True
+            extra = pydantic.Extra.allow

vellum/client/types/named_test_case_image_variable_value_request.py

@@ -0,0 +1,22 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+from .vellum_image_request import VellumImageRequest
+
+
+class NamedTestCaseImageVariableValueRequest(UniversalBaseModel):
+    type: typing.Literal["IMAGE"] = "IMAGE"
+    value: VellumImageRequest
+    name: str
+
+    if IS_PYDANTIC_V2:
+        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
+    else:
+
+        class Config:
+            frozen = True
+            smart_union = True
+            extra = pydantic.Extra.allow

vellum/client/types/named_test_case_variable_value.py

@@ -3,13 +3,17 @@
 import typing
 
 from .named_test_case_array_variable_value import NamedTestCaseArrayVariableValue
+from .named_test_case_audio_variable_value import NamedTestCaseAudioVariableValue
 from .named_test_case_chat_history_variable_value import NamedTestCaseChatHistoryVariableValue
+from .named_test_case_document_variable_value import NamedTestCaseDocumentVariableValue
 from .named_test_case_error_variable_value import NamedTestCaseErrorVariableValue
 from .named_test_case_function_call_variable_value import NamedTestCaseFunctionCallVariableValue
+from .named_test_case_image_variable_value import NamedTestCaseImageVariableValue
 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
+from .named_test_case_video_variable_value import NamedTestCaseVideoVariableValue
 
 NamedTestCaseVariableValue = typing.Union[
     NamedTestCaseStringVariableValue,
@@ -20,4 +24,8 @@ NamedTestCaseVariableValue = typing.Union[
     NamedTestCaseErrorVariableValue,
     NamedTestCaseFunctionCallVariableValue,
     NamedTestCaseArrayVariableValue,
+    NamedTestCaseAudioVariableValue,
+    NamedTestCaseVideoVariableValue,
+    NamedTestCaseImageVariableValue,
+    NamedTestCaseDocumentVariableValue,
 ]

vellum/client/types/named_test_case_variable_value_request.py

@@ -3,13 +3,17 @@
 import typing
 
 from .named_test_case_array_variable_value_request import NamedTestCaseArrayVariableValueRequest
+from .named_test_case_audio_variable_value_request import NamedTestCaseAudioVariableValueRequest
 from .named_test_case_chat_history_variable_value_request import NamedTestCaseChatHistoryVariableValueRequest
+from .named_test_case_document_variable_value_request import NamedTestCaseDocumentVariableValueRequest
 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_image_variable_value_request import NamedTestCaseImageVariableValueRequest
 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
 from .named_test_case_string_variable_value_request import NamedTestCaseStringVariableValueRequest
+from .named_test_case_video_variable_value_request import NamedTestCaseVideoVariableValueRequest
 
 NamedTestCaseVariableValueRequest = typing.Union[
     NamedTestCaseStringVariableValueRequest,
@@ -20,4 +24,8 @@ NamedTestCaseVariableValueRequest = typing.Union[
     NamedTestCaseErrorVariableValueRequest,
     NamedTestCaseFunctionCallVariableValueRequest,
     NamedTestCaseArrayVariableValueRequest,
+    NamedTestCaseAudioVariableValueRequest,
+    NamedTestCaseVideoVariableValueRequest,
+    NamedTestCaseImageVariableValueRequest,
+    NamedTestCaseDocumentVariableValueRequest,
 ]

vellum/client/types/named_test_case_video_variable_value.py

@@ -0,0 +1,22 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+from .vellum_video import VellumVideo
+
+
+class NamedTestCaseVideoVariableValue(UniversalBaseModel):
+    type: typing.Literal["VIDEO"] = "VIDEO"
+    value: VellumVideo
+    name: str
+
+    if IS_PYDANTIC_V2:
+        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
+    else:
+
+        class Config:
+            frozen = True
+            smart_union = True
+            extra = pydantic.Extra.allow

vellum/client/types/named_test_case_video_variable_value_request.py

@@ -0,0 +1,22 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+from .vellum_video_request import VellumVideoRequest
+
+
+class NamedTestCaseVideoVariableValueRequest(UniversalBaseModel):
+    type: typing.Literal["VIDEO"] = "VIDEO"
+    value: VellumVideoRequest
+    name: str
+
+    if IS_PYDANTIC_V2:
+        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
+    else:
+
+        class Config:
+            frozen = True
+            smart_union = True
+            extra = pydantic.Extra.allow

vellum/client/types/node_execution_span_attributes.py

@@ -8,6 +8,7 @@ from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
 
 class NodeExecutionSpanAttributes(UniversalBaseModel):
     label: str
+    filepath: typing.Optional[str] = None
     node_id: str
 
     if IS_PYDANTIC_V2:

vellum/client/types/scenario_input.py

@@ -2,10 +2,20 @@
 
 import typing
 
+from .scenario_input_audio_variable_value import ScenarioInputAudioVariableValue
 from .scenario_input_chat_history_variable_value import ScenarioInputChatHistoryVariableValue
+from .scenario_input_document_variable_value import ScenarioInputDocumentVariableValue
+from .scenario_input_image_variable_value import ScenarioInputImageVariableValue
 from .scenario_input_json_variable_value import ScenarioInputJsonVariableValue
 from .scenario_input_string_variable_value import ScenarioInputStringVariableValue
+from .scenario_input_video_variable_value import ScenarioInputVideoVariableValue
 
 ScenarioInput = typing.Union[
-    ScenarioInputStringVariableValue, ScenarioInputJsonVariableValue, ScenarioInputChatHistoryVariableValue
+    ScenarioInputStringVariableValue,
+    ScenarioInputJsonVariableValue,
+    ScenarioInputChatHistoryVariableValue,
+    ScenarioInputAudioVariableValue,
+    ScenarioInputVideoVariableValue,
+    ScenarioInputImageVariableValue,
+    ScenarioInputDocumentVariableValue,
 ]

vellum/client/types/scenario_input_audio_variable_value.py

@@ -0,0 +1,22 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+from .vellum_audio import VellumAudio
+
+
+class ScenarioInputAudioVariableValue(UniversalBaseModel):
+    type: typing.Literal["AUDIO"] = "AUDIO"
+    value: VellumAudio
+    input_variable_id: str
+
+    if IS_PYDANTIC_V2:
+        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
+    else:
+
+        class Config:
+            frozen = True
+            smart_union = True
+            extra = pydantic.Extra.allow

vellum/client/types/scenario_input_document_variable_value.py

@@ -0,0 +1,22 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+from .vellum_document import VellumDocument
+
+
+class ScenarioInputDocumentVariableValue(UniversalBaseModel):
+    type: typing.Literal["DOCUMENT"] = "DOCUMENT"
+    value: VellumDocument
+    input_variable_id: str
+
+    if IS_PYDANTIC_V2:
+        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
+    else:
+
+        class Config:
+            frozen = True
+            smart_union = True
+            extra = pydantic.Extra.allow

vellum/client/types/scenario_input_image_variable_value.py

@@ -0,0 +1,22 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+from .vellum_image import VellumImage
+
+
+class ScenarioInputImageVariableValue(UniversalBaseModel):
+    type: typing.Literal["IMAGE"] = "IMAGE"
+    value: VellumImage
+    input_variable_id: str
+
+    if IS_PYDANTIC_V2:
+        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
+    else:
+
+        class Config:
+            frozen = True
+            smart_union = True
+            extra = pydantic.Extra.allow

vellum/client/types/scenario_input_video_variable_value.py

@@ -0,0 +1,22 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+from .vellum_video import VellumVideo
+
+
+class ScenarioInputVideoVariableValue(UniversalBaseModel):
+    type: typing.Literal["VIDEO"] = "VIDEO"
+    value: VellumVideo
+    input_variable_id: str
+
+    if IS_PYDANTIC_V2:
+        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
+    else:
+
+        class Config:
+            frozen = True
+            smart_union = True
+            extra = pydantic.Extra.allow

vellum/client/types/span_link.py

@@ -11,7 +11,7 @@ from .span_link_type_enum import SpanLinkTypeEnum
 
 class SpanLink(UniversalBaseModel):
     trace_id: str
-    type: SpanLinkTypeEnum = "TRIGGERED_BY"
+    type: SpanLinkTypeEnum
     span_context: "ParentContext"
 
     if IS_PYDANTIC_V2:

vellum/client/types/span_link_type_enum.py

@@ -2,4 +2,4 @@
 
 import typing
 
-SpanLinkTypeEnum = typing.Literal["TRIGGERED_BY"]
+SpanLinkTypeEnum = typing.Union[typing.Literal["TRIGGERED_BY", "PREVIOUS_SPAN", "ROOT_SPAN"], typing.Any]

vellum/client/types/test_case_audio_variable_value.py

@@ -0,0 +1,27 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+from .vellum_audio import VellumAudio
+
+
+class TestCaseAudioVariableValue(UniversalBaseModel):
+    """
+    An audio value for a variable in a Test Case.
+    """
+
+    variable_id: str
+    name: typing.Optional[str] = None
+    type: typing.Literal["AUDIO"] = "AUDIO"
+    value: VellumAudio
+
+    if IS_PYDANTIC_V2:
+        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
+    else:
+
+        class Config:
+            frozen = True
+            smart_union = True
+            extra = pydantic.Extra.allow

vellum/client/types/test_case_document_variable_value.py

@@ -0,0 +1,27 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+from .vellum_document import VellumDocument
+
+
+class TestCaseDocumentVariableValue(UniversalBaseModel):
+    """
+    A document value for a variable in a Test Case.
+    """
+
+    variable_id: str
+    name: typing.Optional[str] = None
+    type: typing.Literal["DOCUMENT"] = "DOCUMENT"
+    value: VellumDocument
+
+    if IS_PYDANTIC_V2:
+        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
+    else:
+
+        class Config:
+            frozen = True
+            smart_union = True
+            extra = pydantic.Extra.allow

vellum/client/types/test_case_image_variable_value.py

@@ -0,0 +1,27 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+from .vellum_image import VellumImage
+
+
+class TestCaseImageVariableValue(UniversalBaseModel):
+    """
+    An image value for a variable in a Test Case.
+    """
+
+    variable_id: str
+    name: typing.Optional[str] = None
+    type: typing.Literal["IMAGE"] = "IMAGE"
+    value: VellumImage
+
+    if IS_PYDANTIC_V2:
+        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
+    else:
+
+        class Config:
+            frozen = True
+            smart_union = True
+            extra = pydantic.Extra.allow

vellum/client/types/test_case_variable_value.py

@@ -3,13 +3,17 @@
 import typing
 
 from .test_case_array_variable_value import TestCaseArrayVariableValue
+from .test_case_audio_variable_value import TestCaseAudioVariableValue
 from .test_case_chat_history_variable_value import TestCaseChatHistoryVariableValue
+from .test_case_document_variable_value import TestCaseDocumentVariableValue
 from .test_case_error_variable_value import TestCaseErrorVariableValue
 from .test_case_function_call_variable_value import TestCaseFunctionCallVariableValue
+from .test_case_image_variable_value import TestCaseImageVariableValue
 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
 from .test_case_string_variable_value import TestCaseStringVariableValue
+from .test_case_video_variable_value import TestCaseVideoVariableValue
 
 TestCaseVariableValue = typing.Union[
     TestCaseStringVariableValue,
@@ -20,4 +24,8 @@ TestCaseVariableValue = typing.Union[
     TestCaseErrorVariableValue,
     TestCaseFunctionCallVariableValue,
     TestCaseArrayVariableValue,
+    TestCaseAudioVariableValue,
+    TestCaseImageVariableValue,
+    TestCaseVideoVariableValue,
+    TestCaseDocumentVariableValue,
 ]

vellum/client/types/test_case_video_variable_value.py

@@ -0,0 +1,27 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+from .vellum_video import VellumVideo
+
+
+class TestCaseVideoVariableValue(UniversalBaseModel):
+    """
+    A video value for a variable in a Test Case.
+    """
+
+    variable_id: str
+    name: typing.Optional[str] = None
+    type: typing.Literal["VIDEO"] = "VIDEO"
+    value: VellumVideo
+
+    if IS_PYDANTIC_V2:
+        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
+    else:
+
+        class Config:
+            frozen = True
+            smart_union = True
+            extra = pydantic.Extra.allow