bedrock

Sep 12, 2024

Changes in this version

+ const CohereInputTypeQuery

+ const CohereInputTypeText

+ const ModelCohereEn

+ const ModelCohereMulti

+ const ModelTitanEmbedG1

+ func FetchAmazonTextEmbeddings(ctx context.Context, client *bedrockruntime.Client, modelID string, ...) ([][]float32, error)

+ func FetchCohereTextEmbeddings(ctx context.Context, client *bedrockruntime.Client, modelID string, ...) ([][]float32, error)

+ type Bedrock struct

+ BatchSize int

+ ModelID string

+ StripNewLines bool

+ func NewBedrock(opts ...Option) (*Bedrock, error)

+ func (b *Bedrock) EmbedDocuments(ctx context.Context, texts []string) ([][]float32, error)

+ func (b *Bedrock) EmbedQuery(ctx context.Context, text string) ([]float32, error)

+ type Option func(p *Bedrock)

+ func WithBatchSize(batchSize int) Option

+ func WithClient(client *bedrockruntime.Client) Option

+ func WithModel(model string) Option

+ func WithStripNewLines(stripNewLines bool) Option