Object Storage Service:OSS SDK for Python 2.0 を使用したベクターのクエリ

権限

Alibaba Cloud アカウントには、デフォルトですべての権限があります。Alibaba Cloud アカウント配下の RAM ユーザーまたは RAM ロールには、デフォルトで権限がありません。Alibaba Cloud アカウントまたは管理者は、RAM ポリシーの概要またはバケットポリシーを使用して、この操作の権限を付与する必要があります。

メソッド定義

query_vectors(request: QueryVectorsRequest, **kwargs) → QueryVectorsResult

リクエストパラメーター

戻り値

メソッドの完全な定義については、「query_vectors」をご参照ください。

サンプルコード

import argparse
import alibabacloud_oss_v2 as oss
import alibabacloud_oss_v2.vectors as oss_vectors

parser = argparse.ArgumentParser(description="vector query vectors sample")
parser.add_argument('--region', help='バケットが配置されているリージョン。', required=True)
parser.add_argument('--bucket', help='バケットの名前。', required=True)
parser.add_argument('--endpoint', help='他のサービスが OSS にアクセスするために使用できるドメイン名')
parser.add_argument('--index_name', help='ベクターインデックスの名前。', required=True)
parser.add_argument('--account_id', help='アカウント ID。', required=True)

def main():
    args = parser.parse_args()

    # 環境変数から認証情報を読み込みます
    credentials_provider = oss.credentials.EnvironmentVariableCredentialsProvider()

    # SDK のデフォルト設定を使用します
    cfg = oss.config.load_default()
    cfg.credentials_provider = credentials_provider
    cfg.region = args.region
    cfg.account_id = args.account_id
    cfg.use_internal_endpoint = True  # パブリックネットワーク経由でサービスにアクセスするには、これを False に設定するか、この行を削除します。
    if args.endpoint is not None:
        cfg.endpoint = args.endpoint

    vector_client = oss_vectors.Client(cfg)

    query_filter = {
        "$and": [{
            "type": {
                "$nin": ["comedy", "documentary"]
            }
        }]
    }

    query_vector = {"float32": [0.1] * 128}

    result = vector_client.query_vectors(oss_vectors.models.QueryVectorsRequest(
        bucket=args.bucket,
        index_name=args.index_name,
        filter=query_filter,
        query_vector=query_vector,
        return_distance=True,
        return_metadata=True,
        top_k=10
    ))

    print(f'status code: {result.status_code},'
          f' request id: {result.request_id},'
          )

    if result.vectors:
        for vector in result.vectors:
            print(f'vector: {vector}')


if __name__ == "__main__":
    main()

関連ドキュメント

ベクターのクエリに関する完全なサンプルコードについては、「query_vectors.py」をご参照ください。