huggingface

package
v0.1.3 Latest Latest
Warning

This package is not in the latest version of its module.

 Go to latest Published: Jan 4, 2024 License: MIT Imports: 6 Imported by: 20

Documentation

Index

Constants

This section is empty.

Variables

View Source 
var (
	ErrEmptyResponse            = errors.New("empty response")
	ErrMissingToken             = errors.New("missing the Hugging Face API token. Set it in the HUGGINGFACEHUB_API_TOKEN environment variable") //nolint:lll
	ErrUnexpectedResponseLength = errors.New("unexpected length of response")
)

Functions

This section is empty.

Types

type LLM 

type LLM struct {
	CallbacksHandler callbacks.Handler
	// contains filtered or unexported fields
}

func New 

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

func (*LLM) Call 

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

func (*LLM) CreateEmbedding 

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

CreateEmbedding creates embeddings for the given input texts.

func (*LLM) Generate 

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

type Option 

type Option func(*options)

func WithModel 

func WithModel(model string) Option

WithModel passes the HuggingFace model to the client. If not set, then will be used default model.

func WithToken 

func WithToken(token string) Option

WithToken passes the HuggingFace API token to the client. If not set, the token is read from the HUGGINGFACEHUB_API_TOKEN environment variable.

Source Files

View all Source files

Directories

Path Synopsis
internal
huggingfaceclient

Jump to

Keyboard shortcuts

? : This menu
/ : Search site
f or F : Jump to
y or Y : Canonical URL
go.dev uses cookies from Google to deliver and enhance the quality of its services and to analyze traffic. Learn more.