fix subprotocol for granian

2025-01-27 11:34:42 -08:00 · 2025-01-27 11:34:42 -08:00 · ed3955a59a
commit ed3955a59a
parent 709c6dedf2
1 changed files with 5 additions and 4 deletions
--- a/reflex/app.py
+++ b/reflex/app.py
@ -419,11 +419,12 @@ class App(MiddlewareMixin, LifespanMixin):
                        if (
                            message["type"] == "websocket.accept"
                            and protocol_key in headers
+                            and isinstance(
+                                (subprotocol := headers[protocol_key]), bytes
+                            )
                        ):
-                            message["headers"] = [
-                                *message.get("headers", []),
-                                (b"sec-websocket-protocol", headers[protocol_key]),
-                            ]
+                            message["subprotocol"] = subprotocol.decode()
+
                        return await original_send(message)

                    return await self.app(scope, receive, modified_send)