Spring Boot集成Elasticsearch实现KNN搜索报错：double转boolean失败

报错原因 #

你使用的ScriptQueryBuilder属于过滤查询组件，它要求脚本返回boolean值（用于判断文档是否符合过滤条件），但你的脚本返回的是double类型的相似度得分，这直接触发了Cannot cast from [double] to [boolean]的类型转换异常。

同时你的实现思路存在核心偏差：

• KNN搜索针对的是向量字段（如dense_vector类型），直接对文本字段name做数值计算完全错误，且将字符串"Harry potter"强制转为double的操作本身就会引发异常。
• 遍历doc.keySet()的逻辑毫无意义，KNN不需要遍历所有字段，只需要针对目标向量字段计算距离。

正确实现方式 #

1. 先准备向量字段 #

如果要实现文本的语义KNN搜索，需要先把name字段的文本转换成向量，存储为dense_vector类型的字段（比如命名为name_vector）。可以通过Elasticsearch官方的文本嵌入插件（如Elastic Learned Sparse Encoder），或者提前用预训练模型生成向量存入索引。

2. 使用ES原生KNN查询 #

Elasticsearch 7.10+支持原生KNN查询，无需自定义脚本，直接调用API即可：

String hostname = "localhost";
int port = 9200;

RestHighLevelClient client = new RestHighLevelClient(
        RestClient.builder(new HttpHost(hostname, port, "http")));
SearchRequest searchRequest = new SearchRequest("book");
SearchSourceBuilder sourceBuilder = new SearchSourceBuilder();

// 替换为"Harry potter"对应的实际向量数组（由文本嵌入模型生成）
float[] queryVector = {0.12f, 0.34f, 0.56f, ...};

// 构建KNN查询，返回top3相似文档
KnnSearchBuilder knnSearch = KnnSearchBuilder.knn("name_vector")
        .vector(queryVector)
        .k(3);

sourceBuilder.query(knnSearch);
searchRequest.source(sourceBuilder);

SearchResponse searchResponse = client.search(searchRequest, RequestOptions.DEFAULT);
System.out.println(searchResponse.getHits());
client.close(); // 记得关闭客户端资源
return searchResponse.getHits();

3. 若仅需自定义文本匹配得分（非KNN） #

如果只是想基于文本匹配结果做自定义得分排序，使用FunctionScoreQueryBuilder而非ScriptQueryBuilder，它支持返回数值作为得分：

Map<String, Object> params = new HashMap<>();
params.put("targetText", "Harry potter");

// 脚本逻辑：匹配到目标文本则提升得分，否则得0分
String scriptSource = "return doc['name'].value.contains(params.targetText) ? 10.0 : 0.0;";

Script script = new Script(ScriptType.INLINE, "painless", scriptSource, params);

FunctionScoreQueryBuilder functionScoreQuery = QueryBuilders.functionScoreQuery(
        QueryBuilders.matchAllQuery(), // 可替换为match查询缩小范围
        new ScriptScoreFunctionBuilder(script)
);

sourceBuilder.query(functionScoreQuery);
searchRequest.source(sourceBuilder);

关键注意事项 #

• 文本字段无法直接执行KNN搜索，必须先转换为向量字段。
• ScriptQueryBuilder是用于过滤文档的工具，不能用来计算或调整得分。

内容的提问来源于stack exchange，提问作者Gnanaprakash S