def get_auth_headers()

in aidial_adapter_openai/utils/auth.py [0:0]

• 6 lines of code
• 3 McCabe index (conditional complexity)

def get_auth_headers(creds: OpenAICreds) -> dict[str, str]:
    if "api_key" in creds:
        return {"api-key": creds["api_key"]}

    if "azure_ad_token" in creds:
        return {"Authorization": f"Bearer {creds['azure_ad_token']}"}

    raise ValueError("Invalid credentials")