mirror of
https://github.com/tencentmusic/supersonic.git
synced 2025-12-12 12:37:55 +00:00
[improvement][chat]Opt log infos.
This commit is contained in:
@@ -32,10 +32,11 @@ public class LLMSqlCorrector extends BaseSemanticCorrector {
|
|||||||
private static final String INSTRUCTION = ""
|
private static final String INSTRUCTION = ""
|
||||||
+ "#Role: You are a senior data engineer experienced in writing SQL."
|
+ "#Role: You are a senior data engineer experienced in writing SQL."
|
||||||
+ "\n#Task: Your will be provided with a user question and the SQL written by a junior engineer,"
|
+ "\n#Task: Your will be provided with a user question and the SQL written by a junior engineer,"
|
||||||
+ "please take a review and help correct it if necessary." + "\n#Rules: "
|
+ "please take a review and help correct it if necessary."
|
||||||
+ "\n1.ALWAYS follow the output format: `opinion=(POSITIVE|NEGATIVE),sql=(corrected sql if NEGATIVE; empty string if POSITIVE)`."
|
+ "\n#Rules: "
|
||||||
+ "\n2.NO NEED to check date filters as the junior engineer seldom makes mistakes in this regard."
|
+ "1.ALWAYS specify time range using `>`,`<`,`>=`,`<=` operator."
|
||||||
+ "\n3.SQL columns and values must be mentioned in the `#Schema`."
|
+ "2.DO NOT calculate date range using functions."
|
||||||
|
+ "3.SQL columns and values must be mentioned in the `#Schema`."
|
||||||
+ "\n#Question:{{question}} #Schema:{{schema}} #InputSQL:{{sql}} #Response:";
|
+ "\n#Question:{{question}} #Schema:{{schema}} #InputSQL:{{sql}} #Response:";
|
||||||
|
|
||||||
public LLMSqlCorrector() {
|
public LLMSqlCorrector() {
|
||||||
@@ -46,10 +47,10 @@ public class LLMSqlCorrector extends BaseSemanticCorrector {
|
|||||||
@Data
|
@Data
|
||||||
@ToString
|
@ToString
|
||||||
static class SemanticSql {
|
static class SemanticSql {
|
||||||
@Description("positive or negative opinion about the sql")
|
@Description("either positive or negative")
|
||||||
private String opinion;
|
private String opinion;
|
||||||
|
|
||||||
@Description("corrected sql")
|
@Description("corrected sql if negative")
|
||||||
private String sql;
|
private String sql;
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -76,7 +77,7 @@ public class LLMSqlCorrector extends BaseSemanticCorrector {
|
|||||||
semanticParseInfo, chatApp.getPrompt(), exemplar);
|
semanticParseInfo, chatApp.getPrompt(), exemplar);
|
||||||
SemanticSql s2Sql = extractor.generateSemanticSql(prompt.toUserMessage().singleText());
|
SemanticSql s2Sql = extractor.generateSemanticSql(prompt.toUserMessage().singleText());
|
||||||
keyPipelineLog.info("LLMSqlCorrector modelReq:\n{} \nmodelResp:\n{}", prompt.text(), s2Sql);
|
keyPipelineLog.info("LLMSqlCorrector modelReq:\n{} \nmodelResp:\n{}", prompt.text(), s2Sql);
|
||||||
if ("NEGATIVE".equals(s2Sql.getOpinion()) && StringUtils.isNotBlank(s2Sql.getSql())) {
|
if ("NEGATIVE".equalsIgnoreCase(s2Sql.getOpinion()) && StringUtils.isNotBlank(s2Sql.getSql())) {
|
||||||
semanticParseInfo.getSqlInfo().setCorrectedS2SQL(s2Sql.getSql());
|
semanticParseInfo.getSqlInfo().setCorrectedS2SQL(s2Sql.getSql());
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -1,5 +1,6 @@
|
|||||||
package com.tencent.supersonic.headless.core.translator.parser;
|
package com.tencent.supersonic.headless.core.translator.parser;
|
||||||
|
|
||||||
|
import com.google.common.collect.Lists;
|
||||||
import com.google.common.collect.Sets;
|
import com.google.common.collect.Sets;
|
||||||
import com.tencent.supersonic.common.jsqlparser.SqlReplaceHelper;
|
import com.tencent.supersonic.common.jsqlparser.SqlReplaceHelper;
|
||||||
import com.tencent.supersonic.common.jsqlparser.SqlSelectFunctionHelper;
|
import com.tencent.supersonic.common.jsqlparser.SqlSelectFunctionHelper;
|
||||||
@@ -49,8 +50,12 @@ public class SqlQueryParser implements QueryParser {
|
|||||||
// check if there are fields not matched with any metric or dimension
|
// check if there are fields not matched with any metric or dimension
|
||||||
if (queryFields.size() > ontologyQuery.getMetrics().size()
|
if (queryFields.size() > ontologyQuery.getMetrics().size()
|
||||||
+ ontologyQuery.getDimensions().size()) {
|
+ ontologyQuery.getDimensions().size()) {
|
||||||
queryStatement
|
List<String> semanticFields = Lists.newArrayList();
|
||||||
.setErrMsg("There are fields in the SQL not matched with any semantic column.");
|
ontologyQuery.getMetrics().forEach(m -> semanticFields.add(m.getName()));
|
||||||
|
ontologyQuery.getDimensions().forEach(d -> semanticFields.add(d.getName()));
|
||||||
|
String errMsg = String.format("Querying columns[%s] not matched with semantic fields[%s].",
|
||||||
|
queryFields, semanticFields);
|
||||||
|
queryStatement.setErrMsg(errMsg);
|
||||||
queryStatement.setStatus(QueryState.INVALID);
|
queryStatement.setStatus(QueryState.INVALID);
|
||||||
return;
|
return;
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -152,7 +152,7 @@ public class ChatWorkflowEngine {
|
|||||||
StringUtils.normalizeSpace(parseInfo.getSqlInfo().getCorrectedS2SQL()),
|
StringUtils.normalizeSpace(parseInfo.getSqlInfo().getCorrectedS2SQL()),
|
||||||
StringUtils.normalizeSpace(parseInfo.getSqlInfo().getQuerySQL()));
|
StringUtils.normalizeSpace(parseInfo.getSqlInfo().getQuerySQL()));
|
||||||
} catch (Exception e) {
|
} catch (Exception e) {
|
||||||
log.warn("get sql info failed:{}", parseInfo, e);
|
log.warn("get sql info failed:{}", e);
|
||||||
errorMsg.add(String.format("S2SQL:%s %s", parseInfo.getSqlInfo().getParsedS2SQL(),
|
errorMsg.add(String.format("S2SQL:%s %s", parseInfo.getSqlInfo().getParsedS2SQL(),
|
||||||
e.getMessage()));
|
e.getMessage()));
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -119,11 +119,17 @@ public class QueryUtils {
|
|||||||
if (StringUtils.isBlank(type)) {
|
if (StringUtils.isBlank(type)) {
|
||||||
return false;
|
return false;
|
||||||
}
|
}
|
||||||
return type.equalsIgnoreCase("int") || type.equalsIgnoreCase("bigint")
|
return type.equalsIgnoreCase("int")
|
||||||
|| type.equalsIgnoreCase("tinyint") || type.equalsIgnoreCase("smallint")
|
|| type.equalsIgnoreCase("bigint")
|
||||||
|| type.equalsIgnoreCase("float") || type.equalsIgnoreCase("double") || type.equalsIgnoreCase("real")
|
|| type.equalsIgnoreCase("tinyint")
|
||||||
|| type.equalsIgnoreCase("numeric") || type.toLowerCase().startsWith("decimal")
|
|| type.equalsIgnoreCase("smallint")
|
||||||
|| type.toLowerCase().startsWith("uint") || type.toLowerCase().startsWith("int");
|
|| type.equalsIgnoreCase("float")
|
||||||
|
|| type.equalsIgnoreCase("double")
|
||||||
|
|| type.equalsIgnoreCase("real")
|
||||||
|
|| type.equalsIgnoreCase("numeric")
|
||||||
|
|| type.toLowerCase().startsWith("decimal")
|
||||||
|
|| type.toLowerCase().startsWith("uint")
|
||||||
|
|| type.toLowerCase().startsWith("int");
|
||||||
}
|
}
|
||||||
|
|
||||||
private String getName(String nameEn) {
|
private String getName(String nameEn) {
|
||||||
|
|||||||
Reference in New Issue
Block a user