Timothy J. Baek 7 月之前
父節點
當前提交
657d443a3e
共有 1 個文件被更改,包括 10 次插入1 次删除
  1. 10 1
      backend/open_webui/apps/socket/main.py

+ 10 - 1
backend/open_webui/apps/socket/main.py

@@ -12,7 +12,16 @@ from open_webui.utils.utils import decode_token
 
 
 if WEBSOCKET_MANAGER == "redis":
 if WEBSOCKET_MANAGER == "redis":
     mgr = socketio.AsyncRedisManager(WEBSOCKET_REDIS_URL)
     mgr = socketio.AsyncRedisManager(WEBSOCKET_REDIS_URL)
-    sio = socketio.AsyncServer(client_manager=mgr)
+    sio = socketio.AsyncServer(
+        cors_allowed_origins=[],
+        async_mode="asgi",
+        transports=(
+            ["polling", "websocket"] if ENABLE_WEBSOCKET_SUPPORT else ["polling"]
+        ),
+        allow_upgrades=ENABLE_WEBSOCKET_SUPPORT,
+        always_connect=True,
+        client_manager=mgr,
+    )
 else:
 else:
     sio = socketio.AsyncServer(
     sio = socketio.AsyncServer(
         cors_allowed_origins=[],
         cors_allowed_origins=[],