メインコンテンツまでスキップ
バージョン: User Guides (Cloud)

主キー検索

類似性検索を実行する場合、クエリベクトルがターゲットコレクションにすでに存在する場合でも、常に1つ以上のクエリベクトルを提供するよう求められます。検索前にベクトルを取得するのを避けるために、代わりに主キーを使用できます。

概要​

eコマースプラットフォームでは、ユーザーはキーワードを入力して、それに一致する製品を取得できます。ユーザーが製品詳細ページを表示すると、プラットフォームは、比較したいユーザーのために、ページの最下部に類似製品のリストも表示します。

推奨事項は、キーワードまたは現在の製品との類似性によってソートされます。これを実現するために、プラットフォーム開発者は、実際の類似性検索の前に、キーワードまたは現在の製品のベクトル表現をMilvusから取得する必要があります。これにより、プラットフォームとMilvus間の往復が増加し、大量の高次元浮動小数点数がネットワークを介して送信されます。

アプリケーションとMilvus間のインタラクションロジックを簡素化し、往復回数を減らし、大量の高次元浮動小数点値をネットワークを介して送信するのを避けるために、主キー検索の使用を検討してください。

主キー検索では、クエリベクトルを提供する必要はありません。代わりに、クエリベクトルを含むエンティティの主キー（ids）を提供するよう求められます。

制限事項​

  • 主キーを使用した検索は、BM25関数のようにVarCharフィールドから派生したスパースベクトルフィールドを除く、すべてのベクトルデータ型に適用されます。

  • フィルタリング、範囲、およびグループ化検索で、オプションでページネーションを有効にして、クエリベクトルの代わりに主キーを使用できます。ただし、この機能はハイブリッド検索および検索イテレータには適用されません。

  • 埋め込みリストを含む類似性検索の場合、クエリベクトルを取得し、それらを埋め込みリストに配置し、検索を実行する必要があります。

  • 存在しない主キーまたは誤った形式の主キーの場合、Milvusはエラーを通知します。

  • 主キーとクエリベクトルは相互に排他的です。両方を提供するとエラーが発生します。

例​

以下の例では、提供されるすべてのInt64 IDがターゲットコレクションで利用可能であることを前提としています。

📘Notes

主キーはフィルタリングには使用されません。ベクトル取得にのみ使用されます。

基本的な主キー検索を実行するには、クエリベクトルを主キーに置き換えるだけです。

from pymilvus import MilvusClient

client = MilvusClient(
    uri="YOUR_CLUSTER_ENDPOINT",
    token="YOUR_CLUSTER_TOKEN"
)

res = client.search(
    collection_name="my_collection",
    anns_field="vector",
    ids=[551, 296, 43], # a list of primary keys
    limit=3,
    search_params={"metric_type": "IP"}
)

for hits in res:
    for hit in hits:
        print(hit)

例2：主キーを使用したフィルタリング検索​

以下の例では、colorとlikesがターゲットコレクションでスキーマ定義された2つのフィールドであると仮定しています。

res = client.search(
    collection_name="my_collection",
    ids=[551, 296, 43], #
    filter='color like "red%" and likes > 50',
    output_fields=["color", "likes"],
    limit=3,
)

例3：主キーを使用した範囲検索​

res = client.search(
    collection_name="my_collection",
    ids=[551, 296, 43],
    limit=3,
    search_params={
        "params": {
            "radius": 0.4,
            "range_filter": 0.6
        }
    }
)

例 4: 主キーを使用したグループ化検索​

以下の例では、docId がターゲットコレクションのスキーマ定義フィールドであることを前提としています。

res = client.search(
    collection_name="my_collection",
    ids=[551, 296, 43],
    limit=3,
    group_by_field="docId",
    output_fields=["docId"]
)