From cacfeb7b4b3890fab82b28ea6f034c13c401e5a4 Mon Sep 17 00:00:00 2001
From: lexluo09 <39718951+lexluo09@users.noreply.github.com>
Date: Thu, 18 Apr 2024 22:35:36 +0800
Subject: [PATCH] (improvement)(Headless) Fix queryBySql to skip
 SchemaCorrector. (#921)

---
 .../headless/server/service/impl/ChatQueryServiceImpl.java     | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/headless/server/src/main/java/com/tencent/supersonic/headless/server/service/impl/ChatQueryServiceImpl.java b/headless/server/src/main/java/com/tencent/supersonic/headless/server/service/impl/ChatQueryServiceImpl.java
index 6412e7115..4bdeed395 100644
--- a/headless/server/src/main/java/com/tencent/supersonic/headless/server/service/impl/ChatQueryServiceImpl.java
+++ b/headless/server/src/main/java/com/tencent/supersonic/headless/server/service/impl/ChatQueryServiceImpl.java
@@ -40,6 +40,7 @@ import com.tencent.supersonic.headless.api.pojo.response.QueryResult;
 import com.tencent.supersonic.headless.api.pojo.response.QueryState;
 import com.tencent.supersonic.headless.api.pojo.response.SemanticQueryResp;
 import com.tencent.supersonic.headless.core.chat.corrector.GrammarCorrector;
+import com.tencent.supersonic.headless.core.chat.corrector.SchemaCorrector;
 import com.tencent.supersonic.headless.core.chat.corrector.SemanticCorrector;
 import com.tencent.supersonic.headless.core.chat.knowledge.HanlpMapResult;
 import com.tencent.supersonic.headless.core.chat.knowledge.KnowledgeService;
@@ -669,7 +670,7 @@ public class ChatQueryServiceImpl implements ChatQueryService {
         semanticParseInfo.setDataSet(dataSet);
 
         ComponentFactory.getSemanticCorrectors().forEach(corrector -> {
-            if (!(corrector instanceof GrammarCorrector)) {
+            if (!(corrector instanceof GrammarCorrector || (corrector instanceof SchemaCorrector))) {
                 corrector.correct(queryCtx, semanticParseInfo);
             }
         });