refactor: simplify model prefix checks and update message role for o-series models

This commit is contained in:
CaIon committed 2025-04-17 16:50:52 +08:00
1 parent 45e8e04d56
commit b364e9e3b1
3 files changed
+11 -11

No files matched your search

+1 -1
View File
@@ -43,7 +43,7 @@ func InitTokenEncoders() {
} else {
tokenEncoderMap[model] = defaultTokenEncoder
}
} else if strings.HasPrefix(model, "o1") {
} else if strings.HasPrefix(model, "o") {
tokenEncoderMap[model] = o200kTokenEncoder
} else {
tokenEncoderMap[model] = defaultTokenEncoder