ti.sil/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui.git synced 2025-12-13 04:45:19 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 3

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 urllib.parse import quote
from fastapi import Depends, HTTPException, Request, APIRouter 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 pydantic import BaseModel
from starlette.background import BackgroundTask from starlette.background import BackgroundTask
@ -600,9 +605,13 @@ async def verify_connection(
if r.status != 200: if r.status != 200:
if isinstance(response_data, (dict, list)): 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: else:
return PlainTextResponse(status_code=r.status, content=response_data) return PlainTextResponse(
status_code=r.status, content=response_data
)
return response_data return response_data
else: else:
@ -620,9 +629,13 @@ async def verify_connection(
if r.status != 200: if r.status != 200:
if isinstance(response_data, (dict, list)): 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: else:
return PlainTextResponse(status_code=r.status, content=response_data) return PlainTextResponse(
status_code=r.status, content=response_data
)
return response_data return response_data
@ -973,7 +986,9 @@ async def embeddings(request: Request, form_data: dict, user):
if isinstance(response_data, (dict, list)): 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: else:
return PlainTextResponse(status_code=r.status, content=response_data) return PlainTextResponse(
status_code=r.status, content=response_data
)
return response_data return response_data
except Exception as e: 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)): 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: else:
return PlainTextResponse(status_code=r.status, content=response_data) return PlainTextResponse(
status_code=r.status, content=response_data
)
return response_data return response_data