diff --git a/proto/status.go b/proto/status.go index cf1fbe2..7b07867 100644 --- a/proto/status.go +++ b/proto/status.go @@ -146,7 +146,7 @@ const ( ModelTypeOllama = "ollama" ModelTypeQianfan = "qianfan" ModelTypeTongyi = "tongyi" - ModelTypeHuanyun = "huanyun" + ModelTypeHunyuan = "hunyuan" KnowledgeBaseServerResponseType = "kbase_query_resp" ) diff --git a/service/imService.go b/service/imService.go index 042d580..cb2295a 100644 --- a/service/imService.go +++ b/service/imService.go @@ -172,7 +172,7 @@ func WSReceiveMessageServiceV2(modelParam proto.ModelParam, ctx *proto.IMParamCo Qianfan(modelParam, ctx) case proto.ModelTypeTongyi: Tongyi(modelParam, ctx) - case proto.ModelTypeHuanyun: + case proto.ModelTypeHunyuan: generalModel.OpenAI(modelParam, ctx, ctx.ModelType) default: resErr, resID = errors.New("model type not exist:"+ctx.ModelType), 0