diff --git a/relay/channel/aws/dto.go b/relay/channel/aws/dto.go index 7a267b1f..5bf35518 100644 --- a/relay/channel/aws/dto.go +++ b/relay/channel/aws/dto.go @@ -1,6 +1,8 @@ package aws -import "one-api/relay/channel/claude" +import ( + "one-api/relay/channel/claude" +) type AwsClaudeRequest struct { // AnthropicVersion should be "bedrock-2023-05-31" @@ -12,4 +14,6 @@ type AwsClaudeRequest struct { TopP float64 `json:"top_p,omitempty"` TopK int `json:"top_k,omitempty"` StopSequences []string `json:"stop_sequences,omitempty"` + Tools []claude.Tool `json:"tools,omitempty"` + ToolChoice any `json:"tool_choice,omitempty"` }