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

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

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

{vellum_workflow_server-0.14.71.post1.dist-info → vellum_workflow_server-0.14.71.post3.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=wlVFBmKcoI-RdzfGPioeW46k6zaXyUeIerPc6m4aQls,7150
+workflow_server/api/tests/test_workflow_view.py,sha256=39-Q7aSZwzTSpDJD13Ol17ZOga3dUTSuKHc-K0wrwKk,8716
 workflow_server/api/tests/test_workflow_view_stream_workflow_route.py,sha256=2gro4GD3FBuaA8T2-0oQxOOXh6zTf6hwxKb9CGU3x8g,24813
-workflow_server/api/workflow_view.py,sha256=r6XCOlEsEe5bJ_qRcCPKPdsaU99jji-wUt1L7-H9pxY,16202
+workflow_server/api/workflow_view.py,sha256=WRsD8Wjs2O9G2p_X5USx1D3b2HzBqgb1bfSzjcZ7O2o,16193
 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.post1.dist-info/METADATA,sha256=7IZTwcIoUpNaBKCvdYdRnUcawnGwakNBrBodwxivxic,2243
-vellum_workflow_server-0.14.71.post1.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-vellum_workflow_server-0.14.71.post1.dist-info/entry_points.txt,sha256=uB_0yPkr7YV6RhEXzvFReUM8P4OQBlVXD6TN6eb9-oc,277
-vellum_workflow_server-0.14.71.post1.dist-info/RECORD,,
+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,,

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"

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 "",
                                 }
                             )