vertexai

package

v0.4.7 Latest Latest Go to latest Published: Nov 24, 2023 License: MIT Imports: 4 Imported by: 0

Details

Valid go.mod file

The Go module system was introduced in Go 1.11 and is the official dependency management solution for Go.
Redistributable license

Redistributable licenses place minimal restrictions on how software can be used, modified, and redistributed.
Tagged version

Modules with tagged versions give importers more predictable builds.
Stable version

When a project reaches major version v1 it is considered stable.
Learn more about best practices

This section is empty.

This section is empty.

This section is empty.

type Option func(p *VertexAIPaLM)

Option is a function type that can be used to modify the client.

func WithBatchSize(batchSize int) Option

WithBatchSize is an option for specifying the batch size.

func WithClient(client vertexai.LLM) Option

WithClient is an option for providing the LLM client.

func WithStripNewLines(stripNewLines bool) Option

WithStripNewLines is an option for specifying the should it strip new lines.

type VertexAIPaLM struct {
	StripNewLines bool
	BatchSize     int
	// contains filtered or unexported fields
}

VertexAIPaLM is the embedder using the Google PaLM api.

func NewVertexAIPaLM(opts ...Option) (*VertexAIPaLM, error)

NewVertexAIPaLM creates a new VertexAI with options. Options for client, strip new lines and batch size.

func (e VertexAIPaLM) EmbedDocuments(ctx context.Context, texts []string) ([][]float64, error)

EmbedDocuments creates one vector embedding for each of the texts.

func (e VertexAIPaLM) EmbedQuery(ctx context.Context, text string) ([]float64, error)

EmbedQuery embeds a single text.

Path	Synopsis
vertexaichat