vellum-ai 0.12.9__tar.gz → 0.12.13__tar.gz

{vellum_ai-0.12.9 → vellum_ai-0.12.13}/PKG-INFO

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

{vellum_ai-0.12.9 → vellum_ai-0.12.13}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "vellum-ai"
-version = "0.12.9"
+version = "0.12.13"
 description = ""
 readme = "README.md"
 authors = []

{vellum_ai-0.12.9 → vellum_ai-0.12.13}/src/vellum/__init__.py

@@ -120,7 +120,6 @@ from .types import (
     FolderEntityWorkflowSandbox,
     FolderEntityWorkflowSandboxData,
     FulfilledAdHocExecutePromptEvent,
-    FulfilledEnum,
     FulfilledExecutePromptEvent,
     FulfilledExecutePromptResponse,
     FulfilledExecuteWorkflowWorkflowResultEvent,
@@ -389,6 +388,7 @@ from .types import (
     TestSuiteRunExternalExecConfigData,
     TestSuiteRunExternalExecConfigDataRequest,
     TestSuiteRunExternalExecConfigRequest,
+    TestSuiteRunMetricArrayOutput,
     TestSuiteRunMetricErrorOutput,
     TestSuiteRunMetricJsonOutput,
     TestSuiteRunMetricNumberOutput,
@@ -649,7 +649,6 @@ __all__ = [
     "FolderEntityWorkflowSandboxData",
     "ForbiddenError",
     "FulfilledAdHocExecutePromptEvent",
-    "FulfilledEnum",
     "FulfilledExecutePromptEvent",
     "FulfilledExecutePromptResponse",
     "FulfilledExecuteWorkflowWorkflowResultEvent",
@@ -922,6 +921,7 @@ __all__ = [
     "TestSuiteRunExternalExecConfigData",
     "TestSuiteRunExternalExecConfigDataRequest",
     "TestSuiteRunExternalExecConfigRequest",
+    "TestSuiteRunMetricArrayOutput",
     "TestSuiteRunMetricErrorOutput",
     "TestSuiteRunMetricJsonOutput",
     "TestSuiteRunMetricNumberOutput",

{vellum_ai-0.12.9 → vellum_ai-0.12.13}/src/vellum/client/__init__.py

@@ -1079,6 +1079,7 @@ class Vellum:
         index_id: typing.Optional[str] = OMIT,
         index_name: typing.Optional[str] = OMIT,
         options: typing.Optional[SearchRequestOptionsRequest] = OMIT,
+        document_index: typing.Optional[str] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> SearchResponse:
         """
@@ -1090,14 +1091,17 @@ class Vellum:
             The query to search for.
 
         index_id : typing.Optional[str]
-            The ID of the index to search against. Must provide either this or index_name.
+            The ID of the index to search against. Must provide either this, index_name or document_index.
 
         index_name : typing.Optional[str]
-            The name of the index to search against. Must provide either this or index_id.
+            The name of the index to search against. Must provide either this, index_id or document_index.
 
         options : typing.Optional[SearchRequestOptionsRequest]
             Configuration options for the search.
 
+        document_index : typing.Optional[str]
+            Either the index name or index ID to search against. Must provide either this, index_id or index_name.
+
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
@@ -1128,6 +1132,7 @@ class Vellum:
                 "options": convert_and_respect_annotation_metadata(
                     object_=options, annotation=SearchRequestOptionsRequest, direction="write"
                 ),
+                "document_index": document_index,
             },
             request_options=request_options,
             omit=OMIT,
@@ -2395,6 +2400,7 @@ class AsyncVellum:
         index_id: typing.Optional[str] = OMIT,
         index_name: typing.Optional[str] = OMIT,
         options: typing.Optional[SearchRequestOptionsRequest] = OMIT,
+        document_index: typing.Optional[str] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> SearchResponse:
         """
@@ -2406,14 +2412,17 @@ class AsyncVellum:
             The query to search for.
 
         index_id : typing.Optional[str]
-            The ID of the index to search against. Must provide either this or index_name.
+            The ID of the index to search against. Must provide either this, index_name or document_index.
 
         index_name : typing.Optional[str]
-            The name of the index to search against. Must provide either this or index_id.
+            The name of the index to search against. Must provide either this, index_id or document_index.
 
         options : typing.Optional[SearchRequestOptionsRequest]
             Configuration options for the search.
 
+        document_index : typing.Optional[str]
+            Either the index name or index ID to search against. Must provide either this, index_id or index_name.
+
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
@@ -2452,6 +2461,7 @@ class AsyncVellum:
                 "options": convert_and_respect_annotation_metadata(
                     object_=options, annotation=SearchRequestOptionsRequest, direction="write"
                 ),
+                "document_index": document_index,
             },
             request_options=request_options,
             omit=OMIT,

{vellum_ai-0.12.9 → vellum_ai-0.12.13}/src/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.12.9",
+            "X-Fern-SDK-Version": "0.12.13",
         }
         headers["X_API_KEY"] = self.api_key
         return headers

{vellum_ai-0.12.9 → vellum_ai-0.12.13}/src/vellum/client/types/__init__.py

@@ -127,7 +127,6 @@ from .folder_entity_test_suite_data import FolderEntityTestSuiteData
 from .folder_entity_workflow_sandbox import FolderEntityWorkflowSandbox
 from .folder_entity_workflow_sandbox_data import FolderEntityWorkflowSandboxData
 from .fulfilled_ad_hoc_execute_prompt_event import FulfilledAdHocExecutePromptEvent
-from .fulfilled_enum import FulfilledEnum
 from .fulfilled_execute_prompt_event import FulfilledExecutePromptEvent
 from .fulfilled_execute_prompt_response import FulfilledExecutePromptResponse
 from .fulfilled_execute_workflow_workflow_result_event import FulfilledExecuteWorkflowWorkflowResultEvent
@@ -402,6 +401,7 @@ from .test_suite_run_external_exec_config import TestSuiteRunExternalExecConfig
 from .test_suite_run_external_exec_config_data import TestSuiteRunExternalExecConfigData
 from .test_suite_run_external_exec_config_data_request import TestSuiteRunExternalExecConfigDataRequest
 from .test_suite_run_external_exec_config_request import TestSuiteRunExternalExecConfigRequest
+from .test_suite_run_metric_array_output import TestSuiteRunMetricArrayOutput
 from .test_suite_run_metric_error_output import TestSuiteRunMetricErrorOutput
 from .test_suite_run_metric_json_output import TestSuiteRunMetricJsonOutput
 from .test_suite_run_metric_number_output import TestSuiteRunMetricNumberOutput
@@ -641,7 +641,6 @@ __all__ = [
     "FolderEntityWorkflowSandbox",
     "FolderEntityWorkflowSandboxData",
     "FulfilledAdHocExecutePromptEvent",
-    "FulfilledEnum",
     "FulfilledExecutePromptEvent",
     "FulfilledExecutePromptResponse",
     "FulfilledExecuteWorkflowWorkflowResultEvent",
@@ -910,6 +909,7 @@ __all__ = [
     "TestSuiteRunExternalExecConfigData",
     "TestSuiteRunExternalExecConfigDataRequest",
     "TestSuiteRunExternalExecConfigRequest",
+    "TestSuiteRunMetricArrayOutput",
     "TestSuiteRunMetricErrorOutput",
     "TestSuiteRunMetricJsonOutput",
     "TestSuiteRunMetricNumberOutput",

{vellum_ai-0.12.9 → vellum_ai-0.12.13}/src/vellum/client/types/function_call.py

@@ -2,7 +2,6 @@
 
 from ..core.pydantic_utilities import UniversalBaseModel
 import typing
-from .fulfilled_enum import FulfilledEnum
 from ..core.pydantic_utilities import IS_PYDANTIC_V2
 import pydantic
 
@@ -15,7 +14,6 @@ class FunctionCall(UniversalBaseModel):
     arguments: typing.Dict[str, typing.Optional[typing.Any]]
     id: typing.Optional[str] = None
     name: str
-    state: typing.Optional[FulfilledEnum] = None
 
     if IS_PYDANTIC_V2:
         model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2

{vellum_ai-0.12.9 → vellum_ai-0.12.13}/src/vellum/client/types/function_call_request.py

@@ -2,7 +2,6 @@
 
 from ..core.pydantic_utilities import UniversalBaseModel
 import typing
-from .fulfilled_enum import FulfilledEnum
 from ..core.pydantic_utilities import IS_PYDANTIC_V2
 import pydantic
 
@@ -15,7 +14,6 @@ class FunctionCallRequest(UniversalBaseModel):
     arguments: typing.Dict[str, typing.Optional[typing.Any]]
     id: typing.Optional[str] = None
     name: str
-    state: typing.Optional[FulfilledEnum] = None
 
     if IS_PYDANTIC_V2:
         model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2

{vellum_ai-0.12.9 → vellum_ai-0.12.13}/src/vellum/client/types/metric_definition_execution.py

@@ -1,10 +1,13 @@
 # This file was auto-generated by Fern from our API Definition.
 
+from __future__ import annotations
 from ..core.pydantic_utilities import UniversalBaseModel
+from .array_vellum_value import ArrayVellumValue
 import typing
 from .test_suite_run_metric_output import TestSuiteRunMetricOutput
 from ..core.pydantic_utilities import IS_PYDANTIC_V2
 import pydantic
+from ..core.pydantic_utilities import update_forward_refs
 
 
 class MetricDefinitionExecution(UniversalBaseModel):
@@ -18,3 +21,6 @@ class MetricDefinitionExecution(UniversalBaseModel):
             frozen = True
             smart_union = True
             extra = pydantic.Extra.allow
+
+
+update_forward_refs(ArrayVellumValue, MetricDefinitionExecution=MetricDefinitionExecution)

{vellum_ai-0.12.9 → vellum_ai-0.12.13}/src/vellum/client/types/test_suite_run_execution_metric_result.py

@@ -1,11 +1,14 @@
 # This file was auto-generated by Fern from our API Definition.
 
+from __future__ import annotations
 from ..core.pydantic_utilities import UniversalBaseModel
+from .array_vellum_value import ArrayVellumValue
 import typing
 from .test_suite_run_metric_output import TestSuiteRunMetricOutput
 from .test_suite_run_execution_metric_definition import TestSuiteRunExecutionMetricDefinition
 from ..core.pydantic_utilities import IS_PYDANTIC_V2
 import pydantic
+from ..core.pydantic_utilities import update_forward_refs
 
 
 class TestSuiteRunExecutionMetricResult(UniversalBaseModel):
@@ -22,3 +25,6 @@ class TestSuiteRunExecutionMetricResult(UniversalBaseModel):
             frozen = True
             smart_union = True
             extra = pydantic.Extra.allow
+
+
+update_forward_refs(ArrayVellumValue, TestSuiteRunExecutionMetricResult=TestSuiteRunExecutionMetricResult)

vellum_ai-0.12.13/src/vellum/client/types/test_suite_run_metric_array_output.py

@@ -0,0 +1,32 @@
+# This file was auto-generated by Fern from our API Definition.
+
+from __future__ import annotations
+from ..core.pydantic_utilities import UniversalBaseModel
+from .array_vellum_value import ArrayVellumValue
+import typing
+from .vellum_value import VellumValue
+from ..core.pydantic_utilities import IS_PYDANTIC_V2
+import pydantic
+from ..core.pydantic_utilities import update_forward_refs
+
+
+class TestSuiteRunMetricArrayOutput(UniversalBaseModel):
+    """
+    Output for a test suite run metric that is of type ARRAY
+    """
+
+    value: typing.Optional[typing.List[VellumValue]] = None
+    type: typing.Literal["ARRAY"] = "ARRAY"
+    name: str
+
+    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
+
+
+update_forward_refs(ArrayVellumValue, TestSuiteRunMetricArrayOutput=TestSuiteRunMetricArrayOutput)

{vellum_ai-0.12.9 → vellum_ai-0.12.13}/src/vellum/client/types/test_suite_run_metric_output.py

@@ -5,10 +5,12 @@ from .test_suite_run_metric_string_output import TestSuiteRunMetricStringOutput
 from .test_suite_run_metric_number_output import TestSuiteRunMetricNumberOutput
 from .test_suite_run_metric_json_output import TestSuiteRunMetricJsonOutput
 from .test_suite_run_metric_error_output import TestSuiteRunMetricErrorOutput
+from .test_suite_run_metric_array_output import TestSuiteRunMetricArrayOutput
 
 TestSuiteRunMetricOutput = typing.Union[
     TestSuiteRunMetricStringOutput,
     TestSuiteRunMetricNumberOutput,
     TestSuiteRunMetricJsonOutput,
     TestSuiteRunMetricErrorOutput,
+    TestSuiteRunMetricArrayOutput,
 ]

vellum_ai-0.12.13/src/vellum/types/test_suite_run_metric_array_output.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.test_suite_run_metric_array_output import *

{vellum_ai-0.12.9 → vellum_ai-0.12.13}/src/vellum/workflows/exceptions.py

@@ -2,7 +2,7 @@ from vellum.workflows.errors import WorkflowError, WorkflowErrorCode
 
 
 class NodeException(Exception):
-    def __init__(self, message: str, code: WorkflowErrorCode):
+    def __init__(self, message: str, code: WorkflowErrorCode = WorkflowErrorCode.INTERNAL_ERROR):
         self.message = message
         self.code = code
         super().__init__(message)

{vellum_ai-0.12.9 → vellum_ai-0.12.13}/src/vellum/workflows/nodes/displayable/bases/search_node.py

@@ -12,7 +12,6 @@ from vellum import (
     SearchWeightsRequest,
 )
 from vellum.core import ApiError, RequestOptions
-from vellum.utils.uuid import is_valid_uuid
 from vellum.workflows.errors import WorkflowErrorCode
 from vellum.workflows.exceptions import NodeException
 from vellum.workflows.nodes.bases import BaseNode
@@ -74,13 +73,10 @@ class BaseSearchNode(BaseNode[StateType], Generic[StateType]):
         results: List[SearchResult]
 
     def _perform_search(self) -> SearchResponse:
-        index_is_uuid = True if isinstance(self.document_index, UUID) else is_valid_uuid(self.document_index)
-
         try:
             return self._context.vellum_client.search(
                 query=self.query,
-                index_id=str(self.document_index) if index_is_uuid else None,
-                index_name=str(self.document_index) if not index_is_uuid else None,
+                document_index=str(self.document_index),
                 options=self.options,
             )
         except NotFoundError:

{vellum_ai-0.12.9 → vellum_ai-0.12.13}/src/vellum/workflows/nodes/displayable/inline_prompt_node/node.py

@@ -1,3 +1,4 @@
+import json
 from typing import Iterator
 
 from vellum.workflows.errors import WorkflowErrorCode
@@ -37,13 +38,19 @@ class InlinePromptNode(BaseInlinePromptNode[StateType]):
                 code=WorkflowErrorCode.INTERNAL_ERROR,
             )
 
-        string_output = next((output for output in outputs if output.type == "STRING"), None)
-        if not string_output or string_output.value is None:
-            output_types = {output.type for output in outputs}
-            is_plural = len(output_types) > 1
-            raise NodeException(
-                message=f"Expected to receive a non-null string output from Prompt. Only found outputs of type{'s' if is_plural else ''}: {', '.join(output_types)}",  # noqa: E501
-                code=WorkflowErrorCode.INTERNAL_ERROR,
-            )
-
-        yield BaseOutput(name="text", value=string_output.value)
+        string_outputs = []
+        for output in outputs:
+            if output.value is None:
+                continue
+
+            if output.type == "STRING":
+                string_outputs.append(output.value)
+            elif output.type == "JSON":
+                string_outputs.append(json.dumps(output.value, indent=4))
+            elif output.type == "FUNCTION_CALL":
+                string_outputs.append(output.value.model_dump_json(indent=4))
+            else:
+                string_outputs.append(output.value.message)
+
+        value = "\n".join(string_outputs)
+        yield BaseOutput(name="text", value=value)

{vellum_ai-0.12.9 → vellum_ai-0.12.13}/src/vellum/workflows/nodes/displayable/inline_prompt_node/tests/test_node.py

@@ -12,7 +12,6 @@ from vellum.client.types.initiated_execute_prompt_event import InitiatedExecuteP
 from vellum.client.types.prompt_output import PromptOutput
 from vellum.client.types.prompt_request_json_input import PromptRequestJsonInput
 from vellum.client.types.string_vellum_value import StringVellumValue
-from vellum.workflows.nodes.displayable.bases.inline_prompt_node.node import BaseInlinePromptNode
 from vellum.workflows.nodes.displayable.inline_prompt_node.node import InlinePromptNode
 
 
@@ -74,7 +73,7 @@ def test_inline_prompt_node__function_definitions(vellum_adhoc_prompt_client):
         pass
 
     # AND a prompt node with a accepting that function definition
-    class MyNode(BaseInlinePromptNode):
+    class MyNode(InlinePromptNode):
         ml_model = "gpt-4o"
         functions = [my_function]
         prompt_inputs = {}
@@ -99,7 +98,7 @@ def test_inline_prompt_node__function_definitions(vellum_adhoc_prompt_client):
     vellum_adhoc_prompt_client.adhoc_execute_prompt_stream.side_effect = generate_prompt_events
 
     # WHEN the node is run
-    list(MyNode().run())
+    outputs = list(MyNode().run())
 
     # THEN the prompt is executed with the correct inputs
     mock_api = vellum_adhoc_prompt_client.adhoc_execute_prompt_stream
@@ -117,3 +116,14 @@ def test_inline_prompt_node__function_definitions(vellum_adhoc_prompt_client):
             },
         ),
     ]
+    assert (
+        outputs[-1].value
+        == """{
+    "arguments": {
+        "foo": "hello",
+        "bar": 1
+    },
+    "id": null,
+    "name": "my_function"
+}"""
+    )

{vellum_ai-0.12.9 → vellum_ai-0.12.13}/src/vellum/workflows/nodes/displayable/tests/test_search_node_wth_text_output.py

@@ -118,8 +118,7 @@ witness or missing person.\
 
     # AND we should have made the expected call to Vellum search
     vellum_search_client.assert_called_once_with(
-        index_id=None,
-        index_name="vellum-trust-center-policies",
+        document_index="vellum-trust-center-policies",
         query="How often is employee training?",
         options=SearchRequestOptionsRequest(
             limit=8,

{vellum_ai-0.12.9 → vellum_ai-0.12.13}/src/vellum/workflows/runner/runner.py

@@ -4,7 +4,7 @@ import logging
 from queue import Empty, Queue
 from threading import Event as ThreadingEvent, Thread
 from uuid import UUID
-from typing import TYPE_CHECKING, Any, Dict, Generic, Iterable, Iterator, Optional, Sequence, Set, Type, Union
+from typing import TYPE_CHECKING, Any, Dict, Generic, Iterable, Iterator, List, Optional, Sequence, Set, Type, Union
 
 from vellum.workflows.constants import UNDEF
 from vellum.workflows.context import execution_context, get_parent_context
@@ -72,6 +72,7 @@ class WorkflowRunner(Generic[StateType]):
         entrypoint_nodes: Optional[RunFromNodeArg] = None,
         external_inputs: Optional[ExternalInputsArg] = None,
         cancel_signal: Optional[ThreadingEvent] = None,
+        node_output_mocks: Optional[List[BaseOutputs]] = None,
         parent_context: Optional[ParentContext] = None,
     ):
         if state and external_inputs:
@@ -123,6 +124,9 @@ class WorkflowRunner(Generic[StateType]):
 
         self._dependencies: Dict[Type[BaseNode], Set[Type[BaseNode]]] = defaultdict(set)
         self._state_forks: Set[StateType] = {self._initial_state}
+        self._mocks_by_node_outputs_class = (
+            {mock.__class__: mock for mock in node_output_mocks} if node_output_mocks else {}
+        )
 
         self._active_nodes_by_execution_id: Dict[UUID, BaseNode[StateType]] = {}
         self._cancel_signal = cancel_signal
@@ -178,8 +182,11 @@ class WorkflowRunner(Generic[StateType]):
                 node_definition=node.__class__,
                 parent=parent_context,
             )
-            with execution_context(parent_context=updated_parent_context):
-                node_run_response = node.run()
+            if node.Outputs not in self._mocks_by_node_outputs_class:
+                with execution_context(parent_context=updated_parent_context):
+                    node_run_response = node.run()
+            else:
+                node_run_response = self._mocks_by_node_outputs_class[node.Outputs]
             ports = node.Ports()
             if not isinstance(node_run_response, (BaseOutputs, Iterator)):
                 raise NodeException(

{vellum_ai-0.12.9 → vellum_ai-0.12.13}/src/vellum/workflows/state/encoder.py

@@ -2,6 +2,7 @@ from dataclasses import asdict, is_dataclass
 from datetime import datetime
 import enum
 from json import JSONEncoder
+from queue import Queue
 from uuid import UUID
 from typing import Any, Callable, Dict, Type
 
@@ -44,6 +45,9 @@ class DefaultStateEncoder(JSONEncoder):
         if isinstance(obj, enum.Enum):
             return obj.value
 
+        if isinstance(obj, Queue):
+            return list(obj.queue)
+
         if is_dataclass(obj):
             # Technically, obj is DataclassInstance | type[DataclassInstance], but asdict expects a DataclassInstance
             # in practice, we only ever pass the former

{vellum_ai-0.12.9 → vellum_ai-0.12.13}/src/vellum/workflows/state/tests/test_state.py

@@ -2,6 +2,7 @@ import pytest
 from collections import defaultdict
 from copy import deepcopy
 import json
+from queue import Queue
 from typing import Dict
 
 from vellum.workflows.nodes.bases import BaseNode
@@ -113,3 +114,21 @@ def test_state_json_serialization__with_node_output_updates():
 
     # THEN the state is serialized correctly
     assert json_state["meta"]["node_outputs"] == {"MockNode.Outputs.baz": "hello"}
+
+
+def test_state_json_serialization__with_queue():
+    # GIVEN an initial state instance
+    state = MockState(foo="bar")
+
+    # AND we add a Node Output queue to state
+    queue: Queue[str] = Queue()
+    queue.put("test1")
+    queue.put("test2")
+
+    state.meta.node_outputs[MockNode.Outputs.baz] = queue
+
+    # WHEN we serialize the state
+    json_state = json.loads(json.dumps(state, cls=DefaultStateEncoder))
+
+    # THEN the state is serialized correctly with the queue turned into a list
+    assert json_state["meta"]["node_outputs"] == {"MockNode.Outputs.baz": ["test1", "test2"]}

{vellum_ai-0.12.9 → vellum_ai-0.12.13}/src/vellum/workflows/workflows/base.py

@@ -194,6 +194,7 @@ class BaseWorkflow(Generic[WorkflowInputsType, StateType], metaclass=_BaseWorkfl
         entrypoint_nodes: Optional[RunFromNodeArg] = None,
         external_inputs: Optional[ExternalInputsArg] = None,
         cancel_signal: Optional[ThreadingEvent] = None,
+        node_output_mocks: Optional[List[BaseOutputs]] = None,
     ) -> TerminalWorkflowEvent:
         """
         Invoke a Workflow, returning the last event emitted, which should be one of:
@@ -218,6 +219,9 @@ class BaseWorkflow(Generic[WorkflowInputsType, StateType], metaclass=_BaseWorkfl
 
         cancel_signal: Optional[ThreadingEvent] = None
             A threading event that can be used to cancel the Workflow Execution.
+
+        node_output_mocks: Optional[List[Outputs]] = None
+            A list of Outputs to mock for Nodes during Workflow Execution.
         """
 
         events = WorkflowRunner(
@@ -227,6 +231,7 @@ class BaseWorkflow(Generic[WorkflowInputsType, StateType], metaclass=_BaseWorkfl
             entrypoint_nodes=entrypoint_nodes,
             external_inputs=external_inputs,
             cancel_signal=cancel_signal,
+            node_output_mocks=node_output_mocks,
             parent_context=self._context.parent_context,
         ).stream()
         first_event: Optional[Union[WorkflowExecutionInitiatedEvent, WorkflowExecutionResumedEvent]] = None
@@ -290,6 +295,7 @@ class BaseWorkflow(Generic[WorkflowInputsType, StateType], metaclass=_BaseWorkfl
         entrypoint_nodes: Optional[RunFromNodeArg] = None,
         external_inputs: Optional[ExternalInputsArg] = None,
         cancel_signal: Optional[ThreadingEvent] = None,
+        node_output_mocks: Optional[List[BaseOutputs]] = None,
     ) -> WorkflowEventStream:
         """
         Invoke a Workflow, yielding events as they are emitted.
@@ -315,6 +321,9 @@ class BaseWorkflow(Generic[WorkflowInputsType, StateType], metaclass=_BaseWorkfl
 
         cancel_signal: Optional[ThreadingEvent] = None
             A threading event that can be used to cancel the Workflow Execution.
+
+        node_output_mocks: Optional[List[Outputs]] = None
+            A list of Outputs to mock for Nodes during Workflow Execution.
         """
 
         should_yield = event_filter or workflow_event_filter
@@ -325,6 +334,7 @@ class BaseWorkflow(Generic[WorkflowInputsType, StateType], metaclass=_BaseWorkfl
             entrypoint_nodes=entrypoint_nodes,
             external_inputs=external_inputs,
             cancel_signal=cancel_signal,
+            node_output_mocks=node_output_mocks,
             parent_context=self.context.parent_context,
         ).stream():
             if should_yield(self.__class__, event):
@@ -416,7 +426,6 @@ class BaseWorkflow(Generic[WorkflowInputsType, StateType], metaclass=_BaseWorkfl
     def load_from_module(module_path: str) -> Type["BaseWorkflow"]:
         workflow_path = f"{module_path}.workflow"
         module = importlib.import_module(workflow_path)
-
         workflows: List[Type[BaseWorkflow]] = []
         for name in dir(module):
             if name.startswith("__"):
@@ -435,9 +444,40 @@ class BaseWorkflow(Generic[WorkflowInputsType, StateType], metaclass=_BaseWorkfl
             raise ValueError(f"No workflows found in {module_path}")
         elif len(workflows) > 1:
             raise ValueError(f"Multiple workflows found in {module_path}")
+        try:
+            BaseWorkflow.import_node_display(module_path)
+        except ModuleNotFoundError:
+            pass
 
         return workflows[0]
 
+    @staticmethod
+    def import_node_display(module_path):
+        # Import the nodes package
+        nodes_package = importlib.import_module(f"{module_path}.display.nodes")
+        # Use the loader to get the code
+        if hasattr(nodes_package, "__spec__") and nodes_package.__spec__ and nodes_package.__spec__.loader:
+            loader = nodes_package.__spec__.loader
+
+            # Check if the loader has a code attribute
+            if hasattr(loader, "code"):
+                code = loader.code
+
+                # Parse the code to find import statements
+                import_lines = [line.strip() for line in code.splitlines() if line.startswith("from ")]
+
+                # Import each module specified in the code
+                for line in import_lines:
+                    try:
+                        # Extract module name from the import line
+                        module_name = line.split(" ")[1]
+                        full_module_path = f"{module_path}.display.nodes{module_name}"
+                        importlib.import_module(full_module_path)
+                    except Exception:
+                        continue
+        # Also import from workflow.py
+        importlib.import_module(f"{module_path}.display.workflow")
+
 
 WorkflowExecutionInitiatedBody.model_rebuild()
 WorkflowExecutionFulfilledBody.model_rebuild()

vellum_ai-0.12.9/src/vellum/client/types/fulfilled_enum.py

@@ -1,5 +0,0 @@
-# This file was auto-generated by Fern from our API Definition.
-
-import typing
-
-FulfilledEnum = typing.Literal["FULFILLED"]

vellum_ai-0.12.9/src/vellum/types/fulfilled_enum.py

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