fix ut

Author: starocean999

fe/fe-core/src/main/java/org/apache/doris/httpv2/rest/TableQueryPlanAction.java

@@ -253,7 +253,6 @@ private void handleQuery(ConnectContext context, String requestDb, String reques
             context.setQueryId(queryId);
             context.setStartTime();
             context.setSqlHash(DigestUtils.md5Hex(sql));
-            context.getState().setIsQuery(true);
 
             NereidsPlanner nereidsPlanner = new NereidsPlanner(context.getStatementContext());
             LogicalPlan rewrittenPlan = (LogicalPlan) nereidsPlanner.planWithLock(parsedPlan,
@@ -383,7 +382,7 @@ private static void addToAuditLog(ConnectContext ctx, String origStmt, Statement
                 .setStmtId(ctx.getStmtId())
                 .setSqlHash(ctx.getSqlHash())
                 .setIsQuery(true)
-                .setIsNereids(false)
+                .setIsNereids(true)
                 .setisInternal(false)
                 .setCloudCluster(Strings.isNullOrEmpty(cluster) ? "UNKNOWN" : cluster)
                 .setWorkloadGroup(ctx.getWorkloadGroupName());