Update handlers.go

handlers.go

@@ -153,8 +153,11 @@ func MessagesHandler(c *gin.Context) {
 	if cfg.OpenAIAPIKey != "" {
 		req.Header.Set("Authorization", "Bearer "+cfg.OpenAIAPIKey)
 	}
-	// TODO: Consider adding Anthropic-Version header if cfg.UpstreamNativeClaudeFormat is true and if needed by the upstream.
-	// Example: if cfg.UpstreamNativeClaudeFormat { req.Header.Set("anthropic-version", "2023-06-01") }
+	if cfg.UpstreamNativeClaudeFormat {
+		// Add Anthropic-Version header if upstream is native Claude and it's required
+		// Example: req.Header.Set("anthropic-version", "2023-06-01")
+		// For now, this is commented out as it depends on specific upstream requirements.
+	}
 
 	// Giữ lại: Log INFO quan trọng về việc gửi yêu cầu upstream
 	log.Printf("INFO: [%s] Gửi yêu cầu upstream (Stream=%t) đến %s...", requestID, isStreaming, maskURL(upstreamURL))
@@ -210,15 +213,13 @@ func MessagesHandler(c *gin.Context) {
 	if isStreaming {
 		if cfg.UpstreamNativeClaudeFormat {
 			// Giữ lại: Log INFO quan trọng
-			log.Printf("INFO: [%s] Upstream Claude native stream nhận được. Bắt đầu proxy SSE trực tiếp.", requestID)
-			proxyClaudeNativeSSE(c, upstreamResp, requestID)
+			log.Printf("INFO: [%s] Upstream Claude native stream nhận được. Bắt đầu proxy SSE (với hack usage).", requestID)
+			proxyClaudeNativeSSE(c, upstreamResp, requestID, &claudeReq) // SỬA Ở ĐÂY: Thêm &claudeReq
 		} else {
 			// Giữ lại: Log INFO quan trọng
 			log.Printf("INFO: [%s] Upstream OpenAI stream nhận được. Bắt đầu chuyển đổi SSE.", requestID)
-			// Use claudeReq.Model as the source of truth for the model name in conversion,
-			// openAIReqForLogging might be nil if native format.
 			modelForConversion := claudeReq.Model
-			if openAIReqForLogging != nil && openAIReqForLogging.Model != "" { // Prefer model from actual OpenAI request if available
+			if openAIReqForLogging != nil && openAIReqForLogging.Model != "" {
 				modelForConversion = openAIReqForLogging.Model
 			}
 			streamOpenAIResponseToClaudeSSE(c, upstreamResp, requestID, modelForConversion, &claudeReq)
@@ -239,18 +240,16 @@ func MessagesHandler(c *gin.Context) {
 		if cfg.UpstreamNativeClaudeFormat {
 			// Giữ lại: Log INFO
 			log.Printf("INFO: [%s] Upstream Claude native non-stream response nhận được. Chuyển tiếp trực tiếp.", requestID)
-			// Validate if it's actually JSON before sending
-			var tempClaudeResp map[string]interface{} // Use map for validation, not full struct
+			var tempClaudeResp map[string]interface{}
 			if errJson := json.Unmarshal(respBodyBytes, &tempClaudeResp); errJson != nil {
 				// Giữ lại: Log lỗi quan trọng
 				log.Printf("ERROR: [%s] Upstream Claude native response không phải JSON hợp lệ: %v. Body: %s", requestID, errJson, string(respBodyBytes))
 				sendClaudeError(c, http.StatusBadGateway, "api_error", "API upstream (Claude native) trả về JSON không hợp lệ.")
 				return
 			}
-			// Send the original Claude response bytes
 			clientContentType := upstreamResp.Header.Get("Content-Type")
 			if clientContentType == "" {
-				clientContentType = "application/json; charset=utf-8" // Default
+				clientContentType = "application/json; charset=utf-8"
 			}
 			c.Data(http.StatusOK, clientContentType, respBodyBytes)
 		} else {