vellum-ai 0.14.4__py3-none-any.whl → 0.14.6__py3-none-any.whl

vellum/__init__.py

@@ -259,6 +259,7 @@ from .types import (
     PaginatedTestSuiteRunExecutionList,
     PaginatedTestSuiteTestCaseList,
     PaginatedWorkflowReleaseTagReadList,
+    PaginatedWorkflowSandboxExampleList,
     PdfSearchResultMetaSource,
     PdfSearchResultMetaSourceRequest,
     PlainTextPromptBlock,
@@ -495,6 +496,7 @@ from .types import (
     WorkflowResultEventOutputDataNumber,
     WorkflowResultEventOutputDataSearchResults,
     WorkflowResultEventOutputDataString,
+    WorkflowSandboxExample,
     WorkflowStreamEvent,
     WorkspaceRead,
     WorkspaceSecretRead,
@@ -506,6 +508,7 @@ from .resources import (
     FolderEntitiesListRequestEntityStatus,
     ListDeploymentReleaseTagsRequestSource,
     ListWorkflowReleaseTagsRequestSource,
+    ListWorkflowSandboxExamplesRequestTag,
     WorkflowDeploymentsListRequestStatus,
     WorkflowsPullRequestFormat,
     ad_hoc,
@@ -713,6 +716,7 @@ __all__ = [
     "JsonVellumValueRequest",
     "ListDeploymentReleaseTagsRequestSource",
     "ListWorkflowReleaseTagsRequestSource",
+    "ListWorkflowSandboxExamplesRequestTag",
     "LogicalOperator",
     "LogprobsEnum",
     "MapNodeResult",
@@ -796,6 +800,7 @@ __all__ = [
     "PaginatedTestSuiteRunExecutionList",
     "PaginatedTestSuiteTestCaseList",
     "PaginatedWorkflowReleaseTagReadList",
+    "PaginatedWorkflowSandboxExampleList",
     "PdfSearchResultMetaSource",
     "PdfSearchResultMetaSourceRequest",
     "PlainTextPromptBlock",
@@ -1035,6 +1040,7 @@ __all__ = [
     "WorkflowResultEventOutputDataNumber",
     "WorkflowResultEventOutputDataSearchResults",
     "WorkflowResultEventOutputDataString",
+    "WorkflowSandboxExample",
     "WorkflowStreamEvent",
     "WorkflowsPullRequestFormat",
     "WorkspaceRead",

vellum/client/__init__.py

@@ -360,7 +360,7 @@ class Vellum:
             Optionally specify a release tag if you want to pin to a specific release of the Prompt Deployment
 
         external_id : typing.Optional[str]
-            Optionally include a unique identifier for tracking purposes. Must be unique within a given Prompt Deployment.
+            Optionally include a unique identifier for tracking purposes. Must be unique within a given Workspace.
 
         expand_meta : typing.Optional[PromptDeploymentExpandMetaRequest]
             An optionally specified configuration used to opt in to including additional metadata about this prompt execution in the API response. Corresponding values will be returned under the `meta` key of the API response.
@@ -508,7 +508,7 @@ class Vellum:
             Optionally specify a release tag if you want to pin to a specific release of the Prompt Deployment
 
         external_id : typing.Optional[str]
-            Optionally include a unique identifier for tracking purposes. Must be unique within a given Prompt Deployment.
+            Optionally include a unique identifier for tracking purposes. Must be unique within a given Workspace.
 
         expand_meta : typing.Optional[PromptDeploymentExpandMetaRequest]
             An optionally specified configuration used to opt in to including additional metadata about this prompt execution in the API response. Corresponding values will be returned under the `meta` key of the API response.
@@ -667,7 +667,7 @@ class Vellum:
             Optionally specify a release tag if you want to pin to a specific release of the Workflow Deployment
 
         external_id : typing.Optional[str]
-            Optionally include a unique identifier for tracking purposes. Must be unique for a given workflow deployment.
+            Optionally include a unique identifier for tracking purposes. Must be unique within a given Workspace.
 
         metadata : typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]]
             Arbitrary JSON metadata associated with this request. Can be used to capture additional monitoring data such as user id, session id, etc. for future analysis.
@@ -794,7 +794,7 @@ class Vellum:
             Optionally specify a release tag if you want to pin to a specific release of the Workflow Deployment
 
         external_id : typing.Optional[str]
-            Optionally include a unique identifier for tracking purposes. Must be unique for a given workflow deployment.
+            Optionally include a unique identifier for tracking purposes. Must be unique within a given Workspace.
 
         event_types : typing.Optional[typing.Sequence[WorkflowExecutionEventType]]
             Optionally specify which events you want to receive. Defaults to only WORKFLOW events. Note that the schema of non-WORKFLOW events is unstable and should be used with caution.
@@ -1717,7 +1717,7 @@ class AsyncVellum:
             Optionally specify a release tag if you want to pin to a specific release of the Prompt Deployment
 
         external_id : typing.Optional[str]
-            Optionally include a unique identifier for tracking purposes. Must be unique within a given Prompt Deployment.
+            Optionally include a unique identifier for tracking purposes. Must be unique within a given Workspace.
 
         expand_meta : typing.Optional[PromptDeploymentExpandMetaRequest]
             An optionally specified configuration used to opt in to including additional metadata about this prompt execution in the API response. Corresponding values will be returned under the `meta` key of the API response.
@@ -1873,7 +1873,7 @@ class AsyncVellum:
             Optionally specify a release tag if you want to pin to a specific release of the Prompt Deployment
 
         external_id : typing.Optional[str]
-            Optionally include a unique identifier for tracking purposes. Must be unique within a given Prompt Deployment.
+            Optionally include a unique identifier for tracking purposes. Must be unique within a given Workspace.
 
         expand_meta : typing.Optional[PromptDeploymentExpandMetaRequest]
             An optionally specified configuration used to opt in to including additional metadata about this prompt execution in the API response. Corresponding values will be returned under the `meta` key of the API response.
@@ -2040,7 +2040,7 @@ class AsyncVellum:
             Optionally specify a release tag if you want to pin to a specific release of the Workflow Deployment
 
         external_id : typing.Optional[str]
-            Optionally include a unique identifier for tracking purposes. Must be unique for a given workflow deployment.
+            Optionally include a unique identifier for tracking purposes. Must be unique within a given Workspace.
 
         metadata : typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]]
             Arbitrary JSON metadata associated with this request. Can be used to capture additional monitoring data such as user id, session id, etc. for future analysis.
@@ -2175,7 +2175,7 @@ class AsyncVellum:
             Optionally specify a release tag if you want to pin to a specific release of the Workflow Deployment
 
         external_id : typing.Optional[str]
-            Optionally include a unique identifier for tracking purposes. Must be unique for a given workflow deployment.
+            Optionally include a unique identifier for tracking purposes. Must be unique within a given Workspace.
 
         event_types : typing.Optional[typing.Sequence[WorkflowExecutionEventType]]
             Optionally specify which events you want to receive. Defaults to only WORKFLOW events. Note that the schema of non-WORKFLOW events is unstable and should be used with caution.

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

vellum/client/resources/__init__.py

@@ -23,6 +23,7 @@ from .deployments import DeploymentsListRequestStatus, ListDeploymentReleaseTags
 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__ = [
@@ -31,6 +32,7 @@ __all__ = [
     "FolderEntitiesListRequestEntityStatus",
     "ListDeploymentReleaseTagsRequestSource",
     "ListWorkflowReleaseTagsRequestSource",
+    "ListWorkflowSandboxExamplesRequestTag",
     "WorkflowDeploymentsListRequestStatus",
     "WorkflowsPullRequestFormat",
     "ad_hoc",

vellum/client/resources/workflow_sandboxes/__init__.py

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

vellum/client/resources/workflow_sandboxes/client.py

@@ -8,6 +8,8 @@ from ...core.jsonable_encoder import jsonable_encoder
 from ...core.pydantic_utilities import parse_obj_as
 from json.decoder import JSONDecodeError
 from ...core.api_error import ApiError
+from .types.list_workflow_sandbox_examples_request_tag import ListWorkflowSandboxExamplesRequestTag
+from ...types.paginated_workflow_sandbox_example_list import PaginatedWorkflowSandboxExampleList
 from ...core.client_wrapper import AsyncClientWrapper
 
 # this is used as the default value for optional parameters
@@ -97,6 +99,74 @@ class WorkflowSandboxesClient:
             raise ApiError(status_code=_response.status_code, body=_response.text)
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
+    def list_workflow_sandbox_examples(
+        self,
+        *,
+        limit: typing.Optional[int] = None,
+        offset: typing.Optional[int] = None,
+        ordering: typing.Optional[str] = None,
+        tag: typing.Optional[ListWorkflowSandboxExamplesRequestTag] = None,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> PaginatedWorkflowSandboxExampleList:
+        """
+        List Workflow Sandbox examples that were previously cloned into the User's Workspace
+
+        Parameters
+        ----------
+        limit : typing.Optional[int]
+            Number of results to return per page.
+
+        offset : typing.Optional[int]
+            The initial index from which to return the results.
+
+        ordering : typing.Optional[str]
+            Which field to use when ordering the results.
+
+        tag : typing.Optional[ListWorkflowSandboxExamplesRequestTag]
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        PaginatedWorkflowSandboxExampleList
+
+
+        Examples
+        --------
+        from vellum import Vellum
+
+        client = Vellum(
+            api_key="YOUR_API_KEY",
+        )
+        client.workflow_sandboxes.list_workflow_sandbox_examples()
+        """
+        _response = self._client_wrapper.httpx_client.request(
+            "v1/workflow-sandboxes/examples",
+            base_url=self._client_wrapper.get_environment().default,
+            method="GET",
+            params={
+                "limit": limit,
+                "offset": offset,
+                "ordering": ordering,
+                "tag": tag,
+            },
+            request_options=request_options,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                return typing.cast(
+                    PaginatedWorkflowSandboxExampleList,
+                    parse_obj_as(
+                        type_=PaginatedWorkflowSandboxExampleList,  # type: ignore
+                        object_=_response.json(),
+                    ),
+                )
+            _response_json = _response.json()
+        except JSONDecodeError:
+            raise ApiError(status_code=_response.status_code, body=_response.text)
+        raise ApiError(status_code=_response.status_code, body=_response_json)
+
 
 class AsyncWorkflowSandboxesClient:
     def __init__(self, *, client_wrapper: AsyncClientWrapper):
@@ -188,3 +258,79 @@ class AsyncWorkflowSandboxesClient:
         except JSONDecodeError:
             raise ApiError(status_code=_response.status_code, body=_response.text)
         raise ApiError(status_code=_response.status_code, body=_response_json)
+
+    async def list_workflow_sandbox_examples(
+        self,
+        *,
+        limit: typing.Optional[int] = None,
+        offset: typing.Optional[int] = None,
+        ordering: typing.Optional[str] = None,
+        tag: typing.Optional[ListWorkflowSandboxExamplesRequestTag] = None,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> PaginatedWorkflowSandboxExampleList:
+        """
+        List Workflow Sandbox examples that were previously cloned into the User's Workspace
+
+        Parameters
+        ----------
+        limit : typing.Optional[int]
+            Number of results to return per page.
+
+        offset : typing.Optional[int]
+            The initial index from which to return the results.
+
+        ordering : typing.Optional[str]
+            Which field to use when ordering the results.
+
+        tag : typing.Optional[ListWorkflowSandboxExamplesRequestTag]
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        PaginatedWorkflowSandboxExampleList
+
+
+        Examples
+        --------
+        import asyncio
+
+        from vellum import AsyncVellum
+
+        client = AsyncVellum(
+            api_key="YOUR_API_KEY",
+        )
+
+
+        async def main() -> None:
+            await client.workflow_sandboxes.list_workflow_sandbox_examples()
+
+
+        asyncio.run(main())
+        """
+        _response = await self._client_wrapper.httpx_client.request(
+            "v1/workflow-sandboxes/examples",
+            base_url=self._client_wrapper.get_environment().default,
+            method="GET",
+            params={
+                "limit": limit,
+                "offset": offset,
+                "ordering": ordering,
+                "tag": tag,
+            },
+            request_options=request_options,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                return typing.cast(
+                    PaginatedWorkflowSandboxExampleList,
+                    parse_obj_as(
+                        type_=PaginatedWorkflowSandboxExampleList,  # type: ignore
+                        object_=_response.json(),
+                    ),
+                )
+            _response_json = _response.json()
+        except JSONDecodeError:
+            raise ApiError(status_code=_response.status_code, body=_response.text)
+        raise ApiError(status_code=_response.status_code, body=_response_json)

vellum/client/resources/workflow_sandboxes/types/__init__.py

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

vellum/client/resources/workflow_sandboxes/types/list_workflow_sandbox_examples_request_tag.py

@@ -0,0 +1,5 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+ListWorkflowSandboxExamplesRequestTag = typing.Union[typing.Literal["ONBOARDING", "TEMPLATES"], typing.Any]

vellum/client/types/__init__.py

@@ -267,6 +267,7 @@ from .paginated_slim_workflow_deployment_list import PaginatedSlimWorkflowDeploy
 from .paginated_test_suite_run_execution_list import PaginatedTestSuiteRunExecutionList
 from .paginated_test_suite_test_case_list import PaginatedTestSuiteTestCaseList
 from .paginated_workflow_release_tag_read_list import PaginatedWorkflowReleaseTagReadList
+from .paginated_workflow_sandbox_example_list import PaginatedWorkflowSandboxExampleList
 from .pdf_search_result_meta_source import PdfSearchResultMetaSource
 from .pdf_search_result_meta_source_request import PdfSearchResultMetaSourceRequest
 from .plain_text_prompt_block import PlainTextPromptBlock
@@ -519,6 +520,7 @@ from .workflow_result_event_output_data_json import WorkflowResultEventOutputDat
 from .workflow_result_event_output_data_number import WorkflowResultEventOutputDataNumber
 from .workflow_result_event_output_data_search_results import WorkflowResultEventOutputDataSearchResults
 from .workflow_result_event_output_data_string import WorkflowResultEventOutputDataString
+from .workflow_sandbox_example import WorkflowSandboxExample
 from .workflow_stream_event import WorkflowStreamEvent
 from .workspace_read import WorkspaceRead
 from .workspace_secret_read import WorkspaceSecretRead
@@ -779,6 +781,7 @@ __all__ = [
     "PaginatedTestSuiteRunExecutionList",
     "PaginatedTestSuiteTestCaseList",
     "PaginatedWorkflowReleaseTagReadList",
+    "PaginatedWorkflowSandboxExampleList",
     "PdfSearchResultMetaSource",
     "PdfSearchResultMetaSourceRequest",
     "PlainTextPromptBlock",
@@ -1015,6 +1018,7 @@ __all__ = [
     "WorkflowResultEventOutputDataNumber",
     "WorkflowResultEventOutputDataSearchResults",
     "WorkflowResultEventOutputDataString",
+    "WorkflowSandboxExample",
     "WorkflowStreamEvent",
     "WorkspaceRead",
     "WorkspaceSecretRead",

vellum/client/types/paginated_workflow_sandbox_example_list.py

@@ -0,0 +1,23 @@
+# This file was auto-generated by Fern from our API Definition.
+
+from ..core.pydantic_utilities import UniversalBaseModel
+import typing
+from .workflow_sandbox_example import WorkflowSandboxExample
+from ..core.pydantic_utilities import IS_PYDANTIC_V2
+import pydantic
+
+
+class PaginatedWorkflowSandboxExampleList(UniversalBaseModel):
+    count: typing.Optional[int] = None
+    next: typing.Optional[str] = None
+    previous: typing.Optional[str] = None
+    results: typing.Optional[typing.List[WorkflowSandboxExample]] = None
+
+    if IS_PYDANTIC_V2:
+        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
+    else:
+
+        class Config:
+            frozen = True
+            smart_union = True
+            extra = pydantic.Extra.allow

vellum/client/types/workflow_sandbox_example.py

@@ -0,0 +1,22 @@
+# This file was auto-generated by Fern from our API Definition.
+
+from ..core.pydantic_utilities import UniversalBaseModel
+import typing
+from ..core.pydantic_utilities import IS_PYDANTIC_V2
+import pydantic
+
+
+class WorkflowSandboxExample(UniversalBaseModel):
+    id: str
+    label: str
+    description: typing.Optional[str] = None
+    icon_name: typing.Optional[str] = None
+
+    if IS_PYDANTIC_V2:
+        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
+    else:
+
+        class Config:
+            frozen = True
+            smart_union = True
+            extra = pydantic.Extra.allow

vellum/resources/workflow_sandboxes/types/__init__.py

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

vellum/resources/workflow_sandboxes/types/list_workflow_sandbox_examples_request_tag.py

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

vellum/types/paginated_workflow_sandbox_example_list.py

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

vellum/types/workflow_sandbox_example.py

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

vellum/workflows/context.py

@@ -1,5 +1,6 @@
 from contextlib import contextmanager
 import threading
+from uuid import UUID
 from typing import Iterator, Optional, cast
 
 from vellum.client.core import UniversalBaseModel
@@ -8,6 +9,7 @@ from vellum.workflows.events.types import ParentContext
 
 class ExecutionContext(UniversalBaseModel):
     parent_context: Optional[ParentContext] = None
+    trace_id: Optional[UUID] = None
 
 
 _CONTEXT_KEY = "_execution_context"
@@ -30,11 +32,14 @@ def get_parent_context() -> ParentContext:
 
 
 @contextmanager
-def execution_context(parent_context: Optional[ParentContext] = None) -> Iterator[None]:
+def execution_context(
+    parent_context: Optional[ParentContext] = None, trace_id: Optional[UUID] = None
+) -> Iterator[None]:
     """Context manager for handling execution context."""
     prev_context = get_execution_context()
-    set_context = ExecutionContext(parent_context=parent_context) if parent_context else prev_context
-
+    set_trace_id = prev_context.trace_id or trace_id
+    set_parent_context = parent_context or prev_context.parent_context
+    set_context = ExecutionContext(parent_context=set_parent_context, trace_id=set_trace_id)
     try:
         set_execution_context(set_context)
         yield

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

@@ -1,12 +1,16 @@
 import pytest
 import os
-from typing import Any, Union
+from typing import Any, List, Union
+
+from pydantic import BaseModel
 
 from vellum import CodeExecutorResponse, NumberVellumValue, StringInput, StringVellumValue
+from vellum.client.types.chat_message import ChatMessage
 from vellum.client.types.code_execution_package import CodeExecutionPackage
 from vellum.client.types.code_executor_secret_input import CodeExecutorSecretInput
 from vellum.client.types.function_call import FunctionCall
 from vellum.client.types.number_input import NumberInput
+from vellum.client.types.string_chat_message_content import StringChatMessageContent
 from vellum.workflows.errors import WorkflowErrorCode
 from vellum.workflows.exceptions import NodeException
 from vellum.workflows.inputs.base import BaseInputs
@@ -610,3 +614,79 @@ def main(arg1: list[bool]) -> int:
 
     # AND the error should contain the execution error details
     assert outputs == {"result": 3, "log": ""}
+
+
+def test_run_node__union_output_type__pydantic_children():
+    # GIVEN a node that is a union type with a pydantic child
+    class OptionOne(BaseModel):
+        foo: str
+
+    class OptionTwo(BaseModel):
+        bar: int
+
+    class ExampleCodeExecutionNode(CodeExecutionNode[BaseState, Union[OptionOne, OptionTwo]]):
+        code = """\
+def main():
+    return { "foo": "hello" }
+"""
+        runtime = "PYTHON_3_11_6"
+        code_inputs = {}
+
+    # WHEN we run the node
+    node = ExampleCodeExecutionNode()
+
+    # THEN it should run successfully
+    outputs = node.run()
+
+    # AND the result should be the correct type
+    assert outputs == {"result": OptionOne(foo="hello"), "log": ""}
+
+
+def test_run_node__union_output_type__miss():
+    # GIVEN a node that is a union type
+    class ExampleCodeExecutionNode(CodeExecutionNode[BaseState, Union[int, float]]):
+        code = """\
+def main():
+    return "hello"
+"""
+        runtime = "PYTHON_3_11_6"
+        code_inputs = {}
+
+    # WHEN we run the node
+    node = ExampleCodeExecutionNode()
+
+    # THEN it should raise a NodeException with the execution error
+    with pytest.raises(NodeException) as exc_info:
+        node.run()
+
+    # AND the error should contain the execution error details
+    assert exc_info.value.message == "Expected an output of type 'int | float', but received 'str'"
+
+
+def test_run_node__chat_history_output_type():
+    # GIVEN a node that that has a chat history return type
+    class ExampleCodeExecutionNode(CodeExecutionNode[BaseState, List[ChatMessage]]):
+        code = """\
+def main():
+    return [
+        {
+            "role": "USER",
+            "content": {
+                "type": "STRING",
+                "value": "Hello, world!",
+            }
+        }
+    ]
+"""
+        code_inputs = {}
+        runtime = "PYTHON_3_11_6"
+
+    # WHEN we run the node
+    node = ExampleCodeExecutionNode()
+    outputs = node.run()
+
+    # AND the error should contain the execution error details
+    assert outputs == {
+        "result": [ChatMessage(role="USER", content=StringChatMessageContent(value="Hello, world!"))],
+        "log": "",
+    }

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

@@ -67,6 +67,49 @@ def _clean_for_dict_wrapper(obj):
     return obj
 
 
+def _get_type_name(obj: Any) -> str:
+    if isinstance(obj, type):
+        return obj.__name__
+
+    if get_origin(obj) is Union:
+        children = [_get_type_name(child) for child in get_args(obj)]
+        return " | ".join(children)
+
+    return str(obj)
+
+
+def _cast_to_output_type(result: Any, output_type: Any) -> Any:
+    is_valid_output_type = isinstance(output_type, type)
+    if get_origin(output_type) is Union:
+        allowed_types = get_args(output_type)
+        for allowed_type in allowed_types:
+            try:
+                return _cast_to_output_type(result, allowed_type)
+            except NodeException:
+                continue
+    elif get_origin(output_type) is list:
+        allowed_item_type = get_args(output_type)[0]
+        if isinstance(result, list):
+            return [_cast_to_output_type(item, allowed_item_type) for item in result]
+    elif is_valid_output_type and issubclass(output_type, BaseModel) and not isinstance(result, output_type):
+        try:
+            return output_type.model_validate(result)
+        except ValidationError as e:
+            raise NodeException(
+                code=WorkflowErrorCode.INVALID_OUTPUTS,
+                message=re.sub(r"\s+For further information visit [^\s]+", "", str(e)),
+            ) from e
+    elif is_valid_output_type and isinstance(result, output_type):
+        return result
+
+    output_type_name = _get_type_name(output_type)
+    result_type_name = _get_type_name(type(result))
+    raise NodeException(
+        code=WorkflowErrorCode.INVALID_OUTPUTS,
+        message=f"Expected an output of type '{output_type_name}', but received '{result_type_name}'",
+    )
+
+
 def run_code_inline(
     code: str,
     inputs: EntityInputsInterface,
@@ -112,25 +155,6 @@ __arg__out = main({", ".join(run_args)})
     result = exec_globals["__arg__out"]
 
     if output_type != Any:
-        if get_origin(output_type) is Union:
-            allowed_types = get_args(output_type)
-            if not isinstance(result, allowed_types):
-                raise NodeException(
-                    code=WorkflowErrorCode.INVALID_OUTPUTS,
-                    message=f"Expected output to be in types {allowed_types}, but received '{type(result).__name__}'",
-                )
-        elif issubclass(output_type, BaseModel) and not isinstance(result, output_type):
-            try:
-                result = output_type.model_validate(result)
-            except ValidationError as e:
-                raise NodeException(
-                    code=WorkflowErrorCode.INVALID_OUTPUTS,
-                    message=re.sub(r"\s+For further information visit [^\s]+", "", str(e)),
-                ) from e
-        elif not isinstance(result, output_type):
-            raise NodeException(
-                code=WorkflowErrorCode.INVALID_OUTPUTS,
-                message=f"Expected an output of type '{output_type.__name__}', but received '{type(result).__name__}'",
-            )
+        result = _cast_to_output_type(result, output_type)
 
     return logs, result