vellum-workflow-server 1.8.6.post2__tar.gz → 1.8.6.post4__tar.gz

{vellum_workflow_server-1.8.6.post2 → vellum_workflow_server-1.8.6.post4}/PKG-INFO

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

{vellum_workflow_server-1.8.6.post2 → vellum_workflow_server-1.8.6.post4}/pyproject.toml

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

{vellum_workflow_server-1.8.6.post2 → vellum_workflow_server-1.8.6.post4}/src/workflow_server/api/auth_middleware.py

@@ -5,7 +5,7 @@ from flask import Flask, Request, Response
 import jwt
 from jwt import ExpiredSignatureError
 
-from workflow_server.config import IS_VPC, NAMESPACE, VEMBDA_PUBLIC_KEY, is_development
+from workflow_server.config import IS_ASYNC_MODE, IS_VPC, NAMESPACE, VEMBDA_PUBLIC_KEY, is_development
 
 
 class AuthMiddleware:
@@ -15,7 +15,7 @@ class AuthMiddleware:
     def __call__(self, environ: Dict[str, Any], start_response: Any) -> Any:
         try:
             request = Request(environ)
-            if not request.path.startswith("/healthz") and not is_development() and not IS_VPC:
+            if not request.path.startswith("/healthz") and not is_development() and not IS_VPC and not IS_ASYNC_MODE:
                 token = request.headers.get("X-Vembda-Signature")
                 if token:
                     decoded = jwt.decode(token, VEMBDA_PUBLIC_KEY, algorithms=["RS256"])

{vellum_workflow_server-1.8.6.post2 → vellum_workflow_server-1.8.6.post4}/src/workflow_server/api/tests/test_workflow_view.py

@@ -537,6 +537,57 @@ def test_serialize_route__with_invalid_workspace_api_key():
     assert "exec_config" in response.json
 
 
+def test_serialize_route__with_is_new_server_header():
+    """
+    Tests that the serialize route returns the is_new_server header.
+    """
+    # GIVEN a Flask application
+    flask_app = create_app()
+
+    workflow_files = {
+        "__init__.py": "",
+        "workflow.py": (
+            "from vellum.workflows import BaseWorkflow\n\n"
+            "class Workflow(BaseWorkflow):\n"
+            "    class Outputs(BaseWorkflow.Outputs):\n"
+            "        foo = 'hello'\n"
+        ),
+    }
+
+    # WHEN we make a request with is_new_server=True
+    with flask_app.test_client() as test_client:
+        response = test_client.post("/workflow/serialize", json={"files": workflow_files, "is_new_server": True})
+
+    # THEN we should get a successful response
+    assert response.status_code == 200
+
+    # AND the response should contain the is_new_server header set to true
+    assert "X-Vellum-Is-New-Server" in response.headers
+    assert response.headers["X-Vellum-Is-New-Server"] == "true"
+
+    # WHEN we make a request with is_new_server=False
+    with flask_app.test_client() as test_client:
+        response = test_client.post("/workflow/serialize", json={"files": workflow_files, "is_new_server": False})
+
+    # THEN we should get a successful response
+    assert response.status_code == 200
+
+    # AND the response should contain the is_new_server header set to false
+    assert "X-Vellum-Is-New-Server" in response.headers
+    assert response.headers["X-Vellum-Is-New-Server"] == "false"
+
+    # WHEN we make a request without is_new_server
+    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 the is_new_server header set to false (default)
+    assert "X-Vellum-Is-New-Server" in response.headers
+    assert response.headers["X-Vellum-Is-New-Server"] == "false"
+
+
 def test_stream_node_route__with_node_id():
     """
     Tests that the stream-node endpoint works with node_id.

{vellum_workflow_server-1.8.6.post2 → vellum_workflow_server-1.8.6.post4}/src/workflow_server/api/workflow_view.py

@@ -435,6 +435,7 @@ def serialize_route() -> Response:
 
     files = data.get("files", {})
     workspace_api_key = data.get("workspace_api_key")
+    is_new_server = data.get("is_new_server", False)
 
     if not files:
         return Response(
@@ -448,6 +449,11 @@ def serialize_route() -> Response:
     # Generate a unique namespace for this serialization request
     namespace = get_random_namespace()
     virtual_finder = VirtualFileFinder(files, namespace)
+
+    headers = {
+        "X-Vellum-Is-New-Server": str(is_new_server).lower(),
+    }
+
     try:
         sys.meta_path.append(virtual_finder)
         result = BaseWorkflowDisplay.serialize_module(namespace, client=client, dry_run=True)
@@ -456,6 +462,7 @@ def serialize_route() -> Response:
             json.dumps(result.model_dump()),
             status=200,
             content_type="application/json",
+            headers=headers,
         )
 
     except WorkflowInitializationException as e:
@@ -465,6 +472,7 @@ def serialize_route() -> Response:
             json.dumps({"detail": error_message}),
             status=400,
             content_type="application/json",
+            headers=headers,
         )
 
     except Exception as e:
@@ -473,6 +481,7 @@ def serialize_route() -> Response:
             json.dumps({"detail": f"Serialization failed: {str(e)}"}),
             status=500,
             content_type="application/json",
+            headers=headers,
         )
 
     finally:

{vellum_workflow_server-1.8.6.post2 → vellum_workflow_server-1.8.6.post4}/src/workflow_server/config.py

@@ -42,6 +42,8 @@ LOCAL_WORKFLOW_MODULE = os.getenv("LOCAL_WORKFLOW_MODULE")
 # The deployment name to match against when using local mode so you can still run your normal workflow
 LOCAL_DEPLOYMENT = os.getenv("LOCAL_DEPLOYMENT")
 
+IS_ASYNC_MODE = os.getenv("IS_ASYNC_MODE", "false").lower() == "true"
+
 
 def is_development() -> bool:
     return os.getenv("FLASK_ENV", "local") == "local"