jalr/tabbyAPI-ollama
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Auth: Fix an oops with headers

Browse source 
I copy pasted the code wrong.

Signed-off-by: kingbri <bdashore3@proton.me>
This commit is contained in:
kingbri 2023-11-20 00:04:10 -05:00
parent 39ea730be5
commit 2aa9c145be
1 changed files with 2 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

4
auth.py
View file

@ -55,7 +55,7 @@ def load_auth_keys():
def check_api_key(x_api_key: str = Header(None), authorization: str = Header(None)):
if x_api_key:
if auth_keys.verify_key(split_key[1], "api_key"):
if auth_keys.verify_key(x_api_key, "api_key"):
return x_api_key
else:
raise HTTPException(401, "Invalid API key")
@ -73,7 +73,7 @@ def check_api_key(x_api_key: str = Header(None), authorization: str = Header(Non
def check_admin_key(x_admin_key: str = Header(None), authorization: str = Header(None)):
if x_admin_key:
if auth_keys.verify_key(split_key[1], "api_key"):
if auth_keys.verify_key(x_admin_key, "admin_key"):
return x_admin_key
else:
raise HTTPException(401, "Invalid admin key")