関連性ブースティング

ブースティングを使うと、特定のメタデータ特性を持つドキュメントへ検索結果を寄せられます。たとえば、新しいドキュメントを優先したり、優先度の高いページを出したり、下書きの優先度を下げたりできます。ブースティングは意味的な関連性を置き換えず、結果を再ランクします。

ブースティングは、最初の取得ステップのあと、reranking（有効な場合）の前に適用されます。

1. 検索: AI Search はベクトル検索、キーワード検索、またはその両方で、最大 50 件の候補チャンクを取得します。
2. ブースト: 各候補を、boost_by で指定したメタデータフィールドで再スコアします。ブーストは元の取得スコアに加算されます。
3. Rerank: reranking が有効な場合、ブースト後の結果を reranking モデルで再ランクします。
4. 返却: 上位 max_num_results 件を返します。

ブースティングは候補セット内の順序を変えられますが、最初の検索ステップで取得しなかったチャンクを引き上げることはできません。

組み込みの timestamp フィールド、または カスタムメタデータスキーマ で定義した任意のフィールドでブーストできます。

方向は、フィールド値が各結果のランキングにどう影響するかを制御します。

direction を省略すると、AI Search はフィールド型に応じたデフォルトを適用します。

text または boolean フィールドに asc または desc を使うと、エラーになります。

インスタンスの作成または更新時に、boost_by を最大 3 件のオブジェクトの配列として指定します。各オブジェクトは一意のフィールドを参照する必要があります。

const instance = await env.AI_SEARCH.create({
	id: "my-instance",
	retrieval_options: {
		boost_by: [
			{ field: "timestamp", direction: "desc" },
			{ field: "priority", direction: "desc" },
		],
	},
});

ブースティングを外すには、インスタンス更新時に boost_by を空配列にします。

個別リクエストでは、ai_search_options.retrieval で boost_by を上書きできます。リクエスト単位の値は、インスタンスレベルのデフォルトを完全に置き換えます。

const instance = env.AI_SEARCH.get("my-instance");

const results = await instance.search({
	messages: [{ role: "user", content: "What is Cloudflare?" }],
	ai_search_options: {
		retrieval: {
			boost_by: [{ field: "timestamp", direction: "desc" }],
		},
	},
});

1 件のリクエストだけブースティングを無効にするには、空配列を渡します。

const results = await instance.search({
	messages: [{ role: "user", content: "What is Cloudflare?" }],
	ai_search_options: {
		retrieval: {
			boost_by: [],
		},
	},
});

関連性ブースティングのよくある使い方は次のとおりです。

• リクエストあたりのブーストフィールドは最大 3 つです。
• フィールド名は、カスタムメタデータスキーマのフィールド、または組み込みの timestamp フィールドと一致する必要があります。
• text と boolean フィールドが使える方向は exists と not_exists だけです。
• 1 リクエスト内のブーストフィールドは一意である必要があります。
• ブースティングは、最初の検索で得た候補セットを再ランクします。取得しなかったドキュメントを出すことはできません。