写入与查询自动Embedding

SearchRequest searchRequest = new SearchRequest();
SearchSourceBuilder searchSourceBuilder = new SearchSourceBuilder();
Map<String, Object> queryBody = Map.of(
  "knn", Map.of(
    "text_field_embedding", Map.of(
      "query_text", "存储卡",
      "k", 10
    )
  )
);
searchSourceBuilder.size(10);
searchSourceBuilder.query(QueryBuilders.wrapperQuery(new Gson().toJson(queryBody)));
Map<String, String> ext = Map.of("ef_search", "200");
searchSourceBuilder.ext(Collections.singletonList(new LVectorExtBuilder("lvector", ext)));
searchRequest.source(searchSourceBuilder);
searchRequest.indices("search_vector_test");
SearchResponse searchResponse = client.search(searchRequest, RequestOptions.DEFAULT);
System.out.println(searchResponse);

search responseBody = {
  "took": 46,
  "timed_out": false,
  "terminated_early": false,
  "num_reduce_phases": 0,
  "_shards": {
    "total": 2,
    "successful": 2,
    "skipped": 0,
    "failed": 0
  },
  "hits": {
    "total": {
      "value": 7,
      "relation": "eq"
    },
    "max_score": 0.7433592,
    "hits": [
      {
        "_index": "search_vector_test",
        "_id": "8195",
        "_score": 0.7433592,
        "_source": {
          "text_field": "品牌E Class4 32G TF卡(micro SD)手机存储卡",
          "merit": "容量挺大的、速度快、好用、质量好",
          "tag": [
            "存储设备",
            "存储卡",
            "SD卡"
          ],
          "brand": "品牌E"
        }
      },
      {
        "_index": "search_vector_test",
        "_id": "14103",
        "_score": 0.7116537,
        "_source": {
          "text_field": "品牌B 64GB至尊高速移动存储卡 UHS-1制式 读写速度最高可达30MB",
          "merit": "容量挺大的、速度快、好用",
          "tag": [
            "存储设备",
            "存储卡",
            "SD卡"
          ],
          "brand": "品牌B"
        }
      },
      {
        "_index": "search_vector_test",
        "_id": "13316",
        "_score": 0.6831677,
        "_source": {
          "text_field": "品牌E 101 G2 32GB 优盘",
          "merit": "好用、容量挺大的、速度快",
          "tag": [
            "存储设备",
            "U盘",
            "优盘"
          ],
          "brand": "品牌E"
        }
      },
      {
        "_index": "search_vector_test",
        "_id": "3982",
        "_score": 0.64234203,
        "_source": {
          "text_field": "品牌A 时尚节能无线鼠标(草绿)(眩光.悦动.时尚炫舞鼠标 12个月免换电池 高精度光学寻迹引擎 超细微接收器10米传输距离)",
          "merit": "好用、外观漂亮",
          "tag": [
            "鼠标",
            "电子产品"
          ],
          "brand": "品牌A"
        }
      },
      {
        "_index": "search_vector_test",
        "_id": "6797",
        "_score": 0.6357207,
        "_source": {
          "text_field": "品牌D 两刀头充电式电动剃须刀",
          "merit": "好用、外观漂亮",
          "tag": [
            "家用电器",
            "电动剃须刀"
          ],
          "brand": "品牌D"
        }
      },
      {
        "_index": "search_vector_test",
        "_id": "323519",
        "_score": 0.62445086,
        "_source": {
          "text_field": "品牌B 光学鼠标(经典黑)(智能自动对码/1000DPI高精度光学引擎)",
          "merit": "质量好、到货速度快、外观漂亮、好用",
          "tag": [
            "鼠标",
            "电子产品"
          ],
          "brand": "品牌B"
        }
      },
      {
        "_index": "search_vector_test",
        "_id": "300265",
        "_score": 0.62144196,
        "_source": {
          "text_field": "品牌C 耳塞式耳机 白色(经典时尚)",
          "merit": "外观漂亮、质量好",
          "tag": [
            "耳机",
            "电子产品"
          ],
          "brand": "品牌C"
        }
      }
    ]
  }
}

SearchRequest searchRequest = new SearchRequest();
SearchSourceBuilder searchSourceBuilder = new SearchSourceBuilder();
Map<String, Object> queryBody = Map.of(
  "knn", Map.of(
    "text_field_embedding", Map.of(
      "query_text", "存储卡",
      "k", 10,
      "filter", Map.of(
        "bool", Map.of(
          "filter", List.of(
            Map.of("match", Map.of("merit", "质量好")),
            Map.of("term", Map.of("brand", "品牌E")),
            Map.of("terms", Map.of("tag", List.of("SD卡", "存储卡")))
          )
        )
      )
    )
  )
);

searchSourceBuilder.size(10);
searchSourceBuilder.query(QueryBuilders.wrapperQuery(new Gson().toJson(queryBody)));
Map<String, String> ext = Map.of("filter_type", "efficient_filter", "ef_search", "200");
searchSourceBuilder.ext(Collections.singletonList(new LVectorExtBuilder("lvector", ext)));
searchRequest.source(searchSourceBuilder);
searchRequest.indices("search_vector_test");
SearchResponse searchResponse = client.search(searchRequest, RequestOptions.DEFAULT);
System.out.println(searchResponse);

SearchRequest searchRequest = new SearchRequest();
SearchSourceBuilder searchSourceBuilder = new SearchSourceBuilder();
Map<String, Object> queryBody = Map.of(
  "knn", Map.of(
    "text_field_embedding", Map.of(
      "query_text", "存储卡",
      "filter", Map.of(
        "bool", Map.of(
          "must", List.of(
            Map.of(
              "bool", Map.of(
                "must", List.of(
                  Map.of("match", Map.of("text_field", Map.of("query", "存储卡")))
                )
              )
            ),
            Map.of(
              "bool", Map.of(
                "filter", List.of(
                  Map.of("match", Map.of("merit", "质量好")),
                  Map.of("term", Map.of("brand", "品牌E")),
                  Map.of("terms", Map.of("tag", List.of("SD卡", "存储卡")))
                )
              )
            )
          )
        )
      ),
      "k", 10
    )
  )
);
searchSourceBuilder.size(10);
searchSourceBuilder.query(QueryBuilders.wrapperQuery(new Gson().toJson(queryBody)));
Map<String, String> ext = Map.of(
  "filter_type", "efficient_filter",
  "hybrid_search_type", "filter_rrf",
  "rrf_rank_constant", "1",
  "ef_search", "200");
searchSourceBuilder.ext(Collections.singletonList(new LVectorExtBuilder("lvector", ext)));
searchRequest.source(searchSourceBuilder);
searchRequest.indices("search_vector_test");
SearchResponse searchResponse = client.search(searchRequest, RequestOptions.DEFAULT);
System.out.println(searchResponse);