42 lines
1.1 KiB
Go
42 lines
1.1 KiB
Go
package service
|
|
|
|
import "strings"
|
|
|
|
// resolveOpenAIForwardModel determines the upstream model for OpenAI-compatible
|
|
// forwarding. Group-level default mapping only applies when the account itself
|
|
// did not match any explicit model_mapping rule.
|
|
func resolveOpenAIForwardModel(account *Account, requestedModel, defaultMappedModel string) string {
|
|
if account == nil {
|
|
if defaultMappedModel != "" {
|
|
return defaultMappedModel
|
|
}
|
|
return requestedModel
|
|
}
|
|
|
|
mappedModel, matched := account.ResolveMappedModel(requestedModel)
|
|
if !matched && defaultMappedModel != "" && !isExplicitCodexModel(requestedModel) {
|
|
return defaultMappedModel
|
|
}
|
|
return mappedModel
|
|
}
|
|
|
|
func isExplicitCodexModel(model string) bool {
|
|
model = strings.TrimSpace(model)
|
|
if model == "" {
|
|
return false
|
|
}
|
|
if strings.Contains(model, "/") {
|
|
parts := strings.Split(model, "/")
|
|
model = parts[len(parts)-1]
|
|
}
|
|
model = strings.ToLower(strings.TrimSpace(model))
|
|
if getNormalizedCodexModel(model) != "" {
|
|
return true
|
|
}
|
|
if strings.HasSuffix(model, "-openai-compact") {
|
|
base := strings.TrimSuffix(model, "-openai-compact")
|
|
return getNormalizedCodexModel(base) != ""
|
|
}
|
|
return false
|
|
}
|