refactor: simplify model prefix checks and update message role for o-series models
This commit is contained in:
1 parent
45e8e04d56
commit
b364e9e3b1
3 files changed
+11
-11
No files matched your search
@@ -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
|
||||
|
||||
Reference in new issue
Block a user