topdog/supersonic
1
0
Fork 0
mirror of https://github.com/tencentmusic/supersonic.git synced 2025-12-11 12:07:42 +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

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);
}
}