From 65e65097b2da3b81bf97b104413cba5a80f711f8 Mon Sep 17 00:00:00 2001
From: "1808837298@qq.com" <1808837298@qq.com>
Date: Thu, 24 Oct 2024 00:19:08 +0800
Subject: [PATCH] feat: aws claude tools

---
 relay/channel/aws/dto.go | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/relay/channel/aws/dto.go b/relay/channel/aws/dto.go
index 7a267b1..5bf3551 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"`
 }