vellum-workflow-server 0.14.71.post3__tar.gz → 0.14.71.post5__tar.gz

{vellum_workflow_server-0.14.71.post3 → vellum_workflow_server-0.14.71.post5}/PKG-INFO

@@ -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 → vellum_workflow_server-0.14.71.post5}/pyproject.toml

@@ -3,7 +3,7 @@ name = "vellum-workflow-server"
 
 [tool.poetry]
 name = "vellum-workflow-server"
-version = "0.14.71.post3"
+version = "0.14.71.post5"
 description = ""
 readme = "README.md"
 authors = []

{vellum_workflow_server-0.14.71.post3 → vellum_workflow_server-0.14.71.post5}/src/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"

{vellum_workflow_server-0.14.71.post3 → vellum_workflow_server-0.14.71.post5}/src/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 "",
                                 }
                             )