vellum-ai 0.10.9__py3-none-any.whl → 0.11.0__py3-none-any.whl

vellum/client/core/client_wrapper.py

@@ -17,7 +17,7 @@ class BaseClientWrapper:
         headers: typing.Dict[str, str] = {
             "X-Fern-Language": "Python",
             "X-Fern-SDK-Name": "vellum-ai",
-            "X-Fern-SDK-Version": "0.10.9",
+            "X-Fern-SDK-Version": "0.11.0",
         }
         headers["X_API_KEY"] = self.api_key
         return headers

vellum/evaluations/resources.py

@@ -1,17 +1,14 @@
 from __future__ import annotations
 
+from functools import cached_property
 import logging
 import time
-from functools import cached_property
-from typing import Callable, Generator, List, cast, Iterable
 from uuid import UUID
+from typing import Callable, Generator, Iterable, List, cast
 
-from vellum import TestSuiteRunRead, TestSuiteRunMetricNumberOutput
-from vellum.client import Vellum, OMIT
-from vellum.evaluations.constants import (
-    DEFAULT_MAX_POLLING_DURATION_MS,
-    DEFAULT_POLLING_INTERVAL_MS,
-)
+from vellum import TestSuiteRunMetricNumberOutput, TestSuiteRunRead
+from vellum.client import OMIT, Vellum
+from vellum.evaluations.constants import DEFAULT_MAX_POLLING_DURATION_MS, DEFAULT_POLLING_INTERVAL_MS
 from vellum.evaluations.exceptions import TestSuiteRunResultsException
 from vellum.evaluations.utils.env import get_api_key
 from vellum.evaluations.utils.paginator import PaginatedResults, get_all_results
@@ -21,9 +18,9 @@ from vellum.types import (
     ExternalTestCaseExecutionRequest,
     NamedTestCaseVariableValueRequest,
     TestCaseVariableValue,
-    TestSuiteRunExternalExecConfigRequest,
     TestSuiteRunExecution,
     TestSuiteRunExternalExecConfigDataRequest,
+    TestSuiteRunExternalExecConfigRequest,
     TestSuiteRunMetricOutput,
     TestSuiteRunState,
 )
@@ -161,9 +158,7 @@ class VellumTestSuiteRunResults:
 
         metric_outputs: list[TestSuiteRunMetricNumberOutput] = []
 
-        for output in self.get_metric_outputs(
-            metric_identifier=metric_identifier, output_identifier=output_identifier
-        ):
+        for output in self.get_metric_outputs(metric_identifier=metric_identifier, output_identifier=output_identifier):
             if output.type != "NUMBER":
                 raise TestSuiteRunResultsException(
                     f"Expected a numeric metric output, but got a {output.type} output instead."

vellum/evaluations/utils/env.py

@@ -6,8 +6,6 @@ from .exceptions import VellumClientException
 def get_api_key() -> str:
     api_key = os.environ.get("VELLUM_API_KEY")
     if api_key is None:
-        raise VellumClientException(
-            "`VELLUM_API_KEY` environment variable is required to be set."
-        )
+        raise VellumClientException("`VELLUM_API_KEY` environment variable is required to be set.")
 
     return api_key

vellum/evaluations/utils/paginator.py

@@ -1,7 +1,6 @@
 from dataclasses import dataclass
 from typing import Callable, Generator, Generic, List, TypeVar, Union
 
-
 Result = TypeVar("Result")
 
 

vellum/evaluations/utils/typing.py

@@ -1,4 +1,4 @@
-from typing import TypeVar, Optional
+from typing import Optional, TypeVar
 
 _T = TypeVar("_T")
 

vellum/evaluations/utils/uuid.py

@@ -1,5 +1,5 @@
-from typing import Union
 import uuid
+from typing import Union
 
 
 def is_valid_uuid(val: Union[str, uuid.UUID, None]) -> bool:

vellum/plugins/vellum_mypy.py

@@ -154,7 +154,9 @@ class VellumMypyPlugin(Plugin):
     def _base_class_hook(self, ctx: ClassDefContext) -> None:
         if _is_subclass(ctx.cls.info, "vellum.workflows.nodes.core.templating_node.node.TemplatingNode"):
             self._dynamic_output_node_class_hook(ctx, "result")
-        elif _is_subclass(ctx.cls.info, "vellum.workflows.nodes.displayable.code_execution_node.node.CodeExecutionNode"):
+        elif _is_subclass(
+            ctx.cls.info, "vellum.workflows.nodes.displayable.code_execution_node.node.CodeExecutionNode"
+        ):
             self._dynamic_output_node_class_hook(ctx, "result")
         elif _is_subclass(ctx.cls.info, "vellum.workflows.nodes.displayable.final_output_node.node.FinalOutputNode"):
             self._dynamic_output_node_class_hook(ctx, "value")

vellum/workflows/events/node.py

@@ -1,12 +1,10 @@
-from typing import Any, Dict, Generic, Iterable, List, Literal, Optional, Set, Type, Union
+from typing import TYPE_CHECKING, Any, Dict, Generic, List, Literal, Optional, Set, Type, Union
 
-from pydantic import ConfigDict, SerializerFunctionWrapHandler, field_serializer, model_serializer
-from pydantic.main import IncEx
+from pydantic import SerializerFunctionWrapHandler, field_serializer, model_serializer
 
 from vellum.core.pydantic_utilities import UniversalBaseModel
 from vellum.workflows.errors import VellumError
 from vellum.workflows.expressions.accessor import AccessorExpression
-from vellum.workflows.nodes.bases import BaseNode
 from vellum.workflows.outputs.base import BaseOutput
 from vellum.workflows.ports.port import Port
 from vellum.workflows.references.node import NodeReference
@@ -14,9 +12,12 @@ from vellum.workflows.types.generics import OutputsType
 
 from .types import BaseEvent, default_serializer, serialize_type_encoder
 
+if TYPE_CHECKING:
+    from vellum.workflows.nodes.bases import BaseNode
+
 
 class _BaseNodeExecutionBody(UniversalBaseModel):
-    node_definition: Type[BaseNode]
+    node_definition: Type["BaseNode"]
 
     @field_serializer("node_definition")
     def serialize_node_definition(self, node_definition: Type, _info: Any) -> Dict[str, Any]:
@@ -36,7 +37,7 @@ class _BaseNodeEvent(BaseEvent):
     body: _BaseNodeExecutionBody
 
     @property
-    def node_definition(self) -> Type[BaseNode]:
+    def node_definition(self) -> Type["BaseNode"]:
         return self.body.node_definition
 
 

vellum/workflows/events/tests/test_event.py

@@ -1,6 +1,5 @@
 import pytest
 from datetime import datetime
-import json
 from uuid import UUID
 
 from deepdiff import DeepDiff

vellum/workflows/events/types.py

@@ -1,5 +1,4 @@
 from datetime import datetime
-from enum import Enum
 import json
 from uuid import UUID, uuid4
 from typing import Annotated, Any, Dict, List, Literal, Optional, Union

vellum/workflows/events/workflow.py

@@ -6,7 +6,7 @@ from vellum.core.pydantic_utilities import UniversalBaseModel
 from vellum.workflows.errors import VellumError
 from vellum.workflows.outputs.base import BaseOutput
 from vellum.workflows.references import ExternalInputReference
-from vellum.workflows.types.generics import OutputsType, WorkflowInputsType
+from vellum.workflows.types.generics import OutputsType, StateType, WorkflowInputsType
 
 from .node import (
     NodeExecutionFulfilledEvent,
@@ -124,6 +124,23 @@ class WorkflowExecutionResumedEvent(_BaseWorkflowEvent):
     body: WorkflowExecutionResumedBody
 
 
+class WorkflowExecutionSnapshottedBody(_BaseWorkflowExecutionBody, Generic[StateType]):
+    state: StateType
+
+    @field_serializer("state")
+    def serialize_state(self, state: StateType, _info: Any) -> Dict[str, Any]:
+        return default_serializer(state)
+
+
+class WorkflowExecutionSnapshottedEvent(_BaseWorkflowEvent, Generic[StateType]):
+    name: Literal["workflow.execution.snapshotted"] = "workflow.execution.snapshotted"
+    body: WorkflowExecutionSnapshottedBody[StateType]
+
+    @property
+    def state(self) -> StateType:
+        return self.body.state
+
+
 GenericWorkflowEvent = Union[
     WorkflowExecutionStreamingEvent,
     WorkflowExecutionRejectedEvent,
@@ -141,6 +158,7 @@ WorkflowEvent = Union[
     GenericWorkflowEvent,
     WorkflowExecutionInitiatedEvent,
     WorkflowExecutionFulfilledEvent,
+    WorkflowExecutionSnapshottedEvent,
 ]
 
 WorkflowEventStream = Generator[WorkflowEvent, None, None]

vellum/workflows/nodes/bases/base.py

@@ -7,9 +7,7 @@ from typing import Any, Dict, Generic, Iterator, Optional, Set, Tuple, Type, Typ
 from vellum.workflows.constants import UNDEF
 from vellum.workflows.descriptors.base import BaseDescriptor
 from vellum.workflows.descriptors.utils import is_unresolved, resolve_value
-from vellum.workflows.edges.edge import Edge
 from vellum.workflows.errors.types import VellumErrorCode
-from vellum.workflows.events.types import ParentContext
 from vellum.workflows.exceptions import NodeException
 from vellum.workflows.graph import Graph
 from vellum.workflows.graph.graph import GraphTarget
@@ -32,15 +30,9 @@ def is_nested_class(nested: Any, parent: Type) -> bool:
         inspect.isclass(nested)
         # If a class is defined within a function, we don't consider it nested in the class defining that function
         # The example of this is a Subworkflow defined within TryNode.wrap()
-        and (
-            len(nested.__qualname__.split(".")) < 2
-            or nested.__qualname__.split(".")[-2] != "<locals>"
-        )
+        and (len(nested.__qualname__.split(".")) < 2 or nested.__qualname__.split(".")[-2] != "<locals>")
         and nested.__module__ == parent.__module__
-        and (
-            nested.__qualname__.startswith(parent.__name__)
-            or nested.__qualname__.startswith(parent.__qualname__)
-        )
+        and (nested.__qualname__.startswith(parent.__name__) or nested.__qualname__.startswith(parent.__qualname__))
     ) or any(is_nested_class(nested, base) for base in parent.__bases__)
 
 
@@ -92,9 +84,7 @@ class BaseNodeMeta(type):
                         **base.Trigger.__dict__,
                         "__module__": dct["__module__"],
                     }
-                    dct["Trigger"] = type(
-                        f"{name}.Trigger", (base.Trigger,), trigger_dct
-                    )
+                    dct["Trigger"] = type(f"{name}.Trigger", (base.Trigger,), trigger_dct)
                     break
 
         cls = super().__new__(mcs, name, bases, dct)
@@ -139,9 +129,7 @@ class BaseNodeMeta(type):
 
     def __rshift__(cls, other_cls: GraphTarget) -> Graph:
         if not issubclass(cls, BaseNode):
-            raise ValueError(
-                "BaseNodeMeta can only be extended from subclasses of BaseNode"
-            )
+            raise ValueError("BaseNodeMeta can only be extended from subclasses of BaseNode")
 
         if not cls.Ports._default_port:
             raise ValueError("No default port found on node")
@@ -153,9 +141,7 @@ class BaseNodeMeta(type):
 
     def __rrshift__(cls, other_cls: GraphTarget) -> Graph:
         if not issubclass(cls, BaseNode):
-            raise ValueError(
-                "BaseNodeMeta can only be extended from subclasses of BaseNode"
-            )
+            raise ValueError("BaseNodeMeta can only be extended from subclasses of BaseNode")
 
         if not isinstance(other_cls, set):
             other_cls = {other_cls}
@@ -193,18 +179,13 @@ class _BaseNodeTriggerMeta(type):
         if not isinstance(other, _BaseNodeTriggerMeta):
             return False
 
-        if not self.__name__.endswith(".Trigger") or not other.__name__.endswith(
-            ".Trigger"
-        ):
+        if not self.__name__.endswith(".Trigger") or not other.__name__.endswith(".Trigger"):
             return super().__eq__(other)
 
         self_trigger_class = cast(Type["BaseNode.Trigger"], self)
         other_trigger_class = cast(Type["BaseNode.Trigger"], other)
 
-        return (
-            self_trigger_class.node_class.__name__
-            == other_trigger_class.node_class.__name__
-        )
+        return self_trigger_class.node_class.__name__ == other_trigger_class.node_class.__name__
 
 
 class _BaseNodeExecutionMeta(type):
@@ -222,18 +203,13 @@ class _BaseNodeExecutionMeta(type):
         if not isinstance(other, _BaseNodeExecutionMeta):
             return False
 
-        if not self.__name__.endswith(".Execution") or not other.__name__.endswith(
-            ".Execution"
-        ):
+        if not self.__name__.endswith(".Execution") or not other.__name__.endswith(".Execution"):
             return super().__eq__(other)
 
         self_execution_class = cast(Type["BaseNode.Execution"], self)
         other_execution_class = cast(Type["BaseNode.Execution"], other)
 
-        return (
-            self_execution_class.node_class.__name__
-            == other_execution_class.node_class.__name__
-        )
+        return self_execution_class.node_class.__name__ == other_execution_class.node_class.__name__
 
 
 class BaseNode(Generic[StateType], metaclass=BaseNodeMeta):
@@ -271,9 +247,7 @@ class BaseNode(Generic[StateType], metaclass=BaseNodeMeta):
             """
 
             if cls.merge_behavior == MergeBehavior.AWAIT_ATTRIBUTES:
-                if state.meta.node_execution_cache.is_node_execution_initiated(
-                    cls.node_class, node_span_id
-                ):
+                if state.meta.node_execution_cache.is_node_execution_initiated(cls.node_class, node_span_id):
                     return False
 
                 is_ready = True
@@ -281,9 +255,7 @@ class BaseNode(Generic[StateType], metaclass=BaseNodeMeta):
                     if not descriptor.instance:
                         continue
 
-                    resolved_value = resolve_value(
-                        descriptor.instance, state, path=descriptor.name
-                    )
+                    resolved_value = resolve_value(descriptor.instance, state, path=descriptor.name)
                     if is_unresolved(resolved_value):
                         is_ready = False
                         break
@@ -291,29 +263,22 @@ class BaseNode(Generic[StateType], metaclass=BaseNodeMeta):
                 return is_ready
 
             if cls.merge_behavior == MergeBehavior.AWAIT_ANY:
-                if state.meta.node_execution_cache.is_node_execution_initiated(
-                    cls.node_class, node_span_id
-                ):
+                if state.meta.node_execution_cache.is_node_execution_initiated(cls.node_class, node_span_id):
                     return False
 
                 return True
 
             if cls.merge_behavior == MergeBehavior.AWAIT_ALL:
-                if state.meta.node_execution_cache.is_node_execution_initiated(
-                    cls.node_class, node_span_id
-                ):
+                if state.meta.node_execution_cache.is_node_execution_initiated(cls.node_class, node_span_id):
                     return False
 
                 """
                 A node utilizing an AWAIT_ALL merge strategy will only be considered ready for the Nth time
                 when all of its dependencies have been executed N times.
                 """
-                current_node_execution_count = (
-                    state.meta.node_execution_cache.get_execution_count(cls.node_class)
-                )
+                current_node_execution_count = state.meta.node_execution_cache.get_execution_count(cls.node_class)
                 return all(
-                    state.meta.node_execution_cache.get_execution_count(dep)
-                    == current_node_execution_count + 1
+                    state.meta.node_execution_cache.get_execution_count(dep) == current_node_execution_count + 1
                     for dep in dependencies
                 )
 
@@ -353,9 +318,7 @@ class BaseNode(Generic[StateType], metaclass=BaseNodeMeta):
             if not descriptor.instance:
                 continue
 
-            resolved_value = resolve_value(
-                descriptor.instance, self.state, path=descriptor.name, memo=inputs
-            )
+            resolved_value = resolve_value(descriptor.instance, self.state, path=descriptor.name, memo=inputs)
             setattr(self, descriptor.name, resolved_value)
 
         # Resolve descriptors set as defaults to the outputs class
@@ -379,9 +342,7 @@ class BaseNode(Generic[StateType], metaclass=BaseNodeMeta):
         for key, value in inputs.items():
             path_parts = key.split(".")
             node_attribute_discriptor = getattr(self.__class__, path_parts[0])
-            inputs_key = reduce(
-                lambda acc, part: acc[part], path_parts[1:], node_attribute_discriptor
-            )
+            inputs_key = reduce(lambda acc, part: acc[part], path_parts[1:], node_attribute_discriptor)
             all_inputs[inputs_key] = value
 
         self._inputs = MappingProxyType(all_inputs)

vellum/workflows/nodes/bases/tests/test_base_node.py

@@ -1,7 +1,6 @@
 from typing import Optional
 
 from vellum.core.pydantic_utilities import UniversalBaseModel
-
 from vellum.workflows.inputs.base import BaseInputs
 from vellum.workflows.nodes.bases.base import BaseNode
 from vellum.workflows.state.base import BaseState, StateMeta

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

@@ -87,6 +87,7 @@ class TemplatingNode(BaseNode[StateType], Generic[StateType, _OutputType], metac
 
         result: _OutputType - The result of the template rendering
         """
+
         # We use our mypy plugin to override the _OutputType with the actual output type
         # for downstream references to this output.
         result: _OutputType  # type: ignore[valid-type]

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

@@ -1,6 +1,6 @@
 import sys
-from types import MappingProxyType, ModuleType
-from typing import TYPE_CHECKING, Any, Callable, Dict, Generic, Iterator, Optional, Set, Tuple, Type, TypeVar, cast
+from types import ModuleType
+from typing import TYPE_CHECKING, Any, Callable, Dict, Generic, Iterator, Optional, Set, Tuple, Type, TypeVar
 
 from vellum.workflows.errors.types import VellumError, VellumErrorCode
 from vellum.workflows.exceptions import NodeException

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

@@ -30,9 +30,7 @@ def test_try_node__on_error_code__successfully_caught():
     assert len(outputs) == 2
     assert set(outputs) == {
         BaseOutput(name="value"),
-        BaseOutput(
-            name="error", value=VellumError(message="This will be caught", code=VellumErrorCode.PROVIDER_ERROR)
-        ),
+        BaseOutput(name="error", value=VellumError(message="This will be caught", code=VellumErrorCode.PROVIDER_ERROR)),
     }
 
 

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

@@ -8,7 +8,7 @@ from vellum.workflows.errors.types import VellumErrorCode
 from vellum.workflows.exceptions import NodeException
 from vellum.workflows.nodes.bases import BaseNode
 from vellum.workflows.outputs import BaseOutputs
-from vellum.workflows.types.core import Json, JsonObject, VellumSecret
+from vellum.workflows.types.core import Json, VellumSecret
 from vellum.workflows.types.generics import StateType
 
 

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

@@ -3,7 +3,6 @@ from typing import ClassVar, Generator, Generic, Iterator, List, Optional, Union
 
 from vellum import AdHocExecutePromptEvent, ExecutePromptEvent, PromptOutput
 from vellum.core import RequestOptions
-
 from vellum.workflows.errors.types import VellumErrorCode
 from vellum.workflows.exceptions import NodeException
 from vellum.workflows.nodes.bases import BaseNode

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

@@ -14,7 +14,6 @@ from vellum import (
     PromptRequestStringInput,
     VellumVariable,
 )
-
 from vellum.workflows.constants import OMIT
 from vellum.workflows.errors import VellumErrorCode
 from vellum.workflows.exceptions import NodeException

vellum/workflows/nodes/displayable/bases/prompt_deployment_node.py

@@ -25,7 +25,8 @@ class BasePromptDeploymentNode(BasePromptNode, Generic[StateType]):
     prompt_inputs: EntityInputsInterface - The inputs for the Prompt
     deployment: Union[UUID, str] - Either the Prompt Deployment's UUID or its name.
     release_tag: str - The release tag to use for the Prompt Execution
-    external_id: Optional[str] - Optionally include a unique identifier for tracking purposes. Must be unique within a given Prompt Deployment.
+    external_id: Optional[str] - Optionally include a unique identifier for tracking purposes.
+        Must be unique within a given Prompt Deployment.
     expand_meta: Optional[PromptDeploymentExpandMetaRequest] - Expandable execution fields to include in the response
     raw_overrides: Optional[RawPromptExecutionOverridesRequest] - The raw overrides to use for the Prompt Execution
     expand_raw: Optional[Sequence[str]] - Expandable raw fields to include in the response

vellum/workflows/nodes/displayable/bases/search_node.py

@@ -12,7 +12,6 @@ from vellum import (
     SearchWeightsRequest,
 )
 from vellum.core import ApiError, RequestOptions
-
 from vellum.workflows.errors import VellumErrorCode
 from vellum.workflows.exceptions import NodeException
 from vellum.workflows.nodes.bases import BaseNode

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

@@ -1,7 +1,6 @@
 import os
 
 from vellum import CodeExecutorResponse, NumberVellumValue, StringInput
-
 from vellum.workflows.inputs.base import BaseInputs
 from vellum.workflows.nodes.displayable.code_execution_node import CodeExecutionNode
 from vellum.workflows.references.vellum_secret import VellumSecretReference

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

@@ -4,12 +4,13 @@ from typing import Union
 
 def get_project_root() -> str:
     current_dir = os.getcwd()
-    while current_dir != '/':
+    while current_dir != "/":
         if ".git" in os.listdir(current_dir):
             return current_dir
         current_dir = os.path.dirname(current_dir)
     raise FileNotFoundError("Project root not found.")
 
+
 def read_file_from_path(filepath: str) -> Union[str, None]:
     project_root = get_project_root()
     relative_filepath = os.path.join(project_root, filepath)
@@ -17,5 +18,5 @@ def read_file_from_path(filepath: str) -> Union[str, None]:
     if not os.path.exists(relative_filepath):
         return None
 
-    with open(relative_filepath, 'r') as file:
+    with open(relative_filepath) as file:
         return file.read()

vellum/workflows/nodes/displayable/conditional_node/node.py

@@ -15,7 +15,7 @@ class ConditionalNode(BaseNode):
     """
 
     class Ports(NodePorts):
-        def __call__(self, outputs: BaseOutputs, state: BaseState) -> Set[Port]:    
+        def __call__(self, outputs: BaseOutputs, state: BaseState) -> Set[Port]:
             all_ports = [port for port in self.__class__]
             enforce_single_invoked_port = validate_ports(all_ports)
 

vellum/workflows/nodes/displayable/guardrail_node/node.py

@@ -3,7 +3,6 @@ from typing import Any, ClassVar, Dict, Generic, List, Optional, Union, cast
 
 from vellum import ChatHistoryInput, ChatMessage, JsonInput, MetricDefinitionInput, NumberInput, StringInput
 from vellum.core import RequestOptions
-
 from vellum.workflows.constants import LATEST_RELEASE_TAG
 from vellum.workflows.errors.types import VellumErrorCode
 from vellum.workflows.exceptions import NodeException

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

@@ -26,6 +26,7 @@ class InlinePromptNode(BaseInlinePromptNode[StateType]):
 
         text: str - The result of the Prompt Execution
         """
+
         text: str
 
     def run(self) -> Iterator[BaseOutput]:

vellum/workflows/nodes/displayable/prompt_deployment_node/node.py

@@ -14,7 +14,8 @@ class PromptDeploymentNode(BasePromptDeploymentNode[StateType]):
     prompt_inputs: EntityInputsInterface - The inputs for the Prompt
     deployment: Union[UUID, str] - Either the Prompt Deployment's UUID or its name.
     release_tag: str - The release tag to use for the Prompt Execution
-    external_id: Optional[str] - Optionally include a unique identifier for tracking purposes. Must be unique within a given Prompt Deployment.
+    external_id: Optional[str] - Optionally include a unique identifier for tracking purposes.
+        Must be unique within a given Prompt Deployment.
     expand_meta: Optional[PromptDeploymentExpandMetaRequest] - Expandable execution fields to include in the response
     raw_overrides: Optional[RawPromptExecutionOverridesRequest] - The raw overrides to use for the Prompt Execution
     expand_raw: Optional[Sequence[str]] - Expandable raw fields to include in the response
@@ -28,6 +29,7 @@ class PromptDeploymentNode(BasePromptDeploymentNode[StateType]):
 
         text: str - The result of the Prompt Execution
         """
+
         text: str
 
     def run(self) -> Iterator[BaseOutput]:

vellum/workflows/nodes/displayable/search_node/node.py

@@ -25,6 +25,7 @@ class SearchNode(BaseSearchNode[StateType]):
         results: List[SearchResult] - The raw search results
         text: str - The text of the search results joined by the chunk_separator
         """
+
         text: str
 
     def run(self) -> Outputs:

vellum/workflows/nodes/displayable/subworkflow_deployment_node/node.py

@@ -27,8 +27,9 @@ class SubworkflowDeploymentNode(BaseSubworkflowNode[StateType], Generic[StateTyp
     subworkflow_inputs: EntityInputsInterface - The inputs for the Subworkflow
     deployment: Union[UUID, str] - Either the Workflow Deployment's UUID or its name.
     release_tag: str = LATEST_RELEASE_TAG - The release tag to use for the Workflow Execution
-    external_id: Optional[str] = OMIT - Optionally include a unique identifier for tracking purposes. Must be unique within a given Workflow Deployment.
-    expand_meta: Optional[WorkflowExpandMetaRequest] = OMIT - Expandable execution fields to include in the respownse
+    external_id: Optional[str] = OMIT - Optionally include a unique identifier for tracking purposes.
+        Must be unique within a given Workflow Deployment.
+    expand_meta: Optional[WorkflowExpandMetaRequest] = OMIT - Expandable execution fields to include in the response
     metadata: Optional[Dict[str, Optional[Any]]] = OMIT - The metadata to use for the Workflow Execution
     request_options: Optional[RequestOptions] = None - The request options to use for the Workflow Execution
     """

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

@@ -138,10 +138,13 @@ def test_inline_text_prompt_node__catch_provider_error(vellum_adhoc_prompt_clien
     outputs = list(node.run())
 
     # THEN the node should have produced the outputs we expect
-    assert BaseOutput(
-        name="error",
-        value=SdkVellumError(
-            message="OpenAI failed",
-            code=VellumErrorCode.PROVIDER_ERROR,
-        ),
-    ) in outputs
+    assert (
+        BaseOutput(
+            name="error",
+            value=SdkVellumError(
+                message="OpenAI failed",
+                code=VellumErrorCode.PROVIDER_ERROR,
+            ),
+        )
+        in outputs
+    )

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

@@ -11,7 +11,6 @@ from vellum import (
     SearchResultMergingRequest,
     SearchWeightsRequest,
 )
-
 from vellum.workflows.inputs import BaseInputs
 from vellum.workflows.nodes.displayable.search_node import SearchNode as BaseSearchNode
 from vellum.workflows.state import BaseState

vellum/workflows/outputs/base.py

@@ -90,7 +90,7 @@ class BaseOutput(Generic[_Delta, _Accumulated]):
             return False
 
         return self.name == other.name and self.value == other.value and self.delta == other.delta
-    
+
     def __hash__(self) -> int:
         return hash((self._name, self._value, self._value))
 
@@ -193,9 +193,7 @@ class BaseOutputs(metaclass=_BaseOutputsMeta):
         if not isinstance(other, dict):
             return super().__eq__(other)
 
-        outputs = {
-            name: value for name, value in vars(self).items() if not name.startswith("_") and value is not UNDEF
-        }
+        outputs = {name: value for name, value in vars(self).items() if not name.startswith("_") and value is not UNDEF}
         return outputs == other
 
     def __repr__(self) -> str:

vellum/workflows/ports/node_ports.py

@@ -1,4 +1,4 @@
-from typing import Any, Dict, Iterable, Iterator, Optional, Set, Tuple, Type
+from typing import Any, Dict, Iterator, Optional, Set, Tuple, Type
 
 from vellum.workflows.outputs.base import BaseOutput, BaseOutputs
 from vellum.workflows.ports.port import Port