DashVector:ドキュメントの検索

import com.aliyun.dashvector.DashVectorClient;
import com.aliyun.dashvector.DashVectorCollection;
import com.aliyun.dashvector.common.DashVectorException;
import com.aliyun.dashvector.models.Doc;
import com.aliyun.dashvector.models.Vector;
import com.aliyun.dashvector.models.requests.QueryDocRequest;
import com.aliyun.dashvector.models.responses.Response;

import java.util.Arrays;
import java.util.List;

public class Main {
    public static void main(String[] args) throws DashVectorException {
        DashVectorClient client = new DashVectorClient("YOUR_API_KEY", "YOUR_CLUSTER_ENDPOINT");
        DashVectorCollection collection = client.get("quickstart");
        
        // ベクトルを構築します。
        Vector vector = Vector.builder().value(Arrays.asList(0.1f, 0.2f, 0.3f, 0.4f)).build();
      	
      	// QueryDocRequest オブジェクトを構築します。
      	QueryDocRequest request = QueryDocRequest.builder()
          .vector(vector)
          .topk(100)
          .includeVector(true)
          .build();
        
        // ドキュメントを検索します。
        Response<List<Doc>> response = collection.query(request);
      
      	// 操作が成功したかどうかを確認します。
      	assert response.isSuccess() 

        System.out.println(response);
        // 出力例：
        // {
        //   "code":0,
        //   "message":"Success",
        //   "requestId":"b26ce0b8-0caf-4836-8136-df889d79ae91",
        //   "output":[
        //     {
        //       "id":"1",
        //       "vector":{
        //         "value":[
        //           0.10000000149011612,
        //           0.20000000298023224,
        //           0.30000001192092896,
        //           0.4000000059604645
        //         ]
        //       },
        //       "fields":{
        //         "name":"zhangsan",
        //         "age":20,
        //         "weight":100.0,
        //         "anykey1":"String",
        //         "anykey2":1,
        //         "anykey3":true,
        //         "anykey4":3.1415926
        //       },
        //       "score":1.1920929E-7
        //     }
        //   ]
        // }
    }
}

// QueryDocRequest オブジェクトを構築します。
QueryDocRequest request = QueryDocRequest.builder()
  .id("1")
  .topk(100)
  .outputFields(Arrays.asList("name", "age")) // name フィールドと age フィールドのみを返す必要があります。
  .includeVector(true)
  .build();

// プライマリキーに関連付けられたベクトルに基づいて類似検索を実行します。
Response<List<Doc>> response = collection.query(request);

// 検索が成功したかどうかを確認します。
assert response.isSuccess()

Vector vector = Vector.builder().value(Arrays.asList(0.1f, 0.2f, 0.3f, 0.4f)).build();

// QueryDocRequest オブジェクトを構築します。
QueryDocRequest request = QueryDocRequest.builder()
  .vector(vector) // 検索するベクトルを指定します。または、検索するプライマリキーを指定することもできます。
  .topk(100)
  .filter("age > 18") // age フィールドの値が 18 より大きいドキュメントに対して一致クエリを実行するための条件フィルターを指定します。
  .outputFields(Arrays.asList("name", "age")) // name フィールドと age フィールドのみを返す必要があります。
  .includeVector(true)
  .build();

// 条件フィルターと指定されたベクトルまたはプライマリキーを使用して類似検索を実行します。
Response<List<Doc>> response = collection.query(request);

// 検索が成功したかどうかを確認します。
assert response.isSuccess()

Vector vector = Vector.builder().value(Arrays.asList(0.1f, 0.2f, 0.3f, 0.4f)).build();

// 疎ベクトルを含む QueryDocRequest オブジェクトを構築します。
QueryDocRequest request = QueryDocRequest.builder()
  .vector(vector) // 検索するベクトルを指定します。
  .sparseVector(
  	new Map<Integer, Float>() {
      {
        put(1, 0.4f);
        put(10000, 0.6f);
        put(222222, 0.8f);
      }
  	}) // 疎ベクトルを指定します。
  .build();

// 密ベクトルと疎ベクトルの両方を使用して類似検索を実行します。
Response<List<Doc>> response = collection.query(request);

// 検索が成功したかどうかを確認します。
assert response.isSuccess()

// 条件フィルターを含むが、プライマリキーまたはベクトルを含まない QueryDocRequest オブジェクトを構築します。
QueryDocRequest request = QueryDocRequest.builder()
  .topk(100)
  .filter("age > 18") // age フィールドの値が 18 より大きいドキュメントに対して一致クエリを実行するための条件フィルターを指定します。
  .outputFields(Arrays.asList("name", "age")) // name フィールドと age フィールドのみを返す必要があります。
  .includeVector(true)
  .build();

// ベクトルもプライマリキーも指定されていないため、条件フィルターのみに基づいて検索を実行します。
Response<List<Doc>> response = collection.query(request);

// 検索が成功したかどうかを確認します。
assert response.isSuccess()