使用 Azure Active Directory 认证的 Azure OpenAI

本指南将展示如何通过 Azure Active Directory (AAD) 认证使用 Azure OpenAI 客户端。

所使用的身份必须被分配 Cognitive Services OpenAI User 角色。

安装 Azure Identity 客户端

Azure identity 客户端用于与 Azure Active Directory 进行认证。

pip install azure-identity

使用模型客户端

from autogen_ext.models.openai import AzureOpenAIChatCompletionClient
from azure.identity import DefaultAzureCredential, get_bearer_token_provider


# 创建令牌提供器
token_provider = get_bearer_token_provider(
    DefaultAzureCredential(), "https://cognitiveservices.azure.com/.default"
)

client = AzureOpenAIChatCompletionClient(
    azure_deployment="{your-azure-deployment}",
    model="{model-name, such as gpt-4o}",
    api_version="2024-02-01",
    azure_endpoint="https://{your-custom-endpoint}.openai.azure.com/",
    azure_ad_token_provider=token_provider,
)

备注

关于如何直接使用 Azure 客户端或获取更多信息，请参阅此处。