diff --git a/milvus-plus-boot-starter/src/main/java/io/github/javpower/milvus/plus/core/conditions/LambdaSearchWrapper.java b/milvus-plus-boot-starter/src/main/java/io/github/javpower/milvus/plus/core/conditions/LambdaSearchWrapper.java index e84840e..036e130 100644 --- a/milvus-plus-boot-starter/src/main/java/io/github/javpower/milvus/plus/core/conditions/LambdaSearchWrapper.java +++ b/milvus-plus-boot-starter/src/main/java/io/github/javpower/milvus/plus/core/conditions/LambdaSearchWrapper.java @@ -330,10 +330,24 @@ public class LambdaSearchWrapper extends AbstractChainWrapper implements return this; } + public LambdaSearchWrapper annsField(String annsField){ + this.annsField=annsField; + return this; + } public LambdaSearchWrapper vector(List vector) { vectors.add(vector); return this; } + public LambdaSearchWrapper vector(String annsField,List vector) { + this.annsField=annsField; + vectors.add(vector); + return this; + } + public LambdaSearchWrapper vector(FieldFunction annsField,List vector) { + this.annsField=annsField.getFieldName(annsField); + vectors.add(vector); + return this; + } public LambdaSearchWrapper limit(Long limit) { this.setLimit(limit); return this; @@ -348,9 +362,11 @@ public class LambdaSearchWrapper extends AbstractChainWrapper implements */ private SearchReq build() { SearchReq.SearchReqBuilder builder = SearchReq.builder() - .collectionName(collectionName) - .annsField(annsField); + .collectionName(collectionName); + if(!annsField.isEmpty()){ + builder.annsField(annsField); + } if (!vectors.isEmpty()) { builder.data(vectors); }