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

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

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

{vellum_workflow_server-0.14.71.post2 → vellum_workflow_server-0.14.71.post3}/pyproject.toml

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

{vellum_workflow_server-0.14.71.post2 → vellum_workflow_server-0.14.71.post3}/src/workflow_server/api/tests/test_workflow_view.py

@@ -226,3 +226,58 @@ class BrokenNode(BaseNode)
     error_message = caplog.records[0].message
     assert "Failed to load node from module invalid_node" in error_message
     assert "invalid_node.py, line 4" in error_message
+
+
+def test_version_route__with_inputs(tmp_path):
+    # GIVEN a temporary custom_nodes directory
+    custom_nodes_dir = tmp_path / "vellum_custom_nodes"
+    custom_nodes_dir.mkdir()
+
+    # AND an addition node file
+    node_file = custom_nodes_dir / "addition_node.py"
+    node_file.write_text(
+        """
+from vellum.workflows.inputs import BaseInputs
+from vellum.workflows.nodes import BaseNode
+
+
+class Inputs(BaseInputs):
+    arg1: int
+    arg2: int
+
+
+class MyAdditionNode(BaseNode):
+    \"\"\"Custom node that performs simple addition.\"\"\"
+    label = "Addition Node"
+    arg1: int = Inputs.arg1
+    arg2: int = Inputs.arg2
+
+    class Outputs(BaseNode.Outputs):
+        result: int
+
+    def run(self) -> BaseNode.Outputs:
+        result = self.arg1 + self.arg2
+        return self.Outputs(result=result)
+"""
+    )
+
+    flask_app = create_app()
+
+    # WHEN we make a request to the version route
+    with patch("os.getcwd", return_value=str(tmp_path)), flask_app.test_client() as test_client:
+        response = test_client.get("/workflow/version")
+
+    # THEN we should get a successful response
+    assert response.status_code == 200
+
+    # AND we should find the addition node
+    nodes = response.json["nodes"]
+    assert len(nodes) == 1
+
+    # AND the node should have the correct metadata
+    node = nodes[0]
+    assert node["label"] == "Addition Node"
+    assert node["description"] == "Custom node that performs simple addition."
+    assert UUID(node["id"])
+    assert node["module"] == "vellum_custom_nodes"
+    assert node["name"] == "MyAdditionNode"

{vellum_workflow_server-0.14.71.post2 → vellum_workflow_server-0.14.71.post3}/src/workflow_server/api/workflow_view.py

@@ -378,7 +378,7 @@ def get_version_route() -> tuple[dict, int]:
                                     "id": str(uuid4()),
                                     "module": CUSTOM_NODES_DIRECTORY,
                                     "name": obj.__name__,
-                                    "label": obj().label if hasattr(obj, "label") else obj.__name__,  # type: ignore
+                                    "label": obj.label.instance if hasattr(obj, "label") else obj.__name__,
                                     "description": inspect.getdoc(obj) or "",
                                 }
                             )