ti.sil/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui.git synced 2025-12-12 12:25:20 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 4

refactor: change MAX_RETRY_COUNT as env

Browse source 
Signed-off-by: Sihyeon Jang <sihyeon.jang@navercorp.com>
This commit is contained in:
Sihyeon Jang 2025-07-16 09:11:08 +09:00
parent 1b7ac7c739
commit 65882c30cb
2 changed files with 14 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

9
backend/open_webui/env.py
View file

@ -349,6 +349,15 @@ REDIS_KEY_PREFIX = os.environ.get("REDIS_KEY_PREFIX", "open-webui")
REDIS_SENTINEL_HOSTS = os.environ.get("REDIS_SENTINEL_HOSTS", "") REDIS_SENTINEL_HOSTS = os.environ.get("REDIS_SENTINEL_HOSTS", "")
REDIS_SENTINEL_PORT = os.environ.get("REDIS_SENTINEL_PORT", "26379") REDIS_SENTINEL_PORT = os.environ.get("REDIS_SENTINEL_PORT", "26379")
# Maximum number of retries for Redis operations when using Sentinel fail-over
REDIS_SENTINEL_MAX_RETRY_COUNT = os.environ.get("REDIS_SENTINEL_MAX_RETRY_COUNT", "2")
try:
REDIS_SENTINEL_MAX_RETRY_COUNT = int(REDIS_SENTINEL_MAX_RETRY_COUNT)
if REDIS_SENTINEL_MAX_RETRY_COUNT < 1:
REDIS_SENTINEL_MAX_RETRY_COUNT = 2
except ValueError:
REDIS_SENTINEL_MAX_RETRY_COUNT = 2
#################################### ####################################
# UVICORN WORKERS # UVICORN WORKERS
#################################### ####################################

11
backend/open_webui/utils/redis.py
View file

@ -3,8 +3,7 @@ from urllib.parse import urlparse
import redis import redis
from open_webui.env import REDIS_SENTINEL_MAX_RETRY_COUNT
MAX_RETRY_COUNT = 2
class SentinelRedisProxy: class SentinelRedisProxy:
@ -31,7 +30,7 @@ class SentinelRedisProxy:
if self._async_mode: if self._async_mode:
async def _wrapped(*args, **kwargs): async def _wrapped(*args, **kwargs):
for i in range(MAX_RETRY_COUNT): for i in range(REDIS_SENTINEL_MAX_RETRY_COUNT):
try: try:
method = getattr(self._master(), item) method = getattr(self._master(), item)
result = method(*args, **kwargs) result = method(*args, **kwargs)
@ -42,7 +41,7 @@ class SentinelRedisProxy:
redis.exceptions.ConnectionError, redis.exceptions.ConnectionError,
redis.exceptions.ReadOnlyError, redis.exceptions.ReadOnlyError,
) as e: ) as e:
if i < MAX_RETRY_COUNT - 1: if i < REDIS_SENTINEL_MAX_RETRY_COUNT - 1:
continue continue
raise e from e raise e from e
@ -51,7 +50,7 @@ class SentinelRedisProxy:
else: else:
def _wrapped(*args, **kwargs): def _wrapped(*args, **kwargs):
for i in range(MAX_RETRY_COUNT): for i in range(REDIS_SENTINEL_MAX_RETRY_COUNT):
try: try:
method = getattr(self._master(), item) method = getattr(self._master(), item)
return method(*args, **kwargs) return method(*args, **kwargs)
@ -59,7 +58,7 @@ class SentinelRedisProxy:
redis.exceptions.ConnectionError, redis.exceptions.ConnectionError,
redis.exceptions.ReadOnlyError, redis.exceptions.ReadOnlyError,
) as e: ) as e:
if i < MAX_RETRY_COUNT - 1: if i < REDIS_SENTINEL_MAX_RETRY_COUNT - 1:
continue continue
raise e from e raise e from e