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

将spark移动到service

This commit is contained in:
junleea 2025-03-25 14:44:58 +08:00
parent e8a268ca22
commit d0d9cbd294
2 changed files with 5 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
service/imService.go
View File

@ -171,6 +171,6 @@ func ReceiveOllamaSession(userID, sessionID int, channel string, msg proto.WSMes
func ReceiveSparkSession(userID, sessionID int, channel string, msg proto.WSMessageReq, modelParam proto.ModelParam) (error, uint) {
var resErr error
var resID uint
worker.Spark(modelParam, msg.Msg, channel, sessionID, userID)
Spark(modelParam, msg.Msg, channel, sessionID, userID)
return resErr, resID
}

8
worker/spark.go → service/spark.go
View File

@ -1,9 +1,9 @@
package worker
package service
import (
"StuAcaWorksAI/dao"
"StuAcaWorksAI/proto"
"StuAcaWorksAI/service"
"StuAcaWorksAI/worker"
"crypto/hmac"
"crypto/sha256"
"encoding/base64"
@ -164,7 +164,7 @@ func Spark(modelParam proto.ModelParam, question, channel string, SessionID, Use
} else {
answer += data.Payload.Choices.Text[0].Content
}
go service.CreateAIStreamMsg(UserID, 4, SessionID, data, modelParam.Model)
go CreateAIStreamMsg(UserID, 4, SessionID, data, modelParam.Model)
SparkToGeneralMassageAndSendMsgQueue(&data, channel, SessionID, UserID)
}
@ -198,7 +198,7 @@ func SparkToGeneralMassageAndSendMsgQueue(data *SparkResponse, channel string, S
//发送消息
wsMsgStr, _ := json.Marshal(wsMsg)
Publish(channel, string(wsMsgStr), time.Second*60)
worker.Publish(channel, string(wsMsgStr), time.Second*60)
}
// 生成参数