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

vellum/client/types/upsert_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 UpsertTestSuiteTestCaseRequest(UniversalBaseModel):
@@ -42,3 +43,8 @@ class UpsertTestSuiteTestCaseRequest(UniversalBaseModel):
             frozen = True
             smart_union = True
             extra = pydantic.Extra.allow
+
+
+from .array_vellum_value_request import ArrayVellumValueRequest  # noqa: E402, F401, I001
+
+update_forward_refs(UpsertTestSuiteTestCaseRequest)

vellum/client/types/variable_prompt_block.py

@@ -1,11 +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
-from .ephemeral_prompt_cache_config import EphemeralPromptCacheConfig
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
+
 import pydantic
+from vellum.client.utils import convert_input_variable_to_uuid
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+from .ephemeral_prompt_cache_config import EphemeralPromptCacheConfig
+from .prompt_block_state import PromptBlockState
 
 
 class VariablePromptBlock(UniversalBaseModel):
@@ -26,3 +27,9 @@ class VariablePromptBlock(UniversalBaseModel):
             frozen = True
             smart_union = True
             extra = pydantic.Extra.allow
+
+
+    @pydantic.field_serializer("input_variable")
+    def serialize_input_variable(self, value: str, info: pydantic.SerializationInfo) -> str:
+        """Convert input_variable to UUID using executable_id from context."""
+        return convert_input_variable_to_uuid(value, info)

vellum/client/types/vellum_audio.py

@@ -1,15 +1,20 @@
 # 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
+from vellum.utils.files.mixin import VellumFileMixin
+
+import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
 
 
-class VellumAudio(UniversalBaseModel):
+class VellumAudio(VellumFileMixin):
     src: str = pydantic.Field()
     """
-    A valid data URL containing the audio data.
+    The reference to the source data. This can be one of the following formats:
+    1. A base64-encoded data URL (e.g., 'data:application/pdf;base64,JVBERi0xLjQKJcfs...').
+    2. A url pointing to a file accessible over HTTP or HTTPS.
+    3. A reference to a file that's been previously uploaded to Vellum, in the form of 'vellum:uploaded-file:{uploaded_file_id}'.
+        You can use the Uploaded Files API to retrieve the url of the uploaded file given its ID. See https://docs.vellum.ai/developers/client-sdk/uploaded-files/retrieve for details.
     """
 
     metadata: typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]] = None

vellum/client/types/vellum_audio_request.py

@@ -1,15 +1,19 @@
 # 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 VellumAudioRequest(UniversalBaseModel):
     src: str = pydantic.Field()
     """
-    A valid data URL containing the audio data.
+    The reference to the source data. This can be one of the following formats:
+    1. A base64-encoded data URL (e.g., 'data:application/pdf;base64,JVBERi0xLjQKJcfs...').
+    2. A url pointing to a file accessible over HTTP or HTTPS.
+    3. A reference to a file that's been previously uploaded to Vellum, in the form of 'vellum:uploaded-file:{uploaded_file_id}'.
+        You can use the Uploaded Files API to retrieve the url of the uploaded file given its ID. See https://docs.vellum.ai/developers/client-sdk/uploaded-files/retrieve for details.
     """
 
     metadata: typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]] = None

vellum/client/types/vellum_code_resource_definition.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 VellumCodeResourceDefinition(UniversalBaseModel):
@@ -13,6 +13,11 @@ class VellumCodeResourceDefinition(UniversalBaseModel):
     The module that this resource is defined in.
     """
 
+    exclude_from_monitoring: typing.Optional[bool] = pydantic.Field(default=None)
+    """
+    Whether this node should be excluded from monitoring views.
+    """
+
     id: str
 
     if IS_PYDANTIC_V2:

vellum/client/types/vellum_document.py

@@ -1,13 +1,22 @@
 # 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
+from vellum.utils.files.mixin import VellumFileMixin
+
 import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+
 
+class VellumDocument(VellumFileMixin):
+    src: str = pydantic.Field()
+    """
+    The reference to the source data. This can be one of the following formats:
+    1. A base64-encoded data URL (e.g., 'data:application/pdf;base64,JVBERi0xLjQKJcfs...').
+    2. A url pointing to a file accessible over HTTP or HTTPS.
+    3. A reference to a file that's been previously uploaded to Vellum, in the form of 'vellum:uploaded-file:{uploaded_file_id}'.
+        You can use the Uploaded Files API to retrieve the url of the uploaded file given its ID. See https://docs.vellum.ai/developers/client-sdk/uploaded-files/retrieve for details.
+    """
 
-class VellumDocument(UniversalBaseModel):
-    src: str
     metadata: typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]] = None
 
     if IS_PYDANTIC_V2:

vellum/client/types/vellum_document_request.py

@@ -1,13 +1,21 @@
 # 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 VellumDocumentRequest(UniversalBaseModel):
-    src: str
+    src: str = pydantic.Field()
+    """
+    The reference to the source data. This can be one of the following formats:
+    1. A base64-encoded data URL (e.g., 'data:application/pdf;base64,JVBERi0xLjQKJcfs...').
+    2. A url pointing to a file accessible over HTTP or HTTPS.
+    3. A reference to a file that's been previously uploaded to Vellum, in the form of 'vellum:uploaded-file:{uploaded_file_id}'.
+        You can use the Uploaded Files API to retrieve the url of the uploaded file given its ID. See https://docs.vellum.ai/developers/client-sdk/uploaded-files/retrieve for details.
+    """
+
     metadata: typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]] = None
 
     if IS_PYDANTIC_V2:

vellum/client/types/vellum_error.py

@@ -1,15 +1,16 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from ..core.pydantic_utilities import UniversalBaseModel
-from .vellum_error_code_enum import VellumErrorCodeEnum
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
 import typing
+
 import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+from .vellum_error_code_enum import VellumErrorCodeEnum
 
 
 class VellumError(UniversalBaseModel):
-    message: str
     code: VellumErrorCodeEnum
+    message: str
+    raw_data: 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

vellum/client/types/vellum_error_code_enum.py

@@ -7,9 +7,15 @@ VellumErrorCodeEnum = typing.Union[
         "INVALID_REQUEST",
         "INVALID_INPUTS",
         "PROVIDER_ERROR",
+        "PROVIDER_CREDENTIALS_UNAVAILABLE",
+        "INTEGRATION_CREDENTIALS_UNAVAILABLE",
         "REQUEST_TIMEOUT",
         "INTERNAL_SERVER_ERROR",
         "USER_DEFINED_ERROR",
+        "WORKFLOW_CANCELLED",
+        "NODE_CANCELLED",
+        "PROVIDER_QUOTA_EXCEEDED",
+        "CHAT_QUOTA_EXCEEDED",
     ],
     typing.Any,
 ]

vellum/client/types/vellum_error_request.py

@@ -1,15 +1,16 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from ..core.pydantic_utilities import UniversalBaseModel
-from .vellum_error_code_enum import VellumErrorCodeEnum
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
 import typing
+
 import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+from .vellum_error_code_enum import VellumErrorCodeEnum
 
 
 class VellumErrorRequest(UniversalBaseModel):
-    message: str
     code: VellumErrorCodeEnum
+    message: str
+    raw_data: 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

vellum/client/types/vellum_image.py

@@ -1,13 +1,22 @@
 # 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
+from vellum.utils.files.mixin import VellumFileMixin
+
 import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+
 
+class VellumImage(VellumFileMixin):
+    src: str = pydantic.Field()
+    """
+    The reference to the source data. This can be one of the following formats:
+    1. A base64-encoded data URL (e.g., 'data:application/pdf;base64,JVBERi0xLjQKJcfs...').
+    2. A url pointing to a file accessible over HTTP or HTTPS.
+    3. A reference to a file that's been previously uploaded to Vellum, in the form of 'vellum:uploaded-file:{uploaded_file_id}'.
+        You can use the Uploaded Files API to retrieve the url of the uploaded file given its ID. See https://docs.vellum.ai/developers/client-sdk/uploaded-files/retrieve for details.
+    """
 
-class VellumImage(UniversalBaseModel):
-    src: str
     metadata: typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]] = None
 
     if IS_PYDANTIC_V2:

vellum/client/types/vellum_image_request.py

@@ -1,13 +1,21 @@
 # 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 VellumImageRequest(UniversalBaseModel):
-    src: str
+    src: str = pydantic.Field()
+    """
+    The reference to the source data. This can be one of the following formats:
+    1. A base64-encoded data URL (e.g., 'data:application/pdf;base64,JVBERi0xLjQKJcfs...').
+    2. A url pointing to a file accessible over HTTP or HTTPS.
+    3. A reference to a file that's been previously uploaded to Vellum, in the form of 'vellum:uploaded-file:{uploaded_file_id}'.
+        You can use the Uploaded Files API to retrieve the url of the uploaded file given its ID. See https://docs.vellum.ai/developers/client-sdk/uploaded-files/retrieve for details.
+    """
+
     metadata: typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]] = None
 
     if IS_PYDANTIC_V2:

vellum/client/types/vellum_node_execution_event.py

@@ -1,12 +1,14 @@
 # This file was auto-generated by Fern from our API Definition.
 
 import typing
-from .node_execution_initiated_event import NodeExecutionInitiatedEvent
-from .node_execution_streaming_event import NodeExecutionStreamingEvent
+
 from .node_execution_fulfilled_event import NodeExecutionFulfilledEvent
-from .node_execution_rejected_event import NodeExecutionRejectedEvent
+from .node_execution_initiated_event import NodeExecutionInitiatedEvent
+from .node_execution_log_event import NodeExecutionLogEvent
 from .node_execution_paused_event import NodeExecutionPausedEvent
+from .node_execution_rejected_event import NodeExecutionRejectedEvent
 from .node_execution_resumed_event import NodeExecutionResumedEvent
+from .node_execution_streaming_event import NodeExecutionStreamingEvent
 
 VellumNodeExecutionEvent = typing.Union[
     NodeExecutionInitiatedEvent,
@@ -15,4 +17,5 @@ VellumNodeExecutionEvent = typing.Union[
     NodeExecutionRejectedEvent,
     NodeExecutionPausedEvent,
     NodeExecutionResumedEvent,
+    NodeExecutionLogEvent,
 ]

vellum/client/types/vellum_sdk_error.py

@@ -1,15 +1,16 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from ..core.pydantic_utilities import UniversalBaseModel
-from .vellum_sdk_error_code_enum import VellumSdkErrorCodeEnum
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
 import typing
+
 import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+from .vellum_sdk_error_code_enum import VellumSdkErrorCodeEnum
 
 
 class VellumSdkError(UniversalBaseModel):
     message: str
     code: VellumSdkErrorCodeEnum
+    raw_data: 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

vellum/client/types/vellum_sdk_error_code_enum.py

@@ -12,9 +12,14 @@ VellumSdkErrorCodeEnum = typing.Union[
         "INVALID_TEMPLATE",
         "INTERNAL_ERROR",
         "PROVIDER_CREDENTIALS_UNAVAILABLE",
+        "INTEGRATION_CREDENTIALS_UNAVAILABLE",
         "PROVIDER_ERROR",
+        "PROVIDER_QUOTA_EXCEEDED",
         "USER_DEFINED_ERROR",
         "WORKFLOW_CANCELLED",
+        "WORKFLOW_TIMEOUT",
+        "NODE_CANCELLED",
+        "NODE_TIMEOUT",
         "NODE_EXECUTION",
     ],
     typing.Any,

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

vellum/client/types/vellum_span.py

@@ -1,7 +1,8 @@
 # This file was auto-generated by Fern from our API Definition.
 
 import typing
-from .workflow_execution_span import WorkflowExecutionSpan
+
 from .node_execution_span import NodeExecutionSpan
+from .workflow_execution_span import WorkflowExecutionSpan
 
 VellumSpan = typing.Union[WorkflowExecutionSpan, NodeExecutionSpan]

vellum/client/types/vellum_value.py

@@ -1,18 +1,21 @@
 # This file was auto-generated by Fern from our API Definition.
 
 from __future__ import annotations
+
 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 .chat_history_vellum_value import ChatHistoryVellumValue
 from .document_vellum_value import DocumentVellumValue
-from .function_call_vellum_value import FunctionCallVellumValue
 from .error_vellum_value import ErrorVellumValue
-from .chat_history_vellum_value import ChatHistoryVellumValue
+from .function_call_vellum_value import FunctionCallVellumValue
+from .image_vellum_value import ImageVellumValue
+from .json_vellum_value import JsonVellumValue
+from .number_vellum_value import NumberVellumValue
 from .search_results_vellum_value import SearchResultsVellumValue
-import typing
+from .string_vellum_value import StringVellumValue
+from .thinking_vellum_value import ThinkingVellumValue
+from .video_vellum_value import VideoVellumValue
 
 if typing.TYPE_CHECKING:
     from .array_vellum_value import ArrayVellumValue
@@ -20,12 +23,14 @@ VellumValue = typing.Union[
     StringVellumValue,
     NumberVellumValue,
     JsonVellumValue,
-    ImageVellumValue,
     AudioVellumValue,
+    VideoVellumValue,
+    ImageVellumValue,
     DocumentVellumValue,
     FunctionCallVellumValue,
     ErrorVellumValue,
     "ArrayVellumValue",
     ChatHistoryVellumValue,
     SearchResultsVellumValue,
+    ThinkingVellumValue,
 ]

vellum/client/types/vellum_value_logical_condition_group_request.py

@@ -1,13 +1,12 @@
 # This file was auto-generated by Fern from our API Definition.
 
 from __future__ import annotations
-from ..core.pydantic_utilities import UniversalBaseModel
-from .array_vellum_value_request import ArrayVellumValueRequest
+
 import typing
-from .condition_combinator import ConditionCombinator
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
+
 import pydantic
-from ..core.pydantic_utilities import update_forward_refs
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel, update_forward_refs
+from .condition_combinator import ConditionCombinator
 
 
 class VellumValueLogicalConditionGroupRequest(UniversalBaseModel):
@@ -30,6 +29,7 @@ class VellumValueLogicalConditionGroupRequest(UniversalBaseModel):
             extra = pydantic.Extra.allow
 
 
-from .vellum_value_logical_expression_request import VellumValueLogicalExpressionRequest  # noqa: E402
+from .array_vellum_value_request import ArrayVellumValueRequest  # noqa: E402, F401, I001
+from .vellum_value_logical_expression_request import VellumValueLogicalExpressionRequest  # noqa: E402, F401, I001
 
 update_forward_refs(VellumValueLogicalConditionGroupRequest)

vellum/client/types/vellum_value_logical_condition_request.py

@@ -1,12 +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
-from .vellum_value_request import VellumValueRequest
-from .logical_operator import LogicalOperator
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
+
 import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel, update_forward_refs
+from .logical_operator import LogicalOperator
 
 
 class VellumValueLogicalConditionRequest(UniversalBaseModel):
@@ -15,9 +15,9 @@ class VellumValueLogicalConditionRequest(UniversalBaseModel):
     """
 
     type: typing.Literal["LOGICAL_CONDITION"] = "LOGICAL_CONDITION"
-    lhs_variable: VellumValueRequest
+    lhs_variable: "VellumValueRequest"
     operator: LogicalOperator
-    rhs_variable: VellumValueRequest
+    rhs_variable: "VellumValueRequest"
 
     if IS_PYDANTIC_V2:
         model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
@@ -27,3 +27,9 @@ class VellumValueLogicalConditionRequest(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(VellumValueLogicalConditionRequest)

vellum/client/types/vellum_value_logical_expression_request.py

@@ -1,9 +1,10 @@
 # This file was auto-generated by Fern from our API Definition.
 
 from __future__ import annotations
+
 import typing
+
 from .vellum_value_logical_condition_request import VellumValueLogicalConditionRequest
-import typing
 
 if typing.TYPE_CHECKING:
     from .vellum_value_logical_condition_group_request import VellumValueLogicalConditionGroupRequest

vellum/client/types/vellum_value_request.py

@@ -1,18 +1,21 @@
 # This file was auto-generated by Fern from our API Definition.
 
 from __future__ import annotations
+
 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 .chat_history_vellum_value_request import ChatHistoryVellumValueRequest
 from .document_vellum_value_request import DocumentVellumValueRequest
-from .function_call_vellum_value_request import FunctionCallVellumValueRequest
 from .error_vellum_value_request import ErrorVellumValueRequest
-from .chat_history_vellum_value_request import ChatHistoryVellumValueRequest
+from .function_call_vellum_value_request import FunctionCallVellumValueRequest
+from .image_vellum_value_request import ImageVellumValueRequest
+from .json_vellum_value_request import JsonVellumValueRequest
+from .number_vellum_value_request import NumberVellumValueRequest
 from .search_results_vellum_value_request import SearchResultsVellumValueRequest
-import typing
+from .string_vellum_value_request import StringVellumValueRequest
+from .thinking_vellum_value_request import ThinkingVellumValueRequest
+from .video_vellum_value_request import VideoVellumValueRequest
 
 if typing.TYPE_CHECKING:
     from .array_vellum_value_request import ArrayVellumValueRequest
@@ -20,12 +23,14 @@ VellumValueRequest = typing.Union[
     StringVellumValueRequest,
     NumberVellumValueRequest,
     JsonVellumValueRequest,
-    ImageVellumValueRequest,
     AudioVellumValueRequest,
+    VideoVellumValueRequest,
+    ImageVellumValueRequest,
     DocumentVellumValueRequest,
     FunctionCallVellumValueRequest,
     ErrorVellumValueRequest,
     "ArrayVellumValueRequest",
     ChatHistoryVellumValueRequest,
     SearchResultsVellumValueRequest,
+    ThinkingVellumValueRequest,
 ]

vellum/client/types/vellum_variable.py

@@ -1,13 +1,15 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from ..core.pydantic_utilities import UniversalBaseModel
-from .array_vellum_value import ArrayVellumValue
-from .vellum_variable_type import VellumVariableType
+from __future__ import annotations
+
 import typing
-from .vellum_value import VellumValue
-from .vellum_variable_extensions import VellumVariableExtensions
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
+
 import pydantic
+import typing_extensions
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel, update_forward_refs
+from ..core.serialization import FieldMetadata
+from .vellum_variable_extensions import VellumVariableExtensions
+from .vellum_variable_type import VellumVariableType
 
 
 class VellumVariable(UniversalBaseModel):
@@ -15,8 +17,11 @@ class VellumVariable(UniversalBaseModel):
     key: str
     type: VellumVariableType
     required: typing.Optional[bool] = None
-    default: typing.Optional[VellumValue] = None
+    default: typing.Optional["VellumValue"] = None
     extensions: typing.Optional[VellumVariableExtensions] = None
+    schema_: typing_extensions.Annotated[
+        typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]], FieldMetadata(alias="schema")
+    ] = None
 
     if IS_PYDANTIC_V2:
         model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
@@ -26,3 +31,9 @@ class VellumVariable(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(VellumVariable)

vellum/client/types/vellum_variable_extensions.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 VellumVariableExtensions(UniversalBaseModel):
@@ -12,6 +12,8 @@ class VellumVariableExtensions(UniversalBaseModel):
     """
 
     color: typing.Optional[str] = None
+    description: typing.Optional[str] = None
+    title: typing.Optional[str] = None
 
     if IS_PYDANTIC_V2:
         model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2

vellum/client/types/vellum_variable_type.py

@@ -12,10 +12,13 @@ VellumVariableType = typing.Union[
         "ERROR",
         "ARRAY",
         "FUNCTION_CALL",
-        "IMAGE",
         "AUDIO",
+        "VIDEO",
+        "IMAGE",
         "DOCUMENT",
         "NULL",
+        "THINKING",
+        "REFERENCE",
     ],
     typing.Any,
 ]