Parcourir la source

Merge pull request #2226 from QuantumNous/omit-anthropic_beta-empty

fix(relay/channel/aws): 修复AnthropicBeta字段的omitempty处理
Seefs il y a 3 mois
Parent
commit
35192e5675
1 fichiers modifiés avec 5 ajouts et 2 suppressions
  1. 5 2
      relay/channel/aws/dto.go

+ 5 - 2
relay/channel/aws/dto.go

@@ -12,7 +12,7 @@ import (
 type AwsClaudeRequest struct {
 type AwsClaudeRequest struct {
 	// AnthropicVersion should be "bedrock-2023-05-31"
 	// AnthropicVersion should be "bedrock-2023-05-31"
 	AnthropicVersion string              `json:"anthropic_version"`
 	AnthropicVersion string              `json:"anthropic_version"`
-	AnthropicBeta    json.RawMessage     `json:"anthropic_beta"`
+	AnthropicBeta    json.RawMessage     `json:"anthropic_beta,omitempty"`
 	System           any                 `json:"system,omitempty"`
 	System           any                 `json:"system,omitempty"`
 	Messages         []dto.ClaudeMessage `json:"messages"`
 	Messages         []dto.ClaudeMessage `json:"messages"`
 	MaxTokens        uint                `json:"max_tokens,omitempty"`
 	MaxTokens        uint                `json:"max_tokens,omitempty"`
@@ -40,7 +40,10 @@ func formatRequest(requestBody io.Reader, requestHeader http.Header) (*AwsClaude
 		if err != nil {
 		if err != nil {
 			return nil, err
 			return nil, err
 		}
 		}
-		awsClaudeRequest.AnthropicBeta = json.RawMessage(betaJson)
+		var tempArray []string
+		if err := json.Unmarshal(betaJson, &tempArray); err == nil && len(tempArray) != 0 && len(betaJson) > 0 {
+			awsClaudeRequest.AnthropicBeta = json.RawMessage(betaJson)
+		}
 	}
 	}
 	return &awsClaudeRequest, nil
 	return &awsClaudeRequest, nil
 }
 }