openai

Documentation

Index

• Constants
• func EstimateCost(model string, promptTokens int, completionTokens int) float64
• type OpenAiClient
  • func NewOpenAiClient(apiCredential string) OpenAiClient
  • func (c OpenAiClient) Send(rc *config.OpenAiRouteConfig, prompts []*config.OpenAiPrompt, ...) (*OpenAiResponse, error)
• type OpenAiError
  • func NewOpenAiError(message string, errorType string, code int) *OpenAiError
  • func (e *OpenAiError) Error() string
  • func (e *OpenAiError) StatusCode() int
• type OpenAiErrorContent
• type OpenAiErrorResponse
• type OpenAiResponse

Constants

const (
	AuthorizationHeader string = "Authorization"
	ContentTypeHeader   string = "Content-Type"
)

Functions

func EstimateCost

func EstimateCost(model string, promptTokens int, completionTokens int) float64

Types

type OpenAiClient

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

func NewOpenAiClient

func NewOpenAiClient(apiCredential string) OpenAiClient

func (OpenAiClient) Send

func (c OpenAiClient) Send(rc *config.OpenAiRouteConfig, prompts []*config.OpenAiPrompt, lm *logger.LlmMessage) (*OpenAiResponse, error)

type OpenAiError

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

func NewOpenAiError

func NewOpenAiError(message string, errorType string, code int) *OpenAiError

func (*OpenAiError) Error

func (e *OpenAiError) Error() string

func (*OpenAiError) StatusCode

func (e *OpenAiError) StatusCode() int

type OpenAiErrorContent

type OpenAiErrorContent struct {
	Message string `json:"message"`
	Type    string `json:"type"`
}

type OpenAiErrorResponse

type OpenAiErrorResponse struct {
	Error *OpenAiErrorContent `json:"error"`
}

type OpenAiResponse

type OpenAiResponse struct {
	Id      string   `json:"id"`
	Object  string   `json:"object"`
	Created int64    `json:"created"`
	Choices []choice `json:"choices"`
	Usage   usage    `json:"usage"`
}