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

(improvement)(chat) Make it compatible with the case when the agent is empty. (#596)

Browse Source
This commit is contained in:
lexluo09
2024-01-04 21:39:56 +08:00
committed by GitHub
parent ade96c3adc
commit 602b9547b8
3 changed files with 16 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
chat/core/src/main/java/com/tencent/supersonic/chat/core/agent/Agent.java
View File

@@ -22,7 +22,9 @@ public class Agent extends RecordInfo {
private String name;
private String description;
//0 offline, 1 online
/**
* 0 offline, 1 online
*/
private Integer status;
private List<String> examples;
private String agentConfig;
@@ -49,7 +51,7 @@ public class Agent extends RecordInfo {
return enableSearch != null && enableSearch == 1;
}
public boolean containsAllModel(Set<Long> detectModelIds) {
public static boolean containsAllModel(Set<Long> detectModelIds) {
return !CollectionUtils.isEmpty(detectModelIds) && detectModelIds.contains(-1L);
}

7
chat/core/src/main/java/com/tencent/supersonic/chat/core/mapper/MapperHelper.java
View File

@@ -85,9 +85,12 @@ public class MapperHelper {
public Set<Long> getModelIds(QueryReq request, Agent agent) {
Long modelId = request.getModelId();
Set<Long> detectModelIds = agent.getModelIds(null);
Set<Long> detectModelIds = new HashSet<>();
if (Objects.nonNull(agent)) {
detectModelIds = agent.getModelIds(null);
}
//contains all
if (agent.containsAllModel(detectModelIds)) {
if (Agent.containsAllModel(detectModelIds)) {
if (Objects.nonNull(modelId) && modelId > 0) {
Set<Long> result = new HashSet<>();
result.add(modelId);

10
chat/core/src/main/java/com/tencent/supersonic/chat/core/parser/sql/llm/LLMRequestService.java
View File

@@ -61,8 +61,12 @@ public class LLMRequestService {
}
public ModelCluster getModelCluster(QueryContext queryCtx, ChatContext chatCtx) {
Set<Long> distinctModelIds = queryCtx.getAgent().getModelIds(AgentToolType.NL2SQL_LLM);
if (queryCtx.getAgent().containsAllModel(distinctModelIds)) {
Agent agent = queryCtx.getAgent();
Set<Long> distinctModelIds = new HashSet<>();
if (Objects.nonNull(agent)) {
distinctModelIds = agent.getModelIds(AgentToolType.NL2SQL_LLM);
}
if (Agent.containsAllModel(distinctModelIds)) {
distinctModelIds = new HashSet<>();
}
ModelResolver modelResolver = ComponentFactory.getModelResolver();
@@ -77,7 +81,7 @@ public class LLMRequestService {
Optional<NL2SQLTool> llmParserTool = commonAgentTools.stream()
.filter(tool -> {
List<Long> modelIds = tool.getModelIds();
if (agent.containsAllModel(new HashSet<>(modelIds))) {
if (Agent.containsAllModel(new HashSet<>(modelIds))) {
return true;
}
for (Long modelId : modelIdSet) {