topdog/supersonic
1
0
Fork 0
mirror of https://github.com/tencentmusic/supersonic.git synced 2026-04-30 04:54:25 +08: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