From f1a64ccfc2eb2a58086c55fe413a455bb488c35a Mon Sep 17 00:00:00 2001 From: Timothy Jaeryang Baek Date: Mon, 17 Aug 2026 00:35:12 -0700 Subject: [PATCH] refac --- backend/open_webui/utils/tools.py | 20 ++++++++++++-------- 1 file changed, 12 insertions(+), 8 deletions(-) diff --git a/backend/open_webui/utils/tools.py b/backend/open_webui/utils/tools.py index 8a8e980d06..49f6c46aec 100644 --- a/backend/open_webui/utils/tools.py +++ b/backend/open_webui/utils/tools.py @@ -1170,15 +1170,17 @@ async def set_tool_servers(request: Request): async def get_tool_servers(request: Request): try: - tool_servers = [] + tool_servers = None if request.app.state.redis is not None: try: - tool_servers = JSONCodec.loads(await request.app.state.redis.get(f'{REDIS_KEY_PREFIX}:tool_servers')) - request.app.state.TOOL_SERVERS = tool_servers + data = await request.app.state.redis.get(f'{REDIS_KEY_PREFIX}:tool_servers') + if data is not None: + tool_servers = JSONCodec.loads(data) + request.app.state.TOOL_SERVERS = tool_servers except Exception as e: log.error(f'Error fetching tool_servers from Redis: {e}') - if not tool_servers: + if tool_servers is None: tool_servers = await set_tool_servers(request) return tool_servers @@ -1313,15 +1315,17 @@ async def set_terminal_servers(request: Request): async def get_terminal_servers(request: Request): """Return cached terminal server specs, loading if needed.""" - terminal_servers = [] + terminal_servers = None if request.app.state.redis is not None: try: - terminal_servers = JSONCodec.loads(await request.app.state.redis.get(f'{REDIS_KEY_PREFIX}:terminal_servers')) - request.app.state.TERMINAL_SERVERS = terminal_servers + data = await request.app.state.redis.get(f'{REDIS_KEY_PREFIX}:terminal_servers') + if data is not None: + terminal_servers = JSONCodec.loads(data) + request.app.state.TERMINAL_SERVERS = terminal_servers except Exception as e: log.error(f'Error fetching terminal_servers from Redis: {e}') - if not terminal_servers: + if terminal_servers is None: terminal_servers = await set_terminal_servers(request) return terminal_servers