Amazon Bedrock

Amazon Bedrock 是 AWS 提供的托管式大模型平台，支持多种 Embedding 模型。seekdb 提供了 Amazon Bedrock 的 Embedding Function 封装，支持在写入 documents 或使用 query_texts 查询时，自动调用 Bedrock 生成向量并完成相似性检索。

提示

使用 Amazon Bedrock 服务需要遵循 AWS 的收费规则，可能产生相应费用。请在继续前，访问其官网或查阅相关文档，确认并接受其收费标准。如不同意，请勿继续操作。

依赖与鉴权

AmazonBedrockEmbeddingFunction 通过 boto3 调用 Amazon Bedrock 的 Embedding 接口，因此需要：

• AWS 账号已开通 Amazon Bedrock，并对目标 Embedding 模型具备调用权限。
• AWS 凭证已按 boto3 的默认凭证链配置（例如环境变量、共享凭证文件、IAM Role 等）。
• 本地 Python 环境安装 pyseekdb 与 boto3。

示例：创建 Amazon Bedrock Embedding Function

import boto3
from pyseekdb.utils.embedding_functions import AmazonBedrockEmbeddingFunction

# 显式创建 boto3 Session
session = boto3.Session(profile_name="profile", region_name="us-east-1")

ef = AmazonBedrockEmbeddingFunction(
    session=session,
    model_name="amazon.titan-embed-text-v2",
)

参数说明：

• session：boto3.Session 实例，用于鉴权与地域配置。
• model_name：Amazon Bedrock 的 Embedding 模型名称。示例使用 amazon.titan-embed-text-v2；可根据您的账号权限与模型可用性进行调整。

提示

Amazon Bedrock 可用的模型与区域支持会随 AWS 更新而变化。更多信息请参考 Amazon Bedrock 官方文档：https://docs.aws.amazon.com/bedrock/。