llm

Documentation

Index

• Variables
• func New(ctx context.Context, config Config) (llms.Model, error)
• func Query(ctx context.Context, model llms.Model, query string, extraContext string) (string, error)
• func QueryForReference(ctx context.Context, model llms.Model, linterOutput string, ...) (string, error)
• type Config

Variables

var ErrAPIKeyRequired = errors.New("API key is required")

var ErrModelIsNil = errors.New("model is nil")

var ErrModelRequired = errors.New("model is required")

var ErrServerURLRequired = errors.New("server URL is required")

var ErrUnsupportedProvider = errors.New("unsupported llm provider")

Functions

func New

func New(ctx context.Context, config Config) (llms.Model, error)

func Query

func Query(ctx context.Context, model llms.Model, query string, extraContext string) (string, error)

func QueryForReference

func QueryForReference(ctx context.Context, model llms.Model, linterOutput string, codeLanguage string) (string, error)

Types

type Config

type Config struct {
	Provider  string
	APIKey    string
	Model     string
	ServerURL string
}