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

(improvement)(chat)Determine if with statement is supported and send explicitly message in the prompt to the LLM.

Browse Source
This commit is contained in:
jerryjzhang
2025-02-17 21:21:43 +08:00
parent f31db98aba
commit 32793ecf69
9 changed files with 49 additions and 25 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -197,6 +197,7 @@ public class SchemaServiceImpl implements SchemaService {
DatabaseResp databaseResp = databaseService
.getDatabase(dataSetSchemaResp.getModelResps().get(0).getDatabaseId());
dataSetSchemaResp.setDatabaseType(databaseResp.getType());
dataSetSchemaResp.setDatabaseVersion(databaseResp.getVersion());
}
dataSetSchemaResps.add(dataSetSchemaResp);
}

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

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