openai

Documentation

Index

• type APIError
  • func AsAPIError(err error) *APIError
  • func (e *APIError) Error() string
  • func (e *APIError) IsOverloaded() bool
  • func (e *APIError) Retryable() bool
• type Client
  • func New(options Options) *Client
  • func (c *Client) Embed(ctx context.Context, model string, texts []string) ([][]float64, error)
• type Options
• type RetryConfig
  • func DefaultRetryConfig() RetryConfig
  • func NoRetry() RetryConfig

Types

type APIError

type APIError struct {
	Status     int
	Type       string
	Code       string
	Message    string
	RetryAfter time.Duration
}

func AsAPIError

func AsAPIError(err error) *APIError

func (*APIError) Error

func (e *APIError) Error() string

func (*APIError) IsOverloaded

func (e *APIError) IsOverloaded() bool

func (*APIError) Retryable

func (e *APIError) Retryable() bool

type Client

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

func New

func New(options Options) *Client

func (*Client) Embed

func (c *Client) Embed(ctx context.Context, model string, texts []string) ([][]float64, error)

type Options

type Options struct {
	APIKey     string
	BaseURL    string
	Dimensions int
	HTTPClient *http.Client
	Retry      *RetryConfig

	Now   func() time.Time
	Sleep func(context.Context, time.Duration) error
}

type RetryConfig

type RetryConfig struct {
	MaxAttempts    int
	BaseDelay      time.Duration
	OverloadedBase time.Duration
	MaxDelay       time.Duration
	MaxElapsed     time.Duration
	Jitter         float64
}

func DefaultRetryConfig

func DefaultRetryConfig() RetryConfig

func NoRetry

func NoRetry() RetryConfig