enhance: Check ANN field loaded in proxy (#36172)

Related to #35415 Currently ANN field is loaded in Parital load is not performed in proxy, this cause error message not clear and no error returned when collection is empty --------- Signed-off-by: Congqi Xia <[email protected]>
milvus-io · Sep 11, 2024 · 700a443 · 700a443
1 parent e7ea1d7
commit 700a443
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/internal/parser/planparserv2/plan_parser_v2.go b/internal/parser/planparserv2/plan_parser_v2.go
@@ -10,6 +10,7 @@ import (
 	"github.com/milvus-io/milvus-proto/go-api/v2/schemapb"
 	"github.com/milvus-io/milvus/internal/proto/planpb"
 	"github.com/milvus-io/milvus/pkg/log"
+	"github.com/milvus-io/milvus/pkg/util/merr"
 	"github.com/milvus-io/milvus/pkg/util/typeutil"
 )
 
@@ -124,6 +125,10 @@ func CreateSearchPlan(schema *typeutil.SchemaHelper, exprStr string, vectorField
 		log.Info("CreateSearchPlan failed", zap.Error(err))
 		return nil, err
 	}
+	// plan ok with schema, check ann field
+	if !schema.IsFieldLoaded(vectorField.GetFieldID()) {
+		return nil, merr.WrapErrParameterInvalidMsg("ann field \"%s\" not loaded", vectorFieldName)
+	}
 	fieldID := vectorField.FieldID
 	dataType := vectorField.DataType