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

2025-04-17 16:50:52 +08:00
parent 45e8e04d56
commit b364e9e3b1
3 changed files with 11 additions and 11 deletions
@@ -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