folding-studio/folding_studio/utils/headers.py

"""Headers for HTTP requests."""

from folding_studio.config import FOLDING_API_KEY
from folding_studio.utils.gcp import get_id_token


def get_auth_headers() -> dict[str, str]:
    """
    Create authentication headers based on available credentials.

    API key is the default authentication.
    If none is provided, we fallback to JWT from Google Cloud.

    Returns:
        dict: Authentication headers for API requests.
    """
    if FOLDING_API_KEY:
        return {"X-API-Key": FOLDING_API_KEY}

    identity_token = get_id_token()
    return {"Authorization": f"Bearer {identity_token}"}