StructArray を使ったフィルタ付き検索
このページでは、StructArray フィールド上のベクトル検索にスカラー フィルタリングを追加する方法を説明します。StructArray のフィルタリングには 2 つのレベルがあります。行レベルのフィルタは親エンティティを選択し、要素レベルのフィルタは要素レベルのベクトル検索に参加する Struct 要素を制約します。
このページでは、StructArray フィールドの作成 にある tech_articles collection を使用します。この collection には chunks という名前の StructArray フィールドがあり、section、page、quality_score、has_code などのスカラー サブフィールドに加え、検索用のベクトル サブフィールドも含まれています。
フィルタの種類を選択する
| 目的 | 使用するもの | 結果の動作 |
|---|---|---|
category のようなトップレベルのスカラー フィールドでフィルタする。 | 通常の filter 式。 | 検索の前または途中で親エンティティを選択します。 |
| スカラー条件に一致する Struct 要素に要素レベルのベクトル検索を制約する。 | element_filter. | 一致する Struct 要素のみを検索し、一致した要素のオフセットを返すことができます。 |
| いずれか、すべて、または特定数の Struct 要素が述語に一致するかどうかでエンティティを選択する。 | MATCH_ANY、MATCH_ALL、MATCH_LEAST、MATCH_MOST、または MATCH_EXACT。 | 行レベルのフィルタリングです。これらの演算子は単独ではオフセットを返しません。 |
このページでは、検索ワークフローで StructArray フィルタを使用する方法を説明します。完全な構文ルール、サポートされる述語の種類、サポートされない述語の一覧については、StructArray Operators を参照してください。
トップレベル フィールドでフィルタする
条件が個々の Struct 要素ではなく親エンティティに属する場合は、通常の filter 式を使用します。これは EmbeddingList 検索と要素レベル検索の両方で機能します。
from pymilvus import MilvusClient
from pymilvus.client.embedding_list import EmbeddingList
client = MilvusClient(
uri="YOUR_CLUSTER_ENDPOINT",
token="YOUR_CLUSTER_TOKEN",
)
query = EmbeddingList()
query.add([0.12, 0.21, 0.32, 0.44])
query.add([0.18, 0.23, 0.29, 0.36])
results = client.search(
collection_name="tech_articles",
data=[query],
anns_field="chunks[emb_list_vector]",
filter='category == "search"',
limit=3,
output_fields=[
"doc_id",
"title",
"category",
"chunks[text]",
"chunks[section]",
],
)
上記のフィルタは、トップレベルの category フィールドが "search" のエンティティのみを選択します。これによって一致した Struct 要素を 1 つ特定することはありません。
要素レベルのベクトル検索をフィルタする
スカラー条件を、要素レベルのベクトル検索に参加する同じ Struct 要素に適用する必要がある場合は、element_filter(structArrayField, predicate) を使用します。述語の内部では、現在の Struct 要素のスカラー サブフィールドを参照するために $[subfield] を使用します。
query_vector = [0.19, 0.24, 0.30, 0.37]
filter_expr = (
'category == "search" && '
'element_filter(chunks, '
'$[section] == "index" && '
'$[quality_score] > 0.9 && '
'$[has_code] == true)'
)
results = client.search(
collection_name="tech_articles",
data=[query_vector],
anns_field="chunks[emb]",
filter=filter_expr,
limit=5,
output_fields=[
"doc_id",
"title",
"chunks[text]",
"chunks[section]",
"chunks[page]",
"chunks[quality_score]",
"chunks[has_code]",
],
)
for hits in results:
for hit in hits:
print(
"doc_id:", hit["id"],
"distance:", hit["distance"],
"offset:", hit.get("offset"),
"entity:", hit["entity"],
)
この例では、トップレベルの述語 category == "search" が候補エンティティを選択し、element_filter が要素レベルのベクトル検索を、section、quality_score、has_code が同じ Struct 要素内ですべて一致する chunk のみに制限します。
トップレベルの述語と element_filter を組み合わせる場合は、式の末尾に element_filter を配置してください。1 つの filter 式に含められる element_filter は 1 つだけであり、element_filter や MATCH_* を別の StructArray 演算子の中にネストすることはできません。
MATCH 演算子でエンティティをフィルタする
フィルタが、その Struct 要素に基づいて親エンティティが条件を満たすかどうかを判断する必要がある場合は、MATCH_* 演算子を使用します。これらの演算子は行レベルのフィルタです。つまり、エンティティを選択しますが、単独では要素オフセットを返しません。
| Operator | 使用する場面 | 例 |
|---|---|---|
MATCH_ANY | 少なくとも 1 つの Struct 要素が述語を満たす必要がある場合。 | MATCH_ANY(chunks, $[section] == "index") |
MATCH_ALL | すべての Struct 要素が述語を満たす必要がある場合。 | MATCH_ALL(chunks, $[quality_score] > 0.5) |
MATCH_LEAST | 少なくとも N 個の Struct 要素が述語を満たす必要がある場合。 | MATCH_LEAST(chunks, $[has_code] == true, threshold=2) |
MATCH_MOST | 最大で N 個の Struct 要素が述語を満たす必要がある場合。 | MATCH_MOST(chunks, $[section] == "appendix", threshold=1) |
MATCH_EXACT | ちょうど N 個の Struct 要素が述語を満たす必要がある場合。 | MATCH_EXACT(chunks, $[section] == "summary", threshold=1) |
filter_expr = (
'category == "search" && '
'MATCH_ANY(chunks, $[section] == "index" && $[quality_score] > 0.9)'
)
results = client.search(
collection_name="tech_articles",
data=[query],
anns_field="chunks[emb_list_vector]",
filter=filter_expr,
limit=3,
output_fields=[
"doc_id",
"title",
"category",
"chunks[text]",
"chunks[section]",
"chunks[quality_score]",
],
)
ここで MATCH_ANY を使用するのは、EmbeddingList 検索の結果がエンティティ レベルだからです。フィルタは、エンティティ内に少なくとも 1 つ "index" で高品質な chunk が存在することを要求しますが、検索結果自体は依然として親エンティティを表します。
ハイブリッド検索でフィルタを使用する
ハイブリッド検索では、条件を適用したい場所に StructArray フィルタを適用します。トップレベルのフィルタはハイブリッド検索全体で共有できます。element_filter は、要素レベルの制約が必要な StructArray 要素レベルのリクエストに付与する必要があります。
from pymilvus import AnnSearchRequest, RRFRanker
query_vector = [0.19, 0.24, 0.30, 0.37]
title_req = AnnSearchRequest(
data=[query_vector],
anns_field="title_vector",
limit=10,
)
chunk_req = AnnSearchRequest(
data=[query_vector],
anns_field="chunks[emb]",
limit=10,
expr='element_filter(chunks, $[section] == "index" && $[quality_score] > 0.9)',
)
results = client.hybrid_search(
collection_name="tech_articles",
reqs=[title_req, chunk_req],
ranker=RRFRanker(),
filter='category == "search"',
limit=5,
output_fields=[
"doc_id",
"title",
"category",
"chunks[text]",
"chunks[section]",
"chunks[quality_score]",
],
)
filter 引数はトップレベルのエンティティ条件を適用し、chunk_req の expr は StructArray の要素レベル ベクトル リクエストのみに制約を適用します。サポートされるハイブリッド検索の組み合わせとバージョン固有の制限については、Hybrid Search with StructArray および StructArray Limits を参照してください。
述語サポートの概要
StructArray の述語ではスカラー サブフィールドを使用してください。ベクトル サブフィールドはスカラー述語の入力としては使えません。
| サブフィールド型 | 代表的な述語の例 |
|---|---|
BOOL | $[has_code] == true, !($[has_code] == true) |
| 整数型 | $[page] >= 2, $[page] in [1, 2, 3] |
FLOAT, DOUBLE | $[quality_score] > 0.9, 0.7 < $[quality_score] < 0.95 |
VARCHAR | $[section] == "index", $[text] like "range%" |
| ベクトル サブフィールド | $[...] のスカラー述語入力としてはサポートされていません。代わりにベクトル検索を通じてベクトル サブフィールドを使用してください。 |
JSON パス、配列コンテナ関数、テキスト一致関数、$[...] に対する null 述語、Geometry 関数、Timestamptz 式、一般的な関数呼び出しなどのサポートされないケースについては、StructArray Operators を参照してください。
よくある間違い
-
element_filterやMATCH_*の外側で$[subfield]を使う。 -
element_filter(chunks, $[section] == "index")のような StructArray 演算子構文ではなく、chunks.sectionを使う。 -
行レベルのフィルタリングだけでよいのに
element_filterを使う。エンティティを選択するだけでよい場合は、代わりにMATCH_ANYを使ってください。 -
MATCH_*が要素オフセットを返すと期待する。これらの演算子はエンティティを選択しますが、単独では一致した要素を 1 つ特定しません。 -
$[has_code]のような裸のブール述語を書く。$[has_code] == trueのような明示的な比較を使用してください。 -
同じ filter 式の中で、トップレベルの述語より前に
element_filterを置く。
次のステップ
-
StructArray フィルタの完全な構文を確認するには、StructArray Operators を参照してください。
-
まずフィルタなしのベクトル検索を実行するには、Basic Vector Search with StructArray を参照してください。
-
よく使う StructArray フィルタ用のスカラー index を作成するには、Index StructArray Fields を参照してください。
-
バージョン固有のフィルタおよび検索の制限を確認するには、StructArray Limits を参照してください。