vellum-workflow-server 0.14.71.post3__py3-none-any.whl → 0.14.71.post5__py3-none-any.whl

{vellum_workflow_server-0.14.71.post3.dist-info → vellum_workflow_server-0.14.71.post5.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: vellum-workflow-server
-Version: 0.14.71.post3
+Version: 0.14.71.post5
 Summary: 
 License: AGPL
 Requires-Python: >=3.9.0,<4

{vellum_workflow_server-0.14.71.post3.dist-info → vellum_workflow_server-0.14.71.post5.dist-info}/RECORD

@@ -4,9 +4,9 @@ workflow_server/api/auth_middleware.py,sha256=IlZaCiwZ5nwQqk5sYQorvOFj7lt0p1ZSSE
 workflow_server/api/healthz_view.py,sha256=itiRvBDBXncrw8Kbbc73UZLwqMAhgHOR3uSre_dAfgY,404
 workflow_server/api/tests/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 workflow_server/api/tests/test_input_display_mapping.py,sha256=drBZqMudFyB5wgiUOcMgRXz7E7ge-Qgxbstw4E4f0zE,2211
-workflow_server/api/tests/test_workflow_view.py,sha256=39-Q7aSZwzTSpDJD13Ol17ZOga3dUTSuKHc-K0wrwKk,8716
+workflow_server/api/tests/test_workflow_view.py,sha256=zMG__ktijsxHAR0qfgp_crueCuqNUDpbYgMuEjTC1Wk,8509
 workflow_server/api/tests/test_workflow_view_stream_workflow_route.py,sha256=2gro4GD3FBuaA8T2-0oQxOOXh6zTf6hwxKb9CGU3x8g,24813
-workflow_server/api/workflow_view.py,sha256=WRsD8Wjs2O9G2p_X5USx1D3b2HzBqgb1bfSzjcZ7O2o,16193
+workflow_server/api/workflow_view.py,sha256=eNNao-TiM5x3-Ikcg9qxaJ5PNjd7NJGCAlRjCAAkVMw,16236
 workflow_server/code_exec_runner.py,sha256=tfijklTVkX4y45jeFTfrY2hVhdwo0VrLFc3SMeIiVYs,3096
 workflow_server/config.py,sha256=K5Tavm7wiqCZt0RWWue7zzb8N6e8aWnFOTNlBqEJPcI,1330
 workflow_server/core/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -26,7 +26,7 @@ workflow_server/utils/tests/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMp
 workflow_server/utils/tests/test_system_utils.py,sha256=MdBxI9gxUOpR_JBAHpEz6dGFY6JjxhMSM2oExpqFvNA,4314
 workflow_server/utils/tests/test_utils.py,sha256=qwK5Rmy3RQyjtlUrYAuGuDlBeRzZKsf1yS-y2IpUizQ,6452
 workflow_server/utils/utils.py,sha256=Wqqn-1l2ugkGgy5paWWdt0AVxAyPMQCYcnRSSOMjXlA,4355
-vellum_workflow_server-0.14.71.post3.dist-info/METADATA,sha256=r4Qhqly9YQRo_edwbtklqfx5H2C-fUY0nsMPF14isQs,2243
-vellum_workflow_server-0.14.71.post3.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-vellum_workflow_server-0.14.71.post3.dist-info/entry_points.txt,sha256=uB_0yPkr7YV6RhEXzvFReUM8P4OQBlVXD6TN6eb9-oc,277
-vellum_workflow_server-0.14.71.post3.dist-info/RECORD,,
+vellum_workflow_server-0.14.71.post5.dist-info/METADATA,sha256=OQeNKXKETaOWTvWrjPSsAHgza3O-LL2j0GI5nWLu-UY,2243
+vellum_workflow_server-0.14.71.post5.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+vellum_workflow_server-0.14.71.post5.dist-info/entry_points.txt,sha256=uB_0yPkr7YV6RhEXzvFReUM8P4OQBlVXD6TN6eb9-oc,277
+vellum_workflow_server-0.14.71.post5.dist-info/RECORD,,

workflow_server/api/tests/test_workflow_view.py

@@ -31,7 +31,6 @@ class TestNode(BaseNode):
 
     This is a detailed description of what the node does.
     \"""
-    label = "Test Node"
 """
     )
 
@@ -51,7 +50,7 @@ class TestNode(BaseNode):
     # AND the node should have the correct metadata
     node = nodes[0]
     assert UUID(node["id"])
-    assert node["module"] == "vellum_custom_nodes"
+    assert node["module"] == ["vellum_custom_nodes"]
     assert node["name"] == "TestNode"
     assert node["label"] == "Test Node"
     assert "A test node for processing data." in node["description"]
@@ -71,7 +70,6 @@ from vellum.workflows.nodes import BaseNode
 
 class SomeNode(BaseNode):
     \"""This is Some Node.\"""
-    label = "Some node"
 """
     )
 
@@ -83,7 +81,6 @@ from vellum.workflows.nodes import BaseNode
 
 class SomeOtherNode(BaseNode):
     \"""This is Some Other Node.\"""
-    label = "Some other node"
 """
     )
 
@@ -102,17 +99,17 @@ class SomeOtherNode(BaseNode):
 
     # AND the first node should have correct metadata
     some_node = nodes[0]
-    assert some_node["label"] == "Some node"
+    assert some_node["label"] == "Some Node"
     assert some_node["description"] == "This is Some Node."
     assert UUID(some_node["id"])
-    assert some_node["module"] == "vellum_custom_nodes"
+    assert some_node["module"] == ["vellum_custom_nodes"]
 
     # AND the second node should have correct metadata
     some_other_node = nodes[1]
-    assert some_other_node["label"] == "Some other node"
+    assert some_other_node["label"] == "Some Other Node"
     assert some_other_node["description"] == "This is Some Other Node."
     assert UUID(some_other_node["id"])
-    assert some_other_node["module"] == "vellum_custom_nodes"
+    assert some_other_node["module"] == ["vellum_custom_nodes"]
 
 
 def test_version_route__no_custom_nodes_dir(tmp_path):
@@ -143,11 +140,9 @@ from vellum.workflows.nodes import BaseNode
 
 class ProcessingNode(BaseNode):
     \"""Processes input data.\"""
-    label = "Processing Node"
 
 class TransformationNode(BaseNode):
     \"""Transforms data format.\"""
-    label = "Transformation Node"
 
 # This class should not be discovered
 class HelperClass:
@@ -185,7 +180,6 @@ from vellum.workflows.nodes import BaseNode
 
 class SomeNode(BaseNode):
     \"\"\"This is Some Node.\"\"\"
-    label = "Some node"
 """
     )
 
@@ -197,7 +191,6 @@ from vellum.workflows.nodes import BaseNode
 
 class BrokenNode(BaseNode)
     \"\"\"This node has a syntax error.\"\"\"
-    label = "Broken Node"
 """
     )
 
@@ -216,10 +209,10 @@ class BrokenNode(BaseNode)
 
     # AND the valid node should have correct metadata
     valid_node = nodes[0]
-    assert valid_node["label"] == "Some node"
+    assert valid_node["label"] == "Some Node"
     assert valid_node["description"] == "This is Some Node."
     assert UUID(valid_node["id"])
-    assert valid_node["module"] == "vellum_custom_nodes"
+    assert valid_node["module"] == ["vellum_custom_nodes"]
 
     # AND the error should be logged with full traceback
     assert len(caplog.records) > 0
@@ -248,7 +241,6 @@ class Inputs(BaseInputs):
 
 class MyAdditionNode(BaseNode):
     \"\"\"Custom node that performs simple addition.\"\"\"
-    label = "Addition Node"
     arg1: int = Inputs.arg1
     arg2: int = Inputs.arg2
 
@@ -276,8 +268,8 @@ class MyAdditionNode(BaseNode):
 
     # AND the node should have the correct metadata
     node = nodes[0]
-    assert node["label"] == "Addition Node"
+    assert node["label"] == "My Addition Node"
     assert node["description"] == "Custom node that performs simple addition."
     assert UUID(node["id"])
-    assert node["module"] == "vellum_custom_nodes"
+    assert node["module"] == ["vellum_custom_nodes"]
     assert node["name"] == "MyAdditionNode"

workflow_server/api/workflow_view.py

@@ -17,6 +17,7 @@ from flask import Blueprint, Response, current_app as app, request, stream_with_
 from pydantic import ValidationError
 
 from vellum.workflows.nodes import BaseNode
+from vellum.workflows.utils.names import pascal_to_title_case
 from workflow_server.config import MEMORY_LIMIT_MB
 from workflow_server.core.events import (
     STREAM_FINISHED_EVENT,
@@ -370,15 +371,15 @@ def get_version_route() -> tuple[dict, int]:
             for _, name, _ in pkgutil.iter_modules([custom_nodes_path]):
                 try:
                     module = importlib.import_module(f"{CUSTOM_NODES_DIRECTORY}.{name}")
-                    for _, obj in inspect.getmembers(module):
+                    for label, obj in inspect.getmembers(module):
                         # Look for classes that inherit from BaseNode
                         if inspect.isclass(obj) and obj != BaseNode and issubclass(obj, BaseNode):
                             nodes.append(
                                 {
                                     "id": str(uuid4()),
-                                    "module": CUSTOM_NODES_DIRECTORY,
+                                    "module": CUSTOM_NODES_DIRECTORY.split("/"),
                                     "name": obj.__name__,
-                                    "label": obj.label.instance if hasattr(obj, "label") else obj.__name__,
+                                    "label": pascal_to_title_case(label),
                                     "description": inspect.getdoc(obj) or "",
                                 }
                             )