Explorar el Código

feat: aws claude tools

1808837298@qq.com hace 1 año
padre
commit
65e65097b2
Se han modificado 1 ficheros con 5 adiciones y 1 borrados
  1. 5 1
      relay/channel/aws/dto.go

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

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