| package reasoning | |
| import ( | |
| "strings" | |
| "github.com/samber/lo" | |
| ) | |
| var EffortSuffixes = []string{"-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 | |
| } | |