tdpeuter/open-webui
tdpeuter/open-webui
1
Fork 0
forked from open-webui/open-webui
Code Issues Pull requests 1 Releases 2 Packages Activity Actions

fix

Browse source
This commit is contained in:
Timothy J. Baek 2024-04-02 09:18:15 -07:00
parent 56369fea3a
commit b11af9ea0e
1 changed files with 6 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
backend/apps/web/models/auths.py
View file

@ -48,9 +48,11 @@ class Token(BaseModel):
token: str token: str
token_type: str token_type: str
class ApiKey(BaseModel): class ApiKey(BaseModel):
api_key: Optional[str] = None api_key: Optional[str] = None
class UserResponse(BaseModel): class UserResponse(BaseModel):
id: str id: str
email: str email: str
@ -132,6 +134,7 @@ class AuthsTable:
# if no api_key, return None # if no api_key, return None
if not api_key: if not api_key:
return None return None
try: try:
auth = Auth.get(Auth.api_key == api_key, Auth.active == True) auth = Auth.get(Auth.api_key == api_key, Auth.active == True)
if auth: if auth:
@ -140,6 +143,9 @@ class AuthsTable:
else: else:
return None return None
except:
return False
def authenticate_user_by_trusted_header(self, email: str) -> Optional[UserModel]: def authenticate_user_by_trusted_header(self, email: str) -> Optional[UserModel]:
log.info(f"authenticate_user_by_trusted_header: {email}") log.info(f"authenticate_user_by_trusted_header: {email}")
try: try: