def initialize_embedding_model()

in migrate.py [0:0]

• 16 lines of code
• 2 McCabe index (conditional complexity)

def initialize_embedding_model():
    embeddings_client = None
    if "openai" in os.environ["AZURE_AI_EMBEDDINGS_ENDPOINT"]:
        print("initializing embeddings client AOAI")
        embeddings_client = EmbeddingsClient(
            endpoint=os.environ["AZURE_AI_EMBEDDINGS_ENDPOINT"],
            credential=AzureKeyCredential(os.environ["AZURE_AI_EMBEDDINGS_KEY"]),
            api_version=os.environ["AZURE_AI_EMBEDDINGS_API_VERSION"]
        )
    else:
        print("initializing embeddings client not AOAI")
        embeddings_client = EmbeddingsClient(
            endpoint=os.environ["AZURE_AI_EMBEDDINGS_ENDPOINT"],
            credential=AzureKeyCredential(os.environ["AZURE_AI_EMBEDDINGS_KEY"])
        )
    return embeddings_client