bhetherman/open-webui
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

refac

Browse Source
This commit is contained in:
Timothy Jaeryang Baek
2026-08-17 00:35:12 -07:00
parent 76d0160295
commit f1a64ccfc2
1 changed files with 12 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files
backend/open_webui/utils/tools.py
+12 -8
View File
@@ -1170,15 +1170,17 @@ async def set_tool_servers(request: Request):
async def get_tool_servers(request: Request): async def get_tool_servers(request: Request):
try: try:
tool_servers = [] tool_servers = None
if request.app.state.redis is not None: if request.app.state.redis is not None:
try: try:
tool_servers = JSONCodec.loads(await request.app.state.redis.get(f'{REDIS_KEY_PREFIX}:tool_servers')) data = await request.app.state.redis.get(f'{REDIS_KEY_PREFIX}:tool_servers')
request.app.state.TOOL_SERVERS = tool_servers if data is not None:
tool_servers = JSONCodec.loads(data)
request.app.state.TOOL_SERVERS = tool_servers
except Exception as e: except Exception as e:
log.error(f'Error fetching tool_servers from Redis: {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) tool_servers = await set_tool_servers(request)
return tool_servers return tool_servers
@@ -1313,15 +1315,17 @@ async def set_terminal_servers(request: Request):
async def get_terminal_servers(request: Request): async def get_terminal_servers(request: Request):
"""Return cached terminal server specs, loading if needed.""" """Return cached terminal server specs, loading if needed."""
terminal_servers = [] terminal_servers = None
if request.app.state.redis is not None: if request.app.state.redis is not None:
try: try:
terminal_servers = JSONCodec.loads(await request.app.state.redis.get(f'{REDIS_KEY_PREFIX}:terminal_servers')) data = await request.app.state.redis.get(f'{REDIS_KEY_PREFIX}:terminal_servers')
request.app.state.TERMINAL_SERVERS = terminal_servers if data is not None:
terminal_servers = JSONCodec.loads(data)
request.app.state.TERMINAL_SERVERS = terminal_servers
except Exception as e: except Exception as e:
log.error(f'Error fetching terminal_servers from Redis: {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) terminal_servers = await set_terminal_servers(request)
return terminal_servers return terminal_servers