From 81d4ed79aed95ef3b078e4b83fae63ee5fa21297 Mon Sep 17 00:00:00 2001
From: Classic298 <27028174+Classic298@users.noreply.github.com>
Date: Sun, 31 May 2026 23:48:51 +0200
Subject: [PATCH] Update main.py (#25271)

---
 backend/open_webui/socket/main.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/backend/open_webui/socket/main.py b/backend/open_webui/socket/main.py
index d86a86be29..2884847a0e 100644
--- a/backend/open_webui/socket/main.py
+++ b/backend/open_webui/socket/main.py
@@ -805,7 +805,7 @@ async def yjs_awareness_update(sid, data):
 
 
 @sio.event
-async def disconnect(sid):
+async def disconnect(sid, reason=None):
     if sid in SESSION_POOL:
         user = SESSION_POOL[sid]
         del SESSION_POOL[sid]