openai

Documentation

Index

• Constants
• func ListModels() []string
• type Client
  • func NewClient(cfg openai.ClientConfig) *Client
  • func (s *Client) CreateChatCompletion(ctx context.Context, req llm.ChatCompletionRequest) (llm.ChatCompletionResponse, error)
  • func (s *Client) CreateChatCompletionStream(ctx context.Context, req llm.ChatCompletionRequest, ...)
  • func (s *Client) ListModels() []string
• type OpenAI
  • func New(cfg openai.ClientConfig, dao llm.Dao) *OpenAI

Constants

const (
	ModelGPT432K          = "gpt-4-32k"
	ModelGPT4             = "gpt-4"
	ModelGPT3Dot5Turbo16K = "gpt-3.5-turbo-16k"
	ModelGPT3Dot5Turbo    = "gpt-3.5-turbo"
)

Functions

func ListModels

func ListModels() []string

Types

type Client

type Client struct {
	*openai.Client
}

func NewClient

func NewClient(cfg openai.ClientConfig) *Client

func (*Client) CreateChatCompletion

func (s *Client) CreateChatCompletion(ctx context.Context, req llm.ChatCompletionRequest) (llm.ChatCompletionResponse, error)

func (*Client) CreateChatCompletionStream

func (s *Client) CreateChatCompletionStream(ctx context.Context, req llm.ChatCompletionRequest, dataChan chan llm.ChatCompletionStreamResponse, errChan chan error)

func (*Client) ListModels

func (s *Client) ListModels() []string

type OpenAI

type OpenAI struct {
	*llm.LLM
}

func New

func New(cfg openai.ClientConfig, dao llm.Dao) *OpenAI