mirror of
https://github.com/QuantumNous/new-api.git
synced 2026-03-30 06:01:16 +00:00
21 lines
504 B
Go
21 lines
504 B
Go
package reasoning
|
|
|
|
import (
|
|
"strings"
|
|
|
|
"github.com/samber/lo"
|
|
)
|
|
|
|
var EffortSuffixes = []string{"-max", "-high", "-medium", "-low", "-minimal"}
|
|
|
|
// TrimEffortSuffix -> modelName level(low) exists
|
|
func TrimEffortSuffix(modelName string) (string, string, bool) {
|
|
suffix, found := lo.Find(EffortSuffixes, func(s string) bool {
|
|
return strings.HasSuffix(modelName, s)
|
|
})
|
|
if !found {
|
|
return modelName, "", false
|
|
}
|
|
return strings.TrimSuffix(modelName, suffix), strings.TrimPrefix(suffix, "-"), true
|
|
}
|