openai

Documentation

Index

• type LLM
  • func NewLLM(apiSecret string, model ModelType, opt *Option) (*LLM, error)
  • func (llm *LLM) CheckSupport(v llms.LLMSupportType) bool
  • func (llm *LLM) Request(ctx context.Context, input llms.Input, tb llms.ToolBox) (*llms.Output, error)
• type ModelType
• type Option
  • func NewOption() *Option
  • func (p *Option) SetHttpClient(val *http.Client) *Option
  • func (p *Option) SetMaxTokens(val int) *Option
  • func (p *Option) SetTemperature(val float32) *Option
  • func (p *Option) SetTopP(val float32) *Option

Types

type LLM

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

func NewLLM

func NewLLM(apiSecret string, model ModelType, opt *Option) (*LLM, error)

func (*LLM) CheckSupport

func (llm *LLM) CheckSupport(v llms.LLMSupportType) bool

func (*LLM) Request

func (llm *LLM) Request(ctx context.Context, input llms.Input, tb llms.ToolBox) (*llms.Output, error)

type ModelType

type ModelType string

const (
	ModelTypeGpt3Dot5Turbo ModelType = "gpt-3.5-turbo"
	ModelTypeGpt35Turbo16k ModelType = "gpt-35-turbo-16k"
	ModelTypeGpt4Pro       ModelType = "gpt-4"
	ModelTypeGpt4Pro32K    ModelType = "gpt-4-32k"
)

type Option

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

func NewOption

func NewOption() *Option

func (*Option) SetHttpClient

func (p *Option) SetHttpClient(val *http.Client) *Option

func (*Option) SetMaxTokens

func (p *Option) SetMaxTokens(val int) *Option

func (*Option) SetTemperature

func (p *Option) SetTemperature(val float32) *Option

func (*Option) SetTopP

func (p *Option) SetTopP(val float32) *Option