vellum-workflow-server 1.6.4__py3-none-any.whl → 1.6.4.post2__py3-none-any.whl

{vellum_workflow_server-1.6.4.dist-info → vellum_workflow_server-1.6.4.post2.dist-info}/METADATA

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

{vellum_workflow_server-1.6.4.dist-info → vellum_workflow_server-1.6.4.post2.dist-info}/RECORD

@@ -14,13 +14,13 @@ workflow_server/core/cancel_workflow.py,sha256=Ffkc3mzmrdMEUcD-sHfEhX4IwVrka-E--
 workflow_server/core/events.py,sha256=24MA66DVQuaLJJcZrS8IL1Zq4Ohi9CoouKZ5VgoH3Cs,1402
 workflow_server/core/executor.py,sha256=Bw6luelJs84-_fBIG9VvxYFxeha4DmwxDpslVeqr2I8,18411
 workflow_server/core/utils.py,sha256=si0NB4Suurc-mn8NYdn59xM9CkPrfOP1aWEVrZvifDI,1929
-workflow_server/core/workflow_executor_context.py,sha256=u6tK6YgKlMrFXEYNBgWF1FMJ9NmN9HgOsPvptoA4b_c,4517
+workflow_server/core/workflow_executor_context.py,sha256=rSjgkWenRGpFOwioRXNnUhGnEI7isOB-Qk7Tw-2JwnM,4521
 workflow_server/server.py,sha256=QBU12AaAfAgLqfCDBd24qIJl_mbheiq0-hfcWV7rZM4,1234
-workflow_server/start.py,sha256=dsGg9998nki_avrow3m_bRnqSTp90wlrEsdjoP0SbrQ,2356
+workflow_server/start.py,sha256=WHZEsLen8mtGX-wF3xUbB91_NpC_GBRtOEi87OR1rv4,2360
 workflow_server/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 workflow_server/utils/exit_handler.py,sha256=_FacDVi4zc3bfTA3D2mJsISePlJ8jpLrnGVo5-xZQFs,743
 workflow_server/utils/log_proxy.py,sha256=nugi6fOgAYKX2X9DIc39TG366rsmmDUPoEtG3gzma_Y,3088
-workflow_server/utils/oom_killer.py,sha256=AprKFXC_wT3lQZcKDxU5O6dtJwi6meRxjo7nhQtQ8T0,2955
+workflow_server/utils/oom_killer.py,sha256=dzaqSzi0jQ3MvALwwiYIO9r6VWLa5Ln9AY6l11WEexo,3050
 workflow_server/utils/sentry.py,sha256=pqx3X_4W3yOzmz8QMJYUEi39skIKWtrTN5nyFhaPkbk,1597
 workflow_server/utils/system_utils.py,sha256=3jNv113zRkKJ0928i2Vm6TqFHrDulteQu1kjseP2B0Y,3271
 workflow_server/utils/tests/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -28,7 +28,7 @@ workflow_server/utils/tests/test_sentry_integration.py,sha256=LGmWiaLhFrx-jslrRj
 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=ZPoM1Suhid22dpB8oEFLux8wx-9iyzmSfWuYxSCrgWk,4774
-vellum_workflow_server-1.6.4.dist-info/METADATA,sha256=kHt8gHgq8Sb30IZ1kYhn_g6bDX-ZJ7VXtgkApq_ETm8,2267
-vellum_workflow_server-1.6.4.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-vellum_workflow_server-1.6.4.dist-info/entry_points.txt,sha256=uB_0yPkr7YV6RhEXzvFReUM8P4OQBlVXD6TN6eb9-oc,277
-vellum_workflow_server-1.6.4.dist-info/RECORD,,
+vellum_workflow_server-1.6.4.post2.dist-info/METADATA,sha256=NQOzPqJUMupffF4wb8lRMqZaBK7dtiAW4WBZbmM8Ccg,2273
+vellum_workflow_server-1.6.4.post2.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+vellum_workflow_server-1.6.4.post2.dist-info/entry_points.txt,sha256=uB_0yPkr7YV6RhEXzvFReUM8P4OQBlVXD6TN6eb9-oc,277
+vellum_workflow_server-1.6.4.post2.dist-info/RECORD,,

workflow_server/core/workflow_executor_context.py

@@ -15,7 +15,7 @@ from vellum.workflows.context import ExecutionContext
 from workflow_server.config import IS_VPC, VELLUM_API_URL_HOST, VELLUM_API_URL_PORT
 from workflow_server.utils.utils import convert_json_inputs_to_vellum
 
-DEFAULT_TIMEOUT_SECONDS = int(os.getenv("REVISION_TIMEOUT_SECONDS", 1800))
+DEFAULT_TIMEOUT_SECONDS = int(os.getenv("MAX_WORKFLOW_RUNTIME_SECONDS", 1800))
 
 
 def create_vellum_client(

workflow_server/start.py

@@ -44,7 +44,7 @@ def start() -> None:
         start_oom_killer_worker()
     init_signal_handlers()
 
-    max_workflow_runtime_seconds = int(os.getenv("GUNICORN_TIMEOUT", os.getenv("REVISION_TIMEOUT_SECONDS", 1800)))
+    max_workflow_runtime_seconds = int(os.getenv("GUNICORN_TIMEOUT", os.getenv("MAX_WORKFLOW_RUNTIME_SECONDS", 1800)))
 
     # gevent didn't play nice with pebble and processes so just gthread here
     options = {

workflow_server/utils/oom_killer.py

@@ -74,8 +74,12 @@ class OomKillerThread(Thread):
                 self._kill_switch.set()
                 active_span_ids = get_active_span_ids()
                 logger.error(
-                    f"Workflow server OOM killed, memory: {memory_mb}MB, Process Count: {get_active_process_count()}",
-                    extra={"active_span_ids": active_span_ids},
+                    "Workflow server OOM killed",
+                    extra={
+                        "active_span_ids": active_span_ids,
+                        "memory_mb": memory_mb,
+                        "process_count": get_active_process_count(),
+                    },
                 )
                 # Give time for the threads to get our kill switch
                 sleep(_KILL_GRACE_PERIOD)