From 2f966d8c389db5c9ea57abb469d6f5ee7b7ca3d5 Mon Sep 17 00:00:00 2001 From: hjlarry Date: Thu, 17 Jul 2025 17:16:52 +0800 Subject: [PATCH] fix websocket auth --- api/controllers/console/app/online_user.py | 1 - api/extensions/ext_login.py | 4 ++-- 2 files changed, 2 insertions(+), 3 deletions(-) diff --git a/api/controllers/console/app/online_user.py b/api/controllers/console/app/online_user.py index fc03649d06..905e4622df 100644 --- a/api/controllers/console/app/online_user.py +++ b/api/controllers/console/app/online_user.py @@ -6,7 +6,6 @@ from extensions.ext_redis import redis_client from extensions.ext_socketio import ext_socketio - @ext_socketio.on("user_connect") def handle_user_connect(data): """ diff --git a/api/extensions/ext_login.py b/api/extensions/ext_login.py index 3ff8ec0390..39be9929c7 100644 --- a/api/extensions/ext_login.py +++ b/api/extensions/ext_login.py @@ -1,9 +1,9 @@ import json import flask_login # type: ignore -from flask import Response, request, g -from flask_socketio import disconnect +from flask import Response, request from flask_login import user_loaded_from_request, user_logged_in +from flask_socketio import disconnect from werkzeug.exceptions import NotFound, Unauthorized from configs import dify_config