From ddbbd6d8bd9c5dc5cf55396707e766d3b160ba51 Mon Sep 17 00:00:00 2001
From: LXW <1264174498@qq.com>
Date: Wed, 20 Mar 2024 20:37:21 +0800
Subject: [PATCH] (improvement)(Chat) Fix MetricRecommendProcessor getting
 parse info (#838)

Co-authored-by: jolunoluo
---
 .../execute/MetricRecommendProcessor.java     |  2 +-
 .../main/resources/META-INF/spring.factories  | 22 +++++++++----------
 2 files changed, 12 insertions(+), 12 deletions(-)

diff --git a/chat/server/src/main/java/com/tencent/supersonic/chat/server/processor/execute/MetricRecommendProcessor.java b/chat/server/src/main/java/com/tencent/supersonic/chat/server/processor/execute/MetricRecommendProcessor.java
index 9cb49d6ce..a97e6cf03 100644
--- a/chat/server/src/main/java/com/tencent/supersonic/chat/server/processor/execute/MetricRecommendProcessor.java
+++ b/chat/server/src/main/java/com/tencent/supersonic/chat/server/processor/execute/MetricRecommendProcessor.java
@@ -33,7 +33,7 @@ public class MetricRecommendProcessor implements ExecuteResultProcessor {
 
     @Override
     public void process(ChatExecuteContext chatExecuteContext, QueryResult queryResult) {
-        fillSimilarMetric(queryResult.getChatContext());
+        fillSimilarMetric(chatExecuteContext.getParseInfo());
     }
 
     private void fillSimilarMetric(SemanticParseInfo parseInfo) {
diff --git a/launchers/standalone/src/main/resources/META-INF/spring.factories b/launchers/standalone/src/main/resources/META-INF/spring.factories
index e8efe39a2..46142a5a8 100644
--- a/launchers/standalone/src/main/resources/META-INF/spring.factories
+++ b/launchers/standalone/src/main/resources/META-INF/spring.factories
@@ -9,14 +9,6 @@ com.tencent.supersonic.headless.core.chat.parser.SemanticParser=\
     com.tencent.supersonic.headless.core.chat.parser.llm.LLMSqlParser, \
     com.tencent.supersonic.headless.core.chat.parser.QueryTypeParser
 
-com.tencent.supersonic.chat.server.parser.ChatParser=\
-    com.tencent.supersonic.chat.server.parser.NL2PluginParser, \
-    com.tencent.supersonic.chat.server.parser.NL2SQLParser
-
-com.tencent.supersonic.chat.server.executor.ChatExecutor=\
-    com.tencent.supersonic.chat.server.executor.PluginExecutor, \
-    com.tencent.supersonic.chat.server.executor.SqlExecutor
-
 com.tencent.supersonic.headless.core.chat.corrector.SemanticCorrector=\
     com.tencent.supersonic.headless.core.chat.corrector.SchemaCorrector, \
     com.tencent.supersonic.headless.core.chat.corrector.TimeCorrector, \
@@ -40,19 +32,27 @@ com.tencent.supersonic.headless.core.parser.SqlParser=\
 com.tencent.supersonic.headless.core.cache.QueryCache=\
     com.tencent.supersonic.headless.core.cache.DefaultQueryCache
 
+com.tencent.supersonic.headless.core.chat.parser.llm.DataSetResolver=\
+  com.tencent.supersonic.headless.core.chat.parser.llm.HeuristicDataSetResolver
+
 com.tencent.supersonic.headless.server.processor.ResultProcessor=\
     com.tencent.supersonic.headless.server.processor.ParseInfoProcessor, \
     com.tencent.supersonic.headless.server.processor.SqlInfoProcessor
 
-com.tencent.supersonic.headless.core.chat.parser.llm.DataSetResolver=\
-  com.tencent.supersonic.headless.core.chat.parser.llm.HeuristicDataSetResolver
-
 com.tencent.supersonic.auth.authentication.interceptor.AuthenticationInterceptor=\
     com.tencent.supersonic.auth.authentication.interceptor.DefaultAuthenticationInterceptor
 
 com.tencent.supersonic.auth.api.authentication.adaptor.UserAdaptor=\
     com.tencent.supersonic.auth.authentication.adaptor.DefaultUserAdaptor
 
+com.tencent.supersonic.chat.server.parser.ChatParser=\
+    com.tencent.supersonic.chat.server.parser.NL2PluginParser, \
+    com.tencent.supersonic.chat.server.parser.NL2SQLParser
+
+com.tencent.supersonic.chat.server.executor.ChatExecutor=\
+    com.tencent.supersonic.chat.server.executor.PluginExecutor, \
+    com.tencent.supersonic.chat.server.executor.SqlExecutor
+
 com.tencent.supersonic.chat.server.plugin.recognize.PluginRecognizer=\
     com.tencent.supersonic.chat.server.plugin.recognize.embedding.EmbeddingRecallRecognizer