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) { func ReceiveSparkSession(userID, sessionID int, channel string, msg proto.WSMessageReq, modelParam proto.ModelParam) (error, uint) {
var resErr error var resErr error
var resID uint var resID uint
worker.Spark(modelParam, msg.Msg, channel, sessionID, userID) Spark(modelParam, msg.Msg, channel, sessionID, userID)
return resErr, resID return resErr, resID
} }

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

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