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

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

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

{vellum_workflow_server-1.8.6.post1 → vellum_workflow_server-1.8.6.post2}/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.post2"
 description = ""
 readme = "README.md"
 authors = []

vellum_workflow_server-1.8.6.post2/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.post2}/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())