in connectors/aoai.py [0:0]
def __init__(self):
"""
Initializes the AzureOpenAI client.
"""
self.openai_service_name = os.getenv('AZURE_OPENAI_RESOURCE')
self.openai_api_base = f"https://{self.openai_service_name}.openai.azure.com"
self.openai_api_version = os.getenv('AZURE_OPENAI_API_VERSION')
token_provider = get_bearer_token_provider(
ChainedTokenCredential(
ManagedIdentityCredential(),
AzureCliCredential()
), "https://cognitiveservices.azure.com/.default"
)
self.client = AzureOpenAI(
api_version=self.openai_api_version,
azure_endpoint=self.openai_api_base,
azure_ad_token_provider=token_provider,
max_retries=MAX_RETRIES
)