forked from phoenix/litellm-mirror
27e18358ab
* fix(pattern_match_deployments.py): default to user input if unable to map based on wildcards * test: fix test * test: reset test name * test: update conftest to reload proxy server module between tests * ci(config.yml): move langfuse out of local_testing reduce ci/cd time * ci(config.yml): cleanup langfuse ci/cd tests * fix: update test to not use global proxy_server app module * ci: move caching to a separate test pipeline speed up ci pipeline * test: update conftest to check if proxy_server attr exists before reloading * build(conftest.py): don't block on inability to reload proxy_server * ci(config.yml): update caching unit test filter to work on 'cache' keyword as well * fix(encrypt_decrypt_utils.py): use function to get salt key * test: mark flaky test * test: handle anthropic overloaded errors * refactor: create separate ci/cd pipeline for proxy unit tests make ci/cd faster * ci(config.yml): add litellm_proxy_unit_testing to build_and_test jobs * ci(config.yml): generate prisma binaries for proxy unit tests * test: readd vertex_key.json * ci(config.yml): remove `-s` from proxy_unit_test cmd speed up test * ci: remove any 'debug' logging flag speed up ci pipeline * test: fix test * test(test_braintrust.py): rerun * test: add delay for braintrust test
22 lines
679 B
Python
22 lines
679 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 == "":
|
|
raise Exception(
|
|
f"CustomAuth - Malformed API Key passed in. Ensure Key has `Bearer` prefix"
|
|
)
|
|
if api_key == f"{os.getenv('PROXY_MASTER_KEY')}-1234":
|
|
return UserAPIKeyAuth(api_key=api_key)
|
|
raise Exception
|
|
except Exception as e:
|
|
if len(str(e)) > 0:
|
|
raise e
|
|
raise Exception("Failed custom auth")
|