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

(improvement)(chat) Fix the SQL fields generated by the large model that contain (#1715)

Browse Source
This commit is contained in:
lexluo09
2024-09-25 22:49:46 +08:00
committed by GitHub
parent e410611a7c
commit 066b9cc1f7
3 changed files with 5 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
common/src/main/java/com/tencent/supersonic/common/jsqlparser/ParseVisitorHelper.java
View File

@@ -1,5 +1,6 @@
package com.tencent.supersonic.common.jsqlparser; package com.tencent.supersonic.common.jsqlparser;
import com.tencent.supersonic.common.util.StringUtil;
import lombok.extern.slf4j.Slf4j; import lombok.extern.slf4j.Slf4j;
import net.sf.jsqlparser.schema.Column; import net.sf.jsqlparser.schema.Column;
import org.apache.commons.lang3.StringUtils; import org.apache.commons.lang3.StringUtils;
@@ -14,7 +15,7 @@ public class ParseVisitorHelper {
public void replaceColumn( public void replaceColumn(
Column column, Map<String, String> fieldNameMap, boolean exactReplace) { Column column, Map<String, String> fieldNameMap, boolean exactReplace) {
String columnName = column.getColumnName(); String columnName = StringUtil.replaceBackticks(column.getColumnName());
String replaceColumn = getReplaceValue(columnName, fieldNameMap, exactReplace); String replaceColumn = getReplaceValue(columnName, fieldNameMap, exactReplace);
if (StringUtils.isNotBlank(replaceColumn)) { if (StringUtils.isNotBlank(replaceColumn)) {
column.setColumnName(replaceColumn); column.setColumnName(replaceColumn);

3
common/src/main/java/com/tencent/supersonic/common/util/StringUtil.java
View File

@@ -46,6 +46,9 @@ public class StringUtil {
} }
public static String replaceBackticks(String sql) { public static String replaceBackticks(String sql) {
if (StringUtils.isBlank(sql)) {
return sql;
}
return sql.replaceAll("`", ""); return sql.replaceAll("`", "");
} }
} }

1
headless/chat/src/main/java/com/tencent/supersonic/headless/chat/corrector/HavingCorrector.java
View File

@@ -66,6 +66,5 @@ public class HavingCorrector extends BaseSemanticCorrector {
SqlAddHelper.addFunctionToSelect(correctS2SQL, havingExpressionList); SqlAddHelper.addFunctionToSelect(correctS2SQL, havingExpressionList);
semanticParseInfo.getSqlInfo().setCorrectedS2SQL(replaceSql); semanticParseInfo.getSqlInfo().setCorrectedS2SQL(replaceSql);
} }
return;
} }
} }