vertexai

Dec 13, 2023

Changes in this version

+ var ErrEmptyResponse = errors.New("no response")

+ var ErrMissingProjectID = errors.New(...)

+ var ErrNotImplemented = errors.New("not implemented")

+ var ErrUnexpectedResponseLength = errors.New("unexpected length of response")

+ type Chat struct

+ func NewChat(opts ...Option) (*Chat, error)

+ func (o *Chat) Call(ctx context.Context, messages []schema.ChatMessage, options ...llms.CallOption) (*schema.AIChatMessage, error)

+ func (o *Chat) CreateEmbedding(ctx context.Context, inputTexts []string) ([][]float32, error)

+ func (o *Chat) Generate(ctx context.Context, messageSets [][]schema.ChatMessage, ...) ([]*llms.Generation, error)

+ func (o *Chat) GeneratePrompt(ctx context.Context, promptValues []schema.PromptValue, ...) (llms.LLMResult, error)

+ func (o *Chat) GetNumTokens(text string) int

+ type ChatMessage = vertexaiclient.ChatMessage

+ type LLM struct

+ CallbacksHandler callbacks.Handler

+ func New(opts ...Option) (*LLM, error)

+ func (o *LLM) Call(ctx context.Context, prompt string, options ...llms.CallOption) (string, error)

+ func (o *LLM) CreateEmbedding(ctx context.Context, inputTexts []string) ([][]float32, error)

+ func (o *LLM) Generate(ctx context.Context, prompts []string, options ...llms.CallOption) ([]*llms.Generation, error)

+ func (o *LLM) GeneratePrompt(ctx context.Context, promptValues []schema.PromptValue, ...) (llms.LLMResult, error)

+ func (o *LLM) GetNumTokens(text string) int

+ type Option func(*options)

+ func WithAPIKey(apiKey string) Option

+ func WithCredentialsFile(path string) Option

+ func WithCredentialsJSON(json []byte) Option

+ func WithGRPCDialOption(opt grpc.DialOption) Option

+ func WithHTTPClient(client *http.Client) Option

+ func WithProjectID(projectID string) Option