def _get_auth_headers()

in aidial_adapter_openai/embeddings/azure_ai_vision.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 {"Ocp-Apim-Subscription-Key": creds["api_key"]}

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

    raise ValueError("Invalid credentials")