山东大学软件学院创新项目实训-基于LLM的药物分子性质分析平台(十三)

本篇博客主要讲解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);
    }

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值