vellum-ai 0.13.9__py3-none-any.whl → 0.13.11__py3-none-any.whl

vellum/client/core/client_wrapper.py

@@ -18,7 +18,7 @@ class BaseClientWrapper:
         headers: typing.Dict[str, str] = {
             "X-Fern-Language": "Python",
             "X-Fern-SDK-Name": "vellum-ai",
-            "X-Fern-SDK-Version": "0.13.9",
+            "X-Fern-SDK-Version": "0.13.11",
         }
         headers["X_API_KEY"] = self.api_key
         return headers

vellum/workflows/descriptors/utils.py

@@ -108,7 +108,7 @@ def is_unresolved(value: Any) -> bool:
     if isinstance(value, Mapping):
         return any(is_unresolved(item) for item in value.values())
 
-    if isinstance(value, Sequence):
+    if isinstance(value, Sequence) and not isinstance(value, str):
         return any(is_unresolved(item) for item in value)
 
     if isinstance(value, Set):

vellum/workflows/errors/types.py

@@ -61,3 +61,24 @@ def workflow_event_error_to_workflow_error(error: WorkflowEventError) -> Workflo
         message=error.message,
         code=_WORKFLOW_EVENT_ERROR_CODE_TO_WORKFLOW_ERROR_CODE.get(error.code, WorkflowErrorCode.INTERNAL_ERROR),
     )
+
+
+_WORKFLOW_ERROR_CODE_TO_VELLUM_ERROR_CODE: Dict[WorkflowErrorCode, VellumErrorCodeEnum] = {
+    WorkflowErrorCode.INVALID_WORKFLOW: "INVALID_REQUEST",
+    WorkflowErrorCode.INVALID_INPUTS: "INVALID_INPUTS",
+    WorkflowErrorCode.INVALID_OUTPUTS: "INVALID_REQUEST",
+    WorkflowErrorCode.INVALID_STATE: "INVALID_REQUEST",
+    WorkflowErrorCode.INVALID_TEMPLATE: "INVALID_INPUTS",
+    WorkflowErrorCode.INTERNAL_ERROR: "INTERNAL_SERVER_ERROR",
+    WorkflowErrorCode.NODE_EXECUTION: "USER_DEFINED_ERROR",
+    WorkflowErrorCode.PROVIDER_ERROR: "PROVIDER_ERROR",
+    WorkflowErrorCode.USER_DEFINED_ERROR: "USER_DEFINED_ERROR",
+    WorkflowErrorCode.WORKFLOW_CANCELLED: "REQUEST_TIMEOUT",
+}
+
+
+def workflow_error_to_vellum_error(error: WorkflowError) -> VellumError:
+    return VellumError(
+        message=error.message,
+        code=_WORKFLOW_ERROR_CODE_TO_VELLUM_ERROR_CODE.get(error.code, "INTERNAL_SERVER_ERROR"),
+    )

vellum/workflows/nodes/bases/base.py

@@ -237,7 +237,7 @@ class BaseNode(Generic[StateType], metaclass=BaseNodeMeta):
 
     class Trigger(metaclass=_BaseNodeTriggerMeta):
         node_class: Type["BaseNode"]
-        merge_behavior = MergeBehavior.AWAIT_ANY
+        merge_behavior = MergeBehavior.AWAIT_ATTRIBUTES
 
         @classmethod
         def should_initiate(

vellum/workflows/nodes/displayable/api_node/node.py

@@ -2,7 +2,7 @@ from typing import Optional, Union
 
 from vellum.workflows.constants import AuthorizationType
 from vellum.workflows.nodes.displayable.bases.api_node import BaseAPINode
-from vellum.workflows.types.core import VellumSecret
+from vellum.workflows.types.core import MergeBehavior, VellumSecret
 
 
 class APINode(BaseAPINode):
@@ -27,6 +27,9 @@ class APINode(BaseAPINode):
     api_key_header_value: Optional[Union[str, VellumSecret]] = None
     bearer_token_value: Optional[Union[str, VellumSecret]] = None
 
+    class Trigger(BaseAPINode.Trigger):
+        merge_behavior = MergeBehavior.AWAIT_ANY
+
     def run(self) -> BaseAPINode.Outputs:
         headers = self.headers or {}
         header_overrides = {}

vellum/workflows/nodes/displayable/bases/api_node/node.py

@@ -8,7 +8,7 @@ from vellum.workflows.errors.types import WorkflowErrorCode
 from vellum.workflows.exceptions import NodeException
 from vellum.workflows.nodes.bases import BaseNode
 from vellum.workflows.outputs import BaseOutputs
-from vellum.workflows.types.core import Json, VellumSecret
+from vellum.workflows.types.core import Json, MergeBehavior, VellumSecret
 from vellum.workflows.types.generics import StateType
 
 
@@ -23,6 +23,9 @@ class BaseAPINode(BaseNode, Generic[StateType]):
     headers: Optional[Dict[str, Union[str, VellumSecret]]] - The headers to send in the request.
     """
 
+    class Trigger(BaseNode.Trigger):
+        merge_behavior = MergeBehavior.AWAIT_ANY
+
     url: str
     method: APIRequestMethod
     data: Optional[str] = None

vellum/workflows/nodes/displayable/bases/base_prompt_node/node.py

@@ -2,12 +2,13 @@ from abc import abstractmethod
 from typing import ClassVar, Generator, Generic, Iterator, List, Optional, Union
 
 from vellum import AdHocExecutePromptEvent, ExecutePromptEvent, PromptOutput
+from vellum.client.core.api_error import ApiError
 from vellum.core import RequestOptions
 from vellum.workflows.errors.types import WorkflowErrorCode, vellum_error_to_workflow_error
 from vellum.workflows.exceptions import NodeException
 from vellum.workflows.nodes.bases import BaseNode
 from vellum.workflows.outputs.base import BaseOutput, BaseOutputs
-from vellum.workflows.types.core import EntityInputsInterface
+from vellum.workflows.types.core import EntityInputsInterface, MergeBehavior
 from vellum.workflows.types.generics import StateType
 
 
@@ -17,6 +18,9 @@ class BasePromptNode(BaseNode, Generic[StateType]):
 
     request_options: Optional[RequestOptions] = None
 
+    class Trigger(BaseNode.Trigger):
+        merge_behavior = MergeBehavior.AWAIT_ANY
+
     class Outputs(BaseOutputs):
         results: List[PromptOutput]
 
@@ -33,7 +37,19 @@ class BasePromptNode(BaseNode, Generic[StateType]):
             )
 
     def _process_prompt_event_stream(self) -> Generator[BaseOutput, None, Optional[List[PromptOutput]]]:
-        prompt_event_stream = self._get_prompt_event_stream()
+        try:
+            prompt_event_stream = self._get_prompt_event_stream()
+        except ApiError as e:
+            if e.status_code and e.status_code >= 400 and e.status_code < 500 and isinstance(e.body, dict):
+                raise NodeException(
+                    message=e.body.get("detail", "Failed to execute prompt"),
+                    code=WorkflowErrorCode.INVALID_INPUTS,
+                ) from e
+
+            raise NodeException(
+                message="Failed to execute prompt",
+                code=WorkflowErrorCode.INTERNAL_ERROR,
+            ) from e
 
         outputs: Optional[List[PromptOutput]] = None
         for event in prompt_event_stream:

vellum/workflows/nodes/displayable/bases/inline_prompt_node/node.py

@@ -23,6 +23,7 @@ from vellum.workflows.events.types import default_serializer
 from vellum.workflows.exceptions import NodeException
 from vellum.workflows.nodes.displayable.bases.base_prompt_node import BasePromptNode
 from vellum.workflows.nodes.displayable.bases.inline_prompt_node.constants import DEFAULT_PROMPT_PARAMETERS
+from vellum.workflows.types import MergeBehavior
 from vellum.workflows.types.generics import StateType
 from vellum.workflows.utils.functions import compile_function_definition
 
@@ -51,6 +52,9 @@ class BaseInlinePromptNode(BasePromptNode[StateType], Generic[StateType]):
     parameters: PromptParameters = DEFAULT_PROMPT_PARAMETERS
     expand_meta: Optional[AdHocExpandMeta] = OMIT
 
+    class Trigger(BasePromptNode.Trigger):
+        merge_behavior = MergeBehavior.AWAIT_ANY
+
     def _get_prompt_event_stream(self) -> Iterator[AdHocExecutePromptEvent]:
         input_variables, input_values = self._compile_prompt_inputs()
         parent_context = get_parent_context()

vellum/workflows/nodes/displayable/bases/prompt_deployment_node.py

@@ -17,6 +17,7 @@ from vellum.workflows.context import get_parent_context
 from vellum.workflows.errors import WorkflowErrorCode
 from vellum.workflows.exceptions import NodeException
 from vellum.workflows.nodes.displayable.bases.base_prompt_node import BasePromptNode
+from vellum.workflows.types import MergeBehavior
 from vellum.workflows.types.generics import StateType
 
 
@@ -47,6 +48,9 @@ class BasePromptDeploymentNode(BasePromptNode, Generic[StateType]):
     expand_raw: Optional[Sequence[str]] = OMIT
     metadata: Optional[Dict[str, Optional[Any]]] = OMIT
 
+    class Trigger(BasePromptNode.Trigger):
+        merge_behavior = MergeBehavior.AWAIT_ANY
+
     def _get_prompt_event_stream(self) -> Iterator[ExecutePromptEvent]:
         current_parent_context = get_parent_context()
         parent_context = current_parent_context.model_dump() if current_parent_context else None

vellum/workflows/nodes/displayable/bases/search_node.py

@@ -17,6 +17,7 @@ from vellum.workflows.exceptions import NodeException
 from vellum.workflows.nodes.bases import BaseNode
 from vellum.workflows.nodes.displayable.bases.types import SearchFilters
 from vellum.workflows.outputs import BaseOutputs
+from vellum.workflows.types import MergeBehavior
 from vellum.workflows.types.generics import StateType
 
 DEFAULT_SEARCH_WEIGHTS = 0.8
@@ -81,6 +82,9 @@ class BaseSearchNode(BaseNode[StateType], Generic[StateType]):
 
     request_options: Optional[RequestOptions] = None
 
+    class Trigger(BaseNode.Trigger):
+        merge_behavior = MergeBehavior.AWAIT_ANY
+
     class Outputs(BaseOutputs):
         """
         The outputs of the SearchNode.

vellum/workflows/nodes/displayable/bases/tests/test_utils.py

@@ -1,8 +1,11 @@
 import pytest
 import enum
 
+from pydantic import BaseModel
+
 from vellum.client.types.chat_history_vellum_value import ChatHistoryVellumValue
 from vellum.client.types.chat_message import ChatMessage
+from vellum.client.types.error_vellum_value import ErrorVellumValue
 from vellum.client.types.json_vellum_value import JsonVellumValue
 from vellum.client.types.number_vellum_value import NumberVellumValue
 from vellum.client.types.search_result import SearchResult
@@ -10,6 +13,8 @@ from vellum.client.types.search_result_document import SearchResultDocument
 from vellum.client.types.search_results_vellum_value import SearchResultsVellumValue
 from vellum.client.types.string_vellum_value import StringVellumValue
 from vellum.client.types.string_vellum_value_request import StringVellumValueRequest
+from vellum.client.types.vellum_error import VellumError
+from vellum.workflows.errors.types import WorkflowError, WorkflowErrorCode
 from vellum.workflows.nodes.displayable.bases.utils import primitive_to_vellum_value, primitive_to_vellum_value_request
 
 
@@ -17,6 +22,10 @@ class MockEnum(enum.Enum):
     FOO = "foo"
 
 
+class RandomPydanticModel(BaseModel):
+    foo: str
+
+
 @pytest.mark.parametrize(
     ["value", "expected_output"],
     [
@@ -51,6 +60,15 @@ class MockEnum(enum.Enum):
         (StringVellumValue(value="hello"), StringVellumValue(value="hello")),
         (StringVellumValueRequest(value="hello"), StringVellumValueRequest(value="hello")),
         ({"foo": "bar"}, JsonVellumValue(value={"foo": "bar"})),
+        (
+            VellumError(message="hello", code="USER_DEFINED_ERROR"),
+            ErrorVellumValue(value=VellumError(message="hello", code="USER_DEFINED_ERROR")),
+        ),
+        (
+            WorkflowError(message="hello", code=WorkflowErrorCode.USER_DEFINED_ERROR),
+            ErrorVellumValue(value=VellumError(message="hello", code="USER_DEFINED_ERROR")),
+        ),
+        (RandomPydanticModel(foo="bar"), JsonVellumValue(value={"foo": "bar"})),
     ],
 )
 def test_primitive_to_vellum_value(value, expected_output):

vellum/workflows/nodes/displayable/bases/utils.py

@@ -25,8 +25,11 @@ from vellum.client.types.search_results_vellum_value import SearchResultsVellumV
 from vellum.client.types.search_results_vellum_value_request import SearchResultsVellumValueRequest
 from vellum.client.types.string_vellum_value import StringVellumValue
 from vellum.client.types.string_vellum_value_request import StringVellumValueRequest
+from vellum.client.types.vellum_error import VellumError
 from vellum.client.types.vellum_value import VellumValue
 from vellum.client.types.vellum_value_request import VellumValueRequest
+from vellum.workflows.errors.types import WorkflowError, workflow_error_to_vellum_error
+from vellum.workflows.state.encoder import DefaultStateEncoder
 
 VELLUM_VALUE_REQUEST_TUPLE = (
     StringVellumValueRequest,
@@ -63,6 +66,10 @@ def primitive_to_vellum_value(value: Any) -> VellumValue:
     ):
         search_results = cast(Union[List[SearchResultRequest], List[SearchResult]], value)
         return SearchResultsVellumValue(value=search_results)
+    elif isinstance(value, VellumError):
+        return ErrorVellumValue(value=value)
+    elif isinstance(value, WorkflowError):
+        return ErrorVellumValue(value=workflow_error_to_vellum_error(value))
     elif isinstance(
         value,
         (
@@ -88,7 +95,7 @@ def primitive_to_vellum_value(value: Any) -> VellumValue:
         return value  # type: ignore
 
     try:
-        json_value = json.dumps(value)
+        json_value = json.dumps(value, cls=DefaultStateEncoder)
     except json.JSONDecodeError:
         raise ValueError(f"Unsupported variable type: {value.__class__.__name__}")
 

vellum/workflows/nodes/displayable/code_execution_node/node.py

@@ -26,7 +26,7 @@ from vellum.workflows.nodes.bases import BaseNode
 from vellum.workflows.nodes.bases.base import BaseNodeMeta
 from vellum.workflows.nodes.displayable.code_execution_node.utils import read_file_from_path
 from vellum.workflows.outputs.base import BaseOutputs
-from vellum.workflows.types.core import EntityInputsInterface, VellumSecret
+from vellum.workflows.types.core import EntityInputsInterface, MergeBehavior, VellumSecret
 from vellum.workflows.types.generics import StateType
 from vellum.workflows.types.utils import get_original_base
 from vellum.workflows.utils.vellum_variables import primitive_type_to_vellum_variable_type
@@ -82,6 +82,9 @@ class CodeExecutionNode(BaseNode[StateType], Generic[StateType, _OutputType], me
 
     request_options: Optional[RequestOptions] = None
 
+    class Trigger(BaseNode.Trigger):
+        merge_behavior = MergeBehavior.AWAIT_ANY
+
     class Outputs(BaseOutputs):
         # We use our mypy plugin to override the _OutputType with the actual output type
         # for downstream references to this output.

vellum/workflows/nodes/displayable/conditional_node/node.py

@@ -6,6 +6,7 @@ from vellum.workflows.ports.node_ports import NodePorts
 from vellum.workflows.ports.port import Port
 from vellum.workflows.ports.utils import validate_ports
 from vellum.workflows.state.base import BaseState
+from vellum.workflows.types import MergeBehavior
 
 
 class ConditionalNode(BaseNode):
@@ -14,6 +15,9 @@ class ConditionalNode(BaseNode):
     Vellum's Conditional Node, and for most cases, you should extend `BaseNode.Ports` directly.
     """
 
+    class Trigger(BaseNode.Trigger):
+        merge_behavior = MergeBehavior.AWAIT_ANY
+
     class Ports(NodePorts):
         def __call__(self, outputs: BaseOutputs, state: BaseState) -> Set[Port]:
             all_ports = [port for port in self.__class__]

vellum/workflows/nodes/displayable/final_output_node/node.py

@@ -2,6 +2,7 @@ from typing import Any, Dict, Generic, Tuple, Type, TypeVar, get_args
 
 from vellum.workflows.nodes.bases import BaseNode
 from vellum.workflows.nodes.bases.base import BaseNodeMeta
+from vellum.workflows.types import MergeBehavior
 from vellum.workflows.types.generics import StateType
 from vellum.workflows.types.utils import get_original_base
 
@@ -41,6 +42,9 @@ class FinalOutputNode(BaseNode[StateType], Generic[StateType, _OutputType], meta
     This provides backward compatibility with Vellum's Final Output Node.
     """
 
+    class Trigger(BaseNode.Trigger):
+        merge_behavior = MergeBehavior.AWAIT_ANY
+
     class Outputs(BaseNode.Outputs):
         # We use our mypy plugin to override the _OutputType with the actual output type
         # for downstream references to this output.

vellum/workflows/nodes/displayable/guardrail_node/node.py

@@ -8,7 +8,7 @@ from vellum.workflows.errors.types import WorkflowErrorCode
 from vellum.workflows.exceptions import NodeException
 from vellum.workflows.nodes.bases import BaseNode
 from vellum.workflows.outputs.base import BaseOutputs
-from vellum.workflows.types.core import EntityInputsInterface
+from vellum.workflows.types.core import EntityInputsInterface, MergeBehavior
 from vellum.workflows.types.generics import StateType
 
 
@@ -29,6 +29,9 @@ class GuardrailNode(BaseNode[StateType], Generic[StateType]):
 
     request_options: Optional[RequestOptions] = None
 
+    class Trigger(BaseNode.Trigger):
+        merge_behavior = MergeBehavior.AWAIT_ANY
+
     class Outputs(BaseOutputs):
         score: float
 

vellum/workflows/nodes/displayable/inline_prompt_node/node.py

@@ -5,6 +5,7 @@ from vellum.workflows.errors import WorkflowErrorCode
 from vellum.workflows.exceptions import NodeException
 from vellum.workflows.nodes.displayable.bases import BaseInlinePromptNode as BaseInlinePromptNode
 from vellum.workflows.outputs import BaseOutput
+from vellum.workflows.types import MergeBehavior
 from vellum.workflows.types.generics import StateType
 
 
@@ -21,6 +22,9 @@ class InlinePromptNode(BaseInlinePromptNode[StateType]):
     request_options: Optional[RequestOptions] - The request options to use for the Prompt Execution
     """
 
+    class Trigger(BaseInlinePromptNode.Trigger):
+        merge_behavior = MergeBehavior.AWAIT_ANY
+
     class Outputs(BaseInlinePromptNode.Outputs):
         """
         The outputs of the InlinePromptNode.

vellum/workflows/nodes/displayable/inline_prompt_node/tests/test_node.py

@@ -1,7 +1,9 @@
+import pytest
 from dataclasses import dataclass
 from uuid import uuid4
 from typing import Any, Iterator, List
 
+from vellum.client.core.api_error import ApiError
 from vellum.client.core.pydantic_utilities import UniversalBaseModel
 from vellum.client.types.execute_prompt_event import ExecutePromptEvent
 from vellum.client.types.fulfilled_execute_prompt_event import FulfilledExecutePromptEvent
@@ -12,6 +14,8 @@ from vellum.client.types.initiated_execute_prompt_event import InitiatedExecuteP
 from vellum.client.types.prompt_output import PromptOutput
 from vellum.client.types.prompt_request_json_input import PromptRequestJsonInput
 from vellum.client.types.string_vellum_value import StringVellumValue
+from vellum.workflows.errors.types import WorkflowErrorCode
+from vellum.workflows.exceptions import NodeException
 from vellum.workflows.nodes.displayable.inline_prompt_node.node import InlinePromptNode
 
 
@@ -126,3 +130,54 @@ def test_inline_prompt_node__function_definitions(vellum_adhoc_prompt_client):
     "name": "my_function"
 }"""
     )
+
+
+@pytest.mark.parametrize(
+    ["exception", "expected_code", "expected_message"],
+    [
+        (
+            ApiError(status_code=404, body={"detail": "Model not found"}),
+            WorkflowErrorCode.INVALID_INPUTS,
+            "Model not found",
+        ),
+        (
+            ApiError(status_code=404, body={"message": "Model not found"}),
+            WorkflowErrorCode.INVALID_INPUTS,
+            "Failed to execute prompt",
+        ),
+        (
+            ApiError(status_code=404, body="Model not found"),
+            WorkflowErrorCode.INTERNAL_ERROR,
+            "Failed to execute prompt",
+        ),
+        (
+            ApiError(status_code=None, body={"detail": "Model not found"}),
+            WorkflowErrorCode.INTERNAL_ERROR,
+            "Failed to execute prompt",
+        ),
+        (
+            ApiError(status_code=500, body={"detail": "Model not found"}),
+            WorkflowErrorCode.INTERNAL_ERROR,
+            "Failed to execute prompt",
+        ),
+    ],
+    ids=["404", "invalid_dict", "invalid_body", "no_status_code", "500"],
+)
+def test_inline_prompt_node__api_error__invalid_inputs_node_exception(
+    vellum_adhoc_prompt_client, exception, expected_code, expected_message
+):
+    # GIVEN a prompt node with an invalid model name
+    class MyNode(InlinePromptNode):
+        ml_model = "my-invalid-model"
+        blocks = []
+
+    # AND the adhoc prompt client raises a 4xx error
+    vellum_adhoc_prompt_client.adhoc_execute_prompt_stream.side_effect = exception
+
+    # WHEN the node is run
+    with pytest.raises(NodeException) as e:
+        list(MyNode().run())
+
+    # THEN the node raises the correct NodeException
+    assert e.value.code == expected_code
+    assert e.value.message == expected_message

vellum/workflows/nodes/displayable/merge_node/node.py

@@ -1,4 +1,5 @@
 from vellum.workflows.nodes.bases import BaseNode
+from vellum.workflows.types import MergeBehavior
 
 
 class MergeNode(BaseNode):
@@ -7,4 +8,5 @@ class MergeNode(BaseNode):
     with Vellum's Merge Node, and for most cases, you should extend from `BaseNode.Trigger` directly.
     """
 
-    pass
+    class Trigger(BaseNode.Trigger):
+        merge_behavior = MergeBehavior.AWAIT_ANY

vellum/workflows/nodes/displayable/note_node/node.py

@@ -1,4 +1,5 @@
 from vellum.workflows.nodes.bases import BaseNode
+from vellum.workflows.types import MergeBehavior
 
 
 class NoteNode(BaseNode):
@@ -6,5 +7,8 @@ class NoteNode(BaseNode):
     A no-op Node purely used to display a note in the Vellum UI.
     """
 
+    class Trigger(BaseNode.Trigger):
+        merge_behavior = MergeBehavior.AWAIT_ANY
+
     def run(self) -> BaseNode.Outputs:
         raise RuntimeError("NoteNode should never be run")

vellum/workflows/nodes/displayable/prompt_deployment_node/node.py

@@ -4,6 +4,7 @@ from vellum.workflows.errors import WorkflowErrorCode
 from vellum.workflows.exceptions import NodeException
 from vellum.workflows.nodes.displayable.bases import BasePromptDeploymentNode as BasePromptDeploymentNode
 from vellum.workflows.outputs import BaseOutput
+from vellum.workflows.types import MergeBehavior
 from vellum.workflows.types.generics import StateType
 
 
@@ -23,6 +24,9 @@ class PromptDeploymentNode(BasePromptDeploymentNode[StateType]):
     request_options: Optional[RequestOptions] - The request options to use for the Prompt Execution
     """
 
+    class Trigger(BasePromptDeploymentNode.Trigger):
+        merge_behavior = MergeBehavior.AWAIT_ANY
+
     class Outputs(BasePromptDeploymentNode.Outputs):
         """
         The outputs of the PromptDeploymentNode.

vellum/workflows/nodes/displayable/search_node/node.py

@@ -1,6 +1,7 @@
 from typing import ClassVar
 
 from vellum.workflows.nodes.displayable.bases import BaseSearchNode as BaseSearchNode
+from vellum.workflows.types import MergeBehavior
 from vellum.workflows.types.generics import StateType
 
 
@@ -18,6 +19,9 @@ class SearchNode(BaseSearchNode[StateType]):
 
     chunk_separator: ClassVar[str] = "\n\n#####\n\n"
 
+    class Trigger(BaseSearchNode.Trigger):
+        merge_behavior = MergeBehavior.AWAIT_ANY
+
     class Outputs(BaseSearchNode.Outputs):
         """
         The outputs of the SearchNode.

vellum/workflows/nodes/displayable/subworkflow_deployment_node/node.py

@@ -19,7 +19,7 @@ from vellum.workflows.errors.types import workflow_event_error_to_workflow_error
 from vellum.workflows.exceptions import NodeException
 from vellum.workflows.nodes.bases.base import BaseNode
 from vellum.workflows.outputs.base import BaseOutput
-from vellum.workflows.types.core import EntityInputsInterface
+from vellum.workflows.types.core import EntityInputsInterface, MergeBehavior
 from vellum.workflows.types.generics import StateType
 
 
@@ -49,6 +49,9 @@ class SubworkflowDeploymentNode(BaseNode[StateType], Generic[StateType]):
 
     request_options: Optional[RequestOptions] = None
 
+    class Trigger(BaseNode.Trigger):
+        merge_behavior = MergeBehavior.AWAIT_ANY
+
     def _compile_subworkflow_inputs(self) -> List[WorkflowRequestInputRequest]:
         # TODO: We may want to consolidate with prompt deployment input compilation
         # https://app.shortcut.com/vellum/story/4117

{vellum_ai-0.13.9.dist-info → vellum_ai-0.13.11.dist-info}/METADATA

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