openai

Documentation

Index

• type ChatOpenAI
  • func New(apiKey string) (*ChatOpenAI, error)
• type Options

Types

type ChatOpenAI

type ChatOpenAI struct {
	*chatmodel.ChatModel
	// contains filtered or unexported fields
}

func New

func New(apiKey string) (*ChatOpenAI, error)

type Options

type Options struct {
	// Model name to use.
	Model string
	// Sampling temperature to use.
	Temperatur float32
	// The maximum number of tokens to generate in the completion.
	// -1 returns as many tokens as possible given the prompt and
	//the models maximal context size.
	MaxTokens int
	// Total probability mass of tokens to consider at each step.
	TopP float32
	// Penalizes repeated tokens.
	PresencePenalty float32
	// Penalizes repeated tokens according to frequency.
	FrequencyPenalty float32
	// How many completions to generate for each prompt.
	N int
	// Batch size to use when passing multiple documents to generate.
	BatchSize int
}