本篇博客主要讲解chat的聊天内容的存储和实现,首先对于前端的内容,首先如果是drugchat机器人,需要提前输入smiles分子才能继续进行,如果是drugassist机器人则直接发送相关消息即可,之后再获得输入框中的内容调用询问函数调用后端springboot的接口,从而获得模型的输出。
sendMsg(msgList) {
let uid = this.frinedInfo.id - 1001
console.log("this.friendInfo.id", uid)
console.log(uid)
if(uid === 1){
this.request.get("/model/ask", {
params: {
user_messages: msgList,
}
}).then(res => {
if (res) {
console.log(res)
let params = {
uid: uid,
question: msgList,
answer: res.data,
}
this.request.post("/chat/add", params).then(res => {
this.getFriendChatMsg();
setTimeout(() => {
this.scrollBottom();
},100)
})
} else {
this.$message.error("保存失败")
console.log(res)
}
})
}
if(uid === 2)
{
console.log(msgList)
this.request.get("/model/ask_chat", {
params: {
user_messages: msgList,
}
}).then(res => {
if (res) {
console.log(res.data)
let params = {
uid: uid,
question: msgList,
answer: res.data,
}
this.request.post("/chat/add", params).then(res => {
this.getFriendChatMsg();
setTimeout(() => {
this.scrollBottom();
},100)
})
} else {
this.$message.error("保存失败")
console.log(res)
}
})
}
},
同时我们还实现了ChatController的两个函数功能:保存聊天记录和获取指定chatbot的聊天记录,对于保存聊天记录,只需把内容进行保存,包括question和answer内容,对于获取指定chatbot的聊天记录获取则是根据uid获取,从而实现历史聊天记录的展示,对应函数的实现:
@PostMapping("/add")
public Result save(@RequestBody Chat chat) {
System.out.println(chat.getAnswer());
chatService.save(chat);
return Result.success();
}
@GetMapping("/find")
public Result findByUid(@RequestParam Integer uid){
System.out.println(uid);
QueryWrapper<Chat> qw = new QueryWrapper<>();
qw.eq("uid",uid);
List<Chat> list = chatService.list(qw);
return Result.success(list);
}