enhance: Use primitive type for vectorType (#33911)

issue: #22837 pr: #33868 Use primitive type instead of proto enum type for queryHook to recognize Signed-off-by: chasingegg <chao.gao@zilliz.com>
milvus-io · Jun 17, 2024 · 08c096c · 08c096c
1 parent e807183
commit 08c096c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/internal/querynodev2/optimizers/query_hook.go b/internal/querynodev2/optimizers/query_hook.go
@@ -62,7 +62,7 @@ func OptimizeSearchParams(ctx context.Context, req *querypb.SearchRequest, query
 			common.SearchParamKey:  queryInfo.GetSearchParams(),
 			common.SegmentNumKey:   estSegmentNum,
 			common.WithFilterKey:   withFilter,
-			common.DataTypeKey:     plan.GetVectorAnns().GetVectorType(),
+			common.DataTypeKey:     int32(plan.GetVectorAnns().GetVectorType()),
 			common.WithOptimizeKey: paramtable.Get().AutoIndexConfig.EnableOptimize.GetAsBool(),
 			common.CollectionKey:   req.GetReq().GetCollectionID(),
 		}