From 3d9dea3ea7069cab6a947cd577f8197de6f7321d Mon Sep 17 00:00:00 2001
From: junleea <354425203@qq.com>
Date: Mon, 24 Mar 2025 21:45:16 +0800
Subject: [PATCH] =?UTF-8?q?ollama=E9=80=9A=E8=BF=87=E6=A8=A1=E5=9E=8B?=
 =?UTF-8?q?=E5=8F=82=E6=95=B0=E9=80=89=E6=8B=A9=E5=AF=B9=E5=BA=94=E6=A8=A1?=
 =?UTF-8?q?=E5=9E=8B?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

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

diff --git a/service/imService.go b/service/imService.go
index 78968c5..668869e 100644
--- a/service/imService.go
+++ b/service/imService.go
@@ -84,7 +84,7 @@ func WSReceiveMessageService(userID, sessionID int, channel string, msg proto.WS
 	var resID uint
 	//处理消息
 	if model.Type == "ollama" {
-		resErr, resID = ReceiveOllamaSession(userID, sessionID, channel, msg)
+		resErr, resID = ReceiveOllamaSession(userID, sessionID, channel, msg, modelParam)
 	} else if model.Type == "spark" {
 		resErr, resID = ReceiveSparkSession(userID, sessionID, channel, msg, modelParam)
 	} else {
@@ -93,7 +93,7 @@ func WSReceiveMessageService(userID, sessionID int, channel string, msg proto.WS
 	return resErr, resID
 }
 
-func ReceiveOllamaSession(userID, sessionID int, channel string, msg proto.WSMessageReq) (error, uint) {
+func ReceiveOllamaSession(userID, sessionID int, channel string, msg proto.WSMessageReq, modelParam proto.ModelParam) (error, uint) {
 	var resErr error
 	var resID uint
 	var err error
@@ -109,7 +109,7 @@ func ReceiveOllamaSession(userID, sessionID int, channel string, msg proto.WSMes
 		log.Println("actx has get will use:", len(actx))
 	}
 	req := &api.GenerateRequest{
-		Model:   "qwen2.5:0.5b",
+		Model:   modelParam.Model,
 		Context: actx,
 		Prompt:  prompt,
 	}