vellum-ai 0.14.22__py3-none-any.whl → 0.14.24__py3-none-any.whl

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

@@ -13,6 +13,7 @@ from vellum.workflows.state.base import BaseState
 from vellum.workflows.state.context import WorkflowContext
 from vellum.workflows.types.core import EntityInputsInterface
 from vellum.workflows.types.generics import InputsType, StateType
+from vellum.workflows.utils.uuids import uuid4_from_hash
 from vellum.workflows.workflows.event_filters import all_workflow_event_filter
 
 if TYPE_CHECKING:
@@ -140,3 +141,8 @@ class InlineSubworkflowNode(
         # Subclasses of InlineSubworkflowNode can override this method to provider their own
         # approach to annotating the outputs class based on the `subworkflow.Outputs`
         setattr(outputs_class, reference.name, reference)
+
+        if cls.__output_ids__ is None:
+            cls.__output_ids__ = {}
+
+        cls.__output_ids__[reference.name] = uuid4_from_hash(f"{cls.__id__}|{reference.name}")

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

@@ -116,3 +116,30 @@ def test_inline_subworkflow_node__base_inputs_validation():
     # AND the error message should indicate the missing required input
     assert e.value.code == WorkflowErrorCode.INVALID_INPUTS
     assert "Required input variables required_input should have defined value" == str(e.value)
+
+
+def test_inline_subworkflow_node__with_adornment():
+    # GIVEN a simple inline subworkflow with an output
+    class InnerNode(BaseNode):
+        class Outputs(BaseNode.Outputs):
+            final_output = "hello"
+
+    class TestSubworkflow(BaseWorkflow):
+        graph = InnerNode
+
+        class Outputs(BaseWorkflow.Outputs):
+            final_output = InnerNode.Outputs.final_output
+
+    # AND it's wrapped in a TryNode
+    @TryNode.wrap()
+    class TestNode(InlineSubworkflowNode):
+        subworkflow = TestSubworkflow
+
+    # THEN the wrapped node should have the correct output IDs
+    assert "final_output" in TestNode.__output_ids__
+
+    # AND when we run the node
+    node = TestNode()
+    outputs = list(node.run())
+
+    assert outputs[-1].name == "final_output" and outputs[-1].value == "hello"

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

@@ -33,29 +33,6 @@ def test_retry_node__retry_on_error_code__successfully_retried():
     assert outputs.execution_count == 3
 
 
-def test_retry_node__retry_on_error_code_all():
-    # GIVEN a retry node that is configured to retry on all errors
-    @RetryNode.wrap(max_attempts=3)
-    class TestNode(BaseNode):
-        attempt_number = RetryNode.SubworkflowInputs.attempt_number
-
-        class Outputs(BaseOutputs):
-            execution_count: int
-
-        def run(self) -> Outputs:
-            if self.attempt_number < 3:
-                raise NodeException(message="This will be retried", code=WorkflowErrorCode.PROVIDER_ERROR)
-
-            return self.Outputs(execution_count=self.attempt_number)
-
-    # WHEN the node is run and throws a None
-    node = TestNode(state=BaseState())
-    outputs = node.run()
-
-    # THEN the exception is retried
-    assert outputs.execution_count == 3
-
-
 def test_retry_node__retry_on_error_code__missed():
     # GIVEN a retry node that is configured to retry on PROVIDER_ERROR
     @RetryNode.wrap(max_attempts=3, retry_on_error_code=WorkflowErrorCode.PROVIDER_ERROR)

vellum/workflows/workflows/base.py

@@ -110,10 +110,10 @@ class _BaseWorkflowMeta(type):
                         nodes.update(node for node in item.nodes)
                     elif inspect.isclass(item) and issubclass(item, BaseNode):
                         nodes.add(item)
-            elif issubclass(graph_item, BaseNode):
+            elif inspect.isclass(graph_item) and issubclass(graph_item, BaseNode):
                 nodes.add(graph_item)
             else:
-                raise ValueError(f"Unexpected graph type: {graph_item.__class__}")
+                raise TypeError(f"Unexpected graph type: {graph_item.__class__}")
             return nodes
 
         graph_nodes = collect_nodes(dct.get("graph", set()))

vellum/workflows/workflows/tests/test_base_workflow.py

@@ -304,3 +304,13 @@ def test_workflow__node_in_both_graph_and_unused():
 
     # THEN it should raise an error
     assert "Node(s) NodeA cannot appear in both graph and unused_graphs" in str(exc_info.value)
+
+
+def test_workflow__unsupported_graph_item():
+    with pytest.raises(TypeError) as exc_info:
+        # GIVEN a workflow with an unsupported graph item
+        class TestWorkflow(BaseWorkflow[BaseInputs, BaseState]):
+            graph = 1  # type: ignore
+
+    # THEN it should raise an error
+    assert "Unexpected graph type: <class 'int'>" in str(exc_info.value)

{vellum_ai-0.14.22.dist-info → vellum_ai-0.14.24.dist-info}/METADATA

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