(improvement)(headless&chat)Execute against SemanticLayerService instead of ChatQueryService in chat-server and rename several classes by the way.

This commit is contained in:
jerryjzhang
2024-07-06 23:32:59 +08:00
parent 6db6aaf98d
commit e0e77a3b64
26 changed files with 185 additions and 176 deletions

View File

@@ -4,6 +4,7 @@ public enum WorkflowState {
MAPPING,
PARSING,
CORRECTING,
TRANSLATING,
PROCESSING,
FINISHED
}

View File

@@ -1,6 +1,9 @@
package com.tencent.supersonic.headless.api.pojo.request;
import lombok.AllArgsConstructor;
import lombok.Builder;
import lombok.Data;
import lombok.NoArgsConstructor;
import lombok.ToString;
import org.apache.commons.collections.CollectionUtils;
@@ -8,8 +11,13 @@ import java.util.Objects;
@Data
@ToString
@Builder
@AllArgsConstructor
@NoArgsConstructor
public class QuerySqlReq extends SemanticQueryReq {
private String sql;
private Integer limit = 1000;
@Override

View File

@@ -155,7 +155,7 @@ public class QueryStructReq extends SemanticQueryReq {
String sql = null;
try {
sql = buildSql(this, isBizName);
} catch (Exception e) {
} catch (JSQLParserException e) {
log.error("buildSql error", e);
}
@@ -164,7 +164,7 @@ public class QueryStructReq extends SemanticQueryReq {
result.setDataSetId(this.getDataSetId());
result.setModelIds(this.getModelIdSet());
result.setParams(new ArrayList<>());
result.setSqlInfo(this.getSqlInfo());
result.getSqlInfo().setCorrectS2SQL(sql);
return result;
}

View File

@@ -16,7 +16,7 @@ import java.util.List;
import java.util.Set;
@Data
public class QueryReq {
public class QueryTextReq {
private String queryText;
private Integer chatId;
private Set<Long> dataSetIds = Sets.newHashSet();

View File

@@ -12,7 +12,7 @@ import lombok.ToString;
@Builder
@AllArgsConstructor
@NoArgsConstructor
public class ExplainSqlReq<T> {
public class TranslateSqlReq<T> {
private QueryMethod queryTypeEnum;

View File

@@ -13,7 +13,7 @@ import java.io.Serializable;
@Builder
@AllArgsConstructor
@NoArgsConstructor
public class ExplainResp implements Serializable {
public class TranslateResp implements Serializable {
private String sql;