跳到主要内容
版本:V1.1.0

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"