Address blocking comment: move bytes validation to VellumJsonEncoder and remove per-chunk overhead

Co-Authored-By: [email protected] <[email protected]>

Author: devin-ai-integration[bot]

src/vellum/utils/json_encoder.py

@@ -48,6 +48,12 @@ def default(self, obj: Any) -> Any:
         if hasattr(obj, "__vellum_encode__") and callable(getattr(obj, "__vellum_encode__")):
             return obj.__vellum_encode__()
 
+        if isinstance(obj, (bytes, bytearray)):
+            raise TypeError(
+                "bytes are not JSON-serializable; convert to text (e.g., .decode('utf-8')) "
+                "or base64-encode before returning from a node output"
+            )
+
         if isinstance(obj, UUID):
             return str(obj)
 

src/vellum/workflows/runner/runner.py

@@ -572,16 +572,6 @@ def initiate_node_streaming_output(
 
                 with execution_context(parent_context=updated_parent_context, trace_id=execution.trace_id):
                     for output in node_run_response:
-                        try:
-                            default_serializer(output)
-                        except (TypeError, ValueError) as exc:
-                            raise NodeException(
-                                message=(
-                                    f"Node {node.__class__.__name__} produced output: "
-                                    f"'{output.name}' that could not be serialized to JSON: {exc}"
-                                ),
-                                code=WorkflowErrorCode.INVALID_OUTPUTS,
-                            ) from exc
                         invoked_ports = output > ports
                         if output.is_initiated:
                             yield from initiate_node_streaming_output(output)