diff --git a/api/app.py b/api/app.py
index 78faf4f193..634f44d5e5 100644
--- a/api/app.py
+++ b/api/app.py
@@ -4,17 +4,13 @@ from configs import dify_config
 
 if os.environ.get("DEBUG", "false").lower() != "true":
     if os.environ.get("REDIS_GEVENT_SUPPORT", "false").lower() == "true":
-        import socket
+        from gevent import monkey
 
-        original_socket_socket = socket.socket
+        monkey.patch_all(socket=False)
+    else:
+        from gevent import monkey
 
-        from extensions.ext_redis import GeventSafeConnection
-
-        GeventSafeConnection.socket_socket_class = original_socket_socket
-
-    from gevent import monkey
-
-    monkey.patch_all()
+        monkey.patch_all()
 
     import grpc.experimental.gevent