vellum-workflow-server 0.14.71.post2__py3-none-any.whl → 0.14.71.post4__py3-none-any.whl

{vellum_workflow_server-0.14.71.post2.dist-info → vellum_workflow_server-0.14.71.post4.dist-info}/METADATA

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

{vellum_workflow_server-0.14.71.post2.dist-info → vellum_workflow_server-0.14.71.post4.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=b9paDtjUZY8IpRLLcFRnlTW6yB8nw09CZKOwxHgC098,8726
 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=8NIqSNgvRBCMwS2aJLlV-E_S_nD0ILxjKZRyMrFtouY,16204
 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.post2.dist-info/METADATA,sha256=G1JCUaiOnxMgutw-1itCINdfdKRIxIPwUxd8n8-VEqI,2243
-vellum_workflow_server-0.14.71.post2.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-vellum_workflow_server-0.14.71.post2.dist-info/entry_points.txt,sha256=uB_0yPkr7YV6RhEXzvFReUM8P4OQBlVXD6TN6eb9-oc,277
-vellum_workflow_server-0.14.71.post2.dist-info/RECORD,,
+vellum_workflow_server-0.14.71.post4.dist-info/METADATA,sha256=wjw3aXITpYnZhhSwje6_X0YitUOkCyAmkOW-3cLEKzI,2243
+vellum_workflow_server-0.14.71.post4.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+vellum_workflow_server-0.14.71.post4.dist-info/entry_points.txt,sha256=uB_0yPkr7YV6RhEXzvFReUM8P4OQBlVXD6TN6eb9-oc,277
+vellum_workflow_server-0.14.71.post4.dist-info/RECORD,,

workflow_server/api/tests/test_workflow_view.py

@@ -51,7 +51,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"]
@@ -105,14 +105,14 @@ class SomeOtherNode(BaseNode):
     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["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):
@@ -219,10 +219,65 @@ class BrokenNode(BaseNode)
     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
     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

@@ -376,9 +376,9 @@ def get_version_route() -> tuple[dict, int]:
                             nodes.append(
                                 {
                                     "id": str(uuid4()),
-                                    "module": CUSTOM_NODES_DIRECTORY,
+                                    "module": CUSTOM_NODES_DIRECTORY.split("/"),
                                     "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 "",
                                 }
                             )