Snowflake Cortex
Snowflake Cortex 将托管基础模型引入 Snowflake 账户,并通过 REST API 公开这些模型。AISIX 为应用提供一个 OpenAI 兼容 API,用于访问你账户中可用的模型。
准备工作
开始前,请准备以下内容:
- 一套 AISIX 环境:
- 对于 AISIX Cloud,需要一个已关联网关的环境和具有写入作用域的 Admin Token。对于 On-Premises,请按照 AISIX Cloud 快速入门操作。如需申请 Hybrid Cloud 访问权限,请联系 API7。
- 对于开源 AISIX 网关,请准备本地 AISIX 安装,或使用开源 AISIX 网关快速入门中的 Docker 环境。配置网关以加载声明式资源文件。
- Snowflake 账户标识符和编程访问 Token(PAT)。
- 允许使用 Cortex 的 Snowflake 角色。Snowflake 文档说明了
SNOWFLAKE.CORTEX_USER数据库角色和 REST API 用户角色要求。 - 在你的 Snowflake 区域中访问所选模型的权限。
curl和jq。
导出 Snowflake 连接信息:
export SNOWFLAKE_ACCOUNT="example-account"
export SNOWFLAKE_API_BASE="https://${SNOWFLAKE_ACCOUNT}.snowflakecomputing.com/api/v2/cortex/v1"
export SNOWFLAKE_PAT="YOUR_SNOWFLAKE_PROGRAMMATIC_ACCESS_TOKEN"
使用 Snowflake 连接信息中的账户主机名。SNOWFLAKE_ACCOUNT 中不要包含 https:// 或其他路径。
使用 AISIX Cloud 配置
导出 AISIX Cloud 连接信息:
# AISIX_CP 是 Admin API 基础 URL;应包含 /api,且末尾不带斜杠
# 本地 On-Premises 快速入门使用 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"
Snowflake 是提供 OpenAI 兼容 Cortex 端点的社区目录服务提供方。AISIX 通过 openai 适配器连接,使用 Bearer Token 对上游请求进行身份认证,并将账户专用主机名用作 api_base。
创建服务提供方密钥
PROVIDER_KEY_ID=$(
curl -sS -X POST "$AISIX_CP/provider_keys" \
-H "Authorization: Bearer $AISIX_TOKEN" \
-H "Content-Type: application/json" \
-d '{
"display_name": "snowflake-cortex-prod",
"provider": "snowflake-cortex",
"api_key": "'"${SNOWFLAKE_PAT}"'",
"api_base": "'"${SNOWFLAKE_API_BASE}"'",
"allowed_environments": ["'"${ENV_ID}"'"]
}' | jq -er '.provider_key.id'
)
echo "$PROVIDER_KEY_ID"
snowflake-cortex 是准确的目录 ID。snowflake 不是它的别名。AISIX 会推导出 openai 适配器,因此请省略 adapter 字段。
实际使用时必须提供账户专用的 api_base,因为 AISIX 无法推断哪个 Snowflake 账户应接收请求。AISIX 会在此 Root 后追加 /chat/completions。
AISIX_TOKEN 是 AISIX 管理 Token。SNOWFLAKE_PAT 是存储在服务提供方密钥中的上游凭证。尽管二者都可能被称为 PAT,但它们是互不相关的 Token 类型。
创建模型
为账户中可用的 Cortex 模型创建别名:
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": "snowflake-claude-prod",
"model_name": "claude-sonnet-4-5",
"provider_key_id": "'"${PROVIDER_KEY_ID}"'"
}' | jq -er '.model.id'
)
echo "$MODEL_ID"
Snowflake 支持的模型列表因区域和版本而异。只能将 claude-sonnet-4-5 替换为你账户可用的 ID,并从当前的 Cortex 模型可用性参考中复制其准确拼写。
创建调用方 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": "snowflake-cortex-caller",
"allowed_models": ["'"${MODEL_ID}"'"]
}' | jq -er '.plaintext'
)
echo "$AISIX_API_KEY"