vellum-ai 1.1.2__py3-none-any.whl → 1.1.3__py3-none-any.whl

vellum/client/resources/workflows/raw_client.py

@@ -9,6 +9,7 @@ from ...errors.bad_request_error import BadRequestError
 from ...core.pydantic_utilities import parse_obj_as
 from json.decoder import JSONDecodeError
 from ...core.api_error import ApiError
+import contextlib
 from ...types.workflow_push_exec_config import WorkflowPushExecConfig
 from ...types.workflow_push_deployment_config_request import WorkflowPushDeploymentConfigRequest
 from ... import core
@@ -24,6 +25,7 @@ class RawWorkflowsClient:
     def __init__(self, *, client_wrapper: SyncClientWrapper):
         self._client_wrapper = client_wrapper
 
+    @contextlib.contextmanager
     def pull(
         self,
         id: str,
@@ -34,7 +36,7 @@ class RawWorkflowsClient:
         include_sandbox: typing.Optional[bool] = None,
         strict: typing.Optional[bool] = None,
         request_options: typing.Optional[RequestOptions] = None,
-    ) -> HttpResponse[typing.Iterator[bytes]]:
+    ) -> typing.Iterator[HttpResponse[typing.Iterator[bytes]]]:
         """
         Parameters
         ----------
@@ -56,7 +58,7 @@ class RawWorkflowsClient:
 
         Returns
         -------
-        HttpResponse[typing.Iterator[bytes]]
+        typing.Iterator[HttpResponse[typing.Iterator[bytes]]]
 
         """
         with self._client_wrapper.httpx_client.stream(
@@ -72,27 +74,31 @@ class RawWorkflowsClient:
             },
             request_options=request_options,
         ) as _response:
-            try:
-                if 200 <= _response.status_code < 300:
-                    _chunk_size = request_options.get("chunk_size", None) if request_options is not None else None
-                    return HttpResponse(
-                        response=_response, data=(_chunk for _chunk in _response.iter_bytes(chunk_size=_chunk_size))
-                    )
-                _response.read()
-                if _response.status_code == 400:
-                    raise BadRequestError(
-                        typing.cast(
-                            typing.Optional[typing.Any],
-                            parse_obj_as(
-                                type_=typing.Optional[typing.Any],  # type: ignore
-                                object_=_response.json(),
-                            ),
+
+            def stream() -> HttpResponse[typing.Iterator[bytes]]:
+                try:
+                    if 200 <= _response.status_code < 300:
+                        _chunk_size = request_options.get("chunk_size", None) if request_options is not None else None
+                        return HttpResponse(
+                            response=_response, data=(_chunk for _chunk in _response.iter_bytes(chunk_size=_chunk_size))
+                        )
+                    _response.read()
+                    if _response.status_code == 400:
+                        raise BadRequestError(
+                            typing.cast(
+                                typing.Optional[typing.Any],
+                                parse_obj_as(
+                                    type_=typing.Optional[typing.Any],  # type: ignore
+                                    object_=_response.json(),
+                                ),
+                            )
                         )
-                    )
-                _response_json = _response.json()
-            except JSONDecodeError:
-                raise ApiError(status_code=_response.status_code, body=_response.text)
-            raise ApiError(status_code=_response.status_code, body=_response_json)
+                    _response_json = _response.json()
+                except JSONDecodeError:
+                    raise ApiError(status_code=_response.status_code, body=_response.text)
+                raise ApiError(status_code=_response.status_code, body=_response_json)
+
+            yield stream()
 
     def push(
         self,
@@ -142,7 +148,7 @@ class RawWorkflowsClient:
                 "strict": strict,
             },
             files={
-                "artifact": artifact,
+                **({"artifact": artifact} if artifact is not None else {}),
             },
             request_options=request_options,
             omit=OMIT,
@@ -167,6 +173,7 @@ class AsyncRawWorkflowsClient:
     def __init__(self, *, client_wrapper: AsyncClientWrapper):
         self._client_wrapper = client_wrapper
 
+    @contextlib.asynccontextmanager
     async def pull(
         self,
         id: str,
@@ -177,7 +184,7 @@ class AsyncRawWorkflowsClient:
         include_sandbox: typing.Optional[bool] = None,
         strict: typing.Optional[bool] = None,
         request_options: typing.Optional[RequestOptions] = None,
-    ) -> AsyncHttpResponse[typing.AsyncIterator[bytes]]:
+    ) -> typing.AsyncIterator[AsyncHttpResponse[typing.AsyncIterator[bytes]]]:
         """
         Parameters
         ----------
@@ -199,7 +206,7 @@ class AsyncRawWorkflowsClient:
 
         Returns
         -------
-        AsyncHttpResponse[typing.AsyncIterator[bytes]]
+        typing.AsyncIterator[AsyncHttpResponse[typing.AsyncIterator[bytes]]]
 
         """
         async with self._client_wrapper.httpx_client.stream(
@@ -215,28 +222,32 @@ class AsyncRawWorkflowsClient:
             },
             request_options=request_options,
         ) as _response:
-            try:
-                if 200 <= _response.status_code < 300:
-                    _chunk_size = request_options.get("chunk_size", None) if request_options is not None else None
-                    return AsyncHttpResponse(
-                        response=_response,
-                        data=(_chunk async for _chunk in _response.aiter_bytes(chunk_size=_chunk_size)),
-                    )
-                await _response.aread()
-                if _response.status_code == 400:
-                    raise BadRequestError(
-                        typing.cast(
-                            typing.Optional[typing.Any],
-                            parse_obj_as(
-                                type_=typing.Optional[typing.Any],  # type: ignore
-                                object_=_response.json(),
-                            ),
+
+            async def stream() -> AsyncHttpResponse[typing.AsyncIterator[bytes]]:
+                try:
+                    if 200 <= _response.status_code < 300:
+                        _chunk_size = request_options.get("chunk_size", None) if request_options is not None else None
+                        return AsyncHttpResponse(
+                            response=_response,
+                            data=(_chunk async for _chunk in _response.aiter_bytes(chunk_size=_chunk_size)),
+                        )
+                    await _response.aread()
+                    if _response.status_code == 400:
+                        raise BadRequestError(
+                            typing.cast(
+                                typing.Optional[typing.Any],
+                                parse_obj_as(
+                                    type_=typing.Optional[typing.Any],  # type: ignore
+                                    object_=_response.json(),
+                                ),
+                            )
                         )
-                    )
-                _response_json = _response.json()
-            except JSONDecodeError:
-                raise ApiError(status_code=_response.status_code, body=_response.text)
-            raise ApiError(status_code=_response.status_code, body=_response_json)
+                    _response_json = _response.json()
+                except JSONDecodeError:
+                    raise ApiError(status_code=_response.status_code, body=_response.text)
+                raise ApiError(status_code=_response.status_code, body=_response_json)
+
+            yield await stream()
 
     async def push(
         self,
@@ -286,7 +297,7 @@ class AsyncRawWorkflowsClient:
                 "strict": strict,
             },
             files={
-                "artifact": artifact,
+                **({"artifact": artifact} if artifact is not None else {}),
             },
             request_options=request_options,
             omit=OMIT,

vellum/client/resources/workspace_secrets/client.py

@@ -60,10 +60,7 @@ class WorkspaceSecretsClient:
             id="id",
         )
         """
-        response = self._raw_client.retrieve(
-            id,
-            request_options=request_options,
-        )
+        response = self._raw_client.retrieve(id, request_options=request_options)
         return response.data
 
     def partial_update(
@@ -106,12 +103,7 @@ class WorkspaceSecretsClient:
             id="id",
         )
         """
-        response = self._raw_client.partial_update(
-            id,
-            label=label,
-            value=value,
-            request_options=request_options,
-        )
+        response = self._raw_client.partial_update(id, label=label, value=value, request_options=request_options)
         return response.data
 
 
@@ -169,10 +161,7 @@ class AsyncWorkspaceSecretsClient:
 
         asyncio.run(main())
         """
-        response = await self._raw_client.retrieve(
-            id,
-            request_options=request_options,
-        )
+        response = await self._raw_client.retrieve(id, request_options=request_options)
         return response.data
 
     async def partial_update(
@@ -223,10 +212,5 @@ class AsyncWorkspaceSecretsClient:
 
         asyncio.run(main())
         """
-        response = await self._raw_client.partial_update(
-            id,
-            label=label,
-            value=value,
-            request_options=request_options,
-        )
+        response = await self._raw_client.partial_update(id, label=label, value=value, request_options=request_options)
         return response.data

vellum/client/resources/workspaces/client.py

@@ -52,9 +52,7 @@ class WorkspacesClient:
         )
         client.workspaces.workspace_identity()
         """
-        response = self._raw_client.workspace_identity(
-            request_options=request_options,
-        )
+        response = self._raw_client.workspace_identity(request_options=request_options)
         return response.data
 
 
@@ -105,7 +103,5 @@ class AsyncWorkspacesClient:
 
         asyncio.run(main())
         """
-        response = await self._raw_client.workspace_identity(
-            request_options=request_options,
-        )
+        response = await self._raw_client.workspace_identity(request_options=request_options)
         return response.data

vellum/client/types/__init__.py

@@ -549,7 +549,14 @@ from .vellum_value_request import VellumValueRequest
 from .vellum_variable import VellumVariable
 from .vellum_variable_extensions import VellumVariableExtensions
 from .vellum_variable_type import VellumVariableType
+from .vellum_video import VellumVideo
+from .vellum_video_request import VellumVideoRequest
 from .vellum_workflow_execution_event import VellumWorkflowExecutionEvent
+from .video_chat_message_content import VideoChatMessageContent
+from .video_chat_message_content_request import VideoChatMessageContentRequest
+from .video_prompt_block import VideoPromptBlock
+from .video_vellum_value import VideoVellumValue
+from .video_vellum_value_request import VideoVellumValueRequest
 from .workflow_deployment_event_executions_response import WorkflowDeploymentEventExecutionsResponse
 from .workflow_deployment_history_item import WorkflowDeploymentHistoryItem
 from .workflow_deployment_parent_context import WorkflowDeploymentParentContext
@@ -586,6 +593,7 @@ from .workflow_execution_streaming_body import WorkflowExecutionStreamingBody
 from .workflow_execution_streaming_event import WorkflowExecutionStreamingEvent
 from .workflow_execution_usage_calculation_error import WorkflowExecutionUsageCalculationError
 from .workflow_execution_usage_calculation_error_code_enum import WorkflowExecutionUsageCalculationErrorCodeEnum
+from .workflow_execution_usage_calculation_fulfilled_body import WorkflowExecutionUsageCalculationFulfilledBody
 from .workflow_execution_usage_result import WorkflowExecutionUsageResult
 from .workflow_execution_view_online_eval_metric_result import WorkflowExecutionViewOnlineEvalMetricResult
 from .workflow_execution_workflow_result_event import WorkflowExecutionWorkflowResultEvent
@@ -1153,7 +1161,14 @@ __all__ = [
     "VellumVariable",
     "VellumVariableExtensions",
     "VellumVariableType",
+    "VellumVideo",
+    "VellumVideoRequest",
     "VellumWorkflowExecutionEvent",
+    "VideoChatMessageContent",
+    "VideoChatMessageContentRequest",
+    "VideoPromptBlock",
+    "VideoVellumValue",
+    "VideoVellumValueRequest",
     "WorkflowDeploymentEventExecutionsResponse",
     "WorkflowDeploymentHistoryItem",
     "WorkflowDeploymentParentContext",
@@ -1190,6 +1205,7 @@ __all__ = [
     "WorkflowExecutionStreamingEvent",
     "WorkflowExecutionUsageCalculationError",
     "WorkflowExecutionUsageCalculationErrorCodeEnum",
+    "WorkflowExecutionUsageCalculationFulfilledBody",
     "WorkflowExecutionUsageResult",
     "WorkflowExecutionViewOnlineEvalMetricResult",
     "WorkflowExecutionWorkflowResultEvent",

vellum/client/types/array_chat_message_content_item.py

@@ -3,14 +3,16 @@
 import typing
 from .string_chat_message_content import StringChatMessageContent
 from .function_call_chat_message_content import FunctionCallChatMessageContent
-from .image_chat_message_content import ImageChatMessageContent
 from .audio_chat_message_content import AudioChatMessageContent
+from .video_chat_message_content import VideoChatMessageContent
+from .image_chat_message_content import ImageChatMessageContent
 from .document_chat_message_content import DocumentChatMessageContent
 
 ArrayChatMessageContentItem = typing.Union[
     StringChatMessageContent,
     FunctionCallChatMessageContent,
-    ImageChatMessageContent,
     AudioChatMessageContent,
+    VideoChatMessageContent,
+    ImageChatMessageContent,
     DocumentChatMessageContent,
 ]

vellum/client/types/array_chat_message_content_item_request.py

@@ -3,14 +3,16 @@
 import typing
 from .string_chat_message_content_request import StringChatMessageContentRequest
 from .function_call_chat_message_content_request import FunctionCallChatMessageContentRequest
-from .image_chat_message_content_request import ImageChatMessageContentRequest
 from .audio_chat_message_content_request import AudioChatMessageContentRequest
+from .video_chat_message_content_request import VideoChatMessageContentRequest
+from .image_chat_message_content_request import ImageChatMessageContentRequest
 from .document_chat_message_content_request import DocumentChatMessageContentRequest
 
 ArrayChatMessageContentItemRequest = typing.Union[
     StringChatMessageContentRequest,
     FunctionCallChatMessageContentRequest,
-    ImageChatMessageContentRequest,
     AudioChatMessageContentRequest,
+    VideoChatMessageContentRequest,
+    ImageChatMessageContentRequest,
     DocumentChatMessageContentRequest,
 ]

vellum/client/types/chat_message_content.py

@@ -4,15 +4,17 @@ import typing
 from .string_chat_message_content import StringChatMessageContent
 from .function_call_chat_message_content import FunctionCallChatMessageContent
 from .array_chat_message_content import ArrayChatMessageContent
-from .image_chat_message_content import ImageChatMessageContent
 from .audio_chat_message_content import AudioChatMessageContent
+from .video_chat_message_content import VideoChatMessageContent
+from .image_chat_message_content import ImageChatMessageContent
 from .document_chat_message_content import DocumentChatMessageContent
 
 ChatMessageContent = typing.Union[
     StringChatMessageContent,
     FunctionCallChatMessageContent,
     ArrayChatMessageContent,
-    ImageChatMessageContent,
     AudioChatMessageContent,
+    VideoChatMessageContent,
+    ImageChatMessageContent,
     DocumentChatMessageContent,
 ]

vellum/client/types/chat_message_content_request.py

@@ -4,15 +4,17 @@ import typing
 from .string_chat_message_content_request import StringChatMessageContentRequest
 from .function_call_chat_message_content_request import FunctionCallChatMessageContentRequest
 from .array_chat_message_content_request import ArrayChatMessageContentRequest
-from .image_chat_message_content_request import ImageChatMessageContentRequest
 from .audio_chat_message_content_request import AudioChatMessageContentRequest
+from .video_chat_message_content_request import VideoChatMessageContentRequest
+from .image_chat_message_content_request import ImageChatMessageContentRequest
 from .document_chat_message_content_request import DocumentChatMessageContentRequest
 
 ChatMessageContentRequest = typing.Union[
     StringChatMessageContentRequest,
     FunctionCallChatMessageContentRequest,
     ArrayChatMessageContentRequest,
-    ImageChatMessageContentRequest,
     AudioChatMessageContentRequest,
+    VideoChatMessageContentRequest,
+    ImageChatMessageContentRequest,
     DocumentChatMessageContentRequest,
 ]

vellum/client/types/node_execution_span.py

@@ -12,6 +12,7 @@ from .workflow_sandbox_parent_context import WorkflowSandboxParentContext
 import typing
 from .vellum_node_execution_event import VellumNodeExecutionEvent
 from .node_execution_span_attributes import NodeExecutionSpanAttributes
+from .workflow_execution_usage_calculation_fulfilled_body import WorkflowExecutionUsageCalculationFulfilledBody
 import datetime as dt
 from ..core.pydantic_utilities import IS_PYDANTIC_V2
 import pydantic
@@ -21,6 +22,7 @@ class NodeExecutionSpan(UniversalBaseModel):
     name: typing.Literal["node.execution"] = "node.execution"
     events: typing.List[VellumNodeExecutionEvent]
     attributes: NodeExecutionSpanAttributes
+    usage_result: typing.Optional[WorkflowExecutionUsageCalculationFulfilledBody] = None
     span_id: str
     start_ts: dt.datetime
     end_ts: dt.datetime

vellum/client/types/prompt_block.py

@@ -5,8 +5,9 @@ import typing
 from .jinja_prompt_block import JinjaPromptBlock
 from .variable_prompt_block import VariablePromptBlock
 from .rich_text_prompt_block import RichTextPromptBlock
-from .audio_prompt_block import AudioPromptBlock
 from .function_call_prompt_block import FunctionCallPromptBlock
+from .audio_prompt_block import AudioPromptBlock
+from .video_prompt_block import VideoPromptBlock
 from .image_prompt_block import ImagePromptBlock
 from .document_prompt_block import DocumentPromptBlock
 import typing
@@ -18,8 +19,9 @@ PromptBlock = typing.Union[
     "ChatMessagePromptBlock",
     VariablePromptBlock,
     RichTextPromptBlock,
-    AudioPromptBlock,
     FunctionCallPromptBlock,
+    AudioPromptBlock,
+    VideoPromptBlock,
     ImagePromptBlock,
     DocumentPromptBlock,
 ]

vellum/client/types/vellum_value.py

@@ -5,8 +5,9 @@ import typing
 from .string_vellum_value import StringVellumValue
 from .number_vellum_value import NumberVellumValue
 from .json_vellum_value import JsonVellumValue
-from .image_vellum_value import ImageVellumValue
 from .audio_vellum_value import AudioVellumValue
+from .video_vellum_value import VideoVellumValue
+from .image_vellum_value import ImageVellumValue
 from .document_vellum_value import DocumentVellumValue
 from .function_call_vellum_value import FunctionCallVellumValue
 from .error_vellum_value import ErrorVellumValue
@@ -21,8 +22,9 @@ VellumValue = typing.Union[
     StringVellumValue,
     NumberVellumValue,
     JsonVellumValue,
-    ImageVellumValue,
     AudioVellumValue,
+    VideoVellumValue,
+    ImageVellumValue,
     DocumentVellumValue,
     FunctionCallVellumValue,
     ErrorVellumValue,

vellum/client/types/vellum_value_request.py

@@ -5,8 +5,9 @@ import typing
 from .string_vellum_value_request import StringVellumValueRequest
 from .number_vellum_value_request import NumberVellumValueRequest
 from .json_vellum_value_request import JsonVellumValueRequest
-from .image_vellum_value_request import ImageVellumValueRequest
 from .audio_vellum_value_request import AudioVellumValueRequest
+from .video_vellum_value_request import VideoVellumValueRequest
+from .image_vellum_value_request import ImageVellumValueRequest
 from .document_vellum_value_request import DocumentVellumValueRequest
 from .function_call_vellum_value_request import FunctionCallVellumValueRequest
 from .error_vellum_value_request import ErrorVellumValueRequest
@@ -21,8 +22,9 @@ VellumValueRequest = typing.Union[
     StringVellumValueRequest,
     NumberVellumValueRequest,
     JsonVellumValueRequest,
-    ImageVellumValueRequest,
     AudioVellumValueRequest,
+    VideoVellumValueRequest,
+    ImageVellumValueRequest,
     DocumentVellumValueRequest,
     FunctionCallVellumValueRequest,
     ErrorVellumValueRequest,

vellum/client/types/vellum_variable_type.py

@@ -12,8 +12,9 @@ VellumVariableType = typing.Union[
         "ERROR",
         "ARRAY",
         "FUNCTION_CALL",
-        "IMAGE",
         "AUDIO",
+        "VIDEO",
+        "IMAGE",
         "DOCUMENT",
         "NULL",
         "THINKING",

vellum/client/types/vellum_video.py

@@ -0,0 +1,24 @@
+# This file was auto-generated by Fern from our API Definition.
+
+from ..core.pydantic_utilities import UniversalBaseModel
+import pydantic
+import typing
+from ..core.pydantic_utilities import IS_PYDANTIC_V2
+
+
+class VellumVideo(UniversalBaseModel):
+    src: str = pydantic.Field()
+    """
+    A valid data URL containing the video data.
+    """
+
+    metadata: typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]] = None
+
+    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/vellum_video_request.py

@@ -0,0 +1,24 @@
+# This file was auto-generated by Fern from our API Definition.
+
+from ..core.pydantic_utilities import UniversalBaseModel
+import pydantic
+import typing
+from ..core.pydantic_utilities import IS_PYDANTIC_V2
+
+
+class VellumVideoRequest(UniversalBaseModel):
+    src: str = pydantic.Field()
+    """
+    A valid data URL containing the video data.
+    """
+
+    metadata: typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]] = None
+
+    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/video_chat_message_content.py

@@ -0,0 +1,25 @@
+# This file was auto-generated by Fern from our API Definition.
+
+from ..core.pydantic_utilities import UniversalBaseModel
+import typing
+from .vellum_video import VellumVideo
+from ..core.pydantic_utilities import IS_PYDANTIC_V2
+import pydantic
+
+
+class VideoChatMessageContent(UniversalBaseModel):
+    """
+    A video value that is used in a chat message.
+    """
+
+    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

vellum/client/types/video_chat_message_content_request.py

@@ -0,0 +1,25 @@
+# This file was auto-generated by Fern from our API Definition.
+
+from ..core.pydantic_utilities import UniversalBaseModel
+import typing
+from .vellum_video_request import VellumVideoRequest
+from ..core.pydantic_utilities import IS_PYDANTIC_V2
+import pydantic
+
+
+class VideoChatMessageContentRequest(UniversalBaseModel):
+    """
+    A video value that is used in a chat message.
+    """
+
+    type: typing.Literal["VIDEO"] = "VIDEO"
+    value: VellumVideoRequest
+
+    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/video_prompt_block.py

@@ -0,0 +1,29 @@
+# This file was auto-generated by Fern from our API Definition.
+
+from ..core.pydantic_utilities import UniversalBaseModel
+import typing
+from .prompt_block_state import PromptBlockState
+from .ephemeral_prompt_cache_config import EphemeralPromptCacheConfig
+from ..core.pydantic_utilities import IS_PYDANTIC_V2
+import pydantic
+
+
+class VideoPromptBlock(UniversalBaseModel):
+    """
+    A block that represents an video file in a prompt template.
+    """
+
+    block_type: typing.Literal["VIDEO"] = "VIDEO"
+    state: typing.Optional[PromptBlockState] = None
+    cache_config: typing.Optional[EphemeralPromptCacheConfig] = None
+    src: str
+    metadata: typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]] = None
+
+    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/video_vellum_value.py

@@ -0,0 +1,25 @@
+# This file was auto-generated by Fern from our API Definition.
+
+from ..core.pydantic_utilities import UniversalBaseModel
+import typing
+from .vellum_video import VellumVideo
+from ..core.pydantic_utilities import IS_PYDANTIC_V2
+import pydantic
+
+
+class VideoVellumValue(UniversalBaseModel):
+    """
+    A base Vellum primitive value representing a video.
+    """
+
+    type: typing.Literal["VIDEO"] = "VIDEO"
+    value: typing.Optional[VellumVideo] = None
+
+    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/video_vellum_value_request.py

@@ -0,0 +1,25 @@
+# This file was auto-generated by Fern from our API Definition.
+
+from ..core.pydantic_utilities import UniversalBaseModel
+import typing
+from .vellum_video_request import VellumVideoRequest
+from ..core.pydantic_utilities import IS_PYDANTIC_V2
+import pydantic
+
+
+class VideoVellumValueRequest(UniversalBaseModel):
+    """
+    A base Vellum primitive value representing a video.
+    """
+
+    type: typing.Literal["VIDEO"] = "VIDEO"
+    value: typing.Optional[VellumVideoRequest] = None
+
+    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