mirror of
https://github.com/linux-do/new-api.git
synced 2025-09-17 16:06:38 +08:00
35 lines
1.2 KiB
Go
35 lines
1.2 KiB
Go
package aws
|
|
|
|
import (
|
|
"one-api/relay/channel/claude"
|
|
)
|
|
|
|
type AwsClaudeRequest struct {
|
|
// AnthropicVersion should be "bedrock-2023-05-31"
|
|
AnthropicVersion string `json:"anthropic_version"`
|
|
System string `json:"system,omitempty"`
|
|
Messages []claude.ClaudeMessage `json:"messages"`
|
|
MaxTokens uint `json:"max_tokens,omitempty"`
|
|
Temperature float64 `json:"temperature,omitempty"`
|
|
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"`
|
|
}
|
|
|
|
func copyRequest(req *claude.ClaudeRequest) *AwsClaudeRequest {
|
|
return &AwsClaudeRequest{
|
|
AnthropicVersion: "bedrock-2023-05-31",
|
|
System: req.System,
|
|
Messages: req.Messages,
|
|
MaxTokens: req.MaxTokens,
|
|
Temperature: req.Temperature,
|
|
TopP: req.TopP,
|
|
TopK: req.TopK,
|
|
StopSequences: req.StopSequences,
|
|
Tools: req.Tools,
|
|
ToolChoice: req.ToolChoice,
|
|
}
|
|
}
|