"...git@developer.sourcefind.cn:chenpangpang/open-webui.git" did not exist on "565f40c64250c69ec1f7a5a296c85f0d7dbd4a6b"
Commit 1066096b authored by Timothy J. Baek's avatar Timothy J. Baek
Browse files

fix

parent 398b2076
......@@ -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,
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment