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

(improvement)(chat) Add the pluginList parameter to QueryContext. (#609)

Browse Source
This commit is contained in:
lexluo09
2024-01-09 15:23:07 +08:00
committed by GitHub
parent ed54d7bae3
commit bbad302efd
3 changed files with 5 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
chat/core/src/main/java/com/tencent/supersonic/chat/core/plugin/PluginManager.java
View File

@@ -51,7 +51,7 @@ public class PluginManager {
} }
public static List<Plugin> getPluginAgentCanSupport(QueryContext queryContext) { public static List<Plugin> getPluginAgentCanSupport(QueryContext queryContext) {
List<Plugin> plugins = new ArrayList<>(); List<Plugin> plugins = queryContext.getPluginList();
if (Objects.isNull(queryContext.getAgent())) { if (Objects.isNull(queryContext.getAgent())) {
return plugins; return plugins;
} }

1
chat/core/src/main/java/com/tencent/supersonic/chat/core/pojo/QueryContext.java
View File

@@ -36,6 +36,7 @@ public class QueryContext {
private Agent agent; private Agent agent;
private Map<Long, ChatConfigRichResp> modelIdToChatRichConfig; private Map<Long, ChatConfigRichResp> modelIdToChatRichConfig;
private Map<String, Plugin> nameToPlugin; private Map<String, Plugin> nameToPlugin;
private List<Plugin> pluginList;
public List<SemanticQuery> getCandidateQueries() { public List<SemanticQuery> getCandidateQueries() {
OptimizationConfig optimizationConfig = ContextUtils.getBean(OptimizationConfig.class); OptimizationConfig optimizationConfig = ContextUtils.getBean(OptimizationConfig.class);

3
chat/server/src/main/java/com/tencent/supersonic/chat/server/service/impl/QueryServiceImpl.java
View File

@@ -194,6 +194,8 @@ public class QueryServiceImpl implements QueryService {
Map<Long, ChatConfigRichResp> modelIdToChatRichConfig = configService.getModelIdToChatRichConfig(); Map<Long, ChatConfigRichResp> modelIdToChatRichConfig = configService.getModelIdToChatRichConfig();
Map<String, Plugin> nameToPlugin = pluginService.getNameToPlugin(); Map<String, Plugin> nameToPlugin = pluginService.getNameToPlugin();
List<Plugin> pluginList = pluginService.getPluginList();
QueryContext queryCtx = QueryContext.builder() QueryContext queryCtx = QueryContext.builder()
.request(queryReq) .request(queryReq)
.queryFilters(queryReq.getQueryFilters()) .queryFilters(queryReq.getQueryFilters())
@@ -203,6 +205,7 @@ public class QueryServiceImpl implements QueryService {
.agent(agent) .agent(agent)
.modelIdToChatRichConfig(modelIdToChatRichConfig) .modelIdToChatRichConfig(modelIdToChatRichConfig)
.nameToPlugin(nameToPlugin) .nameToPlugin(nameToPlugin)
.pluginList(pluginList)
.build(); .build();
return queryCtx; return queryCtx;
} }