在 Amazon Bedrock 中使用重新排名模型 - Amazon Bedrock

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

在 Amazon Bedrock 中使用重新排名模型

您可以直接使用重新排名模型,也可以在知识库查询期间检索结果时使用。选择您首选方法的选项卡,然后按照以下步骤操作:

Console

您不能直接在中使用重新排名模型 AWS Management Console,但您可以通过执行以下操作在查询知识库时使用重新排名模型:

  1. 查询知识库时,选择 Three horizontal sliders with adjustable circular controls for settings or parameters. 图标打开 “配置” 窗格。

  2. 展开 “重新排名” 部分。

  3. 选择 “选择模型”,然后选择重新排名的模型。

  4. 如果您的 Amazon Bedrock 知识库服务角色缺少使用重新排名模型的权限,请选择更新服务角色以使用适当的权限修改该角色。

  5. (可选)在 “其他重新排名选项” 部分,修改所需的所有选项。

  6. 输入提示并选择 “运行”。响应是应用重新排名模型后的结果。

有关执行知识库查询的更多详细说明,请参阅查询知识库并检索数据查询知识库并根据检索到的数据生成响应

API

有关在知识库查询期间使用重新排名模型的说明,请参阅查询知识库并检索数据和。查询知识库并根据检索到的数据生成响应

要直接将重新排名模型与 Amazon Bedrock API 配合使用,请使用 Amazon Bedrock 运行时端点代理发送重新排名请求。

以下字段是必填字段:

字段 Basic description
查询 一个RerankQuery对象的数组。指定TEXT为,type并在textQuery字段中包含查询。
sources 要提交给重新排序模型的RerankSource对象数组。对于每个RerankSource,请指定INLINE为,type并在inlineDocumentSource字段中包含一个RerankDocument对象。有关详细信息,请参见下文RerankDocument
对配置进行重新排名 包括要使用的重新排名模型的 Amazon 资源名称 (ARN)、重新排名后要返回的结果数量,以及模型的推理配置(可选)。您可以将其他模型配置指定为键值对。有关更多信息,请参阅Cohere文档网站上的重新排名

以下字段是可选字段:

字段 应用场景
nextToken 在之前的响应中返回的标记,您可以将其包含在内以提供下一批结果。

包含的RerankSource对象的格式取决于文档的格式。要查看不同RerankSource类型的格式,请选择与文档格式对应的选项卡:

StringJSON object
String

如果文档是字符串,则将RerankDocument对象type字段的值指定为,TEXT并将文档包含在text字段中。例如:

{ "inlineDocumentSource": { "textDocument": { "text": "string" }, "type": "TEXT" }, "type": "INLINE" }
JSON object

如果文档是 JSON 对象,则将该RerankDocument对象中该type字段的值指定为,JSON并将该文档包含在jsonDocument字段中。例如:

{ "inlineDocumentSource": { "jsonDocument": JSON value, "type": "JSON" }, "type": "INLINE" }

对您的Rerank请求的响应会返回该results字段中的RerankResult对象列表。每个对象都包含以下字段:

  • document— 包括有关您提交的文档的信息。

  • relevanceScore— 文档的相关性分数,由重新排序模型分配。

  • index— 表示文档相对于列表中其他文档的排名。分数越低,排名越高。

如果要显示的结果太多,则响应会在nextToken字段中返回一个值。在这种情况下,要查看下一批结果,请在后续请求中包含该令牌。