720fe2dfeb
* feat: refactor AwsClaude to Aws to support both llama3 and claude * fix: aws llama3 ratio
21 lines
954 B
Go
21 lines
954 B
Go
package aws
|
|
|
|
import "github.com/songquanpeng/one-api/relay/adaptor/anthropic"
|
|
|
|
// Request is the request to AWS Claude
|
|
//
|
|
// https://docs.aws.amazon.com/bedrock/latest/userguide/model-parameters-anthropic-claude-messages.html
|
|
type Request struct {
|
|
// AnthropicVersion should be "bedrock-2023-05-31"
|
|
AnthropicVersion string `json:"anthropic_version"`
|
|
Messages []anthropic.Message `json:"messages"`
|
|
System string `json:"system,omitempty"`
|
|
MaxTokens int `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 []anthropic.Tool `json:"tools,omitempty"`
|
|
ToolChoice any `json:"tool_choice,omitempty"`
|
|
}
|