package rag import ( "encoding/json" "eta/eta_api/controllers" "eta/eta_api/controllers/rag/llm_http" "eta/eta_api/models" "eta/eta_api/models/llm" llmService "eta/eta_api/services/llm" "eta/eta_api/utils" "time" ) type UserChatController struct { controllers.BaseAuthController } // NewChat @Title 新建对话框 // @Description 新建对话框 // @Success 101 {object} response.ListResp // @router /chat/new_chat [post] func (ucCtrl *UserChatController) NewChat() { br := new(models.BaseResponse).Init() defer func() { ucCtrl.Data["json"] = br ucCtrl.ServeJSON() }() var req llm_http.UserChatReq err := json.Unmarshal(ucCtrl.Ctx.Input.RequestBody, &req) if err != nil { br.Msg = "参数解析异常!" br.ErrMsg = "参数解析失败,Err:" + err.Error() return } sysUser := ucCtrl.SysUser if sysUser == nil { br.Msg = "请登录" br.ErrMsg = "请登录,SysUser Is Empty" br.Ret = 408 return } if req.ChatTitle == "" { req.ChatTitle = "新会话" } session := llm.UserLlmChat{ UserId: sysUser.AdminId, CreatedTime: time.Now(), ChatTitle: req.ChatTitle, } var chatResp = new(llm_http.UserChatResp) chatResp.ChatTitle = req.ChatTitle chatResp.ChatId, err = session.CreateChatSession() if err != nil { br.Msg = "创建失败" br.ErrMsg = "创建失败,Err:" + err.Error() return } _ = llmService.AddChatRecord(&llm.UserChatRecordRedis{ ChatId: session.Id, ChatUserType: "user", Content: req.ChatTitle, SendTime: time.Now().Format(utils.FormatDateTime), }) br.Data = chatResp br.Ret = 200 br.Success = true br.Msg = "创建成功" } // RenameChat @Title 新建对话框 // @Description 新建对话框 // @Success 101 {object} response.ListResp // @router /chat/rename_chat [post] func (ucCtrl *UserChatController) RenameChat() { br := new(models.BaseResponse).Init() defer func() { ucCtrl.Data["json"] = br ucCtrl.ServeJSON() }() var req llm_http.UserChatReq err := json.Unmarshal(ucCtrl.Ctx.Input.RequestBody, &req) if err != nil { br.Msg = "参数解析异常!" br.ErrMsg = "参数解析失败,Err:" + err.Error() return } sysUser := ucCtrl.SysUser if sysUser == nil { br.Msg = "请登录" br.ErrMsg = "请登录,SysUser Is Empty" br.Ret = 408 return } if req.ChatId <= 0 { br.Msg = "非法的对话框Id" br.ErrMsg = "非法的对话框Id" return } if req.ChatTitle == "" { br.Msg = "重命名不能为空" br.ErrMsg = "重命名不能为空" return } session := llm.UserLlmChat{ Id: req.ChatId, UpdateTime: time.Now(), UserId: sysUser.AdminId, ChatTitle: req.ChatTitle, } err = session.RenameChatSession() if err != nil { br.Msg = "重命名失败" br.ErrMsg = "重命名失败,Err:" + err.Error() return } br.Ret = 200 br.Success = true br.Msg = "重命名成功" } // DeleteChat @Title 删除对话框 // @Description 删除对话框 // @Success 101 {object} response.ListResp // @router /chat/delete_chat [post] func (ucCtrl *UserChatController) DeleteChat() { br := new(models.BaseResponse).Init() defer func() { ucCtrl.Data["json"] = br ucCtrl.ServeJSON() }() var req llm_http.UserChatReq err := json.Unmarshal(ucCtrl.Ctx.Input.RequestBody, &req) if err != nil { br.Msg = "参数解析异常!" br.ErrMsg = "参数解析失败,Err:" + err.Error() return } sysUser := ucCtrl.SysUser if sysUser == nil { br.Msg = "请登录" br.ErrMsg = "请登录,SysUser Is Empty" br.Ret = 408 return } if req.ChatId <= 0 { br.Msg = "非法的对话框Id" br.ErrMsg = "非法的对话框Id" return } session := llm.UserLlmChat{ Id: req.ChatId, UpdateTime: time.Now(), IsDeleted: 1, } err = session.DeleteChatSession() if err != nil { br.Msg = "删除失败" br.ErrMsg = "删除失败,Err:" + err.Error() return } br.Ret = 200 br.Success = true br.Msg = "删除成功" } // GetUserChatList @Title 获取用户对话框列表 // @Description 获取用户对话框列表 // @Success 101 {object} response.ListResp // @router /chat/user_chat_list [get] func (ucCtrl *UserChatController) GetUserChatList() { br := new(models.BaseResponse).Init() defer func() { ucCtrl.Data["json"] = br ucCtrl.ServeJSON() }() sysUser := ucCtrl.SysUser if sysUser == nil { br.Msg = "请登录" br.ErrMsg = "请登录,SysUser Is Empty" br.Ret = 408 return } //周日是0,周六是6 weekDay := time.Now().Weekday() offset := int(time.Monday - weekDay) if offset > 0 { offset -= 7 } today := time.Now().Format(utils.FormatDate) monDay := time.Now().AddDate(0, 0, offset).Format(utils.FormatDate) yesterday := time.Now().AddDate(0, 0, -1).Format(utils.FormatDate) chatList, err := llm.GetUserChatList(sysUser.AdminId, monDay, time.Now().Format(utils.FormatDate)) if err != nil { br.Msg = "获取用户聊天列表失败" br.ErrMsg = "获取用户聊天列表失败,Err:" + err.Error() return } data := new(llm_http.UserChatListResp) data.WeekList = make([]llm.UserLlmChatListViewItem, 0) data.YesterdayList = make([]llm.UserLlmChatListViewItem, 0) data.TodayList = make([]llm.UserLlmChatListViewItem, 0) for _, v := range chatList { if v.CreatedTime.Format(utils.FormatDate) == today { data.TodayList = append(data.TodayList, llm.CovertItemToView(v)) } else if v.CreatedTime.Format(utils.FormatDate) == yesterday { data.YesterdayList = append(data.YesterdayList, llm.CovertItemToView(v)) } else { data.WeekList = append(data.WeekList, llm.CovertItemToView(v)) } } br.Data = data br.Ret = 200 br.Success = true br.Msg = "获取用户聊天列表成功" } // ChatRecordAdd @Title 保存聊天记录 // @Description 保存聊天记录 // @Success 101 {object} response.ListResp // @router /chat/chat_record_save [post] func (ucCtrl *UserChatController) ChatRecordAdd() { br := new(models.BaseResponse).Init() defer func() { ucCtrl.Data["json"] = br ucCtrl.ServeJSON() }() var req llm_http.UserChatRecordReq err := json.Unmarshal(ucCtrl.Ctx.Input.RequestBody, &req) if err != nil { br.Msg = "参数解析异常!" br.ErrMsg = "参数解析失败,Err:" + err.Error() return } sysUser := ucCtrl.SysUser if sysUser == nil { br.Msg = "请登录" br.ErrMsg = "请登录,SysUser Is Empty" br.Ret = 408 return } if req.ChatId <= 0 { br.Msg = "非法的对话框Id" br.ErrMsg = "非法的对话框Id" return } if req.Content == "" { br.Msg = "聊天记录不能为空" br.ErrMsg = "聊天记录不能为空" return } if req.Id < 0 { br.Msg = "非法的Id" br.ErrMsg = "非法的Id" return } if req.ChatUserType != "user" && req.ChatUserType != "assistant" { br.Msg = "非法的用户类型" br.ErrMsg = "非法的用户类型,用户类型支持:user/assistant" return } if req.SendTime == "" { req.SendTime = time.Now().Format(utils.FormatDateTime) } else { _, err = time.Parse(utils.FormatDateTime, req.SendTime) if err != nil { br.Msg = "非法的发送时间" br.ErrMsg = "非法的发送时间,Err:" + err.Error() return } } record := llm.UserChatRecordRedis{ ChatId: req.ChatId, ChatUserType: req.ChatUserType, Content: req.Content, SendTime: req.SendTime, } err = llmService.AddChatRecord(&record) if err != nil { br.Msg = "添加聊天记录失败" br.ErrMsg = "添加聊天记录失败,Err:" + err.Error() return } br.Ret = 200 br.Success = true br.Msg = "添加聊天记录成功" } // ChatRecordList @Title 获取聊天记录 // @Description 获取聊天记录 // @Success 101 {object} response.ListResp // @router /chat/chat_record_list [get] func (ucCtrl *UserChatController) ChatRecordList() { br := new(models.BaseResponse).Init() defer func() { ucCtrl.Data["json"] = br ucCtrl.ServeJSON() }() sysUser := ucCtrl.SysUser if sysUser == nil { br.Msg = "请登录" br.ErrMsg = "请登录,SysUser Is Empty" br.Ret = 408 return } chatId, _ := ucCtrl.GetInt("ChatId", 0) if chatId <= 0 { br.Msg = "非法的对话Id" br.ErrMsg = "非法的对话Id" return } list, err := llmService.GetChatRecordsFromRedis(chatId) if err != nil { br.Msg = "获取聊天记录失败" br.ErrMsg = "获取聊天记录失败,Err:" + err.Error() return } br.Data = list br.Ret = 200 br.Success = true br.Msg = "获取聊天记录成功" }