vellum-ai 1.3.1__py3-none-any.whl → 1.3.3__py3-none-any.whl

vellum/client/resources/ad_hoc/raw_client.py

@@ -16,10 +16,10 @@ from ...errors.forbidden_error import ForbiddenError
 from ...errors.internal_server_error import InternalServerError
 from ...types.ad_hoc_execute_prompt_event import AdHocExecutePromptEvent
 from ...types.ad_hoc_expand_meta import AdHocExpandMeta
+from ...types.deprecated_prompt_request_input import DeprecatedPromptRequestInput
 from ...types.function_definition import FunctionDefinition
 from ...types.prompt_block import PromptBlock
 from ...types.prompt_parameters import PromptParameters
-from ...types.prompt_request_input import PromptRequestInput
 from ...types.prompt_settings import PromptSettings
 from ...types.vellum_variable import VellumVariable
 
@@ -35,7 +35,7 @@ class RawAdHocClient:
         self,
         *,
         ml_model: str,
-        input_values: typing.Sequence[PromptRequestInput],
+        input_values: typing.Sequence[DeprecatedPromptRequestInput],
         input_variables: typing.Sequence[VellumVariable],
         parameters: PromptParameters,
         blocks: typing.Sequence[PromptBlock],
@@ -49,7 +49,7 @@ class RawAdHocClient:
         ----------
         ml_model : str
 
-        input_values : typing.Sequence[PromptRequestInput]
+        input_values : typing.Sequence[DeprecatedPromptRequestInput]
 
         input_variables : typing.Sequence[VellumVariable]
 
@@ -78,7 +78,7 @@ class RawAdHocClient:
             json={
                 "ml_model": ml_model,
                 "input_values": convert_and_respect_annotation_metadata(
-                    object_=input_values, annotation=typing.Sequence[PromptRequestInput], direction="write"
+                    object_=input_values, annotation=typing.Sequence[DeprecatedPromptRequestInput], direction="write"
                 ),
                 "input_variables": convert_and_respect_annotation_metadata(
                     object_=input_variables, annotation=typing.Sequence[VellumVariable], direction="write"
@@ -160,7 +160,7 @@ class RawAdHocClient:
         self,
         *,
         ml_model: str,
-        input_values: typing.Sequence[PromptRequestInput],
+        input_values: typing.Sequence[DeprecatedPromptRequestInput],
         input_variables: typing.Sequence[VellumVariable],
         parameters: PromptParameters,
         blocks: typing.Sequence[PromptBlock],
@@ -174,7 +174,7 @@ class RawAdHocClient:
         ----------
         ml_model : str
 
-        input_values : typing.Sequence[PromptRequestInput]
+        input_values : typing.Sequence[DeprecatedPromptRequestInput]
 
         input_variables : typing.Sequence[VellumVariable]
 
@@ -203,7 +203,7 @@ class RawAdHocClient:
             json={
                 "ml_model": ml_model,
                 "input_values": convert_and_respect_annotation_metadata(
-                    object_=input_values, annotation=typing.Sequence[PromptRequestInput], direction="write"
+                    object_=input_values, annotation=typing.Sequence[DeprecatedPromptRequestInput], direction="write"
                 ),
                 "input_variables": convert_and_respect_annotation_metadata(
                     object_=input_variables, annotation=typing.Sequence[VellumVariable], direction="write"
@@ -306,7 +306,7 @@ class AsyncRawAdHocClient:
         self,
         *,
         ml_model: str,
-        input_values: typing.Sequence[PromptRequestInput],
+        input_values: typing.Sequence[DeprecatedPromptRequestInput],
         input_variables: typing.Sequence[VellumVariable],
         parameters: PromptParameters,
         blocks: typing.Sequence[PromptBlock],
@@ -320,7 +320,7 @@ class AsyncRawAdHocClient:
         ----------
         ml_model : str
 
-        input_values : typing.Sequence[PromptRequestInput]
+        input_values : typing.Sequence[DeprecatedPromptRequestInput]
 
         input_variables : typing.Sequence[VellumVariable]
 
@@ -349,7 +349,7 @@ class AsyncRawAdHocClient:
             json={
                 "ml_model": ml_model,
                 "input_values": convert_and_respect_annotation_metadata(
-                    object_=input_values, annotation=typing.Sequence[PromptRequestInput], direction="write"
+                    object_=input_values, annotation=typing.Sequence[DeprecatedPromptRequestInput], direction="write"
                 ),
                 "input_variables": convert_and_respect_annotation_metadata(
                     object_=input_variables, annotation=typing.Sequence[VellumVariable], direction="write"
@@ -431,7 +431,7 @@ class AsyncRawAdHocClient:
         self,
         *,
         ml_model: str,
-        input_values: typing.Sequence[PromptRequestInput],
+        input_values: typing.Sequence[DeprecatedPromptRequestInput],
         input_variables: typing.Sequence[VellumVariable],
         parameters: PromptParameters,
         blocks: typing.Sequence[PromptBlock],
@@ -445,7 +445,7 @@ class AsyncRawAdHocClient:
         ----------
         ml_model : str
 
-        input_values : typing.Sequence[PromptRequestInput]
+        input_values : typing.Sequence[DeprecatedPromptRequestInput]
 
         input_variables : typing.Sequence[VellumVariable]
 
@@ -474,7 +474,7 @@ class AsyncRawAdHocClient:
             json={
                 "ml_model": ml_model,
                 "input_values": convert_and_respect_annotation_metadata(
-                    object_=input_values, annotation=typing.Sequence[PromptRequestInput], direction="write"
+                    object_=input_values, annotation=typing.Sequence[DeprecatedPromptRequestInput], direction="write"
                 ),
                 "input_variables": convert_and_respect_annotation_metadata(
                     object_=input_variables, annotation=typing.Sequence[VellumVariable], direction="write"

vellum/client/resources/events/client.py

@@ -4,8 +4,8 @@ import typing
 
 from ...core.client_wrapper import AsyncClientWrapper, SyncClientWrapper
 from ...core.request_options import RequestOptions
+from ...types.create_workflow_event_request import CreateWorkflowEventRequest
 from ...types.event_create_response import EventCreateResponse
-from ...types.workflow_event import WorkflowEvent
 from .raw_client import AsyncRawEventsClient, RawEventsClient
 
 # this is used as the default value for optional parameters
@@ -32,14 +32,14 @@ class EventsClient:
         return self._raw_client
 
     def create(
-        self, *, request: WorkflowEvent, request_options: typing.Optional[RequestOptions] = None
+        self, *, request: CreateWorkflowEventRequest, request_options: typing.Optional[RequestOptions] = None
     ) -> EventCreateResponse:
         """
-        Accept an event and publish it to ClickHouse for analytics processing.
+        Accept an event or list of events and publish them to ClickHouse for analytics processing.
 
         Parameters
         ----------
-        request : WorkflowEvent
+        request : CreateWorkflowEventRequest
 
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
@@ -65,22 +65,40 @@ class EventsClient:
             api_key="YOUR_API_KEY",
         )
         client.events.create(
-            request=NodeExecutionInitiatedEvent(
-                body=NodeExecutionInitiatedBody(
-                    node_definition=VellumCodeResourceDefinition(
-                        name="name",
-                        module=["module", "module"],
-                        id="id",
+            request=[
+                NodeExecutionInitiatedEvent(
+                    body=NodeExecutionInitiatedBody(
+                        node_definition=VellumCodeResourceDefinition(
+                            name="name",
+                            module=["module", "module"],
+                            id="id",
+                        ),
+                        inputs={"inputs": {"key": "value"}},
+                    ),
+                    id="id",
+                    timestamp=datetime.datetime.fromisoformat(
+                        "2024-01-15 09:30:00+00:00",
                     ),
-                    inputs={"inputs": {"key": "value"}},
+                    trace_id="trace_id",
+                    span_id="span_id",
                 ),
-                id="id",
-                timestamp=datetime.datetime.fromisoformat(
-                    "2024-01-15 09:30:00+00:00",
+                NodeExecutionInitiatedEvent(
+                    body=NodeExecutionInitiatedBody(
+                        node_definition=VellumCodeResourceDefinition(
+                            name="name",
+                            module=["module", "module"],
+                            id="id",
+                        ),
+                        inputs={"inputs": {"key": "value"}},
+                    ),
+                    id="id",
+                    timestamp=datetime.datetime.fromisoformat(
+                        "2024-01-15 09:30:00+00:00",
+                    ),
+                    trace_id="trace_id",
+                    span_id="span_id",
                 ),
-                trace_id="trace_id",
-                span_id="span_id",
-            ),
+            ],
         )
         """
         _response = self._raw_client.create(request=request, request_options=request_options)
@@ -103,14 +121,14 @@ class AsyncEventsClient:
         return self._raw_client
 
     async def create(
-        self, *, request: WorkflowEvent, request_options: typing.Optional[RequestOptions] = None
+        self, *, request: CreateWorkflowEventRequest, request_options: typing.Optional[RequestOptions] = None
     ) -> EventCreateResponse:
         """
-        Accept an event and publish it to ClickHouse for analytics processing.
+        Accept an event or list of events and publish them to ClickHouse for analytics processing.
 
         Parameters
         ----------
-        request : WorkflowEvent
+        request : CreateWorkflowEventRequest
 
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
@@ -140,22 +158,40 @@ class AsyncEventsClient:
 
         async def main() -> None:
             await client.events.create(
-                request=NodeExecutionInitiatedEvent(
-                    body=NodeExecutionInitiatedBody(
-                        node_definition=VellumCodeResourceDefinition(
-                            name="name",
-                            module=["module", "module"],
-                            id="id",
+                request=[
+                    NodeExecutionInitiatedEvent(
+                        body=NodeExecutionInitiatedBody(
+                            node_definition=VellumCodeResourceDefinition(
+                                name="name",
+                                module=["module", "module"],
+                                id="id",
+                            ),
+                            inputs={"inputs": {"key": "value"}},
                         ),
-                        inputs={"inputs": {"key": "value"}},
+                        id="id",
+                        timestamp=datetime.datetime.fromisoformat(
+                            "2024-01-15 09:30:00+00:00",
+                        ),
+                        trace_id="trace_id",
+                        span_id="span_id",
                     ),
-                    id="id",
-                    timestamp=datetime.datetime.fromisoformat(
-                        "2024-01-15 09:30:00+00:00",
+                    NodeExecutionInitiatedEvent(
+                        body=NodeExecutionInitiatedBody(
+                            node_definition=VellumCodeResourceDefinition(
+                                name="name",
+                                module=["module", "module"],
+                                id="id",
+                            ),
+                            inputs={"inputs": {"key": "value"}},
+                        ),
+                        id="id",
+                        timestamp=datetime.datetime.fromisoformat(
+                            "2024-01-15 09:30:00+00:00",
+                        ),
+                        trace_id="trace_id",
+                        span_id="span_id",
                     ),
-                    trace_id="trace_id",
-                    span_id="span_id",
-                ),
+                ],
             )
 
 

vellum/client/resources/events/raw_client.py

@@ -13,9 +13,9 @@ from ...errors.bad_request_error import BadRequestError
 from ...errors.forbidden_error import ForbiddenError
 from ...errors.too_many_requests_error import TooManyRequestsError
 from ...errors.unauthorized_error import UnauthorizedError
+from ...types.create_workflow_event_request import CreateWorkflowEventRequest
 from ...types.error_detail_response import ErrorDetailResponse
 from ...types.event_create_response import EventCreateResponse
-from ...types.workflow_event import WorkflowEvent
 
 # this is used as the default value for optional parameters
 OMIT = typing.cast(typing.Any, ...)
@@ -26,14 +26,14 @@ class RawEventsClient:
         self._client_wrapper = client_wrapper
 
     def create(
-        self, *, request: WorkflowEvent, request_options: typing.Optional[RequestOptions] = None
+        self, *, request: CreateWorkflowEventRequest, request_options: typing.Optional[RequestOptions] = None
     ) -> HttpResponse[EventCreateResponse]:
         """
-        Accept an event and publish it to ClickHouse for analytics processing.
+        Accept an event or list of events and publish them to ClickHouse for analytics processing.
 
         Parameters
         ----------
-        request : WorkflowEvent
+        request : CreateWorkflowEventRequest
 
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
@@ -47,7 +47,9 @@ class RawEventsClient:
             "monitoring/v1/events",
             base_url=self._client_wrapper.get_environment().default,
             method="POST",
-            json=convert_and_respect_annotation_metadata(object_=request, annotation=WorkflowEvent, direction="write"),
+            json=convert_and_respect_annotation_metadata(
+                object_=request, annotation=CreateWorkflowEventRequest, direction="write"
+            ),
             headers={
                 "content-type": "application/json",
             },
@@ -119,14 +121,14 @@ class AsyncRawEventsClient:
         self._client_wrapper = client_wrapper
 
     async def create(
-        self, *, request: WorkflowEvent, request_options: typing.Optional[RequestOptions] = None
+        self, *, request: CreateWorkflowEventRequest, request_options: typing.Optional[RequestOptions] = None
     ) -> AsyncHttpResponse[EventCreateResponse]:
         """
-        Accept an event and publish it to ClickHouse for analytics processing.
+        Accept an event or list of events and publish them to ClickHouse for analytics processing.
 
         Parameters
         ----------
-        request : WorkflowEvent
+        request : CreateWorkflowEventRequest
 
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
@@ -140,7 +142,9 @@ class AsyncRawEventsClient:
             "monitoring/v1/events",
             base_url=self._client_wrapper.get_environment().default,
             method="POST",
-            json=convert_and_respect_annotation_metadata(object_=request, annotation=WorkflowEvent, direction="write"),
+            json=convert_and_respect_annotation_metadata(
+                object_=request, annotation=CreateWorkflowEventRequest, direction="write"
+            ),
             headers={
                 "content-type": "application/json",
             },

vellum/client/types/__init__.py

@@ -82,6 +82,7 @@ from .container_image_build_config import ContainerImageBuildConfig
 from .container_image_container_image_tag import ContainerImageContainerImageTag
 from .container_image_read import ContainerImageRead
 from .create_test_suite_test_case_request import CreateTestSuiteTestCaseRequest
+from .create_workflow_event_request import CreateWorkflowEventRequest
 from .delimiter_chunker_config import DelimiterChunkerConfig
 from .delimiter_chunker_config_request import DelimiterChunkerConfigRequest
 from .delimiter_chunking import DelimiterChunking
@@ -92,6 +93,7 @@ from .deployment_provider_payload_response_payload import DeploymentProviderPayl
 from .deployment_read import DeploymentRead
 from .deployment_release_tag_deployment_history_item import DeploymentReleaseTagDeploymentHistoryItem
 from .deployment_release_tag_read import DeploymentReleaseTagRead
+from .deprecated_prompt_request_input import DeprecatedPromptRequestInput
 from .docker_service_token import DockerServiceToken
 from .document_chat_message_content import DocumentChatMessageContent
 from .document_chat_message_content_request import DocumentChatMessageContentRequest
@@ -648,6 +650,7 @@ from .workflow_execution_view_online_eval_metric_result import WorkflowExecution
 from .workflow_execution_workflow_result_event import WorkflowExecutionWorkflowResultEvent
 from .workflow_expand_meta_request import WorkflowExpandMetaRequest
 from .workflow_initialization_error import WorkflowInitializationError
+from .workflow_input import WorkflowInput
 from .workflow_node_result_data import WorkflowNodeResultData
 from .workflow_node_result_event import WorkflowNodeResultEvent
 from .workflow_node_result_event_state import WorkflowNodeResultEventState
@@ -765,6 +768,7 @@ __all__ = [
     "ContainerImageContainerImageTag",
     "ContainerImageRead",
     "CreateTestSuiteTestCaseRequest",
+    "CreateWorkflowEventRequest",
     "DelimiterChunkerConfig",
     "DelimiterChunkerConfigRequest",
     "DelimiterChunking",
@@ -775,6 +779,7 @@ __all__ = [
     "DeploymentRead",
     "DeploymentReleaseTagDeploymentHistoryItem",
     "DeploymentReleaseTagRead",
+    "DeprecatedPromptRequestInput",
     "DockerServiceToken",
     "DocumentChatMessageContent",
     "DocumentChatMessageContentRequest",
@@ -1311,6 +1316,7 @@ __all__ = [
     "WorkflowExecutionWorkflowResultEvent",
     "WorkflowExpandMetaRequest",
     "WorkflowInitializationError",
+    "WorkflowInput",
     "WorkflowNodeResultData",
     "WorkflowNodeResultEvent",
     "WorkflowNodeResultEventState",

vellum/client/types/create_workflow_event_request.py

@@ -0,0 +1,7 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+from .workflow_event import WorkflowEvent
+
+CreateWorkflowEventRequest = typing.Union[typing.List[WorkflowEvent], WorkflowEvent]

vellum/client/types/deprecated_prompt_request_input.py

@@ -0,0 +1,8 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+from .prompt_request_input import PromptRequestInput
+from .workflow_input import WorkflowInput
+
+DeprecatedPromptRequestInput = typing.Union[WorkflowInput, PromptRequestInput]

vellum/client/types/event_create_response.py

@@ -16,6 +16,11 @@ class EventCreateResponse(UniversalBaseModel):
     Indicates whether the event was published successfully.
     """
 
+    count: int = pydantic.Field()
+    """
+    Number of events processed
+    """
+
     if IS_PYDANTIC_V2:
         model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
     else:

vellum/client/types/logical_operator.py

@@ -22,6 +22,7 @@ LogicalOperator = typing.Union[
         "notIn",
         "between",
         "notBetween",
+        "concat",
         "blank",
         "notBlank",
         "coalesce",

vellum/client/types/processing_failure_reason_enum.py

@@ -2,4 +2,6 @@
 
 import typing
 
-ProcessingFailureReasonEnum = typing.Union[typing.Literal["EXCEEDED_CHARACTER_LIMIT", "INVALID_FILE"], typing.Any]
+ProcessingFailureReasonEnum = typing.Union[
+    typing.Literal["EXCEEDED_CHARACTER_LIMIT", "INVALID_FILE", "INVALID_CREDENTIALS"], typing.Any
+]

vellum/client/types/slim_document.py

@@ -39,6 +39,7 @@ class SlimDocument(UniversalBaseModel):
     
     * `EXCEEDED_CHARACTER_LIMIT` - Exceeded Character Limit
     * `INVALID_FILE` - Invalid File
+    * `INVALID_CREDENTIALS` - Invalid Credentials
     """
 
     status: typing.Optional[DocumentStatus] = pydantic.Field(default=None)

vellum/client/types/workflow_input.py

@@ -0,0 +1,31 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+from .array_input import ArrayInput
+from .audio_input import AudioInput
+from .chat_history_input import ChatHistoryInput
+from .document_input import DocumentInput
+from .error_input import ErrorInput
+from .function_call_input import FunctionCallInput
+from .image_input import ImageInput
+from .json_input import JsonInput
+from .number_input import NumberInput
+from .search_results_input import SearchResultsInput
+from .string_input import StringInput
+from .video_input import VideoInput
+
+WorkflowInput = typing.Union[
+    StringInput,
+    JsonInput,
+    ChatHistoryInput,
+    NumberInput,
+    SearchResultsInput,
+    ErrorInput,
+    ArrayInput,
+    FunctionCallInput,
+    AudioInput,
+    VideoInput,
+    ImageInput,
+    DocumentInput,
+]

vellum/types/create_workflow_event_request.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.create_workflow_event_request import *

vellum/types/deprecated_prompt_request_input.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.deprecated_prompt_request_input import *

vellum/types/workflow_input.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_input import *

vellum/workflows/constants.py

@@ -22,6 +22,9 @@ class _UndefMeta(type):
 
         return super().__getattribute__(name)
 
+    def __str__(cls) -> str:
+        return ""
+
     def __bool__(cls) -> bool:
         return False
 

vellum/workflows/emitters/vellum_emitter.py

@@ -1,5 +1,6 @@
 import logging
-from typing import Optional
+import threading
+from typing import List, Optional
 
 from vellum.core.request_options import RequestOptions
 from vellum.workflows.emitters.base import BaseWorkflowEmitter
@@ -29,6 +30,7 @@ class VellumEmitter(BaseWorkflowEmitter):
         *,
         timeout: Optional[float] = 30.0,
         max_retries: int = 3,
+        debounce_timeout: float = 0.1,
     ):
         """
         Initialize the VellumEmitter.
@@ -36,14 +38,19 @@ class VellumEmitter(BaseWorkflowEmitter):
         Args:
             timeout: Request timeout in seconds.
             max_retries: Maximum number of retry attempts for failed requests.
+            debounce_timeout: Time in seconds to wait before sending batched events.
         """
         super().__init__()
         self._timeout = timeout
         self._max_retries = max_retries
+        self._debounce_timeout = debounce_timeout
+        self._event_queue: List[SDKWorkflowEvent] = []
+        self._queue_lock = threading.Lock()
+        self._debounce_timer: Optional[threading.Timer] = None
 
     def emit_event(self, event: SDKWorkflowEvent) -> None:
         """
-        Emit a workflow event to Vellum's infrastructure.
+        Queue a workflow event for batched emission to Vellum's infrastructure.
 
         Args:
             event: The workflow event to emit.
@@ -55,10 +62,45 @@ class VellumEmitter(BaseWorkflowEmitter):
             return
 
         try:
-            self._send_event(event)
+            with self._queue_lock:
+                self._event_queue.append(event)
 
+                if self._debounce_timer:
+                    self._debounce_timer.cancel()
+
+                self._debounce_timer = threading.Timer(self._debounce_timeout, self._flush_events)
+                self._debounce_timer.start()
+
+        except Exception as e:
+            logger.exception(f"Failed to queue event {event.name}: {e}")
+
+    def _flush_events(self) -> None:
+        """
+        Send all queued events as a batch to Vellum's infrastructure.
+        """
+        with self._queue_lock:
+            if not self._event_queue:
+                return
+
+            events_to_send = self._event_queue.copy()
+            self._event_queue.clear()
+            self._debounce_timer = None
+
+        try:
+            self._send_events(events_to_send)
         except Exception as e:
-            logger.exception(f"Failed to emit event {event.name}: {e}")
+            logger.exception(f"Failed to send batched events: {e}")
+
+    def __del__(self) -> None:
+        """
+        Cleanup: flush any pending events and cancel timer.
+        """
+        try:
+            if self._debounce_timer:
+                self._debounce_timer.cancel()
+            self._flush_events()
+        except Exception:
+            pass
 
     def snapshot_state(self, state: BaseState) -> None:
         """
@@ -69,23 +111,27 @@ class VellumEmitter(BaseWorkflowEmitter):
         """
         pass
 
-    def _send_event(self, event: SDKWorkflowEvent) -> None:
+    def _send_events(self, events: List[SDKWorkflowEvent]) -> None:
         """
-        Send event to Vellum's events endpoint using client.events.create.
+        Send events to Vellum's events endpoint using client.events.create.
 
         Args:
-            event: The WorkflowEvent object to send.
+            events: List of WorkflowEvent objects to send.
         """
         if not self._context:
-            logger.warning("Cannot send event: No workflow context registered")
+            logger.warning("Cannot send events: No workflow context registered")
+            return
+
+        if not events:
             return
 
         client = self._context.vellum_client
         request_options = RequestOptions(timeout_in_seconds=self._timeout, max_retries=self._max_retries)
+
         client.events.create(
             # The API accepts a ClientWorkflowEvent but our SDK emits an SDKWorkflowEvent. These shapes are
             # meant to be identical, just with different helper methods. We may consolidate the two in the future.
             # But for now, the type ignore allows us to avoid an additional Model -> json -> Model conversion.
-            request=event,  # type: ignore[arg-type]
+            request=events,  # type: ignore[arg-type]
             request_options=request_options,
         )

vellum/workflows/events/node.py

@@ -141,6 +141,7 @@ class NodeExecutionFulfilledEvent(_BaseNodeEvent, Generic[OutputsType]):
 
 class NodeExecutionRejectedBody(_BaseNodeExecutionBody):
     error: WorkflowError
+    traceback: Optional[str] = None
 
 
 class NodeExecutionRejectedEvent(_BaseNodeEvent):

vellum/workflows/events/tests/test_event.py

@@ -261,6 +261,7 @@ mock_node_uuid = str(uuid4_from_hash(MockNode.__qualname__))
                         "message": "Workflow failed",
                         "code": "USER_DEFINED_ERROR",
                     },
+                    "traceback": None,
                 },
                 "parent": None,
                 "links": None,

vellum/workflows/events/workflow.py

@@ -156,6 +156,7 @@ class WorkflowExecutionFulfilledEvent(_BaseWorkflowEvent, Generic[OutputsType]):
 
 class WorkflowExecutionRejectedBody(_BaseWorkflowExecutionBody):
     error: WorkflowError
+    traceback: Optional[str] = None
 
 
 class WorkflowExecutionRejectedEvent(_BaseWorkflowEvent):

vellum/workflows/nodes/core/retry_node/tests/test_node.py

@@ -6,7 +6,6 @@ from vellum.workflows.inputs.base import BaseInputs
 from vellum.workflows.nodes.bases import BaseNode
 from vellum.workflows.nodes.core.retry_node.node import RetryNode
 from vellum.workflows.outputs import BaseOutputs
-from vellum.workflows.references.lazy import LazyReference
 from vellum.workflows.state.base import BaseState, StateMeta
 
 
@@ -102,7 +101,7 @@ def test_retry_node__condition_arg_successfully_retries():
     # AND a retry node that retries on a condition
     @RetryNode.wrap(
         max_attempts=5,
-        retry_on_condition=LazyReference(lambda: State.count.less_than(3)),
+        retry_on_condition=State.count.less_than(3),
     )
     class TestNode(BaseNode[State]):
         attempt_number = RetryNode.SubworkflowInputs.attempt_number