vellum-ai 0.10.9__py3-none-any.whl → 0.11.1__py3-none-any.whl

vellum/workflows/events/workflow.py

@@ -6,7 +6,7 @@ from vellum.core.pydantic_utilities import UniversalBaseModel
 from vellum.workflows.errors import VellumError
 from vellum.workflows.outputs.base import BaseOutput
 from vellum.workflows.references import ExternalInputReference
-from vellum.workflows.types.generics import OutputsType, WorkflowInputsType
+from vellum.workflows.types.generics import OutputsType, StateType, WorkflowInputsType
 
 from .node import (
     NodeExecutionFulfilledEvent,
@@ -124,6 +124,23 @@ class WorkflowExecutionResumedEvent(_BaseWorkflowEvent):
     body: WorkflowExecutionResumedBody
 
 
+class WorkflowExecutionSnapshottedBody(_BaseWorkflowExecutionBody, Generic[StateType]):
+    state: StateType
+
+    @field_serializer("state")
+    def serialize_state(self, state: StateType, _info: Any) -> Dict[str, Any]:
+        return default_serializer(state)
+
+
+class WorkflowExecutionSnapshottedEvent(_BaseWorkflowEvent, Generic[StateType]):
+    name: Literal["workflow.execution.snapshotted"] = "workflow.execution.snapshotted"
+    body: WorkflowExecutionSnapshottedBody[StateType]
+
+    @property
+    def state(self) -> StateType:
+        return self.body.state
+
+
 GenericWorkflowEvent = Union[
     WorkflowExecutionStreamingEvent,
     WorkflowExecutionRejectedEvent,
@@ -141,6 +158,7 @@ WorkflowEvent = Union[
     GenericWorkflowEvent,
     WorkflowExecutionInitiatedEvent,
     WorkflowExecutionFulfilledEvent,
+    WorkflowExecutionSnapshottedEvent,
 ]
 
 WorkflowEventStream = Generator[WorkflowEvent, None, None]

vellum/workflows/nodes/bases/base.py

@@ -7,9 +7,7 @@ from typing import Any, Dict, Generic, Iterator, Optional, Set, Tuple, Type, Typ
 from vellum.workflows.constants import UNDEF
 from vellum.workflows.descriptors.base import BaseDescriptor
 from vellum.workflows.descriptors.utils import is_unresolved, resolve_value
-from vellum.workflows.edges.edge import Edge
 from vellum.workflows.errors.types import VellumErrorCode
-from vellum.workflows.events.types import ParentContext
 from vellum.workflows.exceptions import NodeException
 from vellum.workflows.graph import Graph
 from vellum.workflows.graph.graph import GraphTarget
@@ -32,15 +30,9 @@ def is_nested_class(nested: Any, parent: Type) -> bool:
         inspect.isclass(nested)
         # If a class is defined within a function, we don't consider it nested in the class defining that function
         # The example of this is a Subworkflow defined within TryNode.wrap()
-        and (
-            len(nested.__qualname__.split(".")) < 2
-            or nested.__qualname__.split(".")[-2] != "<locals>"
-        )
+        and (len(nested.__qualname__.split(".")) < 2 or nested.__qualname__.split(".")[-2] != "<locals>")
         and nested.__module__ == parent.__module__
-        and (
-            nested.__qualname__.startswith(parent.__name__)
-            or nested.__qualname__.startswith(parent.__qualname__)
-        )
+        and (nested.__qualname__.startswith(parent.__name__) or nested.__qualname__.startswith(parent.__qualname__))
     ) or any(is_nested_class(nested, base) for base in parent.__bases__)
 
 
@@ -92,9 +84,7 @@ class BaseNodeMeta(type):
                         **base.Trigger.__dict__,
                         "__module__": dct["__module__"],
                     }
-                    dct["Trigger"] = type(
-                        f"{name}.Trigger", (base.Trigger,), trigger_dct
-                    )
+                    dct["Trigger"] = type(f"{name}.Trigger", (base.Trigger,), trigger_dct)
                     break
 
         cls = super().__new__(mcs, name, bases, dct)
@@ -139,9 +129,7 @@ class BaseNodeMeta(type):
 
     def __rshift__(cls, other_cls: GraphTarget) -> Graph:
         if not issubclass(cls, BaseNode):
-            raise ValueError(
-                "BaseNodeMeta can only be extended from subclasses of BaseNode"
-            )
+            raise ValueError("BaseNodeMeta can only be extended from subclasses of BaseNode")
 
         if not cls.Ports._default_port:
             raise ValueError("No default port found on node")
@@ -153,9 +141,7 @@ class BaseNodeMeta(type):
 
     def __rrshift__(cls, other_cls: GraphTarget) -> Graph:
         if not issubclass(cls, BaseNode):
-            raise ValueError(
-                "BaseNodeMeta can only be extended from subclasses of BaseNode"
-            )
+            raise ValueError("BaseNodeMeta can only be extended from subclasses of BaseNode")
 
         if not isinstance(other_cls, set):
             other_cls = {other_cls}
@@ -193,18 +179,13 @@ class _BaseNodeTriggerMeta(type):
         if not isinstance(other, _BaseNodeTriggerMeta):
             return False
 
-        if not self.__name__.endswith(".Trigger") or not other.__name__.endswith(
-            ".Trigger"
-        ):
+        if not self.__name__.endswith(".Trigger") or not other.__name__.endswith(".Trigger"):
             return super().__eq__(other)
 
         self_trigger_class = cast(Type["BaseNode.Trigger"], self)
         other_trigger_class = cast(Type["BaseNode.Trigger"], other)
 
-        return (
-            self_trigger_class.node_class.__name__
-            == other_trigger_class.node_class.__name__
-        )
+        return self_trigger_class.node_class.__name__ == other_trigger_class.node_class.__name__
 
 
 class _BaseNodeExecutionMeta(type):
@@ -222,18 +203,13 @@ class _BaseNodeExecutionMeta(type):
         if not isinstance(other, _BaseNodeExecutionMeta):
             return False
 
-        if not self.__name__.endswith(".Execution") or not other.__name__.endswith(
-            ".Execution"
-        ):
+        if not self.__name__.endswith(".Execution") or not other.__name__.endswith(".Execution"):
             return super().__eq__(other)
 
         self_execution_class = cast(Type["BaseNode.Execution"], self)
         other_execution_class = cast(Type["BaseNode.Execution"], other)
 
-        return (
-            self_execution_class.node_class.__name__
-            == other_execution_class.node_class.__name__
-        )
+        return self_execution_class.node_class.__name__ == other_execution_class.node_class.__name__
 
 
 class BaseNode(Generic[StateType], metaclass=BaseNodeMeta):
@@ -271,9 +247,7 @@ class BaseNode(Generic[StateType], metaclass=BaseNodeMeta):
             """
 
             if cls.merge_behavior == MergeBehavior.AWAIT_ATTRIBUTES:
-                if state.meta.node_execution_cache.is_node_execution_initiated(
-                    cls.node_class, node_span_id
-                ):
+                if state.meta.node_execution_cache.is_node_execution_initiated(cls.node_class, node_span_id):
                     return False
 
                 is_ready = True
@@ -281,9 +255,7 @@ class BaseNode(Generic[StateType], metaclass=BaseNodeMeta):
                     if not descriptor.instance:
                         continue
 
-                    resolved_value = resolve_value(
-                        descriptor.instance, state, path=descriptor.name
-                    )
+                    resolved_value = resolve_value(descriptor.instance, state, path=descriptor.name)
                     if is_unresolved(resolved_value):
                         is_ready = False
                         break
@@ -291,29 +263,22 @@ class BaseNode(Generic[StateType], metaclass=BaseNodeMeta):
                 return is_ready
 
             if cls.merge_behavior == MergeBehavior.AWAIT_ANY:
-                if state.meta.node_execution_cache.is_node_execution_initiated(
-                    cls.node_class, node_span_id
-                ):
+                if state.meta.node_execution_cache.is_node_execution_initiated(cls.node_class, node_span_id):
                     return False
 
                 return True
 
             if cls.merge_behavior == MergeBehavior.AWAIT_ALL:
-                if state.meta.node_execution_cache.is_node_execution_initiated(
-                    cls.node_class, node_span_id
-                ):
+                if state.meta.node_execution_cache.is_node_execution_initiated(cls.node_class, node_span_id):
                     return False
 
                 """
                 A node utilizing an AWAIT_ALL merge strategy will only be considered ready for the Nth time
                 when all of its dependencies have been executed N times.
                 """
-                current_node_execution_count = (
-                    state.meta.node_execution_cache.get_execution_count(cls.node_class)
-                )
+                current_node_execution_count = state.meta.node_execution_cache.get_execution_count(cls.node_class)
                 return all(
-                    state.meta.node_execution_cache.get_execution_count(dep)
-                    == current_node_execution_count + 1
+                    state.meta.node_execution_cache.get_execution_count(dep) == current_node_execution_count + 1
                     for dep in dependencies
                 )
 
@@ -353,9 +318,7 @@ class BaseNode(Generic[StateType], metaclass=BaseNodeMeta):
             if not descriptor.instance:
                 continue
 
-            resolved_value = resolve_value(
-                descriptor.instance, self.state, path=descriptor.name, memo=inputs
-            )
+            resolved_value = resolve_value(descriptor.instance, self.state, path=descriptor.name, memo=inputs)
             setattr(self, descriptor.name, resolved_value)
 
         # Resolve descriptors set as defaults to the outputs class
@@ -379,9 +342,7 @@ class BaseNode(Generic[StateType], metaclass=BaseNodeMeta):
         for key, value in inputs.items():
             path_parts = key.split(".")
             node_attribute_discriptor = getattr(self.__class__, path_parts[0])
-            inputs_key = reduce(
-                lambda acc, part: acc[part], path_parts[1:], node_attribute_discriptor
-            )
+            inputs_key = reduce(lambda acc, part: acc[part], path_parts[1:], node_attribute_discriptor)
             all_inputs[inputs_key] = value
 
         self._inputs = MappingProxyType(all_inputs)

vellum/workflows/nodes/bases/tests/test_base_node.py

@@ -1,7 +1,6 @@
 from typing import Optional
 
 from vellum.core.pydantic_utilities import UniversalBaseModel
-
 from vellum.workflows.inputs.base import BaseInputs
 from vellum.workflows.nodes.bases.base import BaseNode
 from vellum.workflows.state.base import BaseState, StateMeta

vellum/workflows/nodes/core/inline_subworkflow_node/node.py

@@ -1,10 +1,12 @@
 from typing import TYPE_CHECKING, Generic, Iterator, Optional, Set, Type, TypeVar
 
+from vellum.workflows.context import execution_context, get_parent_context
 from vellum.workflows.errors.types import VellumErrorCode
 from vellum.workflows.exceptions import NodeException
 from vellum.workflows.nodes.bases.base_subworkflow_node import BaseSubworkflowNode
 from vellum.workflows.outputs.base import BaseOutput, BaseOutputs
 from vellum.workflows.state.base import BaseState
+from vellum.workflows.state.context import WorkflowContext
 from vellum.workflows.types.generics import StateType, WorkflowInputsType
 
 if TYPE_CHECKING:
@@ -24,18 +26,22 @@ class InlineSubworkflowNode(BaseSubworkflowNode[StateType], Generic[StateType, W
     subworkflow: Type["BaseWorkflow[WorkflowInputsType, InnerStateType]"]
 
     def run(self) -> Iterator[BaseOutput]:
-        subworkflow = self.subworkflow(
-            parent_state=self.state,
-            context=self._context,
-        )
-        subworkflow_stream = subworkflow.stream(
-            inputs=self._compile_subworkflow_inputs(),
-        )
+        with execution_context(parent_context=get_parent_context() or self._context.parent_context):
+            subworkflow = self.subworkflow(
+                parent_state=self.state,
+                context=WorkflowContext(
+                    _vellum_client=self._context._vellum_client,
+                ),
+            )
+            subworkflow_stream = subworkflow.stream(
+                inputs=self._compile_subworkflow_inputs(),
+            )
 
         outputs: Optional[BaseOutputs] = None
         fulfilled_output_names: Set[str] = set()
 
         for event in subworkflow_stream:
+            self._context._emit_subworkflow_event(event)
             if event.name == "workflow.execution.streaming":
                 if event.output.is_fulfilled:
                     fulfilled_output_names.add(event.output.name)

vellum/workflows/nodes/core/templating_node/node.py

@@ -87,6 +87,7 @@ class TemplatingNode(BaseNode[StateType], Generic[StateType, _OutputType], metac
 
         result: _OutputType - The result of the template rendering
         """
+
         # We use our mypy plugin to override the _OutputType with the actual output type
         # for downstream references to this output.
         result: _OutputType  # type: ignore[valid-type]

vellum/workflows/nodes/core/try_node/node.py

@@ -1,6 +1,6 @@
 import sys
-from types import MappingProxyType, ModuleType
-from typing import TYPE_CHECKING, Any, Callable, Dict, Generic, Iterator, Optional, Set, Tuple, Type, TypeVar, cast
+from types import ModuleType
+from typing import TYPE_CHECKING, Any, Callable, Dict, Generic, Iterator, Optional, Set, Tuple, Type, TypeVar
 
 from vellum.workflows.errors.types import VellumError, VellumErrorCode
 from vellum.workflows.exceptions import NodeException

vellum/workflows/nodes/core/try_node/tests/test_node.py

@@ -30,9 +30,7 @@ def test_try_node__on_error_code__successfully_caught():
     assert len(outputs) == 2
     assert set(outputs) == {
         BaseOutput(name="value"),
-        BaseOutput(
-            name="error", value=VellumError(message="This will be caught", code=VellumErrorCode.PROVIDER_ERROR)
-        ),
+        BaseOutput(name="error", value=VellumError(message="This will be caught", code=VellumErrorCode.PROVIDER_ERROR)),
     }
 
 

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

@@ -18,13 +18,14 @@ class APINode(BaseAPINode):
 
     authorization_type: Optional[AuthorizationType] = None - The type of authorization to use for the API call.
     api_key_header_key: Optional[str] = None - The header key to use for the API key authorization.
-    bearer_token_value: Optional[str] = None - The bearer token value to use for the bearer token authorization.
+    bearer_token_value: Optional[Union[str, VellumSecretReference]] = None - The bearer token value to use
+    for the bearer token authorization.
     """
 
     authorization_type: Optional[AuthorizationType] = None
     api_key_header_key: Optional[str] = None
     api_key_header_value: Optional[Union[str, VellumSecretReference]] = None
-    bearer_token_value: Optional[str] = None
+    bearer_token_value: Optional[Union[str, VellumSecretReference]] = None
 
     def run(self) -> BaseAPINode.Outputs:
         headers = self.headers or {}

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

@@ -8,7 +8,7 @@ from vellum.workflows.errors.types import VellumErrorCode
 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, JsonObject, VellumSecret
+from vellum.workflows.types.core import Json, VellumSecret
 from vellum.workflows.types.generics import StateType
 
 

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

@@ -3,7 +3,6 @@ from typing import ClassVar, Generator, Generic, Iterator, List, Optional, Union
 
 from vellum import AdHocExecutePromptEvent, ExecutePromptEvent, PromptOutput
 from vellum.core import RequestOptions
-
 from vellum.workflows.errors.types import VellumErrorCode
 from vellum.workflows.exceptions import NodeException
 from vellum.workflows.nodes.bases import BaseNode

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

@@ -14,8 +14,9 @@ from vellum import (
     PromptRequestStringInput,
     VellumVariable,
 )
-
+from vellum.client import RequestOptions
 from vellum.workflows.constants import OMIT
+from vellum.workflows.context import get_parent_context
 from vellum.workflows.errors import VellumErrorCode
 from vellum.workflows.exceptions import NodeException
 from vellum.workflows.nodes.displayable.bases.base_prompt_node import BasePromptNode
@@ -49,6 +50,13 @@ class BaseInlinePromptNode(BasePromptNode, Generic[StateType]):
 
     def _get_prompt_event_stream(self) -> Iterator[AdHocExecutePromptEvent]:
         input_variables, input_values = self._compile_prompt_inputs()
+        current_parent_context = get_parent_context()
+        parent_context = current_parent_context.model_dump_json() if current_parent_context else None
+        request_options = self.request_options or RequestOptions()
+        request_options["additional_body_parameters"] = {
+            "execution_context": {"parent_context": parent_context},
+            **request_options.get("additional_body_parameters", {}),
+        }
 
         return self._context.vellum_client.ad_hoc.adhoc_execute_prompt_stream(
             ml_model=self.ml_model,

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

@@ -11,7 +11,9 @@ from vellum import (
     RawPromptExecutionOverridesRequest,
     StringInputRequest,
 )
+from vellum.client import RequestOptions
 from vellum.workflows.constants import LATEST_RELEASE_TAG, OMIT
+from vellum.workflows.context import get_parent_context
 from vellum.workflows.errors import VellumErrorCode
 from vellum.workflows.exceptions import NodeException
 from vellum.workflows.nodes.displayable.bases.base_prompt_node import BasePromptNode
@@ -25,7 +27,8 @@ class BasePromptDeploymentNode(BasePromptNode, Generic[StateType]):
     prompt_inputs: EntityInputsInterface - The inputs for the Prompt
     deployment: Union[UUID, str] - Either the Prompt Deployment's UUID or its name.
     release_tag: str - The release tag to use for the Prompt Execution
-    external_id: Optional[str] - Optionally include a unique identifier for tracking purposes. Must be unique within a given Prompt Deployment.
+    external_id: Optional[str] - Optionally include a unique identifier for tracking purposes.
+        Must be unique within a given Prompt Deployment.
     expand_meta: Optional[PromptDeploymentExpandMetaRequest] - Expandable execution fields to include in the response
     raw_overrides: Optional[RawPromptExecutionOverridesRequest] - The raw overrides to use for the Prompt Execution
     expand_raw: Optional[Sequence[str]] - Expandable raw fields to include in the response
@@ -45,6 +48,13 @@ class BasePromptDeploymentNode(BasePromptNode, Generic[StateType]):
     metadata: Optional[Dict[str, Optional[Any]]] = OMIT
 
     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
+        request_options = self.request_options or RequestOptions()
+        request_options["additional_body_parameters"] = {
+            "execution_context": {"parent_context": parent_context},
+            **request_options.get("additional_body_parameters", {}),
+        }
         return self._context.vellum_client.execute_prompt_stream(
             inputs=self._compile_prompt_inputs(),
             prompt_deployment_id=str(self.deployment) if isinstance(self.deployment, UUID) else None,
@@ -55,7 +65,7 @@ class BasePromptDeploymentNode(BasePromptNode, Generic[StateType]):
             raw_overrides=self.raw_overrides,
             expand_raw=self.expand_raw,
             metadata=self.metadata,
-            request_options=self.request_options,
+            request_options=request_options,
         )
 
     def _compile_prompt_inputs(self) -> List[PromptDeploymentInputRequest]:

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

@@ -12,7 +12,6 @@ from vellum import (
     SearchWeightsRequest,
 )
 from vellum.core import ApiError, RequestOptions
-
 from vellum.workflows.errors import VellumErrorCode
 from vellum.workflows.exceptions import NodeException
 from vellum.workflows.nodes.bases import BaseNode

vellum/workflows/nodes/displayable/code_execution_node/tests/test_code_execution_node.py

@@ -1,7 +1,6 @@
 import os
 
 from vellum import CodeExecutorResponse, NumberVellumValue, StringInput
-
 from vellum.workflows.inputs.base import BaseInputs
 from vellum.workflows.nodes.displayable.code_execution_node import CodeExecutionNode
 from vellum.workflows.references.vellum_secret import VellumSecretReference

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

@@ -4,12 +4,13 @@ from typing import Union
 
 def get_project_root() -> str:
     current_dir = os.getcwd()
-    while current_dir != '/':
+    while current_dir != "/":
         if ".git" in os.listdir(current_dir):
             return current_dir
         current_dir = os.path.dirname(current_dir)
     raise FileNotFoundError("Project root not found.")
 
+
 def read_file_from_path(filepath: str) -> Union[str, None]:
     project_root = get_project_root()
     relative_filepath = os.path.join(project_root, filepath)
@@ -17,5 +18,5 @@ def read_file_from_path(filepath: str) -> Union[str, None]:
     if not os.path.exists(relative_filepath):
         return None
 
-    with open(relative_filepath, 'r') as file:
+    with open(relative_filepath) as file:
         return file.read()

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

@@ -15,7 +15,7 @@ class ConditionalNode(BaseNode):
     """
 
     class Ports(NodePorts):
-        def __call__(self, outputs: BaseOutputs, state: BaseState) -> Set[Port]:    
+        def __call__(self, outputs: BaseOutputs, state: BaseState) -> Set[Port]:
             all_ports = [port for port in self.__class__]
             enforce_single_invoked_port = validate_ports(all_ports)
 

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

@@ -3,7 +3,6 @@ from typing import Any, ClassVar, Dict, Generic, List, Optional, Union, cast
 
 from vellum import ChatHistoryInput, ChatMessage, JsonInput, MetricDefinitionInput, NumberInput, StringInput
 from vellum.core import RequestOptions
-
 from vellum.workflows.constants import LATEST_RELEASE_TAG
 from vellum.workflows.errors.types import VellumErrorCode
 from vellum.workflows.exceptions import NodeException

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

@@ -26,6 +26,7 @@ class InlinePromptNode(BaseInlinePromptNode[StateType]):
 
         text: str - The result of the Prompt Execution
         """
+
         text: str
 
     def run(self) -> Iterator[BaseOutput]:

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

@@ -14,7 +14,8 @@ class PromptDeploymentNode(BasePromptDeploymentNode[StateType]):
     prompt_inputs: EntityInputsInterface - The inputs for the Prompt
     deployment: Union[UUID, str] - Either the Prompt Deployment's UUID or its name.
     release_tag: str - The release tag to use for the Prompt Execution
-    external_id: Optional[str] - Optionally include a unique identifier for tracking purposes. Must be unique within a given Prompt Deployment.
+    external_id: Optional[str] - Optionally include a unique identifier for tracking purposes.
+        Must be unique within a given Prompt Deployment.
     expand_meta: Optional[PromptDeploymentExpandMetaRequest] - Expandable execution fields to include in the response
     raw_overrides: Optional[RawPromptExecutionOverridesRequest] - The raw overrides to use for the Prompt Execution
     expand_raw: Optional[Sequence[str]] - Expandable raw fields to include in the response
@@ -28,6 +29,7 @@ class PromptDeploymentNode(BasePromptDeploymentNode[StateType]):
 
         text: str - The result of the Prompt Execution
         """
+
         text: str
 
     def run(self) -> Iterator[BaseOutput]:

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

@@ -25,6 +25,7 @@ class SearchNode(BaseSearchNode[StateType]):
         results: List[SearchResult] - The raw search results
         text: str - The text of the search results joined by the chunk_separator
         """
+
         text: str
 
     def run(self) -> Outputs:

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

@@ -13,6 +13,7 @@ from vellum import (
 )
 from vellum.core import RequestOptions
 from vellum.workflows.constants import LATEST_RELEASE_TAG, OMIT
+from vellum.workflows.context import get_parent_context
 from vellum.workflows.errors import VellumErrorCode
 from vellum.workflows.exceptions import NodeException
 from vellum.workflows.nodes.bases.base_subworkflow_node.node import BaseSubworkflowNode
@@ -27,8 +28,9 @@ class SubworkflowDeploymentNode(BaseSubworkflowNode[StateType], Generic[StateTyp
     subworkflow_inputs: EntityInputsInterface - The inputs for the Subworkflow
     deployment: Union[UUID, str] - Either the Workflow Deployment's UUID or its name.
     release_tag: str = LATEST_RELEASE_TAG - The release tag to use for the Workflow Execution
-    external_id: Optional[str] = OMIT - Optionally include a unique identifier for tracking purposes. Must be unique within a given Workflow Deployment.
-    expand_meta: Optional[WorkflowExpandMetaRequest] = OMIT - Expandable execution fields to include in the respownse
+    external_id: Optional[str] = OMIT - Optionally include a unique identifier for tracking purposes.
+        Must be unique within a given Workflow Deployment.
+    expand_meta: Optional[WorkflowExpandMetaRequest] = OMIT - Expandable execution fields to include in the response
     metadata: Optional[Dict[str, Optional[Any]]] = OMIT - The metadata to use for the Workflow Execution
     request_options: Optional[RequestOptions] = None - The request options to use for the Workflow Execution
     """
@@ -88,6 +90,13 @@ class SubworkflowDeploymentNode(BaseSubworkflowNode[StateType], Generic[StateTyp
         return compiled_inputs
 
     def run(self) -> Iterator[BaseOutput]:
+        current_parent_context = get_parent_context()
+        parent_context = current_parent_context.model_dump(mode="json") if current_parent_context else None
+        request_options = self.request_options or RequestOptions()
+        request_options["additional_body_parameters"] = {
+            "execution_context": {"parent_context": parent_context},
+            **request_options.get("additional_body_parameters", {}),
+        }
         subworkflow_stream = self._context.vellum_client.execute_workflow_stream(
             inputs=self._compile_subworkflow_inputs(),
             workflow_deployment_id=str(self.deployment) if isinstance(self.deployment, UUID) else None,
@@ -96,8 +105,9 @@ class SubworkflowDeploymentNode(BaseSubworkflowNode[StateType], Generic[StateTyp
             external_id=self.external_id,
             event_types=["WORKFLOW"],
             metadata=self.metadata,
-            request_options=self.request_options,
+            request_options=request_options,
         )
+        # for some reason execution context isn't showing as an option? ^ failing mypy
 
         outputs: Optional[List[WorkflowOutput]] = None
         fulfilled_output_names: Set[str] = set()

vellum/workflows/nodes/displayable/tests/test_inline_text_prompt_node.py

@@ -138,10 +138,13 @@ def test_inline_text_prompt_node__catch_provider_error(vellum_adhoc_prompt_clien
     outputs = list(node.run())
 
     # THEN the node should have produced the outputs we expect
-    assert BaseOutput(
-        name="error",
-        value=SdkVellumError(
-            message="OpenAI failed",
-            code=VellumErrorCode.PROVIDER_ERROR,
-        ),
-    ) in outputs
+    assert (
+        BaseOutput(
+            name="error",
+            value=SdkVellumError(
+                message="OpenAI failed",
+                code=VellumErrorCode.PROVIDER_ERROR,
+            ),
+        )
+        in outputs
+    )

vellum/workflows/nodes/displayable/tests/test_search_node_wth_text_output.py

@@ -11,7 +11,6 @@ from vellum import (
     SearchResultMergingRequest,
     SearchWeightsRequest,
 )
-
 from vellum.workflows.inputs import BaseInputs
 from vellum.workflows.nodes.displayable.search_node import SearchNode as BaseSearchNode
 from vellum.workflows.state import BaseState

vellum/workflows/nodes/displayable/tests/test_text_prompt_deployment_node.py

@@ -75,5 +75,5 @@ def test_text_prompt_deployment_node__basic(vellum_client):
         prompt_deployment_name="my-deployment",
         raw_overrides=OMIT,
         release_tag="LATEST",
-        request_options=None,
+        request_options={"additional_body_parameters": {"execution_context": {"parent_context": None}}},
     )

vellum/workflows/outputs/base.py

@@ -90,7 +90,7 @@ class BaseOutput(Generic[_Delta, _Accumulated]):
             return False
 
         return self.name == other.name and self.value == other.value and self.delta == other.delta
-    
+
     def __hash__(self) -> int:
         return hash((self._name, self._value, self._value))
 
@@ -193,9 +193,7 @@ class BaseOutputs(metaclass=_BaseOutputsMeta):
         if not isinstance(other, dict):
             return super().__eq__(other)
 
-        outputs = {
-            name: value for name, value in vars(self).items() if not name.startswith("_") and value is not UNDEF
-        }
+        outputs = {name: value for name, value in vars(self).items() if not name.startswith("_") and value is not UNDEF}
         return outputs == other
 
     def __repr__(self) -> str:

vellum/workflows/ports/node_ports.py

@@ -1,4 +1,4 @@
-from typing import Any, Dict, Iterable, Iterator, Optional, Set, Tuple, Type
+from typing import Any, Dict, Iterator, Optional, Set, Tuple, Type
 
 from vellum.workflows.outputs.base import BaseOutput, BaseOutputs
 from vellum.workflows.ports.port import Port