diff --git a/backend/open_webui/socket/utils.py b/backend/open_webui/socket/utils.py index 5739a8027a..ccbe333656 100644 --- a/backend/open_webui/socket/utils.py +++ b/backend/open_webui/socket/utils.py @@ -190,7 +190,9 @@ class YdocManager: async def remove_user_from_all_documents(self, user_id: str): if self._redis: - keys = await self._redis.keys(f"{self._redis_key_prefix}:*") + keys = [] + async for key in self._redis.scan_iter(match=f"{self._redis_key_prefix}:*", count=100): + keys.append(key) for key in keys: if key.endswith(":users"): await self._redis.srem(key, user_id)