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

[improvement][chat]Move generation of semantic text info and rewrite of error message to dedicated ResultProcessor.

Browse Source
This commit is contained in:
jerryjzhang
2024-10-27 22:34:16 +08:00
parent bb363a0286
commit b69ee81d58
13 changed files with 165 additions and 124 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
headless/api/src/main/java/com/tencent/supersonic/headless/api/pojo/SchemaElementMatch.java
View File

@@ -12,13 +12,13 @@ import lombok.ToString;
@AllArgsConstructor
@NoArgsConstructor
public class SchemaElementMatch {
SchemaElement element;
double offset;
double similarity;
String detectWord;
String word;
Long frequency;
boolean isInherited;
private SchemaElement element;
private double offset;
private double similarity;
private String detectWord;
private String word;
private Long frequency;
private boolean isInherited;
public boolean isFullMatched() {
return 1.0 == similarity;

8
headless/api/src/main/java/com/tencent/supersonic/headless/api/pojo/SchemaMapInfo.java
View File

@@ -2,6 +2,7 @@ package com.tencent.supersonic.headless.api.pojo;
import com.fasterxml.jackson.annotation.JsonIgnore;
import com.google.common.collect.Lists;
import lombok.Getter;
import org.apache.commons.collections4.CollectionUtils;
import java.util.HashMap;
@@ -9,9 +10,10 @@ import java.util.List;
import java.util.Map;
import java.util.Set;
@Getter
public class SchemaMapInfo {
private Map<Long, List<SchemaElementMatch>> dataSetElementMatches = new HashMap<>();
private final Map<Long, List<SchemaElementMatch>> dataSetElementMatches = new HashMap<>();
public Set<Long> getMatchedDataSetInfos() {
return dataSetElementMatches.keySet();
@@ -21,10 +23,6 @@ public class SchemaMapInfo {
return dataSetElementMatches.getOrDefault(dataSet, Lists.newArrayList());
}
public Map<Long, List<SchemaElementMatch>> getDataSetElementMatches() {
return dataSetElementMatches;
}
public void setMatchedElements(Long dataSet, List<SchemaElementMatch> elementMatches) {
dataSetElementMatches.put(dataSet, elementMatches);
}

11
headless/api/src/main/java/com/tencent/supersonic/headless/api/pojo/request/QueryNLReq.java
View File

@@ -31,6 +31,15 @@ public class QueryNLReq extends SemanticQueryReq {
@Override
public String toCustomizedString() {
return "";
StringBuilder stringBuilder = new StringBuilder("{");
stringBuilder.append("\"queryText\":").append(dataSetId);
stringBuilder.append("\"dataSetId\":").append(dataSetId);
stringBuilder.append("\"modelIds\":").append(modelIds);
stringBuilder.append(",\"params\":").append(params);
stringBuilder.append(",\"cacheInfo\":").append(cacheInfo);
stringBuilder.append(",\"mapMode\":").append(mapModeEnum);
stringBuilder.append(",\"dataType\":").append(queryDataType);
stringBuilder.append('}');
return stringBuilder.toString();
}
}

1
headless/api/src/main/java/com/tencent/supersonic/headless/api/pojo/request/QuerySqlReq.java
View File

@@ -17,7 +17,6 @@ import java.util.Objects;
public class QuerySqlReq extends SemanticQueryReq {
private String sql;
private Integer limit = 1000;
@Override

6
headless/api/src/main/java/com/tencent/supersonic/headless/api/pojo/response/MapResp.java
View File

@@ -7,10 +7,10 @@ import lombok.Data;
public class MapResp {
private final String queryText;
private final SchemaMapInfo mapInfo;
private SchemaMapInfo mapInfo = new SchemaMapInfo();
public MapResp(String queryText) {
public MapResp(String queryText, SchemaMapInfo schemaMapInfo) {
this.queryText = queryText;
this.mapInfo = schemaMapInfo;
}
}

2
headless/api/src/main/java/com/tencent/supersonic/headless/api/pojo/response/ParseResp.java
View File

@@ -1,6 +1,7 @@
package com.tencent.supersonic.headless.api.pojo.response;
import com.google.common.collect.Lists;
import com.tencent.supersonic.common.pojo.Text2SQLExemplar;
import com.tencent.supersonic.headless.api.pojo.SemanticParseInfo;
import lombok.Data;
@@ -16,6 +17,7 @@ public class ParseResp {
private String errorMsg;
private List<SemanticParseInfo> selectedParses = Lists.newArrayList();
private ParseTimeCostResp parseTimeCost = new ParseTimeCostResp();
private List<Text2SQLExemplar> usedExemplars;
public enum ParseState {
COMPLETED, PENDING, FAILED