gemini

Documentation

Index

• type Option
  • func WithModel(model string) Option
• type Provider
  • func New(apiKey string, opts ...Option) (*Provider, error)
  • func (p *Provider) Capabilities() provider.Capabilities
  • func (p *Provider) GenerateEmbeddings(ctx context.Context, texts []string) ([][]float32, error)
  • func (p *Provider) GenerateStream(ctx context.Context, messages []llm.Message, opts ...provider.GenerateOption) (<-chan llm.Part, error)
  • func (p *Provider) ListModels(ctx context.Context) ([]string, error)

Types

type Option

type Option func(*Provider)

func WithModel

func WithModel(model string) Option

WithModel sets the target chat completion model.

type Provider

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

func New

func New(apiKey string, opts ...Option) (*Provider, error)

New constructs a Gemini compatible provider.

func (*Provider) Capabilities

func (p *Provider) Capabilities() provider.Capabilities

func (*Provider) GenerateEmbeddings

func (p *Provider) GenerateEmbeddings(ctx context.Context, texts []string) ([][]float32, error)

GenerateEmbeddings implements the Embedder interface for Gemini.

func (*Provider) GenerateStream

func (p *Provider) GenerateStream(ctx context.Context, messages []llm.Message, opts ...provider.GenerateOption) (<-chan llm.Part, error)

func (*Provider) ListModels

func (p *Provider) ListModels(ctx context.Context) ([]string, error)