gemini

Documentation

Overview

Package gemini implements access to Google's Gemini model.

Client implements llm.Embedder. Use NewClient to connect.

Index

• func Scrub(req *http.Request) error
• type Client
  • func NewClient(ctx context.Context, lg *slog.Logger, sdb secret.DB, hc *http.Client) (*Client, error)
  • func (c *Client) EmbedDocs(ctx context.Context, docs []llm.EmbedDoc) ([]llm.Vector, error)

Functions

func Scrub

func Scrub(req *http.Request) error

Scrub is a request scrubber for use with rsc.io/httprr.

Types

type Client

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

A Client represents a connection to Gemini.

func NewClient

func NewClient(ctx context.Context, lg *slog.Logger, sdb secret.DB, hc *http.Client) (*Client, error)

NewClient returns a connection to Gemini, using the given logger and HTTP client. It expects to find a secret of the form "AIza..." or "user:AIza..." in sdb under the name "ai.google.dev".

func (*Client) EmbedDocs

func (c *Client) EmbedDocs(ctx context.Context, docs []llm.EmbedDoc) ([]llm.Vector, error)

EmbedDocs returns the vector embeddings for the docs, implementing llm.Embedder.