使用 Azure API 管理缩放适用于 JavaScript 的 Azure OpenAI

了解如何将企业级负载均衡添加到应用程序,以将聊天应用扩展到 Azure OpenAI 服务令牌和模型配额限制之外。 此方法使用 Azure API 管理在三个 Azure OpenAI 资源之间智能地定向流量。

本文要求部署两个单独的示例:

  • 聊天应用:
    • 如果尚未部署聊天应用,请等到部署负载均衡器示例之后。
    • 如果已部署聊天应用一次,请更改环境变量以支持负载均衡器的自定义终结点,然后重新部署它。
  • 使用 Azure API 管理的负载均衡器。

注意

本文使用一个或多个 AI 应用模板 作为本文中的示例和指南的基础。 AI 应用模板提供易于部署的维护良好的参考实现。 它们有助于确保你的 AI 应用有一个高质量的起点。

使用 Azure API 管理对 Azure OpenAI 进行负载均衡的体系结构

由于 Azure OpenAI 资源具有特定的令牌和模型配额限制,因此使用单个 Azure OpenAI 资源的聊天应用很容易因为这些限制而发生对话失败。

显示聊天应用体系结构的关系图,其中突出显示了 Azure OpenAI 资源。

若要在不达到这些限制的情况下使用聊天应用,请使用 API 管理的负载均衡解决方案。 此解决方案将 API 管理中的单个终结点无缝地公开到聊天应用服务器。

图示展示了聊天应用的体系结构,其中 Azure API 管理位于三个 Azure OpenAI 资源之前。

API 管理资源(作为 API 层)位于一组 Azure OpenAI 资源前面。 API 层适用于两种场景:正常和受限。 在令牌和模型配额可用的正常场景中,Azure OpenAI 资源会通过 API 层和后端应用服务器返回 200。

显示正常方案的示意图,其中显示三个 Azure OpenAI 终结点组。其中第一组的两个终结点会获得成功的流量。

当某个资源因配额限制而受限时,API 层可以立即重试不同的 Azure OpenAI 资源,以满足原始聊天应用的请求。

显示限制方案的示意图,其中包含 429 个失败的响应代码和客户端必须等待重试的秒数的响应标头。

先决条件

打开 Azure API 管理本地负载均衡器示例应用

GitHub Codespaces 运行由 GitHub 托管的开发容器,Visual Studio Code for web 作为用户界面。 对于最直接的开发环境,请使用 GitHub Codespaces,以便预先安装正确的开发人员工具和依赖项来完成本文。

GitHub Codespaces 中打开。

重要

所有 GitHub 帐户每月最多可以使用 GitHub Codespaces 60 小时,其中包含两个核心实例。 有关详细信息,请参阅 GitHub Codespaces 每月包含的存储和核心小时数

部署 Azure API 管理负载均衡器

  1. 若要将负载均衡器部署到 Azure,请登录到 Azure 开发人员 CLI(AZD):

    azd auth login
    
  2. 完成登录说明。

  3. 部署负载均衡器应用:

    azd up
    

    为部署选择订阅和区域。 它们不必与聊天应用的订阅和区域相同。

  4. 等待部署完成,然后再继续。 此过程最多可能需要 30 分钟。

获取负载均衡器端点

运行以下 Bash 命令,查看部署中的环境变量。 稍后需要此信息。

azd env get-values | grep APIM_GATEWAY_URL

使用负载均衡器终结点重新部署聊天应用

这些示例已在聊天应用示例中完成。

  1. 使用以下选项之一打开聊天应用示例的开发容器。

    语言 GitHub Codespaces Visual Studio Code
    .NET 在 GitHub Codespaces 中打开 在开发容器中打开
    JavaScript 在 GitHub Codespaces 中打开 在开发容器中打开
    Python(编程语言) 在 GitHub Codespaces 中打开 在开发容器中打开
  2. 登录到 Azure 开发人员 CLI(AZD):

    azd auth login
    

    完成登录说明。

  3. 创建一个 AZD 环境并命名,例如 chat-app

    azd env new <name>
    
  4. 添加以下环境变量,该变量告知聊天应用的后端对 Azure OpenAI 请求使用自定义 URL:

    azd env set OPENAI_HOST azure_custom
    
  5. 添加以下环境变量,该变量告知聊天应用的后端 Azure OpenAI 请求的自定义 URL 的值:

    azd env set AZURE_OPENAI_CUSTOM_URL <APIM_GATEWAY_URL>
    
  6. 部署聊天应用:

    azd up
    

配置 TPM 配额

默认情况下,负载均衡器中的每个 Azure OpenAI 实例部署的容量为每分钟 30,000 个令牌(TPM)。 您可以安心使用聊天应用程序,因为它被设计为可支持大量用户,而不会超出配额限制。 在以下情况下更改此值:

  • 出现部署容量错误:降低该值。
  • 需要提高容量:增加该值。
  1. 使用以下命令更改值:

    azd env set OPENAI_CAPACITY 50
    
  2. 重新部署负载均衡器:

    azd up
    

清理资源

完成聊天应用和负载均衡器后,清理资源。 本文中创建的 Azure 资源将计费给 Azure 订阅。 如果不希望将来需要这些资源,请将其删除,以避免产生更多费用。

清理聊天应用资源

返回到聊天应用文章以清理这些资源。

清理负载均衡器资源

运行以下 Azure 开发人员 CLI 命令以删除 Azure 资源并删除源代码:

azd down --purge --force

这些开关提供:

  • purge:立即清除已删除的资源。 可以每分钟重复使用 Azure OpenAI 令牌。
  • force:删除以无提示方式进行,无需用户同意。

清理资源

删除 GitHub Codespaces 环境可确保可以最大程度地提高帐户获得的每核心免费小时数权利。

重要

有关 GitHub 帐户权利的详细信息,请参阅 GitHub Codespaces 每月包含的存储和核心小时数

  1. 登录到 GitHub Codespaces 仪表板

  2. 找到当前正在运行的、源自 azure-samples/openai-apim-lb GitHub 存储库的 codespace。

    屏幕截图,显示所有正在运行的代码空间,包括其状态和模板。

  3. 打开 GitHub Codespaces 项的上下文菜单,然后选择 删除

    屏幕截图,其中显示了单个代码空间的上下文菜单,其中突出显示了“删除”选项。

获取帮助

如果在部署 Azure API 管理负载均衡器时遇到问题,请将问题添加到存储库 问题 网页。

示例代码

本文中使用的示例包括:

下一步