vertexaichat

Documentation

Index

• type ChatOption
  • func WithBatchSize(batchSize int) ChatOption
  • func WithClient(client vertexai.Chat) ChatOption
  • func WithStripNewLines(stripNewLines bool) ChatOption
• type ChatVertexAI
  • func NewChatVertexAI(opts ...ChatOption) (ChatVertexAI, error)
  • func (e ChatVertexAI) EmbedDocuments(ctx context.Context, texts []string) ([][]float64, error)
  • func (e ChatVertexAI) EmbedQuery(ctx context.Context, text string) ([]float64, error)

Types

type ChatOption

type ChatOption func(p *ChatVertexAI)

func WithBatchSize

func WithBatchSize(batchSize int) ChatOption

WithBatchSize is an option for specifying the batch size.

func WithClient

func WithClient(client vertexai.Chat) ChatOption

WithClient is an option for providing the LLM client.

func WithStripNewLines

func WithStripNewLines(stripNewLines bool) ChatOption

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

type ChatVertexAI

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

ChatVertexAI is the embedder using the VertexAI api.

func NewChatVertexAI

func NewChatVertexAI(opts ...ChatOption) (ChatVertexAI, error)

NewChatVertexAI creates a new ChatVertexAI with options. Options for client, strip new lines and batch.

func (ChatVertexAI) EmbedDocuments

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

func (ChatVertexAI) EmbedQuery

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