increase max size for websocket

viser_proxy_manager.py

@@ -19,6 +19,7 @@ class ViserProxyManager:
             These ports are used only for internal communication and don't need to be publicly exposed.
         max_local_port: Maximum local port number to use for Viser servers. Defaults to 9000.
             These ports are used only for internal communication and don't need to be publicly exposed.
+        max_message_size: Maximum WebSocket message size in bytes. Defaults to 100MB.
     """
 
     def __init__(
@@ -26,9 +27,11 @@ class ViserProxyManager:
         app: FastAPI,
         min_local_port: int = 8000,
         max_local_port: int = 9000,
+        max_message_size: int = 100 * 1024 * 1024,  # 100MB default
     ) -> None:
         self._min_port = min_local_port
         self._max_port = max_local_port
+        self._max_message_size = max_message_size
         self._server_from_session_hash: dict[str, viser.ViserServer] = {}
         self._last_port = self._min_port - 1  # Track last port tried
 
@@ -95,8 +98,11 @@ class ViserProxyManager:
                 return
 
             try:
-                # Connect to the target WebSocket
-                async with websockets.connect(target_ws_url) as ws_target:
+                # Connect to the target WebSocket with increased message size
+                async with websockets.connect(
+                    target_ws_url,
+                    max_size=self._max_message_size,  # Set max message size for receiving
+                ) as ws_target:
                     # Create tasks for bidirectional communication
                     async def forward_to_target():
                         """Forward messages from the client to the target WebSocket."""