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

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)),
     }
 
 
@@ -111,3 +109,18 @@ def test_try_node__use_parent_execution_context():
     # THEN the inner node had access to the key
     assert len(outputs) == 1
     assert outputs[-1] == BaseOutput(name="key", value="test-key")
+
+
+def test_try_node__resolved_inputs():
+    """
+    This test ensures that node attributes of TryNodes are correctly resolved.
+    """
+
+    class State(BaseState):
+        counter = 3.0
+
+    @TryNode.wrap()
+    class MyNode(BaseNode[State]):
+        foo = State.counter
+
+    assert MyNode.foo.types == (float,)

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