topdog/supersonic
1
0
Fork 0
mirror of https://github.com/tencentmusic/supersonic.git synced 2026-01-06 09:23:02 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

(improvement)(chat) support show logicSql querySql llmParseSql (#148)

Browse Source
This commit is contained in:
lexluo09
2023-09-26 20:17:02 +08:00
committed by GitHub
parent e4af83380b
commit 4ad3e1d9cf
4 changed files with 24 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
chat/core/src/main/java/com/tencent/supersonic/chat/parser/llm/dsl/LLMDslParser.java
View File

@@ -132,7 +132,7 @@ public class LLMDslParser implements SemanticParser {
if (StringUtils.isEmpty(correctorSql)) {
correctorSql = semanticCorrectInfo.getSql();
}
parseInfo.setLogicSql(correctorSql);
parseInfo.getSqlInfo().setLogicSql(correctorSql);
List<FilterExpression> expressions = SqlParserSelectHelper.getFilterExpression(correctorSql);
//set dataInfo
try {
@@ -281,6 +281,7 @@ public class LLMDslParser implements SemanticParser {
parseInfo.setProperties(properties);
parseInfo.setScore(function_bonus_threshold);
parseInfo.setQueryMode(semanticQuery.getQueryMode());
parseInfo.getSqlInfo().setLlmParseSql(dslParseResult.getLlmResp().getSqlOutput());
SemanticSchema semanticSchema = ContextUtils.getBean(SchemaService.class).getSemanticSchema();
Map<Long, String> modelIdToName = semanticSchema.getModelIdToName();

2
chat/core/src/main/java/com/tencent/supersonic/chat/service/impl/QueryServiceImpl.java
View File

@@ -165,7 +165,7 @@ public class QueryServiceImpl implements QueryService {
if (Objects.isNull(explain)) {
return;
}
parseInfo.setQuerySql(explain.getSql());
parseInfo.getSqlInfo().setQuerySql(explain.getSql());
}
@Override