bedrock

Documentation

Index

• func Invoke(p string) (string, error)
• func InvokeClaude3(p string, model string, system string) (string, error)
• type Content
• type Message
• type MessageResponse
• type MessageUsage
• type Request
• type Response
• type Source

Functions

func Invoke

func Invoke(p string) (string, error)

Invoke invokes the Claude V2 model with the provided prompt.

func InvokeClaude3

func InvokeClaude3(p string, model string, system string) (string, error)

InvokeClaude3 invokes the Claude V3 model with the provided prompt.

Types

type Content

type Content struct {
	Type    string    `json:"type"`
	Sources []*Source `json:"source,omitempty"`
	Text    string    `json:"text,omitempty"`
}

type Message

type Message struct {
	Role     string     `json:"role"`
	Contents []*Content `json:"content"`
}

type MessageResponse

type MessageResponse struct {
	Id           string        `json:"id"`
	Type         string        `json:"type"`
	Role         string        `json:"role"`
	Contents     []*Content    `json:"content"`
	Model        string        `json:"model"`
	StopReason   string        `json:"stop_reason"`
	StopSequence string        `json:"stop_sequence"`
	Usage        *MessageUsage `json:"usage"`
}

type MessageUsage

type MessageUsage struct {
	InputTokens  int `json:"input_tokens"`
	OutputTokens int `json:"output_tokens"`
}

type Request

type Request struct {
	Prompt            string     `json:"prompt,omitempty"`
	System            string     `json:"system,omitempty"`
	Messages          []*Message `json:"messages,omitempty"`
	MaxTokensToSample int        `json:"max_tokens_to_sample,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"`
	AnthropicVersion  string     `json:"anthropic_version,omitempty"`
}

type Response

type Response struct {
	Completion string `json:"completion"`
}

type Source

type Source struct {
	Type      string `json:"type"`
	MediaType string `json:"media_type"`
	Data      string `json:"data"`
}