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

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/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.23.dist-info}/METADATA

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