From 2179daf22dca87d6cd939d7e651959eb705c7647 Mon Sep 17 00:00:00 2001
From: xgc <classxgc@mima>
Date: Sat, 11 May 2024 11:53:04 +0800
Subject: [PATCH] =?UTF-8?q?=E6=B7=BB=E5=8A=A0annsField?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../core/conditions/LambdaSearchWrapper.java  | 20 +++++++++++++++++--
 1 file changed, 18 insertions(+), 2 deletions(-)

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<T> extends AbstractChainWrapper<T> implements
         return this;
     }
 
+    public LambdaSearchWrapper<T> annsField(String annsField){
+        this.annsField=annsField;
+        return this;
+    }
     public LambdaSearchWrapper<T> vector(List<Float> vector) {
         vectors.add(vector);
         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) {
         this.setLimit(limit);
         return this;
@@ -348,9 +362,11 @@ public  class LambdaSearchWrapper<T> extends AbstractChainWrapper<T> 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);
         }