UniqAI/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui synced 2024-11-26 14:02:29 +00:00
Code Issues Actions 13 Packages Projects Releases Wiki Activity

refac: auth required for litellm in dev

Browse Source
This commit is contained in:
Timothy J. Baek 2024-03-20 19:33:27 -07:00
parent ed0d4b5a2a
commit 8e52ba8b91
1 changed files with 6 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
backend/apps/litellm/main.py
View File

@ -47,13 +47,12 @@ async def auth_middleware(request: Request, call_next):
auth_header = request.headers.get("Authorization", "") auth_header = request.headers.get("Authorization", "")
request.state.user = None request.state.user = None
if ENV != "dev": try:
try: user = get_current_user(get_http_authorization_cred(auth_header))
user = get_current_user(get_http_authorization_cred(auth_header)) print(user)
print(user) request.state.user = user
request.state.user = user except Exception as e:
except Exception as e: return JSONResponse(status_code=400, content={"detail": str(e)})
return JSONResponse(status_code=400, content={"detail": str(e)})
response = await call_next(request) response = await call_next(request)
return response return response