import dataiku


def get_api_key(secret_name: str = "AZURE_OPENAI_API_KEY") -> str:
    """
    Get an API Key from Dataiku user secrets
    Args:
        secret_name: The secret key you want to retrieve.

    Returns:
        The value of the secret key (if found)
    """
    client = dataiku.api_client()
    auth_info = client.get_auth_info(with_secrets=True)
    secret_value = None
    for secret in auth_info["secrets"]:
        if secret["key"] == secret_name:
            secret_value = secret["value"]
            break
    if not secret_value:
        raise Exception(f"User secret key {secret_name} not found.")
    else:
        return secret_value