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

[fix][launcher]Config chat apps for S2ArtistDemo

Browse Source
This commit is contained in:
jerryjzhang
2024-10-12 20:33:22 +08:00
parent 83cf171ec5
commit a7fe900394
2 changed files with 14 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -19,10 +19,10 @@ public class Agent extends RecordInfo {
private String name; private String name;
private String description; private String description;
/** 0 offline, 1 online */ /** 0 offline, 1 online */
private Integer status; private Integer status = 1;
private List<String> examples; private List<String> examples;
private Integer enableSearch; private Integer enableSearch = 1;
private Integer enableFeedback; private Integer enableFeedback = 0;
private String toolConfig; private String toolConfig;
private Map<String, ChatApp> chatAppConfig = Collections.EMPTY_MAP; private Map<String, ChatApp> chatAppConfig = Collections.EMPTY_MAP;
private VisualConfig visualConfig; private VisualConfig visualConfig;

11
launchers/standalone/src/main/java/com/tencent/supersonic/demo/S2ArtistDemo.java
View File

@@ -2,12 +2,15 @@ package com.tencent.supersonic.demo;
import com.alibaba.fastjson.JSONObject; import com.alibaba.fastjson.JSONObject;
import com.google.common.collect.Lists; import com.google.common.collect.Lists;
import com.google.common.collect.Maps;
import com.tencent.supersonic.chat.server.agent.Agent; import com.tencent.supersonic.chat.server.agent.Agent;
import com.tencent.supersonic.chat.server.agent.AgentToolType; import com.tencent.supersonic.chat.server.agent.AgentToolType;
import com.tencent.supersonic.chat.server.agent.DatasetTool; import com.tencent.supersonic.chat.server.agent.DatasetTool;
import com.tencent.supersonic.chat.server.agent.ToolConfig; import com.tencent.supersonic.chat.server.agent.ToolConfig;
import com.tencent.supersonic.common.pojo.ChatApp;
import com.tencent.supersonic.common.pojo.enums.StatusEnum; import com.tencent.supersonic.common.pojo.enums.StatusEnum;
import com.tencent.supersonic.common.pojo.enums.TypeEnums; import com.tencent.supersonic.common.pojo.enums.TypeEnums;
import com.tencent.supersonic.common.util.ChatAppManager;
import com.tencent.supersonic.headless.api.pojo.*; import com.tencent.supersonic.headless.api.pojo.*;
import com.tencent.supersonic.headless.api.pojo.DetailTypeDefaultConfig; import com.tencent.supersonic.headless.api.pojo.DetailTypeDefaultConfig;
import com.tencent.supersonic.headless.api.pojo.enums.DimensionType; import com.tencent.supersonic.headless.api.pojo.enums.DimensionType;
@@ -31,6 +34,7 @@ import java.util.ArrayList;
import java.util.Arrays; import java.util.Arrays;
import java.util.Collections; import java.util.Collections;
import java.util.List; import java.util.List;
import java.util.Map;
@Component @Component
@Slf4j @Slf4j
@@ -167,6 +171,8 @@ public class S2ArtistDemo extends S2BaseDemo {
agent.setStatus(1); agent.setStatus(1);
agent.setEnableSearch(1); agent.setEnableSearch(1);
agent.setExamples(Lists.newArrayList("国风流派歌手", "港台歌手", "周杰伦流派")); agent.setExamples(Lists.newArrayList("国风流派歌手", "港台歌手", "周杰伦流派"));
// configure tools
ToolConfig toolConfig = new ToolConfig(); ToolConfig toolConfig = new ToolConfig();
DatasetTool datasetTool = new DatasetTool(); DatasetTool datasetTool = new DatasetTool();
datasetTool.setId("1"); datasetTool.setId("1");
@@ -174,6 +180,11 @@ public class S2ArtistDemo extends S2BaseDemo {
datasetTool.setDataSetIds(Lists.newArrayList(dataSetId)); datasetTool.setDataSetIds(Lists.newArrayList(dataSetId));
toolConfig.getTools().add(datasetTool); toolConfig.getTools().add(datasetTool);
agent.setToolConfig(JSONObject.toJSONString(toolConfig)); agent.setToolConfig(JSONObject.toJSONString(toolConfig));
// configure chat apps
Map<String, ChatApp> chatAppConfig = Maps.newHashMap(ChatAppManager.getAllApps());
chatAppConfig.values().forEach(app -> app.setChatModelId(demoChatModel.getId()));
agent.setChatAppConfig(chatAppConfig);
agentService.createAgent(agent, defaultUser); agentService.createAgent(agent, defaultUser);
} }
} }