openai

Documentation

Overview

Package openai provides a wrapper around the OpenAI API.

Index

• Constants
• Variables
• func New(model Model, temperature float32, maxTokens int, verbose bool) (*openAI, error)
• type Model
• type OpenAICallback

Constants

const (
	DefaultOpenAIMaxTokens   = 256
	DefaultOpenAITemperature = 0.7
	DefaultOpenAINumResults  = 1
	DefaultOpenAITopP        = 1.0
)

Variables

var (
	ErrOpenAICompletion = fmt.Errorf("openai completion error")
	ErrOpenAIChat       = fmt.Errorf("openai chat error")
)

Functions

func New

func New(model Model, temperature float32, maxTokens int, verbose bool) (*openAI, error)

Types

type Model

type Model string

const (
	GPT4               Model = openai.GPT4
	GPT3Dot5Turbo      Model = openai.GPT3Dot5Turbo
	GPT3TextDavinci003 Model = openai.GPT3TextDavinci003
	GPT3TextDavinci002 Model = openai.GPT3TextDavinci002
	GPT3TextCurie001   Model = openai.GPT3TextCurie001
	GPT3TextBabbage001 Model = openai.GPT3TextBabbage001
	GPT3TextAda001     Model = openai.GPT3TextAda001
	GPT3TextDavinci001 Model = openai.GPT3TextDavinci001
	GPT3Davinci        Model = openai.GPT3Davinci
	GPT3Curie          Model = openai.GPT3Curie
	GPT3Ada            Model = openai.GPT3Ada
	GPT3Babbage        Model = openai.GPT3Babbage
)

type OpenAICallback

type OpenAICallback func(types.Meta)