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

vellum/client/types/prompt_exec_config.py

@@ -1,18 +1,15 @@
 # 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 import ArrayVellumValue
-from .chat_message_prompt_block import ChatMessagePromptBlock
+
 import typing
-from .vellum_variable import VellumVariable
+
+import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel, update_forward_refs
+from .function_definition import FunctionDefinition
 from .prompt_parameters import PromptParameters
 from .prompt_settings import PromptSettings
-from .prompt_block import PromptBlock
-from .function_definition import FunctionDefinition
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
-import pydantic
-from ..core.pydantic_utilities import update_forward_refs
+from .vellum_variable import VellumVariable
 
 
 class PromptExecConfig(UniversalBaseModel):
@@ -20,7 +17,7 @@ class PromptExecConfig(UniversalBaseModel):
     input_variables: typing.List[VellumVariable]
     parameters: PromptParameters
     settings: typing.Optional[PromptSettings] = None
-    blocks: typing.List[PromptBlock]
+    blocks: typing.List["PromptBlock"]
     functions: typing.Optional[typing.List[FunctionDefinition]] = None
 
     if IS_PYDANTIC_V2:
@@ -33,5 +30,8 @@ class PromptExecConfig(UniversalBaseModel):
             extra = pydantic.Extra.allow
 
 
-update_forward_refs(ArrayVellumValue, PromptExecConfig=PromptExecConfig)
-update_forward_refs(ChatMessagePromptBlock, PromptExecConfig=PromptExecConfig)
+from .array_vellum_value import ArrayVellumValue  # noqa: E402, F401, I001
+from .chat_message_prompt_block import ChatMessagePromptBlock  # noqa: E402, F401, I001
+from .prompt_block import PromptBlock  # noqa: E402, F401, I001
+
+update_forward_refs(PromptExecConfig)

vellum/client/types/prompt_execution_meta.py

@@ -1,12 +1,12 @@
 # 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 .finish_reason_enum import FinishReasonEnum
 from .ml_model_usage import MlModelUsage
 from .price import Price
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
-import pydantic
 
 
 class PromptExecutionMeta(UniversalBaseModel):

vellum/client/types/prompt_node_execution_meta.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 pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
 from .ml_model_usage import MlModelUsage
 from .price import Price
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
-import pydantic
 
 
 class PromptNodeExecutionMeta(UniversalBaseModel):

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

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

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

@@ -0,0 +1,20 @@
+# 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 PromptPushResponse(UniversalBaseModel):
+    prompt_variant_id: str
+    prompt_sandbox_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/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,13 +1,14 @@
 # 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):
     timeout: typing.Optional[float] = None
+    stream_enabled: typing.Optional[bool] = None
 
     if IS_PYDANTIC_V2:
         model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2

vellum/client/types/prompt_version_build_config_sandbox.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
+
+
+class PromptVersionBuildConfigSandbox(UniversalBaseModel):
+    source: typing.Literal["SANDBOX"] = "SANDBOX"
+    sandbox_id: str
+    sandbox_snapshot_id: str
+    prompt_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/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,15 +1,14 @@
 # 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 import ArrayVellumValue
-import typing
+
 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 update_forward_refs
+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):
@@ -36,4 +35,6 @@ class RejectedWorkflowNodeResultEvent(UniversalBaseModel):
             extra = pydantic.Extra.allow
 
 
-update_forward_refs(ArrayVellumValue, RejectedWorkflowNodeResultEvent=RejectedWorkflowNodeResultEvent)
+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,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
+
 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
-from ..core.pydantic_utilities import update_forward_refs
 
 
 class ReplaceTestSuiteTestCaseRequest(UniversalBaseModel):
@@ -50,4 +49,6 @@ class ReplaceTestSuiteTestCaseRequest(UniversalBaseModel):
             extra = pydantic.Extra.allow
 
 
-update_forward_refs(ArrayVellumValueRequest, ReplaceTestSuiteTestCaseRequest=ReplaceTestSuiteTestCaseRequest)
+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]