scriptoriumadmin/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui.git synced 2026-11-07 08:17:07 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 3

fix: log terminal proxy connectivity failures as single lines and handle client disconnects

Browse source
This commit is contained in:
silentoplayz 2026-07-30 16:34:02 -04:00
parent 810378c0b8
commit 07673721d6
1 changed files with 10 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

11
backend/open_webui/routers/terminals.py
View file

@ -22,6 +22,7 @@ from open_webui.utils.auth import get_verified_user
from open_webui.utils.terminals import get_terminal_server_url
from open_webui.utils.tools import bearer_auth_header, normalize_bearer_token
from starlette.background import BackgroundTask
from starlette.requests import ClientDisconnect
log = logging.getLogger(__name__)
@ -153,13 +154,14 @@ async def proxy_terminal(
if content_type:
headers['Content-Type'] = content_type
body = await request.body()
session = aiohttp.ClientSession(
timeout=aiohttp.ClientTimeout(total=300, connect=10),
trust_env=True,
)
try:
body = await request.body()
upstream_response = await session.request(
method=request.method,
url=target_url,
@ -200,6 +202,13 @@ async def proxy_terminal(
return Response(content=response_body, status_code=status_code, headers=filtered_headers)
except ClientDisconnect:
await session.close()
return Response(status_code=499)
except (aiohttp.ClientConnectionError, TimeoutError) as error:
await session.close()
log.error('Terminal proxy error: %s', str(error) or type(error).__name__)
return JSONResponse({'error': f'Terminal proxy error: {error}'}, status_code=502)
except Exception as error:
await session.close()
log.exception('Terminal proxy error: %s', error)