client

Documentation

Index

• type Client
  • func New(apiKey string, opts ...Option) (Client, error)
• type CompleteResult
• type Option
  • func WithClientContext(clientContext int) Option
  • func WithModel(model string) Option
  • func WithRequestLogger(log log.Logger) Option
• type StreamResult

Types

type Client

type Client interface {
	Complete(ctx context.Context, latest []query.Message, content string) (*CompleteResult, error)
	Stream(ctx context.Context, latest []query.Message, content string) (*StreamResult, error)
	Update(opts ...Option)
}

func New

func New(apiKey string, opts ...Option) (Client, error)

type CompleteResult

type CompleteResult struct {
	Req      openai.ChatCompletionRequest
	Response openai.ChatCompletionResponse
}

type Option

type Option func(*client, *roundTripper)

func WithClientContext

func WithClientContext(clientContext int) Option

func WithModel

func WithModel(model string) Option

func WithRequestLogger

func WithRequestLogger(log log.Logger) Option

type StreamResult

type StreamResult struct {
	Req      openai.ChatCompletionRequest
	Response *openai.ChatCompletionStream
}