添加腾讯混元大模型支持

proto/status.go

@@ -146,6 +146,7 @@ const (
 	ModelTypeOllama                 = "ollama"
 	ModelTypeQianfan                = "qianfan"
 	ModelTypeTongyi                 = "tongyi"
+	ModelTypeHuanyun                = "huanyun"
 	KnowledgeBaseServerResponseType = "kbase_query_resp"
 )
 

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
 	}