extract code for saving chat history

api/handler/chatimpl/xunfei_handler.go

@@ -21,13 +21,11 @@ import (
 	"geekai/utils"
 	"github.com/gorilla/websocket"
 	"gorm.io/gorm"
-	"html/template"
 	"io"
 	"net/http"
 	"net/url"
 	"strings"
 	"time"
-	"unicode/utf8"
 )
 
 type xunFeiResp struct {
@@ -181,89 +179,10 @@ func (h *ChatHandler) sendXunFeiMessage(
 		}
 
 	}
-
 	// 消息发送成功
 	if len(contents) > 0 {
-		if message.Role == "" {
-			message.Role = "assistant"
-		}
-		message.Content = strings.Join(contents, "")
-		useMsg := types.Message{Role: "user", Content: prompt}
-
-		// 更新上下文消息，如果是调用函数则不需要更新上下文
-		if h.App.SysConfig.EnableContext {
-			chatCtx = append(chatCtx, useMsg)  // 提问消息
-			chatCtx = append(chatCtx, message) // 回复消息
-			h.App.ChatContexts.Put(session.ChatId, chatCtx)
-		}
-
-		// 追加聊天记录
-		// for prompt
-		promptToken, err := utils.CalcTokens(prompt, req.Model)
-		if err != nil {
-			logger.Error(err)
-		}
-		historyUserMsg := model.ChatMessage{
-			UserId:     userVo.Id,
-			ChatId:     session.ChatId,
-			RoleId:     role.Id,
-			Type:       types.PromptMsg,
-			Icon:       userVo.Avatar,
-			Content:    template.HTMLEscapeString(prompt),
-			Tokens:     promptToken,
-			UseContext: true,
-			Model:      req.Model,
-		}
-		historyUserMsg.CreatedAt = promptCreatedAt
-		historyUserMsg.UpdatedAt = promptCreatedAt
-		res := h.DB.Save(&historyUserMsg)
-		if res.Error != nil {
-			logger.Error("failed to save prompt history message: ", res.Error)
-		}
-
-		// for reply
-		// 计算本次对话消耗的总 token 数量
-		replyTokens, _ := utils.CalcTokens(message.Content, req.Model)
-		totalTokens := replyTokens + getTotalTokens(req)
-		historyReplyMsg := model.ChatMessage{
-			UserId:     userVo.Id,
-			ChatId:     session.ChatId,
-			RoleId:     role.Id,
-			Type:       types.ReplyMsg,
-			Icon:       role.Icon,
-			Content:    message.Content,
-			Tokens:     totalTokens,
-			UseContext: true,
-			Model:      req.Model,
-		}
-		historyReplyMsg.CreatedAt = replyCreatedAt
-		historyReplyMsg.UpdatedAt = replyCreatedAt
-		res = h.DB.Create(&historyReplyMsg)
-		if res.Error != nil {
-			logger.Error("failed to save reply history message: ", res.Error)
-		}
-
-		// 更新用户算力
-		h.subUserPower(userVo, session, promptToken, replyTokens)
-
-		// 保存当前会话
-		var chatItem model.ChatItem
-		res = h.DB.Where("chat_id = ?", session.ChatId).First(&chatItem)
-		if res.Error != nil {
-			chatItem.ChatId = session.ChatId
-			chatItem.UserId = session.UserId
-			chatItem.RoleId = role.Id
-			chatItem.ModelId = session.Model.Id
-			if utf8.RuneCountInString(prompt) > 30 {
-				chatItem.Title = string([]rune(prompt)[:30]) + "..."
-			} else {
-				chatItem.Title = prompt
-			}
-			chatItem.Model = req.Model
-			h.DB.Create(&chatItem)
-		}
+		h.saveChatHistory(req, prompt, contents, message, chatCtx, session, role, userVo, promptCreatedAt, replyCreatedAt)
 	}
-
 	return nil
 }