ewilliams28/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui.git synced 2026-08-20 12:41:01 -06:00
Code Issues 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):
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