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

vellum/client/types/prompt_output.py

@@ -1,9 +1,13 @@
 # This file was auto-generated by Fern from our API Definition.
 
 import typing
-from .string_vellum_value import StringVellumValue
-from .json_vellum_value import JsonVellumValue
+
 from .error_vellum_value import ErrorVellumValue
 from .function_call_vellum_value import FunctionCallVellumValue
+from .json_vellum_value import JsonVellumValue
+from .string_vellum_value import StringVellumValue
+from .thinking_vellum_value import ThinkingVellumValue
 
-PromptOutput = typing.Union[StringVellumValue, JsonVellumValue, ErrorVellumValue, FunctionCallVellumValue]
+PromptOutput = typing.Union[
+    StringVellumValue, JsonVellumValue, ErrorVellumValue, FunctionCallVellumValue, ThinkingVellumValue
+]

vellum/client/types/prompt_parameters.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 PromptParameters(UniversalBaseModel):

vellum/client/types/prompt_push_response.py

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

vellum/client/types/prompt_request_audio_input.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 PromptRequestAudioInput(UniversalBaseModel):
+    key: str = pydantic.Field()
+    """
+    The variable's name, as defined in the Prompt.
+    """
+
+    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/prompt_request_chat_history_input.py

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

vellum/client/types/prompt_request_document_input.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_document import VellumDocument
+
+
+class PromptRequestDocumentInput(UniversalBaseModel):
+    key: str = pydantic.Field()
+    """
+    The variable's name, as defined in the Prompt.
+    """
+
+    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/prompt_request_image_input.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_image import VellumImage
+
+
+class PromptRequestImageInput(UniversalBaseModel):
+    key: str = pydantic.Field()
+    """
+    The variable's name, as defined in the Prompt.
+    """
+
+    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/prompt_request_input.py

@@ -1,8 +1,21 @@
 # This file was auto-generated by Fern from our API Definition.
 
 import typing
-from .prompt_request_string_input import PromptRequestStringInput
-from .prompt_request_json_input import PromptRequestJsonInput
+
+from .prompt_request_audio_input import PromptRequestAudioInput
 from .prompt_request_chat_history_input import PromptRequestChatHistoryInput
+from .prompt_request_document_input import PromptRequestDocumentInput
+from .prompt_request_image_input import PromptRequestImageInput
+from .prompt_request_json_input import PromptRequestJsonInput
+from .prompt_request_string_input import PromptRequestStringInput
+from .prompt_request_video_input import PromptRequestVideoInput
 
-PromptRequestInput = typing.Union[PromptRequestStringInput, PromptRequestJsonInput, PromptRequestChatHistoryInput]
+PromptRequestInput = typing.Union[
+    PromptRequestStringInput,
+    PromptRequestJsonInput,
+    PromptRequestChatHistoryInput,
+    PromptRequestAudioInput,
+    PromptRequestVideoInput,
+    PromptRequestImageInput,
+    PromptRequestDocumentInput,
+]

vellum/client/types/prompt_request_json_input.py

@@ -1,9 +1,9 @@
 # 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
+
+import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
 
 
 class PromptRequestJsonInput(UniversalBaseModel):

vellum/client/types/prompt_request_string_input.py

@@ -1,9 +1,9 @@
 # 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
+
+import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
 
 
 class PromptRequestStringInput(UniversalBaseModel):

vellum/client/types/prompt_request_video_input.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_video import VellumVideo
+
+
+class PromptRequestVideoInput(UniversalBaseModel):
+    key: str = pydantic.Field()
+    """
+    The variable's name, as defined in the Prompt.
+    """
+
+    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/prompt_settings.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 PromptSettings(UniversalBaseModel):

vellum/client/types/prompt_version_build_config_sandbox.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 PromptVersionBuildConfigSandbox(UniversalBaseModel):

vellum/client/types/raw_prompt_execution_overrides_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
+
 import pydantic
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
 
 
 class RawPromptExecutionOverridesRequest(UniversalBaseModel):

vellum/client/types/reducto_chunker_config.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 ReductoChunkerConfig(UniversalBaseModel):

vellum/client/types/reducto_chunker_config_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 ReductoChunkerConfigRequest(UniversalBaseModel):

vellum/client/types/reducto_chunking.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 .reducto_chunker_config import ReductoChunkerConfig
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
+
 import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+from .reducto_chunker_config import ReductoChunkerConfig
 
 
 class ReductoChunking(UniversalBaseModel):

vellum/client/types/reducto_chunking_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 .reducto_chunker_config_request import ReductoChunkerConfigRequest
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
+
 import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+from .reducto_chunker_config_request import ReductoChunkerConfigRequest
 
 
 class ReductoChunkingRequest(UniversalBaseModel):

vellum/client/types/rejected_ad_hoc_execute_prompt_event.py

@@ -1,11 +1,11 @@
 # 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 .ad_hoc_rejected_prompt_execution_meta import AdHocRejectedPromptExecutionMeta
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
+
 import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+from .ad_hoc_rejected_prompt_execution_meta import AdHocRejectedPromptExecutionMeta
+from .vellum_error import VellumError
 
 
 class RejectedAdHocExecutePromptEvent(UniversalBaseModel):

vellum/client/types/rejected_execute_prompt_event.py

@@ -1,11 +1,11 @@
 # 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 .rejected_prompt_execution_meta import RejectedPromptExecutionMeta
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
+
 import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+from .rejected_prompt_execution_meta import RejectedPromptExecutionMeta
+from .vellum_error import VellumError
 
 
 class RejectedExecutePromptEvent(UniversalBaseModel):

vellum/client/types/rejected_execute_prompt_response.py

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

vellum/client/types/rejected_execute_workflow_workflow_result_event.py

@@ -1,11 +1,11 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from ..core.pydantic_utilities import UniversalBaseModel
-import typing
 import datetime as dt
-from .workflow_event_error import WorkflowEventError
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
+import typing
+
 import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+from .workflow_event_error import WorkflowEventError
 
 
 class RejectedExecuteWorkflowWorkflowResultEvent(UniversalBaseModel):

vellum/client/types/rejected_prompt_execution_meta.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 .finish_reason_enum import FinishReasonEnum
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
+
 import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+from .finish_reason_enum import FinishReasonEnum
 
 
 class RejectedPromptExecutionMeta(UniversalBaseModel):

vellum/client/types/rejected_workflow_node_result_event.py

@@ -1,13 +1,14 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from ..core.pydantic_utilities import UniversalBaseModel
-from .array_vellum_value import ArrayVellumValue
-import typing
+from __future__ import annotations
+
 import datetime as dt
-from .workflow_node_result_data import WorkflowNodeResultData
-from .workflow_event_error import WorkflowEventError
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
+import typing
+
 import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel, update_forward_refs
+from .workflow_event_error import WorkflowEventError
+from .workflow_node_result_data import WorkflowNodeResultData
 
 
 class RejectedWorkflowNodeResultEvent(UniversalBaseModel):
@@ -32,3 +33,8 @@ class RejectedWorkflowNodeResultEvent(UniversalBaseModel):
             frozen = True
             smart_union = True
             extra = pydantic.Extra.allow
+
+
+from .array_vellum_value import ArrayVellumValue  # noqa: E402, F401, I001
+
+update_forward_refs(RejectedWorkflowNodeResultEvent)

vellum/client/types/release_created_by.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 ReleaseCreatedBy(UniversalBaseModel):

vellum/client/types/release_environment.py

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

vellum/client/types/release_release_tag.py

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

vellum/client/types/release_review_reviewer.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 ReleaseReviewReviewer(UniversalBaseModel):

vellum/client/types/release_tag_release.py

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

vellum/client/types/replace_test_suite_test_case_request.py

@@ -1,11 +1,12 @@
 # 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
+
 import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel, update_forward_refs
 from .named_test_case_variable_value_request import NamedTestCaseVariableValueRequest
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
 
 
 class ReplaceTestSuiteTestCaseRequest(UniversalBaseModel):
@@ -46,3 +47,8 @@ class ReplaceTestSuiteTestCaseRequest(UniversalBaseModel):
             frozen = True
             smart_union = True
             extra = pydantic.Extra.allow
+
+
+from .array_vellum_value_request import ArrayVellumValueRequest  # noqa: E402, F401, I001
+
+update_forward_refs(ReplaceTestSuiteTestCaseRequest)

vellum/client/types/rich_text_child_block.py

@@ -1,7 +1,8 @@
 # This file was auto-generated by Fern from our API Definition.
 
 import typing
-from .variable_prompt_block import VariablePromptBlock
+
 from .plain_text_prompt_block import PlainTextPromptBlock
+from .variable_prompt_block import VariablePromptBlock
 
 RichTextChildBlock = typing.Union[VariablePromptBlock, PlainTextPromptBlock]

vellum/client/types/rich_text_prompt_block.py

@@ -1,12 +1,12 @@
 # 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
+
+import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
 from .ephemeral_prompt_cache_config import EphemeralPromptCacheConfig
+from .prompt_block_state import PromptBlockState
 from .rich_text_child_block import RichTextChildBlock
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
-import pydantic
 
 
 class RichTextPromptBlock(UniversalBaseModel):

vellum/client/types/sandbox_scenario.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 .scenario_input import ScenarioInput
+
 import pydantic
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+from .scenario_input import ScenarioInput
 
 
 class SandboxScenario(UniversalBaseModel):

vellum/client/types/scenario_input.py

@@ -1,10 +1,21 @@
 # This file was auto-generated by Fern from our API Definition.
 
 import typing
-from .scenario_input_string_variable_value import ScenarioInputStringVariableValue
-from .scenario_input_json_variable_value import ScenarioInputJsonVariableValue
+
+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: typing.Optional[VellumAudio] = None
+    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_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 ScenarioInputChatHistoryVariableValue(UniversalBaseModel):

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