vellum-ai 0.14.63__py3-none-any.whl → 1.13.5__py3-none-any.whl

vellum/client/types/ml_model_usage.py

@@ -1,9 +1,9 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from ..core.pydantic_utilities import UniversalBaseModel
 import typing
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
+
 import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
 
 
 class MlModelUsage(UniversalBaseModel):

vellum/client/types/ml_model_usage_wrapper.py

@@ -1,10 +1,10 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from ..core.pydantic_utilities import UniversalBaseModel
-from .ml_model_usage import MlModelUsage
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
 import typing
+
 import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+from .ml_model_usage import MlModelUsage
 
 
 class MlModelUsageWrapper(UniversalBaseModel):

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_chat_history_variable_value_request.py

@@ -1,10 +1,10 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from ..core.pydantic_utilities import UniversalBaseModel
 import typing
-from .chat_message_request import ChatMessageRequest
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
+
 import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+from .chat_message_request import ChatMessageRequest
 
 
 class NamedScenarioInputChatHistoryVariableValueRequest(UniversalBaseModel):

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_json_variable_value_request.py

@@ -1,9 +1,9 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from ..core.pydantic_utilities import UniversalBaseModel
 import typing
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
+
 import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
 
 
 class NamedScenarioInputJsonVariableValueRequest(UniversalBaseModel):

vellum/client/types/named_scenario_input_request.py

@@ -1,12 +1,21 @@
 # This file was auto-generated by Fern from our API Definition.
 
 import typing
-from .named_scenario_input_string_variable_value_request import NamedScenarioInputStringVariableValueRequest
-from .named_scenario_input_json_variable_value_request import NamedScenarioInputJsonVariableValueRequest
+
+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_string_variable_value_request.py

@@ -1,9 +1,9 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from ..core.pydantic_utilities import UniversalBaseModel
 import typing
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
+
 import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
 
 
 class NamedScenarioInputStringVariableValueRequest(UniversalBaseModel):

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_array_variable_value.py

@@ -1,11 +1,11 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from ..core.pydantic_utilities import UniversalBaseModel
-from .array_vellum_value import ArrayVellumValue
+from __future__ import annotations
+
 import typing
-from .vellum_value import VellumValue
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
+
 import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel, update_forward_refs
 
 
 class NamedTestCaseArrayVariableValue(UniversalBaseModel):
@@ -14,7 +14,7 @@ class NamedTestCaseArrayVariableValue(UniversalBaseModel):
     """
 
     type: typing.Literal["ARRAY"] = "ARRAY"
-    value: typing.Optional[typing.List[VellumValue]] = None
+    value: typing.Optional[typing.List["VellumValue"]] = None
     name: str
 
     if IS_PYDANTIC_V2:
@@ -25,3 +25,9 @@ class NamedTestCaseArrayVariableValue(UniversalBaseModel):
             frozen = True
             smart_union = True
             extra = pydantic.Extra.allow
+
+
+from .array_vellum_value import ArrayVellumValue  # noqa: E402, F401, I001
+from .vellum_value import VellumValue  # noqa: E402, F401, I001
+
+update_forward_refs(NamedTestCaseArrayVariableValue)

vellum/client/types/named_test_case_array_variable_value_request.py

@@ -1,11 +1,11 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from ..core.pydantic_utilities import UniversalBaseModel
-from .array_vellum_value_request import ArrayVellumValueRequest
+from __future__ import annotations
+
 import typing
-from .vellum_value_request import VellumValueRequest
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
+
 import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel, update_forward_refs
 
 
 class NamedTestCaseArrayVariableValueRequest(UniversalBaseModel):
@@ -14,7 +14,7 @@ class NamedTestCaseArrayVariableValueRequest(UniversalBaseModel):
     """
 
     type: typing.Literal["ARRAY"] = "ARRAY"
-    value: typing.Optional[typing.List[VellumValueRequest]] = None
+    value: typing.Optional[typing.List["VellumValueRequest"]] = None
     name: str
 
     if IS_PYDANTIC_V2:
@@ -25,3 +25,9 @@ class NamedTestCaseArrayVariableValueRequest(UniversalBaseModel):
             frozen = True
             smart_union = True
             extra = pydantic.Extra.allow
+
+
+from .array_vellum_value_request import ArrayVellumValueRequest  # noqa: E402, F401, I001
+from .vellum_value_request import VellumValueRequest  # noqa: E402, F401, I001
+
+update_forward_refs(NamedTestCaseArrayVariableValueRequest)

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: typing.Optional[VellumAudio] = None
+    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: typing.Optional[VellumAudioRequest] = None
+    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_chat_history_variable_value.py

@@ -1,10 +1,10 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from ..core.pydantic_utilities import UniversalBaseModel
 import typing
-from .chat_message import ChatMessage
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
+
 import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+from .chat_message import ChatMessage
 
 
 class NamedTestCaseChatHistoryVariableValue(UniversalBaseModel):

vellum/client/types/named_test_case_chat_history_variable_value_request.py

@@ -1,10 +1,10 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from ..core.pydantic_utilities import UniversalBaseModel
 import typing
-from .chat_message_request import ChatMessageRequest
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
+
 import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+from .chat_message_request import ChatMessageRequest
 
 
 class NamedTestCaseChatHistoryVariableValueRequest(UniversalBaseModel):

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: typing.Optional[VellumDocument] = None
+    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: typing.Optional[VellumDocumentRequest] = None
+    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_error_variable_value.py

@@ -1,10 +1,10 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from ..core.pydantic_utilities import UniversalBaseModel
 import typing
-from .vellum_error import VellumError
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
+
 import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+from .vellum_error import VellumError
 
 
 class NamedTestCaseErrorVariableValue(UniversalBaseModel):

vellum/client/types/named_test_case_error_variable_value_request.py

@@ -1,10 +1,10 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from ..core.pydantic_utilities import UniversalBaseModel
 import typing
-from .vellum_error_request import VellumErrorRequest
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
+
 import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+from .vellum_error_request import VellumErrorRequest
 
 
 class NamedTestCaseErrorVariableValueRequest(UniversalBaseModel):

vellum/client/types/named_test_case_function_call_variable_value.py

@@ -1,10 +1,10 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from ..core.pydantic_utilities import UniversalBaseModel
 import typing
-from .function_call import FunctionCall
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
+
 import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+from .function_call import FunctionCall
 
 
 class NamedTestCaseFunctionCallVariableValue(UniversalBaseModel):

vellum/client/types/named_test_case_function_call_variable_value_request.py

@@ -1,10 +1,10 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from ..core.pydantic_utilities import UniversalBaseModel
 import typing
-from .function_call_request import FunctionCallRequest
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
+
 import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+from .function_call_request import FunctionCallRequest
 
 
 class NamedTestCaseFunctionCallVariableValueRequest(UniversalBaseModel):

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: typing.Optional[VellumImage] = None
+    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: typing.Optional[VellumImageRequest] = None
+    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_json_variable_value.py

@@ -1,9 +1,9 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from ..core.pydantic_utilities import UniversalBaseModel
 import typing
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
+
 import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
 
 
 class NamedTestCaseJsonVariableValue(UniversalBaseModel):

vellum/client/types/named_test_case_json_variable_value_request.py

@@ -1,9 +1,9 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from ..core.pydantic_utilities import UniversalBaseModel
 import typing
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
+
 import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
 
 
 class NamedTestCaseJsonVariableValueRequest(UniversalBaseModel):

vellum/client/types/named_test_case_number_variable_value.py

@@ -1,9 +1,9 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from ..core.pydantic_utilities import UniversalBaseModel
 import typing
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
+
 import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
 
 
 class NamedTestCaseNumberVariableValue(UniversalBaseModel):

vellum/client/types/named_test_case_number_variable_value_request.py

@@ -1,9 +1,9 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from ..core.pydantic_utilities import UniversalBaseModel
 import typing
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
+
 import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
 
 
 class NamedTestCaseNumberVariableValueRequest(UniversalBaseModel):

vellum/client/types/named_test_case_search_results_variable_value.py

@@ -1,10 +1,10 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from ..core.pydantic_utilities import UniversalBaseModel
 import typing
-from .search_result import SearchResult
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
+
 import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+from .search_result import SearchResult
 
 
 class NamedTestCaseSearchResultsVariableValue(UniversalBaseModel):

vellum/client/types/named_test_case_search_results_variable_value_request.py

@@ -1,10 +1,10 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from ..core.pydantic_utilities import UniversalBaseModel
 import typing
-from .search_result_request import SearchResultRequest
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
+
 import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+from .search_result_request import SearchResultRequest
 
 
 class NamedTestCaseSearchResultsVariableValueRequest(UniversalBaseModel):

vellum/client/types/named_test_case_string_variable_value.py

@@ -1,9 +1,9 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from ..core.pydantic_utilities import UniversalBaseModel
 import typing
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
+
 import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
 
 
 class NamedTestCaseStringVariableValue(UniversalBaseModel):

vellum/client/types/named_test_case_string_variable_value_request.py

@@ -1,9 +1,9 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from ..core.pydantic_utilities import UniversalBaseModel
 import typing
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
+
 import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
 
 
 class NamedTestCaseStringVariableValueRequest(UniversalBaseModel):

vellum/client/types/named_test_case_variable_value.py

@@ -1,14 +1,19 @@
 # This file was auto-generated by Fern from our API Definition.
 
 import typing
-from .named_test_case_string_variable_value import NamedTestCaseStringVariableValue
-from .named_test_case_number_variable_value import NamedTestCaseNumberVariableValue
-from .named_test_case_json_variable_value import NamedTestCaseJsonVariableValue
+
+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_search_results_variable_value import NamedTestCaseSearchResultsVariableValue
+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_array_variable_value import NamedTestCaseArrayVariableValue
+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,
@@ -19,4 +24,8 @@ NamedTestCaseVariableValue = typing.Union[
     NamedTestCaseErrorVariableValue,
     NamedTestCaseFunctionCallVariableValue,
     NamedTestCaseArrayVariableValue,
+    NamedTestCaseAudioVariableValue,
+    NamedTestCaseVideoVariableValue,
+    NamedTestCaseImageVariableValue,
+    NamedTestCaseDocumentVariableValue,
 ]