View a markdown version of this page

使用代理式检索查询知识库 - Amazon Bedrock
代理检索的工作原理先决条件使用代理检索查询知识库代理检索响应代理检索所需的权限注意事项

本文属于机器翻译版本。若本译文内容与英语原文存在差异，则一律以英文原文为准。

使用代理式检索查询知识库

代理检索使用基础模型智能地将复杂的查询分解为子查询，以迭代方式从知识库中检索相关信息，并评估检索到的结果是否足以回答原始查询。这种方法提高了复杂的多步骤问题的检索准确性，而单次检索过程可能无法完全解决这些问题。

例如，给定查询 “哪本杂志最先创办，《亚瑟杂志》还是《女性第一本杂志》？”，代理检索将其分解为单独的子查询，例如 “《亚瑟杂志》是什么时候成立的？” 以及 “First for Women 是什么时候成立的？”，检索每个结果，并评估合并后的结果是否足够。

代理检索的工作原理

当您向 AgenticRetrieveStream API 发送请求时，会发生以下过程：

  1. 规划 — 基础模型分析您的查询并创建计划以将其分解为一个或多个子查询。每个子查询都以您配置的特定检索器为目标。收集检索结果后，基础模型将评估它们是否足以回答原始查询。如果不是，它将计划并执行额外的检索迭代，但不超过配置的最大值。

  2. 检索 -子查询是针对配置的知识库检索器执行的。每次检索都会收集结果。

  3. 完整文档扩展 — 当基础模型确定需要文档的完整内容（例如，用于摘要、验证完整性或访问特定部分）时，它会调用 GetDocumentContent API 来检索完整的文档内容。

  4. 响应生成 -当设置generateResponse为true（默认）时，基础模型将从检索到的结果中合成自然语言答案。答案将通过responseEvent活动回传给您。

  5. 结果事件 — 所有迭代的去重检索结果、完整的合成自然语言答案和引文将返回给您。跟踪事件在整个过程中进行流式传输，以实现可观测性。

先决条件

在使用代理检索之前，必须具备以下条件：

  • 完全托管的亚马逊 Bedrock 知识库。代理检索目前仅支持托管知识库。

  • 访问亚马逊 Bedrock 中的基础模型，用于查询规划和评估。

  • 所需的 IAM 权限。有关更多信息，请参阅 代理检索所需的权限。

使用代理检索查询知识库

要使用代理检索，AgenticRetrieveStream请发送请求。响应是一个包含检索结果和跟踪事件的流。

下表描述了密钥请求字段：

必填字段
字段 说明
消息 输入查询和对话历史记录。每条消息都包含一个content带text值的role字段和一个字段（user或assistant）。
寻回犬 用于从中获取数据的知识库检索器。您最多可以指定 5 个检索器，每个检索器通过其 ID 指向托管知识库。每个检索器可以选择包括元数据过滤器和最大数量的结果。
代理的 RetrieveConfiguration 代理检索配置，包括用于查询规划和评估的基础模型，以及可选的重新排名模型和最大代理迭代次数。
可选字段
字段 说明
策略配置 将亚马逊基岩护栏配置为在代理检索期间使用。指定guardrailId和guardrailVersion。
用户上下文 为访问控制筛选提供用户上下文。
生成响应 一个布尔字段，当设置为true（默认）时，它指示基础模型从检索到的结果中生成自然语言答案。答案以文本块的形式回传并包含在结果事件中。

有关完整的请求和响应语法，请参阅 API 参考AgenticRetrieveStream中的。

代理检索响应

AgenticRetrieveStream响应是一个包含以下事件类型的流：

  • 结果事件 (AgenticRetrieveResultEvent)-处理完成时交付的最终事件。包含检索结果，如果启用响应生成，则包含生成的响应。结果事件包括：

    • 检索结果 (results)-在所有迭代中检索到的源块。每个结果都包括内容、源检索器标识符和可选元数据。当多个子查询检索到同一个区块时，它在最终结果中只出现一次。

    • 生成的响应 (generatedResponse)-当设置generateResponse为true（默认）时，结果事件包括一个包含以下内容的generatedResponse对象：

      • answer— 完整的合成自然语言答案文本。

      • citations— 一个可选列表，将答案的范围映射到支持的检索结果。每个引文都包含：

        • startIndex— 被引段落在字符串内开始的answer字符偏移量。

        • endIndex— 被引段落结尾处的字符偏移量（不包括——引用的文本从startIndex上到但不包括endIndex）。

        • references— 一个列表，其中每个参考文献都有一个resultIndex字段，该字段对同一结果事件的results数组进行索引，表明哪个检索结果支持引用的跨度。

  • 响应事件 (AgenticRetrieveResponseEvent)-当设置generateResponse为true（默认）时，将在响应生成期间流式传输responseEvent事件。每个事件都包含一个text字段，其中包含合成的自然语言答案的增量部分。

  • 跟踪事件 (AgenticRetrieveTraceEvent)-代理检索过程中流式传输的事件，提供对每个步骤的可见性。以下是跟踪事件的类型：

    • 规划 -表示基础模型正在分析查询并创建子查询。包括计划采取的行动和目标猎犬。

    • 检索 -表示正在对知识库执行检索。包括检索输入、输出以及任何警告或故障。

    • 推测性检索 — 在第一个计划步骤之前运行的初始检索，以减少延迟。对于单一知识库，它使用原始用户查询检索结果。对于多个知识库，它会执行探测搜索，以帮助将查询路由到相应的检索器。

    • 完整文档展开-表示代理正在检索特定文档的完整内容。包括文档 ID、来源检索器和状态（InProgress成功或失败）。

代理检索所需的权限

要使用 AgenticRetrieveStream API，调用的 IAM 身份必须具有以下权限：

{
    "Version": "2012-10-17",
    "Statement": [
        {
            "Effect": "Allow",
            "Action": "bedrock:AgenticRetrieveStream",
            "Resource": "*"
        },
        {
            "Effect": "Allow",
            "Action": [
                "bedrock:Retrieve",
                "bedrock:GetDocumentContent"
            ],
            "Resource": "arn:aws:bedrock:region:account-id:knowledge-base/*"
        },
        {
            "Effect": "Allow",
            "Action": "bedrock:InvokeModelWithResponseStream",
            "Resource": "*"
        }
    ]
}

如果您使用护栏进行代理检索，请添加以下权限：

{
    "Effect": "Allow",
    "Action": [
        "bedrock:GetGuardrail",
        "bedrock:ApplyGuardrail"
    ],
    "Resource": "*"
}

注意事项

使用代理检索时请记住以下几点：

  • 代理检索仅支持托管的亚马逊 Bedrock 知识库。

  • 有关每个请求的检索器配额、每次检索调用的结果以及最大代理迭代次数，请参阅。托管知识库的服务配额

  • 减少最大迭代次数可能会导致代理提前停止，这可能会降低复杂查询的准确性。

  • 配置护栏时，仅支持该BLOCK操作。代理检索不支持该MASK操作。

  • 客户提供并拥有代理检索期间使用的基础模型、嵌入模型和重新排序模型（如果提供）。您的 IAM 证书用于调用这些模型。