admin/MilvusPlus
1
0
Fork 0
mirror of https://gitee.com/dromara/MilvusPlus.git synced 2025-12-08 01:48:23 +08:00
Code Issues Packages Projects Releases Wiki Activity

添加annsField

Browse Source
This commit is contained in:
xgc 2024-05-11 11:53:04 +08:00
parent d95c117d2e
commit 2179daf22d
1 changed files with 18 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
milvus-plus-boot-starter/src/main/java/io/github/javpower/milvus/plus/core/conditions/LambdaSearchWrapper.java
View File

@ -330,10 +330,24 @@ public class LambdaSearchWrapper<T> extends AbstractChainWrapper<T> implements
return this; return this;
} }
public LambdaSearchWrapper<T> annsField(String annsField){
this.annsField=annsField;
return this;
}
public LambdaSearchWrapper<T> vector(List<Float> vector) { public LambdaSearchWrapper<T> vector(List<Float> vector) {
vectors.add(vector); vectors.add(vector);
return this; return this;
} }
public LambdaSearchWrapper<T> vector(String annsField,List<Float> vector) {
this.annsField=annsField;
vectors.add(vector);
return this;
}
public LambdaSearchWrapper<T> vector(FieldFunction<T,?> annsField,List<Float> vector) {
this.annsField=annsField.getFieldName(annsField);
vectors.add(vector);
return this;
}
public LambdaSearchWrapper<T> limit(Long limit) { public LambdaSearchWrapper<T> limit(Long limit) {
this.setLimit(limit); this.setLimit(limit);
return this; return this;
@ -348,9 +362,11 @@ public class LambdaSearchWrapper<T> extends AbstractChainWrapper<T> implements
*/ */
private SearchReq build() { private SearchReq build() {
SearchReq.SearchReqBuilder<?, ?> builder = SearchReq.builder() SearchReq.SearchReqBuilder<?, ?> builder = SearchReq.builder()
.collectionName(collectionName) .collectionName(collectionName);
.annsField(annsField);
if(!annsField.isEmpty()){
builder.annsField(annsField);
}
if (!vectors.isEmpty()) { if (!vectors.isEmpty()) {
builder.data(vectors); builder.data(vectors);
} }