topdog/supersonic
1
0
Fork 0
mirror of https://github.com/tencentmusic/supersonic.git synced 2025-12-12 12:37:55 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

[improvement](chat) fix SqlGeneration format issues (#538)

Browse Source
This commit is contained in:
lexluo09
2023-12-18 22:04:40 +08:00
committed by GitHub
parent b44fa2bf3c
commit e08435902a
3 changed files with 3 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
chat/core/src/main/java/com/tencent/supersonic/chat/parser/sql/llm/OnePassSCSqlGeneration.java
View File

@@ -63,8 +63,7 @@ public class OnePassSCSqlGeneration implements SqlGeneration, InitializingBean {
Pair<String, Map<String, Double>> linkingMap = OutputFormat.selfConsistencyVote(candidateSortedList); Pair<String, Map<String, Double>> linkingMap = OutputFormat.selfConsistencyVote(candidateSortedList);
List<String> sqlList = llmResults.stream() List<String> sqlList = llmResults.stream()
.map(llmResult -> OutputFormat.getSql(llmResult)).collect(Collectors.toList()); .map(llmResult -> OutputFormat.getSql(llmResult)).collect(Collectors.toList());
List<String> sqlListSortedList = OutputFormat.formatList(sqlList); Pair<String, Map<String, Double>> sqlMap = OutputFormat.selfConsistencyVote(sqlList);
Pair<String, Map<String, Double>> sqlMap = OutputFormat.selfConsistencyVote(sqlListSortedList);
log.info("linkingMap result:{},sqlMap:{}", linkingMap, sqlMap); log.info("linkingMap result:{},sqlMap:{}", linkingMap, sqlMap);
return sqlMap.getRight(); return sqlMap.getRight();
} }

3
chat/core/src/main/java/com/tencent/supersonic/chat/parser/sql/llm/TwoPassSCSqlGeneration.java
View File

@@ -68,8 +68,7 @@ public class TwoPassSCSqlGeneration implements SqlGeneration, InitializingBean {
sqlTaskPool.add(result); sqlTaskPool.add(result);
}); });
//4.format response. //4.format response.
List<String> sqlTaskSortedList = OutputFormat.formatList(sqlTaskPool); Pair<String, Map<String, Double>> sqlMap = OutputFormat.selfConsistencyVote(sqlTaskPool);
Pair<String, Map<String, Double>> sqlMap = OutputFormat.selfConsistencyVote(sqlTaskSortedList);
log.info("linkingMap result:{},sqlMap:{}", linkingMap, sqlMap); log.info("linkingMap result:{},sqlMap:{}", linkingMap, sqlMap);
return sqlMap.getRight(); return sqlMap.getRight();
} }

2
launchers/standalone/src/main/resources/application-local.yaml
View File

@@ -49,7 +49,7 @@ s2:
chat-model: chat-model:
provider: open_ai provider: open_ai
openai: openai:
api-key: api_key api-key: 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
model-name: gpt-3.5-turbo-16k model-name: gpt-3.5-turbo-16k
temperature: 0.0 temperature: 0.0
timeout: PT60S timeout: PT60S