yujunjun/autogen
1
0
Fork 0
mirror of https://github.com/microsoft/autogen.git synced 2025-07-26 10:20:57 +00:00
Code Issues Packages Projects Releases Wiki Activity
autogen/python/docs/src/guides/azure-openai-with-aad-auth.md
Jack Gerrits d365a588cb Move python code to subdir (#98)
2024-06-20 15:19:56 -04:00

1.3 KiB
Raw Blame History

Azure OpenAI with AAD Auth

This guide will show you how to use the Azure OpenAI client with Azure Active Directory (AAD) authentication.

The identity used must be assigned the Cognitive Services OpenAI User role.

Install Azure Identity client

The Azure identity client is used to authenticate with Azure Active Directory.

pip install azure-identity

Using the Model Client

from agnext.components.models import AzureOpenAI
from azure.identity import DefaultAzureCredential, get_bearer_token_provider

# Create the token provider
token_provider = get_bearer_token_provider(
    DefaultAzureCredential(), "https://cognitiveservices.azure.com/.default"
)

client = AzureOpenAI(
    model="{your-azure-deployment}",
    api_version="2024-02-01",
    azure_endpoint="https://{your-custom-endpoint}.openai.azure.com/",
    azure_ad_token_provider=token_provider,
    model_capabilities={
        "vision":True,
        "function_calling":True,
        "json_output":True,
    }
)

See [here](https://learn.microsoft.com/en-us/azure/ai-services/openai/how-to/managed-identity#chat-completions) for how to use the Azure client directly or for more info.