watsonx

Documentation

Index

• Variables
• type LLM
  • func New(modelID string, opts ...wx.ClientOption) (*LLM, error)
  • func (wx *LLM) Call(ctx context.Context, prompt string, options ...llms.CallOption) (string, error)
  • func (wx *LLM) GenerateContent(ctx context.Context, messages []llms.MessageContent, ...) (*llms.ContentResponse, error)

Variables

var (
	ErrInvalidPrompt = errors.New("invalid prompt")
	ErrEmptyResponse = errors.New("no response")
)

Types

type LLM

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

func New

func New(modelID string, opts ...wx.ClientOption) (*LLM, error)

func (*LLM) Call

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

Call implements the LLM interface.

func (*LLM) GenerateContent

func (wx *LLM) GenerateContent(ctx context.Context, messages []llms.MessageContent, options ...llms.CallOption) (*llms.ContentResponse, error)

GenerateContent implements the Model interface.