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

(improvement)(Headless) initializing by adding tag data. (#836)

Browse Source
This commit is contained in:
lexluo09
2024-03-20 15:18:56 +08:00
committed by GitHub
parent 69a76ffc04
commit ef298e8e85
5 changed files with 55 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
headless/server/src/main/java/com/tencent/supersonic/headless/server/service/impl/TagMetaServiceImpl.java
View File

@@ -56,9 +56,9 @@ public class TagMetaServiceImpl implements TagMetaService {
private final DomainService domainService;
public TagMetaServiceImpl(TagRepository tagRepository, ModelService modelService,
CollectService collectService, @Lazy DimensionService dimensionService,
@Lazy MetricService metricService, TagObjectService tagObjectService,
DomainService domainService) {
CollectService collectService, @Lazy DimensionService dimensionService,
@Lazy MetricService metricService, TagObjectService tagObjectService,
DomainService domainService) {
this.tagRepository = tagRepository;
this.modelService = modelService;
this.collectService = collectService;
@@ -106,6 +106,9 @@ public class TagMetaServiceImpl implements TagMetaService {
@Override
public TagResp getTag(Long id, User user) {
TagDO tagDO = tagRepository.getTagById(id);
if (Objects.isNull(tagDO)) {
return null;
}
TagResp tagResp = convert2Resp(tagDO);
List<TagResp> tagRespList = Arrays.asList(tagResp);
fillModelInfo(tagRespList);
@@ -141,7 +144,7 @@ public class TagMetaServiceImpl implements TagMetaService {
BeanUtils.copyProperties(tagMarketPageReq, tagFilter);
tagFilter.setModelIds(modelIds);
PageInfo<TagResp> tagDOPageInfo = PageHelper.startPage(tagMarketPageReq.getCurrent(),
tagMarketPageReq.getPageSize())
tagMarketPageReq.getPageSize())
.doSelectPageInfo(() -> getTags(tagFilter));
List<TagResp> tagRespList = tagDOPageInfo.getList();

3
headless/server/src/main/java/com/tencent/supersonic/headless/server/service/impl/TagQueryServiceImpl.java
View File

@@ -62,6 +62,9 @@ public class TagQueryServiceImpl implements TagQueryService {
itemValueResp.setItemId(itemValueReq.getItemId());
itemValueResp.setType(SchemaElementType.TAG);
TagResp tag = tagMetaService.getTag(itemValueReq.getItemId(), user);
if (Objects.isNull(tag)) {
return null;
}
checkTag(tag);
itemValueResp.setName(tag.getName());
itemValueResp.setBizName(tag.getBizName());

17
launchers/standalone/src/main/java/com/tencent/supersonic/ModelDemoDataLoader.java
View File

@@ -40,6 +40,7 @@ import com.tencent.supersonic.headless.api.pojo.enums.DimensionType;
import com.tencent.supersonic.headless.api.pojo.enums.IdentifyType;
import com.tencent.supersonic.headless.api.pojo.enums.MetricDefineType;
import com.tencent.supersonic.headless.api.pojo.enums.SemanticType;
import com.tencent.supersonic.headless.api.pojo.enums.TagDefineType;
import com.tencent.supersonic.headless.api.pojo.request.DataSetReq;
import com.tencent.supersonic.headless.api.pojo.request.DatabaseReq;
import com.tencent.supersonic.headless.api.pojo.request.DimensionReq;
@@ -47,6 +48,7 @@ import com.tencent.supersonic.headless.api.pojo.request.DomainReq;
import com.tencent.supersonic.headless.api.pojo.request.MetricReq;
import com.tencent.supersonic.headless.api.pojo.request.ModelReq;
import com.tencent.supersonic.headless.api.pojo.request.TagObjectReq;
import com.tencent.supersonic.headless.api.pojo.request.TagReq;
import com.tencent.supersonic.headless.server.service.DataSetService;
import com.tencent.supersonic.headless.server.service.DatabaseService;
import com.tencent.supersonic.headless.server.service.DimensionService;
@@ -113,6 +115,7 @@ public class ModelDemoDataLoader {
addModel_4();
updateDimension();
updateMetric();
addTags();
updateMetric_pv();
addDataSet_1();
addDataSet_2();
@@ -370,6 +373,20 @@ public class ModelDemoDataLoader {
modelService.createModel(modelReq, user);
}
private void addTags() {
addTag(4L);
addTag(5L);
addTag(6L);
addTag(7L);
}
private void addTag(Long itemId) {
TagReq tagReq = new TagReq();
tagReq.setTagDefineType(TagDefineType.DIMENSION);
tagReq.setItemId(itemId);
tagMetaService.create(tagReq, User.getFakeUser());
}
public void updateDimension() throws Exception {
DimensionReq dimensionReq = new DimensionReq();
dimensionReq.setType(DimensionType.categorical.name());

4
launchers/standalone/src/main/resources/application-local.yaml
View File

@@ -59,8 +59,8 @@ s2:
langchain4j:
#1.chat-model
chat-model:
provider: open_ai
openai:
provider: local_ai
local_ai:
api-key: api_key
model-name: gpt-3.5-turbo-16k
temperature: 0.0

26
launchers/standalone/src/test/java/com/tencent/supersonic/headless/QueryTagValueTest.java Normal file
View File

@@ -0,0 +1,26 @@
package com.tencent.supersonic.headless;
import static org.junit.Assert.assertTrue;
import com.tencent.supersonic.auth.api.authentication.pojo.User;
import com.tencent.supersonic.headless.api.pojo.SchemaElementType;
import com.tencent.supersonic.headless.api.pojo.request.ItemValueReq;
import com.tencent.supersonic.headless.api.pojo.response.ItemValueResp;
import com.tencent.supersonic.headless.server.service.TagQueryService;
import org.junit.jupiter.api.Test;
import org.springframework.beans.factory.annotation.Autowired;
public class QueryTagValueTest extends BaseTest {
@Autowired
protected TagQueryService tagQueryService;
@Test
public void testQueryTagValue() throws Exception {
ItemValueReq itemValueReq = new ItemValueReq();
itemValueReq.setItemId(1L);
itemValueReq.setType(SchemaElementType.DIMENSION);
ItemValueResp itemValueResp = tagQueryService.queryTagValue(itemValueReq, User.getFakeUser());
assertTrue(itemValueResp != null);
}
}