googai

Documentation

Index

• type Client
  • func NewClient(ctx context.Context) (*Client, error)
  • func (cl *Client) Embed(ctx context.Context, req *EmbedRequest, rsp *EmbedResponse) error
  • func (cl *Client) EmbedSingle(ctx context.Context, text string, model llm.ServiceName) (llm.Embedding, int64, error)
• type EmbedRequest
  • func (req *EmbedRequest) Validate() error
• type EmbedResponse

Types

type Client

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

func NewClient

func NewClient(ctx context.Context) (*Client, error)

func (*Client) Embed

func (cl *Client) Embed(ctx context.Context, req *EmbedRequest, rsp *EmbedResponse) error

Embed generates embeddings for one or more documents in a single request

func (*Client) EmbedSingle

func (cl *Client) EmbedSingle(ctx context.Context, text string, model llm.ServiceName) (llm.Embedding, int64, error)

EmbedSingle is a convenience method for single document embedding

type EmbedRequest

type EmbedRequest struct {
	Documents []string
	Model     llm.ServiceName
}

func (*EmbedRequest) Validate

func (req *EmbedRequest) Validate() error

type EmbedResponse

type EmbedResponse struct {
	Embeddings             []llm.Embedding
	BillableCharacterCount int64
}