OpenWebUI/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui synced 2025-06-26 18:26:48 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

feat: add ENABLE_WEBSOCKET_SUPPORT to force socket.io to ignore websocket upgrades

Browse Source
This commit is contained in:
Jun Siang Cheah
2024-09-08 12:00:36 +01:00
parent 1a6ce1d5d9
commit 698976add0
2 changed files with 15 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
backend/open_webui/apps/socket/main.py
View File

@@ -2,9 +2,17 @@ import asyncio
import socketio
from open_webui.apps.webui.models.users import Users
from open_webui.config import ENABLE_WEBSOCKET_SUPPORT
from open_webui.utils.utils import decode_token
sio = socketio.AsyncServer(cors_allowed_origins=[], async_mode="asgi")
sio = socketio.AsyncServer(
cors_allowed_origins=[],
async_mode="asgi",
transports=(
["polling", "websocket"] if ENABLE_WEBSOCKET_SUPPORT.value else ["polling"]
),
allow_upgrades=ENABLE_WEBSOCKET_SUPPORT.value,
)
app = socketio.ASGIApp(sio, socketio_path="/ws/socket.io")
# Dictionary to maintain the user pool