mirror of
https://github.com/BerriAI/litellm.git
synced 2025-04-25 02:34:29 +00:00
16 lines
501 B
Python
16 lines
501 B
Python
from fastapi import Request
|
|
|
|
|
|
def get_litellm_virtual_key(request: Request) -> str:
|
|
"""
|
|
Extract and format API key from request headers.
|
|
Prioritizes x-litellm-api-key over Authorization header.
|
|
|
|
|
|
Vertex JS SDK uses `Authorization` header, we use `x-litellm-api-key` to pass litellm virtual key
|
|
|
|
"""
|
|
litellm_api_key = request.headers.get("x-litellm-api-key")
|
|
if litellm_api_key:
|
|
return f"Bearer {litellm_api_key}"
|
|
return request.headers.get("Authorization", "")
|