修改模型选择方式

This commit is contained in:
junleea 2025-03-24 16:31:39 +08:00
parent 26126a10c8
commit b0b14fef3c
1 changed files with 2 additions and 2 deletions

View File

@ -83,9 +83,9 @@ func WSReceiveMessageService(userID, sessionID int, channel string, msg proto.WS
var resErr error var resErr error
var resID uint var resID uint
//处理消息 //处理消息
if msg.Type == "ollama" { if model.Type == "ollama" {
resErr, resID = ReceiveOllamaSession(userID, sessionID, channel, msg) resErr, resID = ReceiveOllamaSession(userID, sessionID, channel, msg)
} else if msg.Type == "spark" { } else if model.Type == "spark" {
resErr, resID = ReceiveSparkSession(userID, sessionID, channel, msg, modelParam) resErr, resID = ReceiveSparkSession(userID, sessionID, channel, msg, modelParam)
} else { } else {
return errors.New("model not exist"), 0 return errors.New("model not exist"), 0