vellum-ai 0.10.7__tar.gz → 0.10.8__tar.gz

{vellum_ai-0.10.7 → vellum_ai-0.10.8}/PKG-INFO

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

{vellum_ai-0.10.7 → vellum_ai-0.10.8}/ee/vellum_cli/__init__.py

@@ -42,11 +42,30 @@ def push(
 
 @main.command()
 @click.argument("module", required=False)
-@click.option("--legacy-module", is_flag=True, help="Pull the workflow as a legacy module")
-@click.option("--include-json", is_flag=True, help="Include the JSON representation of the Workflow in the pull response. Should only be used for debugging purposes.")
-def pull(module: Optional[str], legacy_module: Optional[bool], include_json: Optional[bool]) -> None:
+@click.option(
+    "--include-json",
+    is_flag=True,
+    help="Include the JSON representation of the Workflow in the pull response. Should only be used for debugging purposes.",
+)
+@click.option("--workflow-sandbox-id", type=str, help="Pull the Workflow from a specific Sandbox ID")
+@click.option(
+    "--exclude-code",
+    is_flag=True,
+    help="Exclude the code definition of the Workflow from the pull response. Should only be used for debugging purposes.",
+)
+def pull(
+    module: Optional[str],
+    include_json: Optional[bool],
+    workflow_sandbox_id: Optional[str],
+    exclude_code: Optional[bool],
+) -> None:
     """Pull Workflow from Vellum"""
-    pull_command(module, legacy_module, include_json)
+    pull_command(
+        module=module,
+        include_json=include_json,
+        workflow_sandbox_id=workflow_sandbox_id,
+        exclude_code=exclude_code,
+    )
 
 
 @main.group(aliases=["images", "image"])

{vellum_ai-0.10.7 → vellum_ai-0.10.8}/ee/vellum_cli/pull.py

@@ -7,28 +7,38 @@ from typing import Optional
 from dotenv import load_dotenv
 
 from vellum.workflows.vellum_client import create_vellum_client
-from vellum_cli.config import load_vellum_cli_config
+from vellum_cli.config import WorkflowConfig, load_vellum_cli_config
 from vellum_cli.logger import load_cli_logger
 
 
 def pull_command(
-    module: Optional[str], legacy_module: Optional[bool] = None, include_json: Optional[bool] = None
+    module: Optional[str] = None,
+    workflow_sandbox_id: Optional[str] = None,
+    include_json: Optional[bool] = None,
+    exclude_code: Optional[bool] = None,
 ) -> None:
     load_dotenv()
     logger = load_cli_logger()
     config = load_vellum_cli_config()
 
-    if not config.workflows:
-        raise ValueError("No Workflows found in project to pull.")
-
-    if len(config.workflows) > 1 and not module:
-        raise ValueError("Multiple workflows found in project to pull. Pulling only a single workflow is supported.")
-
     workflow_config = (
-        next((w for w in config.workflows if w.module == module), None) if module else config.workflows[0]
+        next((w for w in config.workflows if w.module == module), None)
+        if module
+        else (config.workflows[0] if config.workflows else None)
     )
+    save_lock_file = False
     if workflow_config is None:
-        raise ValueError(f"No workflow config for '{module}' found in project to push.")
+        if module:
+            raise ValueError(f"No workflow config for '{module}' found in project to pull.")
+        elif workflow_sandbox_id:
+            workflow_config = WorkflowConfig(
+                workflow_sandbox_id=workflow_sandbox_id,
+                module=f"workflow_{workflow_sandbox_id.split('-')[0]}",
+            )
+            config.workflows.append(workflow_config)
+            save_lock_file = True
+        else:
+            raise ValueError("No workflow config found in project to pull from.")
 
     if not workflow_config.workflow_sandbox_id:
         raise ValueError("No workflow sandbox ID found in project to pull from.")
@@ -36,10 +46,10 @@ def pull_command(
     logger.info(f"Pulling workflow into {workflow_config.module}")
     client = create_vellum_client()
     query_parameters = {}
-    if legacy_module:
-        query_parameters["legacyModule"] = legacy_module
     if include_json:
         query_parameters["include_json"] = include_json
+    if exclude_code:
+        query_parameters["exclude_code"] = exclude_code
 
     response = client.workflows.pull(
         workflow_config.workflow_sandbox_id,
@@ -81,6 +91,11 @@ def pull_command(
                 target.write(source.read().decode("utf-8"))
 
     if include_json:
-        logger.warning("The pulled JSON representation of the Workflow should be used for debugging purposely only. Its schema should be considered unstable and subject to change at any time.")
+        logger.warning(
+            "The pulled JSON representation of the Workflow should be used for debugging purposely only. Its schema should be considered unstable and subject to change at any time."
+        )
+
+    if save_lock_file:
+        config.save()
 
     logger.info(f"Successfully pulled Workflow into {workflow_config.module}")

{vellum_ai-0.10.7 → vellum_ai-0.10.8}/ee/vellum_cli/tests/test_pull.py

@@ -69,8 +69,33 @@ def test_pull(vellum_client, mock_module):
     pull_command(module)
 
     # THEN the workflow.py file is written to the module directory
-    assert os.path.exists(os.path.join(temp_dir, *module.split("."), "workflow.py"))
-    with open(os.path.join(temp_dir, *module.split("."), "workflow.py")) as f:
+    workflow_py = os.path.join(temp_dir, *module.split("."), "workflow.py")
+    assert os.path.exists(workflow_py)
+    with open(workflow_py) as f:
+        assert f.read() == "print('hello')"
+
+
+def test_pull__sandbox_id_with_no_config(vellum_client):
+    # GIVEN a workflow sandbox id
+    workflow_sandbox_id = "87654321-0000-0000-0000-000000000000"
+
+    # AND the workflow pull API call returns a zip file
+    vellum_client.workflows.pull.return_value = iter([zip_file_map({"workflow.py": "print('hello')"})])
+
+    # AND we are currently in a new directory
+    current_dir = os.getcwd()
+    temp_dir = tempfile.mkdtemp()
+    os.chdir(temp_dir)
+
+    # WHEN the user runs the pull command with the workflow sandbox id and no module
+    pull_command(workflow_sandbox_id=workflow_sandbox_id)
+    os.chdir(current_dir)
+
+    # THEN the pull api is called with exclude_code=True
+    vellum_client.workflows.pull.assert_called_once()
+    workflow_py = os.path.join(temp_dir, "workflow_87654321", "workflow.py")
+    assert os.path.exists(workflow_py)
+    with open(workflow_py) as f:
         assert f.read() == "print('hello')"
 
 
@@ -168,3 +193,21 @@ def test_pull__include_json(vellum_client, mock_module):
     vellum_client.workflows.pull.assert_called_once()
     call_args = vellum_client.workflows.pull.call_args.kwargs
     assert call_args["request_options"]["additional_query_parameters"] == {"include_json": True}
+
+
+def test_pull__exclude_code(vellum_client, mock_module):
+    # GIVEN a module on the user's filesystem
+    _, module = mock_module
+
+    # AND the workflow pull API call returns a zip file
+    vellum_client.workflows.pull.return_value = iter(
+        [zip_file_map({"workflow.py": "print('hello')", "workflow.json": "{}"})]
+    )
+
+    # WHEN the user runs the pull command
+    pull_command(module, exclude_code=True)
+
+    # THEN the pull api is called with exclude_code=True
+    vellum_client.workflows.pull.assert_called_once()
+    call_args = vellum_client.workflows.pull.call_args.kwargs
+    assert call_args["request_options"]["additional_query_parameters"] == {"exclude_code": True}

{vellum_ai-0.10.7 → vellum_ai-0.10.8}/ee/vellum_ee/workflows/display/nodes/base_node_display.py

@@ -1,7 +1,7 @@
 from functools import cached_property
 import inspect
 from uuid import UUID
-from typing import TYPE_CHECKING, Any, Dict, Generic, Optional, Type, TypeVar, get_args, get_origin, cast
+from typing import TYPE_CHECKING, Any, Dict, Generic, Optional, Type, TypeVar, cast, get_args, get_origin
 
 from vellum.workflows.nodes.bases.base import BaseNode
 from vellum.workflows.nodes.utils import get_wrapped_node, has_wrapped_node

{vellum_ai-0.10.7 → vellum_ai-0.10.8}/ee/vellum_ee/workflows/display/nodes/vellum/code_execution_node.py

@@ -1,5 +1,5 @@
 from uuid import UUID
-from typing import ClassVar, Generic, Optional, TypeVar
+from typing import ClassVar, Dict, Generic, Optional, TypeVar
 
 from vellum.workflows.nodes.displayable.code_execution_node import CodeExecutionNode
 from vellum.workflows.nodes.displayable.code_execution_node.utils import read_file_from_path
@@ -20,6 +20,8 @@ class BaseCodeExecutionNodeDisplay(BaseNodeVellumDisplay[_CodeExecutionNodeType]
     output_id: ClassVar[Optional[UUID]] = None
     log_output_id: ClassVar[Optional[UUID]] = None
 
+    node_input_ids_by_name: ClassVar[Dict[str, UUID]] = {}
+
     def serialize(
         self, display_context: WorkflowDisplayContext, error_output_id: Optional[UUID] = None, **kwargs
     ) -> JsonObject:
@@ -27,6 +29,19 @@ class BaseCodeExecutionNodeDisplay(BaseNodeVellumDisplay[_CodeExecutionNodeType]
         node_id = self.node_id
 
         code = read_file_from_path(raise_if_descriptor(node.filepath))
+        code_inputs = raise_if_descriptor(node.code_inputs)
+
+        inputs = [
+            create_node_input(
+                node_id=node_id,
+                input_name=variable_name,
+                value=variable_value,
+                display_context=display_context,
+                input_id=self.node_input_ids_by_name.get(variable_name),
+            )
+            for variable_name, variable_value in code_inputs.items()
+        ]
+
         code_node_input = create_node_input(
             node_id=node_id,
             input_name="code",
@@ -41,7 +56,7 @@ class BaseCodeExecutionNodeDisplay(BaseNodeVellumDisplay[_CodeExecutionNodeType]
             display_context=display_context,
             input_id=self.runtime_input_id,
         )
-        inputs = [code_node_input, runtime_node_input]
+        inputs.extend([code_node_input, runtime_node_input])
 
         packages = raise_if_descriptor(node.packages)
 

{vellum_ai-0.10.7 → vellum_ai-0.10.8}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "vellum-ai"
-version = "0.10.7"
+version = "0.10.8"
 description = ""
 readme = "README.md"
 authors = []

{vellum_ai-0.10.7 → vellum_ai-0.10.8}/src/vellum/client/core/client_wrapper.py

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

{vellum_ai-0.10.7 → vellum_ai-0.10.8}/src/vellum/workflows/events/types.py

@@ -103,4 +103,4 @@ class BaseEvent(UniversalBaseModel):
     api_version: Literal["2024-10-25"] = "2024-10-25"
     trace_id: UUID
     span_id: UUID
-    parent: Optional['ParentContext'] = None
+    parent: Optional[ParentContext] = None

{vellum_ai-0.10.7 → vellum_ai-0.10.8}/src/vellum/workflows/nodes/core/inline_subworkflow_node/node.py

@@ -18,6 +18,7 @@ class InlineSubworkflowNode(BaseSubworkflowNode[StateType], Generic[StateType, W
     Used to execute a Subworkflow defined inline.
 
     subworkflow: Type["BaseWorkflow[WorkflowInputsType, InnerStateType]"] - The Subworkflow to execute
+    subworkflow_inputs: ClassVar[EntityInputsInterface] = {}
     """
 
     subworkflow: Type["BaseWorkflow[WorkflowInputsType, InnerStateType]"]

{vellum_ai-0.10.7 → vellum_ai-0.10.8}/src/vellum/workflows/nodes/core/templating_node/node.py

@@ -82,6 +82,11 @@ class TemplatingNode(BaseNode[StateType], Generic[StateType, _OutputType], metac
     jinja_custom_filters: Mapping[str, Callable[[Union[str, bytes]], bool]] = _DEFAULT_JINJA_CUSTOM_FILTERS
 
     class Outputs(BaseNode.Outputs):
+        """
+        The outputs of the TemplatingNode.
+
+        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_ai-0.10.7 → vellum_ai-0.10.8}/src/vellum/workflows/nodes/displayable/api_node/node.py

@@ -8,7 +8,7 @@ from vellum.workflows.references.vellum_secret import VellumSecretReference
 class APINode(BaseAPINode):
     """
     Used to execute an API call. This node exists to be backwards compatible with Vellum's API Node, and for most cases,
-    you should extend from `APINode` directly.
+    you should extend from `BaseAPINode` directly.
 
     url: str - The URL to send the request to.
     method: APIRequestMethod - The HTTP method to use for the request.

{vellum_ai-0.10.7 → vellum_ai-0.10.8}/src/vellum/workflows/nodes/displayable/bases/prompt_deployment_node.py

@@ -11,7 +11,6 @@ from vellum import (
     RawPromptExecutionOverridesRequest,
     StringInputRequest,
 )
-
 from vellum.workflows.constants import LATEST_RELEASE_TAG, OMIT
 from vellum.workflows.errors import VellumErrorCode
 from vellum.workflows.exceptions import NodeException
@@ -26,7 +25,7 @@ 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] - The external ID 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.
     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

{vellum_ai-0.10.7 → vellum_ai-0.10.8}/src/vellum/workflows/nodes/displayable/code_execution_node/node.py

@@ -67,9 +67,8 @@ class CodeExecutionNode(BaseNode[StateType], Generic[StateType, _OutputType], me
 
     filepath: str - The path to the script to execute.
     code_inputs: EntityInputsInterface - The inputs for the custom script.
-    output_type: VellumVariableType = "STRING" - The type of the output from the custom script.
     runtime: CodeExecutionRuntime = "PYTHON_3_12" - The runtime to use for the custom script.
-    packages: Optional[Sequence[CodeExecutionPackageRequest]] = None - The packages to use for the custom script.
+    packages: Optional[Sequence[CodeExecutionPackage]] = None - The packages to use for the custom script.
     request_options: Optional[RequestOptions] = None - The request options to use for the custom script.
     """
 

vellum_ai-0.10.8/src/vellum/workflows/nodes/displayable/code_execution_node/utils.py

@@ -0,0 +1,21 @@
+import os
+from typing import Union
+
+
+def get_project_root() -> str:
+    current_dir = os.getcwd()
+    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)
+
+    if not os.path.exists(relative_filepath):
+        return None
+
+    with open(relative_filepath, 'r') as file:
+        return file.read()

{vellum_ai-0.10.7 → vellum_ai-0.10.8}/src/vellum/workflows/nodes/displayable/inline_prompt_node/node.py

@@ -9,16 +9,23 @@ from vellum.workflows.types.generics import StateType
 
 class InlinePromptNode(BaseInlinePromptNode[StateType]):
     """
-    Used to execute an Inline Prompt and surface a string output for convenience.
+    Used to execute a Prompt defined inline.
 
     prompt_inputs: EntityInputsInterface - The inputs for the Prompt
     ml_model: str - Either the ML Model's UUID or its name.
-    blocks: List[PromptBlockRequest] - The blocks that make up the Prompt
+    blocks: List[PromptBlock] - The blocks that make up the Prompt
+    functions: Optional[List[FunctionDefinition]] - The functions to include in the Prompt
     parameters: PromptParameters - The parameters for the Prompt
-    expand_meta: Optional[AdHocExpandMetaRequest] - Set of expandable execution fields to include in the response
+    expand_meta: Optional[AdHocExpandMeta] - Expandable execution fields to include in the response
+    request_options: Optional[RequestOptions] - The request options to use for the Prompt Execution
     """
 
     class Outputs(BaseInlinePromptNode.Outputs):
+        """
+        The outputs of the InlinePromptNode.
+
+        text: str - The result of the Prompt Execution
+        """
         text: str
 
     def run(self) -> Iterator[BaseOutput]:

{vellum_ai-0.10.7 → vellum_ai-0.10.8}/src/vellum/workflows/nodes/displayable/prompt_deployment_node/node.py

@@ -14,7 +14,7 @@ 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] - The external ID 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.
     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
@@ -23,6 +23,11 @@ class PromptDeploymentNode(BasePromptDeploymentNode[StateType]):
     """
 
     class Outputs(BasePromptDeploymentNode.Outputs):
+        """
+        The outputs of the PromptDeploymentNode.
+
+        text: str - The result of the Prompt Execution
+        """
         text: str
 
     def run(self) -> Iterator[BaseOutput]:

{vellum_ai-0.10.7 → vellum_ai-0.10.8}/src/vellum/workflows/nodes/displayable/subworkflow_deployment_node/node.py

@@ -12,7 +12,6 @@ from vellum import (
     WorkflowRequestStringInputRequest,
 )
 from vellum.core import RequestOptions
-
 from vellum.workflows.constants import LATEST_RELEASE_TAG, OMIT
 from vellum.workflows.errors import VellumErrorCode
 from vellum.workflows.exceptions import NodeException
@@ -28,7 +27,7 @@ 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 - The external ID 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
     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

{vellum_ai-0.10.7 → vellum_ai-0.10.8}/src/vellum/workflows/nodes/displayable/tests/test_text_prompt_deployment_node.py

@@ -8,7 +8,6 @@ from vellum import (
     PromptOutput,
     StringVellumValue,
 )
-
 from vellum.workflows.constants import OMIT
 from vellum.workflows.inputs import BaseInputs
 from vellum.workflows.nodes import PromptDeploymentNode
@@ -65,7 +64,7 @@ def test_text_prompt_deployment_node__basic(vellum_client):
     assert text_output.name == "text"
     assert text_output.value == "Hello, world!"
 
-    # AND we should have made the expected call to Vellum search
+    # AND we should have made the expected call to stream the prompt execution
     vellum_client.execute_prompt_stream.assert_called_once_with(
         expand_meta=OMIT,
         expand_raw=OMIT,

{vellum_ai-0.10.7 → vellum_ai-0.10.8}/src/vellum/workflows/runner/runner.py

@@ -28,7 +28,7 @@ from vellum.workflows.events.node import (
     NodeExecutionRejectedBody,
     NodeExecutionStreamingBody,
 )
-from vellum.workflows.events.types import BaseEvent
+from vellum.workflows.events.types import BaseEvent, ParentContext, WorkflowParentContext
 from vellum.workflows.events.utils import is_terminal_event
 from vellum.workflows.events.workflow import (
     WorkflowExecutionFulfilledBody,
@@ -71,6 +71,7 @@ class WorkflowRunner(Generic[StateType]):
         entrypoint_nodes: Optional[RunFromNodeArg] = None,
         external_inputs: Optional[ExternalInputsArg] = None,
         cancel_signal: Optional[ThreadingEvent] = None,
+        parent_context: Optional[ParentContext] = None,
     ):
         if state and external_inputs:
             raise ValueError("Can only run a Workflow providing one of state or external inputs, not both")
@@ -115,6 +116,7 @@ class WorkflowRunner(Generic[StateType]):
 
         self._active_nodes_by_execution_id: Dict[UUID, BaseNode[StateType]] = {}
         self._cancel_signal = cancel_signal
+        self._parent_context = parent_context
 
         setattr(
             self._initial_state,
@@ -143,6 +145,11 @@ class WorkflowRunner(Generic[StateType]):
                         node_definition=node.__class__,
                         inputs=node._inputs,
                     ),
+                    parent=WorkflowParentContext(
+                        span_id=span_id,
+                        workflow_definition=self.workflow.__class__,
+                        parent=self._parent_context
+                    )
                 ),
             )
         )
@@ -190,6 +197,11 @@ class WorkflowRunner(Generic[StateType]):
                                     output=BaseOutput(name=output.name),
                                     invoked_ports=invoked_ports,
                                 ),
+                                parent=WorkflowParentContext(
+                                    span_id=span_id,
+                                    workflow_definition=self.workflow.__class__,
+                                    parent=self._parent_context
+                                )
                             ),
                         )
                     )
@@ -214,6 +226,11 @@ class WorkflowRunner(Generic[StateType]):
                                         output=output,
                                         invoked_ports=invoked_ports,
                                     ),
+                                    parent=WorkflowParentContext(
+                                        span_id=span_id,
+                                        workflow_definition=self.workflow.__class__,
+                                        parent=self._parent_context,
+                                    )
                                 ),
                             )
                         )
@@ -233,6 +250,11 @@ class WorkflowRunner(Generic[StateType]):
                                         output=output,
                                         invoked_ports=invoked_ports,
                                     ),
+                                    parent=WorkflowParentContext(
+                                        span_id=span_id,
+                                        workflow_definition=self.workflow.__class__,
+                                        parent=self._parent_context,
+                                    )
                                 ),
                             )
                         )
@@ -259,6 +281,11 @@ class WorkflowRunner(Generic[StateType]):
                             outputs=outputs,
                             invoked_ports=invoked_ports,
                         ),
+                        parent=WorkflowParentContext(
+                            span_id=span_id,
+                            workflow_definition=self.workflow.__class__,
+                            parent=self._parent_context,
+                        )
                     ),
                 )
             )
@@ -273,6 +300,11 @@ class WorkflowRunner(Generic[StateType]):
                             node_definition=node.__class__,
                             error=e.error,
                         ),
+                        parent=WorkflowParentContext(
+                            span_id=span_id,
+                            workflow_definition=self.workflow.__class__,
+                            parent=self._parent_context,
+                        )
                     ),
                 )
             )
@@ -292,6 +324,11 @@ class WorkflowRunner(Generic[StateType]):
                                 code=VellumErrorCode.INTERNAL_ERROR,
                             ),
                         ),
+                        parent=WorkflowParentContext(
+                            span_id=span_id,
+                            workflow_definition=self.workflow.__class__,
+                            parent=self._parent_context
+                        )
                     ),
                 )
             )
@@ -387,6 +424,7 @@ class WorkflowRunner(Generic[StateType]):
                 workflow_definition=self.workflow.__class__,
                 inputs=self._initial_state.meta.workflow_inputs,
             ),
+            parent=self._parent_context,
         )
 
     def _stream_workflow_event(self, output: BaseOutput) -> WorkflowExecutionStreamingEvent:
@@ -397,6 +435,7 @@ class WorkflowRunner(Generic[StateType]):
                 workflow_definition=self.workflow.__class__,
                 output=output,
             ),
+            parent=self._parent_context
         )
 
     def _fulfill_workflow_event(self, outputs: OutputsType) -> WorkflowExecutionFulfilledEvent:
@@ -407,6 +446,7 @@ class WorkflowRunner(Generic[StateType]):
                 workflow_definition=self.workflow.__class__,
                 outputs=outputs,
             ),
+            parent=self._parent_context,
         )
 
     def _reject_workflow_event(self, error: VellumError) -> WorkflowExecutionRejectedEvent:
@@ -417,6 +457,7 @@ class WorkflowRunner(Generic[StateType]):
                 workflow_definition=self.workflow.__class__,
                 error=error,
             ),
+            parent=self._parent_context,
         )
 
     def _resume_workflow_event(self) -> WorkflowExecutionResumedEvent:
@@ -436,6 +477,7 @@ class WorkflowRunner(Generic[StateType]):
                 workflow_definition=self.workflow.__class__,
                 external_inputs=external_inputs,
             ),
+            parent=self._parent_context,
         )
 
     def _stream(self) -> None:

vellum_ai-0.10.8/src/vellum/workflows/types/__init__.py

@@ -0,0 +1,5 @@
+from .core import MergeBehavior
+
+__all__ = [
+    "MergeBehavior",
+]

vellum_ai-0.10.7/src/vellum/workflows/nodes/displayable/code_execution_node/utils.py

@@ -1,10 +0,0 @@
-import os
-from typing import Union
-
-
-def read_file_from_path(filepath: str) -> Union[str, None]:
-    if not os.path.exists(filepath):
-        return None
-
-    with open(filepath) as file:
-        return file.read()