vellum-ai 1.2.4__tar.gz → 1.2.5__tar.gz

{vellum_ai-1.2.4 → vellum_ai-1.2.5}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: vellum-ai
-Version: 1.2.4
+Version: 1.2.5
 Summary: 
 License: MIT
 Requires-Python: >=3.9,<4.0

{vellum_ai-1.2.4 → vellum_ai-1.2.5}/ee/vellum_ee/workflows/display/utils/events.py

@@ -7,6 +7,24 @@ from vellum_ee.workflows.display.utils.registry import (
 from vellum_ee.workflows.display.workflows.get_vellum_workflow_display_class import get_workflow_display
 
 
+def _should_mark_workflow_dynamic(event: WorkflowExecutionInitiatedEvent) -> bool:
+    """
+    Check if workflow should be marked as dynamic based on execution context.
+    Returns True if parent.type == WORKFLOW_RELEASE_TAG and parent.parent.type == WORKFLOW_NODE.
+    """
+    if not event.parent:
+        return False
+
+    parent = event.parent
+    if parent.type != "WORKFLOW_RELEASE_TAG":
+        return False
+
+    if not parent.parent or parent.parent.type != "WORKFLOW_NODE":
+        return False
+
+    return True
+
+
 def event_enricher(event: WorkflowExecutionInitiatedEvent) -> WorkflowExecutionInitiatedEvent:
     workflow_definition = event.body.workflow_definition
     workflow_display = get_workflow_display(
@@ -16,7 +34,7 @@ def event_enricher(event: WorkflowExecutionInitiatedEvent) -> WorkflowExecutionI
     )
     register_workflow_display_context(event.span_id, workflow_display.display_context)
 
-    if event.body.workflow_definition.is_dynamic:
+    if event.body.workflow_definition.is_dynamic or _should_mark_workflow_dynamic(event):
         register_workflow_display_class(workflow_definition, workflow_display.__class__)
         workflow_version_exec_config = workflow_display.serialize()
         setattr(event.body, "workflow_version_exec_config", workflow_version_exec_config)

{vellum_ai-1.2.4 → vellum_ai-1.2.5}/ee/vellum_ee/workflows/display/utils/tests/test_events.py

@@ -2,8 +2,10 @@ import pytest
 from uuid import uuid4
 from typing import Optional
 
+from vellum.workflows.events.types import NodeParentContext, WorkflowDeploymentParentContext
 from vellum.workflows.events.workflow import WorkflowExecutionInitiatedBody, WorkflowExecutionInitiatedEvent
 from vellum.workflows.inputs.base import BaseInputs
+from vellum.workflows.outputs.base import BaseOutputs
 from vellum.workflows.workflows.base import BaseWorkflow
 from vellum_ee.workflows.display.utils.events import event_enricher
 
@@ -67,3 +69,43 @@ def test_event_enricher_static_workflow(is_dynamic: bool, expected_config: Optio
 
     # AND workflow_version_exec_config is set to the expected config
     assert event.body.workflow_version_exec_config == expected_config
+
+
+def test_event_enricher_marks_subworkflow_deployment_as_dynamic():
+    """Test that event_enricher treats subworkflow deployments as dynamic."""
+
+    class TestWorkflow(BaseWorkflow):
+        is_dynamic = False
+
+        class Outputs(BaseOutputs):
+            pass
+
+    event: WorkflowExecutionInitiatedEvent = WorkflowExecutionInitiatedEvent(
+        trace_id=uuid4(),
+        span_id=uuid4(),
+        parent=WorkflowDeploymentParentContext(
+            span_id=uuid4(),
+            deployment_id=uuid4(),
+            deployment_name="test-deployment",
+            deployment_history_item_id=uuid4(),
+            release_tag_id=uuid4(),
+            release_tag_name="test-tag",
+            workflow_version_id=uuid4(),
+            external_id=None,
+            metadata=None,
+            parent=NodeParentContext(
+                span_id=uuid4(),
+                node_definition=TestWorkflow,
+                parent=None,
+            ),
+        ),
+        body=WorkflowExecutionInitiatedBody(
+            workflow_definition=TestWorkflow,
+            inputs=BaseInputs(),
+        ),
+    )
+
+    enriched_event = event_enricher(event)
+
+    assert hasattr(enriched_event.body, "workflow_version_exec_config")
+    assert enriched_event.body.workflow_version_exec_config is not None

{vellum_ai-1.2.4 → vellum_ai-1.2.5}/pyproject.toml

@@ -3,7 +3,7 @@ name = "vellum-ai"
 
 [tool.poetry]
 name = "vellum-ai"
-version = "1.2.4"
+version = "1.2.5"
 description = ""
 readme = "README.md"
 authors = []

{vellum_ai-1.2.4 → vellum_ai-1.2.5}/src/vellum/__init__.py

@@ -26,6 +26,7 @@ from .client.types import (
     ArrayVellumValueRequest,
     AudioChatMessageContent,
     AudioChatMessageContentRequest,
+    AudioInput,
     AudioInputRequest,
     AudioPromptBlock,
     AudioVellumValue,
@@ -96,6 +97,7 @@ from .client.types import (
     DocumentIndexIndexingConfig,
     DocumentIndexIndexingConfigRequest,
     DocumentIndexRead,
+    DocumentInput,
     DocumentInputRequest,
     DocumentProcessingState,
     DocumentPromptBlock,
@@ -195,6 +197,7 @@ from .client.types import (
     HkunlpInstructorXlVectorizerRequest,
     ImageChatMessageContent,
     ImageChatMessageContentRequest,
+    ImageInput,
     ImageInputRequest,
     ImagePromptBlock,
     ImageVellumValue,
@@ -233,18 +236,28 @@ from .client.types import (
     MlModelRead,
     MlModelUsage,
     MlModelUsageWrapper,
+    NamedScenarioInputAudioVariableValueRequest,
     NamedScenarioInputChatHistoryVariableValueRequest,
+    NamedScenarioInputDocumentVariableValueRequest,
+    NamedScenarioInputImageVariableValueRequest,
     NamedScenarioInputJsonVariableValueRequest,
     NamedScenarioInputRequest,
     NamedScenarioInputStringVariableValueRequest,
+    NamedScenarioInputVideoVariableValueRequest,
     NamedTestCaseArrayVariableValue,
     NamedTestCaseArrayVariableValueRequest,
+    NamedTestCaseAudioVariableValue,
+    NamedTestCaseAudioVariableValueRequest,
     NamedTestCaseChatHistoryVariableValue,
     NamedTestCaseChatHistoryVariableValueRequest,
+    NamedTestCaseDocumentVariableValue,
+    NamedTestCaseDocumentVariableValueRequest,
     NamedTestCaseErrorVariableValue,
     NamedTestCaseErrorVariableValueRequest,
     NamedTestCaseFunctionCallVariableValue,
     NamedTestCaseFunctionCallVariableValueRequest,
+    NamedTestCaseImageVariableValue,
+    NamedTestCaseImageVariableValueRequest,
     NamedTestCaseJsonVariableValue,
     NamedTestCaseJsonVariableValueRequest,
     NamedTestCaseNumberVariableValue,
@@ -255,6 +268,8 @@ from .client.types import (
     NamedTestCaseStringVariableValueRequest,
     NamedTestCaseVariableValue,
     NamedTestCaseVariableValueRequest,
+    NamedTestCaseVideoVariableValue,
+    NamedTestCaseVideoVariableValueRequest,
     NewMemberJoinBehaviorEnum,
     NodeExecutionFulfilledBody,
     NodeExecutionFulfilledEvent,
@@ -375,9 +390,13 @@ from .client.types import (
     RichTextPromptBlock,
     SandboxScenario,
     ScenarioInput,
+    ScenarioInputAudioVariableValue,
     ScenarioInputChatHistoryVariableValue,
+    ScenarioInputDocumentVariableValue,
+    ScenarioInputImageVariableValue,
     ScenarioInputJsonVariableValue,
     ScenarioInputStringVariableValue,
+    ScenarioInputVideoVariableValue,
     SearchFiltersRequest,
     SearchNodeResult,
     SearchNodeResultData,
@@ -444,14 +463,18 @@ from .client.types import (
     TerminalNodeSearchResultsResult,
     TerminalNodeStringResult,
     TestCaseArrayVariableValue,
+    TestCaseAudioVariableValue,
     TestCaseChatHistoryVariableValue,
+    TestCaseDocumentVariableValue,
     TestCaseErrorVariableValue,
     TestCaseFunctionCallVariableValue,
+    TestCaseImageVariableValue,
     TestCaseJsonVariableValue,
     TestCaseNumberVariableValue,
     TestCaseSearchResultsVariableValue,
     TestCaseStringVariableValue,
     TestCaseVariableValue,
+    TestCaseVideoVariableValue,
     TestSuiteRunDeploymentReleaseTagExecConfig,
     TestSuiteRunDeploymentReleaseTagExecConfigData,
     TestSuiteRunDeploymentReleaseTagExecConfigDataRequest,
@@ -553,6 +576,7 @@ from .client.types import (
     VellumWorkflowExecutionEvent,
     VideoChatMessageContent,
     VideoChatMessageContentRequest,
+    VideoInput,
     VideoInputRequest,
     VideoPromptBlock,
     VideoVellumValue,
@@ -703,6 +727,7 @@ __all__ = [
     "AsyncVellum",
     "AudioChatMessageContent",
     "AudioChatMessageContentRequest",
+    "AudioInput",
     "AudioInputRequest",
     "AudioPromptBlock",
     "AudioVellumValue",
@@ -776,6 +801,7 @@ __all__ = [
     "DocumentIndexIndexingConfigRequest",
     "DocumentIndexRead",
     "DocumentIndexesListRequestStatus",
+    "DocumentInput",
     "DocumentInputRequest",
     "DocumentProcessingState",
     "DocumentPromptBlock",
@@ -877,6 +903,7 @@ __all__ = [
     "HkunlpInstructorXlVectorizerRequest",
     "ImageChatMessageContent",
     "ImageChatMessageContentRequest",
+    "ImageInput",
     "ImageInputRequest",
     "ImagePromptBlock",
     "ImageVellumValue",
@@ -919,18 +946,28 @@ __all__ = [
     "MlModelRead",
     "MlModelUsage",
     "MlModelUsageWrapper",
+    "NamedScenarioInputAudioVariableValueRequest",
     "NamedScenarioInputChatHistoryVariableValueRequest",
+    "NamedScenarioInputDocumentVariableValueRequest",
+    "NamedScenarioInputImageVariableValueRequest",
     "NamedScenarioInputJsonVariableValueRequest",
     "NamedScenarioInputRequest",
     "NamedScenarioInputStringVariableValueRequest",
+    "NamedScenarioInputVideoVariableValueRequest",
     "NamedTestCaseArrayVariableValue",
     "NamedTestCaseArrayVariableValueRequest",
+    "NamedTestCaseAudioVariableValue",
+    "NamedTestCaseAudioVariableValueRequest",
     "NamedTestCaseChatHistoryVariableValue",
     "NamedTestCaseChatHistoryVariableValueRequest",
+    "NamedTestCaseDocumentVariableValue",
+    "NamedTestCaseDocumentVariableValueRequest",
     "NamedTestCaseErrorVariableValue",
     "NamedTestCaseErrorVariableValueRequest",
     "NamedTestCaseFunctionCallVariableValue",
     "NamedTestCaseFunctionCallVariableValueRequest",
+    "NamedTestCaseImageVariableValue",
+    "NamedTestCaseImageVariableValueRequest",
     "NamedTestCaseJsonVariableValue",
     "NamedTestCaseJsonVariableValueRequest",
     "NamedTestCaseNumberVariableValue",
@@ -941,6 +978,8 @@ __all__ = [
     "NamedTestCaseStringVariableValueRequest",
     "NamedTestCaseVariableValue",
     "NamedTestCaseVariableValueRequest",
+    "NamedTestCaseVideoVariableValue",
+    "NamedTestCaseVideoVariableValueRequest",
     "NewMemberJoinBehaviorEnum",
     "NodeExecutionFulfilledBody",
     "NodeExecutionFulfilledEvent",
@@ -1062,9 +1101,13 @@ __all__ = [
     "RichTextPromptBlock",
     "SandboxScenario",
     "ScenarioInput",
+    "ScenarioInputAudioVariableValue",
     "ScenarioInputChatHistoryVariableValue",
+    "ScenarioInputDocumentVariableValue",
+    "ScenarioInputImageVariableValue",
     "ScenarioInputJsonVariableValue",
     "ScenarioInputStringVariableValue",
+    "ScenarioInputVideoVariableValue",
     "SearchFiltersRequest",
     "SearchNodeResult",
     "SearchNodeResultData",
@@ -1131,14 +1174,18 @@ __all__ = [
     "TerminalNodeSearchResultsResult",
     "TerminalNodeStringResult",
     "TestCaseArrayVariableValue",
+    "TestCaseAudioVariableValue",
     "TestCaseChatHistoryVariableValue",
+    "TestCaseDocumentVariableValue",
     "TestCaseErrorVariableValue",
     "TestCaseFunctionCallVariableValue",
+    "TestCaseImageVariableValue",
     "TestCaseJsonVariableValue",
     "TestCaseNumberVariableValue",
     "TestCaseSearchResultsVariableValue",
     "TestCaseStringVariableValue",
     "TestCaseVariableValue",
+    "TestCaseVideoVariableValue",
     "TestSuiteRunDeploymentReleaseTagExecConfig",
     "TestSuiteRunDeploymentReleaseTagExecConfigData",
     "TestSuiteRunDeploymentReleaseTagExecConfigDataRequest",
@@ -1244,6 +1291,7 @@ __all__ = [
     "VellumWorkflowExecutionEvent",
     "VideoChatMessageContent",
     "VideoChatMessageContentRequest",
+    "VideoInput",
     "VideoInputRequest",
     "VideoPromptBlock",
     "VideoVellumValue",

{vellum_ai-1.2.4 → vellum_ai-1.2.5}/src/vellum/client/core/client_wrapper.py

@@ -27,10 +27,10 @@ class BaseClientWrapper:
 
     def get_headers(self) -> typing.Dict[str, str]:
         headers: typing.Dict[str, str] = {
-            "User-Agent": "vellum-ai/1.2.4",
+            "User-Agent": "vellum-ai/1.2.5",
             "X-Fern-Language": "Python",
             "X-Fern-SDK-Name": "vellum-ai",
-            "X-Fern-SDK-Version": "1.2.4",
+            "X-Fern-SDK-Version": "1.2.5",
             **(self.get_custom_headers() or {}),
         }
         if self._api_version is not None:

{vellum_ai-1.2.4 → vellum_ai-1.2.5}/src/vellum/client/resources/workflows/client.py

@@ -41,7 +41,9 @@ class WorkflowsClient:
         exclude_display: typing.Optional[bool] = None,
         include_json: typing.Optional[bool] = None,
         include_sandbox: typing.Optional[bool] = None,
+        release_tag: typing.Optional[str] = None,
         strict: typing.Optional[bool] = None,
+        version: typing.Optional[str] = None,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> typing.Iterator[bytes]:
         """
@@ -58,8 +60,14 @@ class WorkflowsClient:
 
         include_sandbox : typing.Optional[bool]
 
+        release_tag : typing.Optional[str]
+            Release tag to use when pulling from deployment (implies deployment-only lookup)
+
         strict : typing.Optional[bool]
 
+        version : typing.Optional[str]
+            Semantic version range to validate against the Workflow SDK version (e.g., '>=1.0.0,<1.2.3')
+
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration. You can pass in configuration such as `chunk_size`, and more to customize the request and response.
 
@@ -74,7 +82,9 @@ class WorkflowsClient:
             exclude_display=exclude_display,
             include_json=include_json,
             include_sandbox=include_sandbox,
+            release_tag=release_tag,
             strict=strict,
+            version=version,
             request_options=request_options,
         ) as r:
             yield from r.data
@@ -150,7 +160,9 @@ class AsyncWorkflowsClient:
         exclude_display: typing.Optional[bool] = None,
         include_json: typing.Optional[bool] = None,
         include_sandbox: typing.Optional[bool] = None,
+        release_tag: typing.Optional[str] = None,
         strict: typing.Optional[bool] = None,
+        version: typing.Optional[str] = None,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> typing.AsyncIterator[bytes]:
         """
@@ -167,8 +179,14 @@ class AsyncWorkflowsClient:
 
         include_sandbox : typing.Optional[bool]
 
+        release_tag : typing.Optional[str]
+            Release tag to use when pulling from deployment (implies deployment-only lookup)
+
         strict : typing.Optional[bool]
 
+        version : typing.Optional[str]
+            Semantic version range to validate against the Workflow SDK version (e.g., '>=1.0.0,<1.2.3')
+
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration. You can pass in configuration such as `chunk_size`, and more to customize the request and response.
 
@@ -183,7 +201,9 @@ class AsyncWorkflowsClient:
             exclude_display=exclude_display,
             include_json=include_json,
             include_sandbox=include_sandbox,
+            release_tag=release_tag,
             strict=strict,
+            version=version,
             request_options=request_options,
         ) as r:
             async for _chunk in r.data:

{vellum_ai-1.2.4 → vellum_ai-1.2.5}/src/vellum/client/resources/workflows/raw_client.py

@@ -33,7 +33,9 @@ class RawWorkflowsClient:
         exclude_display: typing.Optional[bool] = None,
         include_json: typing.Optional[bool] = None,
         include_sandbox: typing.Optional[bool] = None,
+        release_tag: typing.Optional[str] = None,
         strict: typing.Optional[bool] = None,
+        version: typing.Optional[str] = None,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> typing.Iterator[HttpResponse[typing.Iterator[bytes]]]:
         """
@@ -50,8 +52,14 @@ class RawWorkflowsClient:
 
         include_sandbox : typing.Optional[bool]
 
+        release_tag : typing.Optional[str]
+            Release tag to use when pulling from deployment (implies deployment-only lookup)
+
         strict : typing.Optional[bool]
 
+        version : typing.Optional[str]
+            Semantic version range to validate against the Workflow SDK version (e.g., '>=1.0.0,<1.2.3')
+
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration. You can pass in configuration such as `chunk_size`, and more to customize the request and response.
 
@@ -69,7 +77,9 @@ class RawWorkflowsClient:
                 "exclude_display": exclude_display,
                 "include_json": include_json,
                 "include_sandbox": include_sandbox,
+                "release_tag": release_tag,
                 "strict": strict,
+                "version": version,
             },
             request_options=request_options,
         ) as _response:
@@ -185,7 +195,9 @@ class AsyncRawWorkflowsClient:
         exclude_display: typing.Optional[bool] = None,
         include_json: typing.Optional[bool] = None,
         include_sandbox: typing.Optional[bool] = None,
+        release_tag: typing.Optional[str] = None,
         strict: typing.Optional[bool] = None,
+        version: typing.Optional[str] = None,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> typing.AsyncIterator[AsyncHttpResponse[typing.AsyncIterator[bytes]]]:
         """
@@ -202,8 +214,14 @@ class AsyncRawWorkflowsClient:
 
         include_sandbox : typing.Optional[bool]
 
+        release_tag : typing.Optional[str]
+            Release tag to use when pulling from deployment (implies deployment-only lookup)
+
         strict : typing.Optional[bool]
 
+        version : typing.Optional[str]
+            Semantic version range to validate against the Workflow SDK version (e.g., '>=1.0.0,<1.2.3')
+
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration. You can pass in configuration such as `chunk_size`, and more to customize the request and response.
 
@@ -221,7 +239,9 @@ class AsyncRawWorkflowsClient:
                 "exclude_display": exclude_display,
                 "include_json": include_json,
                 "include_sandbox": include_sandbox,
+                "release_tag": release_tag,
                 "strict": strict,
+                "version": version,
             },
             request_options=request_options,
         ) as _response:

{vellum_ai-1.2.4 → vellum_ai-1.2.5}/src/vellum/client/types/__init__.py

@@ -22,6 +22,7 @@ from .array_vellum_value import ArrayVellumValue
 from .array_vellum_value_request import ArrayVellumValueRequest
 from .audio_chat_message_content import AudioChatMessageContent
 from .audio_chat_message_content_request import AudioChatMessageContentRequest
+from .audio_input import AudioInput
 from .audio_input_request import AudioInputRequest
 from .audio_prompt_block import AudioPromptBlock
 from .audio_vellum_value import AudioVellumValue
@@ -100,6 +101,7 @@ from .document_index_chunking_request import DocumentIndexChunkingRequest
 from .document_index_indexing_config import DocumentIndexIndexingConfig
 from .document_index_indexing_config_request import DocumentIndexIndexingConfigRequest
 from .document_index_read import DocumentIndexRead
+from .document_input import DocumentInput
 from .document_input_request import DocumentInputRequest
 from .document_processing_state import DocumentProcessingState
 from .document_prompt_block import DocumentPromptBlock
@@ -203,6 +205,7 @@ from .hkunlp_instructor_xl_vectorizer import HkunlpInstructorXlVectorizer
 from .hkunlp_instructor_xl_vectorizer_request import HkunlpInstructorXlVectorizerRequest
 from .image_chat_message_content import ImageChatMessageContent
 from .image_chat_message_content_request import ImageChatMessageContentRequest
+from .image_input import ImageInput
 from .image_input_request import ImageInputRequest
 from .image_prompt_block import ImagePromptBlock
 from .image_vellum_value import ImageVellumValue
@@ -241,18 +244,28 @@ from .metric_node_result import MetricNodeResult
 from .ml_model_read import MlModelRead
 from .ml_model_usage import MlModelUsage
 from .ml_model_usage_wrapper import MlModelUsageWrapper
+from .named_scenario_input_audio_variable_value_request import NamedScenarioInputAudioVariableValueRequest
 from .named_scenario_input_chat_history_variable_value_request import NamedScenarioInputChatHistoryVariableValueRequest
+from .named_scenario_input_document_variable_value_request import NamedScenarioInputDocumentVariableValueRequest
+from .named_scenario_input_image_variable_value_request import NamedScenarioInputImageVariableValueRequest
 from .named_scenario_input_json_variable_value_request import NamedScenarioInputJsonVariableValueRequest
 from .named_scenario_input_request import NamedScenarioInputRequest
 from .named_scenario_input_string_variable_value_request import NamedScenarioInputStringVariableValueRequest
+from .named_scenario_input_video_variable_value_request import NamedScenarioInputVideoVariableValueRequest
 from .named_test_case_array_variable_value import NamedTestCaseArrayVariableValue
 from .named_test_case_array_variable_value_request import NamedTestCaseArrayVariableValueRequest
+from .named_test_case_audio_variable_value import NamedTestCaseAudioVariableValue
+from .named_test_case_audio_variable_value_request import NamedTestCaseAudioVariableValueRequest
 from .named_test_case_chat_history_variable_value import NamedTestCaseChatHistoryVariableValue
 from .named_test_case_chat_history_variable_value_request import NamedTestCaseChatHistoryVariableValueRequest
+from .named_test_case_document_variable_value import NamedTestCaseDocumentVariableValue
+from .named_test_case_document_variable_value_request import NamedTestCaseDocumentVariableValueRequest
 from .named_test_case_error_variable_value import NamedTestCaseErrorVariableValue
 from .named_test_case_error_variable_value_request import NamedTestCaseErrorVariableValueRequest
 from .named_test_case_function_call_variable_value import NamedTestCaseFunctionCallVariableValue
 from .named_test_case_function_call_variable_value_request import NamedTestCaseFunctionCallVariableValueRequest
+from .named_test_case_image_variable_value import NamedTestCaseImageVariableValue
+from .named_test_case_image_variable_value_request import NamedTestCaseImageVariableValueRequest
 from .named_test_case_json_variable_value import NamedTestCaseJsonVariableValue
 from .named_test_case_json_variable_value_request import NamedTestCaseJsonVariableValueRequest
 from .named_test_case_number_variable_value import NamedTestCaseNumberVariableValue
@@ -263,6 +276,8 @@ from .named_test_case_string_variable_value import NamedTestCaseStringVariableVa
 from .named_test_case_string_variable_value_request import NamedTestCaseStringVariableValueRequest
 from .named_test_case_variable_value import NamedTestCaseVariableValue
 from .named_test_case_variable_value_request import NamedTestCaseVariableValueRequest
+from .named_test_case_video_variable_value import NamedTestCaseVideoVariableValue
+from .named_test_case_video_variable_value_request import NamedTestCaseVideoVariableValueRequest
 from .new_member_join_behavior_enum import NewMemberJoinBehaviorEnum
 from .node_execution_fulfilled_body import NodeExecutionFulfilledBody
 from .node_execution_fulfilled_event import NodeExecutionFulfilledEvent
@@ -383,9 +398,13 @@ from .rich_text_child_block import RichTextChildBlock
 from .rich_text_prompt_block import RichTextPromptBlock
 from .sandbox_scenario import SandboxScenario
 from .scenario_input import ScenarioInput
+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
 from .search_filters_request import SearchFiltersRequest
 from .search_node_result import SearchNodeResult
 from .search_node_result_data import SearchNodeResultData
@@ -452,14 +471,18 @@ from .terminal_node_result_output import TerminalNodeResultOutput
 from .terminal_node_search_results_result import TerminalNodeSearchResultsResult
 from .terminal_node_string_result import TerminalNodeStringResult
 from .test_case_array_variable_value import TestCaseArrayVariableValue
+from .test_case_audio_variable_value import TestCaseAudioVariableValue
 from .test_case_chat_history_variable_value import TestCaseChatHistoryVariableValue
+from .test_case_document_variable_value import TestCaseDocumentVariableValue
 from .test_case_error_variable_value import TestCaseErrorVariableValue
 from .test_case_function_call_variable_value import TestCaseFunctionCallVariableValue
+from .test_case_image_variable_value import TestCaseImageVariableValue
 from .test_case_json_variable_value import TestCaseJsonVariableValue
 from .test_case_number_variable_value import TestCaseNumberVariableValue
 from .test_case_search_results_variable_value import TestCaseSearchResultsVariableValue
 from .test_case_string_variable_value import TestCaseStringVariableValue
 from .test_case_variable_value import TestCaseVariableValue
+from .test_case_video_variable_value import TestCaseVideoVariableValue
 from .test_suite_run_deployment_release_tag_exec_config import TestSuiteRunDeploymentReleaseTagExecConfig
 from .test_suite_run_deployment_release_tag_exec_config_data import TestSuiteRunDeploymentReleaseTagExecConfigData
 from .test_suite_run_deployment_release_tag_exec_config_data_request import (
@@ -577,6 +600,7 @@ from .vellum_video_request import VellumVideoRequest
 from .vellum_workflow_execution_event import VellumWorkflowExecutionEvent
 from .video_chat_message_content import VideoChatMessageContent
 from .video_chat_message_content_request import VideoChatMessageContentRequest
+from .video_input import VideoInput
 from .video_input_request import VideoInputRequest
 from .video_prompt_block import VideoPromptBlock
 from .video_vellum_value import VideoVellumValue
@@ -685,6 +709,7 @@ __all__ = [
     "ArrayVellumValueRequest",
     "AudioChatMessageContent",
     "AudioChatMessageContentRequest",
+    "AudioInput",
     "AudioInputRequest",
     "AudioPromptBlock",
     "AudioVellumValue",
@@ -755,6 +780,7 @@ __all__ = [
     "DocumentIndexIndexingConfig",
     "DocumentIndexIndexingConfigRequest",
     "DocumentIndexRead",
+    "DocumentInput",
     "DocumentInputRequest",
     "DocumentProcessingState",
     "DocumentPromptBlock",
@@ -854,6 +880,7 @@ __all__ = [
     "HkunlpInstructorXlVectorizerRequest",
     "ImageChatMessageContent",
     "ImageChatMessageContentRequest",
+    "ImageInput",
     "ImageInputRequest",
     "ImagePromptBlock",
     "ImageVellumValue",
@@ -892,18 +919,28 @@ __all__ = [
     "MlModelRead",
     "MlModelUsage",
     "MlModelUsageWrapper",
+    "NamedScenarioInputAudioVariableValueRequest",
     "NamedScenarioInputChatHistoryVariableValueRequest",
+    "NamedScenarioInputDocumentVariableValueRequest",
+    "NamedScenarioInputImageVariableValueRequest",
     "NamedScenarioInputJsonVariableValueRequest",
     "NamedScenarioInputRequest",
     "NamedScenarioInputStringVariableValueRequest",
+    "NamedScenarioInputVideoVariableValueRequest",
     "NamedTestCaseArrayVariableValue",
     "NamedTestCaseArrayVariableValueRequest",
+    "NamedTestCaseAudioVariableValue",
+    "NamedTestCaseAudioVariableValueRequest",
     "NamedTestCaseChatHistoryVariableValue",
     "NamedTestCaseChatHistoryVariableValueRequest",
+    "NamedTestCaseDocumentVariableValue",
+    "NamedTestCaseDocumentVariableValueRequest",
     "NamedTestCaseErrorVariableValue",
     "NamedTestCaseErrorVariableValueRequest",
     "NamedTestCaseFunctionCallVariableValue",
     "NamedTestCaseFunctionCallVariableValueRequest",
+    "NamedTestCaseImageVariableValue",
+    "NamedTestCaseImageVariableValueRequest",
     "NamedTestCaseJsonVariableValue",
     "NamedTestCaseJsonVariableValueRequest",
     "NamedTestCaseNumberVariableValue",
@@ -914,6 +951,8 @@ __all__ = [
     "NamedTestCaseStringVariableValueRequest",
     "NamedTestCaseVariableValue",
     "NamedTestCaseVariableValueRequest",
+    "NamedTestCaseVideoVariableValue",
+    "NamedTestCaseVideoVariableValueRequest",
     "NewMemberJoinBehaviorEnum",
     "NodeExecutionFulfilledBody",
     "NodeExecutionFulfilledEvent",
@@ -1034,9 +1073,13 @@ __all__ = [
     "RichTextPromptBlock",
     "SandboxScenario",
     "ScenarioInput",
+    "ScenarioInputAudioVariableValue",
     "ScenarioInputChatHistoryVariableValue",
+    "ScenarioInputDocumentVariableValue",
+    "ScenarioInputImageVariableValue",
     "ScenarioInputJsonVariableValue",
     "ScenarioInputStringVariableValue",
+    "ScenarioInputVideoVariableValue",
     "SearchFiltersRequest",
     "SearchNodeResult",
     "SearchNodeResultData",
@@ -1103,14 +1146,18 @@ __all__ = [
     "TerminalNodeSearchResultsResult",
     "TerminalNodeStringResult",
     "TestCaseArrayVariableValue",
+    "TestCaseAudioVariableValue",
     "TestCaseChatHistoryVariableValue",
+    "TestCaseDocumentVariableValue",
     "TestCaseErrorVariableValue",
     "TestCaseFunctionCallVariableValue",
+    "TestCaseImageVariableValue",
     "TestCaseJsonVariableValue",
     "TestCaseNumberVariableValue",
     "TestCaseSearchResultsVariableValue",
     "TestCaseStringVariableValue",
     "TestCaseVariableValue",
+    "TestCaseVideoVariableValue",
     "TestSuiteRunDeploymentReleaseTagExecConfig",
     "TestSuiteRunDeploymentReleaseTagExecConfigData",
     "TestSuiteRunDeploymentReleaseTagExecConfigDataRequest",
@@ -1212,6 +1259,7 @@ __all__ = [
     "VellumWorkflowExecutionEvent",
     "VideoChatMessageContent",
     "VideoChatMessageContentRequest",
+    "VideoInput",
     "VideoInputRequest",
     "VideoPromptBlock",
     "VideoVellumValue",