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

2025-04-17 16:50:52 +08:00
parent 995c19a997
commit ee4f27d01b
3 changed files with 11 additions and 11 deletions
--- a/service/token_counter.go
+++ b/service/token_counter.go
@@ -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