From de7ed81607959e08baba5641654115947a4549c8 Mon Sep 17 00:00:00 2001
From: junleea <354425203@qq.com>
Date: Mon, 12 May 2025 12:11:44 +0800
Subject: [PATCH] =?UTF-8?q?=E6=B7=BB=E5=8A=A0=E8=85=BE=E8=AE=AF=E6=B7=B7?=
 =?UTF-8?q?=E5=85=83=E5=A4=A7=E6=A8=A1=E5=9E=8B=E6=94=AF=E6=8C=81?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 proto/status.go      | 1 +
 service/imService.go | 3 +++
 2 files changed, 4 insertions(+)

diff --git a/proto/status.go b/proto/status.go
index 8f93711..cf1fbe2 100644
--- a/proto/status.go
+++ b/proto/status.go
@@ -146,6 +146,7 @@ const (
 	ModelTypeOllama                 = "ollama"
 	ModelTypeQianfan                = "qianfan"
 	ModelTypeTongyi                 = "tongyi"
+	ModelTypeHuanyun                = "huanyun"
 	KnowledgeBaseServerResponseType = "kbase_query_resp"
 )
 
diff --git a/service/imService.go b/service/imService.go
index 297d73f..86054c1 100644
--- a/service/imService.go
+++ b/service/imService.go
@@ -3,6 +3,7 @@ package service
 import (
 	"StuAcaWorksAI/dao"
 	"StuAcaWorksAI/proto"
+	"StuAcaWorksAI/service/generalModel"
 	"StuAcaWorksAI/worker"
 	"context"
 	"encoding/json"
@@ -138,6 +139,8 @@ func WSReceiveMessageServiceV2(modelParam proto.ModelParam, ctx *proto.IMParamCo
 		Qianfan(modelParam, ctx)
 	case proto.ModelTypeTongyi:
 		Tongyi(modelParam, ctx)
+	case proto.ModelTypeHuanyun:
+		generalModel.OpenAI(modelParam, ctx, ctx.ModelType)
 	default:
 		resErr, resID = errors.New("model type not exist:"+ctx.ModelType), 0
 	}