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

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

vellum/client/resources/document_indexes/client.py

@@ -54,8 +54,8 @@ class DocumentIndexesClient:
         status : typing.Optional[DocumentIndexesListRequestStatus]
             Filter down to only document indices that have a status matching the status specified
 
-            - `ACTIVE` - Active
-            - `ARCHIVED` - Archived
+            * `ACTIVE` - Active
+            * `ARCHIVED` - Archived
 
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
@@ -589,8 +589,8 @@ class AsyncDocumentIndexesClient:
         status : typing.Optional[DocumentIndexesListRequestStatus]
             Filter down to only document indices that have a status matching the status specified
 
-            - `ACTIVE` - Active
-            - `ARCHIVED` - Archived
+            * `ACTIVE` - Active
+            * `ARCHIVED` - Archived
 
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.

vellum/client/resources/documents/client.py

@@ -278,7 +278,6 @@ class DocumentsClient:
         **Note:** Uses a base url of `https://documents.vellum.ai`.
 
         This is a multipart/form-data request. The `contents` field should be a file upload. It also expects a JSON body with the following fields:
-
         - `add_to_index_names: list[str]` - Optionally include the names of all indexes that you'd like this document to be included in
         - `external_id: str | None` - Optionally include an external ID for this document. This is useful if you want to re-upload the same document later when its contents change and would like it to be re-indexed.
         - `label: str` - A human-friendly name for this document. Typically the filename.
@@ -675,7 +674,6 @@ class AsyncDocumentsClient:
         **Note:** Uses a base url of `https://documents.vellum.ai`.
 
         This is a multipart/form-data request. The `contents` field should be a file upload. It also expects a JSON body with the following fields:
-
         - `add_to_index_names: list[str]` - Optionally include the names of all indexes that you'd like this document to be included in
         - `external_id: str | None` - Optionally include an external ID for this document. This is useful if you want to re-upload the same document later when its contents change and would like it to be re-indexed.
         - `label: str` - A human-friendly name for this document. Typically the filename.

vellum/client/resources/folder_entities/client.py

@@ -39,7 +39,6 @@ class FolderEntitiesClient:
 
             To filter by an entity's parent folder, provide the ID of the parent folder. To filter by the root directory, provide
             a string representing the entity type of the root directory. Supported root directories include:
-
             - PROMPT_SANDBOX
             - WORKFLOW_SANDBOX
             - DOCUMENT_INDEX
@@ -48,8 +47,8 @@ class FolderEntitiesClient:
         entity_status : typing.Optional[FolderEntitiesListRequestEntityStatus]
             Filter down to only those objects whose entities have a status matching the status specified.
 
-            - `ACTIVE` - Active
-            - `ARCHIVED` - Archived
+            * `ACTIVE` - Active
+            * `ARCHIVED` - Archived
 
         limit : typing.Optional[int]
             Number of results to return per page.
@@ -119,7 +118,6 @@ class FolderEntitiesClient:
         folder_id : str
             The ID of the folder to which the entity should be added. This can be a UUID of a folder, or the name of a root
             directory. Supported root directories include:
-
             - PROMPT_SANDBOX
             - WORKFLOW_SANDBOX
             - DOCUMENT_INDEX
@@ -190,7 +188,6 @@ class AsyncFolderEntitiesClient:
 
             To filter by an entity's parent folder, provide the ID of the parent folder. To filter by the root directory, provide
             a string representing the entity type of the root directory. Supported root directories include:
-
             - PROMPT_SANDBOX
             - WORKFLOW_SANDBOX
             - DOCUMENT_INDEX
@@ -199,8 +196,8 @@ class AsyncFolderEntitiesClient:
         entity_status : typing.Optional[FolderEntitiesListRequestEntityStatus]
             Filter down to only those objects whose entities have a status matching the status specified.
 
-            - `ACTIVE` - Active
-            - `ARCHIVED` - Archived
+            * `ACTIVE` - Active
+            * `ARCHIVED` - Archived
 
         limit : typing.Optional[int]
             Number of results to return per page.
@@ -278,7 +275,6 @@ class AsyncFolderEntitiesClient:
         folder_id : str
             The ID of the folder to which the entity should be added. This can be a UUID of a folder, or the name of a root
             directory. Supported root directories include:
-
             - PROMPT_SANDBOX
             - WORKFLOW_SANDBOX
             - DOCUMENT_INDEX

vellum/client/resources/test_suite_runs/client.py

@@ -163,7 +163,6 @@ class TestSuiteRunsClient:
 
         expand : typing.Optional[typing.Union[str, typing.Sequence[str]]]
             The response fields to expand for more information.
-
             - 'results.metric_results.metric_label' expands the metric label for each metric result.
             - 'results.metric_results.metric_definition' expands the metric definition for each metric result.
             - 'results.metric_results.metric_definition.name' expands the metric definition name for each metric result.
@@ -381,7 +380,6 @@ class AsyncTestSuiteRunsClient:
 
         expand : typing.Optional[typing.Union[str, typing.Sequence[str]]]
             The response fields to expand for more information.
-
             - 'results.metric_results.metric_label' expands the metric label for each metric result.
             - 'results.metric_results.metric_definition' expands the metric definition for each metric result.
             - 'results.metric_results.metric_definition.name' expands the metric definition name for each metric result.

vellum/client/types/deployment_read.py

@@ -30,17 +30,17 @@ class DeploymentRead(UniversalBaseModel):
     """
     The current status of the deployment
     
-    - `ACTIVE` - Active
-    - `ARCHIVED` - Archived
+    * `ACTIVE` - Active
+    * `ARCHIVED` - Archived
     """
 
     environment: typing.Optional[EnvironmentEnum] = pydantic.Field(default=None)
     """
     The environment this deployment is used in
     
-    - `DEVELOPMENT` - Development
-    - `STAGING` - Staging
-    - `PRODUCTION` - Production
+    * `DEVELOPMENT` - Development
+    * `STAGING` - Staging
+    * `PRODUCTION` - Production
     """
 
     last_deployed_on: dt.datetime

vellum/client/types/deployment_release_tag_read.py

@@ -18,8 +18,8 @@ class DeploymentReleaseTagRead(UniversalBaseModel):
     """
     The source of how the Release Tag was originally created
     
-    - `SYSTEM` - System
-    - `USER` - User
+    * `SYSTEM` - System
+    * `USER` - User
     """
 
     history_item: DeploymentReleaseTagDeploymentHistoryItem = pydantic.Field()

vellum/client/types/document_document_to_document_index.py

@@ -26,11 +26,11 @@ class DocumentDocumentToDocumentIndex(UniversalBaseModel):
     """
     An enum value representing where this document is along its indexing lifecycle for this index.
     
-    - `AWAITING_PROCESSING` - Awaiting Processing
-    - `QUEUED` - Queued
-    - `INDEXING` - Indexing
-    - `INDEXED` - Indexed
-    - `FAILED` - Failed
+    * `AWAITING_PROCESSING` - Awaiting Processing
+    * `QUEUED` - Queued
+    * `INDEXING` - Indexing
+    * `INDEXED` - Indexed
+    * `FAILED` - Failed
     """
 
     extracted_text_file_url: typing.Optional[str] = None

vellum/client/types/document_index_read.py

@@ -27,17 +27,17 @@ class DocumentIndexRead(UniversalBaseModel):
     """
     The current status of the document index
     
-    - `ACTIVE` - Active
-    - `ARCHIVED` - Archived
+    * `ACTIVE` - Active
+    * `ARCHIVED` - Archived
     """
 
     environment: typing.Optional[EnvironmentEnum] = pydantic.Field(default=None)
     """
     The environment this document index is used in
     
-    - `DEVELOPMENT` - Development
-    - `STAGING` - Staging
-    - `PRODUCTION` - Production
+    * `DEVELOPMENT` - Development
+    * `STAGING` - Staging
+    * `PRODUCTION` - Production
     """
 
     indexing_config: DocumentIndexIndexingConfig

vellum/client/types/document_read.py

@@ -28,7 +28,7 @@ class DocumentRead(UniversalBaseModel):
     """
     The current status of the document
     
-    - `ACTIVE` - Active
+    * `ACTIVE` - Active
     """
 
     original_file_url: typing.Optional[str] = None

vellum/client/types/enriched_normalized_completion.py

@@ -29,9 +29,9 @@ class EnrichedNormalizedCompletion(UniversalBaseModel):
     """
     The reason the generation finished.
     
-    - `LENGTH` - LENGTH
-    - `STOP` - STOP
-    - `UNKNOWN` - UNKNOWN
+    * `LENGTH` - LENGTH
+    * `STOP` - STOP
+    * `UNKNOWN` - UNKNOWN
     """
 
     logprobs: typing.Optional[NormalizedLogProbs] = pydantic.Field(default=None)

vellum/client/types/generate_options_request.py

@@ -12,8 +12,8 @@ class GenerateOptionsRequest(UniversalBaseModel):
     """
     Which logprobs to include, if any. Defaults to NONE.
     
-    - `ALL` - ALL
-    - `NONE` - NONE
+    * `ALL` - ALL
+    * `NONE` - NONE
     """
 
     if IS_PYDANTIC_V2:

vellum/client/types/slim_deployment_read.py

@@ -30,17 +30,17 @@ class SlimDeploymentRead(UniversalBaseModel):
     """
     The current status of the deployment
     
-    - `ACTIVE` - Active
-    - `ARCHIVED` - Archived
+    * `ACTIVE` - Active
+    * `ARCHIVED` - Archived
     """
 
     environment: typing.Optional[EnvironmentEnum] = pydantic.Field(default=None)
     """
     The environment this deployment is used in
     
-    - `DEVELOPMENT` - Development
-    - `STAGING` - Staging
-    - `PRODUCTION` - Production
+    * `DEVELOPMENT` - Development
+    * `STAGING` - Staging
+    * `PRODUCTION` - Production
     """
 
     last_deployed_on: dt.datetime

vellum/client/types/slim_document.py

@@ -37,15 +37,15 @@ class SlimDocument(UniversalBaseModel):
     """
     An enum value representing why the document could not be processed. Is null unless processing_state is FAILED.
     
-    - `EXCEEDED_CHARACTER_LIMIT` - Exceeded Character Limit
-    - `INVALID_FILE` - Invalid File
+    * `EXCEEDED_CHARACTER_LIMIT` - Exceeded Character Limit
+    * `INVALID_FILE` - Invalid File
     """
 
     status: typing.Optional[DocumentStatus] = pydantic.Field(default=None)
     """
     The document's current status.
     
-    - `ACTIVE` - Active
+    * `ACTIVE` - Active
     """
 
     keywords: typing.Optional[typing.List[str]] = pydantic.Field(default=None)

vellum/client/types/slim_document_document_to_document_index.py

@@ -26,11 +26,11 @@ class SlimDocumentDocumentToDocumentIndex(UniversalBaseModel):
     """
     An enum value representing where this document is along its indexing lifecycle for this index.
     
-    - `AWAITING_PROCESSING` - Awaiting Processing
-    - `QUEUED` - Queued
-    - `INDEXING` - Indexing
-    - `INDEXED` - Indexed
-    - `FAILED` - Failed
+    * `AWAITING_PROCESSING` - Awaiting Processing
+    * `QUEUED` - Queued
+    * `INDEXING` - Indexing
+    * `INDEXED` - Indexed
+    * `FAILED` - Failed
     """
 
     if IS_PYDANTIC_V2:

vellum/client/types/slim_workflow_deployment.py

@@ -29,17 +29,17 @@ class SlimWorkflowDeployment(UniversalBaseModel):
     """
     The current status of the workflow deployment
     
-    - `ACTIVE` - Active
-    - `ARCHIVED` - Archived
+    * `ACTIVE` - Active
+    * `ARCHIVED` - Archived
     """
 
     environment: typing.Optional[EnvironmentEnum] = pydantic.Field(default=None)
     """
     The environment this workflow deployment is used in
     
-    - `DEVELOPMENT` - Development
-    - `STAGING` - Staging
-    - `PRODUCTION` - Production
+    * `DEVELOPMENT` - Development
+    * `STAGING` - Staging
+    * `PRODUCTION` - Production
     """
 
     created: dt.datetime

vellum/client/types/test_suite_run_read.py

@@ -21,11 +21,11 @@ class TestSuiteRunRead(UniversalBaseModel):
     """
     The current state of this run
     
-    - `QUEUED` - Queued
-    - `RUNNING` - Running
-    - `COMPLETE` - Complete
-    - `FAILED` - Failed
-    - `CANCELLED` - Cancelled
+    * `QUEUED` - Queued
+    * `RUNNING` - Running
+    * `COMPLETE` - Complete
+    * `FAILED` - Failed
+    * `CANCELLED` - Cancelled
     """
 
     exec_config: typing.Optional[TestSuiteRunExecConfig] = pydantic.Field(default=None)

vellum/client/types/workflow_deployment_read.py

@@ -29,17 +29,17 @@ class WorkflowDeploymentRead(UniversalBaseModel):
     """
     The current status of the workflow deployment
     
-    - `ACTIVE` - Active
-    - `ARCHIVED` - Archived
+    * `ACTIVE` - Active
+    * `ARCHIVED` - Archived
     """
 
     environment: typing.Optional[EnvironmentEnum] = pydantic.Field(default=None)
     """
     The environment this workflow deployment is used in
     
-    - `DEVELOPMENT` - Development
-    - `STAGING` - Staging
-    - `PRODUCTION` - Production
+    * `DEVELOPMENT` - Development
+    * `STAGING` - Staging
+    * `PRODUCTION` - Production
     """
 
     created: dt.datetime

vellum/client/types/workflow_release_tag_read.py

@@ -18,8 +18,8 @@ class WorkflowReleaseTagRead(UniversalBaseModel):
     """
     The source of how the Release Tag was originally created
     
-    - `SYSTEM` - System
-    - `USER` - User
+    * `SYSTEM` - System
+    * `USER` - User
     """
 
     history_item: WorkflowReleaseTagWorkflowDeploymentHistoryItem = pydantic.Field()

vellum/prompts/blocks/compilation.py

@@ -19,7 +19,7 @@ from vellum.client.types.vellum_audio import VellumAudio
 from vellum.client.types.vellum_image import VellumImage
 from vellum.prompts.blocks.exceptions import PromptCompilationError
 from vellum.prompts.blocks.types import CompiledChatMessagePromptBlock, CompiledPromptBlock, CompiledValuePromptBlock
-from vellum.utils.templating.constants import DEFAULT_JINJA_CUSTOM_FILTERS
+from vellum.utils.templating.constants import DEFAULT_JINJA_CUSTOM_FILTERS, DEFAULT_JINJA_GLOBALS
 from vellum.utils.templating.render import render_sandboxed_jinja_template
 from vellum.utils.typing import cast_not_optional
 
@@ -74,7 +74,7 @@ def compile_prompt_blocks(
                 template=block.template,
                 input_values={name: inp.value for name, inp in inputs_by_name.items()},
                 jinja_custom_filters=DEFAULT_JINJA_CUSTOM_FILTERS,
-                jinja_globals=DEFAULT_JINJA_CUSTOM_FILTERS,
+                jinja_globals=DEFAULT_JINJA_GLOBALS,
             )
             jinja_content = StringVellumValue(value=rendered_template)
 

vellum/prompts/blocks/tests/test_compilation.py

@@ -37,6 +37,14 @@ from vellum.prompts.blocks.types import CompiledChatMessagePromptBlock, Compiled
                 CompiledValuePromptBlock(content=StringVellumValue(value="Repeat back to me Hello, world!")),
             ],
         ),
+        (
+            [JinjaPromptBlock(template="{{ re.search('test', message).group() }}")],
+            [PromptRequestStringInput(key="message", value="testing")],
+            [VellumVariable(id="1", type="STRING", key="message")],
+            [
+                CompiledValuePromptBlock(content=StringVellumValue(value="test")),
+            ],
+        ),
         # Rich Text
         (
             [
@@ -127,6 +135,7 @@ from vellum.prompts.blocks.types import CompiledChatMessagePromptBlock, Compiled
         "empty",
         "jinja-no-variables",
         "jinja-with-variables",
+        "jinja-with-custom-global",
         "rich-text-no-variables",
         "rich-text-with-variables",
         "chat-message",

vellum/utils/templating/render.py

@@ -9,7 +9,7 @@ from vellum.workflows.state.encoder import DefaultStateEncoder
 
 
 def finalize(obj: Any) -> str:
-    if isinstance(obj, dict):
+    if isinstance(obj, (dict, list)):
         return json.dumps(obj, cls=DefaultStateEncoder)
 
     return str(obj)

vellum/workflows/constants.py

@@ -3,6 +3,15 @@ from typing import Any, cast
 
 
 class _UndefMeta(type):
+    def __new__(cls, name: str, bases: tuple[type, ...], attrs: dict[str, Any]) -> type:
+        cls.__name__ = "undefined"
+        cls.__qualname__ = "undefined"
+
+        undefined_class = super().__new__(cls, name, bases, attrs)
+        undefined_class.__name__ = "undefined"
+        undefined_class.__qualname__ = "undefined"
+        return undefined_class
+
     def __repr__(cls) -> str:
         return "undefined"
 

vellum/workflows/nodes/core/map_node/node.py

@@ -42,8 +42,8 @@ class MapNode(BaseAdornmentNode[StateType], Generic[StateType, MapNodeItemType])
     Used to map over a list of items and execute a Subworkflow on each iteration.
 
     items: List[MapNodeItemType] - The items to map over
-    subworkflow: Type["BaseWorkflow[SubworkflowInputs, BaseState]"] - The Subworkflow to execute on each iteration
     max_concurrency: Optional[int] = None - The maximum number of concurrent subworkflow executions
+    subworkflow: Type["BaseWorkflow"] - The Subworkflow to execute
     """
 
     items: List[MapNodeItemType]

vellum/workflows/nodes/core/retry_node/node.py

@@ -18,9 +18,10 @@ class RetryNode(BaseAdornmentNode[StateType], Generic[StateType]):
     Used to retry a Subworkflow a specified number of times.
 
     max_attempts: int - The maximum number of attempts to retry the Subworkflow
-    delay: float - The number of seconds to wait between retries
-    retry_on_error_code: Optional[VellumErrorCode] = None - The error code to retry on
-    subworkflow: Type["BaseWorkflow[SubworkflowInputs, BaseState]"] - The Subworkflow to execute
+    delay: float = None - The number of seconds to wait between retries
+    retry_on_error_code: Optional[WorkflowErrorCode] = None - The error code to retry on
+    retry_on_condition: Optional[BaseDescriptor] = None - The condition to retry on
+    subworkflow: Type["BaseWorkflow"] - The Subworkflow to execute
     """
 
     max_attempts: int

vellum/workflows/nodes/core/templating_node/tests/test_templating_node.py

@@ -220,3 +220,17 @@ def test_templating_node__replace_filter():
             },
         }
     ]
+
+
+def test_templating_node__last_chat_message():
+    # GIVEN a templating node that outputs a complex object
+    class LastChatMessageTemplateNode(TemplatingNode[BaseState, List[ChatMessage]]):
+        template = """{{ chat_history[:-1] }}"""
+        inputs = {"chat_history": [ChatMessage(role="USER", text="Hello"), ChatMessage(role="ASSISTANT", text="World")]}
+
+    # WHEN the node is run
+    node = LastChatMessageTemplateNode()
+    outputs = node.run()
+
+    # THEN the output is the expected JSON
+    assert outputs.result == [ChatMessage(role="USER", text="Hello")]

vellum/workflows/nodes/core/try_node/node.py

@@ -17,7 +17,7 @@ class TryNode(BaseAdornmentNode[StateType], Generic[StateType]):
     """
     Used to execute a Subworkflow and handle errors.
 
-    on_error_code: Optional[VellumErrorCode] = None - The error code to handle
+    on_error_code: Optional[WorkflowErrorCode] = None - The error code to handle
     subworkflow: Type["BaseWorkflow"] - The Subworkflow to execute
     """
 

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

@@ -17,6 +17,7 @@ from vellum import (
 )
 from vellum.client import RequestOptions
 from vellum.client.types.chat_message_request import ChatMessageRequest
+from vellum.client.types.prompt_settings import PromptSettings
 from vellum.workflows.constants import OMIT
 from vellum.workflows.context import get_parent_context
 from vellum.workflows.errors import WorkflowErrorCode
@@ -53,6 +54,8 @@ class BaseInlinePromptNode(BasePromptNode[StateType], Generic[StateType]):
     parameters: PromptParameters = DEFAULT_PROMPT_PARAMETERS
     expand_meta: Optional[AdHocExpandMeta] = OMIT
 
+    settings: Optional[PromptSettings] = None
+
     class Trigger(BasePromptNode.Trigger):
         merge_behavior = MergeBehavior.AWAIT_ANY
 
@@ -60,6 +63,7 @@ class BaseInlinePromptNode(BasePromptNode[StateType], Generic[StateType]):
         input_variables, input_values = self._compile_prompt_inputs()
         parent_context = get_parent_context()
         request_options = self.request_options or RequestOptions()
+
         request_options["additional_body_parameters"] = {
             "execution_context": {"parent_context": parent_context},
             **request_options.get("additional_body_parameters", {}),
@@ -79,6 +83,7 @@ class BaseInlinePromptNode(BasePromptNode[StateType], Generic[StateType]):
             input_variables=input_variables,
             parameters=self.parameters,
             blocks=self.blocks,
+            settings=self.settings,
             functions=normalized_functions,
             expand_meta=self.expand_meta,
             request_options=request_options,

vellum/workflows/nodes/displayable/tests/test_inline_text_prompt_node.py

@@ -91,6 +91,7 @@ def test_inline_text_prompt_node__basic(vellum_adhoc_prompt_client):
             custom_parameters=None,
         ),
         request_options=mock.ANY,
+        settings=None,
     )
 
 

vellum/workflows/tests/test_undefined.py

@@ -0,0 +1,12 @@
+import pytest
+
+from vellum.workflows.constants import undefined
+
+
+def test_undefined__ensure_sensible_error_messages():
+    # WHEN we invoke an invalid operation on `undefined`
+    with pytest.raises(Exception) as e:
+        len(undefined)
+
+    # THEN we get a sensible error message
+    assert str(e.value) == "object of type 'undefined' has no len()"

vellum/workflows/workflows/base.py

@@ -100,6 +100,30 @@ class _BaseWorkflowMeta(type):
                     new_dct,
                 )
 
+        def collect_nodes(graph_item: Union[GraphAttribute, Set[GraphAttribute]]) -> Set[Type[BaseNode]]:
+            nodes: Set[Type[BaseNode]] = set()
+            if isinstance(graph_item, Graph):
+                nodes.update(node for node in graph_item.nodes)
+            elif isinstance(graph_item, set):
+                for item in graph_item:
+                    if isinstance(item, Graph):
+                        nodes.update(node for node in item.nodes)
+                    elif inspect.isclass(item) and issubclass(item, BaseNode):
+                        nodes.add(item)
+            elif issubclass(graph_item, BaseNode):
+                nodes.add(graph_item)
+            else:
+                raise ValueError(f"Unexpected graph type: {graph_item.__class__}")
+            return nodes
+
+        graph_nodes = collect_nodes(dct.get("graph", set()))
+        unused_nodes = collect_nodes(dct.get("unused_graphs", set()))
+
+        overlap = graph_nodes & unused_nodes
+        if overlap:
+            node_names = [node.__name__ for node in overlap]
+            raise ValueError(f"Node(s) {', '.join(node_names)} cannot appear in both graph and unused_graphs")
+
         cls = super().__new__(mcs, name, bases, dct)
         workflow_class = cast(Type["BaseWorkflow"], cls)
         workflow_class.__id__ = uuid4_from_hash(workflow_class.__qualname__)
@@ -112,6 +136,7 @@ GraphAttribute = Union[Type[BaseNode], Graph, Set[Type[BaseNode]], Set[Graph]]
 class BaseWorkflow(Generic[InputsType, StateType], metaclass=_BaseWorkflowMeta):
     __id__: UUID = uuid4_from_hash(__qualname__)
     graph: ClassVar[GraphAttribute]
+    unused_graphs: ClassVar[Set[GraphAttribute]]  # nodes or graphs that are defined but not used in the graph
     emitters: List[BaseWorkflowEmitter]
     resolvers: List[BaseWorkflowResolver]
 
@@ -196,6 +221,40 @@ class BaseWorkflow(Generic[InputsType, StateType], metaclass=_BaseWorkflowMeta):
                     nodes.add(node)
                     yield node
 
+    @classmethod
+    def get_unused_nodes(cls) -> Iterator[Type[BaseNode]]:
+        """
+        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()
+            for item in cls.unused_graphs:
+                if isinstance(item, Graph):
+                    # Item is a graph
+                    for node in item.nodes:
+                        if node not in nodes:
+                            nodes.add(node)
+                            yield node
+                elif isinstance(item, set):
+                    # Item is a set of graphs or nodes
+                    for subitem in item:
+                        if isinstance(subitem, Graph):
+                            for node in subitem.nodes:
+                                if node not in nodes:
+                                    nodes.add(node)
+                                    yield node
+                        elif issubclass(subitem, BaseNode):
+                            if subitem not in nodes:
+                                nodes.add(subitem)
+                                yield subitem
+                elif issubclass(item, BaseNode):
+                    # Item is a node
+                    if item not in nodes:
+                        nodes.add(item)
+                        yield item
+
     @classmethod
     def get_entrypoints(cls) -> Iterable[Type[BaseNode]]:
         return iter({e for g in cls.get_subgraphs() for e in g.entrypoints})