Coverage for haystack/utils/azure.py: 67%

1# SPDX-FileCopyrightText: 2022-present deepset GmbH <info@deepset.ai> 

2# 

3# SPDX-License-Identifier: Apache-2.0 

4 

5from haystack.lazy_imports import LazyImport 

6 

7with LazyImport(message="Run 'pip install azure-identity") as azure_import: 

8 from azure.identity import DefaultAzureCredential, get_bearer_token_provider 

9 

10 

11def default_azure_ad_token_provider() -> str: 

12 """ 

13 Get a Azure AD token using the DefaultAzureCredential and the "https://cognitiveservices.azure.com/.default" scope. 

14 """ 

15 azure_import.check() 

16 return get_bearer_token_provider(DefaultAzureCredential(), "https://cognitiveservices.azure.com/.default")()