UniqAI
/
open-webui
mirror of https://github.com/open-webui/open-webui
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

fix

This commit is contained in:
Timothy J. Baek 2024-04-02 10:06:58 -07:00
parent 398b2076cd
commit 1066096b33
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
backend/utils/utils.py
View File

@ -2,7 +2,6 @@ from fastapi.security import HTTPBearer, HTTPAuthorizationCredentials
from fastapi import HTTPException, status, Depends
from apps.web.models.users import Users
from apps.web.models.auths import Auths
from pydantic import BaseModel
from typing import Union, Optional
@ -78,6 +77,8 @@ def get_http_authorization_cred(auth_header: str):
def get_current_user(
auth_token: HTTPAuthorizationCredentials = Depends(bearer_security),
):
print(auth_token)
# auth by api key
if auth_token.credentials.startswith("sk-"):
return get_current_user_by_api_key(auth_token.credentials)
@ -99,7 +100,7 @@ def get_current_user(
def get_current_user_by_api_key(api_key: str):
user = Auths.authenticate_user_by_api_key(api_key)
user = Users.get_user_by_api_key(api_key)
if user is None:
raise HTTPException(
status_code=status.HTTP_401_UNAUTHORIZED,