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

[improvement](headless) Optimize the reload embedding interface (#1655)

Browse Source
This commit is contained in:
lexluo09
2024-09-11 22:52:46 +08:00
committed by GitHub
parent 51e7a9bcd7
commit ff0833ddfa
3 changed files with 20 additions and 15 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
headless/server/src/main/java/com/tencent/supersonic/headless/server/rest/KnowledgeController.java
View File

@@ -7,6 +7,7 @@ import javax.validation.Valid;
import com.github.pagehelper.PageInfo;
import com.tencent.supersonic.auth.api.authentication.pojo.User;
import com.tencent.supersonic.auth.api.authentication.utils.UserHolder;
import com.tencent.supersonic.common.service.ExemplarService;
import com.tencent.supersonic.headless.api.pojo.request.DictItemFilter;
import com.tencent.supersonic.headless.api.pojo.request.DictItemReq;
import com.tencent.supersonic.headless.api.pojo.request.DictSingleTaskReq;
@@ -36,10 +37,12 @@ public class KnowledgeController {
@Autowired private DictConfService confService;
@Autowired private MetaEmbeddingTask embeddingTask;
@Autowired private MetaEmbeddingTask metaEmbeddingTask;
@Autowired private DictionaryReloadTask dictionaryReloadTask;
@Autowired private ExemplarService exemplarService;
/**
* addDictConf-新增item的字典配置 Add configuration information for dictionary entries
*
@@ -130,15 +133,16 @@ public class KnowledgeController {
return taskService.queryLatestDictTask(taskReq, user);
}
@GetMapping("/meta/embedding/reload")
public Object reloadMetaEmbedding() {
embeddingTask.reloadMetaEmbedding();
@GetMapping("/embedding/reload")
public Object reloadEmbedding() {
metaEmbeddingTask.reloadMetaEmbedding();
exemplarService.loadSysExemplars();
return true;
}
@GetMapping("/embedding/persistFile")
public Object executePersistFileTask() {
embeddingTask.executePersistFileTask();
metaEmbeddingTask.executePersistFileTask();
return true;
}