Upload main.go

main.go

@@ -65,14 +65,7 @@ var modelRules = []ModelRule{
 		Keywords: []string{"deepseek"},
 		Target:   "deepseek-v3",
 	},
-	{
-		Keywords: []string{"claude", "3", "sonnet"},
-		Target:   "claude-3-sonnet-20240229",
-	},
-	{
-		Keywords: []string{"claude", "3", "7"},
-		Target:   "claude-3-7-sonnet-latest",
-	},
+	// 先放更具体的规则
 	{
 		Keywords: []string{"claude", "3", "5", "sonnet"},
 		Target:   "claude-3-5-sonnet-latest",
@@ -81,6 +74,10 @@ var modelRules = []ModelRule{
 		Keywords: []string{"claude", "3", "5", "haiku"},
 		Target:   "claude-3-5-haiku-latest",
 	},
+	{
+		Keywords: []string{"claude", "3", "7"},
+		Target:   "claude-3-7-sonnet-latest",
+	},
 	{
 		Keywords: []string{"claude", "3", "opus"},
 		Target:   "claude-3-opus-latest",
@@ -89,6 +86,11 @@ var modelRules = []ModelRule{
 		Keywords: []string{"claude", "3", "haiku"},
 		Target:   "claude-3-haiku-20240307",
 	},
+	// 再放一般规则
+	{
+		Keywords: []string{"claude", "3", "sonnet"},
+		Target:   "claude-3-sonnet-20240229",
+	},
 	{
 		Keywords: []string{"gemini", "2.0"},
 		Target:   "gemini-2.0-flash",
@@ -107,17 +109,43 @@ var modelRules = []ModelRule{
 func matchModelName(input string) string {
 	// 转换为小写进行匹配
 	input = strings.ToLower(input)
+	log.Printf("正在匹配模型名称: %s", input)
+
+	// 特殊处理 Claude 3.5 系列
+	if strings.Contains(input, "claude") && strings.Contains(input, "3.5") ||
+		strings.Contains(input, "claude") && strings.Contains(input, "3") && strings.Contains(input, "5") {
+		if strings.Contains(input, "sonnet") {
+			log.Printf("匹配到Claude 3.5 Sonnet")
+			return "claude-3-5-sonnet-latest"
+		} else if strings.Contains(input, "haiku") {
+			log.Printf("匹配到Claude 3.5 Haiku")
+			return "claude-3-5-haiku-latest"
+		} else {
+			// 默认为Sonnet
+			log.Printf("匹配到Claude 3.5 (默认使用Sonnet)")
+			return "claude-3-5-sonnet-latest"
+		}
+	}
+
+	// 特殊处理 Claude 3.7 系列
+	if strings.Contains(input, "claude") && strings.Contains(input, "3.7") ||
+		strings.Contains(input, "claude") && strings.Contains(input, "3") && strings.Contains(input, "7") {
+		log.Printf("匹配到Claude 3.7")
+		return "claude-3-7-sonnet-latest"
+	}
 
 	// 特殊规则：OpenAI GPT-4o
 	if (strings.Contains(input, "gpt") && strings.Contains(input, "4o")) ||
 		strings.Contains(input, "o1") ||
 		strings.Contains(input, "o3") {
+		log.Printf("匹配到GPT-4o")
 		return "gpt-4o"
 	}
 
 	// 特殊规则：OpenAI GPT-4
 	if (strings.Contains(input, "gpt") && strings.Contains(input, "3") && strings.Contains(input, "5")) ||
 		(strings.Contains(input, "gpt") && strings.Contains(input, "4") && !strings.Contains(input, "4o")) {
+		log.Printf("匹配到GPT-4")
 		return "gpt-4"
 	}
 
@@ -131,11 +159,13 @@ func matchModelName(input string) string {
 			}
 		}
 		if matches {
+			log.Printf("通过规则匹配到: %s", rule.Target)
 			return rule.Target
 		}
 	}
 
 	// 如果没有匹配到，返回原始输入
+	log.Printf("没有匹配到任何规则，使用原始输入: %s", input)
 	return input
 }
 
@@ -196,6 +226,16 @@ func countTokensWithClaude(req TokenCountRequest) (TokenCountResponse, error) {
 	// 准备请求Anthropic API
 	log.Printf("开始Claude API请求: 模型=%s, 消息数量=%d", req.Model, len(req.Messages))
 
+	// 验证消息格式
+	for i, msg := range req.Messages {
+		if msg.Content == "" {
+			log.Printf("警告: 消息 #%d 内容为空，可能导致请求失败", i)
+		}
+		if msg.Role != "user" && msg.Role != "assistant" {
+			log.Printf("警告: 消息 #%d 角色'%s'不是标准角色(user/assistant)，可能导致请求失败", i, msg.Role)
+		}
+	}
+
 	client := &http.Client{}
 	data, err := json.Marshal(req)
 	if err != nil {
@@ -230,9 +270,19 @@ func countTokensWithClaude(req TokenCountRequest) (TokenCountResponse, error) {
 		errorBody, _ = io.ReadAll(response.Body)
 		log.Printf("错误: Claude API返回非200状态码: %d, 响应体: %s", response.StatusCode, string(errorBody))
 
-		if response.StatusCode == http.StatusUnauthorized {
-			log.Printf("错误: Claude API密钥可能无效或过期")
+		// 检查常见错误
+		errorStr := string(errorBody)
+		if response.StatusCode == http.StatusUnauthorized || strings.Contains(errorStr, "invalid_api_key") {
+			log.Printf("错误: Claude API密钥无效或过期")
 			return TokenCountResponse{}, fmt.Errorf("Claude API验证失败，请检查API Key是否有效: %s", string(errorBody))
+		} else if response.StatusCode == http.StatusBadRequest {
+			if strings.Contains(errorStr, "empty content") {
+				log.Printf("错误: 请求包含空内容的消息")
+				return TokenCountResponse{}, fmt.Errorf("请求格式错误: 消息不能有空内容: %s", string(errorBody))
+			} else if strings.Contains(errorStr, "invalid_request_error") {
+				log.Printf("错误: 无效的请求格式")
+				return TokenCountResponse{}, fmt.Errorf("无效的请求格式: %s", string(errorBody))
+			}
 		}
 
 		return TokenCountResponse{}, fmt.Errorf("Claude API返回错误状态码: %d, 响应: %s", response.StatusCode, string(errorBody))
@@ -529,8 +579,34 @@ func countTokens(c *gin.Context) {
 
 	if err != nil {
 		log.Printf("计算token失败: %v", err)
-		c.JSON(http.StatusInternalServerError, ErrorResponse{Error: err.Error()})
-		return
+
+		// 对所有API调用失败的情况尝试使用GPT-4o估算
+		log.Printf("API调用失败，尝试使用GPT-4o估算: 原始模型=%s, 错误=%v", req.Model, err)
+
+		// 创建新的请求，使用GPT-4o
+		gptReq := req
+		gptReq.Model = "gpt-4o"
+
+		// 使用OpenAI API进行估算
+		estimatedResult, estimateErr := countTokensWithOpenAI(gptReq)
+
+		if estimateErr == nil {
+			log.Printf("使用GPT-4o估算成功: 模型=%s, 估算tokens=%d", originalModel, estimatedResult.InputTokens)
+
+			// 返回估算值，但添加警告信息和原始错误
+			c.JSON(http.StatusOK, gin.H{
+				"input_tokens":   estimatedResult.InputTokens,
+				"warning":        fmt.Sprintf("Token calculation for model '%s' failed. This is an estimation based on gpt-4o and may not be accurate.", originalModel),
+				"estimated_with": "gpt-4o",
+				"original_error": err.Error(),
+			})
+			return
+		} else {
+			log.Printf("使用GPT-4o估算也失败: %v", estimateErr)
+			// 如果GPT-4o估算也失败，返回原始错误
+			c.JSON(http.StatusInternalServerError, ErrorResponse{Error: err.Error()})
+			return
+		}
 	}
 
 	// 返回结果