From e08435902a59cdf2ffcaf9c6ef1c46e4f7381d26 Mon Sep 17 00:00:00 2001
From: lexluo09 <39718951+lexluo09@users.noreply.github.com>
Date: Mon, 18 Dec 2023 22:04:40 +0800
Subject: [PATCH] [improvement](chat) fix SqlGeneration format issues (#538)

---
 .../supersonic/chat/parser/sql/llm/OnePassSCSqlGeneration.java | 3 +--
 .../supersonic/chat/parser/sql/llm/TwoPassSCSqlGeneration.java | 3 +--
 launchers/standalone/src/main/resources/application-local.yaml | 2 +-
 3 files changed, 3 insertions(+), 5 deletions(-)

diff --git a/chat/core/src/main/java/com/tencent/supersonic/chat/parser/sql/llm/OnePassSCSqlGeneration.java b/chat/core/src/main/java/com/tencent/supersonic/chat/parser/sql/llm/OnePassSCSqlGeneration.java
index 0a22cda95..9c2b9cb72 100644
--- a/chat/core/src/main/java/com/tencent/supersonic/chat/parser/sql/llm/OnePassSCSqlGeneration.java
+++ b/chat/core/src/main/java/com/tencent/supersonic/chat/parser/sql/llm/OnePassSCSqlGeneration.java
@@ -63,8 +63,7 @@ public class OnePassSCSqlGeneration implements SqlGeneration, InitializingBean {
         Pair<String, Map<String, Double>> linkingMap = OutputFormat.selfConsistencyVote(candidateSortedList);
         List<String> sqlList = llmResults.stream()
                 .map(llmResult -> OutputFormat.getSql(llmResult)).collect(Collectors.toList());
-        List<String> sqlListSortedList = OutputFormat.formatList(sqlList);
-        Pair<String, Map<String, Double>> sqlMap = OutputFormat.selfConsistencyVote(sqlListSortedList);
+        Pair<String, Map<String, Double>> sqlMap = OutputFormat.selfConsistencyVote(sqlList);
         log.info("linkingMap result:{},sqlMap:{}", linkingMap, sqlMap);
         return sqlMap.getRight();
     }
diff --git a/chat/core/src/main/java/com/tencent/supersonic/chat/parser/sql/llm/TwoPassSCSqlGeneration.java b/chat/core/src/main/java/com/tencent/supersonic/chat/parser/sql/llm/TwoPassSCSqlGeneration.java
index 98bdcf9d7..9519e3b44 100644
--- a/chat/core/src/main/java/com/tencent/supersonic/chat/parser/sql/llm/TwoPassSCSqlGeneration.java
+++ b/chat/core/src/main/java/com/tencent/supersonic/chat/parser/sql/llm/TwoPassSCSqlGeneration.java
@@ -68,8 +68,7 @@ public class TwoPassSCSqlGeneration implements SqlGeneration, InitializingBean {
             sqlTaskPool.add(result);
         });
         //4.format response.
-        List<String> sqlTaskSortedList = OutputFormat.formatList(sqlTaskPool);
-        Pair<String, Map<String, Double>> sqlMap = OutputFormat.selfConsistencyVote(sqlTaskSortedList);
+        Pair<String, Map<String, Double>> sqlMap = OutputFormat.selfConsistencyVote(sqlTaskPool);
         log.info("linkingMap result:{},sqlMap:{}", linkingMap, sqlMap);
         return sqlMap.getRight();
     }
diff --git a/launchers/standalone/src/main/resources/application-local.yaml b/launchers/standalone/src/main/resources/application-local.yaml
index ce5ce830d..0ad7322e7 100644
--- a/launchers/standalone/src/main/resources/application-local.yaml
+++ b/launchers/standalone/src/main/resources/application-local.yaml
@@ -49,7 +49,7 @@ s2:
     chat-model:
       provider: open_ai
       openai:
-        api-key: api_key
+        api-key: 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
         model-name: gpt-3.5-turbo-16k
         temperature: 0.0
         timeout: PT60S