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

(fix)(chat)Fix bug in creating chat model.

Browse Source
This commit is contained in:
jerryjzhang
2025-03-12 16:47:40 +08:00
parent 9d13038599
commit f67bf3eeac
2 changed files with 9 additions and 19 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
common/src/main/java/com/tencent/supersonic/common/service/impl/ChatModelServiceImpl.java
View File

@@ -57,6 +57,7 @@ public class ChatModelServiceImpl extends ServiceImpl<ChatModelMapper, ChatModel
chatModelDO.setViewer(JsonUtil.toString(chatModel.getViewers())); chatModelDO.setViewer(JsonUtil.toString(chatModel.getViewers()));
} }
save(chatModelDO); save(chatModelDO);
chatModel.setId(chatModelDO.getId());
return chatModel; return chatModel;
} }

27
headless/chat/src/test/java/com/tencent/supersonic/headless/chat/corrector/AggCorrectorTest.java
View File

@@ -51,14 +51,15 @@ class AggCorrectorTest {
dataSet.setDataSetId(dataSetId); dataSet.setDataSetId(dataSetId);
semanticParseInfo.setDataSet(dataSet); semanticParseInfo.setDataSet(dataSet);
SqlInfo sqlInfo = new SqlInfo(); SqlInfo sqlInfo = new SqlInfo();
String sql = "WITH 总停留时长 AS (SELECT 用户, SUM(停留时长) AS _总停留时长_ FROM 超音数数据集 WHERE 用户 IN ('alice', 'lucy') AND 数据日期 >= '2025-03-01' AND 数据日期 <= '2025-03-12' GROUP BY 用户) SELECT 用户, _总停留时长_ FROM 总停留时长"; String sql =
"WITH 总停留时长 AS (SELECT 用户, SUM(停留时长) AS _总停留时长_ FROM 超音数数据集 WHERE 用户 IN ('alice', 'lucy') AND 数据日期 >= '2025-03-01' AND 数据日期 <= '2025-03-12' GROUP BY 用户) SELECT 用户, _总停留时长_ FROM 总停留时长";
sqlInfo.setParsedS2SQL(sql); sqlInfo.setParsedS2SQL(sql);
sqlInfo.setCorrectedS2SQL(sql); sqlInfo.setCorrectedS2SQL(sql);
semanticParseInfo.setSqlInfo(sqlInfo); semanticParseInfo.setSqlInfo(sqlInfo);
corrector.correct(chatQueryContext, semanticParseInfo); corrector.correct(chatQueryContext, semanticParseInfo);
Assert.assertEquals( Assert.assertEquals(
"WITH 总停留时长 AS (SELECT 用户名, SUM(停留时长) AS _总停留时长_ FROM 超音数数据集 WHERE 用户名 IN ('alice', 'lucy') AND 数据日期 " + "WITH 总停留时长 AS (SELECT 用户名, SUM(停留时长) AS _总停留时长_ FROM 超音数数据集 WHERE 用户名 IN ('alice', 'lucy') AND 数据日期 "
">= '2025-03-01' AND 数据日期 <= '2025-03-12' GROUP BY 用户名) SELECT 用户名, _总停留时长_ FROM 总停留时长", + ">= '2025-03-01' AND 数据日期 <= '2025-03-12' GROUP BY 用户名) SELECT 用户名, _总停留时长_ FROM 总停留时长",
semanticParseInfo.getSqlInfo().getCorrectedS2SQL()); semanticParseInfo.getSqlInfo().getCorrectedS2SQL());
} }
@@ -73,29 +74,17 @@ class AggCorrectorTest {
dataSetSchema.setDataSet(schemaElement); dataSetSchema.setDataSet(schemaElement);
Set<SchemaElement> dimensions = new HashSet<>(); Set<SchemaElement> dimensions = new HashSet<>();
dimensions.add(SchemaElement.builder() dimensions.add(SchemaElement.builder().dataSetId(1L).name("部门").build());
.dataSetId(1L)
.name("部门")
.build());
dimensions.add(SchemaElement.builder() dimensions.add(SchemaElement.builder().dataSetId(1L).name("用户名").build());
.dataSetId(1L)
.name("用户名")
.build());
dataSetSchema.setDimensions(dimensions); dataSetSchema.setDimensions(dimensions);
Set<SchemaElement> metrics = new HashSet<>(); Set<SchemaElement> metrics = new HashSet<>();
metrics.add(SchemaElement.builder() metrics.add(SchemaElement.builder().dataSetId(1L).name("访问次数").build());
.dataSetId(1L)
.name("访问次数")
.build());
metrics.add(SchemaElement.builder() metrics.add(SchemaElement.builder().dataSetId(1L).name("停留时长").build());
.dataSetId(1L)
.name("停留时长")
.build());
dataSetSchema.setMetrics(metrics); dataSetSchema.setMetrics(metrics);
dataSetSchemaList.add(dataSetSchema); dataSetSchemaList.add(dataSetSchema);