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

refactor: add system config key 'user_init_calls' to init the new register user's api calls

Browse Source
This commit is contained in:
RockYang
2023-06-15 10:06:21 +08:00
parent 01d61ab19b
commit 1e84332119
5 changed files with 23 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
api/go/handler/chat_handler.go
View File

@@ -167,7 +167,7 @@ func (h *ChatHandler) sendMessage(ctx context.Context, session types.ChatSession
if res.Error == nil {
for _, msg := range historyMessages {
ms := types.Message{Role: "user", Content: msg.Content}
if msg.Type == types.REPLY_MSG {
if msg.Type == types.ReplyMsg {
ms.Role = "assistant"
}
chatCtx = append(chatCtx, ms)
@@ -276,7 +276,7 @@ func (h *ChatHandler) sendMessage(ctx context.Context, session types.ChatSession
UserId: userVo.Id,
ChatId: session.ChatId,
RoleId: role.Id,
Type: types.PROMPT_MSG,
Type: types.PromptMsg,
Icon: user.Avatar,
Content: prompt,
Tokens: token,
@@ -297,7 +297,7 @@ func (h *ChatHandler) sendMessage(ctx context.Context, session types.ChatSession
UserId: userVo.Id,
ChatId: session.ChatId,
RoleId: role.Id,
Type: types.REPLY_MSG,
Type: types.ReplyMsg,
Icon: role.Icon,
Content: message.Content,
Tokens: token,