From 7d1bad1b372b65ea1a818a746e6d9ba8fb8ae822 Mon Sep 17 00:00:00 2001 From: CaIon Date: Wed, 3 Dec 2025 00:25:05 +0800 Subject: [PATCH] fix(token_counter): correct model name reference in image token estimation --- service/token_counter.go | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/service/token_counter.go b/service/token_counter.go index 4f004b5bb..ebf0e243d 100644 --- a/service/token_counter.go +++ b/service/token_counter.go @@ -317,7 +317,7 @@ func EstimateRequestToken(c *gin.Context, meta *types.TokenCountMeta, info *rela for i, file := range meta.Files { switch file.FileType { case types.FileTypeImage: - if common.IsOpenAITextModel(info.UpstreamModelName) { + if common.IsOpenAITextModel(info.OriginModelName) { token, err := getImageToken(file, model, info.IsStream) if err != nil { return 0, fmt.Errorf("error counting image token, media index[%d], original data[%s], err: %v", i, file.OriginData, err)