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

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

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

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

@@ -5,14 +5,14 @@ workflow_server/api/healthz_view.py,sha256=itiRvBDBXncrw8Kbbc73UZLwqMAhgHOR3uSre
 workflow_server/api/tests/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 workflow_server/api/tests/test_input_display_mapping.py,sha256=drBZqMudFyB5wgiUOcMgRXz7E7ge-Qgxbstw4E4f0zE,2211
 workflow_server/api/tests/test_workflow_view.py,sha256=2nscM_QsYPHkkTG8_JhNbE2LmGL5FQKXEtaLLjXouw0,14591
-workflow_server/api/tests/test_workflow_view_stream_workflow_route.py,sha256=wmeH8oGo0lKx3YzEJQ9nGjw7adqQD0EL7-xGLPgYWqc,24872
-workflow_server/api/workflow_view.py,sha256=AOHccJXkhzw-iXBo4Q_EaLRynTuNmeQZaQxB5ttgrY4,15434
+workflow_server/api/tests/test_workflow_view_stream_workflow_route.py,sha256=cUVF3tqV8Ay91YfFr2lT2oXONWRN823Nk5M3EOC8wE8,24133
+workflow_server/api/workflow_view.py,sha256=bvkDQ9h267hqWsHSElplg83NGHfoLuR8VaLlMh6zJ9s,15295
 workflow_server/code_exec_runner.py,sha256=XSs4w_b0vDUt0HqSakc26Gxx9aoG2wmwOo-JGVL5QJ4,2388
 workflow_server/config.py,sha256=K5Tavm7wiqCZt0RWWue7zzb8N6e8aWnFOTNlBqEJPcI,1330
 workflow_server/core/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 workflow_server/core/cancel_workflow.py,sha256=Ffkc3mzmrdMEUcD-sHfEhX4IwVrka-E--SxKA1dUfIU,2185
 workflow_server/core/events.py,sha256=24MA66DVQuaLJJcZrS8IL1Zq4Ohi9CoouKZ5VgoH3Cs,1402
-workflow_server/core/executor.py,sha256=TOnJ5FcmC8mWNdGvoyTwNlNlbhp6s6hk7Nc4M35jEe4,17148
+workflow_server/core/executor.py,sha256=j78gYOgaeJ5Z25nCefnPg4fUPdTLW0tcwaZeWQf0DP0,16588
 workflow_server/core/utils.py,sha256=lgzxkAEjEXPxGXXQlUYTYuCdHht-eDJJmHj5AhEb3_o,1500
 workflow_server/core/workflow_executor_context.py,sha256=a-v48GJbOWUh4JIf_bNwDX-BvfKkg4xwRSPEyRVQmp4,1373
 workflow_server/server.py,sha256=QBU12AaAfAgLqfCDBd24qIJl_mbheiq0-hfcWV7rZM4,1234
@@ -27,7 +27,7 @@ workflow_server/utils/tests/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMp
 workflow_server/utils/tests/test_system_utils.py,sha256=MdBxI9gxUOpR_JBAHpEz6dGFY6JjxhMSM2oExpqFvNA,4314
 workflow_server/utils/tests/test_utils.py,sha256=qwK5Rmy3RQyjtlUrYAuGuDlBeRzZKsf1yS-y2IpUizQ,6452
 workflow_server/utils/utils.py,sha256=Wqqn-1l2ugkGgy5paWWdt0AVxAyPMQCYcnRSSOMjXlA,4355
-vellum_workflow_server-0.14.73.dist-info/METADATA,sha256=d7G7Lir-2rbsAhDvGMUDbX9oV-88h_ViurMycJ_PK_Q,2237
-vellum_workflow_server-0.14.73.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-vellum_workflow_server-0.14.73.dist-info/entry_points.txt,sha256=uB_0yPkr7YV6RhEXzvFReUM8P4OQBlVXD6TN6eb9-oc,277
-vellum_workflow_server-0.14.73.dist-info/RECORD,,
+vellum_workflow_server-0.14.73.post2.dist-info/METADATA,sha256=7J8qwzytgU5lH36SRdvkqMfN35req2uxbKYeZJhVCQI,2243
+vellum_workflow_server-0.14.73.post2.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+vellum_workflow_server-0.14.73.post2.dist-info/entry_points.txt,sha256=uB_0yPkr7YV6RhEXzvFReUM8P4OQBlVXD6TN6eb9-oc,277
+vellum_workflow_server-0.14.73.post2.dist-info/RECORD,,

workflow_server/api/tests/test_workflow_view_stream_workflow_route.py

@@ -10,6 +10,7 @@ import requests_mock
 
 from workflow_server.code_exec_runner import run_code_exec_stream
 from workflow_server.server import create_app
+from workflow_server.utils.system_utils import get_active_process_count
 
 
 def flask_stream(request_body: dict) -> tuple[int, list]:
@@ -775,44 +776,15 @@ class Workflow(BaseWorkflow):
     # WHEN we call the stream route
     status_code, events = flask_stream(request_body)
 
-    # THEN we get a 200 response
-    assert status_code == 200, events
+    # THEN we get a 429 response
+    assert status_code == 429, events
 
-    # THEN we get the expected events
+    # AND we get a simple JSON error response
+    assert len(events) == 1
     assert events[0] == {
-        "id": mock.ANY,
-        "trace_id": mock.ANY,
-        "span_id": str(span_id),
-        "timestamp": mock.ANY,
-        "api_version": "2024-10-25",
-        "parent": None,
-        "name": "vembda.execution.initiated",
-        "body": {
-            "sdk_version": version("vellum-ai"),
-            "server_version": "local",
-        },
+        "detail": f"Workflow server concurrent request rate exceeded. Process count: {get_active_process_count()}"
     }
 
-    # AND we get a vembda.execution.fulfilled event with error
-    assert events[1] == {
-        "id": mock.ANY,
-        "trace_id": events[0]["trace_id"],
-        "span_id": str(span_id),
-        "timestamp": mock.ANY,
-        "api_version": "2024-10-25",
-        "parent": None,
-        "name": "vembda.execution.fulfilled",
-        "body": {
-            "log": "",
-            "exit_code": -1,
-            "stderr": "Workflow server concurrent request rate exceeded. Process count: 0",
-            "container_overhead_latency": mock.ANY,
-            "timed_out": False,
-        },
-    }
-
-    assert len(events) == 2
-
 
 def test_stream_workflow_route__with_environment_variables(both_stream_types):
     # GIVEN a valid request body with environment variables

workflow_server/api/workflow_view.py

@@ -97,16 +97,14 @@ def stream_workflow_route() -> Response:
     # if we detect a memory problem just exit us early
     if not wait_for_available_process():
         return Response(
-            stream_with_context(
-                startup_error_generator(
-                    context=context,
-                    message=f"Workflow server concurrent request rate exceeded. "
-                    f"Process count: {get_active_process_count()}",
-                    vembda_initiated_event=vembda_initiated_event,
-                )
+            json.dumps(
+                {
+                    "detail": f"Workflow server concurrent request rate exceeded. "
+                    f"Process count: {get_active_process_count()}"
+                }
             ),
-            status=200,
-            content_type="application/x-ndjson",
+            status=429,
+            content_type="application/json",
             headers=headers,
         )
 

workflow_server/core/executor.py

@@ -21,7 +21,6 @@ from vellum_ee.workflows.server.virtual_file_loader import VirtualFileFinder
 
 from vellum import Vellum, VellumEnvironment
 from vellum.workflows import BaseWorkflow
-from vellum.workflows.errors import WorkflowError, WorkflowErrorCode
 from vellum.workflows.events.types import BaseEvent
 from vellum.workflows.events.workflow import WorkflowEventDisplayContext
 from vellum.workflows.exceptions import WorkflowInitializationException
@@ -207,13 +206,6 @@ def stream_workflow(
                     if event.name == "workflow.execution.initiated":
                         event.body.display_context = display_context
 
-                if event.name.endswith("rejected") and event.body.error.code.name == "INTERNAL_ERROR":  # type: ignore
-                    sentry_sdk.set_tag("vellum_trace_id", str(executor_context.trace_id))
-                    logger.exception(event.body.error.message)  # type: ignore
-                    event.body.error = WorkflowError(  # type: ignore
-                        code=WorkflowErrorCode.INTERNAL_ERROR.value, message="Internal Error"  # type: ignore
-                    )
-
                 yield _dump_event(
                     event=event,
                     executor_context=executor_context,