您最多选择25个主题 主题必须以字母或数字开头,可以包含连字符 (-),并且长度不得超过35个字符
 
 
 

21 行
504 B

  1. package reasoning
  2. import (
  3. "strings"
  4. "github.com/samber/lo"
  5. )
  6. var EffortSuffixes = []string{"-max", "-high", "-medium", "-low", "-minimal"}
  7. // TrimEffortSuffix -> modelName level(low) exists
  8. func TrimEffortSuffix(modelName string) (string, string, bool) {
  9. suffix, found := lo.Find(EffortSuffixes, func(s string) bool {
  10. return strings.HasSuffix(modelName, s)
  11. })
  12. if !found {
  13. return modelName, "", false
  14. }
  15. return strings.TrimSuffix(modelName, suffix), strings.TrimPrefix(suffix, "-"), true
  16. }