vellum-workflow-server 1.8.6.post1__tar.gz → 1.8.6.post3__tar.gz

{vellum_workflow_server-1.8.6.post1 → vellum_workflow_server-1.8.6.post3}/PKG-INFO

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

{vellum_workflow_server-1.8.6.post1 → vellum_workflow_server-1.8.6.post3}/pyproject.toml

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

vellum_workflow_server-1.8.6.post3/src/workflow_server/api/status_view.py

@@ -0,0 +1,19 @@
+from typing import Tuple
+
+from flask import Blueprint, Response, jsonify
+
+from workflow_server.config import CONCURRENCY
+from workflow_server.utils.system_utils import get_active_process_count
+
+bp = Blueprint("status", __name__)
+
+
+@bp.route("/is_available", methods=["GET"])
+def is_available() -> Tuple[Response, int]:
+    resp = jsonify(
+        available=get_active_process_count() < CONCURRENCY,
+        process_count=get_active_process_count(),
+        max_concurrency=CONCURRENCY,
+    )
+
+    return resp, 200

{vellum_workflow_server-1.8.6.post1 → vellum_workflow_server-1.8.6.post3}/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.post1 → vellum_workflow_server-1.8.6.post3}/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.post1 → vellum_workflow_server-1.8.6.post3}/src/workflow_server/server.py

@@ -5,6 +5,7 @@ from flask import Flask
 
 from workflow_server.api.auth_middleware import AuthMiddleware
 from workflow_server.api.healthz_view import bp as healthz_bp
+from workflow_server.api.status_view import bp as status_bp
 from workflow_server.api.workflow_view import bp as workflow_bp
 from workflow_server.config import is_development
 from workflow_server.logging_config import GCPJsonFormatter
@@ -48,6 +49,7 @@ def create_app() -> Flask:
 
     # Register blueprints
     app.register_blueprint(healthz_bp, url_prefix="/healthz")
+    app.register_blueprint(status_bp, url_prefix="/status")
     app.register_blueprint(workflow_bp, url_prefix="/workflow")
 
     logger.info(is_development())