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

refac: audio error handling

Browse source
This commit is contained in:
Timothy Jaeryang Baek 2025-07-06 14:20:38 +04:00
parent 44754e4c4a
commit 0a1f9966ef
1 changed files with 18 additions and 16 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

20
backend/open_webui/routers/audio.py
View file

@ -328,6 +328,7 @@ async def speech(request: Request, user=Depends(get_verified_user)):
log.exception(e) log.exception(e)
raise HTTPException(status_code=400, detail="Invalid JSON payload") raise HTTPException(status_code=400, detail="Invalid JSON payload")
r = None
if request.app.state.config.TTS_ENGINE == "openai": if request.app.state.config.TTS_ENGINE == "openai":
payload["model"] = request.app.state.config.TTS_MODEL payload["model"] = request.app.state.config.TTS_MODEL
@ -336,7 +337,7 @@ async def speech(request: Request, user=Depends(get_verified_user)):
async with aiohttp.ClientSession( async with aiohttp.ClientSession(
timeout=timeout, trust_env=True timeout=timeout, trust_env=True
) as session: ) as session:
async with session.post( r = await session.post(
url=f"{request.app.state.config.TTS_OPENAI_API_BASE_URL}/audio/speech", url=f"{request.app.state.config.TTS_OPENAI_API_BASE_URL}/audio/speech",
json=payload, json=payload,
headers={ headers={
@ -354,7 +355,8 @@ async def speech(request: Request, user=Depends(get_verified_user)):
), ),
}, },
ssl=AIOHTTP_CLIENT_SESSION_SSL, ssl=AIOHTTP_CLIENT_SESSION_SSL,
) as r: )
r.raise_for_status() r.raise_for_status()
async with aiofiles.open(file_path, "wb") as f: async with aiofiles.open(file_path, "wb") as f:
@ -369,18 +371,18 @@ async def speech(request: Request, user=Depends(get_verified_user)):
log.exception(e) log.exception(e)
detail = None detail = None
try: status_code = 500
if r.status != 200: detail = f"Open WebUI: Server Connection Error"
res = await r.json()
if r is not None:
status_code = r.status
res = await r.json()
if "error" in res: if "error" in res:
detail = f"External: {res['error'].get('message', '')}" detail = f"External: {res['error'].get('message', '')}"
except Exception:
detail = f"External: {e}"
raise HTTPException( raise HTTPException(
status_code=getattr(r, "status", 500) if r else 500, status_code=status_code,
detail=detail if detail else "Open WebUI: Server Connection Error", detail=detail,
) )
elif request.app.state.config.TTS_ENGINE == "elevenlabs": elif request.app.state.config.TTS_ENGINE == "elevenlabs":