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

add ws handler

Browse Source
This commit is contained in:
RockYang
2024-09-25 18:43:12 +08:00
parent dfd2be1265
commit 478bc32ddd
9 changed files with 136 additions and 33 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
api/handler/markmap_handler.go
View File

@@ -87,7 +87,7 @@ func (h *MarkMapHandler) Client(c *gin.Context) {
logger.Error(err)
utils.ReplyErrorMessage(client, err.Error())
} else {
utils.ReplyMessage(client, types.ReplyMessage{Type: types.WsEnd})
utils.SendMessage(client, types.ReplyMessage{Type: types.WsMsgTypeEnd})
}
}
@@ -170,13 +170,13 @@ func (h *MarkMapHandler) sendMessage(client *types.WsClient, prompt string, mode
break
}
utils.ReplyChunkMessage(client, types.ReplyMessage{
Type: types.WsContent,
utils.SendChunkMessage(client, types.ReplyMessage{
Type: types.WsMsgTypeContent,
Content: utils.InterfaceToString(responseBody.Choices[0].Delta.Content),
})
} // end for
utils.ReplyChunkMessage(client, types.ReplyMessage{Type: types.WsEnd})
utils.SendChunkMessage(client, types.ReplyMessage{Type: types.WsMsgTypeEnd})
} else {
body, _ := io.ReadAll(response.Body)