bedrock

Documentation

Index

• Constants
• type Client
  • func New(opts ...Option) (*Client, error)
  • func (c *Client) ConsumedTokens() int
  • func (c *Client) Embedding(ctx context.Context, text string) ([]float32, error)
• type FoundationModel
  • func NewFoundationModel(scope constructs.Construct, id *string, ...) *FoundationModel
  • func NewTitanTextEmbeddingsV1(scope constructs.Construct) *FoundationModel
  • func NewTitanTextEmbeddingsV2(scope constructs.Construct) *FoundationModel
  • func (c *FoundationModel) GrantAccess(grantee awsiam.IGrantable)
• type ModelID
• type Option
  • func WithConfig(cfg aws.Config) Option
  • func WithDimension(d int) Option
  • func WithModel(id ModelID) Option
  • func WithTitanV1() Option
  • func WithTitanV2() Option

Constants

const (
	TITAN_EMBED_TEXT_V1 = ModelID("amazon.titan-embed-text-v1")
	TITAN_EMBED_TEXT_V2 = ModelID("amazon.titan-embed-text-v2:0")
)

See https://docs.aws.amazon.com/bedrock/latest/userguide/model-ids.html

const (
	EMBEDDING_SIZE_256  = 256
	EMBEDDING_SIZE_512  = 512
	EMBEDDING_SIZE_1024 = 1024
)

const Version = "bedrock/v0.1.2"

Types

type Client

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

func New

func New(opts ...Option) (*Client, error)

Creates AWS BedRock embeddings client.

By default `us-east-1` region is used, supply custom `aws.Config` to alter behavior.

The client is configurable using

WithConfig(cfg aws.Config)

func (*Client) ConsumedTokens

func (c *Client) ConsumedTokens() int

Number of tokens consumed within the session

func (*Client) Embedding

func (c *Client) Embedding(ctx context.Context, text string) ([]float32, error)

Calculates embedding vector

type FoundationModel

type FoundationModel struct {
	constructs.Construct
	// contains filtered or unexported fields
}

Foundation Model L3 construct simplify grant access

func NewFoundationModel

func NewFoundationModel(scope constructs.Construct, id *string, foundationModelId awsbedrock.FoundationModelIdentifier) *FoundationModel

func NewTitanTextEmbeddingsV1

func NewTitanTextEmbeddingsV1(scope constructs.Construct) *FoundationModel

func NewTitanTextEmbeddingsV2

func NewTitanTextEmbeddingsV2(scope constructs.Construct) *FoundationModel

func (*FoundationModel) GrantAccess

func (c *FoundationModel) GrantAccess(grantee awsiam.IGrantable)

type ModelID

type ModelID string

type Option

type Option func(*Client)

func WithConfig

func WithConfig(cfg aws.Config) Option

func WithDimension

func WithDimension(d int) Option

func WithModel

func WithModel(id ModelID) Option

func WithTitanV1

func WithTitanV1() Option

func WithTitanV2

func WithTitanV2() Option