feat: add video URL support in MediaContent and update token counting logic

2025-05-03 21:12:07 +08:00
parent f5c2fda22a
commit 4bf0d094c4
3 changed files with 19 additions and 2 deletions
@@ -400,6 +400,8 @@ func CountTokenMessages(info *relaycommon.RelayInfo, messages []dto.Message, mod
 					tokenNum += 100
 				} else if m.Type == dto.ContentTypeFile {
 					tokenNum += 5000
+				} else if m.Type == dto.ContentTypeVideoUrl {
+					tokenNum += 5000
 				} else {
 					tokenNum += getTokenNum(tokenEncoder, m.Text)
 				}