diff --git a/service/imService.go b/service/imService.go
index ec0d797..78968c5 100644
--- a/service/imService.go
+++ b/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