openai

Documentation

Overview

Package openai provides a wrapper around the OpenAI API.

Index

• Constants
• type Model
• type OpenAI
  • func New(model Model, temperature float32, maxTokens int, verbose bool) (*OpenAI, error)
  • func (o *OpenAI) Chat(prompt *chat.Chat) (string, error)
  • func (o *OpenAI) Completion(prompt string) (string, error)

Constants

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

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 OpenAI

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

func New

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

func (*OpenAI) Chat

func (o *OpenAI) Chat(prompt *chat.Chat) (string, error)

func (*OpenAI) Completion

func (o *OpenAI) Completion(prompt string) (string, error)