vellum-workflow-server 1.7.13__tar.gz → 1.7.14__tar.gz

{vellum_workflow_server-1.7.13 → vellum_workflow_server-1.7.14}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: vellum-workflow-server
-Version: 1.7.13
+Version: 1.7.14
 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.7.13)
+Requires-Dist: vellum-ai (==1.7.14)
 Description-Content-Type: text/markdown
 
 # Vellum Workflow Runner Server

{vellum_workflow_server-1.7.13 → vellum_workflow_server-1.7.14}/pyproject.toml

@@ -3,7 +3,7 @@ name = "vellum-workflow-server"
 
 [tool.poetry]
 name = "vellum-workflow-server"
-version = "1.7.13"
+version = "1.7.14"
 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.7.13"
+vellum-ai = "1.7.14"
 python-dotenv = "1.0.1"
 retrying = "1.3.4"
 sentry-sdk = {extras = ["flask"], version = "2.20.0"}

vellum_workflow_server-1.7.14/src/workflow_server/logging_config.py

@@ -0,0 +1,39 @@
+from datetime import datetime
+import json
+import logging
+
+
+class GCPJsonFormatter(logging.Formatter):
+    """
+    Custom JSON formatter for Google Cloud Platform logging.
+
+    Outputs logs in JSON format with a 'severity' field that GCP Cloud Logging
+    can properly parse. This ensures INFO logs show up as INFO in GCP instead of ERROR.
+
+    See: https://cloud.google.com/logging/docs/structured-logging
+    """
+
+    SEVERITY_MAP = {
+        "DEBUG": "DEBUG",
+        "INFO": "INFO",
+        "WARNING": "WARNING",
+        "ERROR": "ERROR",
+        "CRITICAL": "CRITICAL",
+    }
+
+    def format(self, record: logging.LogRecord) -> str:
+        log_obj = {
+            "severity": self.SEVERITY_MAP.get(record.levelname, "DEFAULT"),
+            "message": record.getMessage(),
+            "timestamp": datetime.utcfromtimestamp(record.created).isoformat() + "Z",
+            "logging.googleapis.com/sourceLocation": {
+                "file": record.pathname,
+                "line": record.lineno,
+                "function": record.funcName,
+            },
+        }
+
+        if record.exc_info:
+            log_obj["exception"] = self.formatException(record.exc_info)
+
+        return json.dumps(log_obj)

{vellum_workflow_server-1.7.13 → vellum_workflow_server-1.7.14}/src/workflow_server/server.py

@@ -7,6 +7,7 @@ from workflow_server.api.auth_middleware import AuthMiddleware
 from workflow_server.api.healthz_view import bp as healthz_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
 from workflow_server.utils.sentry import init_sentry
 from workflow_server.utils.utils import get_version
 
@@ -14,11 +15,18 @@ from workflow_server.utils.utils import get_version
 # enable_log_proxy()
 
 logger = logging.getLogger(__name__)
-logging.basicConfig(
-    level=logging.INFO,
-    format="%(asctime)s.%(msecs)03d %(levelname)s %(module)s - %(funcName)s: %(message)s",
-    datefmt="%Y-%m-%d %H:%M:%S",
-)
+
+if is_development():
+    logging.basicConfig(
+        level=logging.INFO,
+        format="%(asctime)s.%(msecs)03d %(levelname)s %(module)s - %(funcName)s: %(message)s",
+        datefmt="%Y-%m-%d %H:%M:%S",
+    )
+else:
+    handler = logging.StreamHandler()
+    handler.setFormatter(GCPJsonFormatter())
+    logging.root.addHandler(handler)
+    logging.root.setLevel(logging.INFO)
 
 
 def create_app() -> Flask: