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

Format openai router

Browse source
This commit is contained in:
Juan Calderon-Perez 2025-08-08 10:04:02 -04:00
parent 2ae7584686
commit 5d7e8c8e5f
1 changed files with 24 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

31
backend/open_webui/routers/openai.py
View file

@ -10,7 +10,12 @@ import requests
from urllib.parse import quote
from fastapi import Depends, HTTPException, Request, APIRouter
from fastapi.responses import FileResponse, StreamingResponse, JSONResponse, PlainTextResponse
from fastapi.responses import (
FileResponse,
StreamingResponse,
JSONResponse,
PlainTextResponse,
)
from pydantic import BaseModel
from starlette.background import BackgroundTask
@ -600,9 +605,13 @@ async def verify_connection(
if r.status != 200:
if isinstance(response_data, (dict, list)):
return JSONResponse(status_code=r.status, content=response_data)
return JSONResponse(
status_code=r.status, content=response_data
)
else:
return PlainTextResponse(status_code=r.status, content=response_data)
return PlainTextResponse(
status_code=r.status, content=response_data
)
return response_data
else:
@ -620,9 +629,13 @@ async def verify_connection(
if r.status != 200:
if isinstance(response_data, (dict, list)):
return JSONResponse(status_code=r.status, content=response_data)
return JSONResponse(
status_code=r.status, content=response_data
)
else:
return PlainTextResponse(status_code=r.status, content=response_data)
return PlainTextResponse(
status_code=r.status, content=response_data
)
return response_data
@ -973,7 +986,9 @@ async def embeddings(request: Request, form_data: dict, user):
if isinstance(response_data, (dict, list)):
return JSONResponse(status_code=r.status, content=response_data)
else:
return PlainTextResponse(status_code=r.status, content=response_data)
return PlainTextResponse(
status_code=r.status, content=response_data
)
return response_data
except Exception as e:
@ -1068,7 +1083,9 @@ async def proxy(path: str, request: Request, user=Depends(get_verified_user)):
if isinstance(response_data, (dict, list)):
return JSONResponse(status_code=r.status, content=response_data)
else:
return PlainTextResponse(status_code=r.status, content=response_data)
return PlainTextResponse(
status_code=r.status, content=response_data
)
return response_data