vellum-ai 0.14.46__py3-none-any.whl → 0.14.48__py3-none-any.whl

vellum/client/types/workflow_execution_initiated_event.py

@@ -1,6 +1,5 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from __future__ import annotations
 from ..core.pydantic_utilities import UniversalBaseModel
 from .api_request_parent_context import ApiRequestParentContext
 from .node_parent_context import NodeParentContext
@@ -16,7 +15,6 @@ import datetime as dt
 from .api_version_enum import ApiVersionEnum
 from ..core.pydantic_utilities import IS_PYDANTIC_V2
 import pydantic
-from ..core.pydantic_utilities import update_forward_refs
 
 
 class WorkflowExecutionInitiatedEvent(UniversalBaseModel):
@@ -38,12 +36,3 @@ class WorkflowExecutionInitiatedEvent(UniversalBaseModel):
             frozen = True
             smart_union = True
             extra = pydantic.Extra.allow
-
-
-update_forward_refs(ApiRequestParentContext, WorkflowExecutionInitiatedEvent=WorkflowExecutionInitiatedEvent)
-update_forward_refs(NodeParentContext, WorkflowExecutionInitiatedEvent=WorkflowExecutionInitiatedEvent)
-update_forward_refs(PromptDeploymentParentContext, WorkflowExecutionInitiatedEvent=WorkflowExecutionInitiatedEvent)
-update_forward_refs(SpanLink, WorkflowExecutionInitiatedEvent=WorkflowExecutionInitiatedEvent)
-update_forward_refs(WorkflowDeploymentParentContext, WorkflowExecutionInitiatedEvent=WorkflowExecutionInitiatedEvent)
-update_forward_refs(WorkflowParentContext, WorkflowExecutionInitiatedEvent=WorkflowExecutionInitiatedEvent)
-update_forward_refs(WorkflowSandboxParentContext, WorkflowExecutionInitiatedEvent=WorkflowExecutionInitiatedEvent)

vellum/client/types/workflow_execution_node_result_event.py

@@ -1,13 +1,11 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from __future__ import annotations
 from ..core.pydantic_utilities import UniversalBaseModel
 from .array_vellum_value import ArrayVellumValue
 import typing
 from .workflow_node_result_event import WorkflowNodeResultEvent
 from ..core.pydantic_utilities import IS_PYDANTIC_V2
 import pydantic
-from ..core.pydantic_utilities import update_forward_refs
 
 
 class WorkflowExecutionNodeResultEvent(UniversalBaseModel):
@@ -29,6 +27,3 @@ class WorkflowExecutionNodeResultEvent(UniversalBaseModel):
             frozen = True
             smart_union = True
             extra = pydantic.Extra.allow
-
-
-update_forward_refs(ArrayVellumValue, WorkflowExecutionNodeResultEvent=WorkflowExecutionNodeResultEvent)

vellum/client/types/workflow_execution_paused_event.py

@@ -1,6 +1,5 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from __future__ import annotations
 from ..core.pydantic_utilities import UniversalBaseModel
 from .api_request_parent_context import ApiRequestParentContext
 from .node_parent_context import NodeParentContext
@@ -16,7 +15,6 @@ import datetime as dt
 from .api_version_enum import ApiVersionEnum
 from ..core.pydantic_utilities import IS_PYDANTIC_V2
 import pydantic
-from ..core.pydantic_utilities import update_forward_refs
 
 
 class WorkflowExecutionPausedEvent(UniversalBaseModel):
@@ -38,12 +36,3 @@ class WorkflowExecutionPausedEvent(UniversalBaseModel):
             frozen = True
             smart_union = True
             extra = pydantic.Extra.allow
-
-
-update_forward_refs(ApiRequestParentContext, WorkflowExecutionPausedEvent=WorkflowExecutionPausedEvent)
-update_forward_refs(NodeParentContext, WorkflowExecutionPausedEvent=WorkflowExecutionPausedEvent)
-update_forward_refs(PromptDeploymentParentContext, WorkflowExecutionPausedEvent=WorkflowExecutionPausedEvent)
-update_forward_refs(SpanLink, WorkflowExecutionPausedEvent=WorkflowExecutionPausedEvent)
-update_forward_refs(WorkflowDeploymentParentContext, WorkflowExecutionPausedEvent=WorkflowExecutionPausedEvent)
-update_forward_refs(WorkflowParentContext, WorkflowExecutionPausedEvent=WorkflowExecutionPausedEvent)
-update_forward_refs(WorkflowSandboxParentContext, WorkflowExecutionPausedEvent=WorkflowExecutionPausedEvent)

vellum/client/types/workflow_execution_rejected_event.py

@@ -1,6 +1,5 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from __future__ import annotations
 from ..core.pydantic_utilities import UniversalBaseModel
 from .api_request_parent_context import ApiRequestParentContext
 from .node_parent_context import NodeParentContext
@@ -16,7 +15,6 @@ import datetime as dt
 from .api_version_enum import ApiVersionEnum
 from ..core.pydantic_utilities import IS_PYDANTIC_V2
 import pydantic
-from ..core.pydantic_utilities import update_forward_refs
 
 
 class WorkflowExecutionRejectedEvent(UniversalBaseModel):
@@ -38,12 +36,3 @@ class WorkflowExecutionRejectedEvent(UniversalBaseModel):
             frozen = True
             smart_union = True
             extra = pydantic.Extra.allow
-
-
-update_forward_refs(ApiRequestParentContext, WorkflowExecutionRejectedEvent=WorkflowExecutionRejectedEvent)
-update_forward_refs(NodeParentContext, WorkflowExecutionRejectedEvent=WorkflowExecutionRejectedEvent)
-update_forward_refs(PromptDeploymentParentContext, WorkflowExecutionRejectedEvent=WorkflowExecutionRejectedEvent)
-update_forward_refs(SpanLink, WorkflowExecutionRejectedEvent=WorkflowExecutionRejectedEvent)
-update_forward_refs(WorkflowDeploymentParentContext, WorkflowExecutionRejectedEvent=WorkflowExecutionRejectedEvent)
-update_forward_refs(WorkflowParentContext, WorkflowExecutionRejectedEvent=WorkflowExecutionRejectedEvent)
-update_forward_refs(WorkflowSandboxParentContext, WorkflowExecutionRejectedEvent=WorkflowExecutionRejectedEvent)

vellum/client/types/workflow_execution_resumed_event.py

@@ -1,6 +1,5 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from __future__ import annotations
 from ..core.pydantic_utilities import UniversalBaseModel
 from .api_request_parent_context import ApiRequestParentContext
 from .node_parent_context import NodeParentContext
@@ -16,7 +15,6 @@ import datetime as dt
 from .api_version_enum import ApiVersionEnum
 from ..core.pydantic_utilities import IS_PYDANTIC_V2
 import pydantic
-from ..core.pydantic_utilities import update_forward_refs
 
 
 class WorkflowExecutionResumedEvent(UniversalBaseModel):
@@ -38,12 +36,3 @@ class WorkflowExecutionResumedEvent(UniversalBaseModel):
             frozen = True
             smart_union = True
             extra = pydantic.Extra.allow
-
-
-update_forward_refs(ApiRequestParentContext, WorkflowExecutionResumedEvent=WorkflowExecutionResumedEvent)
-update_forward_refs(NodeParentContext, WorkflowExecutionResumedEvent=WorkflowExecutionResumedEvent)
-update_forward_refs(PromptDeploymentParentContext, WorkflowExecutionResumedEvent=WorkflowExecutionResumedEvent)
-update_forward_refs(SpanLink, WorkflowExecutionResumedEvent=WorkflowExecutionResumedEvent)
-update_forward_refs(WorkflowDeploymentParentContext, WorkflowExecutionResumedEvent=WorkflowExecutionResumedEvent)
-update_forward_refs(WorkflowParentContext, WorkflowExecutionResumedEvent=WorkflowExecutionResumedEvent)
-update_forward_refs(WorkflowSandboxParentContext, WorkflowExecutionResumedEvent=WorkflowExecutionResumedEvent)

vellum/client/types/workflow_execution_snapshotted_event.py

@@ -1,6 +1,5 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from __future__ import annotations
 from ..core.pydantic_utilities import UniversalBaseModel
 from .api_request_parent_context import ApiRequestParentContext
 from .node_parent_context import NodeParentContext
@@ -16,7 +15,6 @@ import datetime as dt
 from .api_version_enum import ApiVersionEnum
 from ..core.pydantic_utilities import IS_PYDANTIC_V2
 import pydantic
-from ..core.pydantic_utilities import update_forward_refs
 
 
 class WorkflowExecutionSnapshottedEvent(UniversalBaseModel):
@@ -38,14 +36,3 @@ class WorkflowExecutionSnapshottedEvent(UniversalBaseModel):
             frozen = True
             smart_union = True
             extra = pydantic.Extra.allow
-
-
-update_forward_refs(ApiRequestParentContext, WorkflowExecutionSnapshottedEvent=WorkflowExecutionSnapshottedEvent)
-update_forward_refs(NodeParentContext, WorkflowExecutionSnapshottedEvent=WorkflowExecutionSnapshottedEvent)
-update_forward_refs(PromptDeploymentParentContext, WorkflowExecutionSnapshottedEvent=WorkflowExecutionSnapshottedEvent)
-update_forward_refs(SpanLink, WorkflowExecutionSnapshottedEvent=WorkflowExecutionSnapshottedEvent)
-update_forward_refs(
-    WorkflowDeploymentParentContext, WorkflowExecutionSnapshottedEvent=WorkflowExecutionSnapshottedEvent
-)
-update_forward_refs(WorkflowParentContext, WorkflowExecutionSnapshottedEvent=WorkflowExecutionSnapshottedEvent)
-update_forward_refs(WorkflowSandboxParentContext, WorkflowExecutionSnapshottedEvent=WorkflowExecutionSnapshottedEvent)

vellum/client/types/workflow_execution_span.py

@@ -1,6 +1,5 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from __future__ import annotations
 from ..core.pydantic_utilities import UniversalBaseModel
 from .api_request_parent_context import ApiRequestParentContext
 from .node_parent_context import NodeParentContext
@@ -15,7 +14,6 @@ from .workflow_execution_span_attributes import WorkflowExecutionSpanAttributes
 import datetime as dt
 from ..core.pydantic_utilities import IS_PYDANTIC_V2
 import pydantic
-from ..core.pydantic_utilities import update_forward_refs
 
 
 class WorkflowExecutionSpan(UniversalBaseModel):
@@ -35,12 +33,3 @@ class WorkflowExecutionSpan(UniversalBaseModel):
             frozen = True
             smart_union = True
             extra = pydantic.Extra.allow
-
-
-update_forward_refs(ApiRequestParentContext, WorkflowExecutionSpan=WorkflowExecutionSpan)
-update_forward_refs(NodeParentContext, WorkflowExecutionSpan=WorkflowExecutionSpan)
-update_forward_refs(PromptDeploymentParentContext, WorkflowExecutionSpan=WorkflowExecutionSpan)
-update_forward_refs(SpanLink, WorkflowExecutionSpan=WorkflowExecutionSpan)
-update_forward_refs(WorkflowDeploymentParentContext, WorkflowExecutionSpan=WorkflowExecutionSpan)
-update_forward_refs(WorkflowParentContext, WorkflowExecutionSpan=WorkflowExecutionSpan)
-update_forward_refs(WorkflowSandboxParentContext, WorkflowExecutionSpan=WorkflowExecutionSpan)

vellum/client/types/workflow_execution_streaming_event.py

@@ -1,6 +1,5 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from __future__ import annotations
 from ..core.pydantic_utilities import UniversalBaseModel
 from .api_request_parent_context import ApiRequestParentContext
 from .node_parent_context import NodeParentContext
@@ -16,7 +15,6 @@ import datetime as dt
 from .api_version_enum import ApiVersionEnum
 from ..core.pydantic_utilities import IS_PYDANTIC_V2
 import pydantic
-from ..core.pydantic_utilities import update_forward_refs
 
 
 class WorkflowExecutionStreamingEvent(UniversalBaseModel):
@@ -38,12 +36,3 @@ class WorkflowExecutionStreamingEvent(UniversalBaseModel):
             frozen = True
             smart_union = True
             extra = pydantic.Extra.allow
-
-
-update_forward_refs(ApiRequestParentContext, WorkflowExecutionStreamingEvent=WorkflowExecutionStreamingEvent)
-update_forward_refs(NodeParentContext, WorkflowExecutionStreamingEvent=WorkflowExecutionStreamingEvent)
-update_forward_refs(PromptDeploymentParentContext, WorkflowExecutionStreamingEvent=WorkflowExecutionStreamingEvent)
-update_forward_refs(SpanLink, WorkflowExecutionStreamingEvent=WorkflowExecutionStreamingEvent)
-update_forward_refs(WorkflowDeploymentParentContext, WorkflowExecutionStreamingEvent=WorkflowExecutionStreamingEvent)
-update_forward_refs(WorkflowParentContext, WorkflowExecutionStreamingEvent=WorkflowExecutionStreamingEvent)
-update_forward_refs(WorkflowSandboxParentContext, WorkflowExecutionStreamingEvent=WorkflowExecutionStreamingEvent)

vellum/client/types/workflow_execution_view_online_eval_metric_result.py

@@ -1,13 +1,11 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from __future__ import annotations
 from ..core.pydantic_utilities import UniversalBaseModel
 from .array_vellum_value import ArrayVellumValue
 import typing
 from .execution_vellum_value import ExecutionVellumValue
 from ..core.pydantic_utilities import IS_PYDANTIC_V2
 import pydantic
-from ..core.pydantic_utilities import update_forward_refs
 
 
 class WorkflowExecutionViewOnlineEvalMetricResult(UniversalBaseModel):
@@ -23,8 +21,3 @@ class WorkflowExecutionViewOnlineEvalMetricResult(UniversalBaseModel):
             frozen = True
             smart_union = True
             extra = pydantic.Extra.allow
-
-
-update_forward_refs(
-    ArrayVellumValue, WorkflowExecutionViewOnlineEvalMetricResult=WorkflowExecutionViewOnlineEvalMetricResult
-)

vellum/client/types/workflow_execution_workflow_result_event.py

@@ -1,13 +1,11 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from __future__ import annotations
 from ..core.pydantic_utilities import UniversalBaseModel
 from .array_vellum_value import ArrayVellumValue
 import typing
 from .workflow_result_event import WorkflowResultEvent
 from ..core.pydantic_utilities import IS_PYDANTIC_V2
 import pydantic
-from ..core.pydantic_utilities import update_forward_refs
 
 
 class WorkflowExecutionWorkflowResultEvent(UniversalBaseModel):
@@ -29,6 +27,3 @@ class WorkflowExecutionWorkflowResultEvent(UniversalBaseModel):
             frozen = True
             smart_union = True
             extra = pydantic.Extra.allow
-
-
-update_forward_refs(ArrayVellumValue, WorkflowExecutionWorkflowResultEvent=WorkflowExecutionWorkflowResultEvent)

vellum/client/types/workflow_output_array.py

@@ -1,13 +1,11 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from __future__ import annotations
 from ..core.pydantic_utilities import UniversalBaseModel
 from .array_vellum_value import ArrayVellumValue
 import pydantic
 import typing
 from .vellum_value import VellumValue
 from ..core.pydantic_utilities import IS_PYDANTIC_V2
-from ..core.pydantic_utilities import update_forward_refs
 
 
 class WorkflowOutputArray(UniversalBaseModel):
@@ -32,6 +30,3 @@ class WorkflowOutputArray(UniversalBaseModel):
             frozen = True
             smart_union = True
             extra = pydantic.Extra.allow
-
-
-update_forward_refs(ArrayVellumValue, WorkflowOutputArray=WorkflowOutputArray)

vellum/client/types/workflow_parent_context.py

@@ -34,10 +34,4 @@ from .workflow_deployment_parent_context import WorkflowDeploymentParentContext
 from .workflow_sandbox_parent_context import WorkflowSandboxParentContext  # noqa: E402
 from .parent_context import ParentContext  # noqa: E402
 
-update_forward_refs(ApiRequestParentContext, WorkflowParentContext=WorkflowParentContext)
-update_forward_refs(NodeParentContext, WorkflowParentContext=WorkflowParentContext)
-update_forward_refs(PromptDeploymentParentContext, WorkflowParentContext=WorkflowParentContext)
-update_forward_refs(SpanLink, WorkflowParentContext=WorkflowParentContext)
-update_forward_refs(WorkflowDeploymentParentContext, WorkflowParentContext=WorkflowParentContext)
-update_forward_refs(WorkflowSandboxParentContext, WorkflowParentContext=WorkflowParentContext)
 update_forward_refs(WorkflowParentContext)

vellum/client/types/workflow_result_event.py

@@ -1,6 +1,5 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from __future__ import annotations
 from ..core.pydantic_utilities import UniversalBaseModel
 from .array_vellum_value import ArrayVellumValue
 from .workflow_node_result_event_state import WorkflowNodeResultEventState
@@ -12,7 +11,6 @@ from .workflow_output import WorkflowOutput
 from .execution_vellum_value import ExecutionVellumValue
 from ..core.pydantic_utilities import IS_PYDANTIC_V2
 import pydantic
-from ..core.pydantic_utilities import update_forward_refs
 
 
 class WorkflowResultEvent(UniversalBaseModel):
@@ -32,6 +30,3 @@ class WorkflowResultEvent(UniversalBaseModel):
             frozen = True
             smart_union = True
             extra = pydantic.Extra.allow
-
-
-update_forward_refs(ArrayVellumValue, WorkflowResultEvent=WorkflowResultEvent)

vellum/client/types/workflow_result_event_output_data_array.py

@@ -1,6 +1,5 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from __future__ import annotations
 from ..core.pydantic_utilities import UniversalBaseModel
 from .array_vellum_value import ArrayVellumValue
 import typing
@@ -8,7 +7,6 @@ from .workflow_node_result_event_state import WorkflowNodeResultEventState
 import pydantic
 from .vellum_value import VellumValue
 from ..core.pydantic_utilities import IS_PYDANTIC_V2
-from ..core.pydantic_utilities import update_forward_refs
 
 
 class WorkflowResultEventOutputDataArray(UniversalBaseModel):
@@ -36,6 +34,3 @@ class WorkflowResultEventOutputDataArray(UniversalBaseModel):
             frozen = True
             smart_union = True
             extra = pydantic.Extra.allow
-
-
-update_forward_refs(ArrayVellumValue, WorkflowResultEventOutputDataArray=WorkflowResultEventOutputDataArray)

vellum/client/types/workflow_sandbox_parent_context.py

@@ -35,10 +35,4 @@ from .workflow_deployment_parent_context import WorkflowDeploymentParentContext
 from .workflow_parent_context import WorkflowParentContext  # noqa: E402
 from .parent_context import ParentContext  # noqa: E402
 
-update_forward_refs(ApiRequestParentContext, WorkflowSandboxParentContext=WorkflowSandboxParentContext)
-update_forward_refs(NodeParentContext, WorkflowSandboxParentContext=WorkflowSandboxParentContext)
-update_forward_refs(PromptDeploymentParentContext, WorkflowSandboxParentContext=WorkflowSandboxParentContext)
-update_forward_refs(SpanLink, WorkflowSandboxParentContext=WorkflowSandboxParentContext)
-update_forward_refs(WorkflowDeploymentParentContext, WorkflowSandboxParentContext=WorkflowSandboxParentContext)
-update_forward_refs(WorkflowParentContext, WorkflowSandboxParentContext=WorkflowSandboxParentContext)
 update_forward_refs(WorkflowSandboxParentContext)

vellum/workflows/nodes/bases/base.py

@@ -28,7 +28,7 @@ from vellum.workflows.types.utils import get_class_attr_names, get_original_base
 from vellum.workflows.utils.uuids import uuid4_from_hash
 
 
-def is_nested_class(nested: Any, parent: Type) -> bool:
+def _is_nested_class(nested: Any, parent: Type) -> bool:
     return (
         inspect.isclass(nested)
         # If a class is defined within a function, we don't consider it nested in the class defining that function
@@ -36,7 +36,18 @@ def is_nested_class(nested: Any, parent: Type) -> bool:
         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__))
-    ) or any(is_nested_class(nested, base) for base in parent.__bases__)
+    ) or any(_is_nested_class(nested, base) for base in parent.__bases__)
+
+
+def _is_annotated(cls: Type, name: str) -> bool:
+    if name in cls.__annotations__:
+        return True
+
+    for base in cls.__bases__:
+        if _is_annotated(base, name):
+            return True
+
+    return False
 
 
 class BaseNodeMeta(type):
@@ -131,12 +142,21 @@ class BaseNodeMeta(type):
         return {"BaseWorkflow": BaseWorkflow}
 
     def __getattribute__(cls, name: str) -> Any:
-        attribute = super().__getattribute__(name)
+        if name.startswith("_"):
+            return super().__getattribute__(name)
+
+        try:
+            attribute = super().__getattribute__(name)
+        except AttributeError as e:
+            if _is_annotated(cls, name):
+                attribute = None
+            else:
+                raise e
+
         if (
-            name.startswith("_")
-            or inspect.isfunction(attribute)
+            inspect.isfunction(attribute)
             or inspect.ismethod(attribute)
-            or is_nested_class(attribute, cls)
+            or _is_nested_class(attribute, cls)
             or isinstance(attribute, (property, cached_property))
             or not issubclass(cls, BaseNode)
         ):

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

@@ -10,6 +10,7 @@ from vellum.workflows.inputs.base import BaseInputs
 from vellum.workflows.nodes import FinalOutputNode
 from vellum.workflows.nodes.bases.base import BaseNode
 from vellum.workflows.outputs.base import BaseOutputs
+from vellum.workflows.references.node import NodeReference
 from vellum.workflows.references.output import OutputReference
 from vellum.workflows.state.base import BaseState, StateMeta
 
@@ -303,3 +304,32 @@ def test_base_node__iterate_over_attributes__preserves_order():
 
         # THEN the attributes are in the correct order
         assert attribute_names == ["baz", "qux", "quux", "foo", "bar"], f"Iteration {i} failed"
+
+
+def test_base_node__node_reference_of_annotation():
+    # GIVEN a node with an annotated attribute
+    class MyNode(BaseNode):
+        foo: str
+
+    # WHEN we reference the attribute
+    node_reference = MyNode.foo
+
+    # THEN the node reference is of the correct type
+    assert isinstance(node_reference, NodeReference)
+    assert node_reference.name == "foo"
+
+
+def test_base_node__node_reference_of_inherited_annotation():
+    # GIVEN a node with an annotated attribute
+    class MyNode(BaseNode):
+        foo: str
+
+    class InheritedNode(MyNode):
+        bar: str
+
+    # WHEN we reference the attribute
+    node_reference = InheritedNode.foo
+
+    # THEN the node reference is of the correct type
+    assert isinstance(node_reference, NodeReference)
+    assert node_reference.name == "foo"

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

@@ -409,6 +409,56 @@ def main(word: str) -> int:
     assert exc_info.value.message == "Expected an output of type 'int', but received 'str'"
 
 
+def test_run_node__run_inline__incorrect_output_type_list():
+    """Confirm that CodeExecutionNodes raise an error if the output type is incorrect during inline execution."""
+
+    # GIVEN a node that subclasses CodeExecutionNode that returns a list but is defined to return an int
+    class ExampleCodeExecutionNode(CodeExecutionNode[BaseState, int]):
+        code = """\
+def main(output: list[str]) -> int:
+    return output.value
+"""
+        runtime = "PYTHON_3_11_6"
+
+        code_inputs = {
+            "output": ["hello", "world"],
+        }
+
+    # WHEN we run the node
+    node = ExampleCodeExecutionNode()
+    with pytest.raises(NodeException) as exc_info:
+        node.run()
+
+    # THEN the node should have produced the exception we expected
+    assert exc_info.value.message == "Expected an output of type 'int', but received 'list'"
+
+
+def test_run_node__run_inline__incorrect_output_type_dict():
+    """Confirm that CodeExecutionNodes raise an error if the output type is incorrect during inline execution."""
+
+    # GIVEN a node that subclasses CodeExecutionNode that returns a dict but is defined to return an int
+    class ExampleCodeExecutionNode(CodeExecutionNode[BaseState, int]):
+        code = """\
+def main(output: dict[str, str]) -> int:
+    return output.value
+"""
+        runtime = "PYTHON_3_11_6"
+
+        code_inputs = {
+            "output": {
+                "foo": "bar",
+            }
+        }
+
+    # WHEN we run the node
+    node = ExampleCodeExecutionNode()
+    with pytest.raises(NodeException) as exc_info:
+        node.run()
+
+    # THEN the node should have produced the exception we expected
+    assert exc_info.value.message == "Expected an output of type 'int', but received 'dict'"
+
+
 def test_run_node__run_inline__valid_dict_to_pydantic():
     """Confirm that CodeExecutionNodes can convert a dict to a Pydantic model during inline execution."""
 

vellum/workflows/nodes/utils.py

@@ -15,7 +15,7 @@ from vellum.workflows.nodes import BaseNode
 from vellum.workflows.nodes.bases.base_adornment_node import BaseAdornmentNode
 from vellum.workflows.ports.port import Port
 from vellum.workflows.state.base import BaseState
-from vellum.workflows.types.code_execution_node_wrappers import StringValueWrapper
+from vellum.workflows.types.code_execution_node_wrappers import DictWrapper, ListWrapper, StringValueWrapper
 from vellum.workflows.types.core import Json
 from vellum.workflows.types.generics import NodeType
 
@@ -180,6 +180,10 @@ def parse_type_from_str(result_as_str: str, output_type: Any) -> Any:
 def _get_type_name(obj: Any) -> str:
     if inspect.isclass(obj) and issubclass(obj, StringValueWrapper):
         return "str"
+    if inspect.isclass(obj) and issubclass(obj, ListWrapper):
+        return "list"
+    if inspect.isclass(obj) and issubclass(obj, DictWrapper):
+        return "dict"
     if isinstance(obj, type):
         return obj.__name__
 

vellum/workflows/types/code_execution_node_wrappers.py

@@ -18,6 +18,11 @@ class ListWrapper(list):
 
         return super().__getitem__(key)
 
+    def __getattr__(self, attr):
+        if attr == "value":
+            return self
+        raise AttributeError(f"'list' object has no attribute '{attr}'")
+
 
 class DictWrapper(dict):
     """

{vellum_ai-0.14.46.dist-info → vellum_ai-0.14.48.dist-info}/METADATA

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