ljydlv
/
saw-go
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

添加会话错误消息返回

This commit is contained in:
junleea 2025-05-12 13:57:38 +08:00
parent 1df0cd60e7
commit b4ab058ee3
2 changed files with 7 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
handler/im.go
View File

@ -16,6 +16,7 @@ import (
"net/http" "net/http"
"strconv" "strconv"
"sync" "sync"
"time"
) )
type SMessage struct { type SMessage struct {
@ -133,6 +134,10 @@ func subscribeAndHandleIMMessages(ws *websocket.Conn, userId int) {
err = doReceiveGenChatMessage(userId, &sessionID, &data, chanel) err = doReceiveGenChatMessage(userId, &sessionID, &data, chanel)
if err != nil { if err != nil {
log.Println("do receive gen chat message error:", err) log.Println("do receive gen chat message error:", err)
//返回错误消息
errMsg := proto.WSMessage{Code: proto.OperationFailed, ErrorMsg: "服务器处理错误", SessionID: int(sessionID)}
errMsgStr, _ := json.Marshal(errMsg)
worker.Publish(chanel, string(errMsgStr), time.Second*60)
return return
} }
} }
@ -236,7 +241,7 @@ func doReceiveGenChatMessage(userId int, sessionID *uint, data *proto.WSMessageR
} }
log.Println("create session id:", *sessionID) log.Println("create session id:", *sessionID)
//会话名称过长，使用ai进行总结 //会话名称过长，使用ai进行总结
if len(sessionName) > 30 { if len(sessionName) > 30 && data.Function == "gen-chat" { //只有在通用聊天时才进行ai总结
//通过ai总结会话名称 //通过ai总结会话名称
go func() { go func() {
sessionErr := service.UpdateSessionNameByAIAndID(int(*sessionID), userId, sessionName) sessionErr := service.UpdateSessionNameByAIAndID(int(*sessionID), userId, sessionName)

1
proto/im.go
View File

@ -14,6 +14,7 @@ type AIQueueMessage struct {
type WSMessage struct { type WSMessage struct {
Code int `json:"code"` //返回码,0表示成功， 其它状态码 Code int `json:"code"` //返回码,0表示成功， 其它状态码
Type string `json:"type"` //接收及发送消息类型 Type string `json:"type"` //接收及发送消息类型
ErrorMsg string `json:"error_msg"` //错误信息
Msg AIQueueMessage `json:"msg"` //消息内容,只进行转发，不做处理 Msg AIQueueMessage `json:"msg"` //消息内容,只进行转发，不做处理
SessionID int `json:"session_id"` //应用层会话id SessionID int `json:"session_id"` //应用层会话id
ToID int `json:"to_id"` //接收者id ToID int `json:"to_id"` //接收者id