vellum-workflow-server 1.10.7__py3-none-any.whl → 1.11.15__py3-none-any.whl

{vellum_workflow_server-1.10.7.dist-info → vellum_workflow_server-1.11.15.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: vellum-workflow-server
-Version: 1.10.7
+Version: 1.11.15
 Summary: 
 License: AGPL
 Requires-Python: >=3.9.0,<4
@@ -24,12 +24,13 @@ Requires-Dist: cryptography (==43.0.3)
 Requires-Dist: flask (==2.3.3)
 Requires-Dist: gunicorn (==23.0.0)
 Requires-Dist: orderly-set (==5.2.2)
+Requires-Dist: orjson (==3.11.4)
 Requires-Dist: pebble (==5.0.7)
 Requires-Dist: pyjwt (==2.10.0)
 Requires-Dist: python-dotenv (==1.2.1)
 Requires-Dist: retrying (==1.3.4)
 Requires-Dist: sentry-sdk[flask] (==2.20.0)
-Requires-Dist: vellum-ai (==1.10.7)
+Requires-Dist: vellum-ai (==1.11.15)
 Description-Content-Type: text/markdown
 
 # Vellum Workflow Runner Server

{vellum_workflow_server-1.10.7.dist-info → vellum_workflow_server-1.11.15.dist-info}/RECORD

@@ -5,16 +5,16 @@ workflow_server/api/healthz_view.py,sha256=itiRvBDBXncrw8Kbbc73UZLwqMAhgHOR3uSre
 workflow_server/api/status_view.py,sha256=Jah8dBAVL4uOcRfsjKAOyfVONFyk9HQjXeRfjcIqhmA,514
 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=B6B8mCirt3FvpPKRP_AyzPJ199k_gwLzAcQuWRkzEfA,32343
+workflow_server/api/tests/test_workflow_view.py,sha256=I2sd11ptKDqbylzB9rKqkMXeZoh8ttad3zIhNus86vk,32491
 workflow_server/api/tests/test_workflow_view_async_exec.py,sha256=eP_H2xI9SRfJdoJ6HPeynQecnxR50I_8aDCooF-YzIw,11952
-workflow_server/api/tests/test_workflow_view_stream_workflow_route.py,sha256=16ZxP_nuAIP1Cg4-4z6EQttn07SRY1GkVq2m53z6XaE,42389
-workflow_server/api/workflow_view.py,sha256=RiRO0Z_gCIbdcG9XX_PcB9j8Qx5K_2dXxxtkib6fezY,24601
-workflow_server/code_exec_runner.py,sha256=DLNNrinCRbnkSvlqVvSZ1wv_etI7r_kKAXNPGMj3jBk,2196
+workflow_server/api/tests/test_workflow_view_stream_workflow_route.py,sha256=QilhiYv3_iaF-xvbmqUqqQqqVIZC4KgBe1u8Ku1q52s,42511
+workflow_server/api/workflow_view.py,sha256=60RWK2VFmmeLpHCGwf3DdBBx2_GP5KIRxE6rAEwG8Jw,24774
+workflow_server/code_exec_runner.py,sha256=vJlCQ8FkcG8RfCZ34Ea2Xt6J7dNkU5EqA-KxRkbVOeo,2219
 workflow_server/config.py,sha256=I4hfTsjIbHxoSKylPCjKnrysPV0jO5nfRKwpKvEcfAE,2193
 workflow_server/core/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 workflow_server/core/cancel_workflow.py,sha256=QcEeYUIrxq4pub-z9BlGi5fLI3gVRml-56rMCW7j5Hc,2212
 workflow_server/core/events.py,sha256=24MA66DVQuaLJJcZrS8IL1Zq4Ohi9CoouKZ5VgoH3Cs,1402
-workflow_server/core/executor.py,sha256=xbySFdb9KHoqFDfiKMR77fViFVo3XEQ5ER54C1PlS8c,16948
+workflow_server/core/executor.py,sha256=7mJDuP8uedCAygNorp2Vscy6w7xHHqNZ_jxJXkJZzeY,17112
 workflow_server/core/utils.py,sha256=mecVPqQkthrC4mpop3r8J3IWnBmKbDgqfCrSagyzVEg,2021
 workflow_server/core/workflow_executor_context.py,sha256=8faOdpU4cBeIbmOvg9VzD3eS5i_PKcH7tyNGzx_rehg,3899
 workflow_server/logging_config.py,sha256=Hvx1t8uhqMMinl-5qcef7ufUvzs6x14VRnCb7YZxEAg,1206
@@ -31,7 +31,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.10.7.dist-info/METADATA,sha256=fmGtKTAbZbiiO93Cw18heCPbkQUjjicY2bEXEg2TB5U,2269
-vellum_workflow_server-1.10.7.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-vellum_workflow_server-1.10.7.dist-info/entry_points.txt,sha256=uB_0yPkr7YV6RhEXzvFReUM8P4OQBlVXD6TN6eb9-oc,277
-vellum_workflow_server-1.10.7.dist-info/RECORD,,
+vellum_workflow_server-1.11.15.dist-info/METADATA,sha256=oIldsrXZvwKAboSlLYy1LRYuEir14fTww2xS7ii3I7I,2304
+vellum_workflow_server-1.11.15.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+vellum_workflow_server-1.11.15.dist-info/entry_points.txt,sha256=uB_0yPkr7YV6RhEXzvFReUM8P4OQBlVXD6TN6eb9-oc,277
+vellum_workflow_server-1.11.15.dist-info/RECORD,,

workflow_server/api/tests/test_workflow_view.py

@@ -389,7 +389,15 @@ class MyAdditionNode(BaseNode):
         },
         "id": "2464b610-fb6d-495b-b17c-933ee147f19f",
         "label": "My Addition Node",
-        "outputs": [{"id": "f39d85c9-e7bf-45e1-bb67-f16225db0118", "name": "result", "type": "NUMBER", "value": None}],
+        "outputs": [
+            {
+                "id": "f39d85c9-e7bf-45e1-bb67-f16225db0118",
+                "name": "result",
+                "type": "NUMBER",
+                "value": None,
+                "schema": {"type": "integer"},
+            }
+        ],
         "ports": [{"id": "bc489295-cd8a-4aa2-88bb-34446374100d", "name": "default", "type": "DEFAULT"}],
         "trigger": {"id": "ff580cad-73d6-44fe-8f2c-4b8dc990ee70", "merge_behavior": "AWAIT_ATTRIBUTES"},
         "type": "GENERIC",

workflow_server/api/tests/test_workflow_view_stream_workflow_route.py

@@ -549,7 +549,10 @@ class Inputs(BaseInputs):
     # AND the third event should be workflow execution rejected
     assert events[2]["name"] == "workflow.execution.rejected"
     assert events[1]["span_id"] == events[2]["span_id"]
-    assert "Required input variables foo should have defined value" in events[2]["body"]["error"]["message"]
+    actual_error_message = events[2]["body"]["error"]["message"]
+    assert "Required input variables" in actual_error_message
+    assert "foo" in actual_error_message
+    assert "should have defined value" in actual_error_message
 
     # AND the fourth event should be vembda execution fulfilled
     assert events[3]["name"] == "vembda.execution.fulfilled"

workflow_server/api/workflow_view.py

@@ -15,6 +15,7 @@ from uuid import uuid4
 from typing import Any, Dict, Generator, Iterator, Optional, Union, cast
 
 from flask import Blueprint, Response, current_app as app, request, stream_with_context
+import orjson
 from pydantic import ValidationError
 from vellum_ee.workflows.display.nodes.get_node_display_class import get_node_display_class
 from vellum_ee.workflows.display.types import WorkflowDisplayContext
@@ -115,7 +116,7 @@ def stream_workflow_route() -> Response:
             for row in workflow_events:
                 yield "\n"
                 if isinstance(row, dict):
-                    dump = json.dumps(row)
+                    dump = orjson.dumps(row).decode("utf-8")
                     yield dump
                 else:
                     yield row
@@ -500,11 +501,11 @@ def stream_node_route() -> Response:
                 break
 
     def generator() -> Generator[str, None, None]:
-        yield json.dumps(vembda_initiated_event.model_dump(mode="json"))
+        yield orjson.dumps(vembda_initiated_event.model_dump(mode="json")).decode("utf-8")
 
         for row in node_events():
             yield "\n"
-            yield json.dumps(row)
+            yield orjson.dumps(row).decode("utf-8")
 
     headers = {
         "X-Vellum-SDK-Version": vembda_initiated_event.body.sdk_version,
@@ -530,11 +531,18 @@ def serialize_route() -> Response:
     is_new_server = data.get("is_new_server", False)
     module = data.get("module")
 
+    headers = {
+        "X-Vellum-Is-New-Server": str(is_new_server).lower(),
+    }
+
     if not files:
+        error_message = "No files received"
+        logger.warning(error_message)
         return Response(
-            json.dumps({"detail": "No files received"}),
+            json.dumps({"detail": error_message}),
             status=400,
             content_type="application/json",
+            headers=headers,
         )
 
     client = create_vellum_client(api_key=workspace_api_key)
@@ -543,10 +551,6 @@ def serialize_route() -> Response:
     namespace = get_random_namespace()
     virtual_finder = VirtualFileFinder(files, namespace, source_module=module)
 
-    headers = {
-        "X-Vellum-Is-New-Server": str(is_new_server).lower(),
-    }
-
     try:
         sys.meta_path.append(virtual_finder)
         result = BaseWorkflowDisplay.serialize_module(namespace, client=client, dry_run=True)

workflow_server/code_exec_runner.py

@@ -1,11 +1,12 @@
 from datetime import datetime
-import json
 import logging
 import os
 from threading import Event as ThreadingEvent
 from uuid import uuid4
 from typing import Optional
 
+import orjson
+
 from workflow_server.core.events import VembdaExecutionInitiatedBody, VembdaExecutionInitiatedEvent
 from workflow_server.core.executor import stream_workflow
 from workflow_server.core.utils import serialize_vembda_rejected_event
@@ -29,7 +30,7 @@ def run_code_exec_stream() -> None:
         split_input = input_raw.split("\n--vellum-input-stop--\n")
         input_json = split_input[0]
 
-        input_data = json.loads(input_json)
+        input_data = orjson.loads(input_json)
         context = WorkflowExecutorContext.model_validate(input_data)
 
         print("--vellum-output-start--")  # noqa: T201
@@ -53,7 +54,7 @@ def run_code_exec_stream() -> None:
             cancel_signal=ThreadingEvent(),
         )
         for line in stream_iterator:
-            print(f"{_EVENT_LINE}{json.dumps(line)}")  # noqa: T201
+            print(f"{_EVENT_LINE}{orjson.dumps(line).decode('utf-8')}")  # noqa: T201
     except Exception as e:
         logger.exception(e)
 

workflow_server/core/executor.py

@@ -1,6 +1,5 @@
 from datetime import datetime, timezone
 from io import StringIO
-import json
 import logging
 from multiprocessing import Process, Queue
 import os
@@ -13,7 +12,9 @@ from traceback import format_exc
 from uuid import UUID, uuid4
 from typing import Any, Callable, Generator, Iterator, Optional, Tuple
 
+import orjson
 from vellum_ee.workflows.display.utils.events import event_enricher
+from vellum_ee.workflows.display.utils.expressions import base_descriptor_validator
 from vellum_ee.workflows.server.virtual_file_loader import VirtualFileFinder
 
 from vellum.workflows import BaseWorkflow
@@ -104,7 +105,7 @@ def _stream_workflow_wrapper(
         span_id_emitted = True
 
         for event in stream_iterator:
-            queue.put(json.dumps(event))
+            queue.put(orjson.dumps(event).decode("utf-8"))
 
     except Exception as e:
         if not span_id_emitted:
@@ -177,6 +178,7 @@ def stream_workflow(
         node_output_mocks = MockNodeExecution.validate_all(
             executor_context.node_output_mocks,
             workflow.__class__,
+            descriptor_validator=base_descriptor_validator,
         )
 
         cancel_signal = cancel_signal or ThreadingEvent()