openai

type APIKey string

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

type ChatCompletion struct {
	ID                string         `json:"id"`
	Object            string         `json:"object"`
	Created           int            `json:"created"`
	Model             string         `json:"model"`
	ResponseFormat    ResponseFormat `json:"response_format"`
	SystemFingerprint string         `json:"system_fingerprint"`
	Choices           []struct {
		FinishReason string      `json:"finish_reason"`
		Index        int         `json:"index"`
		Message      ChatMessage `json:"message"`
	} `json:"choices"`
	Usage struct {
		PromptTokens     int `json:"prompt_tokens"`
		CompletionTokens int `json:"completion_tokens"`
		TotalTokens      int `json:"total_tokens"`
	} `json:"usage"`
}

type ChatMessage struct {
	Role    string `json:"role"`
	Content string `json:"content"`
}

type CreateChatCompletion struct {
	Messages         []ChatMessage      `json:"messages"`
	Model            string             `json:"model"`
	ResponseFormat   *ResponseFormat    `json:"response_format,omitempty"`
	MaxTokens        *int               `json:"max_tokens,omitempty"`
	N                *int               `json:"n,omitempty"`
	Temperature      *float64           `json:"temperature,omitempty"`
	Seed             *int               `json:"seed,omitempty"`
	TopP             *float64           `json:"top_p,omitempty"`
	Stop             []string           `json:"stop,omitempty"`
	FrequencyPenalty *float64           `json:"frequency_penalty,omitempty"`
	LogitBias        map[string]float64 `json:"logit_bias,omitempty"`
	User             *string            `json:"user,omitempty"`
	PresencePenalty  *float64           `json:"presence_penalty,omitempty"`
}

type ErrorDetail struct {
	Message string      `json:"message"`
	Type    string      `json:"type"`
	Param   interface{} `json:"param"` // Param can be of any type, hence using interface{}.
	Code    string      `json:"code"`
}

type ErrorResponse struct {
	Error ErrorDetail `json:"error"`
}

type GenerativeAIClient interface {
	RequestCreateChatCompletion(context.Context, *CreateChatCompletion) (*ChatCompletion, error)
	MakeCreateChatCompletion(string) *CreateChatCompletion
}

type Pricing struct {
	Model                  string
	InputTokensCostDollar  float64
	InputTokens            float64
	OutputTokensCostDollar float64
	OutputTokens           float64
}

type ResponseFormat struct {
	Type string `json:"type"`
}

type TotalUsageCost struct {
	UsageCosts []*UsageCost
}

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