vellum-workflow-server 0.14.70.post81__tar.gz → 1.9.7__tar.gz

{vellum_workflow_server-0.14.70.post81 → vellum_workflow_server-1.9.7}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: vellum-workflow-server
-Version: 0.14.70.post81
+Version: 1.9.7
 Summary: 
 License: AGPL
 Requires-Python: >=3.9.0,<4
@@ -27,8 +27,9 @@ Requires-Dist: orderly-set (==5.2.2)
 Requires-Dist: pebble (==5.0.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 (==0.14.70)
+Requires-Dist: vellum-ai (==1.9.7)
 Description-Content-Type: text/markdown
 
 # Vellum Workflow Runner Server

{vellum_workflow_server-0.14.70.post81 → vellum_workflow_server-1.9.7}/pyproject.toml

@@ -1,6 +1,9 @@
+[project]
+name = "vellum-workflow-server"
+
 [tool.poetry]
 name = "vellum-workflow-server"
-version = "0.14.70.post81"
+version = "1.9.7"
 description = ""
 readme = "README.md"
 authors = []
@@ -29,6 +32,7 @@ packages = [
 
 [tool.poetry.group.dev.dependencies]
 requests-mock = "^1.12.1"
+pytest-mock = "^3.14.1"
 
 [project.urls]
 Repository = 'https://github.com/vellum-ai/vembda-service'
@@ -41,8 +45,9 @@ flask = "2.3.3"
 orderly-set = "5.2.2"
 pebble = "5.0.7"
 gunicorn = "23.0.0"
-vellum-ai = "0.14.70"
+vellum-ai = "1.9.7"
 python-dotenv = "1.0.1"
+retrying = "1.3.4"
 sentry-sdk = {extras = ["flask"], version = "2.20.0"}
 
 [tool.poetry.dev-dependencies]

{vellum_workflow_server-0.14.70.post81 → vellum_workflow_server-1.9.7}/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.9.7/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