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

vellum/client/types/test_suite_test_case_bulk_result.py

@@ -1,10 +1,11 @@
 # This file was auto-generated by Fern from our API Definition.
 
 import typing
+
 from .test_suite_test_case_created_bulk_result import TestSuiteTestCaseCreatedBulkResult
-from .test_suite_test_case_replaced_bulk_result import TestSuiteTestCaseReplacedBulkResult
 from .test_suite_test_case_deleted_bulk_result import TestSuiteTestCaseDeletedBulkResult
 from .test_suite_test_case_rejected_bulk_result import TestSuiteTestCaseRejectedBulkResult
+from .test_suite_test_case_replaced_bulk_result import TestSuiteTestCaseReplacedBulkResult
 
 TestSuiteTestCaseBulkResult = typing.Union[
     TestSuiteTestCaseCreatedBulkResult,

vellum/client/types/test_suite_test_case_create_bulk_operation_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
-import pydantic
+from __future__ import annotations
+
 import typing
+
+import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel, update_forward_refs
 from .create_test_suite_test_case_request import CreateTestSuiteTestCaseRequest
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
 
 
 class TestSuiteTestCaseCreateBulkOperationRequest(UniversalBaseModel):
@@ -29,3 +30,8 @@ class TestSuiteTestCaseCreateBulkOperationRequest(UniversalBaseModel):
             frozen = True
             smart_union = True
             extra = pydantic.Extra.allow
+
+
+from .array_vellum_value_request import ArrayVellumValueRequest  # noqa: E402, F401, I001
+
+update_forward_refs(TestSuiteTestCaseCreateBulkOperationRequest)

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

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

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

vellum/client/types/test_suite_test_case_delete_bulk_operation_request.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 .test_suite_test_case_delete_bulk_operation_data_request import TestSuiteTestCaseDeleteBulkOperationDataRequest
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
 
 
 class TestSuiteTestCaseDeleteBulkOperationRequest(UniversalBaseModel):

vellum/client/types/test_suite_test_case_deleted_bulk_result.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 .test_suite_test_case_deleted_bulk_result_data import TestSuiteTestCaseDeletedBulkResultData
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
 
 
 class TestSuiteTestCaseDeletedBulkResult(UniversalBaseModel):

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

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

vellum/client/types/test_suite_test_case_replace_bulk_operation_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
-import pydantic
+from __future__ import annotations
+
 import typing
+
+import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel, update_forward_refs
 from .replace_test_suite_test_case_request import ReplaceTestSuiteTestCaseRequest
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
 
 
 class TestSuiteTestCaseReplaceBulkOperationRequest(UniversalBaseModel):
@@ -29,3 +30,8 @@ class TestSuiteTestCaseReplaceBulkOperationRequest(UniversalBaseModel):
             frozen = True
             smart_union = True
             extra = pydantic.Extra.allow
+
+
+from .array_vellum_value_request import ArrayVellumValueRequest  # noqa: E402, F401, I001
+
+update_forward_refs(TestSuiteTestCaseReplaceBulkOperationRequest)

vellum/client/types/test_suite_test_case_replaced_bulk_result.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 .test_suite_test_case_replaced_bulk_result_data import TestSuiteTestCaseReplacedBulkResultData
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
 
 
 class TestSuiteTestCaseReplacedBulkResult(UniversalBaseModel):

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

vellum/client/types/test_suite_test_case_upsert_bulk_operation_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
-import pydantic
+from __future__ import annotations
+
 import typing
+
+import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel, update_forward_refs
 from .upsert_test_suite_test_case_request import UpsertTestSuiteTestCaseRequest
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
 
 
 class TestSuiteTestCaseUpsertBulkOperationRequest(UniversalBaseModel):
@@ -29,3 +30,8 @@ class TestSuiteTestCaseUpsertBulkOperationRequest(UniversalBaseModel):
             frozen = True
             smart_union = True
             extra = pydantic.Extra.allow
+
+
+from .array_vellum_value_request import ArrayVellumValueRequest  # noqa: E402, F401, I001
+
+update_forward_refs(TestSuiteTestCaseUpsertBulkOperationRequest)

vellum/client/types/thinking_vellum_value.py

@@ -0,0 +1,25 @@
+# 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 .string_vellum_value import StringVellumValue
+
+
+class ThinkingVellumValue(UniversalBaseModel):
+    """
+    A value representing Thinking mode output.
+    """
+
+    type: typing.Literal["THINKING"] = "THINKING"
+    value: typing.Optional[StringVellumValue] = 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/thinking_vellum_value_request.py

@@ -0,0 +1,25 @@
+# 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 .string_vellum_value_request import StringVellumValueRequest
+
+
+class ThinkingVellumValueRequest(UniversalBaseModel):
+    """
+    A value representing Thinking mode output.
+    """
+
+    type: typing.Literal["THINKING"] = "THINKING"
+    value: typing.Optional[StringVellumValueRequest] = 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/token_overlapping_window_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 TokenOverlappingWindowChunkerConfig(UniversalBaseModel):

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

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

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

vellum/client/types/update_active_workspace_response.py

@@ -0,0 +1,32 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+
+
+class UpdateActiveWorkspaceResponse(UniversalBaseModel):
+    update_active_workspace_id: typing.Optional[str] = pydantic.Field(default=None)
+    """
+    The id of the workspace that the user should update to, or null if no workspace change needed.
+    """
+
+    update_active_environment_id: typing.Optional[str] = pydantic.Field(default=None)
+    """
+    The id of the environment that the user should update to, or null if no environment change needed.
+    """
+
+    is_staff: typing.Optional[bool] = pydantic.Field(default=None)
+    """
+    Whether or not the user is a staff member of Vellum.
+    """
+
+    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/upload_document_response.py

@@ -1,10 +1,10 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from ..core.pydantic_utilities import UniversalBaseModel
-import pydantic
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
 import typing
 
+import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+
 
 class UploadDocumentResponse(UniversalBaseModel):
     document_id: str = pydantic.Field()

vellum/client/types/uploaded_file_read.py

@@ -0,0 +1,27 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+
+
+class UploadedFileRead(UniversalBaseModel):
+    """
+    Represents a file that has been uploaded to a Vellum Workspace.
+    """
+
+    id: typing.Optional[str] = None
+    file_url: typing.Optional[str] = pydantic.Field(default=None)
+    """
+    A signed URL to download the uploaded file.
+    """
+
+    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/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,12 +1,13 @@
 # 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.

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

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,18 @@
 # 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()
+    """
+    A valid data URL containing the document data.
+    """
 
-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,17 @@
 # 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()
+    """
+    A valid data URL containing the document data.
+    """
+
     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,18 @@
 # 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()
+    """
+    A valid data URL containing the image data.
+    """
 
-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,17 @@
 # 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()
+    """
+    A valid data URL containing the image data.
+    """
+
     metadata: typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]] = None
 
     if IS_PYDANTIC_V2: