huggingface

Documentation

Index

• type Completer
  • func NewCompleter(url, model string, options ...Option) (*Completer, error)
• type Config
• type Embedder
  • func NewEmbedder(url, model string, options ...Option) (*Embedder, error)
  • func (e *Embedder) Embed(ctx context.Context, texts []string) (*provider.Embedding, error)
• type Option
  • func WithClient(client *http.Client) Option
  • func WithToken(token string) Option
• type Reranker
  • func NewReranker(url, model string, options ...Option) (*Reranker, error)
  • func (r *Reranker) Rerank(ctx context.Context, query string, texts []string, ...) ([]provider.Ranking, error)
• type Result

Types

type Completer

type Completer = openai.Completer

func NewCompleter

func NewCompleter(url, model string, options ...Option) (*Completer, error)

type Config

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

type Embedder

type Embedder struct {
	*Config
}

func NewEmbedder

func NewEmbedder(url, model string, options ...Option) (*Embedder, error)

func (*Embedder) Embed

func (e *Embedder) Embed(ctx context.Context, texts []string) (*provider.Embedding, error)

type Option

type Option func(*Config)

func WithClient

func WithClient(client *http.Client) Option

func WithToken

func WithToken(token string) Option

type Reranker

type Reranker struct {
	*Config
}

func NewReranker

func NewReranker(url, model string, options ...Option) (*Reranker, error)

func (*Reranker) Rerank

func (r *Reranker) Rerank(ctx context.Context, query string, texts []string, options *provider.RerankOptions) ([]provider.Ranking, error)

type Result

type Result struct {
	Index int     `json:"index"`
	Score float64 `json:"score"`
}