mirror of
https://github.com/songquanpeng/one-api.git
synced 2025-09-17 17:16:38 +08:00
25 lines
1.0 KiB
Go
25 lines
1.0 KiB
Go
package aws
|
|
|
|
import (
|
|
"github.com/songquanpeng/one-api/relay/adaptor/anthropic"
|
|
"github.com/songquanpeng/one-api/relay/model"
|
|
)
|
|
|
|
// 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"`
|
|
Thinking *model.Thinking `json:"thinking,omitempty"`
|
|
}
|