bedrock

Documentation

Index

• func NormalizeToolName(input string) string
• type Client
• type ModelProvider
  • func NewWithClient(client Client) *ModelProvider
  • func (p *ModelProvider) GenerateImage(ctx context.Context, req *estellm.GenerateImageRequest, ...) error
  • func (p *ModelProvider) GenerateText(ctx context.Context, req *estellm.GenerateTextRequest, ...) error
  • func (p *ModelProvider) SetClilent(client Client)
• type StableDiffusionXLArtifact
• type StableDiffusionXLJSONPrompt
• type StableDiffusionXLRequset
• type StableDiffusionXLResponse
• type StableDiffusionXLTextPrompt

Functions

func NormalizeToolName

func NormalizeToolName(input string) string

Types

type Client

type Client interface {
	InvokeModel(ctx context.Context, input *bedrockruntime.InvokeModelInput, optFns ...func(*bedrockruntime.Options)) (*bedrockruntime.InvokeModelOutput, error)
	ConverseStream(ctx context.Context, params *bedrockruntime.ConverseStreamInput, optFns ...func(*bedrockruntime.Options)) (*bedrockruntime.ConverseStreamOutput, error)
}

type ModelProvider

type ModelProvider struct {
	// contains filtered or unexported fields
}

func NewWithClient

func NewWithClient(client Client) *ModelProvider

func (*ModelProvider) GenerateImage

func (p *ModelProvider) GenerateImage(ctx context.Context, req *estellm.GenerateImageRequest, w estellm.ResponseWriter) error

func (*ModelProvider) GenerateText

func (p *ModelProvider) GenerateText(ctx context.Context, req *estellm.GenerateTextRequest, w estellm.ResponseWriter) error

func (*ModelProvider) SetClilent

func (p *ModelProvider) SetClilent(client Client)

type StableDiffusionXLArtifact

type StableDiffusionXLArtifact struct {
	Seed         int64  `json:"seed"`
	Base64       string `json:"base64"`
	FinishReason string `json:"finishReason"`
}

type StableDiffusionXLJSONPrompt

type StableDiffusionXLJSONPrompt struct {
	Prompt         string `json:"prompt"`
	NegativePrompt string `json:"negative_prompt"`
	StylePreset    string `json:"style_preset"`
}

type StableDiffusionXLRequset

type StableDiffusionXLRequset struct {
	TextPrompts []StableDiffusionXLTextPrompt `json:"text_prompts"`
	Height      int                           `json:"height,omitempty"`
	Width       int                           `json:"width,omitempty"`
	StylePreset string                        `json:"style_preset,omitempty"`
	Seed        int64                         `json:"seed,omitempty"`
	CFGScale    float64                       `json:"cfg_scale,omitempty"`
	Steps       int                           `json:"steps,omitempty"`
	Sampler     string                        `json:"sampler,omitempty"`
	Samples     int                           `json:"samples,omitempty"`
	Extra       json.RawMessage               `json:"extra,omitempty"`
}

https://docs.aws.amazon.com/ja_jp/bedrock/latest/userguide/model-parameters-diffusion-1-0-text-image.html

type StableDiffusionXLResponse

type StableDiffusionXLResponse struct {
	Result    string                      `json:"result"`
	Artifacts []StableDiffusionXLArtifact `json:"artifacts"`
}

type StableDiffusionXLTextPrompt

type StableDiffusionXLTextPrompt struct {
	Text   string  `json:"text"`
	Weight float64 `json:"weight"`
}