xinyin025/new-api
1
0
Fork 0
mirror of https://github.com/linux-do/new-api.git synced 2025-09-17 16:06:38 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

fix: close #218

Browse Source
This commit is contained in:
CaIon 2024-04-24 14:44:24 +08:00
parent 77e7d11151
commit a3ccc92f55
1 changed files with 2 additions and 16 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
service/token_counter.go
View File

@ -196,22 +196,8 @@ func CountTokenMessages(messages []dto.Message, model string, checkSensitive boo
if model == "glm-4v" { if model == "glm-4v" {
imageTokenNum = 1047 imageTokenNum = 1047
} else { } else {
if str, ok := m.ImageUrl.(string); ok { imageUrl := m.ImageUrl.(dto.MessageImageUrl)
imageTokenNum, err = getImageToken(&dto.MessageImageUrl{Url: str, Detail: "auto"}) imageTokenNum, err = getImageToken(&imageUrl)
} else {
imageUrlMap := m.ImageUrl.(map[string]interface{})
detail, ok := imageUrlMap["detail"]
if ok {
imageUrlMap["detail"] = detail.(string)
} else {
imageUrlMap["detail"] = "auto"
}
imageUrl := dto.MessageImageUrl{
Url: imageUrlMap["url"].(string),
Detail: imageUrlMap["detail"].(string),
}
imageTokenNum, err = getImageToken(&imageUrl)
}
if err != nil { if err != nil {
return 0, err, false return 0, err, false
} }