vellum-ai 0.14.25__py3-none-any.whl → 0.14.26__py3-none-any.whl

vellum/__init__.py

@@ -322,7 +322,7 @@ from .types import (
     RejectedExecuteWorkflowWorkflowResultEvent,
     RejectedPromptExecutionMeta,
     RejectedWorkflowNodeResultEvent,
-    Release,
+    ReleaseTagRelease,
     ReleaseTagSource,
     ReplaceTestSuiteTestCaseRequest,
     RichTextChildBlock,
@@ -580,7 +580,6 @@ from .resources import (
     ListWorkflowReleaseTagsRequestSource,
     ListWorkflowSandboxExamplesRequestTag,
     WorkflowDeploymentsListRequestStatus,
-    WorkflowsPullRequestFormat,
     ad_hoc,
     container_images,
     deployments,
@@ -933,7 +932,7 @@ __all__ = [
     "RejectedExecuteWorkflowWorkflowResultEvent",
     "RejectedPromptExecutionMeta",
     "RejectedWorkflowNodeResultEvent",
-    "Release",
+    "ReleaseTagRelease",
     "ReleaseTagSource",
     "ReplaceTestSuiteTestCaseRequest",
     "RichTextChildBlock",
@@ -1182,7 +1181,6 @@ __all__ = [
     "WorkflowSandboxExample",
     "WorkflowSandboxParentContext",
     "WorkflowStreamEvent",
-    "WorkflowsPullRequestFormat",
     "WorkspaceRead",
     "WorkspaceSecretRead",
     "__version__",

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.14.25",
+            "X-Fern-SDK-Version": "0.14.26",
         }
         headers["X_API_KEY"] = self.api_key
         return headers

vellum/client/resources/__init__.py

@@ -24,7 +24,6 @@ from .document_indexes import DocumentIndexesListRequestStatus
 from .folder_entities import FolderEntitiesListRequestEntityStatus
 from .workflow_deployments import ListWorkflowReleaseTagsRequestSource, WorkflowDeploymentsListRequestStatus
 from .workflow_sandboxes import ListWorkflowSandboxExamplesRequestTag
-from .workflows import WorkflowsPullRequestFormat
 
 __all__ = [
     "DeploymentsListRequestStatus",
@@ -34,7 +33,6 @@ __all__ = [
     "ListWorkflowReleaseTagsRequestSource",
     "ListWorkflowSandboxExamplesRequestTag",
     "WorkflowDeploymentsListRequestStatus",
-    "WorkflowsPullRequestFormat",
     "ad_hoc",
     "container_images",
     "deployments",

vellum/client/resources/workflows/__init__.py

@@ -1,5 +1,2 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from .types import WorkflowsPullRequestFormat
-
-__all__ = ["WorkflowsPullRequestFormat"]

vellum/client/resources/workflows/client.py

@@ -2,7 +2,6 @@
 
 import typing
 from ...core.client_wrapper import SyncClientWrapper
-from .types.workflows_pull_request_format import WorkflowsPullRequestFormat
 from ...core.request_options import RequestOptions
 from ...core.jsonable_encoder import jsonable_encoder
 from ...errors.bad_request_error import BadRequestError
@@ -28,7 +27,6 @@ class WorkflowsClient:
         id: str,
         *,
         exclude_code: typing.Optional[bool] = None,
-        format: typing.Optional[WorkflowsPullRequestFormat] = None,
         include_json: typing.Optional[bool] = None,
         include_sandbox: typing.Optional[bool] = None,
         strict: typing.Optional[bool] = None,
@@ -42,8 +40,6 @@ class WorkflowsClient:
 
         exclude_code : typing.Optional[bool]
 
-        format : typing.Optional[WorkflowsPullRequestFormat]
-
         include_json : typing.Optional[bool]
 
         include_sandbox : typing.Optional[bool]
@@ -64,7 +60,6 @@ class WorkflowsClient:
             method="GET",
             params={
                 "exclude_code": exclude_code,
-                "format": format,
                 "include_json": include_json,
                 "include_sandbox": include_sandbox,
                 "strict": strict,
@@ -180,7 +175,6 @@ class AsyncWorkflowsClient:
         id: str,
         *,
         exclude_code: typing.Optional[bool] = None,
-        format: typing.Optional[WorkflowsPullRequestFormat] = None,
         include_json: typing.Optional[bool] = None,
         include_sandbox: typing.Optional[bool] = None,
         strict: typing.Optional[bool] = None,
@@ -194,8 +188,6 @@ class AsyncWorkflowsClient:
 
         exclude_code : typing.Optional[bool]
 
-        format : typing.Optional[WorkflowsPullRequestFormat]
-
         include_json : typing.Optional[bool]
 
         include_sandbox : typing.Optional[bool]
@@ -216,7 +208,6 @@ class AsyncWorkflowsClient:
             method="GET",
             params={
                 "exclude_code": exclude_code,
-                "format": format,
                 "include_json": include_json,
                 "include_sandbox": include_sandbox,
                 "strict": strict,

vellum/client/types/__init__.py

@@ -330,7 +330,7 @@ from .rejected_execute_prompt_response import RejectedExecutePromptResponse
 from .rejected_execute_workflow_workflow_result_event import RejectedExecuteWorkflowWorkflowResultEvent
 from .rejected_prompt_execution_meta import RejectedPromptExecutionMeta
 from .rejected_workflow_node_result_event import RejectedWorkflowNodeResultEvent
-from .release import Release
+from .release_tag_release import ReleaseTagRelease
 from .release_tag_source import ReleaseTagSource
 from .replace_test_suite_test_case_request import ReplaceTestSuiteTestCaseRequest
 from .rich_text_child_block import RichTextChildBlock
@@ -914,7 +914,7 @@ __all__ = [
     "RejectedExecuteWorkflowWorkflowResultEvent",
     "RejectedPromptExecutionMeta",
     "RejectedWorkflowNodeResultEvent",
-    "Release",
+    "ReleaseTagRelease",
     "ReleaseTagSource",
     "ReplaceTestSuiteTestCaseRequest",
     "RichTextChildBlock",

vellum/client/types/deployment_release_tag_read.py

@@ -4,6 +4,7 @@ from ..core.pydantic_utilities import UniversalBaseModel
 import pydantic
 from .release_tag_source import ReleaseTagSource
 from .deployment_release_tag_deployment_history_item import DeploymentReleaseTagDeploymentHistoryItem
+from .release_tag_release import ReleaseTagRelease
 from ..core.pydantic_utilities import IS_PYDANTIC_V2
 import typing
 
@@ -24,7 +25,12 @@ class DeploymentReleaseTagRead(UniversalBaseModel):
 
     history_item: DeploymentReleaseTagDeploymentHistoryItem = pydantic.Field()
     """
-    The Deployment History Item that this Release Tag is associated with
+    Deprecated. Reference the `release` field instead.
+    """
+
+    release: ReleaseTagRelease = pydantic.Field()
+    """
+    The Release that this Release Tag points to.
     """
 
     if IS_PYDANTIC_V2:

vellum/client/types/{release.py → release_tag_release.py}

@@ -7,7 +7,7 @@ import typing
 import pydantic
 
 
-class Release(UniversalBaseModel):
+class ReleaseTagRelease(UniversalBaseModel):
     id: str
     timestamp: dt.datetime
 

vellum/client/types/workflow_release_tag_read.py

@@ -4,7 +4,7 @@ from ..core.pydantic_utilities import UniversalBaseModel
 import pydantic
 from .release_tag_source import ReleaseTagSource
 from .workflow_release_tag_workflow_deployment_history_item import WorkflowReleaseTagWorkflowDeploymentHistoryItem
-from .release import Release
+from .release_tag_release import ReleaseTagRelease
 from ..core.pydantic_utilities import IS_PYDANTIC_V2
 import typing
 
@@ -28,7 +28,7 @@ class WorkflowReleaseTagRead(UniversalBaseModel):
     Deprecated. Reference the `release` field instead.
     """
 
-    release: Release = pydantic.Field()
+    release: ReleaseTagRelease = pydantic.Field()
     """
     The Release that this Release Tag points to.
     """

vellum/client/types/workflow_release_tag_workflow_deployment_history_item.py

@@ -1,22 +1,15 @@
 # This file was auto-generated by Fern from our API Definition.
 
 from ..core.pydantic_utilities import UniversalBaseModel
-import pydantic
 import datetime as dt
 from ..core.pydantic_utilities import IS_PYDANTIC_V2
 import typing
+import pydantic
 
 
 class WorkflowReleaseTagWorkflowDeploymentHistoryItem(UniversalBaseModel):
-    id: str = pydantic.Field()
-    """
-    The ID of the Workflow Deployment History Item
-    """
-
-    timestamp: dt.datetime = pydantic.Field()
-    """
-    The timestamp representing when this History Item was created
-    """
+    id: str
+    timestamp: dt.datetime
 
     if IS_PYDANTIC_V2:
         model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2

vellum/types/{release.py → release_tag_release.py}

@@ -1,3 +1,3 @@
 # WARNING: This file will be removed in a future release. Please import from "vellum.client" instead.
 
-from vellum.client.types.release import *
+from vellum.client.types.release_tag_release import *

vellum/workflows/events/types.py

@@ -28,13 +28,16 @@ def serialize_type_encoder(obj: type) -> Dict[str, Any]:
     }
 
 
-def serialize_type_encoder_with_id(obj: type) -> Dict[str, Any]:
-    if not hasattr(obj, "__id__"):
-        raise AttributeError(f"The object of type '{type(obj).__name__}' must have an '__id__' attribute.")
-    return {
-        "id": getattr(obj, "__id__"),
-        **serialize_type_encoder(obj),
-    }
+def serialize_type_encoder_with_id(obj: Union[type, "CodeResourceDefinition"]) -> Dict[str, Any]:
+    if hasattr(obj, "__id__") and isinstance(obj, type):
+        return {
+            "id": getattr(obj, "__id__"),
+            **serialize_type_encoder(obj),
+        }
+    elif isinstance(obj, CodeResourceDefinition):
+        return obj.model_dump(mode="json")
+
+    raise AttributeError(f"The object of type '{type(obj).__name__}' must have an '__id__' attribute.")
 
 
 def default_serializer(obj: Any) -> Any:

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

@@ -87,13 +87,11 @@ class BaseInlinePromptNode(BasePromptNode[StateType], Generic[StateType]):
 
     def _get_prompt_event_stream(self) -> Iterator[AdHocExecutePromptEvent]:
         input_variables, input_values = self._compile_prompt_inputs()
-        current_context = get_execution_context()
-        parent_context = current_context.parent_context
-        trace_id = current_context.trace_id
+        execution_context = get_execution_context()
         request_options = self.request_options or RequestOptions()
 
         request_options["additional_body_parameters"] = {
-            "execution_context": {"parent_context": parent_context, "trace_id": trace_id},
+            "execution_context": execution_context.model_dump(mode="json"),
             **request_options.get("additional_body_parameters", {}),
         }
         normalized_functions = (

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

@@ -55,12 +55,10 @@ class BasePromptDeploymentNode(BasePromptNode, Generic[StateType]):
         merge_behavior = MergeBehavior.AWAIT_ANY
 
     def _get_prompt_event_stream(self) -> Iterator[ExecutePromptEvent]:
-        current_context = get_execution_context()
-        trace_id = current_context.trace_id
-        parent_context = current_context.parent_context.model_dump() if current_context.parent_context else None
+        execution_context = get_execution_context()
         request_options = self.request_options or RequestOptions()
         request_options["additional_body_parameters"] = {
-            "execution_context": {"parent_context": parent_context, "trace_id": trace_id},
+            "execution_context": execution_context.model_dump(mode="json"),
             **request_options.get("additional_body_parameters", {}),
         }
         return self._context.vellum_client.execute_prompt_stream(

vellum/workflows/nodes/displayable/conftest.py

@@ -0,0 +1,117 @@
+import pytest
+from uuid import UUID
+
+from vellum.workflows.events.types import (
+    CodeResourceDefinition,
+    NodeParentContext,
+    WorkflowDeploymentParentContext,
+    WorkflowParentContext,
+)
+
+
+@pytest.fixture
+def mock_complex_parent_context():
+    # TODO: We were able to confirm that this parent context caused our serialization to hang, but don't know why yet.
+    # We should try to reduce this example further to isolate a minimal example that reproduces the issue.
+    return NodeParentContext(
+        span_id=UUID("d697f8c8-b363-4154-8469-eb4f9fb5e445"),
+        parent=WorkflowParentContext(
+            span_id=UUID("a0c68884-22c3-4ac9-8476-8747884d80e1"),
+            parent=NodeParentContext(
+                span_id=UUID("46163407-71f7-40f2-9f66-872d4b338fcc"),
+                parent=WorkflowParentContext(
+                    span_id=UUID("0ddf01e7-d0c3-426c-af27-d8bfb22fcdd5"),
+                    parent=NodeParentContext(
+                        span_id=UUID("79a6c926-b5f3-4ede-b2f8-4bb6f0c086ba"),
+                        parent=WorkflowParentContext(
+                            span_id=UUID("530a56fe-90fd-4f4c-b905-457975fb3e10"),
+                            parent=WorkflowDeploymentParentContext(
+                                span_id=UUID("3e10a8c2-558c-4ef7-926d-7b79ebc7cba9"),
+                                parent=NodeParentContext(
+                                    span_id=UUID("a3cd4086-c0b9-4dff-88f3-3e2191b8a2a7"),
+                                    parent=WorkflowParentContext(
+                                        span_id=UUID("c2ba7577-8d24-49b1-aa92-b9ace8244090"),
+                                        workflow_definition=CodeResourceDefinition(
+                                            id=UUID("2e2d5c56-49b7-48b5-82fa-e80e72768b9c"),
+                                            name="Workflow",
+                                            module=["e81a6124-2c57-4c39-938c-ab6059059ff2", "workflow"],
+                                        ),
+                                    ),
+                                    node_definition=CodeResourceDefinition(
+                                        id=UUID("23d25675-f377-4450-916f-39ebee5c8ea9"),
+                                        name="SubworkflowDeployment",
+                                        module=[
+                                            "e81a6124-2c57-4c39-938c-ab6059059ff2",
+                                            "nodes",
+                                            "subworkflow_deployment",
+                                        ],
+                                    ),
+                                ),
+                                deployment_id=UUID("cfc99610-2869-4506-b106-3fd7ce0bbb15"),
+                                deployment_name="my-deployment",
+                                deployment_history_item_id=UUID("13f31aae-29fd-4066-a4ec-c7687faebae3"),
+                                release_tag_id=UUID("2d03987a-dcb5-49b9-981e-5e871c8f5d97"),
+                                release_tag_name="LATEST",
+                                external_id=None,
+                                metadata=None,
+                                workflow_version_id=UUID("7eaae816-b5f3-436d-8597-e8c3e4a32958"),
+                            ),
+                            workflow_definition=CodeResourceDefinition(
+                                id=UUID("2e2d5c56-49b7-48b5-82fa-e80e72768b9c"),
+                                name="Workflow",
+                                module=["3e10a8c2-558c-4ef7-926d-7b79ebc7cba9", "workflow"],
+                            ),
+                        ),
+                        node_definition=CodeResourceDefinition(
+                            id=UUID("42c8adc2-a0d6-499e-81a4-e2e02d7beba9"),
+                            name="MyNode",
+                            module=[
+                                "3e10a8c2-558c-4ef7-926d-7b79ebc7cba9",
+                                "nodes",
+                                "my_node",
+                            ],
+                        ),
+                    ),
+                    workflow_definition=CodeResourceDefinition(
+                        id=UUID("b8563da0-7fd4-42e0-a75e-9ef037fca5a1"),
+                        name="MyNodeWorkflow",
+                        module=[
+                            "3e10a8c2-558c-4ef7-926d-7b79ebc7cba9",
+                            "nodes",
+                            "my_node",
+                            "workflow",
+                        ],
+                    ),
+                ),
+                node_definition=CodeResourceDefinition(
+                    id=UUID("d44aee53-3b6e-41fd-8b7a-908cb2c77821"),
+                    name="RetryNode",
+                    module=[
+                        "3e10a8c2-558c-4ef7-926d-7b79ebc7cba9",
+                        "nodes",
+                        "my_node",
+                        "nodes",
+                        "my_prompt",
+                        "MyPrompt",
+                        "<adornment>",
+                    ],
+                ),
+            ),
+            workflow_definition=CodeResourceDefinition(
+                id=UUID("568a28dd-7134-436e-a5f4-790675212b51"),
+                name="Subworkflow",
+                module=["vellum", "workflows", "nodes", "utils"],
+            ),
+        ),
+        node_definition=CodeResourceDefinition(
+            id=UUID("86a34e5c-2652-49f0-9f9e-c653cf70029a"),
+            name="MyPrompt",
+            module=[
+                "3e10a8c2-558c-4ef7-926d-7b79ebc7cba9",
+                "nodes",
+                "my_node",
+                "nodes",
+                "my_prompt",
+            ],
+        ),
+    )

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

@@ -1,8 +1,11 @@
 import pytest
 from dataclasses import dataclass
+import json
 from uuid import uuid4
 from typing import Any, Iterator, List
 
+from httpx import Response
+
 from vellum.client.core.api_error import ApiError
 from vellum.client.core.pydantic_utilities import UniversalBaseModel
 from vellum.client.types.chat_message import ChatMessage
@@ -17,6 +20,7 @@ from vellum.client.types.prompt_output import PromptOutput
 from vellum.client.types.prompt_request_chat_history_input import PromptRequestChatHistoryInput
 from vellum.client.types.prompt_request_json_input import PromptRequestJsonInput
 from vellum.client.types.string_vellum_value import StringVellumValue
+from vellum.workflows.context import execution_context
 from vellum.workflows.errors.types import WorkflowErrorCode
 from vellum.workflows.exceptions import NodeException
 from vellum.workflows.nodes.displayable.inline_prompt_node.node import InlinePromptNode
@@ -230,3 +234,48 @@ def test_inline_prompt_node__chat_history_inputs(vellum_adhoc_prompt_client):
         ),
     ]
     assert mock_api.call_args.kwargs["input_variables"][0].type == "CHAT_HISTORY"
+
+
+@pytest.mark.timeout(5)
+def test_inline_prompt_node__parent_context(mock_httpx_transport, mock_complex_parent_context):
+    # GIVEN a prompt node
+    class MyNode(InlinePromptNode):
+        ml_model = "gpt-4o"
+        blocks = []
+        prompt_inputs = {}
+
+    # AND a known response from the httpx client
+    expected_outputs: List[PromptOutput] = [
+        StringVellumValue(value="Test"),
+    ]
+    execution_id = str(uuid4())
+    events: List[ExecutePromptEvent] = [
+        InitiatedExecutePromptEvent(execution_id=execution_id),
+        FulfilledExecutePromptEvent(
+            execution_id=execution_id,
+            outputs=expected_outputs,
+        ),
+    ]
+    text = "\n".join(e.model_dump_json() for e in events)
+
+    mock_httpx_transport.handle_request.return_value = Response(
+        status_code=200,
+        text=text,
+    )
+
+    # WHEN the node is run with the complex parent context
+    trace_id = uuid4()
+    with execution_context(
+        parent_context=mock_complex_parent_context,
+        trace_id=trace_id,
+    ):
+        outputs = list(MyNode().run())
+
+    # THEN the last output is as expected
+    assert outputs[-1].value == "Test"
+
+    # AND the prompt is executed with the correct execution context
+    call_request_args = mock_httpx_transport.handle_request.call_args_list[0][0][0]
+    request_execution_context = json.loads(call_request_args.read().decode("utf-8"))["execution_context"]
+    assert request_execution_context["trace_id"] == str(trace_id)
+    assert request_execution_context["parent_context"]

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

@@ -1,7 +1,10 @@
 import pytest
+import json
 from uuid import uuid4
 from typing import Any, Iterator, List
 
+from httpx import Response
+
 from vellum.client.types.chat_history_input_request import ChatHistoryInputRequest
 from vellum.client.types.chat_message import ChatMessage
 from vellum.client.types.chat_message_request import ChatMessageRequest
@@ -9,7 +12,9 @@ from vellum.client.types.execute_prompt_event import ExecutePromptEvent
 from vellum.client.types.fulfilled_execute_prompt_event import FulfilledExecutePromptEvent
 from vellum.client.types.initiated_execute_prompt_event import InitiatedExecutePromptEvent
 from vellum.client.types.json_input_request import JsonInputRequest
+from vellum.client.types.prompt_output import PromptOutput
 from vellum.client.types.string_vellum_value import StringVellumValue
+from vellum.workflows.context import execution_context
 from vellum.workflows.nodes.displayable.prompt_deployment_node.node import PromptDeploymentNode
 
 
@@ -94,3 +99,47 @@ def test_run_node__any_array_input(vellum_client):
     assert call_kwargs["inputs"] == [
         JsonInputRequest(name="fruits", value=["apple", "banana", "cherry"]),
     ]
+
+
+@pytest.mark.timeout(5)
+def test_prompt_deployment_node__parent_context_serialization(mock_httpx_transport, mock_complex_parent_context):
+    # GIVEN a prompt deployment node
+    class MyNode(PromptDeploymentNode):
+        deployment = "example_prompt_deployment"
+        prompt_inputs = {}
+
+    # AND a known response from the httpx client
+    expected_outputs: List[PromptOutput] = [
+        StringVellumValue(value="Test"),
+    ]
+    execution_id = str(uuid4())
+    events: List[ExecutePromptEvent] = [
+        InitiatedExecutePromptEvent(execution_id=execution_id),
+        FulfilledExecutePromptEvent(
+            execution_id=execution_id,
+            outputs=expected_outputs,
+        ),
+    ]
+    text = "\n".join(e.model_dump_json() for e in events)
+
+    mock_httpx_transport.handle_request.return_value = Response(
+        status_code=200,
+        text=text,
+    )
+
+    # WHEN the node is run with a complex parent context
+    trace_id = uuid4()
+    with execution_context(
+        parent_context=mock_complex_parent_context,
+        trace_id=trace_id,
+    ):
+        outputs = list(MyNode().run())
+
+    # THEN the last output is as expected
+    assert outputs[-1].value == "Test"
+
+    # AND the prompt is executed with the correct execution context
+    call_request_args = mock_httpx_transport.handle_request.call_args_list[0][0][0]
+    request_execution_context = json.loads(call_request_args.read().decode("utf-8"))["execution_context"]
+    assert request_execution_context["trace_id"] == str(trace_id)
+    assert request_execution_context["parent_context"]

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

@@ -122,13 +122,10 @@ class SubworkflowDeploymentNode(BaseNode[StateType], Generic[StateType]):
         return compiled_inputs
 
     def run(self) -> Iterator[BaseOutput]:
-        current_context = get_execution_context()
-        parent_context = (
-            current_context.parent_context.model_dump(mode="json") if current_context.parent_context else None
-        )
+        execution_context = get_execution_context()
         request_options = self.request_options or RequestOptions()
         request_options["additional_body_parameters"] = {
-            "execution_context": {"parent_context": parent_context, "trace_id": current_context.trace_id},
+            "execution_context": execution_context.model_dump(mode="json"),
             **request_options.get("additional_body_parameters", {}),
         }
 

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

@@ -1,8 +1,11 @@
 import pytest
 from datetime import datetime
+import json
 from uuid import uuid4
 from typing import Any, Iterator, List
 
+from httpx import Response
+
 from vellum.client.core.api_error import ApiError
 from vellum.client.types.chat_message import ChatMessage
 from vellum.client.types.chat_message_request import ChatMessageRequest
@@ -13,6 +16,7 @@ from vellum.client.types.workflow_request_json_input_request import WorkflowRequ
 from vellum.client.types.workflow_request_number_input_request import WorkflowRequestNumberInputRequest
 from vellum.client.types.workflow_result_event import WorkflowResultEvent
 from vellum.client.types.workflow_stream_event import WorkflowStreamEvent
+from vellum.workflows.context import execution_context
 from vellum.workflows.errors import WorkflowErrorCode
 from vellum.workflows.exceptions import NodeException
 from vellum.workflows.nodes.displayable.subworkflow_deployment_node.node import SubworkflowDeploymentNode
@@ -405,3 +409,62 @@ def test_subworkflow_deployment_node__immediate_api_error__node_exception(vellum
     # THEN the node raises the correct NodeException
     assert e.value.code == WorkflowErrorCode.INVALID_INPUTS
     assert e.value.message == "Not found"
+
+
+@pytest.mark.timeout(5)
+def test_prompt_deployment_node__parent_context_serialization(mock_httpx_transport, mock_complex_parent_context):
+    # GIVEN a prompt deployment node
+    class MyNode(SubworkflowDeploymentNode):
+        deployment = "example_subworkflow_deployment"
+        subworkflow_inputs = {}
+
+    # AND a known response from the httpx client
+    execution_id = str(uuid4())
+    events: List[WorkflowStreamEvent] = [
+        WorkflowExecutionWorkflowResultEvent(
+            execution_id=execution_id,
+            data=WorkflowResultEvent(
+                id=str(uuid4()),
+                state="INITIATED",
+                ts=datetime.now(),
+            ),
+        ),
+        WorkflowExecutionWorkflowResultEvent(
+            execution_id=execution_id,
+            data=WorkflowResultEvent(
+                id=str(uuid4()),
+                state="FULFILLED",
+                ts=datetime.now(),
+                outputs=[
+                    WorkflowOutputString(
+                        id=str(uuid4()),
+                        name="final-output_copy",  # Note the hyphen here
+                        value="Test",
+                    )
+                ],
+            ),
+        ),
+    ]
+    text = "\n".join(e.model_dump_json() for e in events)
+
+    mock_httpx_transport.handle_request.return_value = Response(
+        status_code=200,
+        text=text,
+    )
+
+    # WHEN the node is run with a complex parent context
+    trace_id = uuid4()
+    with execution_context(
+        parent_context=mock_complex_parent_context,
+        trace_id=trace_id,
+    ):
+        outputs = list(MyNode().run())
+
+    # THEN the last output is as expected
+    assert outputs[-1].value == "Test"
+
+    # AND the prompt is executed with the correct execution context
+    call_request_args = mock_httpx_transport.handle_request.call_args_list[0][0][0]
+    request_execution_context = json.loads(call_request_args.read().decode("utf-8"))["execution_context"]
+    assert request_execution_context["trace_id"] == str(trace_id)
+    assert request_execution_context["parent_context"]

vellum/workflows/references/workflow_input.py

@@ -35,6 +35,9 @@ class WorkflowInputReference(BaseDescriptor[_InputType], Generic[_InputType]):
         if state.meta.parent:
             return self.resolve(state.meta.parent)
 
+        if type(None) in self.types:
+            return cast(_InputType, None)
+
         raise NodeException(f"Missing required Workflow input: {self._name}", code=WorkflowErrorCode.INVALID_INPUTS)
 
     def __repr__(self) -> str:

{vellum_ai-0.14.25.dist-info → vellum_ai-0.14.26.dist-info}/METADATA

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