vellum-ai 0.14.5__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.5",
+            "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/nodes/displayable/subworkflow_deployment_node/node.py

@@ -127,10 +127,20 @@ class SubworkflowDeploymentNode(BaseNode[StateType], Generic[StateType]):
             "execution_context": {"parent_context": parent_context},
             **request_options.get("additional_body_parameters", {}),
         }
+
+        try:
+            deployment_id = str(self.deployment) if isinstance(self.deployment, UUID) else None
+            deployment_name = self.deployment if isinstance(self.deployment, str) else None
+        except AttributeError:
+            raise NodeException(
+                code=WorkflowErrorCode.NODE_EXECUTION,
+                message="Expected subworkflow deployment attribute to be either a UUID or STR, got None instead",
+            )
+
         subworkflow_stream = self._context.vellum_client.execute_workflow_stream(
             inputs=self._compile_subworkflow_inputs(),
-            workflow_deployment_id=str(self.deployment) if isinstance(self.deployment, UUID) else None,
-            workflow_deployment_name=self.deployment if isinstance(self.deployment, str) else None,
+            workflow_deployment_id=deployment_id,
+            workflow_deployment_name=deployment_name,
             release_tag=self.release_tag,
             external_id=self.external_id,
             event_types=["WORKFLOW"],

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

@@ -11,6 +11,8 @@ from vellum.client.types.workflow_request_chat_history_input_request import Work
 from vellum.client.types.workflow_request_json_input_request import WorkflowRequestJsonInputRequest
 from vellum.client.types.workflow_result_event import WorkflowResultEvent
 from vellum.client.types.workflow_stream_event import WorkflowStreamEvent
+from vellum.workflows.errors import WorkflowErrorCode
+from vellum.workflows.exceptions import NodeException
 from vellum.workflows.nodes.displayable.subworkflow_deployment_node.node import SubworkflowDeploymentNode
 
 
@@ -129,3 +131,24 @@ def test_run_workflow__any_array(vellum_client):
     assert call_kwargs["inputs"] == [
         WorkflowRequestJsonInputRequest(name="fruits", value=["apple", "banana", "cherry"]),
     ]
+
+
+def test_run_workflow__no_deployment():
+    """Confirm that we raise error when running a subworkflow deployment node with no deployment attribute set"""
+
+    # GIVEN a Subworkflow Deployment Node
+    class ExampleSubworkflowDeploymentNode(SubworkflowDeploymentNode):
+        subworkflow_inputs = {
+            "fruits": ["apple", "banana", "cherry"],
+        }
+
+    # WHEN/THEN running the node should raise a NodeException
+    node = ExampleSubworkflowDeploymentNode()
+    with pytest.raises(NodeException) as exc_info:
+        list(node.run())
+
+    # AND the error message should be correct
+    assert exc_info.value.code == WorkflowErrorCode.NODE_EXECUTION
+    assert "Expected subworkflow deployment attribute to be either a UUID or STR, got None instead" in str(
+        exc_info.value
+    )

vellum/workflows/workflows/base.py

@@ -178,48 +178,64 @@ class BaseWorkflow(Generic[InputsType, StateType], metaclass=_BaseWorkflowMeta):
     def context(self) -> WorkflowContext:
         return self._context
 
-    @classmethod
-    def get_subgraphs(cls) -> List[Graph]:
-        original_graph = cls.graph
-        if isinstance(original_graph, Graph):
-            return [original_graph]
-        if isinstance(original_graph, set):
-            return [
-                subgraph if isinstance(subgraph, Graph) else Graph.from_node(subgraph) for subgraph in original_graph
-            ]
-        if issubclass(original_graph, BaseNode):
-            return [Graph.from_node(original_graph)]
-
-        raise ValueError(f"Unexpected graph type: {original_graph.__class__}")
-
-    @classmethod
-    def get_edges(cls) -> Iterator[Edge]:
+    @staticmethod
+    def _resolve_graph(graph: GraphAttribute) -> List[Graph]:
         """
-        Returns an iterator over the edges in the workflow. We use a set to
-        ensure uniqueness, and the iterator to preserve order.
+        Resolves a single graph source to a list of Graph objects.
         """
+        if isinstance(graph, Graph):
+            return [graph]
+        if isinstance(graph, set):
+            graphs = []
+            for item in graph:
+                if isinstance(item, Graph):
+                    graphs.append(item)
+                elif issubclass(item, BaseNode):
+                    graphs.append(Graph.from_node(item))
+                else:
+                    raise ValueError(f"Unexpected graph type: {type(item)}")
+            return graphs
+        if issubclass(graph, BaseNode):
+            return [Graph.from_node(graph)]
+        raise ValueError(f"Unexpected graph type: {type(graph)}")
 
+    @staticmethod
+    def _get_edges_from_subgraphs(subgraphs: Iterable[Graph]) -> Iterator[Edge]:
         edges = set()
-        subgraphs = cls.get_subgraphs()
         for subgraph in subgraphs:
             for edge in subgraph.edges:
                 if edge not in edges:
                     edges.add(edge)
                     yield edge
 
+    @staticmethod
+    def _get_nodes_from_subgraphs(subgraphs: Iterable[Graph]) -> Iterator[Type[BaseNode]]:
+        nodes = set()
+        for subgraph in subgraphs:
+            for node in subgraph.nodes:
+                if node not in nodes:
+                    nodes.add(node)
+                    yield node
+
+    @classmethod
+    def get_subgraphs(cls) -> List[Graph]:
+        return cls._resolve_graph(cls.graph)
+
+    @classmethod
+    def get_edges(cls) -> Iterator[Edge]:
+        """
+        Returns an iterator over the edges in the workflow. We use a set to
+        ensure uniqueness, and the iterator to preserve order.
+        """
+        return cls._get_edges_from_subgraphs(cls.get_subgraphs())
+
     @classmethod
     def get_nodes(cls) -> Iterator[Type[BaseNode]]:
         """
         Returns an iterator over the nodes in the workflow. We use a set to
         ensure uniqueness, and the iterator to preserve order.
         """
-
-        nodes = set()
-        for subgraph in cls.get_subgraphs():
-            for node in subgraph.nodes:
-                if node not in nodes:
-                    nodes.add(node)
-                    yield node
+        return cls._get_nodes_from_subgraphs(cls.get_subgraphs())
 
     @classmethod
     def get_unused_subgraphs(cls) -> List[Graph]:
@@ -228,19 +244,9 @@ class BaseWorkflow(Generic[InputsType, StateType], metaclass=_BaseWorkflowMeta):
         """
         if not hasattr(cls, "unused_graphs"):
             return []
-
         graphs = []
         for item in cls.unused_graphs:
-            if isinstance(item, Graph):
-                graphs.append(item)
-            elif isinstance(item, set):
-                for subitem in item:
-                    if isinstance(subitem, Graph):
-                        graphs.append(subitem)
-                    elif issubclass(subitem, BaseNode):
-                        graphs.append(Graph.from_node(subitem))
-            elif issubclass(item, BaseNode):
-                graphs.append(Graph.from_node(item))
+            graphs.extend(cls._resolve_graph(item))
         return graphs
 
     @classmethod
@@ -248,29 +254,14 @@ class BaseWorkflow(Generic[InputsType, StateType], metaclass=_BaseWorkflowMeta):
         """
         Returns an iterator over the nodes that are defined but not used in the graph.
         """
-        if not hasattr(cls, "unused_graphs"):
-            yield from ()
-        else:
-            nodes = set()
-            subgraphs = cls.get_unused_subgraphs()
-            for subgraph in subgraphs:
-                for node in subgraph.nodes:
-                    if node not in nodes:
-                        nodes.add(node)
-                        yield node
+        return cls._get_nodes_from_subgraphs(cls.get_unused_subgraphs())
 
     @classmethod
     def get_unused_edges(cls) -> Iterator[Edge]:
         """
         Returns an iterator over edges that are defined but not used in the graph.
         """
-        edges = set()
-        subgraphs = cls.get_unused_subgraphs()
-        for subgraph in subgraphs:
-            for edge in subgraph.edges:
-                if edge not in edges:
-                    edges.add(edge)
-                    yield edge
+        return cls._get_edges_from_subgraphs(cls.get_unused_subgraphs())
 
     @classmethod
     def get_entrypoints(cls) -> Iterable[Type[BaseNode]]:

{vellum_ai-0.14.5.dist-info → vellum_ai-0.14.6.dist-info}/METADATA

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

{vellum_ai-0.14.5.dist-info → vellum_ai-0.14.6.dist-info}/RECORD

@@ -50,9 +50,9 @@ vellum_ee/workflows/display/nodes/vellum/tests/__init__.py,sha256=47DEQpj8HBSa-_
 vellum_ee/workflows/display/nodes/vellum/tests/test_error_node.py,sha256=ulrpoYUW-5kIxfG4Lf5F2p0k_EoYKhmahEbF3P_eruM,1648
 vellum_ee/workflows/display/nodes/vellum/tests/test_prompt_node.py,sha256=bg9INsXiWfyK047u8TD1oEOFYrqDq8GC7Hvgz69n7BE,1988
 vellum_ee/workflows/display/nodes/vellum/tests/test_try_node.py,sha256=mtzB8LJlFCHVFM4H5AanLp29gQfaVmnN4A4iaRGJHoI,2427
-vellum_ee/workflows/display/nodes/vellum/tests/test_utils.py,sha256=3uT7Gbc0f_mQ3u8uZuCWd0mJ4GtWbz2gbUMySYaVlNE,3774
+vellum_ee/workflows/display/nodes/vellum/tests/test_utils.py,sha256=4YUaTeD_OWF-UaPMyOTBTu9skGC1jgSHlAYrzbH7Z04,5039
 vellum_ee/workflows/display/nodes/vellum/try_node.py,sha256=HBfGz4yt9GlmMW9JxzaCacPnHBDNIeXE8Jhqr9DqLLw,6191
-vellum_ee/workflows/display/nodes/vellum/utils.py,sha256=OEGHjQSbuUgJexXI1aubYW33z2F_YdkhQ8REahfz864,4320
+vellum_ee/workflows/display/nodes/vellum/utils.py,sha256=F_0BrlSszllK_BhryPbojIleLq2dGXOfQD1rVp3fNFg,4733
 vellum_ee/workflows/display/tests/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 vellum_ee/workflows/display/tests/test_vellum_workflow_display.py,sha256=1EEvkKQRfOKlnpLxE9-hKSsVLLaelM39LY7007LM5dg,4983
 vellum_ee/workflows/display/tests/workflow_serialization/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -115,12 +115,12 @@ vellum_ee/workflows/tests/local_workflow/workflow.py,sha256=A4qOzOPNwePYxWbcAgIP
 vellum_ee/workflows/tests/test_display_meta.py,sha256=pzdqND4KLWs7EUIbpXuqgso7BIRpoUsO3T_bgeENs0Q,2205
 vellum_ee/workflows/tests/test_server.py,sha256=SvKUrUPmOf3sIInXcFjETekql60npb4cAn1GPbF0bPs,391
 vellum_ee/workflows/tests/test_virtual_files.py,sha256=TJEcMR0v2S8CkloXNmCHA0QW0K6pYNGaIjraJz7sFvY,2762
-vellum/__init__.py,sha256=NPSLAe9-b6SSiyf-FrvMxVy-cKsAEerg8m_HhINTqmE,35794
+vellum/__init__.py,sha256=BGZ28ICgCMzo3Qjj3IN3eMsvylstMCl9C1YKzrQNRnk,36024
 vellum/client/README.md,sha256=JkCJjmMZl4jrPj46pkmL9dpK4gSzQQmP5I7z4aME4LY,4749
-vellum/client/__init__.py,sha256=j6zi0NZ4BMC6JrwckvzMWuG5x8KoOvO4KqsLhvVCa68,117624
+vellum/client/__init__.py,sha256=tKtdM1_GqmGq1gpi9ydWD_T-MM7fPn8QdHh8ww19cNI,117564
 vellum/client/core/__init__.py,sha256=SQ85PF84B9MuKnBwHNHWemSGuy-g_515gFYNFhvEE0I,1438
 vellum/client/core/api_error.py,sha256=RE8LELok2QCjABadECTvtDp7qejA1VmINCh6TbqPwSE,426
-vellum/client/core/client_wrapper.py,sha256=XW9zEBd9XlCTI0wEOgZ6LI4dCsGwvpW4NO7KogWDgao,1868
+vellum/client/core/client_wrapper.py,sha256=Hye69mrkKpKvBVFEfZKH3c08doQVIZ0VGlBRZZmgr0k,1868
 vellum/client/core/datetime_utils.py,sha256=nBys2IsYrhPdszxGKCNRPSOCwa-5DWOHG95FB8G9PKo,1047
 vellum/client/core/file.py,sha256=X9IbmkZmB2bB_DpmZAO3crWdXagOakAyn6UCOCImCPg,2322
 vellum/client/core/http_client.py,sha256=R0pQpCppnEtxccGvXl4uJ76s7ro_65Fo_erlNNLp_AI,19228
@@ -136,7 +136,7 @@ vellum/client/errors/bad_request_error.py,sha256=_EbO8mWqN9kFZPvIap8qa1lL_EWkRcs
 vellum/client/errors/forbidden_error.py,sha256=QO1kKlhClAPES6zsEK7g9pglWnxn3KWaOCAawWOg6Aw,263
 vellum/client/errors/internal_server_error.py,sha256=8USCagXyJJ1MOm9snpcXIUt6eNXvrd_aq7Gfcu1vlOI,268
 vellum/client/errors/not_found_error.py,sha256=tBVCeBC8n3C811WHRj_n-hs3h8MqwR5gp0vLiobk7W8,262
-vellum/client/resources/__init__.py,sha256=j-SQQ4yYDgxbT8MdAhz7BePlsQOX7uvbWM9mluf1FWs,1452
+vellum/client/resources/__init__.py,sha256=g95miLphTYIegm6D-G3sWk-Sf0-5EulFpMGNJMxpp1Y,1567
 vellum/client/resources/ad_hoc/__init__.py,sha256=FTtvy8EDg9nNNg9WCatVgKTRYV8-_v1roeGPAKoa_pw,65
 vellum/client/resources/ad_hoc/client.py,sha256=_liorv4AsoJ55kVu0a5oWB3Qeff0iUKXqoHEIyDWLxc,14173
 vellum/client/resources/container_images/__init__.py,sha256=FTtvy8EDg9nNNg9WCatVgKTRYV8-_v1roeGPAKoa_pw,65
@@ -173,8 +173,10 @@ vellum/client/resources/workflow_deployments/client.py,sha256=H9MhGYZVx1WLHE5j83
 vellum/client/resources/workflow_deployments/types/__init__.py,sha256=W7DKJ1nduwhRckYLvH7wHLdaGH9MXHTZkxwG7FdTngY,340
 vellum/client/resources/workflow_deployments/types/list_workflow_release_tags_request_source.py,sha256=LPETHLX9Ygha_JRT9oWZAZR6clv-W1tTelXzktkTBX8,178
 vellum/client/resources/workflow_deployments/types/workflow_deployments_list_request_status.py,sha256=FXVkVmGM6DZ2RpTGnZXWJYiVlLQ-K5fDtX3WMaBPaWk,182
-vellum/client/resources/workflow_sandboxes/__init__.py,sha256=FTtvy8EDg9nNNg9WCatVgKTRYV8-_v1roeGPAKoa_pw,65
-vellum/client/resources/workflow_sandboxes/client.py,sha256=3wVQxkjrJ5bIS8fB5FpKXCP2dX38299ghWrJ8YmXxwQ,7435
+vellum/client/resources/workflow_sandboxes/__init__.py,sha256=OR3wE3pTgsZlTS-0ukeMWzSuEZF8PszuQTCHDh6JybI,175
+vellum/client/resources/workflow_sandboxes/client.py,sha256=4FfB7DCAo8DDd3CDlmRiycMjnZhP4oWEbfGuhtzVfwo,12404
+vellum/client/resources/workflow_sandboxes/types/__init__.py,sha256=EaGVRU1w6kJiiHrbZOeEa0c3ggjfgv_jBqsyOkCRWOI,212
+vellum/client/resources/workflow_sandboxes/types/list_workflow_sandbox_examples_request_tag.py,sha256=TEwWit20W3X-zWPPLAhmUG05UudG9gaBSJ4Q4-rNJws,188
 vellum/client/resources/workflows/__init__.py,sha256=Z4xi8Nxd9U4t35FQSepTt1p-ns0X1xtdNs168kUcuBk,153
 vellum/client/resources/workflows/client.py,sha256=hM7FDn05XHhQk599ti8CI4moIg0RVoEFil3Wp9v9UZk,11215
 vellum/client/resources/workflows/types/__init__.py,sha256=-uFca4ypncAOvfsg6sjD-5C9zWdA5qNvU6m675GphVg,177
@@ -183,7 +185,7 @@ vellum/client/resources/workspace_secrets/__init__.py,sha256=FTtvy8EDg9nNNg9WCat
 vellum/client/resources/workspace_secrets/client.py,sha256=h7UzXLyTttPq1t-JZGMg1BWxypxJvBGUdqg7KGT7MK4,8027
 vellum/client/resources/workspaces/__init__.py,sha256=FTtvy8EDg9nNNg9WCatVgKTRYV8-_v1roeGPAKoa_pw,65
 vellum/client/resources/workspaces/client.py,sha256=RthwzN1o-Jxwg5yyNNodavFyNUSxfLoTv26w3mRR5g8,3595
-vellum/client/types/__init__.py,sha256=HsQa33k5NOOW9zQdCCxp6REWnFHvqqR-vJl0RlNmxh0,53975
+vellum/client/types/__init__.py,sha256=w4LPPCUDsYhlZyPcoN7B3cCwPDL87A-UIlYTtiO0zO4,54198
 vellum/client/types/ad_hoc_execute_prompt_event.py,sha256=bCjujA2XsOgyF3bRZbcEqV2rOIymRgsLoIRtZpB14xg,607
 vellum/client/types/ad_hoc_expand_meta.py,sha256=1gv-NCsy_6xBYupLvZH979yf2VMdxAU-l0y0ynMKZaw,1331
 vellum/client/types/ad_hoc_fulfilled_prompt_execution_meta.py,sha256=Bfvf1d_dkmshxRACVM5vcxbH_7AQY23RmrrnPc0ytYY,939
@@ -439,6 +441,7 @@ vellum/client/types/paginated_slim_workflow_deployment_list.py,sha256=b0SGPewaOK
 vellum/client/types/paginated_test_suite_run_execution_list.py,sha256=_NCKlKzs-8h0oZFhbGLO4sMt3xh9jicPqJdYu-NN-8c,1019
 vellum/client/types/paginated_test_suite_test_case_list.py,sha256=9KrCCQKy0egMmVx5U2k6o1GjNrUYpVvGG_hm2cHqIzc,995
 vellum/client/types/paginated_workflow_release_tag_read_list.py,sha256=dH24ESWyAMVtyHsBkxG8kJ9oORY04Wn3IN-7jvV7Lu4,818
+vellum/client/types/paginated_workflow_sandbox_example_list.py,sha256=rCivuKp5fzVV8PdRwyiet7bEmLxX_1hv8N0vn0xaT70,817
 vellum/client/types/pdf_search_result_meta_source.py,sha256=EMVhqdN1bwE6Ujdx4VhlmKQtJvitN-57kY8oZPxh9dI,1126
 vellum/client/types/pdf_search_result_meta_source_request.py,sha256=nUhaD2Kw1paGC6O_ICVNu3R0e1SVgTshRTkGNgmcjXo,1133
 vellum/client/types/plain_text_prompt_block.py,sha256=cqEN-B4mcvMw_9lBN7FQG8pk9b5LBJ9xpM6PTgkGiqs,930
@@ -675,6 +678,7 @@ vellum/client/types/workflow_result_event_output_data_json.py,sha256=8MrgcTSVUby
 vellum/client/types/workflow_result_event_output_data_number.py,sha256=OZYYUF3ayq7gyaesRK3YRaTMVgxFdeFGtOpTPWX10yk,1081
 vellum/client/types/workflow_result_event_output_data_search_results.py,sha256=U34IK7ZvBG70ZBO4SEqbaNzIrV9Zn1NXabNh3M9v_hg,1172
 vellum/client/types/workflow_result_event_output_data_string.py,sha256=tM3kgh6tEhD0dFEb_7UU0-UspeN4pUdINCcCrD64W74,1228
+vellum/client/types/workflow_sandbox_example.py,sha256=ZYFI7ZahyRYMNVmF8LZcF-_lkhqt95WiWXT6SpxnFHk,666
 vellum/client/types/workflow_stream_event.py,sha256=Wn3Yzuy9MqWAeo8tEaXDTKDEbJoA8DdYdMVq8EKuhu8,361
 vellum/client/types/workspace_read.py,sha256=ocPtWvOwadqkU3z21bJgE4JeLYTAkOqBlKkc9lDDFFg,697
 vellum/client/types/workspace_secret_read.py,sha256=3CnHDG72IAY0KRNvc31F0xLmhnpwjQHnDYCfQJzCxI0,714
@@ -755,6 +759,8 @@ vellum/resources/workflow_deployments/types/list_workflow_release_tags_request_s
 vellum/resources/workflow_deployments/types/workflow_deployments_list_request_status.py,sha256=mLfHzwK3uBQfw8nku6YmceHoaWksRPRZzpNBy9MGBv4,209
 vellum/resources/workflow_sandboxes/__init__.py,sha256=jHqdk62iizWsMzrhlXMhAON5eX4CRgL7ipux4MqoK2Y,160
 vellum/resources/workflow_sandboxes/client.py,sha256=6hUU770XG4lOoZZOwvAqmUz5XZDgRd3MPWZs1wieWEM,167
+vellum/resources/workflow_sandboxes/types/__init__.py,sha256=jMwkDLd5Xc4hheGoU-h087t1L3qxp4llmT6MG5MeufA,166
+vellum/resources/workflow_sandboxes/types/list_workflow_sandbox_examples_request_tag.py,sha256=ylKYQDob7ivDO05K9Y21qfAYQmWvsuhu25C6CKjby5c,209
 vellum/resources/workflows/__init__.py,sha256=NhC5vYlg2Jre4lekOubbziseG5_k2c9-tlxGJ2zNVHU,151
 vellum/resources/workflows/client.py,sha256=HVVkYDhHnodVD-MAj1oABf4PGVHr4XNGJrMnShuX-Oc,158
 vellum/resources/workflows/types/__init__.py,sha256=n3dH29XbL5XgAZesnDu2WVWUd_JqPU-s1vOKU_56oDs,157
@@ -1019,6 +1025,7 @@ vellum/types/paginated_slim_workflow_deployment_list.py,sha256=3QgvxRFqcOw9z-cl0
 vellum/types/paginated_test_suite_run_execution_list.py,sha256=XEr928_4w9Rw9_q6dshxPWfXXptLdRlDp-frKIIcdYQ,177
 vellum/types/paginated_test_suite_test_case_list.py,sha256=LoyXDEr2yXrkniJ25HctBWvhqKQ987XItukUwPYUIhQ,173
 vellum/types/paginated_workflow_release_tag_read_list.py,sha256=XUeQn_6JPJ6K2qts-NZIEEZF94C3U2AosStc2k57eWY,178
+vellum/types/paginated_workflow_sandbox_example_list.py,sha256=FlPNK6QtzQL9yD-k_qpQrE8yMARrJRjk5aGf6ZTbGyY,177
 vellum/types/pdf_search_result_meta_source.py,sha256=tkM53z99Zky8ifkcfj1HoS7k-scjy6xeVInVDjMTBzM,167
 vellum/types/pdf_search_result_meta_source_request.py,sha256=TW9FWTdqZi-SuERKkjMdrKBgQEq3RA4W9nwefm8it6k,175
 vellum/types/plain_text_prompt_block.py,sha256=K5tGXMDCVTLDIDOL17TjLvZD6pMaHnRtcSYfXOrzQMM,161
@@ -1255,6 +1262,7 @@ vellum/types/workflow_result_event_output_data_json.py,sha256=z5uMkd759__fTOKWDA
 vellum/types/workflow_result_event_output_data_number.py,sha256=TtQhFjBxV8zc036NkvBrYhWWHEmaUJk2RWAd1NKVtz0,178
 vellum/types/workflow_result_event_output_data_search_results.py,sha256=UNfCHLQ0jd5advLYdV7IBgmsRNzJ0PdDd3r2dNiIlgY,186
 vellum/types/workflow_result_event_output_data_string.py,sha256=rHEVbN0nyf-xoDoSIUEKlUKh6DDoguer4w0iN18JQ2I,178
+vellum/types/workflow_sandbox_example.py,sha256=PixQSt8aczB-oUT8qTL6k9nY-H2Mx7XvispKV0thUHM,162
 vellum/types/workflow_stream_event.py,sha256=PjHGgN0eJm5w-5FJ6__ASC1FU94Gsav_ko5JWkpVvK8,159
 vellum/types/workspace_read.py,sha256=9CvgvK8Li8vL6qC5KX7f3-nEHslJ4lw2w07bvXcrjA0,152
 vellum/types/workspace_secret_read.py,sha256=Z6QNXHxVHRdrLXSI31KxngePRwJTVoJYMXVbtPQwrxs,159
@@ -1406,9 +1414,9 @@ vellum/workflows/nodes/displayable/search_node/node.py,sha256=_VHHuTNN4icZBgc7O5
 vellum/workflows/nodes/displayable/search_node/tests/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 vellum/workflows/nodes/displayable/search_node/tests/test_node.py,sha256=2-QCV7Vk_-YMls33p0GOUtCv3f2uPNZCjkB2CRjek7o,6562
 vellum/workflows/nodes/displayable/subworkflow_deployment_node/__init__.py,sha256=9yYM6001YZeqI1VOk1QuEM_yrffk_EdsO7qaPzINKds,92
-vellum/workflows/nodes/displayable/subworkflow_deployment_node/node.py,sha256=sSRo_zX5QVI7V0qmReWMBkEa78HfQfjNIKwvKh7-Om8,8307
+vellum/workflows/nodes/displayable/subworkflow_deployment_node/node.py,sha256=9yqklkehUWLFhjIAC2WrkoMNqqvzHj1-hsuIVLXJHfo,8657
 vellum/workflows/nodes/displayable/subworkflow_deployment_node/tests/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-vellum/workflows/nodes/displayable/subworkflow_deployment_node/tests/test_node.py,sha256=c8RP-QnsERzIinVytAc0jVZ9nd7Jl3hbc9-_yG91Ros,5445
+vellum/workflows/nodes/displayable/subworkflow_deployment_node/tests/test_node.py,sha256=qHnwn1KI4XpxM1b1zmCXEZUBxe5Sjk88ehxZ8HIGJQc,6353
 vellum/workflows/nodes/displayable/tests/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 vellum/workflows/nodes/displayable/tests/test_inline_text_prompt_node.py,sha256=LaxohBcKfSW2PSiBBlx67FdW_q4YC2BM2ouH-vuGPAA,4700
 vellum/workflows/nodes/displayable/tests/test_search_node_wth_text_output.py,sha256=VepO5z1277c1y5N6LLIC31nnWD1aak2m5oPFplfJHHs,6935
@@ -1471,13 +1479,13 @@ vellum/workflows/utils/uuids.py,sha256=DFzPv9RCvsKhvdTEIQyfSek2A31D6S_QcmeLPbgrg
 vellum/workflows/utils/vellum_variables.py,sha256=fC2aSLvlS31D15dOWu43LBRR0QsgUKNXBiCUvvaLXSs,3231
 vellum/workflows/vellum_client.py,sha256=ODrq_TSl-drX2aezXegf7pizpWDVJuTXH-j6528t75s,683
 vellum/workflows/workflows/__init__.py,sha256=KY45TqvavCCvXIkyCFMEc0dc6jTMOUci93U2DUrlZYc,66
-vellum/workflows/workflows/base.py,sha256=MNwWXt0xosrGAjbkQ_lhmBPlHzsJVX1Cr2qy7guFRK8,22446
+vellum/workflows/workflows/base.py,sha256=k3GfMDbhsa0pNGGHEaqJFbACyIJf_-wZMOz3FFtr_ls,22192
 vellum/workflows/workflows/event_filters.py,sha256=GSxIgwrX26a1Smfd-6yss2abGCnadGsrSZGa7t7LpJA,2008
 vellum/workflows/workflows/tests/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 vellum/workflows/workflows/tests/test_base_workflow.py,sha256=DTfXqHUF0SJELTVSA9PCOWHVoAylqpDNFZTkQM44CdU,6215
 vellum/workflows/workflows/tests/test_context.py,sha256=VJBUcyWVtMa_lE5KxdhgMu0WYNYnUQUDvTF7qm89hJ0,2333
-vellum_ai-0.14.5.dist-info/LICENSE,sha256=hOypcdt481qGNISA784bnAGWAE6tyIf9gc2E78mYC3E,1574
-vellum_ai-0.14.5.dist-info/METADATA,sha256=_53R_GRMep--gP15bdsMgW6SYTC9AD_W9yvPkQp94w0,5407
-vellum_ai-0.14.5.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-vellum_ai-0.14.5.dist-info/entry_points.txt,sha256=HCH4yc_V3J_nDv3qJzZ_nYS8llCHZViCDP1ejgCc5Ak,42
-vellum_ai-0.14.5.dist-info/RECORD,,
+vellum_ai-0.14.6.dist-info/LICENSE,sha256=hOypcdt481qGNISA784bnAGWAE6tyIf9gc2E78mYC3E,1574
+vellum_ai-0.14.6.dist-info/METADATA,sha256=gJeL8u5StNSUb5b8gENXqxj5mH0etlqQWLtWKBvE_eY,5407
+vellum_ai-0.14.6.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+vellum_ai-0.14.6.dist-info/entry_points.txt,sha256=HCH4yc_V3J_nDv3qJzZ_nYS8llCHZViCDP1ejgCc5Ak,42
+vellum_ai-0.14.6.dist-info/RECORD,,

vellum_ee/workflows/display/nodes/vellum/tests/test_utils.py

@@ -7,6 +7,7 @@ from vellum.workflows.descriptors.base import BaseDescriptor
 from vellum.workflows.inputs import BaseInputs
 from vellum.workflows.nodes.bases import BaseNode
 from vellum.workflows.outputs import BaseOutputs
+from vellum.workflows.references import LazyReference
 from vellum_ee.workflows.display.nodes.base_node_vellum_display import BaseNodeVellumDisplay
 from vellum_ee.workflows.display.nodes.types import NodeOutputDisplay
 from vellum_ee.workflows.display.nodes.vellum.utils import create_node_input_value_pointer_rules
@@ -43,6 +44,10 @@ class MyNodeADisplay(BaseNodeVellumDisplay[MyNodeA]):
 class MyNodeB(BaseNode):
     example = MyNodeA.Outputs.output
     fallback_example = MyNodeA.Outputs.output.coalesce(Inputs.example_workflow_input).coalesce("fallback")
+    constant_coalesce = Inputs.example_workflow_input.coalesce("default_value")
+    lazy_coalesce: BaseDescriptor[str] = LazyReference(
+        lambda: MyNodeA.Outputs.output.coalesce(Inputs.example_workflow_input)
+    )
 
 
 @pytest.mark.parametrize(
@@ -76,6 +81,32 @@ class MyNodeB(BaseNode):
                 ConstantValuePointer(type="CONSTANT_VALUE", data=StringVellumValue(value="fallback")),
             ],
         ),
+        (
+            MyNodeB.constant_coalesce,
+            [
+                InputVariablePointer(
+                    type="INPUT_VARIABLE",
+                    data=InputVariableData(input_variable_id="a154c29d-fac0-4cd0-ba88-bc52034f5470"),
+                ),
+                ConstantValuePointer(type="CONSTANT_VALUE", data=StringVellumValue(value="default_value")),
+            ],
+        ),
+        (
+            MyNodeB.lazy_coalesce,
+            [
+                NodeOutputPointer(
+                    type="NODE_OUTPUT",
+                    data=NodeOutputData(
+                        node_id="b48fa5e0-d7d3-4fe3-ae48-615415011cc5",
+                        output_id="4b16a629-11a1-4b3f-a965-a57b872d13b8",
+                    ),
+                ),
+                InputVariablePointer(
+                    type="INPUT_VARIABLE",
+                    data=InputVariableData(input_variable_id="a154c29d-fac0-4cd0-ba88-bc52034f5470"),
+                ),
+            ],
+        ),
     ],
 )
 def test_create_node_input_value_pointer_rules(

vellum_ee/workflows/display/nodes/vellum/utils.py

@@ -5,8 +5,10 @@ from vellum.workflows.descriptors.base import BaseDescriptor
 from vellum.workflows.expressions.coalesce_expression import CoalesceExpression
 from vellum.workflows.nodes.displayable.bases.utils import primitive_to_vellum_value
 from vellum.workflows.references import NodeReference
+from vellum.workflows.references.lazy import LazyReference
 from vellum.workflows.utils.uuids import uuid4_from_hash
 from vellum_ee.workflows.display.types import WorkflowDisplayContext
+from vellum_ee.workflows.display.utils.expressions import get_child_descriptor
 from vellum_ee.workflows.display.utils.vellum import create_node_input_value_pointer_rule
 from vellum_ee.workflows.display.vellum import (
     ConstantValuePointer,
@@ -56,6 +58,12 @@ def create_node_input_value_pointer_rules(
                 raise ValueError(f"Expected NodeReference {value.name} to have an instance")
             value = cast(BaseDescriptor, value.instance)
 
+        if isinstance(value, LazyReference):
+            child_descriptor = get_child_descriptor(value, display_context)
+            return create_node_input_value_pointer_rules(
+                child_descriptor, display_context, [], pointer_type=pointer_type
+            )
+
         if isinstance(value, CoalesceExpression):
             # Recursively handle the left-hand side
             lhs_rules = create_node_input_value_pointer_rules(value.lhs, display_context, [], pointer_type=pointer_type)