Upload main.go

main.go

@@ -226,23 +226,50 @@ func countTokensWithClaude(req TokenCountRequest) (TokenCountResponse, error) {
 	// 准备请求Anthropic API
 	log.Printf("开始Claude API请求: 模型=%s, 消息数量=%d", req.Model, len(req.Messages))
 
-	// 验证消息格式
+	// 验证并过滤空内容的消息
+	var filteredMessages []Message
 	for i, msg := range req.Messages {
 		if msg.Content == "" {
-			log.Printf("警告: 消息 #%d 内容为空，可能导致请求失败", i)
+			log.Printf("警告: 消息 #%d 内容为空，将被过滤掉", i)
+			continue // 跳过空内容消息
 		}
 		if msg.Role != "user" && msg.Role != "assistant" {
 			log.Printf("警告: 消息 #%d 角色'%s'不是标准角色(user/assistant)，可能导致请求失败", i, msg.Role)
 		}
+		filteredMessages = append(filteredMessages, msg)
+	}
+
+	if len(filteredMessages) == 0 {
+		log.Printf("错误: 过滤后没有有效消息")
+		return TokenCountResponse{}, fmt.Errorf("没有有效消息：所有消息内容都为空")
+	}
+
+	// 创建新请求，使用过滤后的消息
+	filteredReq := TokenCountRequest{
+		Model:    req.Model,
+		Messages: filteredMessages,
+		System:   req.System,
+	}
+
+	// 记录过滤后的消息数量
+	if len(filteredMessages) != len(req.Messages) {
+		log.Printf("消息过滤: 原始消息数=%d, 过滤后消息数=%d", len(req.Messages), len(filteredMessages))
 	}
 
 	client := &http.Client{}
-	data, err := json.Marshal(req)
+	data, err := json.Marshal(filteredReq)
 	if err != nil {
 		log.Printf("错误: 序列化Claude请求失败: %v", err)
 		return TokenCountResponse{}, fmt.Errorf("序列化请求失败: %v", err)
 	}
 
+	// 记录请求内容用于调试
+	if len(data) < 1000 {
+		log.Printf("Claude请求内容: %s", string(data))
+	} else {
+		log.Printf("Claude请求内容较大，长度=%d字节", len(data))
+	}
+
 	// 创建请求
 	request, err := http.NewRequest("POST", "https://api.anthropic.com/v1/messages/count_tokens", bytes.NewBuffer(data))
 	if err != nil {