openai

type AudioRequestBody struct {
	File           string  `json:"file"`
	Model          string  `json:"model"`
	Prompt         string  `json:"prompt,omitempty"`
	ResponseFormat string  `json:"response_format,omitempty"`
	Temperature    float32 `json:"temperature,omitempty"`
	Language       string  `json:"language,omitempty"` // just create transcription
}

type AudioResponseBody struct {
	Text string `json:"text"`
}

type ChatChoice struct {
	Index        int         `json:"index"`
	Message      ChatMessage `json:"message"`
	FinishReason string      `json:"finish_reason"`
}

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

type ChatRequestBody struct {
	Model            string         `json:"model"`
	Messages         []ChatMessage  `json:"messages"`
	Temperature      float32        `json:"temperature,omitempty"`
	TopP             float32        `json:"top_p,omitempty"`
	N                int            `json:"n,omitempty"`
	Stream           bool           `json:"stream,omitempty"`
	Stop             string         `json:"stop,omitempty"`
	MaxTokens        int            `json:"max_tokens,omitempty"`
	PresencePenalty  float32        `json:"presence_penalty,omitempty"`
	FrequencyPenalty float32        `json:"frequency_penalty,omitempty"`
	LogitBias        map[string]int `json:"logit_bias,omitempty"`
	User             string         `json:"user,omitempty"`
}

type ChatResponseBody struct {
	Usage   TokensUsage  `json:"usage"`
	ID      string       `json:"id"`
	Object  string       `json:"object"`
	Created int          `json:"created"`
	Choices []ChatChoice `json:"choices"`
}

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

type CompletionChoice struct {
	Text         string `json:"text"`
	Index        int    `json:"index"`
	Logprobs     *int   `json:"logprobs"`
	FinishReason string `json:"finish_reason"`
}

type CompletionRequestBody struct {
	Model           string         `json:"model"`
	Prompt          string         `json:"prompt,omitempty"`
	Suffix          string         `json:"suffix,omitempty"`
	MaxTokens       int            `json:"max_tokens,omitempty"`
	Temperature     float32        `json:"temperature,omitempty"`
	TopP            float32        `json:"top_p,omitempty"`
	N               int            `json:"n,omitempty"`
	Stream          bool           `json:"stream,omitempty"`
	Logprobs        int            `json:"logprobs,omitempty"`
	Echo            bool           `json:"echo,omitempty"`
	Stop            string         `json:"stop,omitempty"`
	PresencePenalty float32        `json:"presence_penalty,omitempty"`
	BestOf          int            `json:"best_of,omitempty"`
	LogitBias       map[string]int `json:"logit_bias,omitempty"`
	User            string         `json:"user,omitempty"`
}

type CompletionResponseBody struct {
	ID      string             `json:"id"`
	Object  string             `json:"object"`
	Created int                `json:"created"`
	Model   string             `json:"model"`
	Choices []CompletionChoice `json:"choices"`
	Usage   TokensUsage        `json:"usage"`
}

type DeleteFileResponseBody struct {
	ID      string `json:"id"`
	Object  string `json:"object"`
	Deleted bool   `json:"deleted"`
}

type EditChoice struct {
	Text  string `json:"text"`
	Index int    `json:"index"`
}

type EditRequestBody struct {
	Model       string  `json:"model"`
	Input       string  `json:"input,omitempty"`
	Instruction string  `json:"instruction"`
	N           int     `json:"n,omitempty"`
	Temperature float32 `json:"temperature,omitempty"`
	TopP        float32 `json:"top_p,omitempty"`
}

type EditResponseBody struct {
	Usage   TokensUsage  `json:"usage"`
	Object  string       `json:"object"`
	Created int          `json:"created"`
	Choices []EditChoice `json:"choices"`
}

type EmbeddingsRequestBody struct {
	Model string `json:"model"`
	Input string `json:"input"`
	User  string `json:"user,omitempty"`
}

type EmbeddingsResponseBody struct {
	Object string      `json:"object"`
	Model  string      `json:"model"`
	Usage  TokensUsage `json:"usage"`
	Data   []struct {
		Object    string    `json:"object"`
		Embedding []float64 `json:"embedding"`
	} `json:"data"`
}

type Error struct {
	Code       *string `json:"code,omitempty"`
	Param      *string `json:"param,omitempty"`
	Message    string  `json:"message"`
	Type       string  `json:"type"`
	StatusCode int     `json:"-"`
}

type ErrorResponseBody struct {
	Error *Error `json:"error,omitempty"`
}

type FileObject struct {
	ID        string `json:"id"`
	Object    string `json:"object"`
	Bytes     int    `json:"bytes"`
	CreatedAt int    `json:"created_at"`
	FileName  string `json:"filename"`
	Purpose   string `json:"purpose"`
}

type ImageData struct {
	URL string `json:"url"`
}

type ImageEditRequestBody struct {
	// [Required]
	// The image to edit. Must be a valid PNG file, less than 4MB, and square.
	// If mask is not provided, image must have transparency, which will be used as the mask.
	Image string `json:"image" multipart:"image,file"`
	// [Optional]
	// An additional image whose fully transparent areas (e.g. where alpha is zero) indicate
	// where `image` should be edited. Must be a valid PNG file, less than 4MB, and have the
	// same dimensions as `image`.
	Mask string `json:"mask,omitempty" multipart:"mask,file"`
	// [Required]
	// A text description of the desired image(s).
	// The maximum length is 1000 characters.
	Prompt string `json:"prompt" multipart:"prompt"`
	// [Optional Defaults to 1]
	// The number of images to generate.
	// Must be between 1 and 10.
	N int `json:"n,omitempty" multipart:"n"`
	// [Optional Defaults to 1024x1024]
	// The size of the generated images.
	// Must be one of `256x256`, `512x512`, or `1024x1024`.
	Size string `json:"size,omitempty" multipart:"size"`
	// [Optional Defaults to url]
	// The format in which the generated images are returned.
	// Must be one of `url` or `b64_json`.
	ResponseFormat string `json:"response_format,omitempty" multipart:"response-format"`
	// [Optional]
	// A unique identifier representing your end-user, which can help OpenAI to monitor and detect abuse.
	User string `json:"user,omitempty" multipart:"user"`
}

type ImageRequestBody struct {
	// [Required]
	// A text description of the desired image(s).
	// The maximum length is 1000 characters.
	Prompt string `json:"prompt"`
	// [Optional Defaults to 1]
	// The number of images to generate.
	// Must be between 1 and 10.
	N int `json:"n,omitempty"`
	// [Optional Defaults to 1024x1024] The size of the generated images.
	//Must be one of `256x256`, `512x512`, or `1024x1024`.
	Size string `json:"size,omitempty"`
	// [Optional Defaults to url]
	// The format in which the generated images are returned.
	// Must be one of `url` or `b64_json`.
	ResponseFormat string `json:"response_format,omitempty"`
	// [Optional]
	// A unique identifier representing your end-user,
	// which can help OpenAI to monitor and detect abuse.
	User string `json:"user,omitempty"`
}

type ImageResponseBody struct {
	Created int         `json:"created"`
	Data    []ImageData `json:"data"`
}

type ImageVariationRequestBody struct {
	// [Required]
	// The image to use as the basic for the variation(s).
	// Must be a valid PNG file, less than 4MB, and square.
	Image          string `json:"image"`
	N              int    `json:"n"`
	Size           string `json:"size"`
	ResponseFormat string `json:"response_format"`
	User           string `json:"user"`
}

type ListFilesResponseBody struct {
	Object string       `json:"object"`
	Data   []FileObject `json:"data"`
}

type ModelObject struct {
	ID         string            `json:"id"`
	Object     string            `json:"object"`
	Created    int               `json:"created"`
	OwnerBy    string            `json:"owner_by"`
	Permission []ModelPermission `json:"permission"`
	Root       string            `json:"root"`
	Parent     interface{}       `json:"parent"`
}

type ModelPermission struct {
	ID                 string      `json:"id"`
	Object             string      `json:"object"`
	Created            int         `json:"created"`
	AllowCreateEngine  bool        `json:"allow_create_engine"`
	AllowSampling      bool        `json:"allow_sampling"`
	AllowLogprobs      bool        `json:"allow_logprobs"`
	AllowSearchIndices bool        `json:"allow_search_indices"`
	AllowView          bool        `json:"allow_view"`
	AllowFineTuning    bool        `json:"allow_fine_tuning"`
	Organization       string      `json:"organization"`
	Group              interface{} `json:"group"`
	IsBlocking         bool        `json:"is_blocking"`
}

type ModelsResponseBody struct {
	Object string        `json:"object"`
	Data   []ModelObject `json:"data"`
}

type ModerationObject struct {
	Hate            bool `json:"hate"`
	HateThreatening bool `json:"hate/threatening"`
	SelfHarm        bool `json:"self-harm"`
	Sexual          bool `json:"sexual"`
	SexualMinors    bool `json:"sexual/minors"`
	Violence        bool `json:"violence"`
	ViolenceGraphic bool `json:"violence/graphic"`
}

type ModerationRequestBody struct {
	Input string
	Model string `json:"model,omitempty"`
}

type ModerationResponseBody struct {
	ID      string             `json:"id"`
	Model   string             `json:"model"`
	Results []ModerationResult `json:"results"`
}

type ModerationResult struct {
	Categories     ModerationObject `json:"categories"`
	CategoryScores ModerationObject `json:"category_scores"`
	Flagged        bool             `json:"flagged"`
}

type RequestError struct {
	StatusCode int
}

type RetrieveFileContentResponseBody struct {
	Data []byte
}

type TokensUsage struct {
	PromptTokens     int `json:"prompt_tokens"`
	CompletionTokens int `json:"completion_tokens"`
	TotalTokens      int `json:"total_tokens"`
}

type UploadFileRequestBody struct {
	File    string `json:"file"`
	Purpose string `json:"purpose"`
}