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

[improvement][headless]Add databaseType into the Schema part of the Text2SQL prompt. #1621

Browse Source
This commit is contained in:
jerryjzhang
2024-09-12 23:20:49 +08:00
parent 47cc933aec
commit 37f12391b0
9 changed files with 44 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -16,6 +16,7 @@ import java.util.stream.Collectors;
@Data
public class DataSetSchema {
private String databaseType;
private SchemaElement dataSet;
private Set<SchemaElement> metrics = new HashSet<>();
private Set<SchemaElement> dimensions = new HashSet<>();

1
headless/api/src/main/java/com/tencent/supersonic/headless/api/pojo/response/DataSetSchemaResp.java
View File

@@ -13,6 +13,7 @@ import java.util.List;
@NoArgsConstructor
public class DataSetSchemaResp extends DataSetResp {
private String databaseType;
private List<MetricSchemaResp> metrics = Lists.newArrayList();
private List<DimSchemaResp> dimensions = Lists.newArrayList();
private List<ModelResp> modelResps = Lists.newArrayList();