xinyin025/geekai
1
0
Fork 0
mirror of https://github.com/yangjian102621/geekai.git synced 2025-11-10 11:13:42 +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 dbcf14b566
commit 5c9b1e8764
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/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 {