From a3ccc92f55317ce5184706e909f0d861ea8a50a4 Mon Sep 17 00:00:00 2001
From: CaIon <1808837298@qq.com>
Date: Wed, 24 Apr 2024 14:44:24 +0800
Subject: [PATCH] fix: close #218

---
 service/token_counter.go | 18 ++----------------
 1 file changed, 2 insertions(+), 16 deletions(-)

diff --git a/service/token_counter.go b/service/token_counter.go
index 52828e9..18fc5a3 100644
--- a/service/token_counter.go
+++ b/service/token_counter.go
@@ -196,22 +196,8 @@ func CountTokenMessages(messages []dto.Message, model string, checkSensitive boo
 						if model == "glm-4v" {
 							imageTokenNum = 1047
 						} else {
-							if str, ok := m.ImageUrl.(string); ok {
-								imageTokenNum, err = getImageToken(&dto.MessageImageUrl{Url: str, Detail: "auto"})
-							} 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)
-							}
+							imageUrl := m.ImageUrl.(dto.MessageImageUrl)
+							imageTokenNum, err = getImageToken(&imageUrl)
 							if err != nil {
 								return 0, err, false
 							}