vellum-workflow-server 1.0.6__tar.gz → 1.0.7__tar.gz

{vellum_workflow_server-1.0.6 → vellum_workflow_server-1.0.7}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: vellum-workflow-server
-Version: 1.0.6
+Version: 1.0.7
 Summary: 
 License: AGPL
 Requires-Python: >=3.9.0,<4
@@ -29,7 +29,7 @@ Requires-Dist: pyjwt (==2.10.0)
 Requires-Dist: python-dotenv (==1.0.1)
 Requires-Dist: retrying (==1.3.4)
 Requires-Dist: sentry-sdk[flask] (==2.20.0)
-Requires-Dist: vellum-ai (==1.0.6)
+Requires-Dist: vellum-ai (==1.0.7)
 Description-Content-Type: text/markdown
 
 # Vellum Workflow Runner Server

{vellum_workflow_server-1.0.6 → vellum_workflow_server-1.0.7}/pyproject.toml

@@ -3,7 +3,7 @@ name = "vellum-workflow-server"
 
 [tool.poetry]
 name = "vellum-workflow-server"
-version = "1.0.6"
+version = "1.0.7"
 description = ""
 readme = "README.md"
 authors = []
@@ -45,7 +45,7 @@ flask = "2.3.3"
 orderly-set = "5.2.2"
 pebble = "5.0.7"
 gunicorn = "23.0.0"
-vellum-ai = "1.0.6"
+vellum-ai = "1.0.7"
 python-dotenv = "1.0.1"
 retrying = "1.3.4"
 sentry-sdk = {extras = ["flask"], version = "2.20.0"}

{vellum_workflow_server-1.0.6 → vellum_workflow_server-1.0.7}/src/workflow_server/api/tests/test_workflow_view.py

@@ -3,6 +3,8 @@ import re
 from unittest.mock import patch
 from uuid import UUID
 
+from deepdiff import DeepDiff
+
 from workflow_server.server import create_app
 
 
@@ -376,3 +378,80 @@ class MyAdditionNode(BaseNode):
         "trigger": {"id": "a5298668-d808-4a45-a62e-790943948e8a", "merge_behavior": "AWAIT_ATTRIBUTES"},
         "type": "GENERIC",
     }
+
+
+def test_serialize_route__with_no_files():
+    # GIVEN a Flask application
+    flask_app = create_app()
+
+    # WHEN we make a request with no files
+    with flask_app.test_client() as test_client:
+        response = test_client.post("/workflow/serialize", json={"files": {}})
+
+    # THEN we should get a bad request response
+    assert response.status_code == 400
+
+    # AND the response should contain an error message
+    assert "detail" in response.json
+    assert "No files received" in response.json["detail"]
+
+
+def test_serialize_route__with_invalid_python_syntax():
+    # GIVEN a Flask application
+    flask_app = create_app()
+
+    # AND a file with invalid Python syntax
+    invalid_content = """
+from vellum.workflows.nodes import BaseNode
+
+class BrokenNode(BaseNode)  # Missing colon
+    \"\"\"This node has a syntax error.\"\"\"
+"""
+
+    # WHEN we make a request to the serialize route
+    with flask_app.test_client() as test_client:
+        response = test_client.post("/workflow/serialize", json={"files": {"broken_node.py": invalid_content}})
+
+    # THEN we should get a server error response
+    assert response.status_code == 500
+
+    # AND the response should contain an error message
+    assert "detail" in response.json
+    assert "Serialization failed" in response.json["detail"]
+
+
+def test_serialize_route__with__workflow():
+    # GIVEN a Flask application
+    flask_app = create_app()
+
+    # AND a complete workflow with multiple files
+    workflow_files = {
+        "__init__.py": "# flake8: noqa: F401, F403\n\n",
+        "inputs.py": "from typing import Any, Optional\n\nfrom vellum.workflows.inputs import BaseInputs\n\n\nclass Inputs(BaseInputs):\n    text: str\n    var_1: Optional[Any]\n",  # noqa: E501
+        "workflow.py": "from vellum.workflows import BaseWorkflow\nfrom vellum.workflows.state import BaseState\n\nfrom .inputs import Inputs\nfrom .nodes.final_output import FinalOutput\nfrom .nodes.templating_node import TemplatingNode\n\n\nclass Workflow(BaseWorkflow[Inputs, BaseState]):\n    graph = TemplatingNode >> FinalOutput\n\n    class Outputs(BaseWorkflow.Outputs):\n        final_output = FinalOutput.Outputs.value\n",  # noqa: E501
+        "nodes/__init__.py": 'from .final_output import FinalOutput\nfrom .templating_node import TemplatingNode\n\n__all__ = [\n    "FinalOutput",\n    "TemplatingNode",\n]\n',  # noqa: E501
+        "nodes/templating_node.py": 'from vellum.workflows.nodes.displayable import TemplatingNode as BaseTemplatingNode\nfrom vellum.workflows.state import BaseState\n\nfrom ..inputs import Inputs\n\n\nclass TemplatingNode(BaseTemplatingNode[BaseState, str]):\n    template = """{{ text }}"""\n    inputs = {\n        "text": Inputs.text,\n    }\n',  # noqa: E501
+        "nodes/final_output.py": "from vellum.workflows.nodes.displayable import FinalOutputNode\nfrom vellum.workflows.state import BaseState\n\nfrom .templating_node import TemplatingNode\n\n\nclass FinalOutput(FinalOutputNode[BaseState, str]):\n    class Outputs(FinalOutputNode.Outputs):\n        value = TemplatingNode.Outputs.result\n",  # noqa: E501
+    }
+
+    # WHEN we make a request to the serialize route
+    with flask_app.test_client() as test_client:
+        response = test_client.post("/workflow/serialize", json={"files": workflow_files})
+
+    # THEN we should get a successful response
+    assert response.status_code == 200
+
+    # AND the response should contain exec_config
+    assert "exec_config" in response.json
+    exec_config = response.json["exec_config"]
+
+    # AND the exec_config should have workflow_raw_data
+    assert "workflow_raw_data" in exec_config
+    nodes = exec_config["workflow_raw_data"]["nodes"]
+
+    # AND we should find the workflow nodes
+    node_labels = {node["data"]["label"] for node in nodes}
+    expected_nodes = {"Templating Node", "Final Output", "Entrypoint Node"}
+
+    # AND at least some of the expected nodes should be present
+    assert not DeepDiff(node_labels, expected_nodes, ignore_order=True)

{vellum_workflow_server-1.0.6 → vellum_workflow_server-1.0.7}/src/workflow_server/api/workflow_view.py

@@ -7,6 +7,8 @@ from multiprocessing import Process, Queue, set_start_method
 import os
 import pkgutil
 from queue import Empty
+import random
+import string
 import sys
 from threading import Event as ThreadingEvent
 import time
@@ -18,6 +20,8 @@ from flask import Blueprint, Response, current_app as app, has_request_context,
 from pydantic import ValidationError
 from vellum_ee.workflows.display.nodes.get_node_display_class import get_node_display_class
 from vellum_ee.workflows.display.types import WorkflowDisplayContext
+from vellum_ee.workflows.display.workflows import BaseWorkflowDisplay
+from vellum_ee.workflows.server.virtual_file_loader import VirtualFileFinder
 
 from vellum.workflows.exceptions import WorkflowInitializationException
 from vellum.workflows.nodes import BaseNode
@@ -410,6 +414,47 @@ def stream_node_route() -> Response:
     return resp
 
 
+@bp.route("/serialize", methods=["POST"])
+def serialize_route() -> Response:
+    data = request.get_json()
+
+    files = data.get("files", {})
+
+    if not files:
+        return Response(
+            json.dumps({"detail": "No files received"}),
+            status=400,
+            content_type="application/json",
+        )
+
+    try:
+        # Generate a unique namespace for this serialization request
+        namespace = "".join(random.choice(string.ascii_letters + string.digits) for i in range(14))
+        virtual_finder = VirtualFileFinder(files, namespace)
+
+        try:
+            sys.meta_path.append(virtual_finder)
+            exec_config = BaseWorkflowDisplay.serialize_module(namespace).exec_config
+
+            return Response(
+                json.dumps({"exec_config": exec_config}),
+                status=200,
+                content_type="application/json",
+            )
+
+        finally:
+            if virtual_finder in sys.meta_path:
+                sys.meta_path.remove(virtual_finder)
+
+    except Exception as e:
+        logger.exception(f"Error during serialization: {str(e)}")
+        return Response(
+            json.dumps({"detail": f"Serialization failed: {str(e)}"}),
+            status=500,
+            content_type="application/json",
+        )
+
+
 @bp.route("/version", methods=["GET"])
 def get_version_route() -> tuple[dict, int]:
     resp = get_version()