Cerebras
Cerebras Inference 在 Cerebras 推理系统上托管开放权重模型。AISIX 将这些模型置于统一的 OpenAI 兼容 API 之后,并集中管理上游凭证、调用方访问权限、限流和用量核算。
准备工作
开始前,请准备以下内容:
- 一套 AISIX 环境:
- 对于 AISIX Cloud,需要一个已关联网关的环境和具有写入作用域的 Admin Token。对于 On-Premises,请按照 AISIX Cloud 快速入门操作。如需申请 Hybrid Cloud 访问权限,请联系 API7。
- 对于开源 AISIX 网关,请准备本地 AISIX 安装,或使用开源 AISIX 网关快速入门中的 Docker 配置。配置网关以加载声明式资源文件。
- 从 Cerebras Cloud 控制台获取的 Cerebras API Key。
- 已安装
curl和jq。
使用 AISIX Cloud 配置
导出 AISIX Cloud 连接信息:
# AISIX_CP 是 Admin API 基础 URL;应包含 /api,且不含尾部斜杠
# 本地私有化部署快速入门使用 http://localhost:8080/api
export AISIX_CP="YOUR_AISIX_CLOUD_ADMIN_API_URL"
export AISIX_TOKEN="YOUR_ADMIN_TOKEN"
export ENV_ID="YOUR_ENVIRONMENT_ID"
为 Cerebras 支持的 Chat Completions 路由创建服务提供方密钥、模型别名和调用方 API Key。
Cerebras API 与 OpenAI 兼容,因此 AISIX 通过 openai 适配器连接,并使用 Cerebras API 根路径作为 api_base。
创建服务提供方密钥
创建用于存储 Cerebras 凭证和 API 根路径的服务提供方密钥:
# 请替换为实际值
export CEREBRAS_API_KEY="YOUR_PROVIDER_API_KEY"
PROVIDER_KEY_ID=$(curl -sS -X POST "$AISIX_CP/provider_keys" \
-H "Authorization: Bearer $AISIX_TOKEN" \
-H "Content-Type: application/json" \
-d '{
"display_name": "cerebras-prod",
"provider": "cerebras",
"api_key": "'"${CEREBRAS_API_KEY}"'",
"api_base": "https://api.cerebras.ai/v1",
"allowed_environments": ["'"${ENV_ID}"'"]
}' | jq -r '.provider_key.id')
echo "$PROVIDER_KEY_ID"
❶ provider 为 cerebras。AISIX Cloud Admin API 从目录服务提供方派生适配器;adapter 字段仅接受 BYO 服务提供方密钥。
❷ api_key 存储 Cerebras API Key。Cerebras 使用 HTTP Bearer 身份认证,这正是 openai 适配器已经发送的形式。该值遵循服务提供方密钥中的凭证处理行为。
❸ api_base 为 https://api.cerebras.ai/v1,与 Cerebras 为 OpenAI 客户端库记录的 baseURL 根路径相同。它已包含 /v1 路径,因此 AISIX 会直接向其追加 /chat/completions 等端点路径。对于 cerebras 目录服务提供方,此字段为可选项;省略时 AISIX Cloud Admin API 会填入相同的值。示例仍显式设置该字段,使配置中始终可以看到每个密钥指向的根路径。
该命令会把返回的服务提供方密钥 ID 保存到 PROVIDER_KEY_ID。
创建模型
Cerebras 模型 ID 是不带厂商前缀的裸标识符,即使权重来自其他厂商也是如此。OpenAI 开放权重模型在 Cerebras 上为 gpt-oss-120b,Google 模型为 gemma-4-31b。创建别名前,请在 Cerebras 模型目录中查看当前列表;该目录较短,并会随模型的新增和退役而轮换。当前 ID 包括 gpt-oss-120b、gemma-4-31b 和 zai-glm-4.7。
创建调用方将在请求中发送的模型别名:
MODEL_ID=$(curl -sS -X POST "$AISIX_CP/environments/$ENV_ID/models" \
-H "Authorization: Bearer $AISIX_TOKEN" \
-H "Content-Type: application/json" \
-d '{
"display_name": "cerebras-gptoss-prod",
"model_name": "gpt-oss-120b",
"provider_key_id": "'"${PROVIDER_KEY_ID}"'"
}' | jq -r '.model.id')
echo "$MODEL_ID"
❶ display_name 是调用方在 model 中发送的别名。
❷ model_name 是 Cerebras 模型 ID,例如 gpt-oss-120b 或 gemma-4-31b。不要从托管相同权重的其他服务提供方沿用 openai/gpt-oss-120b 等带前缀的 ID。
❸ provider_key_id 将别名关联到 Cerebras 服务提供方密钥。
创建调用方 API Key
创建能够访问该模型别名的调用方 API Key。明文密钥由服务器生成,并只在创建响应中返回一次,因此请立即保存:
AISIX_API_KEY=$(curl -sS -X POST "$AISIX_CP/environments/$ENV_ID/api_keys" \
-H "Authorization: Bearer $AISIX_TOKEN" \
-H "Content-Type: application/json" \
-d '{
"display_name": "cerebras-caller",
"allowed_models": ["'"${MODEL_ID}"'"]
}' | jq -r '.plaintext')
echo "$AISIX_API_KEY"
allowed_models 值必须引用上一步保存的模型 ID。写入后,配置会自动投射到已关联的网关。
使用开源 AISIX 网关配置
导出上游凭证,并选择应用将发送给网关的调用方 API Key:
export CEREBRAS_API_KEY="YOUR_PROVIDER_API_KEY"
export CALLER_API_KEY="YOUR_CALLER_API_KEY"
对于新网关,请使用此完整资源文件。对于现有网关,请将这些条目合并到其当前文件中,并保留其他资源:
_format_version: "1"
provider_keys:
- display_name: "cerebras-prod"
provider: "cerebras"
adapter: "openai"
api_key: ${CEREBRAS_API_KEY}
api_base: "https://api.cerebras.ai/v1"
models:
- display_name: "cerebras-gptoss-prod"
provider: "cerebras"
model_name: "gpt-oss-120b"
provider_key: "cerebras-prod"
api_keys:
- display_name: "cerebras-caller"
key_env: CALLER_API_KEY
allowed_models:
- "cerebras-gptoss-prod"
如果 AISIX 安装在本地,请在加载前验证文件:
aisix validate --resources resources.yaml
验证后,在网关进程环境中提供文件引用的环境变量,再启动网关。仅当这些变量已经可供进程使用时,才重新加载现有网关;否则,请使用更新后的环境重启网关。
如果使用 Docker,请调整开源 AISIX 网关快速入门中的验证和启动命令。挂载此 resources.yaml 文件,并在两条命令中使用 -e 传入它引用的每个环境变量。资源加载后,准备下文共用的验证请求:
export AISIX_API_KEY="$CALLER_API_KEY"