vellum-workflow-server 1.7.12__py3-none-any.whl → 1.7.13.post1__py3-none-any.whl

{vellum_workflow_server-1.7.12.dist-info → vellum_workflow_server-1.7.13.post1.dist-info}/METADATA

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

{vellum_workflow_server-1.7.12.dist-info → vellum_workflow_server-1.7.13.post1.dist-info}/RECORD

@@ -15,7 +15,8 @@ workflow_server/core/events.py,sha256=24MA66DVQuaLJJcZrS8IL1Zq4Ohi9CoouKZ5VgoH3C
 workflow_server/core/executor.py,sha256=rMpVgP3PqMnjHVYeuTdHvbYGwJW5C9O5TFGW926jFM0,16106
 workflow_server/core/utils.py,sha256=si0NB4Suurc-mn8NYdn59xM9CkPrfOP1aWEVrZvifDI,1929
 workflow_server/core/workflow_executor_context.py,sha256=Q2R0T2KkYZ1z52v8erDMysJfxSODbzPhDxBxX--k4Zw,3202
-workflow_server/server.py,sha256=lhHPmK1PhRZd6eCkj1C0acK3YwaApZgoPHghMChw0fc,1461
+workflow_server/logging_config.py,sha256=Hvx1t8uhqMMinl-5qcef7ufUvzs6x14VRnCb7YZxEAg,1206
+workflow_server/server.py,sha256=UnldeFS6iINBc-Kxl8LQcdAB2SGV6u-Xi5gy5EN466E,1729
 workflow_server/start.py,sha256=xSIobowtSLoZI86bbMkmEw3pqJHQaFdDyNffk4kGYL8,2544
 workflow_server/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 workflow_server/utils/exit_handler.py,sha256=_FacDVi4zc3bfTA3D2mJsISePlJ8jpLrnGVo5-xZQFs,743
@@ -28,7 +29,7 @@ workflow_server/utils/tests/test_sentry_integration.py,sha256=14PfuW8AaQNNtqLmBs
 workflow_server/utils/tests/test_system_utils.py,sha256=_4GwXvVvU5BrATxUEWwQIPg0bzQXMWBtiBmjP8MTxJM,4314
 workflow_server/utils/tests/test_utils.py,sha256=0Nq6du8o-iBtTrip9_wgHES53JSiJbVdSXaBnPobw3s,6930
 workflow_server/utils/utils.py,sha256=m7iMJtor5SQLWu7jlJw-X5Q3nmbq69BCxTMv6qnFYrA,4835
-vellum_workflow_server-1.7.12.dist-info/METADATA,sha256=yZT9CluUxLUwMmq-Gs3R8WVEXX8PXL99urVqdXgonfw,2269
-vellum_workflow_server-1.7.12.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-vellum_workflow_server-1.7.12.dist-info/entry_points.txt,sha256=uB_0yPkr7YV6RhEXzvFReUM8P4OQBlVXD6TN6eb9-oc,277
-vellum_workflow_server-1.7.12.dist-info/RECORD,,
+vellum_workflow_server-1.7.13.post1.dist-info/METADATA,sha256=N9S9IyIcQAncCjAxdj9BicVIvd7HyANVvVE1bO4eO8k,2275
+vellum_workflow_server-1.7.13.post1.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+vellum_workflow_server-1.7.13.post1.dist-info/entry_points.txt,sha256=uB_0yPkr7YV6RhEXzvFReUM8P4OQBlVXD6TN6eb9-oc,277
+vellum_workflow_server-1.7.13.post1.dist-info/RECORD,,

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)

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: