mirror of
https://github.com/BerriAI/litellm.git
synced 2025-04-26 19:24:27 +00:00
16 lines
455 B
Python
16 lines
455 B
Python
from litellm.proxy._types import UserAPIKeyAuth
|
|
from fastapi import Request
|
|
from dotenv import load_dotenv
|
|
import os
|
|
|
|
load_dotenv()
|
|
|
|
|
|
async def user_api_key_auth(request: Request, api_key: str) -> UserAPIKeyAuth:
|
|
try:
|
|
print(f"api_key: {api_key}")
|
|
if api_key == f"{os.getenv('PROXY_MASTER_KEY')}-1234":
|
|
return UserAPIKeyAuth(api_key=api_key)
|
|
raise Exception
|
|
except:
|
|
raise Exception("Failed custom auth")
|