diff --git a/internal/proxy/impl.go b/internal/proxy/impl.go
index 538de821e8..224603cda8 100644
--- a/internal/proxy/impl.go
+++ b/internal/proxy/impl.go
@@ -1612,6 +1612,7 @@ func (node *Proxy) Flush(ctx context.Context, request *milvuspb.FlushRequest) (*
 	return ft.result, nil
 }
 
+// Query get the records by primary keys.
 func (node *Proxy) Query(ctx context.Context, request *milvuspb.QueryRequest) (*milvuspb.QueryResults, error) {
 	if !node.checkHealthy() {
 		return &milvuspb.QueryResults{