xinyin025/geekai
1
0
Fork 0
mirror of https://github.com/yangjian102621/geekai.git synced 2025-11-08 18:23:45 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

feat: allow to view chat message in manager console

Browse Source
This commit is contained in:
RockYang
2024-02-22 17:16:44 +08:00
parent b99b310306
commit e6bbd83c6b
23 changed files with 688 additions and 26 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
api/handler/chatimpl/azure_handler.go
View File

@@ -126,7 +126,7 @@ func (h *ChatHandler) sendAzureMessage(
if err != nil {
logger.Error(err)
}
historyUserMsg := model.HistoryMessage{
historyUserMsg := model.ChatMessage{
UserId: userVo.Id,
ChatId: session.ChatId,
RoleId: role.Id,
@@ -148,7 +148,7 @@ func (h *ChatHandler) sendAzureMessage(
totalTokens, _ := utils.CalcTokens(message.Content, req.Model)
totalTokens += getTotalTokens(req)
historyReplyMsg := model.HistoryMessage{
historyReplyMsg := model.ChatMessage{
UserId: userVo.Id,
ChatId: session.ChatId,
RoleId: role.Id,

4
api/handler/chatimpl/baidu_handler.go
View File

@@ -151,7 +151,7 @@ func (h *ChatHandler) sendBaiduMessage(
if err != nil {
logger.Error(err)
}
historyUserMsg := model.HistoryMessage{
historyUserMsg := model.ChatMessage{
UserId: userVo.Id,
ChatId: session.ChatId,
RoleId: role.Id,
@@ -173,7 +173,7 @@ func (h *ChatHandler) sendBaiduMessage(
// 计算本次对话消耗的总 token 数量
replyToken, _ := utils.CalcTokens(message.Content, req.Model)
totalTokens := replyToken + getTotalTokens(req)
historyReplyMsg := model.HistoryMessage{
historyReplyMsg := model.ChatMessage{
UserId: userVo.Id,
ChatId: session.ChatId,
RoleId: role.Id,

4
api/handler/chatimpl/chat_handler.go
View File

@@ -325,7 +325,7 @@ func (h *ChatHandler) sendMessage(ctx context.Context, session *types.ChatSessio
// loading recent chat history as chat context
if chatConfig.ContextDeep > 0 {
var historyMessages []model.HistoryMessage
var historyMessages []model.ChatMessage
res := h.db.Debug().Where("chat_id = ? and use_context = 1", session.ChatId).Limit(chatConfig.ContextDeep).Order("id desc").Find(&historyMessages)
if res.Error == nil {
for i := len(historyMessages) - 1; i >= 0; i-- {
@@ -390,7 +390,7 @@ func (h *ChatHandler) Tokens(c *gin.Context) {
// 如果没有传入 text 字段，则说明是获取当前 reply 总的 token 消耗（带上下文）
if data.Text == "" && data.ChatId != "" {
var item model.HistoryMessage
var item model.ChatMessage
userId, _ := c.Get(types.LoginUserID)
res := h.db.Where("user_id = ?", userId).Where("chat_id = ?", data.ChatId).Last(&item)
if res.Error != nil {

4
api/handler/chatimpl/chat_item_handler.go
View File

@@ -95,7 +95,7 @@ func (h *ChatHandler) Clear(c *gin.Context) {
return res.Error
}
res = h.db.Where("user_id = ? AND chat_id IN ?", user.Id, chatIds).Delete(&model.HistoryMessage{})
res = h.db.Where("user_id = ? AND chat_id IN ?", user.Id, chatIds).Delete(&model.ChatMessage{})
if res.Error != nil {
return res.Error
}
@@ -116,7 +116,7 @@ func (h *ChatHandler) Clear(c *gin.Context) {
// History 获取聊天历史记录
func (h *ChatHandler) History(c *gin.Context) {
chatId := c.Query("chat_id") // 会话 ID
var items []model.HistoryMessage
var items []model.ChatMessage
var messages = make([]vo.HistoryMessage, 0)
res := h.db.Where("chat_id = ?", chatId).Find(&items)
if res.Error != nil {

4
api/handler/chatimpl/chatglm_handler.go
View File

@@ -130,7 +130,7 @@ func (h *ChatHandler) sendChatGLMMessage(
if err != nil {
logger.Error(err)
}
historyUserMsg := model.HistoryMessage{
historyUserMsg := model.ChatMessage{
UserId: userVo.Id,
ChatId: session.ChatId,
RoleId: role.Id,
@@ -152,7 +152,7 @@ func (h *ChatHandler) sendChatGLMMessage(
// 计算本次对话消耗的总 token 数量
replyToken, _ := utils.CalcTokens(message.Content, req.Model)
totalTokens := replyToken + getTotalTokens(req)
historyReplyMsg := model.HistoryMessage{
historyReplyMsg := model.ChatMessage{
UserId: userVo.Id,
ChatId: session.ChatId,
RoleId: role.Id,

6
api/handler/chatimpl/openai_handler.go
View File

@@ -46,6 +46,8 @@ func (h *ChatHandler) sendOpenAiMessage(
utils.ReplyMessage(ws, ErrorMsg)
utils.ReplyMessage(ws, ErrImg)
all, _ := io.ReadAll(response.Body)
logger.Error(string(all))
return err
} else {
defer response.Body.Close()
@@ -197,7 +199,7 @@ func (h *ChatHandler) sendOpenAiMessage(
if err != nil {
logger.Error(err)
}
historyUserMsg := model.HistoryMessage{
historyUserMsg := model.ChatMessage{
UserId: userVo.Id,
ChatId: session.ChatId,
RoleId: role.Id,
@@ -227,7 +229,7 @@ func (h *ChatHandler) sendOpenAiMessage(
}
totalTokens += getTotalTokens(req)
historyReplyMsg := model.HistoryMessage{
historyReplyMsg := model.ChatMessage{
UserId: userVo.Id,
ChatId: session.ChatId,
RoleId: role.Id,

4
api/handler/chatimpl/qwen_handler.go
View File

@@ -151,7 +151,7 @@ func (h *ChatHandler) sendQWenMessage(
if err != nil {
logger.Error(err)
}
historyUserMsg := model.HistoryMessage{
historyUserMsg := model.ChatMessage{
UserId: userVo.Id,
ChatId: session.ChatId,
RoleId: role.Id,
@@ -173,7 +173,7 @@ func (h *ChatHandler) sendQWenMessage(
// 计算本次对话消耗的总 token 数量
replyToken, _ := utils.CalcTokens(message.Content, req.Model)
totalTokens := replyToken + getTotalTokens(req)
historyReplyMsg := model.HistoryMessage{
historyReplyMsg := model.ChatMessage{
UserId: userVo.Id,
ChatId: session.ChatId,
RoleId: role.Id,

4
api/handler/chatimpl/xunfei_handler.go
View File

@@ -189,7 +189,7 @@ func (h *ChatHandler) sendXunFeiMessage(
if err != nil {
logger.Error(err)
}
historyUserMsg := model.HistoryMessage{
historyUserMsg := model.ChatMessage{
UserId: userVo.Id,
ChatId: session.ChatId,
RoleId: role.Id,
@@ -211,7 +211,7 @@ func (h *ChatHandler) sendXunFeiMessage(
// 计算本次对话消耗的总 token 数量
replyToken, _ := utils.CalcTokens(message.Content, req.Model)
totalTokens := replyToken + getTotalTokens(req)
historyReplyMsg := model.HistoryMessage{
historyReplyMsg := model.ChatMessage{
UserId: userVo.Id,
ChatId: session.ChatId,
RoleId: role.Id,