diff --git a/model/channel.go b/model/channel.go index 19af2263..cdff2edf 100644 --- a/model/channel.go +++ b/model/channel.go @@ -3,6 +3,8 @@ package model import ( "encoding/json" "fmt" + + "github.com/pkg/errors" "github.com/songquanpeng/one-api/common" "github.com/songquanpeng/one-api/common/config" "github.com/songquanpeng/one-api/common/helper" @@ -122,8 +124,12 @@ func (channel *Channel) Update() error { return err } DB.Model(channel).First(channel, "id = ?", channel.Id) - err = channel.UpdateAbilities() - return err + if err = channel.UpdateAbilities(); err != nil { + logger.SysError("failed to update abilities: " + err.Error()) + return errors.Wrap(err, "failed to update abilities") + } + + return nil } func (channel *Channel) UpdateResponseTime(responseTime int64) { diff --git a/relay/controller/text.go b/relay/controller/text.go index e19b571f..36ecd16c 100644 --- a/relay/controller/text.go +++ b/relay/controller/text.go @@ -29,9 +29,16 @@ func RelayTextHelper(c *gin.Context) *model.ErrorWithStatusCode { meta.IsStream = textRequest.Stream // map model name - var isModelMapped bool + var ( + originRequestModel = textRequest.Model + isModelMapped bool + ) meta.OriginModelName = textRequest.Model textRequest.Model, isModelMapped = util.GetMappedModelName(textRequest.Model, meta.ModelMapping) + if isModelMapped { + logger.Info(c.Request.Context(), fmt.Sprintf("rewrite model name from %s to %s", originRequestModel, textRequest.Model)) + } + meta.ActualModelName = textRequest.Model // get model ratio & group ratio modelRatio := common.GetModelRatio(textRequest.Model)