def __init_

def init()

in src/azure_config.py [0:0]
34 lines of code
16 McCabe index (conditional complexity)

    def __init__(self):
        """
        Initializes the AzureConfig object by loading environment variables and setting up
        the necessary Azure clients, including MLClient and CognitiveServicesManagementClient.
        """
        # Load essential environment variables, ensuring necessary configurations are set
        self.subscription_id = self.get_env_var("AZURE_SUBSCRIPTION_ID")
        self.resource_group = self.get_env_var("AZURE_RESOURCE_GROUP")
        self.workspace_name = self.get_env_var("AZUREAI_PROJECT_NAME")
        self.check_missing_vars()

        # If essential variables are provided, initialize Azure clients
        if self.subscription_id and self.resource_group and self.workspace_name:
            # Import necessary Azure libraries only when needed to avoid unnecessary dependencies
            from azure.ai.ml import MLClient
            from azure.identity import DefaultAzureCredential
            from azure.mgmt.cognitiveservices import CognitiveServicesManagementClient

            # Initialize the MLClient using DefaultAzureCredential and configuration
            self.ml_client = MLClient(
                DefaultAzureCredential(),
                self.subscription_id,
                self.resource_group,
                self.workspace_name
            )

            # Retrieve workspace details and update location if available
            self.workspace = self.ml_client.workspaces.get(
                name=self.workspace_name,
                resource_group_name=self.resource_group
            )
            self.location = self.workspace.location  # Use workspace location if available

            # Retrieve service connections for Azure OpenAI and AI Search
            self.aoai_connection = self.ml_client.connections.get('aoai-connection')
            self.search_connection = self.ml_client.connections.get('rag-search')

            # Extract Azure OpenAI endpoint and API version from the connection metadata
            self.aoai_endpoint = self.aoai_connection.target
            self.aoai_api_version = self.aoai_connection.metadata.get('ApiVersion', '')

            # Extract the account name from the OpenAI endpoint for Cognitive Services API keys
            hostname = self.aoai_endpoint.split("://")[1].split("/")[0]
            account_name = hostname.split('.')[0]

            # Initialize the CognitiveServicesManagementClient to retrieve API keys
            self.cognitive_client = CognitiveServicesManagementClient(
                DefaultAzureCredential(), self.subscription_id
            )
            keys = self.cognitive_client.accounts.list_keys(self.resource_group, account_name)
            self.aoai_api_key = keys.key1  # Use the first key for authentication

            # Extract the Azure AI Search endpoint from the search connection
            self.search_endpoint = self.search_connection.target

            # Extract domain prefixes for OpenAI and Search services for easier identification
            self.aoai_account_name = self.get_domain_prefix(self.aoai_endpoint)
            self.search_account_name = self.get_domain_prefix(self.search_endpoint)
def __init__()

def init()
