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

{vellum_workflow_server-1.7.13 → vellum_workflow_server-1.7.13.post1}/PKG-INFO

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

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

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

vellum_workflow_server-1.7.13.post1/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.13.post1}/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: