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

(improvement)(Chat) Return agent info after creating agent (#1272)

Browse Source 
Co-authored-by: lxwcodemonkey
This commit is contained in:
LXW
2024-06-28 19:15:36 +08:00
committed by GitHub
parent 5a052248dc
commit 5bcb76a1de
8 changed files with 17 additions and 23 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
chat/server/src/main/java/com/tencent/supersonic/chat/server/rest/AgentController.java
View File

@@ -28,21 +28,19 @@ public class AgentController {
private AgentService agentService;
@PostMapping
public boolean createAgent(@RequestBody Agent agent,
public Agent createAgent(@RequestBody Agent agent,
HttpServletRequest httpServletRequest,
HttpServletResponse httpServletResponse) {
User user = UserHolder.findUser(httpServletRequest, httpServletResponse);
agentService.createAgent(agent, user);
return true;
return agentService.createAgent(agent, user);
}
@PutMapping
public boolean updateAgent(@RequestBody Agent agent,
public Agent updateAgent(@RequestBody Agent agent,
HttpServletRequest httpServletRequest,
HttpServletResponse httpServletResponse) {
User user = UserHolder.findUser(httpServletRequest, httpServletResponse);
agentService.updateAgent(agent, user);
return true;
return agentService.updateAgent(agent, user);
}
@DeleteMapping("/{id}")

4
chat/server/src/main/java/com/tencent/supersonic/chat/server/service/AgentService.java
View File

@@ -8,9 +8,9 @@ public interface AgentService {
List<Agent> getAgents();
Integer createAgent(Agent agent, User user);
Agent createAgent(Agent agent, User user);
void updateAgent(Agent agent, User user);
Agent updateAgent(Agent agent, User user);
Agent getAgent(Integer id);

8
chat/server/src/main/java/com/tencent/supersonic/chat/server/service/impl/AgentServiceImpl.java
View File

@@ -45,19 +45,21 @@ public class AgentServiceImpl extends ServiceImpl<AgentDOMapper, AgentDO>
}
@Override
public Integer createAgent(Agent agent, User user) {
public Agent createAgent(Agent agent, User user) {
agent.createdBy(user.getName());
AgentDO agentDO = convert(agent);
save(agentDO);
agent.setId(agentDO.getId());
executeAgentExamplesAsync(agent);
return agentDO.getId();
return agent;
}
@Override
public void updateAgent(Agent agent, User user) {
public Agent updateAgent(Agent agent, User user) {
agent.updatedBy(user.getName());
updateById(convert(agent));
executeAgentExamplesAsync(agent);
return agent;
}
@Override

2
chat/server/src/main/java/com/tencent/supersonic/chat/server/util/LLMConnHelper.java
View File

@@ -18,7 +18,7 @@ public class LLMConnHelper {
String response = chatLanguageModel.generate("Hi there");
return StringUtils.isNotEmpty(response) ? true : false;
} catch (Exception e) {
log.warn("connect llm failed:", e);
log.warn("connect to llm failed:", e);
throw new InvalidArgumentException(e.getMessage());
}
}