topdog/supersonic
1
0
Fork 0
mirror of https://github.com/tencentmusic/supersonic.git synced 2025-12-14 13:47:09 +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

6
headless/server/src/main/java/com/tencent/supersonic/headless/server/service/impl/SchemaServiceImpl.java
View File

@@ -246,6 +246,12 @@ public class SchemaServiceImpl implements SchemaService {
.collect(Collectors.toList()));
dataSetSchemaResp.setTermResps(
termMaps.getOrDefault(dataSetResp.getDomainId(), Lists.newArrayList()));
if (!CollectionUtils.isEmpty(dataSetSchemaResp.getModelResps())) {
DatabaseResp databaseResp =
databaseService.getDatabase(
dataSetSchemaResp.getModelResps().get(0).getDatabaseId());
dataSetSchemaResp.setDatabaseType(databaseResp.getType());
}
dataSetSchemaResps.add(dataSetSchemaResp);
}
fillStaticInfo(dataSetSchemaResps);

1
headless/server/src/main/java/com/tencent/supersonic/headless/server/utils/DataSetSchemaBuilder.java
View File

@@ -41,6 +41,7 @@ public class DataSetSchemaBuilder {
.type(SchemaElementType.DATASET)
.build();
dataSetSchema.setDataSet(dataSet);
dataSetSchema.setDatabaseType(resp.getDatabaseType());
Set<SchemaElement> metrics = getMetrics(resp);
dataSetSchema.getMetrics().addAll(metrics);