xinyin025/one-api
1
0
Fork 0
mirror of https://github.com/songquanpeng/one-api.git synced 2025-11-19 22:53:41 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

refactor: use model_mapping instead of deployment_mapping

Browse Source
This commit is contained in:
WqyJh
2023-11-23 16:29:21 +08:00
parent 9042b7c4e4
commit 10291630f1
5 changed files with 26 additions and 66 deletions
Download Patch File Download Diff File Expand all files Collapse all files

31
model/channel.go
View File

@@ -27,7 +27,6 @@ type Channel struct {
UsedQuota int64 `json:"used_quota" gorm:"bigint;default:0"`
ModelMapping *string `json:"model_mapping" gorm:"type:varchar(1024);default:''"`
Priority *int64 `json:"priority" gorm:"bigint;default:0"`
DeploymentMapping *string `json:"deployment_mapping" gorm:"type:varchar(1024);default:''"`
}
func GetAllChannels(startIdx int, num int, selectAll bool) ([]*Channel, error) {
@@ -76,26 +75,6 @@ func BatchInsertChannels(channels []Channel) error {
return nil
}
func (channel *Channel) GetDeploymentMapping() (deploymentMapping map[string]string) {
if channel.DeploymentMapping == nil || *channel.DeploymentMapping == "" {
return
}
err := json.Unmarshal([]byte(*channel.DeploymentMapping), &deploymentMapping)
if err != nil {
common.SysError("failed to unmarshal deployment mapping: " + err.Error())
}
return
}
func (channel *Channel) GetDeployment(model string) string {
deploymentMapping := channel.GetDeploymentMapping()
deployment, ok := deploymentMapping[model]
if !ok {
return ModelToDeployment(model)
}
return deployment
}
func ModelToDeployment(model string) string {
model = strings.Replace(model, ".", "", -1)
// https://github.com/songquanpeng/one-api/issues/67
@@ -119,11 +98,15 @@ func (channel *Channel) GetBaseURL() string {
return *channel.BaseURL
}
func (channel *Channel) GetModelMapping() string {
func (channel *Channel) GetModelMapping() (m map[string]string) {
if channel.ModelMapping == nil {
return ""
return
}
return *channel.ModelMapping
err := json.Unmarshal([]byte(*channel.ModelMapping), &m)
if err != nil {
common.SysError("failed to unmarshal model mapping: " + err.Error())
}
return
}
func (channel *Channel) Insert() error {