Amazon Bedrock
Amazon Bedrock 是 AWS 提供的托管式大模型平台,支持多种 Embedding 模型。seekdb 提供 AmazonBedrockEmbeddingFunction 封装,用于在写入 documents 或使用 query_texts 查询时,自动调用 Bedrock 生成向量并完成相似性检索。
提示
使用 Amazon Bedrock 服务需要遵循 AWS 的收费规则,可能产生相应费用。请在继续前,访问其官网或查阅相关文档,确认并接受其收费标准。如不同意,请勿继续操作。
依赖与鉴权
使用前需确保您的 AWS 账号已开通 Amazon Bedrock,并对目标 Embedding 模型具备调用权限。
安装
npm i seekdb @seekdb/amazon-bedrock
示例:创建 Amazon Bedrock Embedding Function
import { AmazonBedrockEmbeddingFunction } from "@seekdb/amazon-bedrock";
const ef = new AmazonBedrockEmbeddingFunction({
region: "us-east-1",
modelName: "amazon.titan-embed-text-v2",
});
配置项说明:
- apiKey:Bedrock API Key(可选;若不提供,则尝试从环境变量读取)
- apiKeyEnv:API Key 的环境变量名(默认:
"AMAZON_BEDROCK_API_KEY") - region:AWS Region(必填,例如
"us-east-1") - modelName:Embedding 模型名称(默认:
"amazon.titan-embed-text-v2")