WHartTest 测试自动化平台接入 TaoToken：统一 Key 打通 Django REST Framework 与大模型调用

1. WHartTest 接入大模型时Key 管理为什么容易乱WHartTest 是麦港数据开源的一套 AI 测试自动化平台后端基于 Django REST Framework 构建前端用 Vue3整体走前后端分离的 Monorepo 架构。它能做的事情很具体把需求文档喂进去让大模型生成结构化测试用例再通过 Playwright 或接口自动化模块把用例跑起来最后把执行结果和截图回传知识库。适合谁用适合那些已经在做测试自动化、但被多模型 Key 管理折腾得够呛的测试开发团队。问题就出在“多模型”这三个字上。WHartTest 支持 OpenAI、Anthropic、Claude、Ollama 等多种模型服务社区版可以配一套全局模型专业版还能按模块独立配置不同 LLM。听起来很灵活但实际落地时你会在 settings.json、config.toml、环境变量、数据库配置表里到处散落 API Key。换一个模型供应商就要改一遍配置某个 Key 额度用完了得翻半天才知道是哪个模块在用。更麻烦的是测试自动化场景对稳定性要求高。用例生成、需求评审、执行报告分析这些环节都依赖大模型调用。如果 Key 分散管理一旦某个供应商接口抖动排查链路会很长——你甚至不确定是网络问题、Key 失效还是模型返回格式变了。我试过把多个供应商的 Key 硬编码在 Django 的 settings 里结果每次轮换 Key 都要重新部署。后来改成统一走一个 API 通道所有模型调用都从同一个入口出去配置量直接降下来。这篇就按这个思路讲清楚 WHartTest 在 Django REST Framework 后端下怎么用 TaoToken 统一 Key 打通大模型调用并给出一份可复制的配置骨架和验证步骤。2. TaoToken 在 WHartTest 里的定位与前置准备TaoToken 在这里扮演的角色是“统一 API 通道”。WHartTest 的 Django 后端本来要分别对接 OpenAI、Anthropic 等不同厂商的接口地址和鉴权方式现在改成只对接 TaoToken 一个入口由它来路由到具体模型。对 WHartTest 来说配置项从“N 个供应商 × M 个参数”收敛成“一个 base_url 一个 Key 模型名”。这样做的好处很直接。第一Key 只有一份轮换时改一个地方。第二模型切换不用动代码改配置里的模型名就行。第三调用链路统一后日志和用量统计也集中了排查问题不用在多个供应商后台之间跳。前置准备需要你确认几件事。WHartTest 后端能正常启动Django REST Framework 的接口能访问Python 环境里 requests 或 httpx 可用然后去 TaoToken 控制台创建一个 API Key。创建入口在控制台的 API Keys 页面地址是 https://taotoken.net/console/api-keys?utm_sourcetaotoken_aicg_blog_endutm_contentapi_keysutm_campaignrewrite 。拿到 Key 之后先别急着写进代码放到环境变量里后面配置会引用它。模型名这块要注意TaoToken 的模型对话接口兼容主流命名你在 WHartTest 里填的模型名要和 TaoToken 支持的名称一致。不确定的话可以先去模型对话页面确认可用模型列表https://taotoken.net/models?utm_sourcetaotoken_aicg_blog_endutm_contentmodelsutm_campaignrewrite 。接入文档在 https://taotoken.net/doc?utm_sourcetaotoken_aicg_blog_endutm_contentdocutm_campaignrewrite 里面有完整的请求格式说明。3. settings.json 与 config.toml 可复制配置骨架WHartTest 的配置分两层前端或工具链侧用 settings.json后端服务侧用 config.toml。下面给的是骨架字段名按你实际项目调整但结构可以直接抄。先看 settings.json这个文件通常放在项目根目录或前端配置目录用来告诉 WHartTest 的 AI 模块去哪里调模型{ llm: { provider: taotoken, base_url: https://taotoken.net/api, api_key_env: TAOTOKEN_API_KEY, default_model: claude-sonnet-4-20250514, timeout: 60, max_retries: 2, models: { case_generation: claude-sonnet-4-20250514, requirement_review: claude-sonnet-4-20250514, report_analysis: claude-sonnet-4-20250514 } }, embedding: { provider: taotoken, base_url: https://taotoken.net/api, api_key_env: TAOTOKEN_API_KEY, model: text-embedding-3-small } }这里的关键点是api_key_env指向环境变量名而不是把 Key 明文写进 JSON。base_url用 https://taotoken.net/api 注意这个地址不带 UTM 参数是纯 API 端点。models里按 WHartTest 的功能模块分别指定模型专业版可以在这里做模块级模型路由。再看 config.toml这个文件给 Django 后端读取通常放在wharttest_backend/config/或类似路径[llm] provider taotoken base_url https://taotoken.net/api api_key_env TAOTOKEN_API_KEY default_model claude-sonnet-4-20250514 timeout 60 max_retries 2 [llm.module_models] case_generation claude-sonnet-4-20250514 requirement_review claude-sonnet-4-20250514 report_analysis claude-sonnet-4-20250514 [embedding] provider taotoken base_url https://taotoken.net/api api_key_env TAOTOKEN_API_KEY model text-embedding-3-small [security] mask_api_key_in_logs true audit_llm_calls truemask_api_key_in_logs建议打开避免日志里泄露 Key。audit_llm_calls打开后每次模型调用都会记审计日志方便后面排查。环境变量在启动 Django 前设置export TAOTOKEN_API_KEY你的Key如果你用 Docker 部署 WHartTest在 docker-compose.yml 的 environment 段里加这一条或者用.env文件挂载。4. Django REST Framework 侧读取配置并完成一次调用验证配置写好了接下来要让 Django 后端真正读进去。WHartTest 的 AI 调用通常封装在一个 service 层比如apps/ai/services/llm_client.py。下面给一个最小可用的客户端封装用 requests 直接调 TaoToken 的对话接口import os import requests from django.conf import settings class TaoTokenClient: def __init__(self): self.base_url settings.LLM_CONFIG[base_url] self.api_key os.environ.get(settings.LLM_CONFIG[api_key_env]) self.default_model settings.LLM_CONFIG[default_model] self.timeout settings.LLM_CONFIG.get(timeout, 60) def chat(self, messages, modelNone): url f{self.base_url}/v1/messages headers { Authorization: fBearer {self.api_key}, Content-Type: application/json, } payload { model: model or self.default_model, messages: messages, max_tokens: 1024, } resp requests.post(url, jsonpayload, headersheaders, timeoutself.timeout) resp.raise_for_status() return resp.json()然后在 Django 的 settings.py 里加载 config.tomlimport tomllib with open(BASE_DIR / config / config.toml, rb) as f: _cfg tomllib.load(f) LLM_CONFIG { base_url: _cfg[llm][base_url], api_key_env: _cfg[llm][api_key_env], default_model: _cfg[llm][default_model], timeout: _cfg[llm].get(timeout, 60), }验证请求可以写一个 Django management command或者直接在 shell 里跑python manage.py shellfrom apps.ai.services.llm_client import TaoTokenClient client TaoTokenClient() result client.chat([ {role: user, content: 用一句话说明什么是测试用例} ]) print(result)如果返回结构里包含正常的文本内容说明 Key 生效、请求链路通了。返回体里一般会有content数组取第一个元素的text字段就是模型输出。这一步跑通之后WHartTest 的用例生成、需求评审等模块就可以复用这个 client。对于需要长期跑编码任务或 Agent 场景的团队可以考虑 Coding Plan入口在 https://taotoken.net/coding-plan?utm_sourcetaotoken_aicg_blog_endutm_contentcoding_planutm_campaignrewrite 适合把模型调用纳入持续集成流程。5. 本篇常见错误排查接入过程中最容易碰到几类问题这里按现象、原因、处理方式列出来。第一类是 401 鉴权失败。现象是请求返回 401 或invalid api key。原因通常是环境变量没生效或者 Key 复制时带了空格。处理方式在 Django shell 里打印os.environ.get(TAOTOKEN_API_KEY)确认值存在且无多余字符如果用的是 Docker确认 environment 段写对了。第二类是 404 路径错误。现象是返回 404 或not found。原因多半是 base_url 拼错了比如多加了/v1或者少了/api。TaoToken 的 API 根地址是 https://taotoken.net/api 具体路径在代码里拼/v1/messages。检查base_url和请求路径的拼接结果。第三类是模型名不匹配。现象是返回model not found或类似提示。原因是你填的模型名不在 TaoToken 支持的列表里。处理方式去模型对话页面确认可用模型名或者查接入文档里的模型列表把 config.toml 里的default_model改成正确的名称。第四类是超时。现象是请求长时间无响应后抛 timeout。原因可能是网络链路波动或者max_tokens设得太大导致生成时间长。处理方式先把timeout调到 120 秒试一次如果稳定复现检查是不是模型本身响应慢换一个模型名再试。第五类是 Django 读不到配置。现象是启动时报KeyError: llm或LLM_CONFIG为空。原因通常是 config.toml 路径不对或者 tomllib 在 Python 3.11 以下不可用。处理方式确认 Python 版本3.11 以下用tomli替代确认BASE_DIR / config / config.toml文件真实存在。第六类是日志里出现明文 Key。现象是审计日志或错误日志里能看到完整 Key。处理方式把mask_api_key_in_logs设为 true并检查 client 里打印请求头的地方确保 Authorization 字段被脱敏。6. 统一 Key 之后的维护建议与入口汇总配置跑通之后日常维护会轻松很多。Key 只有一份轮换时改环境变量重启服务即可。模型切换改 config.toml 里的模型名不用动 Python 代码。用量和调用日志集中在 TaoToken 侧排查问题时先看审计日志再定位到具体模块。如果你在 WHartTest 里做的是接口自动化或 UI 自动化建议把模型调用和测试执行分开配置。模型调用走 TaoToken 统一通道测试执行走 WHartTest 自己的执行器两者互不干扰。这样即使模型侧有波动也不会影响已经生成好的用例执行。需要进一步操作的话几个入口按场景分流排查接入问题、管理 Key 去 API Keys 页面 https://taotoken.net/console/api-keys?utm_sourcetaotoken_aicg_blog_endutm_contentapi_keysutm_campaignrewrite 验证模型是否可用去模型对话 https://taotoken.net/models?utm_sourcetaotoken_aicg_blog_endutm_contentmodelsutm_campaignrewrite 查请求格式和参数说明去接入文档 https://taotoken.net/doc?utm_sourcetaotoken_aicg_blog_endutm_contentdocutm_campaignrewrite 长期编码或 Agent 任务看 Coding Plan https://taotoken.net/coding-plan?utm_sourcetaotoken_aicg_blog_endutm_contentcoding_planutm_campaignrewrite 。官网首页在 https://taotoken.net/?utm_sourcetaotoken_aicg_blog_endutm_mediumcsdnutm_campaignrewriteutm_content 需要整体了解时可以从这里进。