fix: text filed check for 4v request (#1634)

2024-07-13 14:57:08 +08:00 · 2024-07-13 14:57:08 +08:00 · 65acb94f45
commit 65acb94f45
parent 6ad169975f
1 changed files with 5 additions and 1 deletions
--- a/relay/adaptor/openai/token.go
+++ b/relay/adaptor/openai/token.go
@ -97,7 +97,11 @@ func CountTokenMessages(messages []model.Message, model string) int {
 				m := it.(map[string]any)
 				switch m["type"] {
 				case "text":
-					tokenNum += getTokenNum(tokenEncoder, m["text"].(string))
+					if textValue, ok := m["text"]; ok {
+						if textString, ok := textValue.(string); ok {
+							tokenNum += getTokenNum(tokenEncoder, textString)
+						}
+					}
 				case "image_url":
 					imageUrl, ok := m["image_url"].(map[string]any)
 					if ok {