From b0b14fef3c53f29870296afbe1b60ab5eeb168a1 Mon Sep 17 00:00:00 2001
From: junleea <354425203@qq.com>
Date: Mon, 24 Mar 2025 16:31:39 +0800
Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E6=94=B9=E6=A8=A1=E5=9E=8B=E9=80=89?=
 =?UTF-8?q?=E6=8B=A9=E6=96=B9=E5=BC=8F?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 service/imService.go | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

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