ai-gateway/relay/util/model_mapping.go

13 lines
278 B
Go
Raw Normal View History

2024-01-21 14:56:20 +00:00
package util
func GetMappedModelName(modelName string, mapping map[string]string) (string, bool) {
if mapping == nil {
return modelName, false
}
mappedModelName := mapping[modelName]
if mappedModelName != "" {
return mappedModelName, true
}
return modelName, false
}