Upload main.go

main.go

@@ -545,15 +545,18 @@ func countTokens(c *gin.Context) {
 
 		if estimateErr == nil {
 			log.Printf("使用GPT-4o估算成功: 模型=%s, 估算tokens=%d", originalModel, estimatedResult.InputTokens)
-			// 返回估算值，但添加警告信息
-			c.JSON(http.StatusOK, gin.H{
+			// 返回估算值，但添加警告信息，使用400状态码
+			c.JSON(http.StatusBadRequest, gin.H{
 				"input_tokens":   estimatedResult.InputTokens,
 				"warning":        fmt.Sprintf("The tokenizer for model '%s' is not supported yet. This is an estimation based on gpt-4o and may not be accurate.", originalModel),
 				"estimated_with": "gpt-4o",
+				"error":          fmt.Sprintf("Unsupported model: %s", originalModel),
 			})
 			return
 		} else {
 			log.Printf("使用GPT-4o估算失败: %v", estimateErr)
+			c.JSON(http.StatusBadRequest, ErrorResponse{Error: fmt.Sprintf("Failed to estimate tokens for unsupported model: %s", originalModel)})
+			return
 		}
 	} else {
 		log.Printf("完全不支持的模型: %s, 将尝试使用GPT-4o估算", originalModel)
@@ -565,10 +568,11 @@ func countTokens(c *gin.Context) {
 		estimatedResult, estimateErr := countTokensWithOpenAI(gptReq)
 		if estimateErr == nil {
 			log.Printf("使用GPT-4o估算成功: 模型=%s, 估算tokens=%d", originalModel, estimatedResult.InputTokens)
-			c.JSON(http.StatusOK, gin.H{
+			c.JSON(http.StatusBadRequest, gin.H{
 				"input_tokens":   estimatedResult.InputTokens,
 				"warning":        fmt.Sprintf("The tokenizer for model '%s' is not supported yet. This is an estimation based on gpt-4o and may not be accurate.", originalModel),
 				"estimated_with": "gpt-4o",
+				"error":          fmt.Sprintf("Unsupported model: %s", originalModel),
 			})
 		} else {
 			log.Printf("使用GPT-4o估算失败: %v", estimateErr)
@@ -593,12 +597,12 @@ func countTokens(c *gin.Context) {
 		if estimateErr == nil {
 			log.Printf("使用GPT-4o估算成功: 模型=%s, 估算tokens=%d", originalModel, estimatedResult.InputTokens)
 
-			// 返回估算值，但添加警告信息和原始错误
-			c.JSON(http.StatusOK, gin.H{
+			// 返回估算值，但添加警告信息和原始错误，使用400状态码
+			c.JSON(http.StatusBadRequest, 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(),
+				"error":          err.Error(),
 			})
 			return
 		} else {