首页
/ CodeceptJS 中使用 Bearer Token 认证访问 Azure OpenAI 服务的实践指南

CodeceptJS 中使用 Bearer Token 认证访问 Azure OpenAI 服务的实践指南

2025-06-15 11:12:19作者:董宙帆

背景介绍

在现代自动化测试中,人工智能辅助测试正变得越来越普遍。CodeceptJS 作为一款流行的 Node.js 端到端测试框架,提供了与 AI 服务集成的能力。然而,当开发者尝试使用 Bearer Token 而非 API Key 来认证 Azure OpenAI 服务时,可能会遇到一些技术挑战。

核心问题分析

传统上,CodeceptJS 的 AI 辅助功能默认使用 API Key 进行认证。但在企业级 Azure OpenAI 服务部署中,出于安全考虑,组织往往会要求使用更安全的 Bearer Token 认证方式。这种认证方式基于 OAuth 2.0 协议,通过短期有效的访问令牌来保护 API 访问。

解决方案实现

方法一:使用 Azure 身份认证库

最推荐的解决方案是利用 Azure 官方提供的 @azure/identity 库来获取 Bearer Token:

const {getBearerTokenProvider, DefaultAzureCredential} = require("@azure/identity");
const AzureOpenAI = require("openai");

const scope = "https://cognitiveservices.azure.com/.default";
const azureADTokenProvider = getBearerTokenProvider(
  new DefaultAzureCredential(), 
  scope
);

const client = new AzureOpenAI({ 
  azureADTokenProvider,
  deployment: "YOUR_DEPLOYMENT_NAME",
  apiVersion: "2024-04-01-preview"
});

这种方式会自动处理令牌的获取和刷新,是最安全可靠的方法。

方法二:手动获取 Bearer Token

如果无法使用 Azure 默认凭证,可以手动实现令牌获取逻辑:

async function getAzureBearerToken() {
  const axios = require("axios").default;
  const FormData = require("form-data");

  const form = new FormData();
  form.append("grant_type", "client_credentials");
  form.append("client_id", process.env.CLIENT_ID);
  form.append("client_secret", process.env.CLIENT_SECRET);

  const response = await axios.post(
    `https://login.microsoftonline.com/${process.env.RESOURCE_ID}/oauth2/token`,
    form
  );
  return response.data;
}

然后在 CodeceptJS 配置中使用:

AI: {
  request: async (messages) => {
    const token = await getAzureBearerToken();
    const client = new AzureOpenAI({
      azureADTokenProvider: token,
      deployment: process.env.DEPLOYMENT_NAME,
      apiVersion: "2024-04-01-preview"
    });
    
    const result = await client.chat.completions.create({
      messages,
      model: 'gpt-3.5-turbo'
    });
    
    return result.choices[0]?.message?.content;
  }
}

模型兼容性注意事项

当切换到 Bearer Token 认证后,开发者还需要注意:

  1. Azure OpenAI 部署的模型名称可能与原生 OpenAI 不同
  2. 确保使用的 API 版本与 Azure 服务端兼容
  3. 某些模型可能只支持特定类型的操作(如聊天补全或文本补全)

最佳实践建议

  1. 始终将敏感凭证存储在环境变量中,而非硬编码在配置文件里
  2. 为不同的测试环境(开发、测试、生产)配置不同的 Azure AD 应用
  3. 定期轮换客户端密钥以提高安全性
  4. 监控令牌使用情况,避免频繁获取新令牌

总结

通过本文介绍的方法,开发者可以成功地在 CodeceptJS 测试框架中集成使用 Bearer Token 认证的 Azure OpenAI 服务。这种集成方式不仅提高了安全性,还能更好地适应企业级开发环境的要求。在实际实施时,建议根据具体项目需求和 Azure 环境配置选择合适的实现方案。

登录后查看全文
热门项目推荐

项目优选

收起
ohos_react_nativeohos_react_native
React Native鸿蒙化仓库
C++
177
262
RuoYi-Vue3RuoYi-Vue3
🎉 (RuoYi)官方仓库 基于SpringBoot,Spring Security,JWT,Vue3 & Vite、Element Plus 的前后端分离权限管理系统
Vue
864
512
ShopXO开源商城ShopXO开源商城
🔥🔥🔥ShopXO企业级免费开源商城系统,可视化DIY拖拽装修、包含PC、H5、多端小程序(微信+支付宝+百度+头条&抖音+QQ+快手)、APP、多仓库、多商户、多门店、IM客服、进销存,遵循MIT开源协议发布、基于ThinkPHP8框架研发
JavaScript
93
15
openGauss-serveropenGauss-server
openGauss kernel ~ openGauss is an open source relational database management system
C++
129
182
openHiTLSopenHiTLS
旨在打造算法先进、性能卓越、高效敏捷、安全可靠的密码套件,通过轻量级、可剪裁的软件技术架构满足各行业不同场景的多样化要求,让密码技术应用更简单,同时探索后量子等先进算法创新实践,构建密码前沿技术底座!
C
261
302
kernelkernel
deepin linux kernel
C
22
5
cherry-studiocherry-studio
🍒 Cherry Studio 是一款支持多个 LLM 提供商的桌面客户端
TypeScript
596
57
CangjieCommunityCangjieCommunity
为仓颉编程语言开发者打造活跃、开放、高质量的社区环境
Markdown
1.07 K
0
HarmonyOS-ExamplesHarmonyOS-Examples
本仓将收集和展示仓颉鸿蒙应用示例代码,欢迎大家投稿,在仓颉鸿蒙社区展现你的妙趣设计!
Cangjie
398
371
Cangjie-ExamplesCangjie-Examples
本仓将收集和展示高质量的仓颉示例代码,欢迎大家投稿,让全世界看到您的妙趣设计,也让更多人通过您的编码理解和喜爱仓颉语言。
Cangjie
332
1.08 K