修改模型选择方式

service/imService.go

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